#include "fesa/analysis/analysis_state.hpp"

#include <gtest/gtest.h>

#include <array>
#include <filesystem>
#include <utility>

namespace {

fesa::DofManager makeEmptyDofs() {
    fesa::ModelDefinition definition{};
    definition.sourcePath = "models/result-records.inp";
    definition.sourceContentIdentity = "fnv1a64:0123456789abcdef";
    definition.steps = {{
        "Step-1", {}, {}, 0.1, 1.0, 0.01, 1.0,
        {definition.sourcePath, 10U}}};

    auto domain = fesa::Domain::create(std::move(definition));
    EXPECT_TRUE(domain.hasValue());
    auto model = fesa::AnalysisModel::create(domain.value());
    EXPECT_TRUE(model.hasValue());
    auto dofs = fesa::DofManager::create(model.value());
    EXPECT_TRUE(dofs.hasValue());
    return std::move(dofs.value());
}

} // namespace

TEST(AnalysisState, PreservesStepFrameAndStableRowOrder) {
    const auto dofs = makeEmptyDofs();
    auto state = fesa::AnalysisState::create(dofs, {"Step-1", 0U});

    const fesa::EndpointResultRow firstEndpoint{
        2U,
        -1,
        {"Beam-1", 10, "010"},
        {1.0, 2.0, 3.0, 4.0, 5.0, 6.0},
        {7.0, 8.0, 9.0, 10.0}};
    const fesa::EndpointResultRow secondEndpoint{
        2U,
        1,
        {"Beam-1", 20, "020"},
        {11.0, 12.0, 13.0, 14.0, 15.0, 16.0},
        {17.0, 18.0, 19.0, 20.0}};
    const fesa::GaussResultRow firstGauss{
        2U, 1, {0.1, 0.2, 0.3, 0.4}, {1.1, 1.2, 1.3, 1.4}};
    const fesa::GaussResultRow secondGauss{
        2U, 2, {0.5, 0.6, 0.7, 0.8}, {1.5, 1.6, 1.7, 1.8}};
    const fesa::StressS11Row firstStress{
        2U, 1, 1U, -0.25, 0.5, 12.5, "input"};
    const fesa::StressS11Row secondStress{
        2U, 1, 2U, 0.25, -0.5, -7.5, "input"};

    state.endpointResults().push_back(firstEndpoint);
    state.endpointResults().push_back(secondEndpoint);
    state.gaussResults().push_back(firstGauss);
    state.gaussResults().push_back(secondGauss);
    state.stressResults().push_back(firstStress);
    state.stressResults().push_back(secondStress);

    const auto* const endpointStorage = state.endpointResults().data();
    const auto* const gaussStorage = state.gaussResults().data();
    const auto* const stressStorage = state.stressResults().data();
    const fesa::AnalysisState& constState = state;

    EXPECT_EQ(constState.identity().stepName, "Step-1");
    EXPECT_EQ(constState.identity().frameIndex, 0U);
    ASSERT_EQ(constState.endpointResults().size(), 2U);
    EXPECT_EQ(constState.endpointResults().data(), endpointStorage);
    EXPECT_EQ(constState.endpointResults()[0].element, 2U);
    EXPECT_EQ(constState.endpointResults()[0].endpoint, -1);
    EXPECT_EQ(constState.endpointResults()[0].node.instanceName, "Beam-1");
    EXPECT_EQ(constState.endpointResults()[0].node.sourceLabel, 10);
    EXPECT_EQ(constState.endpointResults()[0].node.sourceLabelText, "010");
    EXPECT_EQ(
        constState.endpointResults()[0].endAction,
        (std::array<double, 6>{1.0, 2.0, 3.0, 4.0, 5.0, 6.0}));
    EXPECT_EQ(
        constState.endpointResults()[1].sectionResultant,
        (std::array<double, 4>{17.0, 18.0, 19.0, 20.0}));

    ASSERT_EQ(constState.gaussResults().size(), 2U);
    EXPECT_EQ(constState.gaussResults().data(), gaussStorage);
    EXPECT_EQ(constState.gaussResults()[0].gaussPoint, 1);
    EXPECT_EQ(constState.gaussResults()[1].gaussPoint, 2);
    EXPECT_EQ(
        constState.gaussResults()[0].generalizedStrain,
        (std::array<double, 4>{0.1, 0.2, 0.3, 0.4}));
    EXPECT_EQ(
        constState.gaussResults()[1].generalizedResultant,
        (std::array<double, 4>{1.5, 1.6, 1.7, 1.8}));

    ASSERT_EQ(constState.stressResults().size(), 2U);
    EXPECT_EQ(constState.stressResults().data(), stressStorage);
    EXPECT_EQ(constState.stressResults()[0].sectionPoint, 1U);
    EXPECT_DOUBLE_EQ(constState.stressResults()[0].x1, -0.25);
    EXPECT_DOUBLE_EQ(constState.stressResults()[0].x2, 0.5);
    EXPECT_DOUBLE_EQ(constState.stressResults()[0].s11, 12.5);
    EXPECT_EQ(constState.stressResults()[0].source, "input");
    EXPECT_EQ(constState.stressResults()[1].sectionPoint, 2U);
}