M tools/make-dotcode.py → tools/make-dotcode.py

@@ -3,11 +3,101 @@ import PIL.Image
 import PIL.ImageChops
 import sys
 
+blocksize = 104
+pow = [0] * 256
+rev = [0] * 256
+rev[0] = 0xFF
+x = 1
+for i in range(0xFF):
+    pow[i]=x
+    rev[x]=i
+    x *= 2
+    if x >= 0x100:
+        x ^= 0x187
+gg = [0] * 16
+gg[0] = pow[0x78]
+for i in range(16):
+    gg[i] = 1
+    for j in range(i + 1):
+        j = i - j
+        if j == 0:
+            y = 0
+        else:
+            y = gg[j - 1]
+        x = gg[j]
+        if x != 0:
+            x = rev[x] + 0x78 + i
+            if x >= 0xFF:
+                x -= 0xFF
+            y ^= pow[x]
+        gg[j] = y
+for i in range(16):
+    gg[i] = rev[gg[i]]
+
+print(*[hex(x) for x in gg])
+
+def interleave(data, header):
+    data = data.reshape([-1, 48]).T
+    new_data = np.zeros((64, data.shape[1]), dtype=data.dtype)
+    for i, row in enumerate(data.T):
+        new_data[:,i] = rs(row)
+    data = new_data.reshape([-1])
+    new_data = np.zeros(data.shape[0] + (102 - data.shape[0] % 102), dtype=data.dtype)
+    new_data[:data.shape[0]] = data
+    new_data = new_data.reshape([-1, 102])
+    data = new_data
+    new_data = np.zeros((data.shape[0], 104), dtype=data.dtype)
+    new_data[:,2:] = data
+    for i in range(new_data.shape[0]):
+        x = (i * 2) % len(header)
+        new_data[i,:2] = header[x:x + 2]
+    data = new_data.reshape([-1])
+    return data
+
+def rs(data):
+    new_data = np.zeros(data.shape[0] + 16, dtype=data.dtype)
+    new_data[:data.shape[0]] = data
+    new_data = np.flipud(new_data)
+    for i in range(data.shape[0]):
+        i = new_data.shape[0] - i - 1
+        z = rev[new_data[i] ^ new_data[15]]
+        for j in range(16):
+            j = 15 - j
+            if j == 0:
+                x = 0
+            else:
+                x = new_data[j - 1]
+            if z != 0xFF:
+                y = gg[j]
+                if y != 0xFF:
+                    y += z
+                    if y >= 0xFF:
+                        y -= 0xFF
+                    x ^= pow[y]
+            new_data[j] = x
+    new_data[:16] = ~new_data[:16]
+    new_data = np.flipud(new_data)
+    return new_data
+
+def bin2raw(data):
+    if len(data) == 1344:
+        header = np.array([0x00, 0x02, 0x00, 0x01, 0x40, 0x10, 0x00, 0x1C], dtype=np.uint8)
+    else:
+        header = np.array([0x00, 0x03, 0x00, 0x19, 0x40, 0x10, 0x00, 0x2C], dtype=np.uint8)
+    header = rs(header)
+    new_data = interleave(np.frombuffer(data, np.uint8), header)
+    return new_data.tobytes()
+
 with open(sys.argv[1], 'rb') as f:
     data = f.read()
 size = len(data)
 
-blocksize = 104
+if size in (1344, 2112):
+    data = bin2raw(data)
+    size = len(data)
+    with open('dotcode.raw', 'wb') as f:
+        f.write(data)
+
 blocks = size // blocksize
 height = 36
 width = 35