phoenixuprising

AoC day3 Input to Factorio Blueprint

Dec 6th, 2020
831
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 5.05 KB | None | 0 0
  1. import zlib
  2. import base64
  3. import json
  4.  
  5.  
  6. def read_blueprint(data):
  7.     decoded_data = base64.b64decode(data.encode('ascii')[1:])
  8.     decompressed = zlib.decompress(decoded_data)
  9.     return json.dumps(json.loads(decompressed), indent=2)
  10.  
  11.  
  12. def write_blueprint(blueprint_dict):
  13.     print('0' + base64.b64encode(zlib.compress(json.dumps(blueprint_dict).encode('utf-8'))).decode('ascii'))
  14.  
  15.  
  16. def add_combinator(ent_num, value, x, x_offset, y, y_offset):
  17.     return {
  18.         "entity_number": ent_num,
  19.         "name": "constant-combinator",
  20.         "position": {
  21.             "x": x_offset + 0.5,
  22.             "y": y_offset + 0.5
  23.         },
  24.         "control_behavior": {
  25.             "filters": [
  26.                 {
  27.                     "signal": {
  28.                         "type": "virtual",
  29.                         "name": "signal-I"
  30.                     },
  31.                     "count": value,
  32.                     "index": 1
  33.                 },
  34.                 {
  35.                     "signal": {
  36.                         "type": "item",
  37.                         "name": "iron-plate"
  38.                     },
  39.                     "count": y,
  40.                     "index": 11
  41.                 },
  42.                 {
  43.                     "signal": {
  44.                         "type": "virtual",
  45.                         "name": "signal-L"
  46.                     },
  47.                     "count": 1,
  48.                     "index": 12
  49.                 }
  50.             ]
  51.         }
  52.     }
  53.  
  54. def reverse_bits(val):
  55.     BitReverseTable256 = [0x00, 0x80, 0x40, 0xC0, 0x20, 0xA0, 0x60, 0xE0, 0x10, 0x90, 0x50, 0xD0, 0x30, 0xB0, 0x70,
  56.                           0xF0,
  57.                           0x08, 0x88, 0x48, 0xC8, 0x28, 0xA8, 0x68, 0xE8, 0x18, 0x98, 0x58, 0xD8, 0x38, 0xB8, 0x78,
  58.                           0xF8,
  59.                           0x04, 0x84, 0x44, 0xC4, 0x24, 0xA4, 0x64, 0xE4, 0x14, 0x94, 0x54, 0xD4, 0x34, 0xB4, 0x74,
  60.                           0xF4,
  61.                           0x0C, 0x8C, 0x4C, 0xCC, 0x2C, 0xAC, 0x6C, 0xEC, 0x1C, 0x9C, 0x5C, 0xDC, 0x3C, 0xBC, 0x7C,
  62.                           0xFC,
  63.                           0x02, 0x82, 0x42, 0xC2, 0x22, 0xA2, 0x62, 0xE2, 0x12, 0x92, 0x52, 0xD2, 0x32, 0xB2, 0x72,
  64.                           0xF2,
  65.                           0x0A, 0x8A, 0x4A, 0xCA, 0x2A, 0xAA, 0x6A, 0xEA, 0x1A, 0x9A, 0x5A, 0xDA, 0x3A, 0xBA, 0x7A,
  66.                           0xFA,
  67.                           0x06, 0x86, 0x46, 0xC6, 0x26, 0xA6, 0x66, 0xE6, 0x16, 0x96, 0x56, 0xD6, 0x36, 0xB6, 0x76,
  68.                           0xF6,
  69.                           0x0E, 0x8E, 0x4E, 0xCE, 0x2E, 0xAE, 0x6E, 0xEE, 0x1E, 0x9E, 0x5E, 0xDE, 0x3E, 0xBE, 0x7E,
  70.                           0xFE,
  71.                           0x01, 0x81, 0x41, 0xC1, 0x21, 0xA1, 0x61, 0xE1, 0x11, 0x91, 0x51, 0xD1, 0x31, 0xB1, 0x71,
  72.                           0xF1,
  73.                           0x09, 0x89, 0x49, 0xC9, 0x29, 0xA9, 0x69, 0xE9, 0x19, 0x99, 0x59, 0xD9, 0x39, 0xB9, 0x79,
  74.                           0xF9,
  75.                           0x05, 0x85, 0x45, 0xC5, 0x25, 0xA5, 0x65, 0xE5, 0x15, 0x95, 0x55, 0xD5, 0x35, 0xB5, 0x75,
  76.                           0xF5,
  77.                           0x0D, 0x8D, 0x4D, 0xCD, 0x2D, 0xAD, 0x6D, 0xED, 0x1D, 0x9D, 0x5D, 0xDD, 0x3D, 0xBD, 0x7D,
  78.                           0xFD,
  79.                           0x03, 0x83, 0x43, 0xC3, 0x23, 0xA3, 0x63, 0xE3, 0x13, 0x93, 0x53, 0xD3, 0x33, 0xB3, 0x73,
  80.                           0xF3,
  81.                           0x0B, 0x8B, 0x4B, 0xCB, 0x2B, 0xAB, 0x6B, 0xEB, 0x1B, 0x9B, 0x5B, 0xDB, 0x3B, 0xBB, 0x7B,
  82.                           0xFB,
  83.                           0x07, 0x87, 0x47, 0xC7, 0x27, 0xA7, 0x67, 0xE7, 0x17, 0x97, 0x57, 0xD7, 0x37, 0xB7, 0x77,
  84.                           0xF7,
  85.                           0x0F, 0x8F, 0x4F, 0xCF, 0x2F, 0xAF, 0x6F, 0xEF, 0x1F, 0x9F, 0x5F, 0xDF, 0x3F, 0xBF, 0x7F,
  86.                           0xFF]
  87.  
  88.     lav = BitReverseTable256[val & 0xff] << 24 | \
  89.         BitReverseTable256[(val >> 8) & 0xff] << 16 | \
  90.         BitReverseTable256[(val >> 16) & 0xff] << 8 | \
  91.         BitReverseTable256[(val >> 24) & 0xff]
  92.  
  93.     lav = lav >> 1
  94.     print(format(val, '031b') + " - {}".format(val))
  95.     print(format(lav, '031b') + " - {}".format(lav))
  96.     return lav
  97.  
  98.  
  99. input = []
  100.  
  101. with open('day03_input', 'r') as input_file:
  102.     for line in input_file:
  103.         input.append(line.strip())
  104.  
  105. blueprint = {
  106.     "blueprint": {
  107.         "icons": [
  108.  
  109.         ],
  110.         "entities": [
  111.  
  112.         ]
  113.     }
  114. }
  115.  
  116. values = []
  117. # input.append('...............................')
  118. # input.append('###############################')
  119.  
  120. for y in range(len(input)):
  121.     value = 0
  122.     for x in range(31):
  123.         bit = input[y][x]
  124.         if bit == '#':
  125.             value = value << 1
  126.             value = value + 1
  127.         else:
  128.             value = value << 1
  129.     print(input[y])
  130.     eulav = reverse_bits(value)
  131.     values.append([y, value])
  132.     x_offset = 0
  133.     y_offset = y * 2
  134.     entity_num = y + 1
  135.     blueprint['blueprint']['entities'].append(add_combinator(entity_num, eulav, x, x_offset, y + 1, y_offset))
  136.  
  137. write_blueprint(blueprint)
Add Comment
Please, Sign In to add comment