codemicro/adventOfCode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Formatting

Browse source
This commit is contained in:
akp 2024-12-08 13:45:19 +00:00
parent 8f7d4b9278
commit 8c2606b60b
No known key found for this signature in database
GPG key ID: CF8D58F3DEB20755
1 changed files with 30 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

25
challenges/2024/08-resonantCollinearity/vis.py
View file

@ -29,8 +29,15 @@ colour_diffs = tuple(map(lambda x: x[1] - x[0], zip(highest_colour, lowest_colou
def get_colour_for(n):
return tuple(
map(int, map(lambda x: x[0] - x[1], zip(lowest_colour, map(lambda x: x * n, colour_diffs))))
map(
int,
map(
lambda x: x[0] - x[1],
zip(lowest_colour, map(lambda x: x * n, colour_diffs)),
),
)
)
scale_factor = 8
@ -41,7 +48,7 @@ def get_highlight_for(n):
def generate_frame(i, base_img, highlight_locs, hc, sequence) -> int:
for n in range(len(sequence)):
s = sequence[:n+1]
s = sequence[: n + 1]
img = base_img.copy()
sl = len(s) + 1
@ -52,7 +59,9 @@ def generate_frame(i, base_img, highlight_locs, hc, sequence) -> int:
img.putpixel(h, hc)
maxx, maxy = img.size
img = img.resize((maxx * scale_factor, maxy * scale_factor), resample=Image.NEAREST)
img = img.resize(
(maxx * scale_factor, maxy * scale_factor), resample=Image.NEAREST
)
img.save(f"frames/{str(i).zfill(5)}.png")
i += 1
return i
@ -75,7 +84,7 @@ if __name__ == "__main__":
except FileExistsError:
pass
base_img = Image.new("RGB", (max_x+1, max_y+1), color=lowest_colour)
base_img = Image.new("RGB", (max_x + 1, max_y + 1), color=lowest_colour)
i = 0
for antenna_type in tqdm(antenna_by_type):
@ -102,5 +111,11 @@ if __name__ == "__main__":
x_cursor += diff.x
y_cursor += diff.y
i = generate_frame(i, base_img, (a, b), get_highlight_for(ns.index(antenna_type) / nns), this_iter)
i = generate_frame(
i,
base_img,
(a, b),
get_highlight_for(ns.index(antenna_type) / nns),
this_iter,
)
update_base(base_img, this_iter)