aoc_ocr

tools/aoc_ocr.py

@@ -0,0 +1,61 @@
+# Copyright (c) 2020-present Benjamin Soyka
+# Original and Licence at https://github.com/bsoyka/advent-of-code-ocr
+
+from collections.abc import Sequence
+
+ALPHABET_6 = {
+    ".##.\n#..#\n#..#\n####\n#..#\n#..#": "A",
+    "###.\n#..#\n###.\n#..#\n#..#\n###.": "B",
+    ".##.\n#..#\n#...\n#...\n#..#\n.##.": "C",
+    "####\n#...\n###.\n#...\n#...\n####": "E",
+    "####\n#...\n###.\n#...\n#...\n#...": "F",
+    ".##.\n#..#\n#...\n#.##\n#..#\n.###": "G",
+    "#..#\n#..#\n####\n#..#\n#..#\n#..#": "H",
+    ".###\n..#.\n..#.\n..#.\n..#.\n.###": "I",
+    "..##\n...#\n...#\n...#\n#..#\n.##.": "J",
+    "#..#\n#.#.\n##..\n#.#.\n#.#.\n#..#": "K",
+    "#...\n#...\n#...\n#...\n#...\n####": "L",
+    ".##.\n#..#\n#..#\n#..#\n#..#\n.##.": "O",
+    "###.\n#..#\n#..#\n###.\n#...\n#...": "P",
+    "###.\n#..#\n#..#\n###.\n#.#.\n#..#": "R",
+    ".###\n#...\n#...\n.##.\n...#\n###.": "S",
+    "#..#\n#..#\n#..#\n#..#\n#..#\n.##.": "U",
+    "#...\n#...\n.#.#\n..#.\n..#.\n..#.": "Y",
+    "####\n...#\n..#.\n.#..\n#...\n####": "Z",
+}
+
+
+def convert_6(input_text: str, *, fill_pixel: str = "#", empty_pixel: str = ".") -> str:
+    """Convert height 6 text to characters"""
+    input_text = input_text.replace(fill_pixel, "#").replace(empty_pixel, ".")
+    prepared_array = [list(line) for line in input_text.split("\n")]
+    return _convert_6(prepared_array)
+
+
+def convert_array_6(array: Sequence[Sequence[str | int]], *, fill_pixel: str | int = "#", empty_pixel: str | int = ".") -> str:
+    """Convert a height 6 NumPy array or nested list to characters"""
+    prepared_array = [
+        [
+            "#" if pixel == fill_pixel else "." if pixel == empty_pixel else ""
+            for pixel in line
+        ]
+        for line in array
+    ]
+    return _convert_6(prepared_array)
+
+
+def _convert_6(array: list[list[str]]) -> str:
+    """Convert a prepared height 6 array to characters"""
+    rows, cols = len(array), len(array[0])
+    if any(len(row) != cols for row in array):
+        raise ValueError("all rows should have the same number of columns")
+    if rows != 6:
+        raise ValueError("incorrect number of rows (expected 6)")
+
+    indices = [slice(start, start + 4) for start in range(0, cols, 5)]
+    result = [
+        ALPHABET_6["\n".join("".join(row[index]) for row in array)]
+        for index in indices
+    ]
+
+    return "".join(result)

tools/grid.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+from .aoc_ocr import convert_array_6
 from .coordinate import Coordinate, DistanceAlgorithm, Shape
 from .types import Numeric
 from enum import Enum
@@ -367,6 +368,9 @@ class Grid:
 
             print()
 
+    def get_aoc_ocr_string(self, x_shift: int = 0, y_shift: int = 0):
+        return convert_array_6([['#' if self.get(Coordinate(x + x_shift, y + y_shift)) else '.' for x in self.rangeX()] for y in self.rangeY()])
+
     def __str__(self, true_char: str = '#', false_char: str = "."):
         return "/".join(
             "".join(