import pytest
from mathutils import Vector

from blenderbim.bim.helper import clip_segment


BOUNDS = (10, 30, 10, 30)

SEGMENTS_INSIDE = (
    (Vector((15, 20)), Vector((25, 20))),
    (Vector((20, 15)), Vector((20, 25))),
    (Vector((15, 15)), Vector((25, 25))),
    (Vector((15, 25)), Vector((25, 15))),
)

SEGMENTS_OUTSIDE = (
    (Vector((15, 35)), Vector((25, 35))),
    (Vector((25, 35)), Vector((35, 25))),
    (Vector((35, 25)), Vector((35, 15))),
    (Vector((35, 15)), Vector((25, 5))),
    (Vector((25, 5)), Vector((15, 5))),
    (Vector((15, 5)), Vector((5, 15))),
    (Vector((5, 15)), Vector((5, 25))),
    (Vector((5, 25)), Vector((15, 35)))
)

SEGMENTS_CLIPPED = (
    ((Vector((5, 20)), Vector((15, 20))), (Vector((10, 20)), Vector((15, 20)))),
    ((Vector((25, 20)), Vector((35, 20))), (Vector((25, 20)), Vector((30, 20)))),
    ((Vector((20, 5)), Vector((20, 15))), (Vector((20, 10)), Vector((20, 15)))),
    ((Vector((20, 15)), Vector((20, 35))), (Vector((20, 15)), Vector((20, 30)))),
    ((Vector((5, 20)), Vector((20, 35))), (Vector((10, 25)), Vector((15, 30)))),
    ((Vector((5, 20)), Vector((20, 5))), (Vector((10, 15)), Vector((15, 10)))),
    ((Vector((35, 20)), Vector((20, 35))), (Vector((30, 25)), Vector((25, 30)))),
    ((Vector((35, 20)), Vector((20, 5))), (Vector((30, 15)), Vector((25, 10)))),
)

@pytest.mark.parametrize('segment', SEGMENTS_INSIDE)
def test_clip_inside(segment):
    clipped = clip_segment(BOUNDS, segment)
    assert clipped == segment


@pytest.mark.parametrize('segment', SEGMENTS_OUTSIDE)
def test_clip_outside(segment):
    clipped = clip_segment(BOUNDS, segment)
    assert clipped is None


@pytest.mark.parametrize('segment,expected', SEGMENTS_CLIPPED)
def test_clip_crossing(segment, expected):
    clipped = clip_segment(BOUNDS, segment)
    assert clipped == expected