# mesh_instance_test.py - Unit test for MeshLayer mesh instances and Viewport3D mesh APIs

import mcrfpy
import sys

def test_viewport3d_add_mesh():
    """Test adding meshes to Viewport3D layers"""
    vp = mcrfpy.Viewport3D()

    # Add a mesh layer first
    vp.add_layer("ground", z_index=0)

    # Create a model to place (simple cube primitive)
    model = mcrfpy.Model3D()

    # Add mesh instance at position
    result = vp.add_mesh("ground", model, pos=(5.0, 0.0, 5.0))

    # Should return the index of the added mesh
    assert result is not None, "Expected add_mesh to return something"
    assert isinstance(result, int), f"Expected int index, got {type(result)}"
    assert result == 0, f"Expected first mesh index 0, got {result}"

    print("[PASS] test_viewport3d_add_mesh")

def test_viewport3d_add_mesh_with_transform():
    """Test adding meshes with rotation and scale"""
    vp = mcrfpy.Viewport3D()
    vp.add_layer("buildings", z_index=0)

    model = mcrfpy.Model3D()

    # Add with rotation (in degrees as per API)
    idx1 = vp.add_mesh("buildings", model, pos=(10.0, 0.0, 10.0), rotation=90)
    assert idx1 == 0, f"Expected first mesh index 0, got {idx1}"

    # Add with scale
    idx2 = vp.add_mesh("buildings", model, pos=(15.0, 0.0, 15.0), scale=2.0)
    assert idx2 == 1, f"Expected second mesh index 1, got {idx2}"

    # Add with both rotation and scale
    idx3 = vp.add_mesh("buildings", model, pos=(5.0, 0.0, 5.0), rotation=45, scale=0.5)
    assert idx3 == 2, f"Expected third mesh index 2, got {idx3}"

    print("[PASS] test_viewport3d_add_mesh_with_transform")

def test_viewport3d_clear_meshes():
    """Test clearing meshes from a layer"""
    vp = mcrfpy.Viewport3D()
    vp.add_layer("objects", z_index=0)

    model = mcrfpy.Model3D()

    # Add several meshes
    vp.add_mesh("objects", model, pos=(1.0, 0.0, 1.0))
    vp.add_mesh("objects", model, pos=(2.0, 0.0, 2.0))
    vp.add_mesh("objects", model, pos=(3.0, 0.0, 3.0))

    # Clear meshes from layer
    vp.clear_meshes("objects")

    # Add a new mesh - should get index 0 since list was cleared
    idx = vp.add_mesh("objects", model, pos=(0.0, 0.0, 0.0))
    assert idx == 0, f"Expected index 0 after clear, got {idx}"

    print("[PASS] test_viewport3d_clear_meshes")

def test_viewport3d_place_blocking():
    """Test placing blocking information on the navigation grid"""
    vp = mcrfpy.Viewport3D()

    # Initialize navigation grid first
    vp.set_grid_size(width=16, depth=16)

    # Place blocking cell (unwalkable, non-transparent)
    vp.place_blocking(grid_pos=(5, 5), footprint=(1, 1))

    # Place larger blocking footprint
    vp.place_blocking(grid_pos=(10, 10), footprint=(2, 2))

    # Place blocking with custom walkability
    vp.place_blocking(grid_pos=(0, 0), footprint=(3, 3), walkable=False, transparent=True)

    # Verify the cells were marked (check via VoxelPoint)
    cell = vp.at(5, 5)
    assert cell.walkable == False, f"Expected cell (5,5) unwalkable, got walkable={cell.walkable}"

    cell_transparent = vp.at(0, 0)
    assert cell_transparent.transparent == True, f"Expected cell (0,0) transparent"

    print("[PASS] test_viewport3d_place_blocking")

def test_viewport3d_mesh_layer_operations():
    """Test various mesh layer operations"""
    vp = mcrfpy.Viewport3D()

    # Create multiple layers
    vp.add_layer("floor", z_index=0)
    vp.add_layer("walls", z_index=1)
    vp.add_layer("props", z_index=2)

    model = mcrfpy.Model3D()

    # Add meshes to different layers
    vp.add_mesh("floor", model, pos=(0.0, 0.0, 0.0))
    vp.add_mesh("walls", model, pos=(1.0, 1.0, 0.0), rotation=0, scale=1.5)
    vp.add_mesh("props", model, pos=(2.0, 0.0, 2.0), scale=0.25)

    # Clear only one layer
    vp.clear_meshes("walls")

    # Other layers should be unaffected
    # (Can verify by adding to them and checking indices)
    idx_floor = vp.add_mesh("floor", model, pos=(5.0, 0.0, 5.0))
    assert idx_floor == 1, f"Expected floor mesh index 1, got {idx_floor}"

    idx_walls = vp.add_mesh("walls", model, pos=(5.0, 0.0, 5.0))
    assert idx_walls == 0, f"Expected walls mesh index 0 after clear, got {idx_walls}"

    print("[PASS] test_viewport3d_mesh_layer_operations")

def test_auto_layer_creation():
    """Test that add_mesh auto-creates layers if they don't exist"""
    vp = mcrfpy.Viewport3D()
    model = mcrfpy.Model3D()

    # Add mesh to a layer that doesn't exist yet - should auto-create it
    idx = vp.add_mesh("auto_created", model, pos=(0.0, 0.0, 0.0))
    assert idx == 0, f"Expected index 0 for auto-created layer, got {idx}"

    # Verify the layer was created
    layer = vp.get_layer("auto_created")
    assert layer is not None, "Expected auto_created layer to exist"

    print("[PASS] test_auto_layer_creation")

def test_invalid_layer_clear():
    """Test error handling for clearing non-existent layers"""
    vp = mcrfpy.Viewport3D()

    # Try to clear meshes from non-existent layer
    try:
        vp.clear_meshes("nonexistent")
        # If it doesn't raise, it might just silently succeed (which is fine too)
        print("[PASS] test_invalid_layer_clear (no exception)")
        return
    except (ValueError, KeyError, RuntimeError):
        print("[PASS] test_invalid_layer_clear (exception raised)")
        return

def run_all_tests():
    """Run all mesh instance tests"""
    tests = [
        test_viewport3d_add_mesh,
        test_viewport3d_add_mesh_with_transform,
        test_viewport3d_clear_meshes,
        test_viewport3d_place_blocking,
        test_viewport3d_mesh_layer_operations,
        test_auto_layer_creation,
        test_invalid_layer_clear,
    ]

    passed = 0
    failed = 0

    for test in tests:
        try:
            test()
            passed += 1
        except AssertionError as e:
            print(f"[FAIL] {test.__name__}: {e}")
            failed += 1
        except Exception as e:
            print(f"[ERROR] {test.__name__}: {e}")
            failed += 1

    print(f"\n=== Results: {passed} passed, {failed} failed ===")
    return failed == 0

if __name__ == "__main__":
    success = run_all_tests()
    sys.exit(0 if success else 1)