3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-05-21 21:16:29 +02:00
Code Issues Projects Releases Packages Wiki Activity

[core] add GraphVisitMessage exceptions mecanism

Browse source 
allow to stop branch or graph visit by throwing specific exceptions in visitor callbacks
This commit is contained in:
Yann Lanthony 2018-07-18 16:12:46 +02:00
parent 52d1d1fb89
commit ab4e82aa88
2 changed files with 27 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
meshroom/core/graph.py
View file

@ -14,7 +14,7 @@ import meshroom
import meshroom.core
from meshroom.common import BaseObject, DictModel, Slot, Signal, Property
from meshroom.core.attribute import Attribute, ListAttribute
from meshroom.core.exception import UnknownNodeTypeError
from meshroom.core.exception import StopGraphVisit, StopBranchVisit
from meshroom.core.node import node_factory, Status, Node, CompatibilityNode
# Replace default encoder to support Enums
@ -586,10 +586,19 @@ class Graph(BaseObject):
assert not self.dirtyTopology
nodes = sorted(nodes, key=lambda item: item.depth)
for node in nodes:
self.dfsVisit(node, visitor, colors, nodeChildren, longestPathFirst)
try:
for node in nodes:
self.dfsVisit(node, visitor, colors, nodeChildren, longestPathFirst)
except StopGraphVisit:
pass
def dfsVisit(self, u, visitor, colors, nodeChildren, longestPathFirst):
try:
self._dfsVisit(u, visitor, colors, nodeChildren, longestPathFirst)
except StopBranchVisit:
pass
def _dfsVisit(self, u, visitor, colors, nodeChildren, longestPathFirst):
colors[u] = GRAY
visitor.discoverVertex(u, self)
# d_time[u] = time = time + 1