computeSize for sfmMerge

meshroom/nodes/aliceVision/SfMMerge.py

@@ -5,9 +5,29 @@
 
 import os.path
 
+class MergeNodeSize(desc.DynamicNodeSize):
+    """
+    MergeNodeSize 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)
+            size = size + param.getLinkParam().node.size
+
+        return size
+
 
 class SfMMerge(desc.AVCommandLineNode):
     commandLine = 'aliceVision_sfmMerge {allParams}'
+    size = MergeNodeSize("inputs")
 
     category = 'Utils'
     documentation = '''