同步更改数据

script/turn_around.py

@@ -0,0 +1,124 @@
+import numpy as np
+import open3d as o3d
+import math,os
+#矫正模型的方向，目前只支持水平旋转，
+#55084 55090 55092 55114 55367 55370 55372 55730 55844 55849  56559 56654 56750 75408
+
+
+
+def rotate_vertex(x, y, z, axis, angle_degrees):
+    """
+    围绕任意轴旋转顶点。
+    axis是一个表示旋转轴的三元组 (ax, ay, az)。
+    angle_degrees是旋转角度，以度为单位。
+    """
+    # 将角度转换为弧度
+    angle_radians = math.radians(angle_degrees)
+
+    # 将旋转轴转换为单位向量
+    axis = np.array(axis)
+    axis = axis / np.linalg.norm(axis)
+
+    # 将顶点位置转换为向量
+    v = np.array([x, y, z])
+
+    # 应用罗德里格斯旋转公式
+    v_rotated = (v * math.cos(angle_radians) +
+                 np.cross(axis, v) * math.sin(angle_radians) +
+                 axis * np.dot(axis, v) * (1 - math.cos(angle_radians)))
+    
+    return v_rotated.tolist()
+
+def transform_obj_file(source_file, target_file, axis, angle_degrees):
+    """
+    读取 OBJ 文件，应用旋转，然后将结果保存到另一个文件。
+    """
+    with open(source_file, 'r') as src:
+        lines = src.readlines()
+    
+    with open(target_file, 'w') as tgt:
+        for line in lines:
+            if line.startswith('v '):
+                parts = line.split()
+                x, y, z = map(float, parts[1:4])
+                x_new, y_new, z_new = rotate_vertex(x, y, z, axis, angle_degrees)
+                tgt.write(f"v {x_new} {y_new} {z_new}\n")
+            else:
+                tgt.write(line)
+
+def list_files(directory):
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            yield os.path.join(root, file)
+
+# # #遍历文件夹下的所有文件
+# folder_path = 'G://obj_fix/print/wrong'
+# nums = 0
+# rotate_axis = (0, 1, 0)  # 比如绕X轴旋转
+# for file_path in list_files(folder_path):
+#     #print(file_path)  G://obj_fix/print/around/preview\9023_preview_117.png
+#     if "_preview.png" in file_path:
+#         continue
+#     if "_right.png" in file_path:
+#         continue
+#     if "_wait.png" in file_path:
+#         continue
+#     #获取pid 从 "G://obj_fix/print/around/preview\9023_preview_117.png" 中获取 9023 he 117
+#     arrData = file_path.split('\\')
+#     lastData = arrData[-1]
+#     arrLastData = lastData.split('_')
+#     if len(arrLastData) < 3:
+#         print(lastData)
+#         continue
+#     pid = arrLastData[0]
+#     angle = arrLastData[2].split('.')[0]
+#     #判断 angle 第一个数字是否为 0 
+
+#     source_obj = f'G://obj_fix/print/{pid}/{pid}.obj'
+#     target_obj = f'G://obj_fix/print/{pid}/{pid}_new.obj'
+    
+#     nums +=1
+#     print(f"当前正在处理{pid},已处理至-{nums}",)
+#     if str(angle[0]) == "0":
+#         rotate_angle = -int(angle)
+#         transform_obj_file(source_obj, target_obj, rotate_axis, rotate_angle)
+#     else:
+#         rotate_angle = int(angle)
+#         transform_obj_file(source_obj, target_obj, rotate_axis, rotate_angle)
+    # print(int(angle))
+    # print(angle,angle[0])
+
+
+
+# # 加载模型
+# mesh = o3d.io.read_triangle_mesh(target_obj, enable_post_processing=True)
+
+# # 创建渲染窗口并添加模型
+# vis = o3d.visualization.Visualizer()
+# vis.create_window()
+# vis.add_geometry(mesh)
+
+# # 设置渲染参数
+# render_option = vis.get_render_option()
+# render_option.point_size = 1.0
+
+# # 渲染模型
+# vis.run()
+# vis.destroy_window()
+
+
+pid = "90864" # 98894  115123
+source_obj = f'G://obj_fix/auto/{pid}/{pid}.obj'
+target_obj = f'G://obj_fix/auto/{pid}/{pid}_90.obj'
+
+#遍历文件夹下的所有文件 /data/datasets/temp/auto_preview_need_around
+for file_path in list_files('G://obj_fix/auto'):
+    pid = file_path.split('\\')[-1].split('_')[0]
+    # 使用示例：
+    rotate_axis = (1, 0, 0)  # 比如绕X轴旋转 0 x 轴
+    rotate_angle = -90  # 以度为单位
+    transform_obj_file(source_obj, target_obj, rotate_axis, rotate_angle)
+
+# rotate_axis = (0, 1, 0)  # 比如绕X轴旋转 0 x 轴
+# rotate_angle = 60  # 以度为单位
+# transform_obj_file(target_obj, target_obj, rotate_axis, rotate_angle)