Create day12.py

2020/Python/day12.py

@@ -0,0 +1,72 @@
+from enum import Enum
+from typing import Tuple
+
+
+class Direction(Enum):
+    NORTH = 1
+    EAST = 2
+    SOUTH = 4
+    WEST = 8
+
+
+def parse_data() -> list[(str, int)]:
+    data = []
+    with open("input.txt") as file:
+        for line in file:
+            char, *number = line.rstrip()
+            data.append((char, int("".join(number))))
+    return data
+
+
+def apply_command(curr_pos: Tuple[int, int], curr_dir: Direction, command: Tuple[str, int]) -> ((int, int), Direction):
+    curr_x, curr_y = curr_pos
+    _char, _number = command
+
+    def find_new_direction(old_direction: Direction, char: str, number: int) -> Direction:
+        if char == "L":
+            result = (old_direction.value << 4) >> ((number % 360) // 90)
+            if result > 8:
+                result = result >> 4
+            return Direction(result)
+        elif char == "R":
+            result = old_direction.value << ((number % 360) // 90)
+            if result > 8:
+                result = result >> 4
+            return Direction(result)
+
+    if _char == 'N':
+        return (curr_x, curr_y + _number), curr_dir
+    elif _char == 'S':
+        return (curr_x, curr_y - _number), curr_dir
+    elif _char == 'E':
+        return (curr_x - _number, curr_y), curr_dir
+    elif _char == 'W':
+        return (curr_x + _number, curr_y), curr_dir
+
+    elif _char in "LR":
+        return curr_pos, find_new_direction(curr_dir, _char, _number)
+
+    elif _char == 'F':
+        if curr_dir is Direction.NORTH:
+            return (curr_x, curr_y + _number), curr_dir
+        elif curr_dir is Direction.SOUTH:
+            return (curr_x, curr_y - _number), curr_dir
+        elif curr_dir is Direction.EAST:
+            return (curr_x - _number, curr_y), curr_dir
+        elif curr_dir is Direction.WEST:
+            return (curr_x + _number, curr_y), curr_dir
+
+
+def solve_p1(data: list[Tuple[str, int]]):
+    current_direction = Direction.EAST
+    current_position = 0, 0
+    for command in data:
+        new_position, new_direction = apply_command(current_position, current_direction, command)
+        current_direction = new_direction
+        current_position = new_position
+    final_x, final_y = current_position
+    return abs(final_x)+abs(final_y)
+
+
+DATA = parse_data()
+print(solve_p1(DATA))