first commit
This commit is contained in:
dongchangxi committed
1 parent
f20f6f3c34
commit
7a2b784979
36 files changed
+712
-3566
No files matched your search
@@ -0,0 +1,202 @@
|
||||
import os, sys, time, bpy, math, requests, bmesh, json, shutil
|
||||
from PIL import Image
|
||||
import platform,redis
|
||||
if platform.system() == 'Windows':
|
||||
sys.path.append('e:\\libs\\')
|
||||
#sys.path.append('libs')
|
||||
else:
|
||||
sys.path.append('/data/deploy/make3d/make2/libs/')
|
||||
import config, libs, libs_db,main_service_db,common,foot_mark_seam,libs_db_gpu
|
||||
|
||||
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 find_pid_objname(pid):
|
||||
for obj in bpy.data.objects:
|
||||
if obj.name.startswith(str(pid)):
|
||||
return obj.name
|
||||
|
||||
#加载obj文件
|
||||
def reload_obj(pid):
|
||||
|
||||
#下载obj文件
|
||||
libs.down_obj_from_oss(config.workdir, pid, "print")
|
||||
# obj = reload_obj(pid, action)
|
||||
|
||||
|
||||
obj_filename = os.path.join(config.workdir,'print',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 = 1
|
||||
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)
|
||||
|
||||
def base_fix(pid):
|
||||
# 统一blender环境
|
||||
reload_obj(pid)
|
||||
# 统一模型方向、位置、大小...
|
||||
pid_objname = find_pid_objname(pid)
|
||||
obj = bpy.data.objects[pid_objname]
|
||||
bpy.data.objects[pid_objname].rotation_euler = (0, 0, 0)
|
||||
# 检查当前模型是否正确使用 z 轴作为高度轴
|
||||
if obj.dimensions.z < obj.dimensions.y:
|
||||
# 如果 z 轴比 y 轴小，说明高度错误，应该旋转调整模型
|
||||
print(f"警告：模型的高度轴（z）似乎不正确，使用 y 轴作为高度。正在修正模型方向...")
|
||||
bpy.ops.object.transform_apply(location=True, rotation=True, scale=True) # 应用当前变换
|
||||
obj.rotation_euler[0] = math.radians(90) # 旋转模型 90 度，交换 z 和 y 轴
|
||||
bpy.ops.object.transform_apply(rotation=True) # 应用旋转
|
||||
|
||||
|
||||
|
||||
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.ops.object.transform_apply(location=True, rotation=True, scale=True)
|
||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 模型基础校正完成')
|
||||
|
||||
|
||||
def export_and_update_glbs(pid):
|
||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始导出并上传审核模型和3D相册模型glb文件...')
|
||||
start_time = time.time()
|
||||
#headcount = libs.getHeadCount(pid)
|
||||
headcount = 1
|
||||
pid_objname = find_pid_objname(pid)
|
||||
obj = bpy.data.objects[pid_objname]
|
||||
obj.select_set(True)
|
||||
|
||||
model_info = {}
|
||||
model_info['headcount'] = headcount
|
||||
model_info['faces'] = round(len(obj.data.polygons) / 10000)
|
||||
model_info['height'] = round(obj.dimensions.y * 100)
|
||||
|
||||
# bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, pid, f'{pid}.blend'))
|
||||
|
||||
# 统一缩放到9cm标准尺寸
|
||||
scale = 90 / bpy.data.objects[pid_objname].dimensions.y
|
||||
bpy.data.objects[pid_objname].scale = (scale, scale, scale)
|
||||
bpy.ops.object.transform_apply(scale=True)
|
||||
|
||||
# bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, pid, f'{pid}-9cm.blend'))
|
||||
|
||||
bm = bmesh_copy_from_object(obj)
|
||||
model_info['volume'] = round(bm.calc_volume(), 2)
|
||||
model_info['weight'] = round(model_info['volume'] * 1.226, 2)
|
||||
print(f'{pid}的模型数据：{model_info}')
|
||||
|
||||
res = requests.get(f'{config.urls["upload_model_info_url"]}?pid={pid}&headcount={headcount}&faces={model_info["faces"]}&volume={model_info["volume"]}&weight={model_info["weight"]}&height={model_info["height"]}')
|
||||
print('上传模型数据：', res.text)
|
||||
# with open(os.path.join(config.sharedir, 'model_info', f'{pid}.json'), 'w') as f:
|
||||
# json.dump(model_info, f)
|
||||
# f.close()
|
||||
|
||||
# 先生成审核模型
|
||||
faces_dest = 500000 * headcount
|
||||
# 减面
|
||||
faces_current = len(bpy.data.objects[pid_objname].data.polygons)
|
||||
print(f'当前面数：{faces_current}，目标面数：{faces_dest}')
|
||||
|
||||
bpy.ops.object.modifier_add(type='DECIMATE')
|
||||
bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current
|
||||
bpy.ops.object.modifier_apply(modifier="Decimate")
|
||||
|
||||
# glb_filename = os.path.join(config.workdir, pid, 'output', f'{pid}.glb')
|
||||
# bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False)
|
||||
|
||||
# 再生成数字模型
|
||||
faces_dest = 120000 * headcount
|
||||
|
||||
# 减面
|
||||
faces_current = len(bpy.data.objects[pid_objname].data.polygons)
|
||||
print(f'当前面数：{faces_current}，目标面数：{faces_dest}')
|
||||
|
||||
bpy.ops.object.modifier_add(type='DECIMATE')
|
||||
bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current
|
||||
bpy.ops.object.modifier_apply(modifier="Decimate")
|
||||
|
||||
glb_filename = os.path.join(config.workdir,'print',pid,f'{pid}-3d.glb')
|
||||
bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False)
|
||||
|
||||
# os.system(f'gltfpack -c -i {os.path.join(config.workdir, pid, "output", f"{pid}.glb")} -o {os.path.join(config.workdir, pid, "output", f"{pid}-pack.glb")}')
|
||||
# config.oss_bucket.put_object_from_file(f'glbs/auto/{pid}.glb', os.path.join(config.workdir, pid, 'output', f'{pid}-pack.glb'))
|
||||
config.oss_bucket.put_object_from_file(f'glbs/3d/{pid}.glb', os.path.join(config.workdir,'print',pid, f'{pid}-3d.glb'))
|
||||
|
||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} glb文件导出并上传完成，共费时{libs.diff_time(start_time)}')
|
||||
|
||||
#移除文件夹
|
||||
shutil.rmtree(os.path.join(config.workdir,'print',pid))
|
||||
|
||||
#更新该笔订单的状态为 3000
|
||||
requests.post("https://shop.api.suwa3d.com/api/printOrder/updateExternalOrderStatus", data={'pid': pid})
|
||||
|
||||
def createGlb(pid):
|
||||
|
||||
#移除文件夹
|
||||
try:
|
||||
shutil.rmtree(os.path.join(config.workdir,'print',pid))
|
||||
except:
|
||||
pass
|
||||
base_fix(pid)
|
||||
export_and_update_glbs(pid)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
arrArgs = sys.argv
|
||||
if len(arrArgs) == 2:
|
||||
pid = arrArgs[1]
|
||||
#判断是否是数字
|
||||
if pid.isdigit():
|
||||
print('请输入正确的pid')
|
||||
createGlb(str(pid))
|
||||
|
||||
else:
|
||||
r = redis.Redis(host="106.14.158.208",password="kcV2000",port=6379,db=6)
|
||||
while True:
|
||||
if r.llen('model:3dglb') == 0:
|
||||
print('队列为空，等待10秒')
|
||||
time.sleep(10)
|
||||
continue
|
||||
|
||||
info = r.lpop('model:3dglb')
|
||||
if info is None:
|
||||
print('队列为空，等待10秒')
|
||||
time.sleep(10)
|
||||
pid = info.decode('utf-8')
|
||||
|
||||
if not pid.isdigit():
|
||||
continue
|
||||
|
||||
createGlb(pid)
|
||||
|
||||
|
||||
Reference in new issue
Block a user