module;

#include <mfem.hpp>

export module mean_field:operators.kernels.hydrostatic_equilibrium;

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

export namespace mean_field::operators::kernels {
    void apply_hydrostatic_equilibrium(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::RigidRotation &rotation,
        const mfem::Vector &enthalpyTrue,
        const mfem::Vector &potentialTrue,
        const mfem::Vector &displacementTrue,
        double bernoulliConstant,
        mfem::Vector &residual
    );

    void apply_hydrostatic_equilibrium_enthalpy_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const mfem::Vector &enthalpyVariationTrue,
        const mfem::Vector &displacementTrue,
        mfem::Vector &action
    );

    void apply_hydrostatic_equilibrium_potential_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const mfem::Vector &potentialVariationTrue,
        const mfem::Vector &displacementTrue,
        mfem::Vector &action
    );

    void apply_hydrostatic_equilibrium_constant_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        double constantVariation,
        const mfem::Vector &displacementTrue,
        mfem::Vector &action
    );

    void apply_hydrostatic_equilibrium_displacement_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::RigidRotation &rotation,
        const mfem::Vector &baseEnthalpyTrue,
        const mfem::Vector &basePotentialTrue,
        const mfem::Vector &baseDisplacementTrue,
        double baseBernoulliConstant,
        const mfem::Vector &displacementVariationTrue,
        mfem::Vector &action
    );

    void apply_hydrostatic_equilibrium_action(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domainMapper,
        const physics::RigidRotation &rotation,
        const mfem::Vector &baseEnthalpyTrue,
        const mfem::Vector &basePotentialTrue,
        const mfem::Vector &baseDisplacementTrue,
        double baseBernoulliConstant,
        const mfem::Vector &enthalpyVariationTrue,
        const mfem::Vector &potentialVariationTrue,
        double constantVariation,
        const mfem::Vector &displacementVariationTrue,
        mfem::Vector &action
    );
} // namespace mean_field::operators::kernels