python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-05-02 15:03:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

improved comments and one logic according to PR 3294 discussion

Browse Source
This commit is contained in:
yo1995 2018-09-04 12:03:20 -04:00
parent 5e2d6c951a
commit 3cf1a4ea87
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/PIL/ImageDraw.py
View File

@ -352,7 +352,7 @@ def floodfill(image, xy, value, border=None, thresh=0):
except (ValueError, IndexError): except (ValueError, IndexError):
return # seed point outside image return # seed point outside image
edge = {(x, y)} edge = {(x, y)}
full_edge = set() # use a set to record each unique pixel processed full_edge = set() # use a set to keep record of current and previous edge pixels to reduce memory consumption
while edge: while edge:
new_edge = set() new_edge = set()
for (x, y) in edge: # 4 adjacent method for (x, y) in edge: # 4 adjacent method
@ -364,6 +364,7 @@ def floodfill(image, xy, value, border=None, thresh=0):
except (ValueError, IndexError): except (ValueError, IndexError):
pass pass
else: else:
full_edge.add((s, t))
if border is None: if border is None:
fill = _color_diff(p, background) <= thresh fill = _color_diff(p, background) <= thresh
else: else:
@ -371,8 +372,7 @@ def floodfill(image, xy, value, border=None, thresh=0):
if fill: if fill:
pixel[s, t] = value pixel[s, t] = value
new_edge.add((s, t)) new_edge.add((s, t))
full_edge.add((s, t)) full_edge = edge # discard pixels proceeded
full_edge = edge # do not record useless pixels to reduce memory consumption
edge = new_edge edge = new_edge