2019-04-22 19:14:39 +00:00
|
|
|
import os, shutil
|
2016-03-24 17:35:29 +00:00
|
|
|
|
|
|
|
from opendm import log
|
|
|
|
from opendm import io
|
|
|
|
from opendm import system
|
|
|
|
from opendm import context
|
2019-04-22 19:14:39 +00:00
|
|
|
from opendm import types
|
2016-03-24 17:35:29 +00:00
|
|
|
|
2019-04-22 19:14:39 +00:00
|
|
|
class ODMMvsTexStage(types.ODM_Stage):
|
|
|
|
def process(self, args, outputs):
|
|
|
|
tree = outputs['tree']
|
|
|
|
reconstruction = outputs['reconstruction']
|
2016-03-24 17:35:29 +00:00
|
|
|
|
|
|
|
# define paths and create working directories
|
|
|
|
system.mkdir_p(tree.odm_texturing)
|
2018-07-03 16:37:39 +00:00
|
|
|
if not args.use_3dmesh: system.mkdir_p(tree.odm_25dtexturing)
|
2016-03-24 17:35:29 +00:00
|
|
|
|
2017-04-05 17:56:48 +00:00
|
|
|
runs = [{
|
|
|
|
'out_dir': tree.odm_texturing,
|
2018-01-03 02:02:43 +00:00
|
|
|
'model': tree.odm_mesh,
|
2018-07-03 16:37:39 +00:00
|
|
|
'nadir': False
|
2017-04-05 17:56:48 +00:00
|
|
|
}]
|
|
|
|
|
2018-07-03 16:37:39 +00:00
|
|
|
if args.skip_3dmodel:
|
2018-01-02 17:38:15 +00:00
|
|
|
runs = []
|
|
|
|
|
2018-07-03 16:37:39 +00:00
|
|
|
if not args.use_3dmesh:
|
2017-04-05 17:56:48 +00:00
|
|
|
runs += [{
|
|
|
|
'out_dir': tree.odm_25dtexturing,
|
2018-01-03 02:02:43 +00:00
|
|
|
'model': tree.odm_25dmesh,
|
2018-07-03 16:37:39 +00:00
|
|
|
'nadir': True
|
2017-04-05 17:56:48 +00:00
|
|
|
}]
|
|
|
|
|
|
|
|
for r in runs:
|
|
|
|
odm_textured_model_obj = os.path.join(r['out_dir'], tree.odm_textured_model_obj)
|
|
|
|
|
2019-04-22 19:14:39 +00:00
|
|
|
if not io.file_exists(odm_textured_model_obj) or self.rerun():
|
2019-06-28 15:10:08 +00:00
|
|
|
log.ODM_INFO('Writing MVS Textured file in: %s'
|
2017-04-05 17:56:48 +00:00
|
|
|
% odm_textured_model_obj)
|
2018-01-26 19:38:26 +00:00
|
|
|
|
2017-04-05 17:56:48 +00:00
|
|
|
# Format arguments to fit Mvs-Texturing app
|
|
|
|
skipGeometricVisibilityTest = ""
|
|
|
|
skipGlobalSeamLeveling = ""
|
|
|
|
skipLocalSeamLeveling = ""
|
|
|
|
skipHoleFilling = ""
|
|
|
|
keepUnseenFaces = ""
|
2018-07-03 16:37:39 +00:00
|
|
|
nadir = ""
|
2018-07-01 21:28:55 +00:00
|
|
|
|
2019-04-22 19:14:39 +00:00
|
|
|
if (self.params.get('skip_vis_test')):
|
2017-04-05 17:56:48 +00:00
|
|
|
skipGeometricVisibilityTest = "--skip_geometric_visibility_test"
|
2019-04-22 19:14:39 +00:00
|
|
|
if (self.params.get('skip_glob_seam_leveling')):
|
2017-04-05 17:56:48 +00:00
|
|
|
skipGlobalSeamLeveling = "--skip_global_seam_leveling"
|
2019-04-22 19:14:39 +00:00
|
|
|
if (self.params.get('skip_loc_seam_leveling')):
|
2017-04-05 17:56:48 +00:00
|
|
|
skipLocalSeamLeveling = "--skip_local_seam_leveling"
|
2019-04-22 19:14:39 +00:00
|
|
|
if (self.params.get('skip_hole_fill')):
|
2017-04-05 17:56:48 +00:00
|
|
|
skipHoleFilling = "--skip_hole_filling"
|
2019-04-22 19:14:39 +00:00
|
|
|
if (self.params.get('keep_unseen_faces')):
|
2017-04-05 17:56:48 +00:00
|
|
|
keepUnseenFaces = "--keep_unseen_faces"
|
2018-07-03 16:37:39 +00:00
|
|
|
if (r['nadir']):
|
|
|
|
nadir = '--nadir_mode'
|
2017-04-05 17:56:48 +00:00
|
|
|
|
|
|
|
# mvstex definitions
|
|
|
|
kwargs = {
|
|
|
|
'bin': context.mvstex_path,
|
|
|
|
'out_dir': io.join_paths(r['out_dir'], "odm_textured_model"),
|
|
|
|
'model': r['model'],
|
2019-04-22 19:14:39 +00:00
|
|
|
'dataTerm': self.params.get('data_term'),
|
|
|
|
'outlierRemovalType': self.params.get('outlier_rem_type'),
|
2017-04-05 17:56:48 +00:00
|
|
|
'skipGeometricVisibilityTest': skipGeometricVisibilityTest,
|
|
|
|
'skipGlobalSeamLeveling': skipGlobalSeamLeveling,
|
|
|
|
'skipLocalSeamLeveling': skipLocalSeamLeveling,
|
|
|
|
'skipHoleFilling': skipHoleFilling,
|
|
|
|
'keepUnseenFaces': keepUnseenFaces,
|
2019-04-22 19:14:39 +00:00
|
|
|
'toneMapping': self.params.get('tone_mapping'),
|
2018-07-08 04:16:45 +00:00
|
|
|
'nadirMode': nadir,
|
2018-08-11 16:45:21 +00:00
|
|
|
'nadirWeight': 2 ** args.texturing_nadir_weight - 1,
|
|
|
|
'nvm_file': io.join_paths(tree.opensfm, "reconstruction.nvm")
|
2017-04-05 17:56:48 +00:00
|
|
|
}
|
|
|
|
|
2018-06-03 14:35:07 +00:00
|
|
|
mvs_tmp_dir = os.path.join(r['out_dir'], 'tmp')
|
|
|
|
|
2019-06-05 15:58:38 +00:00
|
|
|
# Make sure tmp directory is empty
|
|
|
|
if io.dir_exists(mvs_tmp_dir):
|
|
|
|
log.ODM_INFO("Removing old tmp directory {}".format(mvs_tmp_dir))
|
|
|
|
shutil.rmtree(mvs_tmp_dir)
|
2019-06-03 15:43:50 +00:00
|
|
|
|
2019-06-05 15:58:38 +00:00
|
|
|
# run texturing binary
|
|
|
|
system.run('{bin} {nvm_file} {model} {out_dir} '
|
|
|
|
'-d {dataTerm} -o {outlierRemovalType} '
|
|
|
|
'-t {toneMapping} '
|
|
|
|
'{skipGeometricVisibilityTest} '
|
|
|
|
'{skipGlobalSeamLeveling} '
|
|
|
|
'{skipLocalSeamLeveling} '
|
|
|
|
'{skipHoleFilling} '
|
|
|
|
'{keepUnseenFaces} '
|
|
|
|
'{nadirMode} '
|
|
|
|
'-n {nadirWeight}'.format(**kwargs))
|
2019-05-15 22:01:46 +00:00
|
|
|
|
|
|
|
self.update_progress(50)
|
2016-09-30 13:08:56 +00:00
|
|
|
else:
|
2017-04-05 17:56:48 +00:00
|
|
|
log.ODM_WARNING('Found a valid ODM Texture file in: %s'
|
|
|
|
% odm_textured_model_obj)
|
2016-03-24 17:35:29 +00:00
|
|
|
|