mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-06-08 13:51:57 +02:00
c498f48e1e
Use CamelCase for all labels, always end descriptions with periods, and replace the mixed use of single and double quotes with double quotes only.
215 lines
9.5 KiB
Python
215 lines
9.5 KiB
Python
__version__ = "3.1"
|
|
|
|
from meshroom.core import desc
|
|
|
|
import os.path
|
|
|
|
|
|
class SfMTransform(desc.AVCommandLineNode):
|
|
commandLine = 'aliceVision_sfmTransform {allParams}'
|
|
size = desc.DynamicNodeSize('input')
|
|
|
|
category = 'Utils'
|
|
documentation = '''
|
|
This node allows to change the coordinate system of one SfM scene.
|
|
|
|
The transformation can be based on:
|
|
* transformation: Apply a given transformation
|
|
* auto_from_cameras: Fit all cameras into a box [-1,1]
|
|
* auto_from_landmarks: Fit all landmarks into a box [-1,1]
|
|
* from_single_camera: Use a specific camera as the origin of the coordinate system
|
|
* from_markers: Align specific markers to custom coordinates
|
|
* from_gps: Align with the gps positions from the image metadata
|
|
* align_ground: Detect ground level and align to it
|
|
|
|
'''
|
|
|
|
inputs = [
|
|
desc.File(
|
|
name="input",
|
|
label="Input",
|
|
description="SfMData file.",
|
|
value="",
|
|
uid=[0],
|
|
),
|
|
desc.ChoiceParam(
|
|
name="method",
|
|
label="Transformation Method",
|
|
description="Transformation method:\n"
|
|
" - transformation: Apply a given transformation.\n"
|
|
" - manual: Apply the gizmo transformation (show the transformed input).\n"
|
|
" - auto: Determines scene orientation from the cameras' X axis, determines north and scale from GPS information if available, and defines ground level from the point cloud.\n"
|
|
" - auto_from_cameras: Defines coordinate system from cameras.\n"
|
|
" - auto_from_cameras_x_axis: Determines scene orientation from the cameras' X axis.\n"
|
|
" - auto_from_landmarks: Defines coordinate system from landmarks.\n"
|
|
" - from_single_camera: Defines the coordinate system from the camera specified by --tranformation.\n"
|
|
" - from_center_camera: Defines the coordinate system from the camera closest to the center of the reconstruction.\n"
|
|
" - from_markers: Defines the coordinate system from markers specified by --markers.\n"
|
|
" - from_gps: Defines coordinate system from GPS metadata.\n"
|
|
" - align_ground: Defines ground level from the point cloud density. It assumes that the scene is oriented.",
|
|
value="auto",
|
|
values=["transformation", "manual", "auto", "auto_from_cameras", "auto_from_cameras_x_axis", "auto_from_landmarks", "from_single_camera", "from_center_camera", "from_markers", "from_gps", "align_ground"],
|
|
exclusive=True,
|
|
uid=[0],
|
|
),
|
|
desc.StringParam(
|
|
name="transformation",
|
|
label="Transformation",
|
|
description="Required only for 'transformation' and 'from_single_camera' methods:\n"
|
|
" - transformation: Align [X,Y,Z] to +Y-axis, rotate around Y by R deg, scale by S; syntax: X,Y,Z;R;S\n"
|
|
" - from_single_camera: Camera UID or simplified regular expression to match image filepath (like '*camera2*.jpg').",
|
|
value="",
|
|
uid=[0],
|
|
enabled=lambda node: node.method.value == "transformation" or node.method.value == "from_single_camera" or node.method.value == "auto_from_cameras_x_axis",
|
|
),
|
|
desc.GroupAttribute(
|
|
name="manualTransform",
|
|
label="Manual Transform (Gizmo)",
|
|
description="Translation, rotation (Euler ZXY) and uniform scale.",
|
|
groupDesc=[
|
|
desc.GroupAttribute(
|
|
name="manualTranslation",
|
|
label="Translation",
|
|
description="Translation in space.",
|
|
groupDesc=[
|
|
desc.FloatParam(
|
|
name="x", label="x", description="X offset.",
|
|
value=0.0,
|
|
uid=[0],
|
|
range=(-20.0, 20.0, 0.01)
|
|
),
|
|
desc.FloatParam(
|
|
name="y", label="y", description="Y offset.",
|
|
value=0.0,
|
|
uid=[0],
|
|
range=(-20.0, 20.0, 0.01)
|
|
),
|
|
desc.FloatParam(
|
|
name="z", label="z", description="Z offset.",
|
|
value=0.0,
|
|
uid=[0],
|
|
range=(-20.0, 20.0, 0.01)
|
|
)
|
|
],
|
|
joinChar=","
|
|
),
|
|
desc.GroupAttribute(
|
|
name="manualRotation",
|
|
label="Euler Rotation",
|
|
description="Rotation in Euler angles.",
|
|
groupDesc=[
|
|
desc.FloatParam(
|
|
name="x", label="x", description="Euler X rotation.",
|
|
value=0.0,
|
|
uid=[0],
|
|
range=(-90.0, 90.0, 1.0)
|
|
),
|
|
desc.FloatParam(
|
|
name="y", label="y", description="Euler Y rotation.",
|
|
value=0.0,
|
|
uid=[0],
|
|
range=(-180.0, 180.0, 1.0)
|
|
),
|
|
desc.FloatParam(
|
|
name="z", label="z", description="Euler Z rotation.",
|
|
value=0.0,
|
|
uid=[0],
|
|
range=(-180.0, 180.0, 1.0)
|
|
)
|
|
],
|
|
joinChar=","
|
|
),
|
|
desc.FloatParam(
|
|
name="manualScale",
|
|
label="Scale",
|
|
description="Uniform scale.",
|
|
value=1.0,
|
|
uid=[0],
|
|
range=(0.0, 20.0, 0.01)
|
|
)
|
|
],
|
|
joinChar=",",
|
|
enabled=lambda node: node.method.value == "manual",
|
|
),
|
|
desc.ChoiceParam(
|
|
name="landmarksDescriberTypes",
|
|
label="Landmarks Describer Types",
|
|
description="Image describer types used to compute the mean of the point cloud (only for 'landmarks' method).",
|
|
value=["sift", "dspsift", "akaze"],
|
|
values=["sift", "sift_float", "sift_upright", "dspsift", "akaze", "akaze_liop", "akaze_mldb", "cctag3", "cctag4", "sift_ocv", "akaze_ocv", "tag16h5", "unknown"],
|
|
exclusive=False,
|
|
uid=[0],
|
|
joinChar=",",
|
|
),
|
|
desc.FloatParam(
|
|
name="scale",
|
|
label="Additional Scale",
|
|
description="Additional scale to apply.",
|
|
value=1.0,
|
|
range=(0.0, 100.0, 0.1),
|
|
uid=[0],
|
|
),
|
|
desc.ListAttribute(
|
|
name="markers",
|
|
elementDesc=desc.GroupAttribute(name="markerAlign", label="Marker Align", description="", joinChar=":", groupDesc=[
|
|
desc.IntParam(name="markerId", label="Marker", description="Marker ID.", value=0, uid=[0], range=(0, 32, 1)),
|
|
desc.GroupAttribute(name="markerCoord", label="Coord", description="Marker coordinates.", joinChar=",", groupDesc=[
|
|
desc.FloatParam(name="x", label="x", description="X coordinates for the marker.", value=0.0, uid=[0], range=(-2.0, 2.0, 1.0)),
|
|
desc.FloatParam(name="y", label="y", description="Y coordinates for the marker.", value=0.0, uid=[0], range=(-2.0, 2.0, 1.0)),
|
|
desc.FloatParam(name="z", label="z", description="Z coordinates for the marker.", value=0.0, uid=[0], range=(-2.0, 2.0, 1.0)),
|
|
])
|
|
]),
|
|
label="Markers",
|
|
description="Markers alignment points.",
|
|
),
|
|
desc.BoolParam(
|
|
name="applyScale",
|
|
label="Scale",
|
|
description="Apply scale transformation.",
|
|
value=True,
|
|
uid=[0],
|
|
enabled=lambda node: node.method.value != "manual",
|
|
),
|
|
desc.BoolParam(
|
|
name="applyRotation",
|
|
label="Rotation",
|
|
description="Apply rotation transformation.",
|
|
value=True,
|
|
uid=[0],
|
|
enabled=lambda node: node.method.value != "manual",
|
|
),
|
|
desc.BoolParam(
|
|
name="applyTranslation",
|
|
label="Translation",
|
|
description="Apply translation transformation.",
|
|
value=True,
|
|
uid=[0],
|
|
enabled=lambda node: node.method.value != "manual",
|
|
),
|
|
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="Aligned 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=[],
|
|
),
|
|
]
|