#pragma once

#include "fesa/core/status.hpp"
#include "fesa/math/matrix.hpp"
#include "fesa/model/model_types.hpp"

#include <array>

namespace fesa {

struct Mitc4ShapeFunctions {
    std::array<double, 4> values;
    std::array<double, 4> xiDerivatives;
    std::array<double, 4> etaDerivatives;
};

struct Mitc4LocalFrame {
    std::array<double, 3> e1;
    std::array<double, 3> e2;
    std::array<double, 3> e3;
};

struct Mitc4TyingWeights {
    std::array<double, 2> xiZeta;
    std::array<double, 2> etaZeta;
};

struct Mitc4QuadraturePoint {
    std::array<double, 3> naturalCoordinates;
    double weight;
};

// Concrete small-rotation MITC4 kinematics and material value kernel. Global
// equation ownership, drilling stiffness, recovery, and assembly remain outside.
class Mitc4Shell {
public:
    static Result<Mitc4Shell> create(
        std::array<const Node*, 4> nodes,
        std::array<std::array<double, 3>, 4> initialDirectors,
        const ShellSection& section,
        const LinearElasticMaterial& material);

    static Mitc4ShapeFunctions shapeFunctions(double xi, double eta) noexcept;
    static Mitc4TyingWeights tyingWeights(double xi, double eta) noexcept;
    static const std::array<Mitc4QuadraturePoint, 8>&
    volumeQuadrature() noexcept;

    [[nodiscard]] Mitc4LocalFrame localFrame(double xi, double eta) const;
    [[nodiscard]] Matrix physicalTransformation20() const;
    [[nodiscard]] Matrix drillingTransformation4() const;
    [[nodiscard]] Matrix directStrainDisplacement20(
        double xi,
        double eta,
        double zeta) const;
    [[nodiscard]] Matrix covariantTyingShearSamples20() const;
    [[nodiscard]] Matrix strainDisplacement20(
        double xi,
        double eta,
        double zeta) const;

    [[nodiscard]] Matrix planeStressConstitutive() const;
    [[nodiscard]] Matrix materialConstitutive5() const;
    [[nodiscard]] Matrix membraneSectionMatrix() const;
    [[nodiscard]] Matrix bendingSectionMatrix() const;
    [[nodiscard]] Matrix transverseShearSectionMatrix() const;

private:
    using Vector3 = std::array<double, 3>;

    struct GeometryData {
        std::array<Vector3, 3> covariant;
        std::array<Vector3, 3> reciprocal;
        Mitc4LocalFrame frame;
        double jacobian;
    };

    Mitc4Shell(
        std::array<Vector3, 4> coordinates,
        std::array<Vector3, 4> directors,
        std::array<Vector3, 4> tangentA,
        std::array<Vector3, 4> tangentB,
        Vector3 normalCandidate,
        double thickness,
        double youngsModulus,
        double poissonRatio);

    bool evaluateGeometry(
        double xi,
        double eta,
        double zeta,
        GeometryData& result) const noexcept;
    std::array<std::array<Vector3, 3>, 20> basisDerivatives(
        double xi,
        double eta,
        double zeta) const noexcept;
    Matrix strainDisplacement(
        double xi,
        double eta,
        double zeta,
        const Matrix* tyingSamples) const;

    std::array<Vector3, 4> coordinates_;
    std::array<Vector3, 4> directors_;
    std::array<Vector3, 4> tangentA_;
    std::array<Vector3, 4> tangentB_;
    Vector3 normalCandidate_;
    double thickness_;
    double youngsModulus_;
    double poissonRatio_;
};

} // namespace fesa