import os, sys, time, bpy, bmesh, shutil,requests,json 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 import pandas as pd 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,2) model_info['weight'] = round(model_info['volume'] * 1.226,2) print(f'{size/10}cm：体积 {model_info["volume"]}cm³, 克重 {model_info["weight"]}g') return model_info["weight"] def main(action, pid, sizes,orderId=0): 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}的体积与克重估算信息:') arrData = {} excel_file = 'weights.xlsx' # 替换为你的文件路径 df = pd.read_excel(excel_file) new_data = { 'pid': [pid], # 假设新的 pid 是 '002' '1/4尺寸': [0], '1/4': [0], '1/6尺寸': [0], '1/6': [0], '1/10尺寸': [0], '1/10': [0], '1/12尺寸': [0], '1/12': [0], '1/14尺寸': [0], '1/14': [0], '1/16尺寸': [0], '1/16': [0], '1/18尺寸': [0], '1/18': [0], '1/24尺寸': [0], '1/24': [0], '1/32尺寸': [0], '1/32': [0], '1/43尺寸': [0], '1/43': [0], '1/64尺寸': [0], '1/64': [0], } for sizeV in sizes: arrSize = sizeV.split("_") size = float(arrSize[0]) weight = cal_weight(obj, size) # size = str(size/10)+"cm" arrData[arrSize[1]] = str(weight)+"g" if arrSize[1] in df.columns: new_data[arrSize[1]+"尺寸"] = [str(size/10)+"cm"] new_data[arrSize[1]] = [str(weight)+'g'] # 更新数据 # 将新数据转换为 DataFrame new_df = pd.DataFrame(new_data) # 将新数据附加到原始 DataFrame 中 df = pd.concat([df, new_df], ignore_index=True) # 保存更新后的 Excel 文件 df.to_excel(excel_file, index=False) # #请求接口进行更新数据 # arrParams = {"pid":pid,"order_id":orderId} # if action == "auto": # arrParams["auto_weight"] = json.dumps(arrData) # if action == "print": # arrParams["print_weight"] = json.dumps(arrData) # #发起请求 # print(f"请求的参数-{arrParams}") # url = "https://mp.api.suwa3d.com/api/physical/addPhysicalWeight" # res = requests.post(url,data=arrParams) # print('res:', res.text) #获取真实身高 def get_real_height(pid): get_real_height_url = 'https://mp.api.suwa3d.com/api/physical/infoByPid' print(f'pid: {pid}') res = requests.get(get_real_height_url, params={'pid': pid}) res = json.loads(res.text) if res['code'] == -1: print(f'Error: {res["message"]}, return default height 160') return 0 height = res['data']['height'] if height == 0: print(f'Error: height=0, return default height 160') return 0 print(f'height: {height}') return height if __name__ == '__main__': pids = [235777,235759,235272,235022,234683,234678,233973,233634,233618,233589,233588,233106,232673,231611,231604,231394,231306,231061,230812,230528,230523,230513,230329,230328,230327,230325,230272,230268,230259,229967,229935,229634,229380,229309,229308,229202,228698,228693,227372,227357,227295,227103,227026,226503,226502,226280,226137,225927,225438,225077,225059,224944,224898,224894,224493,224237,224229,224136,224131,223139,223137,222902,222897,222636,222516,222507,222488,222467,222278,222251,222248,222232,222196,221876,221679,221672,220781,220748,218310,217964,217050,216319,216311,215959,215488,215477,215466,214955,214951,214729,214726,212891,212169,211726,209906,208801,208798,208765,208706,208702,208640,208634,207864,207724,207722,207510,207363,204787,204700,203993,203657,203373,203340,203320,203243,203239,203235,203223,202784,202362,202351,200744,200205,200202,200152,200081] for pid in pids: #获取真实身高 realHeight = get_real_height(pid) if realHeight == 0: continue sizes = [] #根据真实身高计算不同的比例的高度 arrBiLie = [["1/4",1/4],["1/6",1/6],["1/8",1/8],["1/10",1/10],["1/12",1/12],["1/16",1/16],["1/18",1/18],["1/24",1/24],["1/32",1/32],["1/43",1/43],["1/64",1/64]] for biLieV in arrBiLie: biLie = biLieV[1] if biLie == 1/43: if realHeight < 140: continue if biLie == 1/64: if realHeight < 160: continue tempHeight = realHeight * biLie #取整 tempHeight = round(tempHeight) #拼接字符串 tempHeight = str(tempHeight)+"_"+str(biLieV[0]) sizes.append(tempHeight) print(sizes) # break main("print", pid, sizes)