107 lines
3.7 KiB
Python
107 lines
3.7 KiB
Python
from PIL import Image
|
|
import sys
|
|
|
|
sys.setrecursionlimit(10**5)
|
|
|
|
img = Image.open("laketour2.png")
|
|
img = Image.open("laketour2_fju.png")
|
|
# img = Image.open("exlaketour.png")
|
|
|
|
noir = (0, 0, 0)
|
|
rouge = (255, 0, 0)
|
|
vert = (0, 255, 0)
|
|
bleu = (0, 0, 255)
|
|
|
|
width, height = img.size
|
|
|
|
img = img.convert("RGB")
|
|
|
|
def fill(x, y, pixels_to_match, pixel_to_fill, remplir_countour=False):
|
|
global img
|
|
global width
|
|
global height
|
|
|
|
res = 0
|
|
|
|
p = img.getpixel((x,y))
|
|
if remplir_countour or p in pixels_to_match:
|
|
img.putpixel((x,y), pixel_to_fill)
|
|
res += 1
|
|
|
|
contour = list()
|
|
|
|
if p in pixels_to_match:
|
|
if y > 0 and img.getpixel((x,y-1)) in pixels_to_match:
|
|
pixels, new_contour = fill(x, y-1, pixels_to_match, pixel_to_fill, remplir_countour)
|
|
res += pixels
|
|
contour += new_contour
|
|
if y > 0 and img.getpixel((x,y-1)) not in pixels_to_match and img.getpixel((x,y-1)) != pixel_to_fill:
|
|
contour.append((x,y-1))
|
|
if remplir_countour:
|
|
img.putpixel((x,y-1), pixel_to_fill)
|
|
res += 1
|
|
|
|
if x < width-1 and img.getpixel((x+1,y)) in pixels_to_match:
|
|
pixels, new_contour = fill(x+1, y, pixels_to_match, pixel_to_fill, remplir_countour)
|
|
res += pixels
|
|
contour += new_contour
|
|
if x < width-1 and img.getpixel((x+1,y)) not in pixels_to_match and img.getpixel((x+1,y)) != pixel_to_fill:
|
|
contour.append((x+1,y))
|
|
if remplir_countour:
|
|
img.putpixel((x+1,y), pixel_to_fill)
|
|
res += 1
|
|
|
|
if x > 0 and img.getpixel((x-1,y)) in pixels_to_match:
|
|
pixels, new_contour = fill(x-1, y, pixels_to_match, pixel_to_fill, remplir_countour)
|
|
res += pixels
|
|
contour += new_contour
|
|
if x > 0 and img.getpixel((x-1,y)) not in pixels_to_match and img.getpixel((x-1,y)) != pixel_to_fill:
|
|
contour.append((x-1,y))
|
|
if remplir_countour:
|
|
img.putpixel((x-1,y), pixel_to_fill)
|
|
res += 1
|
|
|
|
if y < height-1 and img.getpixel((x,y+1)) in pixels_to_match:
|
|
pixels, new_contour = fill(x, y+1, pixels_to_match, pixel_to_fill, remplir_countour)
|
|
res += pixels
|
|
contour += new_contour
|
|
if y < height-1 and img.getpixel((x,y+1)) not in pixels_to_match and img.getpixel((x,y+1)) != pixel_to_fill:
|
|
contour.append((x,y+1))
|
|
if remplir_countour:
|
|
img.putpixel((x,y+1), pixel_to_fill)
|
|
res += 1
|
|
|
|
if x == 0 or y == 0 or x == width-1 or y == height-1:
|
|
contour.append(None)
|
|
|
|
return res, contour
|
|
|
|
#On commence par remplir toutes les zones en vert, une à une
|
|
#en regardant si elles ont un seul type de contour ou plusieurs
|
|
#si plusieurs, ce ne sont pas des lacs, on les passe en rouge
|
|
#sinon on les passe de la couleur du contour
|
|
for x in range(width):
|
|
for y in range(height):
|
|
if img.getpixel((x, y)) == noir:
|
|
print((x,y))
|
|
pixels, contour = fill(x, y, [noir,], vert) # on passe en vert les zones noires et on liste les points du contour
|
|
contour_colors = list({img.getpixel((p[0], p[1])) for p in contour if p != None})
|
|
# print(pixels, contour, contour_colors)
|
|
if None in contour:
|
|
fill(x, y, [vert,], rouge)
|
|
elif len(contour_colors) != 1:
|
|
fill(x, y, [vert,], rouge)
|
|
else:
|
|
fill(x, y, [vert,], contour_colors[0])
|
|
# img.show()
|
|
|
|
fleches = 0
|
|
for x in range(width):
|
|
for y in range(height):
|
|
if img.getpixel((x, y)) != rouge:
|
|
fleches += img.getpixel((x, y))[0]
|
|
print(fleches, "flèches")
|
|
|
|
img.save("laketour2_tmp.png")
|
|
img.show()
|