"""Tests for Polyline / PolylineShape (the swept trace copper)."""

from __future__ import annotations

from freeroute.geometry import IntBox, IntPoint, Polyline, PolylineShape, segment_box


def test_polyline_segments():
    pl = Polyline([IntPoint(0, 0), IntPoint(10, 0), IntPoint(10, 10)])
    segs = list(pl.segments())
    assert len(segs) == 2
    assert pl.corner_count() == 3
    assert not pl.is_empty()
    assert Polyline([IntPoint(0, 0)]).is_empty()


def test_segment_box_horizontal_is_flush_in_x():
    box = segment_box(IntPoint(0, 0), IntPoint(100, 0), 10)
    # flush in the travel direction (x), offset by half-width in y
    assert box == IntBox(0, -10, 100, 10)


def test_segment_box_vertical_is_flush_in_y():
    box = segment_box(IntPoint(0, 0), IntPoint(0, 100), 10)
    assert box == IntBox(-10, 0, 10, 100)


def test_polyline_shape_tiles_include_corner_squares():
    pl = Polyline([IntPoint(0, 0), IntPoint(100, 0), IntPoint(100, 100)])
    tiles = PolylineShape(pl, 10).tiles()
    # 2 segment boxes + 1 interior-corner square
    assert len(tiles) == 3
    # the corner square is centred on the turn at (100, 0)
    assert IntBox(90, -10, 110, 10) in tiles


def test_polyline_shape_bounding_box():
    pl = Polyline([IntPoint(0, 0), IntPoint(100, 0)])
    assert PolylineShape(pl, 5).bounding_box() == IntBox(-5, -5, 105, 5)


def test_diagonal_segment_box_is_conservative_bbox():
    box = segment_box(IntPoint(0, 0), IntPoint(50, 50), 10)
    assert box == IntBox(-10, -10, 60, 60)