day10_stars.py

import re
from typing import List, NamedTuple

import pandas as pd

TEST_INPUT = """position=< 9,  1> velocity=< 0,  2>
position=< 7,  0> velocity=<-1,  0>
position=< 3, -2> velocity=<-1,  1>
position=< 6, 10> velocity=<-2, -1>
position=< 2, -4> velocity=< 2,  2>
position=<-6, 10> velocity=< 2, -2>
position=< 1,  8> velocity=< 1, -1>
position=< 1,  7> velocity=< 1,  0>
position=<-3, 11> velocity=< 1, -2>
position=< 7,  6> velocity=<-1, -1>
position=<-2,  3> velocity=< 1,  0>
position=<-4,  3> velocity=< 2,  0>
position=<10, -3> velocity=<-1,  1>
position=< 5, 11> velocity=< 1, -2>
position=< 4,  7> velocity=< 0, -1>
position=< 8, -2> velocity=< 0,  1>
position=<15,  0> velocity=<-2,  0>
position=< 1,  6> velocity=< 1,  0>
position=< 8,  9> velocity=< 0, -1>
position=< 3,  3> velocity=<-1,  1>
position=< 0,  5> velocity=< 0, -1>
position=<-2,  2> velocity=< 2,  0>
position=< 5, -2> velocity=< 1,  2>
position=< 1,  4> velocity=< 2,  1>
position=<-2,  7> velocity=< 2, -2>
position=< 3,  6> velocity=<-1, -1>
position=< 5,  0> velocity=< 1,  0>
position=<-6,  0> velocity=< 2,  0>
position=< 5,  9> velocity=< 1, -2>
position=<14,  7> velocity=<-2,  0>
position=<-3,  6> velocity=< 2, -1>""".splitlines()


class Star(NamedTuple):
    x: int
    y: int
    x_velocity: int
    y_velocity: int


test_stars = [
    Star(*map(int, re.findall(r"-?\d+", line))) for line in TEST_INPUT
]


def extract_points(stars: List[Star], *, t: int):
    xs = [star.x + t * star.x_velocity for star in stars]
    ys = [star.y + t * star.y_velocity for star in stars]
    return {"x": xs, "y": ys}


def driver(stars: List[Star]):
    for time in range(0, 5):
        points = extract_points(stars, t=time)


if __name__ == "__main__":
    points = pd.DataFrame.from_dict(extract_points(test_stars, t=3))