import open3d as o3d
from open3d.visualization.gui import Application
from open3d.visualization import O3DVisualizer
import sys, os


# 假设 obj_file_path 是你的 OBJ 文件路径，image_save_path 是保存图片的路径
filePath = "G://obj_fix/print"
pid = "8160"
obj_file_path = f"{filePath}/{pid}/{pid}.obj"
image_save_path = f"{filePath}/{pid}/{pid}_test.png"



# import open3d as o3d

# 加载OBJ文件
mesh = o3d.io.read_triangle_mesh(obj_file_path)

# 创建一个OffscreenRenderer对象
renderer = o3d.visualization.rendering.OffscreenRenderer()

# 创建一个Open3DScene对象
scene = o3d.visualization.rendering.Open3DScene()

# 将几何体添加到场景中
scene.add_geometry(mesh)

# 创建一个材质并应用到几何体上
material = o3d.visualization.rendering.MaterialRecord()
material.base_color = [1.0, 1.0, 1.0]  # 白色材质
mesh.materials.append(material)

# 创建一个灯光对象
light = o3d.visualization.rendering.Light()
light.type = "directional"  # 方向光
light.direction = [0.0, -1.0, 0.0]  # 光照方向
light.color = [1.0, 1.0, 1.0]  # 白色光线

# 将灯光添加到场景中
scene.add_light(light)

# 设置渲染参数
renderer.set_scene(scene)

# 渲染场景
image = renderer.render()

# 显示渲染结果
o3d.geometry.image.axial_slice(image, 2, (128, 128, 128))

# # 计算顶点法线，这对于渲染是非常重要的
# mesh.compute_vertex_normals()

# # 可选：可以对网格进行进一步处理，例如滤波或下采样
# # mesh = o3d.geometry.filter_triangle_mesh(mesh, "default")

# # 创建一个Open3D的可视化对象
# vis = o3d.visualization.Visualizer()
# vis.create_window()

# # 将网格添加到可视化对象中
# vis.add_geometry(mesh)

# # 创建一个环境光源
# light = get_light_source("default")
# light.set_intensity([1.0, 1.0, 1.0])  # 设置光照强度

# # 将灯光添加到可视化对象中
# vis.add_light(light)

# # 调整灯光位置和方向
# light.set_position([0.0, 0.0, 0.0])
# light.set_direction([0.0, 0.0, 1.0])
# screenshot = vis.capture_screen()
# o3d.io.write_image("screenshot.png", screenshot)

# # 渲染并显示窗口
# vis.run()


# # 加载一个点云或者网格
# mesh = o3d.io.read_triangle_mesh(obj_file_path)

# # 创建 O3DVisualizer 实例
# vis = o3d.visualization.O3DVisualizer("O3D Visualizer Example", 1024, 768)

# # 添加几何体到可视化窗口
# vis.add_geometry("Mesh", mesh)

# # 显示可视化窗口
# Application.instance.add_window(vis)
# Application.instance.run()

# # Derive the object path set the model, material, and shader
# model_dir = sys.argv[1]
# model_name = os.path.join(model_dir, os.path.basename(model_dir) + ".obj")
# model = o3d.io.read_triangle_mesh(obj_file_path)
# material = o3d.visualization.rendering.Material()
# material.shader = "defaultLit"

# # Derive the texture paths
# albedo_name = os.path.join(model_dir, "albedo.png")
# normal_name = os.path.join(model_dir, "normal.png")
# ao_name = os.path.join(model_dir, "ao.png")
# metallic_name = os.path.join(model_dir, "metallic.png")
# roughness_name = os.path.join(model_dir, "roughness.png")

# # Check if the textures are available and loads the texture. For example, if metallic exists then load metallic texture
# if os.path.exists(albedo_name):
#     material.albedo_img = o3d.io.read_image(albedo_name)
# if os.path.exists(normal_name):
#     material.normal_img = o3d.io.read_image(normal_name)
# if os.path.exists(ao_name):
#     material.ao_img = o3d.io.read_image(ao_name)
# if os.path.exists(metallic_name):
#     material.base_metallic = 1.0
#     material.metallic_img = o3d.io.read_image(metallic_name)
# if os.path.exists(roughness_name):
#     material.roughness_img = o3d.io.read_image(roughness_name)

# # Draw an object named cube using the available model and texture
# o3d.visualization.draw([{"name": "cube", "geometry": model, "material": material}])