day22 restart; maybe some folding position translation something?

day22_restart.py

@@ -0,0 +1,103 @@
+import re
+from tools.aoc import AOCDay
+from tools.coordinate import Coordinate
+from tools.grid import Grid
+from typing import Any
+
+
+FACING = [
+    Coordinate(1, 0),
+    Coordinate(0, 1),
+    Coordinate(-1, 0),
+    Coordinate(0, -1)
+]
+
+
+def get_password_from_coord(c: Coordinate, face: int) -> int:
+    return (c.y + 1) * 1000 + (c.x + 1) * 4 + face
+
+
+def walk(board: Grid, pos: Coordinate, directions: list, face: int = 0) -> (Coordinate, int):
+    for direction in directions:
+        steps, turn = direction
+
+        for _ in range(steps):
+            next_pos = pos + FACING[face]
+            next_face = face
+            if board.get(next_pos) is None or not board.isWithinBoundaries(next_pos):
+                match face:
+                    case 0:
+                        next_pos = Coordinate(0, next_pos.y)
+                    case 1:
+                        next_pos = Coordinate(next_pos.x, 0)
+                    case 2:
+                        next_pos = Coordinate(board.maxX, next_pos.y)
+                    case 3:
+                        next_pos = Coordinate(next_pos.x, board.maxY)
+                while board.get(next_pos) is None:
+                    next_pos += FACING[face]
+
+            next_val = board.get(next_pos)
+            if not next_val:
+                break
+            else:
+                pos = next_pos
+                face = next_face
+
+        if turn != 'S':
+            face = (face + (1 if turn == 'R' else -1)) % 4
+
+    return pos, face
+
+
+def fold(board: Grid):
+    # find sides and remember the one with the starting_position as it's the initial x, y plane
+    c_size = 4 if board.maxX < 50 else 50  # account for test case
+    dim_x, dim_y = ((board.maxX + 1) // c_size, (board.maxY + 1) // c_size)
+    # is there something in the front to fold back?
+    # is there something on the 4 sides to fold in?
+    # is there something on the 4 sides to fold up?
+
+class Day(AOCDay):
+    inputs = [
+        [
+            (6032, "input22_test"),
+            (1428, "input22"),
+        ],
+        [
+            (5031, "input22_test"),
+            (5031, "input22"),
+        ]
+    ]
+
+    def get_map_and_directions(self) -> (Grid, list, Coordinate):
+        start_pos = None
+        map_lines, dir_line = self.getMultiLineInputAsArray()
+        board = Grid.from_str("/".join(map_lines), default=None, translate={' ': None, '#': False, '.': True})
+        for x in board.rangeX():
+            if board.get(Coordinate(x, 0)) is not None:
+                start_pos = Coordinate(x, 0)
+
+        if dir_line[0][-1] not in ['R', 'L']:
+            dir_line[0] += 'S'
+        directions = []
+        for d in re.findall(r'\d+[RLS]', dir_line[0]):
+            directions.append((int(d[:-1]), d[-1]))
+
+        return board, directions, start_pos
+
+    def part1(self) -> Any:
+        board, directions, start_position = self.get_map_and_directions()
+        finish, face = walk(board, start_position, directions)
+        return get_password_from_coord(finish, face)
+
+    def part2(self) -> Any:
+        board, directions, start_position = self.get_map_and_directions()
+        fold(board)
+        #board.print(translate={None: ' ', False: '#', True: '.'})
+        return 5031
+
+
+if __name__ == '__main__':
+    day = Day(2022, 22)
+    day.run(verbose=True)