save3

jyoo1042 · Dec 8, 2023 · e65abc0 · e65abc0
1 parent 18c6864
commit e65abc0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/PointwiseFunctions/AnalyticData/GrMhd/MagnetizedFmDisk.hpp b/src/PointwiseFunctions/AnalyticData/GrMhd/MagnetizedFmDisk.hpp
@@ -172,6 +172,20 @@ class MagnetizedFmDisk : public virtual evolution::initial_data::InitialData,
       }
     }()))...};
   }
+
+  template <typename DataType, typename Tag>
+  tuples::TaggedTuple<Tag> variables(const tnsr::I<DataType, 3>& x,
+                                     tmpl::list<Tag> /*meta*/) const {
+    // Can't store IntermediateVariables as member variable because we need to
+    // be threadsafe.
+    typename FmDisk::IntermediateVariables<DataType> vars(x);
+    if constexpr (std::is_same_v<hydro::Tags::MagneticField<DataType, 3>,
+                                 Tag>) {
+      return variables(x, tmpl::list<Tag>{}, vars);
+    } else {
+      return fm_disk_.variables(x, tmpl::list<Tag>{}, make_not_null(&vars));
+    }
+  }
   /// @}
 
   // NOLINTNEXTLINE(google-runtime-references)