feat(libmeanfield): centrifugal + pressure

libmeanfield/interface/integrators/gravity.cppm

@@ -1,14 +1,29 @@
 module;
+#include <cstdint>
 #include <mfem.hpp>
 export module mean_field:integrators.gravity;
 import :mapping.domain_mapper;
 
 export namespace mean_field::integrators {
-    class GravityForceIntegrator : public mfem::BlockNonlinearFormIntegrator {
-    public:
-        GravityForceIntegrator(const mapping::DomainMapper& map, const mfem::GridFunction& phi);
+    enum class GravityForceJacobianMode : std::uint8_t {
+        minimal,
+        field_coupled,
+        exact
+    };
 
-        void SetPotential(const mfem::GridFunction& phi);
+    class GravityMomentumIntegrator
+        : public mfem::BlockNonlinearFormIntegrator {
+    public:
+        explicit GravityMomentumIntegrator(
+            const mapping::DomainMapper &map,
+            GravityForceJacobianMode jacobian_mode =
+                GravityForceJacobianMode::field_coupled
+        );
+
+        void SetJacobianMode(GravityForceJacobianMode jacobian_mode);
+        void SetIntegrationRule(const mfem::IntegrationRule &integration_rule);
+
+        [[nodiscard]] GravityForceJacobianMode GetJacobianMode() const;
 
         void AssembleElementVector(
             const mfem::Array<const mfem::FiniteElement *> &el,
@@ -16,7 +31,6 @@ export namespace mean_field::integrators {
             const mfem::Array<const mfem::Vector *> &elfun,
             const mfem::Array<mfem::Vector *> &elvec
         ) override;
-
         void AssembleElementGrad(
             const mfem::Array<const mfem::FiniteElement *> &el,
             mfem::ElementTransformation &Tr,
@@ -24,9 +38,9 @@ export namespace mean_field::integrators {
             const mfem::Array2D<mfem::DenseMatrix *> &elmats
         ) override;
 
-
     private:
-        const mapping::DomainMapper& m_map;
-        const mfem::GridFunction* m_phi;
+        const mapping::DomainMapper &m_map;
+        GravityForceJacobianMode m_jacobian_mode;
+        const mfem::IntegrationRule *m_integration_rule{nullptr};
     };
-}
+} // namespace mean_field::integrators