module;

#include <mfem.hpp>

export module mean_field:operators.kernels.barotropic_closure;

export import :fem;
export import :mapping.domain_mapper;
export import :physics.barotrope;

export namespace mean_field::operators::kernels {
    void apply_barotropic_closure(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::PolytropicBarotrope &barotrope,
        const mfem::Vector &densityTrue,
        const mfem::Vector &enthalpyTrue,
        const mfem::Vector &displacementTrue,
        mfem::Vector &residual
    );

    void apply_barotropic_closure_density_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::PolytropicBarotrope &barotrope,
        const mfem::Vector &densityVariationTrue,
        const mfem::Vector &displacementTrue,
        mfem::Vector &action
    );

    void apply_barotropic_closure_enthalpy_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::PolytropicBarotrope &barotrope,
        const mfem::Vector &baseEnthalpyTrue,
        const mfem::Vector &enthalpyVariationTrue,
        const mfem::Vector &displacementTrue,
        mfem::Vector &action
    );

    void apply_barotropic_closure_displacement_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::PolytropicBarotrope &barotrope,
        const mfem::Vector &baseDensityTrue,
        const mfem::Vector &baseEnthalpyTrue,
        const mfem::Vector &displacementTrue,
        const mfem::Vector &displacementVariationTrue,
        mfem::Vector &action
    );
} // namespace mean_field::operators::kernels