"""Tests for tracking transforms and feature operations."""

import unittest
import numpy as np

import simplecadapi as scad
from simplecadapi.topology import TopoKind, TopoEvent, TopoRef, TopoDelta
from simplecadapi.tracking import (
    tracked_translate,
    tracked_rotate,
    tracked_extrude,
    tracked_fillet,
    tracked_chamfer,
    TrackedResult,
)


class TestTrackedTranslate(unittest.TestCase):
    def setUp(self):
        self.box = scad.make_box_rsolid(10, 10, 10)

    def test_translate_returns_tracked_result(self):
        result = tracked_translate(self.box, (5, 0, 0))
        self.assertIsInstance(result, TrackedResult)
        self.assertIsNotNone(result.shape)

    def test_translate_preserves_volume(self):
        result = tracked_translate(self.box, (5, 0, 0))
        self.assertAlmostEqual(
            result.shape.get_volume(), self.box.get_volume(), places=6
        )

    def test_translate_all_faces_preserved(self):
        result = tracked_translate(self.box, (5, 0, 0))
        preserved = [r for r in result.delta.preserved if r.kind == TopoKind.FACE]
        original_faces = len(self.box.get_faces())
        self.assertEqual(len(preserved), original_faces)

    def test_translate_no_generated(self):
        result = tracked_translate(self.box, (5, 0, 0))
        self.assertEqual(len(result.delta.generated), 0)
        self.assertEqual(len(result.delta.deleted), 0)

    def test_translate_result_is_solid(self):
        result = tracked_translate(self.box, (5, 0, 0))
        self.assertIsInstance(result.shape, scad.Solid)


class TestTrackedRotate(unittest.TestCase):
    def setUp(self):
        self.box = scad.make_box_rsolid(10, 10, 10)

    def test_rotate_returns_tracked_result(self):
        result = tracked_rotate(self.box, 45.0, (0, 0, 1))
        self.assertIsInstance(result, TrackedResult)
        self.assertIsNotNone(result.shape)

    def test_rotate_preserves_volume(self):
        result = tracked_rotate(self.box, 45.0, (0, 0, 1))
        self.assertAlmostEqual(
            result.shape.get_volume(), self.box.get_volume(), places=4
        )

    def test_rotate_all_faces_preserved(self):
        result = tracked_rotate(self.box, 45.0, (0, 0, 1))
        preserved = [r for r in result.delta.preserved if r.kind == TopoKind.FACE]
        original_faces = len(self.box.get_faces())
        self.assertEqual(len(preserved), original_faces)

    def test_rotate_no_generated(self):
        result = tracked_rotate(self.box, 45.0, (0, 0, 1))
        self.assertEqual(len(result.delta.generated), 0)


class TestTrackedExtrude(unittest.TestCase):
    def setUp(self):
        self.profile = scad.make_rectangle_rface(5.0, 3.0)

    def test_extrude_returns_tracked_result(self):
        result = tracked_extrude(self.profile, (0, 0, 1), 10.0)
        self.assertIsInstance(result, TrackedResult)
        self.assertIsNotNone(result.shape)

    def test_extrude_has_correct_volume(self):
        result = tracked_extrude(self.profile, (0, 0, 1), 10.0)
        expected_vol = self.profile.get_area() * 10.0
        self.assertAlmostEqual(result.shape.get_volume(), expected_vol, places=5)

    def test_extrude_generates_new_faces(self):
        result = tracked_extrude(self.profile, (0, 0, 1), 10.0)
        self.assertGreater(len(result.delta.generated), 0)

    def test_extrude_has_modified_or_preserved_face(self):
        result = tracked_extrude(self.profile, (0, 0, 1), 10.0)
        # The profile face becomes part of the extruded solid (either modified or
        # generated depending on OCC version)
        has_changes = (
            len(result.delta.modified) > 0
            or len(result.delta.preserved) > 0
            or len(result.delta.generated) > 0
        )
        self.assertTrue(has_changes)


class TestTrackedFillet(unittest.TestCase):
    def setUp(self):
        self.box = scad.make_box_rsolid(10, 10, 10)
        # Select some edges for filleting
        self.edges = [self.box.get_edges(i) for i in range(4)]

    def test_fillet_returns_tracked_result(self):
        result = tracked_fillet(self.box, self.edges, 0.5)
        self.assertIsInstance(result, TrackedResult)
        self.assertIsNotNone(result.shape)

    def test_fillet_volume_decreased(self):
        result = tracked_fillet(self.box, self.edges, 0.5)
        self.assertLess(result.shape.get_volume(), self.box.get_volume())

    def test_fillet_has_modified_faces(self):
        result = tracked_fillet(self.box, self.edges, 0.5)
        # Fillet modifies edge-adjacent faces; OCC may report new toroidal faces
        # as Modified rather than Generated
        total_changes = len(result.delta.modified) + len(result.delta.generated)
        self.assertGreater(total_changes, 0)


class TestTrackedChamfer(unittest.TestCase):
    def setUp(self):
        self.box = scad.make_box_rsolid(10, 10, 10)
        self.edges = [self.box.get_edges(i) for i in range(4)]

    def test_chamfer_returns_tracked_result(self):
        result = tracked_chamfer(self.box, self.edges, 0.5)
        self.assertIsInstance(result, TrackedResult)
        self.assertIsNotNone(result.shape)

    def test_chamfer_volume_decreased(self):
        result = tracked_chamfer(self.box, self.edges, 0.5)
        self.assertLess(result.shape.get_volume(), self.box.get_volume())


if __name__ == "__main__":
    unittest.main()