You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
165 lines
6.3 KiB
165 lines
6.3 KiB
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/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) |