first commit

tools/cal_weight.py

@@ -0,0 +1,92 @@
+import os, sys, time, bpy, bmesh, shutil
+import platform
+if platform.system() == 'Windows':
+    sys.path.append('e:\\libs\\')
+else:
+    sys.path.append('/data/deploy/make3d/make2/libs/')
+    
+import config, libs, libs_db
+
+def bmesh_copy_from_object(obj, transform=True, triangulate=True, apply_modifiers=False):
+    """Returns a transformed, triangulated copy of the mesh"""
+    assert obj.type == 'MESH'
+    if apply_modifiers and obj.modifiers:
+        import bpy
+        depsgraph = bpy.context.evaluated_depsgraph_get()
+        obj_eval = obj.evaluated_get(depsgraph)
+        me = obj_eval.to_mesh()
+        bm = bmesh.new()
+        bm.from_mesh(me)
+        obj_eval.to_mesh_clear()
+    else:
+        me = obj.data
+        if obj.mode == 'EDIT':
+            bm_orig = bmesh.from_edit_mesh(me)
+            bm = bm_orig.copy()
+        else:
+            bm = bmesh.new()
+            bm.from_mesh(me)
+    if transform:
+        matrix = obj.matrix_world.copy()
+        if not matrix.is_identity:
+            bm.transform(matrix)
+            matrix.translation.zero()
+            if not matrix.is_identity:
+                bm.normal_update()
+    if triangulate:
+        bmesh.ops.triangulate(bm, faces=bm.faces)
+    return bm
+
+def reload_obj(pid, action):
+    obj_filename = os.path.join(config.workdir, action, pid, f'{pid}.obj')
+    bpy.ops.wm.read_homefile()
+    bpy.ops.object.delete(use_global=False, confirm=False)
+    bpy.ops.import_scene.obj(filepath=obj_filename)
+    bpy.context.scene.unit_settings.scale_length = 0.001
+    bpy.context.scene.unit_settings.length_unit = 'CENTIMETERS'
+    bpy.context.scene.unit_settings.mass_unit = 'GRAMS'
+
+    obj = bpy.context.selected_objects[0]
+    bpy.context.view_layer.objects.active = obj
+    obj.select_set(True)
+
+    return obj
+
+def cal_weight(obj, size):
+        # 统一缩放到9cm标准尺寸
+    scale = size / obj.dimensions.z
+    obj.scale = (scale, scale, scale)
+    bpy.ops.object.transform_apply(scale=True)
+
+    # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, action, pid, f'{pid}-{size/10}cm.blend'))
+    model_info = {}
+    bm = bmesh_copy_from_object(obj)
+    model_info['volume'] = round(bm.calc_volume() / 1000)
+    model_info['weight'] = round(model_info['volume'] * 1.226)
+    print(f'{size/10}cm：体积 {model_info["volume"]}cm³, 克重 {model_info["weight"]}g')
+
+def main(action, pid, sizes):
+    libs.down_obj_from_oss(config.workdir, pid, action)
+    obj = reload_obj(pid, action)
+    
+    bpy.ops.object.transform_apply(location=True, rotation=True, scale=True)
+    print(f'模型{pid}的体积与克重估算信息:')
+    for size in sizes:
+        size = float(size)
+        cal_weight(obj, size)
+    
+if __name__ == '__main__':
+    sizes = (90, 120, 150, 180)
+
+    if len(sys.argv) == 3:
+        action = sys.argv[1]
+        pids = sys.argv[2].split(',')
+        for pid in pids:
+            main(action, pid, sizes)
+    elif len(sys.argv) == 4:
+        action = sys.argv[1]
+        pids = sys.argv[2].split(',')
+        sizes = sys.argv[3].split(',')
+        for pid in pids:
+            main(action, pid, sizes)
+        print('Usage: python cal_weight.py <pids>')