"""Pure-NumPy overlay geometry for Finger 3D arrows."""

from __future__ import annotations

import numpy as np
from orisys_app_shared.scene3d.overlays import (
    arrow_instances_to_mesh,
    build_flow_overlay,
    build_normal_overlay,
    force_visual_strength,
)
from orisys_app_shared.scene3d.synthetic import poc_scene
from orisys_app_shared.scene3d.visual_policy import VisualPolicy, select_flow_samples


def test_force_visual_strength_is_capped() -> None:
    assert force_visual_strength(0.0) == 0.0
    strong = force_visual_strength(1e9)
    mild = force_visual_strength(100.0)
    assert strong >= mild > 0.0
    assert strong <= 1.0 + 1e-9


def test_flow_and_normal_overlays_on_poc_mapping() -> None:
    assets = poc_scene()
    mapping = assets.surface_mapping
    assert mapping is not None
    h, w = mapping.shape_hw
    flow = np.zeros((h, w, 2), dtype=np.float64)
    flow[1, 1] = (0.8, 0.0)
    mask = np.asarray(mapping.valid, dtype=bool)
    policy = VisualPolicy(
        flow_on_px=0.4,
        flow_grid_spacing=1,
        flow_edge_margin=0,
        max_arrows=50,
    )
    samples = select_flow_samples(flow, mask, policy)
    arrows, diags = build_flow_overlay(samples, mapping, style=policy)
    assert not diags
    assert arrows is not None and arrows.origins.shape[0] >= 1
    contact, cdiags = build_normal_overlay(
        contact_uv_xy=(1.0, 1.0),
        contact_found=True,
        contact_valid=True,
        fn=500.0,
        mapping=mapping,
    )
    assert not cdiags
    assert contact is not None and contact.origins.shape[0] == 1
    # Yellow-ish contact colour.
    assert float(contact.colors[0, 0]) > 0.7
    assert float(contact.colors[0, 1]) > 0.7

    positions, triangles, colors = arrow_instances_to_mesh(arrows)
    assert positions.ndim == 2 and positions.shape[1] == 3
    assert triangles.ndim == 2 and triangles.shape[1] == 3
    assert colors.shape[0] == positions.shape[0]


def test_missing_contact_yields_no_arrow() -> None:
    assets = poc_scene()
    mapping = assets.surface_mapping
    assert mapping is not None
    arrows, _ = build_normal_overlay(
        contact_uv_xy=(1.0, 1.0),
        contact_found=False,
        contact_valid=True,
        fn=100.0,
        mapping=mapping,
    )
    assert arrows.origins.shape[0] == 0