2019 Day 13 Part 2; intcode as class; intcode_test to assure functionality; changed old code to adapt intcode computer as object
This commit is contained in:
Akumatic
164
2019/13/code.py
164
2019/13/code.py
@ -1,110 +1,84 @@
|
||||
""" https://adventofcode.com/2019/day/13 """
|
||||
# SPDX-License-Identifier: MIT
|
||||
# Copyright (c) 2019 Akumatic
|
||||
#
|
||||
# https://adventofcode.com/2019/day/13
|
||||
|
||||
class InvalidOpcode(Exception):
|
||||
pass
|
||||
|
||||
class mylist(list):
|
||||
def __getitem__(self, item):
|
||||
try:
|
||||
return super(mylist,self).__getitem__(item)
|
||||
except IndexError as e:
|
||||
if item < 0: raise IndexError()
|
||||
super(mylist,self).extend([0]*(item + 1 - super(mylist,self).__len__()))
|
||||
return super(mylist,self).__getitem__(item)
|
||||
|
||||
def __setitem__(self, idx, val):
|
||||
try:
|
||||
return super(mylist,self).__setitem__(idx,val)
|
||||
except IndexError as e:
|
||||
if idx < 0: raise IndexError()
|
||||
super(mylist,self).extend([0]*(idx + 1 - super(mylist,self).__len__()))
|
||||
return super(mylist,self).__setitem__(idx,val)
|
||||
import sys, os
|
||||
sys.path.insert(1, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
import intcode
|
||||
|
||||
def readFile():
|
||||
with open(f"{__file__.rstrip('code.py')}input.txt", "r") as f:
|
||||
return [int(num) for num in f.readline().split(",")]
|
||||
|
||||
def getOutput(vals : list, input=0, base=0):
|
||||
vals = mylist(vals)
|
||||
i = 0
|
||||
out = []
|
||||
while 1:
|
||||
opcode = (vals[i] % 100,
|
||||
vals[i] // 100 % 10,
|
||||
vals[i] // 1000 % 10,
|
||||
vals[i] // 10000 % 10)
|
||||
|
||||
# 0 Parameter
|
||||
if opcode[0] in [99]: # Termination
|
||||
return vals, out
|
||||
|
||||
# 1 Parameter
|
||||
elif opcode[0] in [3,4,9]:
|
||||
a = vals[i+1] if not opcode[1] else i+1 if opcode[1] == 1 else base+vals[i+1]
|
||||
if opcode[0] == 3: # Input
|
||||
vals[a] = input
|
||||
elif opcode[0] == 4: # Output
|
||||
vals[0] = vals[a]
|
||||
out.append(vals[0])
|
||||
elif opcode[0] == 9: # Adjust Base
|
||||
base += vals[a]
|
||||
i += 2
|
||||
|
||||
# 2 Parameter
|
||||
elif opcode[0] in [5,6]:
|
||||
a = vals[i+1] if not opcode[1] else i+1 if opcode[1] == 1 else base+vals[i+1]
|
||||
b = vals[i+2] if not opcode[2] else i+2 if opcode[2] == 1 else base+vals[i+2]
|
||||
if opcode[0] == 5 and vals[a] != 0: # Jump-if-true
|
||||
i = vals[b]
|
||||
elif opcode[0] == 6 and vals[a] == 0: # Jump-if-false
|
||||
i = vals[b]
|
||||
else:
|
||||
i += 3
|
||||
|
||||
# 3 Parameter
|
||||
elif opcode[0] in [1,2,7,8]:
|
||||
a = vals[i+1] if not opcode[1] else i+1 if opcode[1] == 1 else base+vals[i+1]
|
||||
b = vals[i+2] if not opcode[2] else i+2 if opcode[2] == 1 else base+vals[i+2]
|
||||
c = vals[i+3] if not opcode[3] else i+3 if opcode[3] == 1 else base+vals[i+3]
|
||||
if opcode[0] == 1: # Addition
|
||||
vals[c] = vals[a] + vals[b]
|
||||
elif opcode[0] == 2: # Multiplication
|
||||
vals[c] = vals[a] * vals[b]
|
||||
elif opcode[0] == 7: # Less Than
|
||||
vals[c] = int(vals[a] < vals[b])
|
||||
elif opcode[0] == 8: # Equals
|
||||
vals[c] = int(vals[a] == vals[b])
|
||||
i += 4
|
||||
def getEntities(vals: list):
|
||||
assert len(vals) % 3 == 0
|
||||
entities = {"empty": [], "wall": [], "block": [],
|
||||
"paddle": tuple(), "ball": tuple(), "score": 0}
|
||||
keys = tuple(entities.keys())
|
||||
|
||||
for i in range(0, len(vals), 3):
|
||||
# updated score
|
||||
if (vals[i] == -1 and vals[i+1] == 0):
|
||||
entities["score"] = vals[i+2]
|
||||
continue
|
||||
# updated ball, paddle and other positions
|
||||
pos = (vals[i], vals[i+1])
|
||||
if vals[i+2] < 3:
|
||||
entities[keys[vals[i+2]]].append(pos)
|
||||
else:
|
||||
raise InvalidOpcode()
|
||||
entities[keys[vals[i+2]]] = pos
|
||||
|
||||
def getEntities():
|
||||
output = getOutput(vals.copy())[1]
|
||||
output.reverse()
|
||||
assert len(output) % 3 == 0
|
||||
entities = [set() for _ in range(5)]
|
||||
while output:
|
||||
pos = (output.pop(), output.pop())
|
||||
tile = output.pop()
|
||||
entities[tile].add(pos)
|
||||
return entities
|
||||
|
||||
def part1(entities):
|
||||
print([len(x) for x in entities])
|
||||
print(entities[3], entities[4])
|
||||
return len(entities[2])
|
||||
def part1(pc: intcode.Computer) -> int:
|
||||
pc.run()
|
||||
entities = getEntities(pc.output)
|
||||
return len(entities["block"])
|
||||
|
||||
class Ball:
|
||||
def __init__(self, x, y):
|
||||
self.x, self.y = 0,0
|
||||
self.dx, self.dy = 0,0
|
||||
def part2(pc: intcode.Computer) -> int:
|
||||
pc.reset(wait_for_input=True)
|
||||
pc.data[0] = 2
|
||||
|
||||
def part2(entities):
|
||||
pass
|
||||
# initial setup
|
||||
pc.run()
|
||||
idx = pc.output.index(-1)
|
||||
while pc.output[idx + 1] != 0:
|
||||
idx = pc.output.index(-1, idx + 1)
|
||||
entities = getEntities(pc.output[:idx])
|
||||
pc.output = pc.output[idx:]
|
||||
|
||||
# loop
|
||||
while not pc.stop:
|
||||
pc.run()
|
||||
changes = getEntities(pc.output)
|
||||
if changes["score"]:
|
||||
entities["score"] = changes["score"]
|
||||
|
||||
if changes["ball"]:
|
||||
changes["empty"].remove(entities["ball"])
|
||||
entities["ball"] = changes["ball"]
|
||||
|
||||
if changes["paddle"]:
|
||||
changes["empty"].remove(entities["paddle"])
|
||||
entities["paddle"] = changes["paddle"]
|
||||
|
||||
for e in changes["empty"]:
|
||||
entities["block"].remove(e)
|
||||
|
||||
pc.output.clear()
|
||||
pc.input = entities["ball"][0] - entities["paddle"][0]
|
||||
|
||||
return entities["score"]
|
||||
|
||||
def test():
|
||||
test_input = [1, 2, 3, 6, 5, 4]
|
||||
test_output = getEntities(test_input)
|
||||
assert test_output["empty"] == test_output["wall"] == test_output["block"] == []
|
||||
assert test_output["paddle"] == (1, 2) and test_output["ball"] == (6, 5)
|
||||
|
||||
if __name__ == "__main__":
|
||||
vals = readFile()
|
||||
entities = getEntities()
|
||||
print(f"Part 1: {part1(entities)}")
|
||||
#print(f"Part 2: {part2(entities)}")
|
||||
test()
|
||||
pc = intcode.Computer(readFile())
|
||||
print(f"Part 1: {part1(pc)}")
|
||||
print(f"Part 2: {part2(pc)}")
|
||||
@ -1 +1,2 @@
|
||||
Part 1: 200
|
||||
Part 1: 200
|
||||
Part 2: 9803
|
||||
Reference in New Issue
Block a user