Merge pull request #2696 from alicevision/dev/surveyInjecting

Node to inject survey points in the SFM

Author: fabiencastan

meshroom/core/utils.py

@@ -5,7 +5,7 @@
                "Log3G10 REDWideGamutRGB", "Linear Venice S-Gamut3.Cine", "S-Log3 Venice S-Gamut3.Cine", "no_conversion"]
 
 DESCRIBER_TYPES = ["sift", "sift_float", "sift_upright", "dspsift", "akaze", "akaze_liop", "akaze_mldb", "cctag3",
-                   "cctag4", "sift_ocv", "akaze_ocv", "tag16h5", "unknown"]
+                   "cctag4", "sift_ocv", "akaze_ocv", "tag16h5", "survey", "unknown"]
 
 EXR_STORAGE_DATA_TYPE = ["float", "half", "halfFinite", "auto"]
 

meshroom/nodes/aliceVision/ConvertSfMFormat.py

@@ -80,6 +80,12 @@ class ConvertSfMFormat(desc.AVCommandLineNode):
             description="Export observations.",
             value=True,
         ),
+        desc.BoolParam(
+            name="surveys",
+            label="Surveys",
+            description="Export surveys.",
+            value=True,
+        ),
         desc.ChoiceParam(
             name="verboseLevel",
             label="Verbose Level",

meshroom/nodes/aliceVision/SfMSurveyInjecting.py

@@ -0,0 +1,47 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+from meshroom.core.utils import VERBOSE_LEVEL
+
+import json
+
+class SfMSurveyInjecting(desc.AVCommandLineNode):
+
+    commandLine = "aliceVision_sfmSurveyInjecting {allParams}"
+    size = desc.DynamicNodeSize("input")
+    
+    category = "Utils"
+    documentation = """
+Use a JSON file to inject survey measurements inside the SfMData.
+"""
+
+    inputs = [
+        desc.File(
+            name="input",
+            label="SfMData",
+            description="Input SfMData file.",
+            value="",
+        ),
+        desc.File(
+            name="surveyFilename",
+            label="Survey",
+            description="Input JSON file containing the survey.",
+            value="",
+        ),
+        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",
+            description="Path to the output SfM file.",
+            value="{nodeCacheFolder}/sfmData.abc",
+        ),
+    ]