Samuel Enocsson
549f0c4382
🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
"""Input parsing and common utilities for Advent of Code."""
|
|
|
|
from pathlib import Path
|
|
from typing import Callable, Any
|
|
import numpy as np
|
|
|
|
|
|
def get_input_path(day: int) -> Path:
|
|
"""Get the path to input file for a given day."""
|
|
return Path(__file__).parent.parent / "inputs" / f"day{day:02d}.txt"
|
|
|
|
|
|
def read_input(day: int) -> str:
|
|
"""Read raw input for a given day."""
|
|
return get_input_path(day).read_text().strip()
|
|
|
|
def read_lines_and_split(day: int, split: str) -> list:
|
|
"""Read input as lines, optionally parsing each line."""
|
|
return [line for line in read_input(day).split(split)]
|
|
|
|
def read_lines(day: int, parser: Callable[[str], Any] = str) -> list:
|
|
"""Read input as lines, optionally parsing each line."""
|
|
return [parser(line) for line in read_input(day).splitlines()]
|
|
|
|
|
|
def read_ints(day: int) -> list[int]:
|
|
"""Read input as list of integers (one per line)."""
|
|
return read_lines(day, int)
|
|
|
|
|
|
def read_grid(day: int, dtype=str) -> np.ndarray:
|
|
"""Read input as a 2D numpy grid of characters or digits."""
|
|
lines = read_input(day).splitlines()
|
|
if dtype == int:
|
|
return np.array([[int(c) for c in line] for line in lines])
|
|
return np.array([list(line) for line in lines])
|
|
|
|
|
|
def read_blocks(day: int) -> list[str]:
|
|
"""Read input split by blank lines (paragraphs)."""
|
|
return read_input(day).split("\n\n")
|
|
|
|
|
|
# Common grid directions
|
|
DIRS_4 = [(0, 1), (1, 0), (0, -1), (-1, 0)] # right, down, left, up
|
|
DIRS_8 = DIRS_4 + [(1, 1), (1, -1), (-1, 1), (-1, -1)] # + diagonals
|
|
|
|
# Direction mappings
|
|
DIR_MAP = {
|
|
"R": (0, 1), "L": (0, -1), "U": (-1, 0), "D": (1, 0),
|
|
">": (0, 1), "<": (0, -1), "^": (-1, 0), "v": (1, 0),
|
|
}
|