From 40a39164476b27d46958cda990c9330e2f75a906 Mon Sep 17 00:00:00 2001
From: Fabien Castan <fabcastan@gmail.com>
Date: Wed, 27 Mar 2024 21:29:39 +0100
Subject: [PATCH] [nodes] New SelectConnectedViews node

Select Connected Views based on SfM landmarks.
---
 .../nodes/aliceVision/SelectConnectedViews.py | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 meshroom/nodes/aliceVision/SelectConnectedViews.py

diff --git a/meshroom/nodes/aliceVision/SelectConnectedViews.py b/meshroom/nodes/aliceVision/SelectConnectedViews.py
new file mode 100644
index 00000000..329a7dfa
--- /dev/null
+++ b/meshroom/nodes/aliceVision/SelectConnectedViews.py
@@ -0,0 +1,71 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+from meshroom.core.utils import VERBOSE_LEVEL
+
+
+class SelectConnectedViews(desc.AVCommandLineNode):
+    commandLine = 'aliceVision_selectConnectedViews {allParams}'
+
+    cpu = desc.Level.NORMAL
+    ram = desc.Level.NORMAL
+
+    category = 'Dense Reconstruction'
+    documentation = '''
+Select Connected Views based on SfM landmarks.
+'''
+
+    inputs = [
+        desc.File(
+            name="input",
+            label="SfMData",
+            description="Input SfMData file.",
+            value="",
+            uid=[0],
+        ),
+        desc.IntParam(
+            name="maxTCams",
+            label="Max Nb Neighbour Cameras",
+            description="Maximum number of neighbour cameras per image.",
+            value=10,
+            range=(1, 20, 1),
+            uid=[0],
+        ),
+        desc.FloatParam(
+            name="minViewAngle",
+            label="Min View Angle",
+            description="Minimum angle between two views (select the neighbouring cameras, select depth planes from epipolar segment point).",
+            value=2.0,
+            range=(0.0, 10.0, 0.1),
+            uid=[0],
+            advanced=True,
+        ),
+        desc.FloatParam(
+            name="maxViewAngle",
+            label="Max View Angle",
+            description="Maximum angle between two views (select the neighbouring cameras, select depth planes from epipolar segment point).",
+            value=70.0,
+            range=(10.0, 120.0, 1.0),
+            uid=[0],
+            advanced=True,
+        ),
+        desc.ChoiceParam(
+            name="verboseLevel",
+            label="Verbose Level",
+            description="Verbosity level (fatal, error, warning, info, debug, trace).",
+            values=VERBOSE_LEVEL,
+            value="info",
+            exclusive=True,
+            uid=[],
+        ),
+    ]
+
+    outputs = [
+        desc.File(
+            name="output",
+            label="Connected Views",
+            description="List of connected views in a text file.",
+            value=desc.Node.internalFolder + "connectedViews.txt",
+            uid=[],
+        ),
+    ]