#pragma once

#include <array>
#include <optional>
#include <vector>

#include <fesa/core/diagnostic.hpp>
#include <fesa/core/vec3.hpp>
#include <fesa/fem/beam_frame.hpp>
#include <fesa/model/beam_section.hpp>
#include <fesa/model/material.hpp>

namespace fesa {

struct Beam3D2Input final {
    std::array<Vec3, 2> coordinates;
    IsotropicElastic material;
    BeamSection section;
};

struct Beam3D2Contribution final {
    Matrix12 local_stiffness;
    Matrix12 global_stiffness;
    BeamFrame frame;
};

struct BeamKernelResult final {
    std::optional<Beam3D2Contribution> contribution;
    std::vector<Diagnostic> diagnostics;
};

[[nodiscard]] BeamKernelResult compute_beam3d2(
    const Beam3D2Input& input);

}  // namespace fesa