2019-04-22 19:14:39 +00:00
|
|
|
import os, math
|
2015-11-30 15:53:44 +00:00
|
|
|
|
|
|
|
from opendm import log
|
|
|
|
from opendm import io
|
|
|
|
from opendm import system
|
|
|
|
from opendm import context
|
2018-06-10 18:57:16 +00:00
|
|
|
from opendm import mesh
|
2018-08-08 19:41:08 +00:00
|
|
|
from opendm import gsd
|
2019-04-22 19:14:39 +00:00
|
|
|
from opendm import types
|
2023-02-20 15:40:47 +00:00
|
|
|
from opendm.dem import commands
|
2015-11-30 15:53:44 +00:00
|
|
|
|
2019-04-22 19:14:39 +00:00
|
|
|
class ODMeshingStage(types.ODM_Stage):
|
|
|
|
def process(self, args, outputs):
|
|
|
|
tree = outputs['tree']
|
|
|
|
reconstruction = outputs['reconstruction']
|
2015-11-30 15:53:44 +00:00
|
|
|
|
|
|
|
# define paths and create working directories
|
2015-12-10 11:01:41 +00:00
|
|
|
system.mkdir_p(tree.odm_meshing)
|
2015-11-30 15:53:44 +00:00
|
|
|
|
2018-07-03 16:37:39 +00:00
|
|
|
# Create full 3D model unless --skip-3dmodel is set
|
|
|
|
if not args.skip_3dmodel:
|
2019-04-22 19:14:39 +00:00
|
|
|
if not io.file_exists(tree.odm_mesh) or self.rerun():
|
2019-06-28 15:10:08 +00:00
|
|
|
log.ODM_INFO('Writing ODM Mesh file in: %s' % tree.odm_mesh)
|
2018-01-02 17:38:15 +00:00
|
|
|
|
2019-04-03 18:47:06 +00:00
|
|
|
mesh.screened_poisson_reconstruction(tree.filtered_point_cloud,
|
2018-07-03 15:24:47 +00:00
|
|
|
tree.odm_mesh,
|
2019-04-22 19:14:39 +00:00
|
|
|
depth=self.params.get('oct_tree'),
|
|
|
|
samples=self.params.get('samples'),
|
|
|
|
maxVertexCount=self.params.get('max_vertex'),
|
|
|
|
pointWeight=self.params.get('point_weight'),
|
2022-11-14 16:07:51 +00:00
|
|
|
threads=max(1, self.params.get('max_concurrency') - 1)), # poissonrecon can get stuck on some machines if --threads == all cores
|
2018-01-02 17:38:15 +00:00
|
|
|
else:
|
|
|
|
log.ODM_WARNING('Found a valid ODM Mesh file in: %s' %
|
|
|
|
tree.odm_mesh)
|
2019-05-15 22:01:46 +00:00
|
|
|
|
|
|
|
self.update_progress(50)
|
2016-02-26 18:50:12 +00:00
|
|
|
|
2018-07-03 16:37:39 +00:00
|
|
|
# Always generate a 2.5D mesh
|
|
|
|
# unless --use-3dmesh is set.
|
|
|
|
if not args.use_3dmesh:
|
2019-04-22 19:14:39 +00:00
|
|
|
if not io.file_exists(tree.odm_25dmesh) or self.rerun():
|
2018-06-10 18:57:16 +00:00
|
|
|
|
2019-06-28 15:10:08 +00:00
|
|
|
log.ODM_INFO('Writing ODM 2.5D Mesh file in: %s' % tree.odm_25dmesh)
|
2023-02-20 20:28:35 +00:00
|
|
|
|
|
|
|
multiplier = math.pi / 2.0
|
|
|
|
radius_steps = commands.get_dem_radius_steps(tree.filtered_point_cloud_stats, 3, args.orthophoto_resolution, multiplier=multiplier)
|
|
|
|
dsm_resolution = radius_steps[0] / multiplier
|
|
|
|
|
|
|
|
log.ODM_INFO('ODM 2.5D DSM resolution: %s' % dsm_resolution)
|
2018-10-17 15:35:18 +00:00
|
|
|
|
2018-10-24 16:30:09 +00:00
|
|
|
if args.fast_orthophoto:
|
2023-02-20 15:40:47 +00:00
|
|
|
dsm_resolution *= 8.0
|
2018-10-17 22:54:18 +00:00
|
|
|
|
2019-04-03 18:47:06 +00:00
|
|
|
mesh.create_25dmesh(tree.filtered_point_cloud, tree.odm_25dmesh,
|
2023-02-20 20:28:35 +00:00
|
|
|
radius_steps,
|
2018-10-17 15:35:18 +00:00
|
|
|
dsm_resolution=dsm_resolution,
|
2019-04-22 19:14:39 +00:00
|
|
|
depth=self.params.get('oct_tree'),
|
|
|
|
maxVertexCount=self.params.get('max_vertex'),
|
|
|
|
samples=self.params.get('samples'),
|
2018-12-12 19:24:27 +00:00
|
|
|
available_cores=args.max_concurrency,
|
2019-06-08 23:28:58 +00:00
|
|
|
method='poisson' if args.fast_orthophoto else 'gridded',
|
2021-12-07 19:53:27 +00:00
|
|
|
smooth_dsm=True)
|
2017-04-05 17:56:48 +00:00
|
|
|
else:
|
|
|
|
log.ODM_WARNING('Found a valid ODM 2.5D Mesh file in: %s' %
|
|
|
|
tree.odm_25dmesh)
|
|
|
|
|