"""Smoke tests for the undeformed-reference overlay on both backends."""

from __future__ import annotations

import numpy as np
import pytest

pytest.importorskip("PySide6")

from otko.core import ElasticBeamColumn, ElasticSection, Node, Project


def _frame_project() -> Project:
    b, h = 0.30, 0.50
    return Project(
        nodes=[
            Node(id=1, coords=(0.0, 0.0, 0.0), restraint=(True, True, True, True, True, True)),
            Node(id=2, coords=(6.0, 0.0, 0.0)),
        ],
        sections=[
            ElasticSection(
                id=1,
                name="Rect",
                E=200e9,
                A=b * h,
                Iz=b * h**3 / 12.0,
                Iy=h * b**3 / 12.0,
                G=80e9,
                J=1e-6,
            )
        ],
        elements=[ElasticBeamColumn(id=1, nodes=(1, 2), section_id=1)],
    )


@pytest.mark.gui
def test_pyvista_undeformed_reference_overlay(qtbot) -> None:  # type: ignore[no-untyped-def]
    from otko.services.deformation import DeformationSource
    from otko.views.canvas3d.model_canvas import ModelCanvas
    from otko.views.canvas3d.model_renderer import RendererMode

    canvas = ModelCanvas()
    qtbot.addWidget(canvas)
    canvas.show_project(_frame_project())

    # In MODEL mode there is nothing to reference — the toggle must be a no-op.
    canvas.set_show_undeformed(True)
    assert canvas._renderer._undeformed_actor is None

    # A deformed shape gets a ghost wireframe, removed again on toggle off.
    src = DeformationSource(
        displacements=np.array([[0.0, 0.0, 0.0], [0.0, 0.0, 0.5]]),
        node_id_to_row={1: 0, 2: 1},
    )
    canvas._renderer.set_mode(RendererMode.DEFORMED, src)
    canvas.render()
    canvas.set_show_undeformed(True)
    assert canvas._renderer._undeformed_actor is not None

    canvas.set_show_undeformed(False)
    assert canvas._renderer._undeformed_actor is None