Added 2019 day 07 part 1

2019/07/code.py

@@ -0,0 +1,133 @@
+""" https://adventofcode.com/2019/day/7 """
+
+import queue, threading
+
+class InvalidOpcode(Exception):
+    pass
+
+def readFile():
+    with open(f"{__file__.rstrip('code.py')}input.txt", "r") as f:
+        return [int(num) for num in f.readline().split(",")]
+
+class Amp:
+    def __init__(self, id, vals, phase, ownQueue, nextQueue):
+        self.id = id
+        self.vals = vals.copy()
+        self.phase = phase
+        self.output = 0
+        self.queue = ownQueue
+        self.next = nextQueue
+        self.phased = False
+        self.i = 0
+
+    def run(self):
+        while 1:
+            opcode = (self.vals[self.i] % 100, self.vals[self.i] // 100 % 10, 
+                self.vals[self.i] // 1000 % 10, self.vals[self.i] // 10000 % 10)
+
+            # 99 : Termination
+            # 0 Parameter
+            if opcode[0] == 99:
+                self.output = self.vals[0]
+                self.next.put(self.output)
+                break
+
+            # 1 : Addition
+            # 3 parameter
+            elif opcode[0] == 1:
+                a = self.vals[self.i+1] if opcode[1] == 0 else self.i+1
+                b = self.vals[self.i+2] if opcode[2] == 0 else self.i+2
+                self.vals[self.vals[self.i+3]] = self.vals[a] + self.vals[b]
+                self.i += 4
+
+            # 2 : Multiplication
+            # 3 parameter
+            elif opcode[0] == 2:
+                a = self.vals[self.i+1] if opcode[1] == 0 else self.i+1
+                b = self.vals[self.i+2] if opcode[2] == 0 else self.i+2
+                self.vals[self.vals[self.i+3]] = self.vals[a] * self.vals[b]
+                self.i += 4
+
+            # 3 : Input
+            # 1 parameter
+            elif opcode[0] == 3: 
+                if self.phased:
+                    self.vals[self.vals[self.i+1]] = self.queue.get()
+                else:
+                    self.vals[self.vals[self.i+1]], self.phased = self.phase, True
+                self.i += 2
+
+            # 4 : Output
+            # 1 parameter
+            elif opcode[0] == 4: 
+                a = self.vals[self.i+1] if opcode[1] == 0 else self.i+1
+                self.vals[0] = self.vals[a]
+                self.next.put(self.vals[0])
+                self.i += 2
+
+            # 5 : Jump-if-true
+            # 6 : Jump-if-false
+            # 2 parameter
+            elif opcode[0] in [5, 6]:
+                a = self.vals[self.i+1] if opcode[1] == 0 else self.i+1
+                if opcode[0] == 5 and self.vals[a] or \
+                opcode[0] == 6 and not self.vals[a]:
+                    b = self.vals[self.i+2] if opcode[2] == 0 else self.i+2
+                    self.i = self.vals[b]
+                else:
+                    self.i += 3
+
+            # 7 : less-than
+            # 8 : equals
+            # 3 parameter
+            elif opcode[0] in [7, 8]:
+                a = self.vals[self.i+1] if opcode[1] == 0 else self.i+1
+                b = self.vals[self.i+2] if opcode[2] == 0 else self.i+2
+                self.vals[self.vals[self.i+3]] = 1 if opcode[0] == 7 and self.vals[a] < \
+                    self.vals[b] or opcode[0] == 8 and self.vals[a] == self.vals[b] else 0
+                self.i += 4
+            
+            else:
+                raise InvalidOpcode
+
+def amplify(vals : list, phaseSettings : tuple):
+    q = [queue.Queue() for i in range(5)]
+    amps = [Amp(i, vals, phaseSettings[i], q[i], q[(i+1) % 5]) for i in range(5)]
+    amps[0].queue.put(0)
+    if phaseSettings[0] < 5:
+        amps[4].next = queue.Queue()
+    threads = [threading.Thread(target=amps[i].run()) for i in range(5)]
+
+    for thread in threads:
+        thread.start()
+
+    for thread in threads:
+        thread.join()
+
+    return amps[4].next.get()
+
+def part1(vals : list):
+    return max([amplify(vals, (a, b, c, d, e)) for a in range(5) for b in range(5) for c in range(5) for d in range(5)
+        for e in range(5) if b != a and c not in [a, b] and d not in [a, b, c] and e not in [a, b, c, d]])
+
+def part2(vals : list):
+    pass
+
+def test():
+    vals = [3,15,3,16,1002,16,10,16,1,16,15,15,4,15,99,0,0]
+    assert amplify(vals, (4,3,2,1,0)) == 43210
+    vals = [3,23,3,24,1002,24,10,24,1002,23,-1,23,
+        101,5,23,23,1,24,23,23,4,23,99,0,0]
+    assert amplify(vals, (0,1,2,3,4)) == 54321
+    vals = [3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33,
+        1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0]
+    assert amplify(vals, (1,0,4,3,2)) == 65210
+    vals = [3,26,1001,26,-4,26,3,27,1002,27,2,27,1,27,26,
+        27,4,27,1001,28,-1,28,1005,28,6,99,0,0,5]
+    #print(amplify(vals, (9,8,7,6,5)))
+     
+if __name__ == "__main__":
+    test()
+    vals = readFile()
+    print(f"Part 1: {part1(vals)}")
+    print(f"Part 2: {part2(vals)}")