feat(newton): first newton solver implementation
This commit is contained in:
@@ -1,7 +1,9 @@
|
||||
module;
|
||||
#include <cstdint>
|
||||
#include <expected>
|
||||
#include <memory>
|
||||
#include <mfem.hpp>
|
||||
#include <stdexcept>
|
||||
#include <vector>
|
||||
|
||||
export module mean_field:operators.prepared_gravity_source;
|
||||
@@ -10,6 +12,23 @@ export import :field.mfem;
|
||||
export import :mapping.domain_mapper;
|
||||
|
||||
export namespace mean_field::operators {
|
||||
enum class GravitySourcePreparationRejectionReason : std::uint8_t { invalid_mapping, non_finite_arithmetic };
|
||||
|
||||
struct GravitySourcePreparationRejection final {
|
||||
GravitySourcePreparationRejectionReason reason{GravitySourcePreparationRejectionReason::invalid_mapping};
|
||||
mapping::MappingStatus mappingStatus{mapping::MappingStatus::valid};
|
||||
};
|
||||
|
||||
using GravitySourcePreparationResult = std::expected<void, GravitySourcePreparationRejection>;
|
||||
|
||||
[[noreturn]] inline void
|
||||
throwGravitySourcePreparationRejection(const GravitySourcePreparationRejection &rejection) {
|
||||
if (rejection.reason == GravitySourcePreparationRejectionReason::non_finite_arithmetic) {
|
||||
throw std::domain_error("Prepared gravity-source data contained non-finite arithmetic.");
|
||||
}
|
||||
throw std::domain_error("Prepared gravity-source data could not map the candidate geometry.");
|
||||
}
|
||||
|
||||
class PreparedMappedGravitySourceOperator final : public mfem::Operator {
|
||||
public:
|
||||
PreparedMappedGravitySourceOperator(
|
||||
@@ -19,6 +38,8 @@ export namespace mean_field::operators {
|
||||
|
||||
void Prepare(const mfem::Vector &displacement);
|
||||
void PreparePrimal(const mfem::Vector &displacement);
|
||||
[[nodiscard]] GravitySourcePreparationResult TryPrepare(const mfem::Vector &displacement);
|
||||
[[nodiscard]] GravitySourcePreparationResult TryPreparePrimal(const mfem::Vector &displacement);
|
||||
void Mult(
|
||||
const mfem::Vector &density,
|
||||
mfem::Vector &action
|
||||
@@ -68,7 +89,7 @@ export namespace mean_field::operators {
|
||||
mfem::Vector quadrature_data;
|
||||
};
|
||||
|
||||
void PrepareImpl(
|
||||
[[nodiscard]] GravitySourcePreparationResult TryPrepareImpl(
|
||||
const mfem::Vector &displacement,
|
||||
PreparationMode mode
|
||||
);
|
||||
|
||||
Reference in New Issue
Block a user