"""Integration tests for OpenSees Example 1b elastic portal frame."""

from __future__ import annotations

import tempfile
from pathlib import Path

import pytest

pytest.importorskip("openseespy")

from otko.core import PushoverCase, TransientCase  # noqa: E402
from otko.services import load_project, save_project  # noqa: E402
from otko.services.opensees_runner import OpenSeesRunner  # noqa: E402

pytestmark = pytest.mark.slow


def _reload(proj, tmp_path):  # type: ignore[no-untyped-def]
    path = tmp_path / "ex1b_portal2d.osmodel"
    save_project(proj, path)
    reloaded = load_project(path)
    reloaded.validate_references()
    return reloaded


def test_ex1b_portal2d_pushover_reaches_target(tmp_path) -> None:  # type: ignore[no-untyped-def]
    from examples.ex1b_portal2d import PUSH_STEP, PUSH_TARGET, build_ex1b_portal2d

    proj = _reload(build_ex1b_portal2d(), tmp_path)
    push_case = next(c for c in proj.analyses if isinstance(c, PushoverCase))
    result = OpenSeesRunner(proj).run(push_case)

    expected_pts = int(PUSH_TARGET / PUSH_STEP) + 1
    assert len(result.control_disp) == expected_pts
    assert result.control_disp[-1] == pytest.approx(PUSH_TARGET, rel=1e-6)
    assert result.base_shear[-1] > 0.0

    slope_early = (result.base_shear[10] - result.base_shear[0]) / (
        result.control_disp[10] - result.control_disp[0]
    )
    slope_late = (result.base_shear[-1] - result.base_shear[-11]) / (
        result.control_disp[-1] - result.control_disp[-11]
    )
    assert slope_early == pytest.approx(slope_late, rel=0.02)


def test_ex1b_portal2d_earthquake_runs_and_moves_symmetrically(tmp_path) -> None:  # type: ignore[no-untyped-def]
    from examples.ex1b_portal2d import ANALYSIS_DT, ANALYSIS_STEPS, build_ex1b_portal2d

    proj = _reload(build_ex1b_portal2d(), tmp_path)
    eq_case = next(c for c in proj.analyses if isinstance(c, TransientCase))

    results_dir = Path(tempfile.mkdtemp(prefix="ex1b_portal2d_eq_"))
    result = OpenSeesRunner(proj).run(eq_case, results_dir=results_dir)

    time = result.time()
    left = result.node_disp_history(3)
    right = result.node_disp_history(4)
    ux_left = left[:, 0]
    ux_right = right[:, 0]

    assert len(time) == ANALYSIS_STEPS
    assert result.dt == pytest.approx(ANALYSIS_DT)
    assert ux_left.max() > 1e-4
    assert ux_left.min() < -1e-4
    assert ux_right.max() > 1e-4
    assert ux_right.min() < -1e-4
    assert ux_left == pytest.approx(ux_right, rel=1e-3, abs=1e-6)