Line data    Source code

       1           0 : // Distributed under the MIT License.
       2             : // See LICENSE.txt for details.
       3             : 
       4             : #pragma once
       5             : 
       6             : #include <cstddef>
       7             : #include <mutex>
       8             : #include <string>
       9             : 
      10             : #include "DataStructures/DataVector.hpp"
      11             : #include "DataStructures/Tensor/Tensor.hpp"
      12             : #include "PointwiseFunctions/GeneralRelativity/Tags.hpp"
      13             : #include "PointwiseFunctions/Hydro/Tags.hpp"
      14             : #include "Utilities/TMPL.hpp"
      15             : #include "Utilities/TaggedTuple.hpp"
      16             : 
      17             : namespace io {
      18             : 
      19             : /// Type of FUKA initial data
      20           0 : enum class FukaIdType { Bh, Bbh, Ns, Bns, Bhns };
      21             : 
      22             : namespace detail {
      23             : using fuka_gr_tags =
      24             :     tmpl::list<gr::Tags::Lapse<DataVector>, gr::Tags::Shift<DataVector, 3>,
      25             :                gr::Tags::SpatialMetric<DataVector, 3>,
      26             :                gr::Tags::ExtrinsicCurvature<DataVector, 3>>;
      27             : using fuka_hydro_tags =
      28             :     tmpl::list<hydro::Tags::RestMassDensity<DataVector>,
      29             :                hydro::Tags::SpecificInternalEnergy<DataVector>,
      30             :                hydro::Tags::Pressure<DataVector>,
      31             :                hydro::Tags::SpatialVelocity<DataVector, 3>>;
      32             : template <FukaIdType IdType>
      33             : struct FukaTags {
      34             :   using type =
      35             :       tmpl::conditional_t<IdType <= FukaIdType::Bh or IdType == FukaIdType::Bbh,
      36             :                           fuka_gr_tags,
      37             :                           tmpl::append<fuka_gr_tags, fuka_hydro_tags>>;
      38             : };
      39             : }  // namespace detail
      40             : 
      41             : /// List of tags supplied by FUKA initial data
      42             : template <FukaIdType IdType>
      43           1 : using fuka_tags = typename detail::FukaTags<IdType>::type;
      44             : 
      45             : /*!
      46             :  * \brief Interpolate numerical FUKA initial data to arbitrary points
      47             :  *
      48             :  * \tparam IdType Type of FUKA initial data
      49             :  * \param fuka_lock Lock for accessing FUKA data. This is needed because
      50             :  * FUKA is not thread-safe. Pass in a lock that is shared with other
      51             :  * threads that are also calling this function.
      52             :  * \param info_filename Path to the FUKA info file to load
      53             :  * \param x Coordinates of points to interpolate to
      54             :  * \param interpolation_offset See FUKA documentation for export functions
      55             :  * \param interp_order See FUKA documentation for export functions
      56             :  * \param delta_r_rel See FUKA documentation for export functions
      57             :  * \return Data interpolated to the given points
      58             :  */
      59             : template <FukaIdType IdType>
      60           1 : tuples::tagged_tuple_from_typelist<fuka_tags<IdType>> interpolate_from_fuka(
      61             :     gsl::not_null<std::mutex*> fuka_lock, const std::string& info_filename,
      62             :     const tnsr::I<DataVector, 3, Frame::Inertial>& x,
      63             :     double interpolation_offset = 0., int interp_order = 8,
      64             :     double delta_r_rel = 0.3);
      65             : 
      66             : }  // namespace io