mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-05-16 18:46:27 +02:00
125 lines
4.8 KiB
Python
125 lines
4.8 KiB
Python
__version__ = "2.0"
|
|
|
|
from meshroom.core import desc
|
|
|
|
import os.path
|
|
|
|
|
|
class SfMAlignment(desc.AVCommandLineNode):
|
|
commandLine = 'aliceVision_utils_sfmAlignment {allParams}'
|
|
size = desc.DynamicNodeSize('input')
|
|
|
|
category = 'Utils'
|
|
documentation = '''
|
|
This node allows to change the coordinate system of one SfM scene to align it on another one.
|
|
|
|
The alignment can be based on:
|
|
* from_cameras_viewid: Align cameras in both SfM on the specified viewId
|
|
* from_cameras_poseid: Align cameras in both SfM on the specified poseId
|
|
* from_cameras_filepath: Align cameras with a filepath matching, using 'fileMatchingPattern'
|
|
* from_cameras_metadata: Align cameras with matching metadata, using 'metadataMatchingList'
|
|
* from_markers: Align from markers with the same Id
|
|
|
|
'''
|
|
|
|
inputs = [
|
|
desc.File(
|
|
name='input',
|
|
label='Input',
|
|
description='''SfMData file .''',
|
|
value='',
|
|
uid=[0],
|
|
),
|
|
desc.File(
|
|
name='reference',
|
|
label='Reference',
|
|
description='''Path to the scene used as the reference coordinate system.''',
|
|
value='',
|
|
uid=[0],
|
|
),
|
|
desc.ChoiceParam(
|
|
name='method',
|
|
label='Alignment Method',
|
|
description="Alignment Method:\n"
|
|
" * from_cameras_viewid: Align cameras with same view Id\n"
|
|
" * from_cameras_poseid: Align cameras with same pose Id\n"
|
|
" * from_cameras_filepath: Align cameras with a filepath matching, using 'fileMatchingPattern'\n"
|
|
" * from_cameras_metadata: Align cameras with matching metadata, using 'metadataMatchingList'\n"
|
|
" * from_markers: Align from markers with the same Id\n",
|
|
value='from_cameras_viewid',
|
|
values=['from_cameras_viewid', 'from_cameras_poseid', 'from_cameras_filepath', 'from_cameras_metadata', 'from_markers'],
|
|
exclusive=True,
|
|
uid=[0],
|
|
),
|
|
desc.StringParam(
|
|
name='fileMatchingPattern',
|
|
label='File Matching Pattern',
|
|
description='Matching regular expression for the "from_cameras_filepath" method. '
|
|
'You should capture specific parts of the filepath with parenthesis to define matching elements.\n'
|
|
'Some examples of patterns:\n'
|
|
r' - Match the filename without extension (default value): ".*\/(.*?)\.\w{3}"' + '\n' +
|
|
r' - Match the filename suffix after "_": ".*\/.*(_.*?\.\w{3})"' + '\n' +
|
|
r' - Match the filename prefix before "_": ".*\/(.*?)_.*\.\w{3}"',
|
|
value=r'.*\/(.*?)\.\w{3}',
|
|
uid=[0],
|
|
),
|
|
desc.ListAttribute(
|
|
elementDesc=desc.File(
|
|
name="metadataMatching",
|
|
label="Metadata",
|
|
description="",
|
|
value="",
|
|
uid=[0],
|
|
),
|
|
name="metadataMatchingList",
|
|
label="Metadata Matching List",
|
|
description='List of metadata that should match to create the correspondences. If the list is empty, the default value will be used: ["Make", "Model", "Exif:BodySerialNumber", "Exif:LensSerialNumber"].',
|
|
),
|
|
desc.BoolParam(
|
|
name='applyScale',
|
|
label='Scale',
|
|
description='Apply scale transformation.',
|
|
value=True,
|
|
uid=[0]
|
|
),
|
|
desc.BoolParam(
|
|
name='applyRotation',
|
|
label='Rotation',
|
|
description='Apply rotation transformation.',
|
|
value=True,
|
|
uid=[0]
|
|
),
|
|
desc.BoolParam(
|
|
name='applyTranslation',
|
|
label='Translation',
|
|
description='Apply translation transformation.',
|
|
value=True,
|
|
uid=[0]
|
|
),
|
|
desc.ChoiceParam(
|
|
name='verboseLevel',
|
|
label='Verbose Level',
|
|
description='''verbosity level (fatal, error, warning, info, debug, trace).''',
|
|
value='info',
|
|
values=['fatal', 'error', 'warning', 'info', 'debug', 'trace'],
|
|
exclusive=True,
|
|
uid=[],
|
|
),
|
|
]
|
|
|
|
outputs = [
|
|
desc.File(
|
|
name='output',
|
|
label='SfMData File',
|
|
description='SfMData file.',
|
|
value=lambda attr: desc.Node.internalFolder + (os.path.splitext(os.path.basename(attr.node.input.value))[0] or 'sfmData') + '.abc',
|
|
uid=[],
|
|
),
|
|
desc.File(
|
|
name='outputViewsAndPoses',
|
|
label='Poses',
|
|
description='''Path to the output sfmdata file with cameras (views and poses).''',
|
|
value=desc.Node.internalFolder + 'cameras.sfm',
|
|
uid=[],
|
|
),
|
|
]
|