ruediger/advent-2022-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

day 16 much, much quicker

Browse source
This commit is contained in:
Ruediger Ludwig 2023-01-18 20:17:27 +01:00
parent d02d8190f0
commit eb12a799fd
3 changed files with 164 additions and 128 deletions
Download patch file Download diff file Expand all files Collapse all files

46
advent/__main__.py
View file

@ -1,18 +1,18 @@
import sys
from importlib import import_module
import time
import sys
from advent.common import input
from advent.days.template import Day, ResultType, is_day
def output(day: int, part: int, result: ResultType | None) -> None:
def output(day: int, part: int, result: ResultType | None, delta: float) -> None:
match result:
case int(value):
print('Day {0:02} Part {1}: {2}'.format(day, part, value))
print(f'Day {day:02} Part {part}: {value} ({delta:0.3}s)')
case str(value):
print('Day {0:02} Part {1}: {2}'.format(day, part, value))
print(f'Day {day:02} Part {part}: {value} ({delta:0.3}s)')
case list(value):
print('Day {0:02} Part {1}: {2}'.format(day, part, value[0]))
print(f'Day {day:02} Part {part}: {value[0]} ({delta:0.3}s)')
for line in value[1:]:
print(f' {line}')
case None:
@ -29,23 +29,31 @@ def get_day(day_num: int) -> Day:
return day_module
def run(day: Day, part: int) -> None:
def run(day: Day, part: int) -> float:
data = input.read_lines(day.day_num, 'input.txt')
t0 = time.time()
match part:
case 1: output(day.day_num, 1, day.part1(data))
case 2: output(day.day_num, 2, day.part2(data))
case 1: result = day.part1(data)
case 2: result = day.part2(data)
case _: raise Exception(f'Unknown part {part}')
t1 = time.time()
delta = t1 - t0
output(day.day_num, part, result, delta)
return delta
def run_from_string(day_str: str) -> None:
def run_from_string(day_str: str) -> float:
match day_str.split('/'):
case [d]:
day_num = int(d)
day = get_day(day_num)
if day_num == day.day_num:
run(day, 1)
run(day, 2)
p1 = run(day, 1)
p2 = run(day, 2)
return p1 + p2
assert False, "We should never get here"
case [d, p]:
day_num = int(d)
@ -53,30 +61,36 @@ def run_from_string(day_str: str) -> None:
if day_num == day.day_num:
part = int(p)
run(day, part)
return run(day, part)
assert False, "We should never get here"
case _:
raise Exception(f'{day_str} is not a valid day description')
def main() -> None:
print()
time = 0.0
match sys.argv:
case [_]:
try:
for day_num in range(1, 25):
day = get_day(day_num)
if day_num == day.day_num:
run(day, 1)
run(day, 2)
time += run(day, 1)
time += run(day, 2)
except ModuleNotFoundError:
pass
case [_, argument]:
run_from_string(argument)
time += run_from_string(argument)
case _:
raise Exception(f'Usage: python {sys.argv[0]} [day[/part]]')
print(f"\nTotal time: {time:0.3}s")
if __name__ == '__main__':
main()