Merge pull request #2658 from alicevision/dev/maskProcessing

Mask processing node

meshroom/nodes/aliceVision/MaskProcessing.py

@@ -0,0 +1,85 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+from meshroom.core.utils import DESCRIBER_TYPES, VERBOSE_LEVEL
+
+import os.path
+
+class MaskProcessingNodeSize(desc.DynamicNodeSize):
+    """
+    MaskProcessingNodeSize expresses a dependency to multiple input attributess to define
+    the size of a Node in terms of individual tasks for parallelization.
+    """
+    def __init__(self, param):
+        self._params = param
+
+    def computeSize(self, node):
+        
+        size = 0
+
+        for input in node.attribute(self._params).value:
+            paramName = input.getFullName()
+            param = node.attribute(paramName)
+            if param.isLink:
+                size = max(size, param.getLinkParam().node.size)
+        
+        return size
+
+
+class MaskProcessing(desc.AVCommandLineNode):
+    commandLine = 'aliceVision_maskProcessing {allParams}'
+    size = MaskProcessingNodeSize("inputs")
+
+    category = 'Utils'
+    documentation = '''
+    Perform operations on a list of masks with the same names
+    '''
+
+    inputs = [
+        desc.ListAttribute(
+            elementDesc=desc.File(
+                name="input",
+                label="Input Directory",
+                description="A directory with a set of mask.",
+                value="",
+            ),
+            name="inputs",
+            label="Input directories",
+            description="A set of directories containing masks with the same names.",
+            exposed=True,
+        ),
+        desc.ChoiceParam(
+            name="operator",
+            label="Operator",
+            description="Operator: Binary operator\n"
+                        "OR : applies binary OR between all the masks\n"
+                        "AND : applies binary AND between all the masks\n"
+                        "NOT : applies binary NOT to the first mask in the list\n",
+            value="and",
+            values=["or", "and", "not"],
+        ),
+        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="Output",
+            description="Path to the output directory.",
+            value=desc.Node.internalFolder,
+        ),
+        desc.File(
+            name="masks",
+            label="Masks",
+            description="Processed segmentation masks.",
+            semantic="imageList",
+            value= desc.Node.internalFolder + "*.exr",
+            group="",
+        ),
+    ]