add generic evaluator

lanl · Oct 9, 2023 · 73a6561 · 73a6561
1 parent 790cba7
commit 73a6561
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/singularity-eos/eos/eos_base.hpp b/singularity-eos/eos/eos_base.hpp
@@ -128,6 +128,14 @@ class EosBase {
   struct is_raw_pointer
       : std::is_same<std::remove_reference_t<std::remove_cv_t<T>>, R *> {};
 
+  // Generic evaluator
+  template <typename Functor_t>
+  PORTABLE_INLINE_FUNCTION
+  void Evaluate(const Functor_t &f) const {
+    CRTP copy = *(static_cast<CRTP const *>(this));
+    f(copy);
+  }
+
   // Vector member functions
   template <typename RealIndexer, typename ConstRealIndexer, typename LambdaIndexer>
   inline void

diff --git a/singularity-eos/eos/eos_variant.hpp b/singularity-eos/eos/eos_variant.hpp
@@ -72,6 +72,16 @@ class Variant {
   }
 
   // Place member functions here
+  template <typename Functor_t>
+  PORTABLE_INLINE_FUNCTION
+  void Evaluate(const Functor_t &f) const {
+    return mpark::visit(
+      [&f](const auto &eos) {
+        return eos.Evaluate(f);
+      },
+      eos_);
+  }
+
   PORTABLE_INLINE_FUNCTION
   Real TemperatureFromDensityInternalEnergy(const Real rho, const Real sie,
                                             Real *lambda = nullptr) const {