Merge pull request #2214 from alicevision/dev/panoramaSfm

New nodes for panorama to sfm

meshroom/nodes/aliceVision/SfMMerge.py

@@ -0,0 +1,75 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+import os.path
+
+
+class MergeNodeSize(object):
+    """
+    MergeNodeSize expresses a dependency to two input attributess to define
+    the size of a Node in terms of individual tasks for parallelization.
+    """
+    def __init__(self, param1, param2):
+        self._param1 = param1
+        self._param2 = param2
+
+    def computeSize(self, node):
+
+        size1 = 0
+        size2 = 0
+
+        param1 = node.attribute(self._param1)
+        if param1.isLink:
+            size1 = param1.getLinkParam().node.size
+
+        param2 = node.attribute(self._param2)
+        if param2.isLink:
+            size2 = param2.getLinkParam().node.size
+
+        return size1 + size2
+
+
+class SfMMerge(desc.AVCommandLineNode):
+    commandLine = 'aliceVision_sfmMerge {allParams}'
+    size = MergeNodeSize('firstinput', 'secondinput')
+
+    category = 'Utils'
+    documentation = '''
+Merges two SfMData files into a single one. Fails if some UID is shared among them.
+'''
+
+    inputs = [
+        desc.File(
+            name="firstinput",
+            label="First SfMData",
+            description="First input SfMData file to merge.",
+            value="",
+            uid=[0],
+        ),
+        desc.File(
+            name="secondinput",
+            label="Second SfMData",
+            description="Second input SfMData file to merge.",
+            value="",
+            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",
+            description="Path to the output SfM file (in SfMData format).",
+            value=lambda attr: desc.Node.internalFolder + "sfmData.sfm",
+            uid=[],
+        ),
+    ]

meshroom/nodes/aliceVision/SfMToRig.py

@@ -0,0 +1,42 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+import os.path
+
+class SfMToRig(desc.AVCommandLineNode):
+    commandLine = 'aliceVision_sfmToRig {allParams}'
+    size = desc.DynamicNodeSize('input')
+
+    category = 'Utils'
+    documentation = '''
+Assumes the input SfMData describes a set of cameras capturing a scene at a common time. Transformd the set of cameras into a rig of cameras.
+'''
+
+    inputs = [
+        desc.File(
+            name="input",
+            label="SfMData",
+            description="Input SfMData file.",
+            value="",
+            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",
+            description="Path to the output SfM file (in SfMData format).",
+            value=lambda attr: desc.Node.internalFolder + "sfmData.sfm",
+            uid=[],
+        ),
+    ]