import bpy
import numpy as np
import pytest
from anyimage.operators.cutout_tool.object import load_cutout_node_group
from anyimage.operators.cutout_tool.operators import normal_mode_for_shape
from anyimage.operators.cutout_tool.mesh import build_mesh
from anyimage.operators.cutout_tool.balloon import poisson_balloon_profile


@pytest.mark.parametrize("enabled", [False, True])
def test_depth_presets_always_generate_the_correct_normal_space(enabled):
    assert normal_mode_for_shape("DEPTH_SOLID", enabled) == "OBJECT"
    assert normal_mode_for_shape("DEPTH_BALLOON", enabled) == "TANGENT"
    for shape in ("FLAT", "SOLID"):
        assert normal_mode_for_shape(shape, enabled) == ("TANGENT" if enabled else "NONE")


def test_new_cutout_does_not_reuse_a_saved_wrapper():
    bpy.ops.wm.read_factory_settings(use_empty=True)
    saved = bpy.data.node_groups.new("O Image Cutout", "GeometryNodeTree")
    saved.interface.new_socket(name="Shape", in_out="INPUT", socket_type="NodeSocketMenu")
    saved.use_fake_user = True
    current = load_cutout_node_group("SOLID")
    assert current != saved and current.get("anyimage_asset_revision") == 13
    assert saved in bpy.data.node_groups.values()
    assert current == load_cutout_node_group("FLAT")
    assert any(item.name == "Shape" for item in saved.interface.items_tree)


def test_depth_cutout_loads_uniform_thickness_revision_and_preserves_existing_group():
    bpy.ops.wm.read_factory_settings(use_empty=True)
    saved = bpy.data.node_groups.new("O Image Depth Cutout", "GeometryNodeTree")
    saved["anyimage_asset_revision"] = 11
    current = load_cutout_node_group("DEPTH_SOLID")
    assert current != saved and current["anyimage_asset_revision"] == 26
    assert saved in bpy.data.node_groups.values()
    assert saved["anyimage_asset_revision"] == 11
    assert load_cutout_node_group("DEPTH_SOLID") == current


def test_small_quality_mesh_has_positive_balloon_support():
    points, faces, boundary, _ = build_mesh(np.ones((2, 2)), 32, .9)
    profile = poisson_balloon_profile(points, faces, boundary)
    assert np.all(profile[boundary] == 0)
    assert profile.max() > 0