import sys, os, oss2, bpy, bmesh, requests, json, shutil, time, platform from PIL import Image 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: 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 down_obj_fromoss(pid, print_type=1, order_id=None): # print_type：// 打印状态 1:正常打印 2:重打 3:加打，4: 样品 print('开始下载obj文件...' , pid) if not order_id is None: path = os.path.join(workdir, f'{pid}_{order_id}') else: path = os.path.join(workdir, pid) if not os.path.exists(path): os.makedirs(path) # 根据前缀获取文件列表 prefix = f'objs/print/{pid}/' filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix) find = False for file in filelist: filename = file.key.split('/')[-1] if filename == '': continue if filename.endswith(f'{pid}.obj'): find = True localfile = os.path.join(path, filename) res = oss_client.get_object_to_file(file.key, localfile) print(f'下载文件：{file.key}，状态：{res.status}') def down_auto_obj_fromoss(pid, print_type=1, order_id=None): # print_type：// 打印状态 1:正常打印 2:重打 3:加打，4: 样品 print('开始下载obj文件...' , pid) if not order_id is None: path = os.path.join(workdir, f'{pid}_{order_id}') else: path = os.path.join(workdir, pid) if not os.path.exists(path): os.makedirs(path) # 根据前缀获取文件列表 prefix = f'objs/auto/{pid}/' filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix) find = False for file in filelist: filename = file.key.split('/')[-1] if filename == '': continue if filename.endswith(f'{pid}.obj'): find = True localfile = os.path.join(path, filename) res = oss_client.get_object_to_file(file.key, localfile) print(f'下载文件：{localfile}，状态：{res.status}') def find_obj(pid): find = False if not os.path.exists(os.path.join(workdir, pid, f'{pid}.mtl')): print('没有找到obj模型文件，开始下载') down_obj_fromoss(pid) if os.path.exists(os.path.join(workdir, pid, f'{pid}.jpg')): shutil.move(os.path.join(workdir, pid, f'{pid}.jpg'), os.path.join(workdir, pid, f'{pid}Tex1.jpg')) with open(os.path.join(workdir, pid, f'{pid}.mtl'), 'r') as f: lines = f.readlines() lines = [line.replace(f'map_Kd {pid}.jpg', f'map_Kd {pid}Tex1.jpg') for line in lines] with open(os.path.join(workdir, pid, f'{pid}.mtl'), 'w') as f: f.writelines(lines) filelist = os.listdir(os.path.join(workdir, pid)) for filename in filelist: if '9cm' in filename: find = True return filename for filename in filelist: if f'{pid}.obj' in filename: find = True return filename for filename in filelist: if '.obj' in filename: find = True return filename print('没有找到obj模型文件') return '' def resize_photo(filename, ratio, quality_value=80): print('正在压缩图片:', filename) img = Image.open(filename) width, height = img.size width = int(width * ratio) height = int(height * ratio) img = img.resize((width, height)) img.save(filename, optimize=True, quality=quality_value) def fix_link_texture(pid): path = os.path.join(workdir,action,pid) if os.path.exists(os.path.join(path, f'{pid}Tex1_old.jpg')): shutil.copy(os.path.join(path, f'{pid}Tex1_old.jpg'), os.path.join(path, f'{pid}Tex1_decimate.jpg')) elif os.path.exists(os.path.join(path, f'{pid}.jpg')): shutil.copy(os.path.join(path, f'{pid}.jpg'), os.path.join(path, f'{pid}Tex1_decimate.jpg')) else: shutil.copy(os.path.join(path, f'{pid}Tex1.jpg'), os.path.join(path, f'{pid}Tex1_decimate.jpg')) resize_photo(os.path.join(path, f'{pid}Tex1_decimate.jpg'), 0.5) # 修改pid_original.mtl文件中的贴图为pid_old.jpg with open(os.path.join(path, f'{pid}_decimate.mtl'), 'r') as f: lines = f.readlines() for i in range(len(lines)): if lines[i].startswith('map_Kd'): lines[i] = f'map_Kd {pid}Tex1_decimate.jpg\n' break with open(os.path.join(path, f'{pid}_decimate.mtl'), 'w') as f: f.writelines(lines) f.close() def main(pid, faces_per_person,action="print"): start_time = time.time() print('{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 开始处理pid: {pid}') # 读取obj文件 path = os.path.join(workdir,action,pid) libs.down_obj_from_oss(workdir,pid,action) prefixJpg = "Tex1.jpg" if action == "auto": prefixJpg = ".jpg" resize_photo(os.path.join(path, f'{pid}{prefixJpg}'), 0.5) obj_file = f'{pid}.obj' bpy.ops.wm.read_homefile() # bpy.context.preferences.view.language = 'en_US' bpy.ops.object.delete(use_global=False, confirm=False) 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' bpy.ops.import_scene.obj(filepath=os.path.join(path, obj_file)) obj = bpy.context.selected_objects[0] bpy.context.view_layer.objects.active = obj obj.select_set(True) scale = 90 / obj.dimensions.y obj.scale = (scale, scale, scale) bm = bmesh_copy_from_object(obj) obj_volume = round(bm.calc_volume() / 1000, 3) print('volume:', obj_volume) print('weight:', obj_volume * 1.26, 'g') faces = len(obj.data.polygons) print('faces:', faces) with open(os.path.join(path, f'{pid}_decimate.txt'), 'w') as f: lines = [f'faces:{faces}\n', f'volume:{obj_volume}\n'] f.writelines(lines) f.close() # upload_res = requests.get(f'{upload_obj_volume_url}?pid={pid}&order_id={order_id}&faces={faces}&volume={obj_volume}') # print('上传模型体积:', upload_res.text) # 生成数字模型 headcount = get_headcount(pid) faces_dest = faces_per_person * headcount # 减面 faces_current = len(obj.data.polygons) bpy.ops.object.modifier_add(type='DECIMATE') bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current bpy.ops.object.modifier_apply(modifier="Decimate") bpy.ops.object.origin_set(type='ORIGIN_CENTER_OF_VOLUME', center='MEDIAN') bpy.context.object.location[0] = 0 bpy.context.object.location[1] = 0 # bpy.context.object.rotation_euler = (0, 0, 0) bpy.ops.object.transform_apply(location=True, rotation=True, scale=True) bpy.ops.export_scene.obj(filepath=os.path.join(path, f'{pid}_decimate.obj')) fix_link_texture(pid) # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(path, f'{pid}.blend')) bpy.ops.export_scene.gltf(filepath=os.path.join(path, f'{pid}_decimate.glb'), export_format='GLB', export_apply=True, export_jpeg_quality=80) bpy.ops.wm.quit_blender() input = os.path.join(path, f'{pid}_decimate.obj') output = os.path.join(path, f'{pid}_decimate.glb') # os.system(f'gltfpack -c -i {input} -o {output}') oss_client.put_object_from_file(f'glbs/3d/{pid}.glb', output) # print('上传glb文件:', f'glbs/3d/{pid}.glb 完成') for filename in os.listdir(path): if not 'decimate' in filename: os.remove(os.path.join(path, filename)) print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 处理完成，共费时{libs.diff_time(start_time)}') def get_headcount(pid): res = requests.get(get_printinfo_url, params={'id': pid}) print('get_printsize_url:', res.url) print('res:', res.text) if res.status_code != 200: print('获取人数失败，程序退出') exit(1) res = json.loads(res.text) return res['data']['headcount'] if __name__ == '__main__': if platform.system() == 'Windows': workdir = 'd:\\3d\\' else: workdir = '/data/datasets/3d/' get_printinfo_url = config.urls['get_printinfo_url'] upload_obj_volume_url = 'https://mp.api.suwa3d.com/api/physical/add' # ?pid=1&order_id=1&faces=1&volume=1 ali_oss = { 'access_key_id': 'LTAI5tSReWm8hz7dSYxxth8f', 'access_key_secret': '8ywTDF9upPAtvgXtLKALY2iMYHIxdS', 'facebody_endpoint': 'facebody.cn-shanghai.aliyuncs.com', 'endpoint': 'oss-cn-shanghai.aliyuncs.com', 'bucket_name': 'suwa3d-securedata', } oss_client = oss2.Bucket(oss2.Auth(ali_oss['access_key_id'], ali_oss['access_key_secret']), ali_oss['endpoint'], ali_oss['bucket_name']) faces_per_person = 120000 action = "print" if len(sys.argv) == 3: pids = sys.argv[2].split(',') action = sys.argv[1] for pid in pids: main(pid, faces_per_person,action) else: pass # # 根据前缀获取文件列表 # prefix = f'glbs/3d/' # filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix) # start = False # for file in filelist: # filename = file.key.split('/')[-1] # if filename.endswith('.glb'): # pid = filename.split('.')[0] # if pid == '99531': # start = True # continue # elif not start: # print('跳过pid:', pid) # continue # if start: # print('正在处理pid:', pid) # main(pid, faces_per_person) # print('处理pid:', pid, '完成')