#ifndef CHRONOSEULERSIMULATIONMODEL_HPP
#define CHRONOSEULERSIMULATIONMODEL_HPP

#include "simulationmodel.hpp"

namespace chrono {
class ChSystemSMC;
namespace fea {
class ChMesh;
class ChNodeFEAxyzrot;
} // namespace fea
} // namespace chrono

class ChronosEulerSimulationModel : public SimulationModel
{
    std::shared_ptr<chrono::fea::ChMesh> mesh_chronos;
    std::vector<std::shared_ptr<chrono::fea::ChNodeFEAxyzrot>> edgeMeshVertsToChronoNodes;

public:
    ChronosEulerSimulationModel();
    SimulationResults executeSimulation(const std::shared_ptr<SimulationJob> &pJob) override;
    void setStructure(const std::shared_ptr<SimulationMesh> &pMesh) override;
    static std::shared_ptr<chrono::fea::ChMesh> convertToChronosMesh_Euler(
        const std::shared_ptr<SimulationMesh> &pMesh,
        std::vector<std::shared_ptr<chrono::fea::ChNodeFEAxyzrot>> &edgeMeshVertsToChronosNodes);

private:
    static void parseConstrainedVertices(
        const std::shared_ptr<SimulationJob> &pJob,
        const std::vector<std::shared_ptr<chrono::fea::ChNodeFEAxyzrot>> &edgeMeshVertsToChronoNodes,
        chrono::ChSystemSMC &my_system);
    static void parseForces(
        const std::shared_ptr<chrono::fea::ChMesh> &mesh_chronos,
        const std::vector<std::shared_ptr<chrono::fea::ChNodeFEAxyzrot>> &edgeMeshVertsToChronoNodes,
        const std::unordered_map<VertexIndex, Vector6d> &nodalExternalForces);
};

#endif // CHRONOSEULERSIMULATIONMODEL_HPP