46 lines
1.3 KiB
Python
Executable File
46 lines
1.3 KiB
Python
Executable File
#!/usr/bin/python3
|
|
import time
|
|
|
|
t0 = time.clock_gettime(time.CLOCK_MONOTONIC)
|
|
|
|
|
|
from PIL import Image
|
|
im = Image.open("bodensee-m3.png")
|
|
print(time.clock_gettime(time.CLOCK_MONOTONIC) - t0)
|
|
#print(im.format, im.size, im.mode)
|
|
region_colors = {
|
|
( 0, 0, 255, 255),
|
|
( 0, 255, 255, 255),
|
|
( 0, 255, 0, 255),
|
|
(255, 255, 0, 255),
|
|
(255, 0, 0, 255),
|
|
}
|
|
seeds = {c: [] for c in region_colors}
|
|
for y in range(0, im.size[1]):
|
|
for x in range(0, im.size[0]):
|
|
p = im.getpixel((x, y))
|
|
if p in region_colors:
|
|
seeds[p].append((x, y))
|
|
print(time.clock_gettime(time.CLOCK_MONOTONIC) - t0)
|
|
for y in range(0, im.size[1]):
|
|
#print(f"{y}")
|
|
for x in range(0, im.size[0]):
|
|
p = im.getpixel((x, y))
|
|
if p[3] == 0:
|
|
best_color = (0, 0, 0, 255)
|
|
min_d2 = 10_000_000
|
|
for color, points in seeds.items():
|
|
for p in points:
|
|
dx = p[0] - x
|
|
dy = p[1] - y
|
|
d2 = dx * dx + dy * dy
|
|
if d2 < min_d2:
|
|
best_color = color
|
|
min_d2 = d2
|
|
best_color = (*best_color[:3], 127)
|
|
im.putpixel((x, y), best_color)
|
|
|
|
print(time.clock_gettime(time.CLOCK_MONOTONIC) - t0)
|
|
im.save("bodensee-r3.png")
|
|
print(time.clock_gettime(time.CLOCK_MONOTONIC) - t0)
|