Function gen_points

examples/Chromancer/gen.py:68–98

(
    input: list[HexagonAngle], leds_per_strip: int, startPos: Point,
    exclude: list[int] | None = None,
    add_last: bool = False
)

def gen_points(
    input: list[HexagonAngle], leds_per_strip: int, startPos: Point,
    exclude: list[int] | None = None,
    add_last: bool = False
) -> list[Point]:
    points: list[Point] = []
    if (not input) or (not leds_per_strip):
        return points
    exclude = exclude or []
    # Start FSM. Start pointer get's put into the accumulator.
    curr_point: Point = Point(startPos.x, startPos.y)
    # points.append(curr_point)
    last_angle = input[0]
    for i,angle in enumerate(input):
        excluded = i in exclude
        values = list(range(leds_per_strip))
        last_angle = angle
        for v in values:
            last_angle = angle
            curr_point = next_point(curr_point, angle, SPACE_PER_LED)
            if not excluded:
                points.append(curr_point)
        #if i == len(input) - 1:
        #    break
        # Next starting point
        curr_point = next_point(curr_point, last_angle, SPACE_PER_LED)
        #if not excluded:
        #    points.append(curr_point)
    if add_last:
        points.append(curr_point)
    return points