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.
197 lines
8.9 KiB
197 lines
8.9 KiB
import os, sys, time, shutil, subprocess, shlex |
|
import platform |
|
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,common,main_service_db |
|
redisLocal = config.redis_local |
|
def load_model(pid): |
|
cmd = f'{config.rcbin} {config.r1["init"]} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"' |
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
res = subprocess.run(cmd) |
|
|
|
def get_rcver(): |
|
rcbin = '"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe"' |
|
if os.path.getsize(rcbin[1:-1]) == 20783616: |
|
return 1 |
|
else: |
|
return 2 |
|
|
|
def make3d(pid): |
|
if get_rcver() == 1: # old version |
|
#修改重建区域的大小 |
|
common.change_rcbox_s(pid,"0.999") |
|
#获取影棚id |
|
# psid = libs.getPSid(pid) |
|
# if int(psid) == 80: |
|
# change_rcbox_deepth(str(pid),0.03) |
|
|
|
simplify_value = 1000000 * libs.getHeadCount(pid) |
|
add_photo3 = ' ' |
|
#判断是否存在photo3 |
|
if os.path.exists(os.path.join(config.workdir, pid, 'photo3')): |
|
add_photo3 = ' -addFolder "' + os.path.join(config.workdir, pid, 'photo3') + '" -align -align ' |
|
|
|
if get_rcver() == 1: # old version |
|
|
|
#判断oss 上是否存在 controlpoints 文件 |
|
isExistPoint = common.isExistControlPointsOss(pid) |
|
cmd = f'{config.rcbin} {config.r1["init"]} \ |
|
-addFolder "{os.path.join(config.workdir, pid, "photo1")}" -addFolder "{os.path.join(config.workdir, pid, "photo2")}" {add_photo3} \ |
|
-importControlPointsMeasurements "{os.path.join(config.workdir, pid, f"{pid}.controlPoints.csv")}" \ |
|
-align -save' |
|
|
|
#不存在point文件的时候就要用自动点击的方式 |
|
if isExistPoint == False: |
|
cmd = cmd + f' "{os.path.join(config.workdir, pid, f"{pid}_wait.rcproj")}"' |
|
else: |
|
cmd = cmd + f' "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
res = subprocess.run(cmd) |
|
time.sleep(2) |
|
|
|
if isExistPoint == False: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 定位点导入完成') |
|
time.sleep(3) |
|
# defind_distance |
|
#死循环阻塞获取 |
|
while True: |
|
print("循环阻塞开始") |
|
time.sleep(3) |
|
#判断是否有 pid_1 的的值 |
|
print(pid+"_1") |
|
if redisLocal.lpos('model:auto_distance',pid+"_1") == None: |
|
continue |
|
shutil.move(os.path.join(config.workdir, pid, f'{pid}_wait.rcproj'), os.path.join(config.workdir, pid, f'{pid}.rcproj')) |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 定义定位点距离完成') |
|
#将 controlpoints_0.dat 文件拷贝到 oss 上作为公共的使用 |
|
common.uploadControlPointsOss(pid) |
|
|
|
#最后处理掉redis中的值 |
|
redisLocal.lrem('model:auto_distance', 0, pid+"_1") |
|
break |
|
print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),"循环阻塞结束,开始建模") |
|
else: |
|
#修改rcproj文件 controlpoints fileName 的引入 |
|
flag = common.changeRcprojControlpointsFile(pid) |
|
if flag == False: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 获取及修改controlpoints失败') |
|
return |
|
|
|
print("使用公共point文件的方式进行建模") |
|
|
|
#区域的设置 建模 |
|
#update |
|
cmdSmall = "-align" |
|
#使用公共point文件的时候，就不能使用update ， 而是要用 align |
|
if isExistPoint == True: |
|
cmdSmall = "-align" |
|
#{config.r1["init"]} |
|
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" {cmdSmall} \ |
|
-set "sfmEnableCameraPrior=True" -align -set "sfmEnableCameraPrior=False" -align -setReconstructionRegion "{os.path.join(config.workdir, pid, f"{pid}.rcbox")}" \ |
|
-mvs -modelSelectMaximalConnectedComponent -modelInvertSelection -modelRemoveSelectedTriangles -closeHoles -clean -simplify {simplify_value} -smooth -unwrap -calculateTexture -renameModel {pid} -save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
res = subprocess.run(cmd) |
|
|
|
time.sleep(3) |
|
#修改rcproj文件 |
|
flag = common.changeRcprojFile(pid) |
|
if flag == False: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} rcproj文件不存在') |
|
return |
|
#保存在导出 |
|
|
|
#创建指定文件夹 |
|
if not os.path.exists(os.path.join(config.workdir, pid, "output")): |
|
os.makedirs(os.path.join(config.workdir, pid, "output")) |
|
#执行导出 |
|
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"\ |
|
-exportModel "{pid}" "{os.path.join(config.workdir, pid, "output", f"{pid}.obj")}" "d:\\make2\\config\\ModelExportParams102.xml" -quit' |
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
res = subprocess.run(cmd) |
|
|
|
|
|
|
|
else: # new version |
|
#判断是否要进行高精模 |
|
if common.task_need_high_model(pid): |
|
calulate_type = 'calculateHighModel' |
|
else: |
|
calulate_type = 'calculateNormalModel' |
|
|
|
#创建指定文件夹 |
|
if not os.path.exists(os.path.join(config.workdir, pid, "output")): |
|
os.makedirs(os.path.join(config.workdir, pid, "output")) |
|
|
|
cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
-load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
-{calulate_type} \ |
|
-selectLargestModelComponent -invertTrianglesSelection -removeSelectedTriangles -simplify {simplify_value} -smooth -closeHoles -cleanModel -calculateTexture \ |
|
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
-exportSelectedModel "{os.path.join(config.workdir, pid, "output", f"{pid}.obj")}" "d:\\make2\\config\\ModelExportParams.xml" -quit' |
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
res = subprocess.run(cmd) |
|
|
|
|
|
#阻塞判断是否导出完成 |
|
while True: |
|
#判断 output 目录下是否存在 三个文件 |
|
files = os.listdir(os.path.join(config.workdir, pid, "output")) |
|
if len(files) >= 3: |
|
break |
|
|
|
def step2(pid,task_distributed_id=""): |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始建模任务step2') |
|
|
|
#判断是否要从共享目录拷贝数据 |
|
if os.path.exists(os.path.join(config.sharedir, pid)) and not os.path.exists(os.path.join(config.workdir, pid)): |
|
shutil.move(os.path.join(config.sharedir, pid), config.workdir) |
|
else: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 目录{os.path.join(config.sharedir, pid)}不存在，或{os.path.join(config.workdir, pid)}已存在') |
|
# return |
|
#最后还是要判断有没有存在目录 |
|
if not os.path.exists(os.path.join(config.workdir, pid)): |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 目录{os.path.join(config.workdir, pid)}不存在') |
|
return |
|
|
|
start_time = time.time() |
|
make3d(pid) |
|
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 建模任务step2完成，共费时{libs.diff_time(start_time)}，任务已提交到step3') |
|
# 更新本地任务状态，加入step3任务队列 |
|
if task_distributed_id == "": |
|
os.system(f'python d:\\make2\\main_step3.py {pid}') |
|
else: |
|
#暂时 step2 step3 一起连续执行 |
|
print('step2 执行完,开始执行step3') |
|
os.system(f'python d:\\make2\\main_step3.py {pid}') |
|
main_service_db.update_task_distributed({"id":task_distributed_id,"status":2,"finished_at":time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}) |
|
main_service_db.update_task_distributed_detail({"task_distributed_id":task_distributed_id,"finished_at":time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}) |
|
#return |
|
|
|
def main(pid): |
|
if pid == '0': |
|
while True: |
|
# 取本地mysql队列任务，完成第二步的建模任务 |
|
|
|
step2(pid) |
|
else: |
|
step2(pid) |
|
|
|
if __name__ == '__main__': |
|
# 取本地mysql队列任务，完成第二步的建模任务 |
|
# 默认循环值守，可传参数运行单一任务，以方便调试 |
|
pid = '0' |
|
if len(sys.argv) > 1: |
|
pids = sys.argv[1].split(',') |
|
for pid in pids: |
|
main(pid) |
|
exit() |
|
main(pid) |