precomputed_scalar_reference_finite_element.h

 
#ifndef INCGe281cd0ab7fb476e9315a3dda7f45ffe
#define INCGe281cd0ab7fb476e9315a3dda7f45ffe
 
#include <Eigen/src/Core/util/ForwardDeclarations.h>
#include <lf/quad/quad.h>
 
#include <memory>
 
#include "uniform_scalar_fe_space.h"
 
namespace lf::uscalfe {

template <class SCALAR>
class PrecomputedScalarReferenceFiniteElement
    : public lf::fe::ScalarReferenceFiniteElement<SCALAR> {
 public:
  PrecomputedScalarReferenceFiniteElement() = default;
 
  PrecomputedScalarReferenceFiniteElement(
      const PrecomputedScalarReferenceFiniteElement&) = delete;
 
  PrecomputedScalarReferenceFiniteElement(
      PrecomputedScalarReferenceFiniteElement&&) noexcept = default;
 
  PrecomputedScalarReferenceFiniteElement& operator=(
      const PrecomputedScalarReferenceFiniteElement&) = delete;
 
  PrecomputedScalarReferenceFiniteElement& operator=(
      PrecomputedScalarReferenceFiniteElement&&) noexcept = default;
  PrecomputedScalarReferenceFiniteElement(
      lf::fe::ScalarReferenceFiniteElement<SCALAR> const* fe, quad::QuadRule qr)
      : lf::fe::ScalarReferenceFiniteElement<SCALAR>(),
        fe_(fe),
        qr_(std::move(qr)),
        shap_fun_(fe_->EvalReferenceShapeFunctions(qr_.Points())),
        grad_shape_fun_(fe_->GradientsReferenceShapeFunctions(qr_.Points())) {}

  [[nodiscard]] bool isInitialized() const { return (fe_ != nullptr); }
 
  [[nodiscard]] base::RefEl RefEl() const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->RefEl();
  }
 
  [[nodiscard]] unsigned Degree() const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->Degree();
  }
 
  [[nodiscard]] size_type NumRefShapeFunctions() const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->NumRefShapeFunctions();
  }
 
  [[nodiscard]] size_type NumRefShapeFunctions(dim_t codim) const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->NumRefShapeFunctions(codim);
  }
 
  [[nodiscard]] size_type NumRefShapeFunctions(
      dim_t codim, sub_idx_t subidx) const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->NumRefShapeFunctions(codim, subidx);
  }
 
  [[nodiscard]] Eigen::Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic>
  EvalReferenceShapeFunctions(const Eigen::MatrixXd& local) const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->EvalReferenceShapeFunctions(local);
  }
  [[nodiscard]] Eigen::Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic>
  GradientsReferenceShapeFunctions(
      const Eigen::MatrixXd& local) const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->GradientsReferenceShapeFunctions(local);
  }
  [[nodiscard]] Eigen::MatrixXd EvaluationNodes() const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->EvaluationNodes();
  }
  [[nodiscard]] size_type NumEvaluationNodes() const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->NumEvaluationNodes();
  }
 
  [[nodiscard]] Eigen::Matrix<SCALAR, 1, Eigen::Dynamic> NodalValuesToDofs(
      const Eigen::Matrix<SCALAR, 1, Eigen::Dynamic>& nodvals) const override {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return fe_->NodalValuesToDofs(nodvals);
  }

  [[nodiscard]] const quad::QuadRule& Qr() const {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return qr_;
  }

  [[nodiscard]] const Eigen::Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic>&
  PrecompReferenceShapeFunctions() const {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return shap_fun_;
  }
 
  // clang-format off
  // clang-format on
  [[nodiscard]] const Eigen::Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic>&
  PrecompGradientsReferenceShapeFunctions() const {
    LF_ASSERT_MSG(fe_ != nullptr, "Not initialized.");
    return grad_shape_fun_;
  }

  ~PrecomputedScalarReferenceFiniteElement() override = default;
 
 private:
  fe::ScalarReferenceFiniteElement<SCALAR> const* fe_ = nullptr;
  quad::QuadRule qr_;
  Eigen::Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic> shap_fun_;
  Eigen::Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic> grad_shape_fun_;
};
 
}  // namespace lf::uscalfe
 
#endif  // INCGe281cd0ab7fb476e9315a3dda7f45ffe