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,9 +29,16 @@ colour_diffs = tuple(map(lambda x: x[1] - x[0], zip(highest_colour, lowest_colou
def get_colour_for(n): def get_colour_for(n):
return tuple( 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 scale_factor = 8
@ -41,7 +48,7 @@ def get_highlight_for(n):
def generate_frame(i, base_img, highlight_locs, hc, sequence) -> int: def generate_frame(i, base_img, highlight_locs, hc, sequence) -> int:
for n in range(len(sequence)): for n in range(len(sequence)):
s = sequence[:n+1] s = sequence[: n + 1]
img = base_img.copy() img = base_img.copy()
sl = len(s) + 1 sl = len(s) + 1
@ -52,7 +59,9 @@ def generate_frame(i, base_img, highlight_locs, hc, sequence) -> int:
img.putpixel(h, hc) img.putpixel(h, hc)
maxx, maxy = img.size 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") img.save(f"frames/{str(i).zfill(5)}.png")
i += 1 i += 1
return i return i
@ -75,7 +84,7 @@ if __name__ == "__main__":
except FileExistsError: except FileExistsError:
pass 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 i = 0
for antenna_type in tqdm(antenna_by_type): for antenna_type in tqdm(antenna_by_type):
@ -102,5 +111,11 @@ if __name__ == "__main__":
x_cursor += diff.x x_cursor += diff.x
y_cursor += diff.y 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) update_base(base_img, this_iter)