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.
91 lines
3.8 KiB
91 lines
3.8 KiB
import os, sys, datetime, oss2, requests, time |
|
import subprocess, shlex, shutil |
|
|
|
def find_blender_bin_path(): |
|
base_path = 'C:\\Program Files\\Blender Foundation\\' |
|
if os.path.exists(base_path): |
|
for dir in os.listdir(base_path): |
|
if dir.startswith('Blender'): |
|
blender_bin_path = base_path + dir + '\\blender.exe' |
|
return f'"{blender_bin_path}"' |
|
else: |
|
print('未找到blender安装目录') |
|
exit(1) |
|
|
|
def diff_time(start_time, end_time): |
|
d = end_time - start_time |
|
m = d.microseconds |
|
return str(d.seconds // 60) + ":" + str(d.seconds % 60) + ":" + str(m)[:-3] |
|
|
|
def delete_lines_in_file(filename, count): |
|
with open(filename, 'r') as f: |
|
lines = f.readlines() |
|
lines = lines[count:] |
|
with open(filename, 'w') as f: |
|
f.writelines(lines) |
|
|
|
def fix_mtl_file(pid): |
|
mtl_file = os.path.join(input_path, 'output', pid + '.mtl') |
|
with open(mtl_file, 'r') as f: |
|
lines = f.readlines() |
|
lines = [line.replace('_u0_v0_diffuse.jpg', '.jpg') for line in lines] |
|
with open(mtl_file, 'w') as f: |
|
f.writelines(lines) |
|
os.rename(os.path.join(input_path, 'output', pid + '_u0_v0_diffuse.jpg'), os.path.join(input_path, 'output', pid + '.jpg')) |
|
|
|
if __name__ == '__main__': |
|
AccessKeyId = 'LTAI5tSReWm8hz7dSYxxth8f' |
|
AccessKeySecret = '8ywTDF9upPAtvgXtLKALY2iMYHIxdS' |
|
Endpoint = 'oss-cn-shanghai.aliyuncs.com' |
|
Bucket = 'suwa3d-securedata' |
|
oss_client = oss2.Bucket(oss2.Auth(AccessKeyId, AccessKeySecret), Endpoint, Bucket) |
|
update_status_modelsuccess_url = 'https://repair.api.suwa3d.com/api/modelRepairOrder/toModelMakeSucceed' |
|
|
|
workdir = 'd:\\' |
|
|
|
start = datetime.datetime.now() |
|
if len(sys.argv) == 2: |
|
pid = sys.argv[1] |
|
else: |
|
print('usage: python rc_export.py [pid]') |
|
exit(1) |
|
input_path = os.path.join(workdir, pid) |
|
output_path = input_path + '\\output\\' |
|
ImagesGeometry = os.path.join(input_path, "photo1") |
|
ImagesTexture = os.path.join(input_path, "photo2") |
|
|
|
rcbin = '"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe"' |
|
blenderbin = find_blender_bin_path() |
|
config_path = 'D:\\apps\\config\\' |
|
|
|
shutil.rmtree(output_path, ignore_errors=True) |
|
os.system('mkdir ' + output_path) |
|
|
|
cmd = rcbin + \ |
|
' -delegateTo ' + pid + \ |
|
' -exportSelectedModel "' + output_path + pid + '.obj" "' + config_path + 'ModelExportParams.xml" ' + \ |
|
' -quit' |
|
|
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
subprocess.run(cmd) |
|
time.sleep(20) |
|
fix_mtl_file(pid) |
|
delete_lines_in_file(os.path.join(output_path, pid + '.mtl'), 6) |
|
delete_lines_in_file(os.path.join(output_path, pid + '.obj'), 6) |
|
os.system(f'{blenderbin} -b -P d:\\apps\\blender\\autofix.py -- {pid}') |
|
os.system(f'gltfpack -c -i {os.path.join(output_path, pid + ".obj")} -o {os.path.join(output_path, pid + ".glb")}') |
|
print(f'开始上传{pid}模型文件...') |
|
oss_client.put_object_from_file(f'objs/auto/{pid}/{pid}.obj.rcInfo', os.path.join(input_path, 'output', pid + '.obj.rcInfo')) |
|
oss_client.put_object_from_file(f'objs/auto/{pid}/{pid}.obj', os.path.join(input_path, 'output', pid + '.obj')) |
|
oss_client.put_object_from_file(f'objs/auto/{pid}/{pid}.mtl', os.path.join(input_path, 'output', pid + '.mtl')) |
|
oss_client.put_object_from_file(f'objs/auto/{pid}/{pid}.jpg', os.path.join(input_path, 'output', pid + '.jpg')) |
|
oss_client.put_object_from_file(f'glbs/auto/{pid}.glb', os.path.join(input_path, 'output', pid + '.glb')) |
|
print(f'{update_status_modelsuccess_url}?id={pid}') |
|
res = requests.post(update_status_modelsuccess_url, data={'id': pid}) |
|
print('上传完成更新建模成功状态：', res.text) |
|
|
|
end = datetime.datetime.now() |
|
print('Time: ' + diff_time(start, end)) |
|
|
|
shutil.rmtree(input_path, ignore_errors=True) |