diff --git a/tools/cal_weight_excel.py b/tools/cal_weight_excel.py new file mode 100644 index 0000000..780c289 --- /dev/null +++ b/tools/cal_weight_excel.py @@ -0,0 +1,163 @@ +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) + model_info['weight'] = round(model_info['volume'] * 1.226) + 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' + '6cm': [0], # 新的 6cm 数据 + '7cm': [0], # 新的 7cm 数据 + '8cm': [0], # 新的 8cm 数据 + } + + for size in sizes: + size = float(size) + weight = cal_weight(obj, size) + size = str(size/10)+"cm" + arrData[size] = str(weight)+"g" + + if size+"cm" in df.columns: + new_data[size+"cm"] = [weight] + + # 更新数据 + # 将新数据转换为 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/6.1/8,1/10,1/12,1/16,1/18,1/24,1/32,1/43,1/64] + for biLie in arrBiLie: + 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(biLie) + + sizes.append(tempHeight) + + print(sizes) + #main("print", pid, sizes) \ No newline at end of file