Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <array>
       7              : #include <cstddef>
       8              : #include <memory>
       9              : #include <random>
      10              : #include <string>
      11              : #include <unordered_map>
      12              : 
      13              : #include "DataStructures/DataBox/DataBox.hpp"
      14              : #include "DataStructures/DataVector.hpp"
      15              : #include "DataStructures/Tensor/Tensor.hpp"
      16              : #include "Domain/ElementMap.hpp"
      17              : #include "Domain/FunctionsOfTime/FunctionOfTime.hpp"
      18              : #include "Domain/FunctionsOfTime/Tags.hpp"
      19              : #include "Domain/Tags.hpp"
      20              : #include "Domain/TagsTimeDependent.hpp"
      21              : #include "Evolution/Initialization/Tags.hpp"
      22              : #include "Evolution/Systems/GeneralizedHarmonic/GaugeSourceFunctions/Tags/GaugeCondition.hpp"
      23              : #include "Evolution/Systems/GeneralizedHarmonic/Tags.hpp"
      24              : #include "Parallel/AlgorithmExecution.hpp"
      25              : #include "Parallel/GlobalCache.hpp"
      26              : #include "Utilities/TMPL.hpp"
      27              : 
      28              : /// \cond
      29              : namespace Tags {
      30              : struct Time;
      31              : }  // namespace Tags
      32              : /// \endcond
      33              : 
      34              : namespace gh {
      35              : namespace Tags {
      36              : /// DataBox tag for holding whether or not to set GH variables $\Pi$ and $\Phi$
      37              : /// from constraints.
      38            1 : struct SetPiAndPhiFromConstraints : db::SimpleTag {
      39            0 :   using type = bool;
      40              : 
      41            0 :   using option_tags = tmpl::list<>;
      42            0 :   static constexpr bool pass_metavariables = false;
      43              : 
      44            0 :   static bool create_from_options() { return true; }
      45              : };
      46              : }  // namespace Tags
      47              : 
      48              : namespace gauges {
      49              : /*!
      50              :  * \brief GlobalCache mutator to set the value of the
      51              :  * `gh::Tags::SetPiAndPhiFromConstraints` tag.
      52              :  */
      53            1 : struct SetPiAndPhiFromConstraintsCacheMutator {
      54            0 :   static void apply(gsl::not_null<bool*> value, bool new_value);
      55              : };
      56              : 
      57              : /*!
      58              :  * \brief Set \f$\Pi_{ab}\f$ from the gauge source function (or 1-index
      59              :  * constraint) and \f$\Phi_{iab}\f$ from the 3-index constraint.
      60              :  *
      61              :  * This is necessary to ensure the initial data is in the desired evolution
      62              :  * gauge and that the 1- and 3-index constraints are satisfied.
      63              :  */
      64              : template <class AllSolutionsForChristoffelAnalytic, size_t Dim>
      65            1 : struct SetPiAndPhiFromConstraints {
      66            0 :   using const_global_cache_tags = tmpl::list<gh::gauges::Tags::GaugeCondition>;
      67            0 :   using mutable_global_cache_tags =
      68              :       tmpl::list<gh::Tags::SetPiAndPhiFromConstraints>;
      69              : 
      70              :   template <typename DbTags, typename... InboxTags, typename Metavariables,
      71              :             typename ArrayIndex, typename ActionList,
      72              :             typename ParallelComponent>
      73            0 :   static Parallel::iterable_action_return_t apply(
      74              :       db::DataBox<DbTags>& box, tuples::TaggedTuple<InboxTags...>& /*inboxes*/,
      75              :       const Parallel::GlobalCache<Metavariables>& cache,
      76              :       const ArrayIndex& /*array_index*/, ActionList /*meta*/,
      77              :       const ParallelComponent* const /*meta*/) {
      78              :     db::mutate_apply<
      79              :         tmpl::list<gh::Tags::Pi<DataVector, Dim>,
      80              :                    gh::Tags::Phi<DataVector, Dim>>,
      81              :         tmpl::list<::Tags::Time, domain::Tags::Mesh<Dim>,
      82              :                    domain::Tags::ElementMap<Dim, Frame::Grid>,
      83              :                    domain::CoordinateMaps::Tags::CoordinateMap<Dim, Frame::Grid,
      84              :                                                                Frame::Inertial>,
      85              :                    domain::Tags::FunctionsOfTime,
      86              :                    domain::Tags::Coordinates<Dim, Frame::ElementLogical>,
      87              :                    gr::Tags::SpacetimeMetric<DataVector, Dim>,
      88              :                    gh::gauges::Tags::GaugeCondition>>(
      89              :         [](const auto&... args) { impl(args...); }, make_not_null(&box),
      90              :         Parallel::get<gh::Tags::SetPiAndPhiFromConstraints>(cache));
      91              :     return {Parallel::AlgorithmExecution::Continue, std::nullopt};
      92              :   }
      93              : 
      94            0 :   static void impl(
      95              :       gsl::not_null<tnsr::aa<DataVector, Dim, Frame::Inertial>*> pi,
      96              :       gsl::not_null<tnsr::iaa<DataVector, Dim, Frame::Inertial>*> phi,
      97              :       double time, const Mesh<Dim>& mesh,
      98              :       const ElementMap<Dim, Frame::Grid>& logical_to_grid_map,
      99              :       const domain::CoordinateMapBase<Frame::Grid, Frame::Inertial, Dim>&
     100              :           grid_to_inertial_map,
     101              :       const std::unordered_map<
     102              :           std::string,
     103              :           std::unique_ptr<domain::FunctionsOfTime::FunctionOfTime>>&
     104              :           functions_of_time,
     105              :       const tnsr::I<DataVector, Dim, Frame::ElementLogical>&
     106              :           logical_coordinates,
     107              :       const tnsr::aa<DataVector, Dim, Frame::Inertial>& spacetime_metric,
     108              :       const gauges::GaugeCondition& gauge_condition,
     109              :       bool set_pi_and_phi_from_constraints);
     110              : };
     111              : }  // namespace gauges
     112              : }  // namespace gh