32 lines
708 B
Python
32 lines
708 B
Python
|
from grid import Grid
|
||
|
|
from intcode import Intcode
|
||
|
|
from tools import parse_intcode
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
codes = parse_intcode("day19.txt")
|
||
|
|
area = Grid()
|
||
|
|
for y in range(50):
|
||
|
|
for x in range(50):
|
||
|
|
mgr = DroneMgr(codes)
|
||
|
|
mgr.run(inputs=[x, y])
|
||
|
|
area.setv(x, y, mgr.output)
|
||
|
|
area.dumb_print()
|
||
|
|
num_affected = sum(v for _, _, v in area.values_gen())
|
||
|
|
print("Num affected:", num_affected)
|
||
|
|
|
||
|
|
|
||
|
|
class DroneMgr(Intcode):
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.inputs = [0, 0]
|
||
|
|
self.output = None
|
||
|
|
|
||
|
|
def output_data(self, data):
|
||
|
|
self.output = data
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|