diff --git a/src/go/engine.py b/src/go/engine.py --- a/src/go/engine.py +++ b/src/go/engine.py @@ -15,7 +15,7 @@ class SpecGo(core.Go): Additions can be taken as relevant right away. Deletions and replacements had to be captured, so we add their liberties. Also any non-missing stones of partially deleted (or replaced) groups had to be replayed, so add them too. - Needs to handle snapback. + Needs to handle snapback, throw-in. There's no end to what could be theoretically relevant, but such sequences are long and we will pretend they won't happen.""" res=(set(),set()) for d in diff: @@ -23,7 +23,10 @@ class SpecGo(core.Go): colorKey=(1-color)>>1 # {-1,1}->{1,0} if action!="-" and (r,c) not in res[colorKey]: res[colorKey].add((r,c)) + for (ri,ci) in self.listNeighbours(r,c): # in case a stone was played and captured. !! might want to add even more + res[1-colorKey].add((ri,ci)) # this is rather sloppy but correct. the time will show if it is effective enough + # just floodFill from the current intersection, add everything you find and also all the neighbours to be sure if action!="+" and (r,c) not in res[colorKey] and (r,c) not in res[1-colorKey]: self._helper.clear() self._helper.floodFill(color if action=="-" else 1-color, r, c) @@ -31,20 +34,17 @@ class SpecGo(core.Go): for (ri,ci) in self._helper.getContinuousArea(): res[colorKey].add((ri,ci)) res[1-colorKey].add((ri,ci)) - if ri>0: - res[colorKey].add((ri-1,ci)) - res[1-colorKey].add((ri-1,ci)) - if ri+10: - res[colorKey].add((ri,ci-1)) - res[1-colorKey].add((ri,ci-1)) - if ci+10: yield (r-1,c) + if r+10: yield (r,c-1) + if c+10 else None, g.board[r][c+1] if c+1