Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <type_traits>
       7              : #include <utility>
       8              : 
       9              : #include "DataStructures/DataBox/MetavariablesTag.hpp"
      10              : #include "DataStructures/DataBox/Tag.hpp"
      11              : #include "DataStructures/Tensor/Tensor.hpp"
      12              : #include "DataStructures/Variables.hpp"
      13              : #include "DataStructures/VariablesTag.hpp"
      14              : #include "Domain/Tags.hpp"
      15              : #include "Elliptic/Tags.hpp"
      16              : #include "PointwiseFunctions/GeneralRelativity/Tags.hpp"
      17              : #include "PointwiseFunctions/Hydro/Tags.hpp"
      18              : #include "PointwiseFunctions/InitialDataUtilities/Background.hpp"
      19              : #include "Utilities/CallWithDynamicType.hpp"
      20              : #include "Utilities/Gsl.hpp"
      21              : #include "Utilities/TMPL.hpp"
      22              : 
      23            1 : namespace Xcts::Tags {
      24              : namespace detail {
      25              : 
      26              : template <typename Background, typename HydroTags, typename = std::void_t<>>
      27              : struct has_hydro_variables : std::false_type {};
      28              : 
      29              : template <typename Background, typename HydroTags>
      30              : struct has_hydro_variables<
      31              :     Background, HydroTags,
      32              :     std::void_t<decltype(std::declval<const Background&>().variables(
      33              :         std::declval<const tnsr::I<DataVector, 3, Frame::Inertial>&>(),
      34              :         HydroTags{}))>> : std::true_type {};
      35              : 
      36              : }  // namespace detail
      37              : 
      38              : /*!
      39              :  * \brief MHD quantities retrieved from the background solution/data
      40              :  *
      41              :  * Retrieve the `HydroTags` from the background solution/data so they can be
      42              :  * written out to disk. These quantities don't take part in a pure XCTS solve,
      43              :  * (which only solves for the gravity sector given the matter profile) except
      44              :  * for their contributions to the matter source terms in the XCTS equations.
      45              :  * However, the matter source terms are computed and stored separately, so these
      46              :  * hydro quantities are only used for observations.
      47              :  */
      48              : template <typename HydroTags>
      49            1 : struct HydroQuantitiesCompute : ::Tags::Variables<HydroTags>, db::ComputeTag {
      50            0 :   using base = ::Tags::Variables<HydroTags>;
      51            0 :   using argument_tags = tmpl::list<
      52              :       domain::Tags::Coordinates<3, Frame::Inertial>,
      53              :       elliptic::Tags::Background<elliptic::analytic_data::Background>,
      54              :       Parallel::Tags::Metavariables>;
      55              :   template <typename Metavariables>
      56            0 :   static void function(const gsl::not_null<Variables<HydroTags>*> result,
      57              :                        const tnsr::I<DataVector, 3>& inertial_coords,
      58              :                        const elliptic::analytic_data::Background& background,
      59              :                        const Metavariables& /*meta*/) {
      60              :     using background_classes =
      61              :         tmpl::at<typename Metavariables::factory_creation::factory_classes,
      62              :                  elliptic::analytic_data::Background>;
      63              :     *result = call_with_dynamic_type<Variables<HydroTags>, background_classes>(
      64              :         &background, [&inertial_coords](const auto* const derived) {
      65              :           using derived_type = std::decay_t<decltype(*derived)>;
      66              :           if constexpr (detail::has_hydro_variables<derived_type,
      67              :                                                     HydroTags>::value) {
      68              :             return variables_from_tagged_tuple(
      69              :                 derived->variables(inertial_coords, HydroTags{}));
      70              :           } else {
      71              :             return Variables<HydroTags>{inertial_coords.begin()->size(), 0.};
      72              :           }
      73              :         });
      74              :   }
      75              : };
      76              : 
      77              : }  // namespace Xcts::Tags