1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#!/usr/bin/env python3
import sys
import time
def read_input(args, day):
if len(args) > 0:
if args[0] == "-":
return sys.stdin.readlines()
else:
return open(args[0]).readlines()
else:
return open(f"../input/{day:02}", "r").readlines()
if __name__ == "__main__":
import importlib
skip = set()
only = set()
only_part = 0
run_times = 1
input_root = "../input"
decorate = True
argv, argc = sys.argv, len(sys.argv)
i = 1
while i < argc:
if argv[i] == "--help":
print(f"usage: {argv[0]} [--help] [--time [times]] [--skip <n> <n> ...]\n" +
" [--only <n> <n> ...] [--part 0|1|2] [--input <dir>]\n" +
" [--no-decorate]")
sys.exit(0)
elif argv[i] == "--time":
i += 1
if i < argc and not argv[i].startswith("-"):
run_times = int(argv[i])
i += 1
else:
run_times = 10
elif argv[i] == "--skip":
i += 1
while i < argc and not argv[i].startswith("-"):
skip.add(int(argv[i]))
i += 1
elif argv[i] == "--only":
i += 1
while i < argc and not argv[i].startswith("-"):
only.add(int(argv[i]))
i += 1
elif argv[i] == "--part":
i += 1
only_part = int(argv[i])
i += 1
elif argv[i] == "--input":
i += 1
input_root = argv[i]
i += 1
elif argv[i] == "--no-decorate":
i += 1
decorate = False
else:
print(f"unknown argument {argv[i]}")
print(f"maybe try {argv[0]} --help ?")
i += 1
def running_part(part):
return only_part == 0 or part == only_part
days = []
for day in range(25):
if day+1 in skip or (only and day+1 not in only):
continue
try:
days.append((day+1, importlib.import_module(f"d{day+1:02}")))
except:
break
if decorate:
print(" ", end="")
if running_part(1):
print(" -------Part 1--------", end="")
if only_part == 0:
print(" ", end="") # :(
if running_part(2):
print(" -------Part 2--------", end="")
print()
print("Day", end="")
print(" Time Ans ", end="")
if only_part == 0:
print(" Time Ans", end="")
print()
tot_time = [0, 0]
for day, mod in days:
input = open(f"{input_root}/{day:02}").readlines()
print(f" {day:2}", end="")
for part, part_func in enumerate((mod.pt1, mod.pt2)):
if not running_part(part+1):
continue
times = []
for i in range(run_times):
start = time.time()
ans = part_func(input)
ans_time = time.time()
times.append(ans_time-start)
avg_time = sum(times) / len(times)
tot_time[part] += avg_time
print(f" {avg_time*1000:6.3f} {ans:14}", end="")
print()
if decorate:
print(" -------", end="")
if only_part == 0:
print(" -------", end="")
print()
print("tot ", end="")
if running_part(1):
print(f"{tot_time[0]*1000:6.3f}", end="")
if only_part == 0:
print(" ", end="")
if running_part(2):
print(f"{tot_time[1]*1000:6.3f}", end="")
print()
|
