Adding day 16 part 1

Author: Joe Webster

16/__init__.py

@@ -0,0 +1,127 @@
+class BITSPacketDecoder(object):
+    def __init__(self, packet):
+        self.packet = packet
+        self.bits = self.hex2bin(self.packet)
+        self.index = 0
+
+    def hex2bin(self, hex_string):
+        ret = ''
+        for c in hex_string:
+            ret += bin(int(c, 16))[2:].rjust(4, '0') 
+
+        return ret
+
+    def bits_left(self):
+        ret = len(self.bits) - self.index
+        return ret
+
+    def read(self, read_len):
+        if self.bits_left() < read_len:
+            print("read underrun")
+            return None
+
+        ret = self.bits[self.index:self.index+read_len]
+
+        #print(self.bits)
+        #print(' ' * self.index + '*' * read_len + ' = ' + ret + ' (' + str(int(ret, 2)) + ')')
+        #print()
+
+        self.index += read_len
+
+        return ret
+
+    def read_packet(self):
+        version = int(self.read(3), 2)
+        type_id = int(self.read(3), 2)
+        
+        print('version: %s type_id: %s' % (version, type_id))
+
+        if type_id == 4:
+            ret = self.parse_payload_4(version, type_id)
+        else:
+            ret = self.parse_payload_op(version, type_id)
+
+        return ret
+
+    def parse_payload_4(self, version, type_id):
+        bit_string = ""
+        more = 1
+        while more:
+            more = int(self.read(1))
+            bit_string += self.read(4)
+
+        return {'version': version, 'type_id': type_id, 'literal': int(bit_string, 2)}
+
+    def parse_payload_op(self, version, type_id):
+        subpackets = []
+
+        length_type_id = int(self.read(1))
+
+        if length_type_id == 0:
+            subpacket_length = int(self.read(15), 2)
+        elif length_type_id == 1:
+            subpacket_count = int(self.read(11), 2)
+
+        while self.bits_left() > 12:
+            subpacket = self.read_packet()
+            if subpacket:
+                subpackets.append(subpacket)
+            else:
+                print("no subpacket?, breaking")
+                break
+
+        return {'version': version, 'type_id': type_id, 'subpackets': subpackets}
+
+class BITS(object):
+    def __init__(self):
+        self.transmissions = []
+
+    def load_data(self, data):
+        for line in data:
+            self.transmissions.append(line)
+
+    def get_version_sum(self, packet):
+        version_sum = packet.get('version') 
+
+        if 'subpackets' in packet:
+            for subpacket in packet['subpackets']:
+                version_sum += self.get_version_sum(subpacket)
+
+        return version_sum
+
+    def parse_packets(self, data):
+        print(data)
+        bd = BITSPacketDecoder(data)
+        print(bd.bits)
+        print()
+
+        packets = bd.read_packet()
+
+        return packets
+
+def get_data():
+    return [l.rstrip() for l in open('data.txt', 'r').readlines()]
+
+def get_test_data():
+    return [l.rstrip() for l in open('test_data.txt', 'r').readlines()]
+
+def solution_part_1(data):
+    bits = BITS()
+    bits.load_data(data) 
+    packets = bits.parse_packets(data)
+    version_sum = bits.get_version_sum(packets)
+    print('version_sum', version_sum)
+
+def solution_part_2(data):
+    bits = BITS()
+    bits.load_data(data) 
+    packets = bits.parse_packets(data)
+    result = bits.evaluate_packets(packets)
+    print('eval results', result)
+
+if __name__ == '__main__':
+    solution_part_1(get_test_data())
+    solution_part_1(get_data())
+
+    #solution_part_2(get_test_data())
+    #solution_part_2(get_data())

16/data.txt

@@ -0,0 +1 @@
+6051639005B56008C1D9BB3CC9DAD5BE97A4A9104700AE76E672DC95AAE91425EF6AD8BA5591C00F92073004AC0171007E0BC248BE0008645982B1CA680A7A0CC60096802723C94C265E5B9699E7E94D6070C016958F99AC015100760B45884600087C6E88B091C014959C83E740440209FC89C2896A50765A59CE299F3640D300827902547661964D2239180393AF92A8B28F4401BCC8ED52C01591D7E9D2591D7E9D273005A5D127C99802C095B044D5A19A73DC0E9C553004F000DE953588129E372008F2C0169FDB44FA6C9219803E00085C378891F00010E8FF1AE398803D1BE25C743005A6477801F59CC4FA1F3989F420C0149ED9CF006A000084C5386D1F4401F87310E313804D33B4095AFBED32ABF2CA28007DC9D3D713300524BCA940097CA8A4AF9F4C00F9B6D00088654867A7BC8BCA4829402F9D6895B2E4DF7E373189D9BE6BF86B200B7E3C68021331CD4AE6639A974232008E663C3FE00A4E0949124ED69087A848002749002151561F45B3007218C7A8FE600FC228D50B8C01097EEDD7001CF9DE5C0E62DEB089805330ED30CD3C0D3A3F367A40147E8023221F221531C9681100C717002100B36002A19809D15003900892601F950073630024805F400150D400A70028C00F5002C00252600698400A700326C0E44590039687B313BF669F35C9EF974396EF0A647533F2011B340151007637C46860200D43085712A7E4FE60086003E5234B5A56129C91FC93F1802F12EC01292BD754BCED27B92BD754BCED27B100264C4C40109D578CA600AC9AB5802B238E67495391D5CFC402E8B325C1E86F266F250B77ECC600BE006EE00085C7E8DF044001088E31420BCB08A003A72BF87D7A36C994CE76545030047801539F649BF4DEA52CBCA00B4EF3DE9B9CFEE379F14608

16/test_data.txt

@@ -0,0 +1,7 @@
+D2FE28
+38006F45291200
+EE00D40C823060
+8A004A801A8002F478
+620080001611562C8802118E34
+C0015000016115A2E0802F182340
+A0016C880162017C3686B18A3D4780