mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-04-29 10:17:27 +02:00
266 lines
11 KiB
Python
266 lines
11 KiB
Python
__version__ = "3.1"
|
|
|
|
from meshroom.core import desc
|
|
from meshroom.core.utils import DESCRIBER_TYPES, VERBOSE_LEVEL
|
|
|
|
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
|
|
* from_lineup: Align using a camera pose (json line up file), tracks and a mesh
|
|
'''
|
|
|
|
inputs = [
|
|
desc.File(
|
|
name="input",
|
|
label="Input",
|
|
description="SfMData file.",
|
|
value="",
|
|
),
|
|
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"
|
|
" - from_lineup: Defines coordinate system using lineup json file.\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", "from_lineup", "align_ground"],
|
|
),
|
|
desc.File(
|
|
name="lineUp",
|
|
label="Line Up File",
|
|
description="LineUp Json file.",
|
|
value="",
|
|
enabled=lambda node: node.method.value == "from_lineup"
|
|
),
|
|
desc.File(
|
|
name="tracksFile",
|
|
label="Tracks File",
|
|
description="Tracks file for lineup.",
|
|
value="",
|
|
enabled=lambda node: node.method.value == "from_lineup"
|
|
),
|
|
desc.File(
|
|
name="objectFile",
|
|
label="Mesh File",
|
|
description="Mesh file for lineup.",
|
|
value="",
|
|
enabled=lambda node: node.method.value == "from_lineup"
|
|
),
|
|
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="",
|
|
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,
|
|
range=(-20.0, 20.0, 0.01),
|
|
),
|
|
desc.FloatParam(
|
|
name="y",
|
|
label="y",
|
|
description="Y offset.",
|
|
value=0.0,
|
|
range=(-20.0, 20.0, 0.01),
|
|
),
|
|
desc.FloatParam(
|
|
name="z",
|
|
label="z",
|
|
description="Z offset.",
|
|
value=0.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,
|
|
range=(-90.0, 90.0, 1.0),
|
|
),
|
|
desc.FloatParam(
|
|
name="y",
|
|
label="y",
|
|
description="Euler Y rotation.",
|
|
value=0.0,
|
|
range=(-180.0, 180.0, 1.0),
|
|
),
|
|
desc.FloatParam(
|
|
name="z",
|
|
label="z",
|
|
description="Euler Z rotation.",
|
|
value=0.0,
|
|
range=(-180.0, 180.0, 1.0),
|
|
),
|
|
],
|
|
joinChar=",",
|
|
),
|
|
desc.FloatParam(
|
|
name="manualScale",
|
|
label="Scale",
|
|
description="Uniform scale.",
|
|
value=1.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).",
|
|
values=DESCRIBER_TYPES,
|
|
value=["sift", "dspsift", "akaze"],
|
|
exclusive=False,
|
|
joinChar=",",
|
|
),
|
|
desc.FloatParam(
|
|
name="scale",
|
|
label="Additional Scale",
|
|
description="Additional scale to apply.",
|
|
value=1.0,
|
|
range=(0.0, 100.0, 0.1),
|
|
),
|
|
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,
|
|
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,
|
|
range=(-2.0, 2.0, 1.0),
|
|
),
|
|
desc.FloatParam(
|
|
name="y",
|
|
label="y",
|
|
description="Y coordinates for the marker.",
|
|
value=0.0,
|
|
range=(-2.0, 2.0, 1.0),
|
|
),
|
|
desc.FloatParam(
|
|
name="z",
|
|
label="z",
|
|
description="Z coordinates for the marker.",
|
|
value=0.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,
|
|
enabled=lambda node: node.method.value != "manual",
|
|
),
|
|
desc.BoolParam(
|
|
name="applyRotation",
|
|
label="Rotation",
|
|
description="Apply rotation transformation.",
|
|
value=True,
|
|
enabled=lambda node: node.method.value != "manual",
|
|
),
|
|
desc.BoolParam(
|
|
name="applyTranslation",
|
|
label="Translation",
|
|
description="Apply translation transformation.",
|
|
value=True,
|
|
enabled=lambda node: node.method.value != "manual",
|
|
),
|
|
desc.ChoiceParam(
|
|
name="verboseLevel",
|
|
label="Verbose Level",
|
|
description="Verbosity level (fatal, error, warning, info, debug, trace).",
|
|
values=VERBOSE_LEVEL,
|
|
value="info",
|
|
),
|
|
]
|
|
|
|
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",
|
|
),
|
|
desc.File(
|
|
name="outputViewsAndPoses",
|
|
label="Poses",
|
|
description="Path to the output SfMData file with cameras (views and poses).",
|
|
value=desc.Node.internalFolder + "cameras.sfm",
|
|
),
|
|
]
|