From d18cadc5ec4afd2126e272002789b12c604f5b60 Mon Sep 17 00:00:00 2001
From: Fabien Servant <fabien.servant@technicolor.com>
Date: Thu, 5 Oct 2023 11:15:56 +0200
Subject: [PATCH] Update chunks with merge

---
 meshroom/nodes/aliceVision/SfMMerge.py | 28 +++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/meshroom/nodes/aliceVision/SfMMerge.py b/meshroom/nodes/aliceVision/SfMMerge.py
index df0fa3eb8c..2152c587a2 100644
--- a/meshroom/nodes/aliceVision/SfMMerge.py
+++ b/meshroom/nodes/aliceVision/SfMMerge.py
@@ -3,9 +3,35 @@
 from meshroom.core import desc
 import os.path
 
+
+class MergeNodeSize(object):
+    """
+    MergeNodeSize expresses a dependency to two input attributess to define
+    the size of a Node in terms of individual tasks for parallelization.
+    """
+    def __init__(self, param1, param2):
+        self._param1 = param1
+        self._param2 = param2
+
+    def computeSize(self, node):
+        
+        size1 = 0
+        size2 = 0
+
+        param1 = node.attribute(self._param1)
+        if param1.isLink:
+            size1 = param1.getLinkParam().node.size
+
+        param2 = node.attribute(self._param2)
+        if param2.isLink:
+            size2 = param2.getLinkParam().node.size
+        
+        return size1 + size2
+
+
 class SfmMerge(desc.AVCommandLineNode):
     commandLine = 'aliceVision_sfmMerge {allParams}'
-    size = desc.DynamicNodeSize('firstinput')
+    size = MergeNodeSize('firstinput', 'secondinput')
 
     category = 'Utils'
     documentation = '''