From e73505273f6620b116403a203d17211e9c13715a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radek=20Hu=C5=A1ek?= <PitelVonSacek@gmail.com> Date: Sun, 13 Dec 2015 19:27:12 +0100 Subject: [PATCH] guidedSubdivisionIterator: add context parametr --- groupConnectivity.pyx | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/groupConnectivity.pyx b/groupConnectivity.pyx index 69b87ef..edcf2a1 100644 --- a/groupConnectivity.pyx +++ b/groupConnectivity.pyx @@ -163,7 +163,7 @@ def subdivisionIterator(G, edges = None): yield L -def guidedSubdivisionIterator(G, edges = None, guide = lambda H: (True, None)): +def guidedSubdivisionIterator(G, edges = None, guide = lambda H: (True, None), context = None): class DeadEnd(Exception): pass @@ -172,7 +172,10 @@ def guidedSubdivisionIterator(G, edges = None, guide = lambda H: (True, None)): def impl(G, edges): if len(edges) == 0: - cont, val = guide(G) + if context is None: + cont, val = guide(G) + else: + cont, val = guide(G, context) if not cont: raise DeadEnd() yield val -- GitLab