Creation of the sfm Checking node

Author: Fabien Servant

meshroom/nodes/aliceVision/SfmChecking.py

@@ -0,0 +1,82 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+from meshroom.core.utils import VERBOSE_LEVEL
+import os.path
+
+
+
+class SfmChecking(desc.Node):
+    
+    category = 'Utils'
+    documentation = '''
+    Check an input Sfm for validity.
+    Throw an error if the sfm does not satisfy constraints
+    '''
+
+    inputs = [
+        desc.File(
+            name="input",
+            label="SfMData",
+            description="Input SfMData file.",
+            value="",
+        ),
+        desc.FloatParam(
+            name="posecompletion",
+            label="Completion percentage",
+            description="Minimal percent of the views reconstructed.",
+            value=80.0,
+            range=(0.0, 100.0, 1.0),
+        ),
+        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="SfM File",
+            description="Path to the output SfM file.",
+            value=desc.Node.internalFolder + "sfmData.abc",
+        )
+    ]
+
+    def processChunk(self, chunk):
+        from pyalicevision import sfmData as avsfmdata
+        from pyalicevision import sfmDataIO as avsfmdataio
+
+        error = False
+
+        chunk.logManager.start(chunk.node.verboseLevel.value)
+        chunk.logger.error("open input")
+        data = avsfmdata.SfMData()
+        ret = avsfmdataio.load(data, chunk.node.input.value, avsfmdataio.ALL)
+        if not ret:
+            chunk.logger.error("Cannot open input")
+            chunk.logManager.end()
+            raise RuntimeError()
+
+        total = len(data.getViews())
+        valid = len(data.getValidViews())
+        ratio = (100.0 * float(valid))/float(total)
+
+        chunk.logger.info(f"Total views : {total}")
+        chunk.logger.info(f"Reconstructed views : {valid}")
+        chunk.logger.info(f"Percentage : {ratio}")
+
+        if ratio < chunk.node.posecompletion.value:
+            chunk.logger.error("Percentage of reconstructed views is insufficient")
+            error = True
+
+        avsfmdataio.save(data, chunk.node.output.value, avsfmdataio.ALL)
+        
+        chunk.logManager.end()
+
+        if error:
+            raise RuntimeError()
+