[submitter] Add tractor API

Author: Alxiice

meshroom/core/graph.py

@@ -1711,7 +1711,7 @@ def submitGraph(graph, submitter, toNodes=None, submitLabel="{projectName}"):
         res = sub.submit(nodesToProcess, edgesToProcess, graph.filepath, submitLabel=submitLabel)
         if res:
             for node in nodesToProcess:
-                node.submit()  # update node status
+                node.initStatusOnSubmit()  # update node status
     except Exception as e:
         logging.error(f"Error on submit : {e}")
 

meshroom/core/node.py

@@ -1315,7 +1315,8 @@ def updateStatusFromCache(self):
         # logging.warning(f"updateStatusFromCache: {self.name}, status: {s} => {self.globalStatus}")
         self.updateOutputAttr()
 
-    def submit(self, forceCompute=False):
+    def initStatusOnSubmit(self, forceCompute=False):
+        """ Prepare chunks status when the node is in a graph that was submitted """
         for chunk in self._chunks:
             if forceCompute or chunk.status.status != Status.SUCCESS:
                 chunk._status.setNode(self)

meshroom/core/taskManager.py

@@ -434,7 +434,7 @@ def submit(self, graph, submitter=None, toNodes=None, submitLabel="{projectName}
             if res:
                 for node in nodesToProcess:
                     node.destroyed.connect(lambda obj=None, name=node.name: self.onNodeDestroyed(obj, name))
-                    node.submit()  # update node status
+                    node.initStatusOnSubmit()  # update node status
             self._nodes.update(nodesToProcess)
             self._nodesExtern.extend(nodesToProcess)
 

meshroom/submitters/simpleFarmSubmitter.py

@@ -75,15 +75,11 @@ def createTask(self, meshroomFile, node):
         allRequirements.extend(self.config['RAM'].get(node.nodeDesc.ram.name, []))
         allRequirements.extend(self.config['GPU'].get(node.nodeDesc.gpu.name, []))
 
+        executable = 'meshroom_compute' if self.reqPackages else os.path.join(binDir, 'meshroom_compute')
+        taskCommand = f"{executable} --node {node.name} \"{meshroomFile}\" {parallelArgs} --extern"
         task = simpleFarm.Task(
-            name=node.name,
-            command='{exe} --node {nodeName} "{meshroomFile}" {parallelArgs} --extern'.format(
-                exe='meshroom_compute' if self.reqPackages else os.path.join(binDir, 'meshroom_compute'),
-                nodeName=node.name, meshroomFile=meshroomFile, parallelArgs=parallelArgs),
-            tags=tags,
-            rezPackages=self.reqPackages,
-            requirements={'service': str(','.join(allRequirements))},
-            **arguments)
+            name=node.name, command=taskCommand, tags=tags, rezPackages=self.reqPackages,
+            requirements={'service': str(','.join(allRequirements))}, **arguments)
         return task
 
     def submit(self, nodes, edges, filepath, submitLabel="{projectName}"):