277 lines
13 KiB
Python
277 lines
13 KiB
Python
import os, sys, time, shutil, subprocess, shlex
|
|
import platform
|
|
import pyautogui as ag
|
|
from export_build_info import export_already_build
|
|
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,computerRecboxCenterByPoint,libs_db_repeat_texture
|
|
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,"1")
|
|
#获取影棚id
|
|
# psid = libs.getPSid(pid)
|
|
# if int(psid) == 80:
|
|
# change_rcbox_deepth(str(pid),0.03)
|
|
psid = libs.getPSid(pid)
|
|
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 '
|
|
#存在photo3 的情况下，photo2 不参与贴图
|
|
#遍历获取photo2 目录下的所有文件
|
|
directory = os.path.join(config.workdir, pid, 'photo2')
|
|
for root, dirs, files in os.walk(directory):
|
|
for file in files:
|
|
#判断file 是否以 _8.xmp 结尾
|
|
if file.endswith("_8.xmp") == False:
|
|
continue
|
|
#修改文件内容不参与贴图
|
|
libs.set_photo_join_type(config.workdir, pid, 'photo2', file.split("_")[0], mesh='0', texture='0')
|
|
else:
|
|
pass
|
|
#设置photo2参与贴图，因为有的时候贴图会黑
|
|
directory = os.path.join(config.workdir, pid, 'photo2')
|
|
for root, dirs, files in os.walk(directory):
|
|
for file in files:
|
|
#判断file 是否以 _8.xmp 结尾
|
|
if file.endswith("_8.xmp") == False:
|
|
continue
|
|
#修改文件内容不参与贴图
|
|
libs.set_photo_join_type(config.workdir, pid, 'photo2', file.split("_")[0], mesh='0', texture='1')
|
|
|
|
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 -exportRegistration "{os.path.join(config.workdir,"make2", "config","exportRegistration.xml")}" "{os.path.join(config.workdir, pid,str(pid)+"_align.csv")}" -save'
|
|
|
|
cmd = cmd + f' "{os.path.join(config.workdir, pid, f"{pid}_wait.rcproj")}"'
|
|
print(cmd)
|
|
cmd = shlex.split(cmd)
|
|
res = subprocess.run(cmd)
|
|
time.sleep(2)
|
|
|
|
|
|
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()),"循环阻塞结束,判断是否对齐成功")
|
|
|
|
|
|
#查看对齐的数量
|
|
alignRes = common.isAlignNums(str(pid))
|
|
if alignRes == "rebuild":
|
|
os.system(f'python d:\\make2\\tools\push_cmd.py rebuild {pid}')
|
|
os.system(f'python d:\\make2\\main_service.py')
|
|
return
|
|
#区域的设置 建模
|
|
#update
|
|
cmdSmall = "-align"
|
|
#使用公共point文件的时候，就不能使用update ， 而是要用 align
|
|
if isExistPoint == True:
|
|
cmdSmall = "-align"
|
|
#{config.r1["init"]}
|
|
|
|
if common.task_need_high_model(pid):
|
|
if str(psid) == "41" or str(psid) == "85":
|
|
calulate_type = "-mvs"
|
|
else:
|
|
calulate_type = '-mvsHigh'
|
|
else:
|
|
calulate_type = '-mvs'
|
|
|
|
#翻转相机的线向下
|
|
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" {cmdSmall} \
|
|
-set "sfmEnableCameraPrior=True" -align -set "sfmEnableCameraPrior=False" -align -save "{os.path.join(config.workdir, pid, f"{pid}_point.rcproj")}" '
|
|
cmd = shlex.split(cmd)
|
|
res = subprocess.run(cmd)
|
|
#执行完成后命令会触发导出点云的定时程序
|
|
while True:
|
|
print("判断是否导出点云文件循环阻塞开始")
|
|
time.sleep(3)
|
|
#判断是否有 pid_1 的的值
|
|
print(pid+"_3")
|
|
if redisLocal.lpos('model:auto_distance',pid+"_3") == None:
|
|
continue
|
|
shutil.move(os.path.join(config.workdir, pid, f'{pid}_point.rcproj'), os.path.join(config.workdir, pid, f'{pid}.rcproj'))
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 导出点云文件完成')
|
|
#最后处理掉redis中的值
|
|
redisLocal.lrem('model:auto_distance', 0, pid+"_3")
|
|
break
|
|
|
|
#处理点云文件
|
|
centerRcboxValue = computerRecboxCenterByPoint.boxCenter(pid)
|
|
if centerRcboxValue > 0:
|
|
print("修改中心点的坐标",centerRcboxValue)
|
|
#修改中心点的值
|
|
common.change_rcbox_center(pid,centerRcboxValue)
|
|
|
|
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -align \
|
|
-set "sfmEnableCameraPrior=True" -align -set "sfmEnableCameraPrior=False" -align -setReconstructionRegion "{os.path.join(config.workdir, pid, f"{pid}.rcbox")}" \
|
|
{calulate_type} -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)
|
|
#保存在导出
|
|
# common.saveScreenImg(pid)
|
|
# time.sleep(1)
|
|
# ag.hotkey('alt', 'f4')
|
|
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
|
|
|
|
#复制文件
|
|
shutil.copy(os.path.join(config.workdir, pid, f'{pid}.rcproj'), os.path.join(config.workdir, pid, f'{pid}_screen.rcproj'))
|
|
time.sleep(2)
|
|
#打开工程文件
|
|
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}_screen.rcproj")}"'
|
|
cmd = shlex.split(cmd)
|
|
res = subprocess.run(cmd)
|
|
|
|
#创建指定文件夹
|
|
if not os.path.exists(os.path.join(config.workdir, pid, "output")):
|
|
os.makedirs(os.path.join(config.workdir, pid, "output"))
|
|
|
|
#判断output目录下是否有文件，有的话删除
|
|
files = os.listdir(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):
|
|
if str(psid) == "41" or str(psid) == "85":
|
|
calulate_type = "calculateNormalModel"
|
|
else:
|
|
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
|
|
|
|
# #插入需要重新贴图的任务表
|
|
# libs_db_repeat_texture.add_reapeat_texture_task({"pid":pid,"heads":libs.getHeadCount(pid)})
|
|
# #然后将导出的obj文件上传到oss上
|
|
# config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/repeat_texture/{pid}.obj', os.path.join(config.workdir, pid, 'output', f'{pid}.obj'))
|
|
# config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/repeat_texture/{pid}.mtl', os.path.join(config.workdir, pid, 'output', f'{pid}.mtl'))
|
|
# config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/repeat_texture/{pid}_u0_v0_diffuse.jpg', os.path.join(config.workdir, pid, 'output', f'{pid}_u0_v0_diffuse.jpg'))
|
|
# config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/repeat_texture/{pid}.obj.rcInfo', os.path.join(config.workdir, pid, 'output', f'{pid}.obj.rcInfo'))
|
|
# #导出建模信息
|
|
export_already_build.exportInfo(str(pid))
|
|
|
|
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) |