AdventOfCode/2022/helpers.py

from numpy.typing import ArrayLike
import numpy as np
from functools import cmp_to_key, cache
from itertools import permutations
from math import prod
from time import perf_counter_ns
import re
import requests
import sys
sys.setrecursionlimit(100000)

try:
	import datetime
	today = datetime.date.today()
	day = today.day
	year = today.year

	import browser_cookie3
	def download_input(day: int = day):
		filename = f'input/{day:02}'
		r = requests.get(f"https://adventofcode.com/{year}/day/{day}/input", cookies=browser_cookie3.firefox())
		if r.status_code != 200:
			print(r)
		else:
			with open(filename,'w') as f:
				f.write(r.text)
except ModuleNotFoundError as mod:
	print(f'{mod.args[0]}, automatic input fetching will not work.')

numbers_pattern = re.compile(r'((?:(?<!\d)-)?\d+)')

def line_to_numbers(line: str) -> list[int]:
	return [int(x) for x in numbers_pattern.findall(line)]

def lines_to_numbers(lines: list[str]) -> list[list[int]]:
	return [line_to_numbers(line) for line in lines]

def transpose_array_of_strings(aos: list[str], reverse_x = False, reverse_y = False, strip = '') -> list[str]:
	return [''.join(l[i] for l in aos[::-1 if reverse_y else 1]).strip(strip) for i in range(len(aos[0]))[::-1 if reverse_x else 1]]

def read_day(day: int, strip=True) -> str:
	with open(f'input/{day:02}', 'r') as file:
		if strip:
			return file.read().strip()
		else:
			return file.read()

def print_time(name: str, ns: int):
	print(f'{name} took {ns}ns = {ns/1000000:.2f}ms')

dtype = np.int32
directions_array = np.array([[1,0], [-1,0], [0,1], [0,-1]], dtype=dtype)
directions_dict = {'R': directions_array[0], 'L': directions_array[1], 'D': directions_array[2], 'U': directions_array[3]}  # Positive down for visualization

def visualise_sparse_cells_set(cells: set[tuple[int, int]], size=20, sym_hit='#', sym_miss='.'):
	vis_lol = [[sym_hit if (x,y) in cells else sym_miss for x in range(-size, size)] for y in range(-size, size)]
	vis_str = '\n'.join(''.join(line) for line in vis_lol)
	print(vis_str)