Emily Boudreaux
177ae8b38a
mass normaliztion, gravity, displacement, and hydrostatic equilibrium are now migrated
116 lines
4.3 KiB
C++
116 lines
4.3 KiB
C++
module;
|
|
#include <memory>
|
|
#include <mfem.hpp>
|
|
|
|
export module mean_field:operators.gravity_field;
|
|
export import :fem;
|
|
export import :mapping.domain_mapper;
|
|
export import :operators.gravity_field_jacobian;
|
|
export import :operators.context.gravity_field;
|
|
|
|
export namespace mean_field::operators {
|
|
class GravityFieldOperator final : public mfem::Operator {
|
|
public:
|
|
GravityFieldOperator(
|
|
fem::FEM &f,
|
|
const mapping::DomainMapperStateless &domain_mapper,
|
|
context::gravity_field::GravityFieldLinearizationContext &linearization_context,
|
|
const mfem::Array<int> &state_offsets,
|
|
GravityFieldJacobianOperator &jacobian
|
|
);
|
|
|
|
context::gravity_field::GravityFieldPreparationReport Prepare(
|
|
const mfem::Vector &state,
|
|
const context::gravity_field::GravityFieldRevisions &revisions
|
|
);
|
|
|
|
void Mult(
|
|
const mfem::Vector &state,
|
|
mfem::Vector &residual
|
|
) const override;
|
|
|
|
Operator &GetGradient(const mfem::Vector &state) const override;
|
|
|
|
[[nodiscard]] const mfem::Array<int> &GetStateOffsets() const noexcept;
|
|
|
|
[[nodiscard]] const mfem::Array<int> &GetResidualOffsets() const noexcept;
|
|
|
|
[[nodiscard]] context::gravity_field::GravityFieldLinearizationContext &GetLinearizationContext() noexcept;
|
|
|
|
[[nodiscard]] const context::gravity_field::GravityFieldLinearizationContext &
|
|
GetLinearizationContext() const noexcept;
|
|
|
|
void ApplyGravityUnknowns(
|
|
const mfem::Vector &gravity_gradient,
|
|
const mfem::Vector &gravity_potential,
|
|
const context::gravity_field::GravityFieldGeometryContext &geometry_context,
|
|
mfem::Vector &action
|
|
) const;
|
|
|
|
void ApplyDensitySource(
|
|
const mfem::Vector &density,
|
|
const context::gravity_field::GravityFieldGeometryContext &geometry_context,
|
|
mfem::Vector &action
|
|
) const;
|
|
|
|
private:
|
|
fem::FEM &m_fem;
|
|
const mapping::DomainMapperStateless &m_domain_mapper;
|
|
context::gravity_field::GravityFieldLinearizationContext &m_linearization_context;
|
|
mfem::Array<int> m_state_offsets;
|
|
mfem::Array<int> m_residual_offsets;
|
|
GravityFieldJacobianOperator &m_jacobian;
|
|
};
|
|
|
|
class ReducedGravityFieldOperator final : public mfem::Operator {
|
|
public:
|
|
ReducedGravityFieldOperator(
|
|
GravityFieldOperator &gravity_field_operator,
|
|
context::gravity_field::GravityFieldGeometryContext &gravity_field_geometry_context,
|
|
const mfem::Vector &displacement
|
|
);
|
|
|
|
ReducedGravityFieldOperator(const ReducedGravityFieldOperator &) = delete;
|
|
ReducedGravityFieldOperator &operator=(const ReducedGravityFieldOperator &) = delete;
|
|
ReducedGravityFieldOperator(ReducedGravityFieldOperator &&) = delete;
|
|
ReducedGravityFieldOperator &operator=(ReducedGravityFieldOperator &&) = delete;
|
|
|
|
void SetDisplacement(const mfem::Vector &displacement);
|
|
|
|
[[nodiscard]] const mfem::Vector &GetDisplacement() const;
|
|
|
|
void BuildRightHandSide(
|
|
const mfem::Vector &density,
|
|
mfem::Vector &right_hand_side
|
|
) const;
|
|
|
|
void Mult(
|
|
const mfem::Vector &gravity_state,
|
|
mfem::Vector &action
|
|
) const override;
|
|
|
|
[[nodiscard]] GravityFieldOperator &GetGravityFieldOperator() noexcept;
|
|
|
|
[[nodiscard]] const GravityFieldOperator &GetGravityFieldOperator() const noexcept;
|
|
|
|
[[nodiscard]] context::gravity_field::GravityFieldGeometryContext &GetGeometryContext() noexcept;
|
|
|
|
[[nodiscard]] const context::gravity_field::GravityFieldGeometryContext &GetGeometryContext() const noexcept;
|
|
|
|
[[nodiscard]] const mfem::Array<int> &GetGravityOffsets() const noexcept;
|
|
|
|
private:
|
|
void ValidateDisplacement(const mfem::Vector &displacement) const;
|
|
|
|
void ValidateDensity(const mfem::Vector &density) const;
|
|
|
|
void ValidateGravityState(const mfem::Vector &gravity_state) const;
|
|
|
|
private:
|
|
GravityFieldOperator &m_gravity_field_operator;
|
|
mfem::Array<int> m_gravity_offsets;
|
|
context::gravity_field::GravityFieldGeometryContext &m_gravity_field_geometry_context;
|
|
mfem::Vector m_displacement;
|
|
};
|
|
} // namespace mean_field::operators
|