module;
#include <mfem.hpp>
export module mean_field:operators.kernels.gravity_field;
export import :mapping.domain_mapper;
export import :fem;

export namespace mean_field::operators::kernels {

    void apply_mapped_hdiv_mass(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domain_mapper,
        const mfem::Vector &gravity_gradient_true,
        const mfem::Vector &displacement_true,
        mfem::Vector &action
    );

    void apply_mapped_source(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domain_mapper,
        const mfem::Vector &density_true,
        const mfem::Vector &displacement_true,
        mfem::Vector &action
    );

    void apply_mapped_hdiv_mass_variation(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domain_mapper,
        const mfem::Vector &gravity_gradient_true,
        const mfem::Vector &displacement_true,
        const mfem::Vector &displacement_variation_true,
        mfem::Vector &action
    );

    void apply_mapped_source_variation(
        const fem::FEM &f,
        const mapping::DomainMapperStateless &domain_mapper,
        const mfem::Vector &density_true,
        const mfem::Vector &displacement_true,
        const mfem::Vector &displacement_variation_true,
        mfem::Vector &action_variation
    );
} // namespace mean_field::operators::kernels