56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
|
from queue import Empty, SimpleQueue
|
||
|
|
from threading import Thread
|
||
|
|
|
||
|
|
from intcode import Intcode
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
codes = Intcode.parse_file("day23.txt")
|
||
|
|
network = Network(50, codes)
|
||
|
|
for nic in network.nics:
|
||
|
|
thread = Thread(target=lambda: nic.run(), name=f"NIC {nic.address}")
|
||
|
|
thread.start()
|
||
|
|
|
||
|
|
|
||
|
|
class Network:
|
||
|
|
|
||
|
|
def __init__(self, size, codes):
|
||
|
|
self.nics = [NIC(address, self, codes) for address in range(size)]
|
||
|
|
|
||
|
|
def route(self, sender, dest, x, y):
|
||
|
|
if dest == 255:
|
||
|
|
print(f"Packet to 255: {x, y}.")
|
||
|
|
return
|
||
|
|
print(f"Send {x, y} from {sender} to {dest}")
|
||
|
|
self.nics[dest].inq.put(x)
|
||
|
|
self.nics[dest].inq.put(y)
|
||
|
|
|
||
|
|
|
||
|
|
class NIC(Intcode):
|
||
|
|
|
||
|
|
def __init__(self, address, network, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.address = address
|
||
|
|
self.network = network
|
||
|
|
self.inq = SimpleQueue()
|
||
|
|
self.inq.put(address)
|
||
|
|
self.outq = []
|
||
|
|
|
||
|
|
def input_data(self):
|
||
|
|
try:
|
||
|
|
data = self.inq.get(timeout=0.1)
|
||
|
|
except Empty:
|
||
|
|
data = -1
|
||
|
|
return data
|
||
|
|
|
||
|
|
def output_data(self, data):
|
||
|
|
self.outq.append(data)
|
||
|
|
if len(self.outq) == 3:
|
||
|
|
self.network.route(self.address, *self.outq)
|
||
|
|
self.outq = []
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|