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/DgSubcell/ActiveGrid.hpp"
      22              : #include "Evolution/DgSubcell/Tags/ActiveGrid.hpp"
      23              : #include "Evolution/DgSubcell/Tags/Coordinates.hpp"
      24              : #include "Evolution/DgSubcell/Tags/Mesh.hpp"
      25              : #include "Evolution/Initialization/Tags.hpp"
      26              : #include "Evolution/Systems/GeneralizedHarmonic/GaugeSourceFunctions/Gauges.hpp"
      27              : #include "Evolution/Systems/GeneralizedHarmonic/GaugeSourceFunctions/SetPiAndPhiFromConstraints.hpp"
      28              : #include "Evolution/Systems/GeneralizedHarmonic/GaugeSourceFunctions/Tags/GaugeCondition.hpp"
      29              : #include "Evolution/Systems/GeneralizedHarmonic/Tags.hpp"
      30              : #include "Evolution/Systems/GrMhd/GhValenciaDivClean/AllSolutions.hpp"
      31              : #include "Parallel/AlgorithmExecution.hpp"
      32              : #include "Utilities/TMPL.hpp"
      33              : 
      34              : /// \cond
      35              : namespace Tags {
      36              : struct Time;
      37              : }  // namespace Tags
      38              : /// \endcond
      39              : 
      40              : namespace grmhd::GhValenciaDivClean {
      41              : /*!
      42              :  * \brief Set \f$\Pi_{ab}\f$ from the gauge source function.
      43              :  *
      44              :  * This is necessary to ensure the initial data is in the desired evolution
      45              :  * gauge.
      46              :  */
      47            1 : struct SetPiAndPhiFromConstraints {
      48            0 :   using const_global_cache_tags =
      49              :       typename gh::gauges::SetPiAndPhiFromConstraints<
      50              :           ghmhd::GhValenciaDivClean::InitialData::
      51              :               analytic_solutions_and_data_list,
      52              :           3>::const_global_cache_tags;
      53            0 :   using mutable_global_cache_tags =
      54              :       typename gh::gauges::SetPiAndPhiFromConstraints<
      55              :           ghmhd::GhValenciaDivClean::InitialData::
      56              :               analytic_solutions_and_data_list,
      57              :           3>::mutable_global_cache_tags;
      58              : 
      59              :   template <typename DbTags, typename... InboxTags, typename Metavariables,
      60              :             typename ArrayIndex, typename ActionList,
      61              :             typename ParallelComponent>
      62            0 :   static Parallel::iterable_action_return_t apply(
      63              :       db::DataBox<DbTags>& box, tuples::TaggedTuple<InboxTags...>& /*inboxes*/,
      64              :       const Parallel::GlobalCache<Metavariables>& cache,
      65              :       const ArrayIndex& /*array_index*/, ActionList /*meta*/,
      66              :       const ParallelComponent* const /*meta*/) {
      67              :     db::mutate_apply<
      68              :         tmpl::list<gh::Tags::Pi<DataVector, 3>, gh::Tags::Phi<DataVector, 3>>,
      69              :         tmpl::list<
      70              :             ::Tags::Time, domain::Tags::Mesh<3>,
      71              :             domain::Tags::ElementMap<3, Frame::Grid>,
      72              :             domain::CoordinateMaps::Tags::CoordinateMap<3, Frame::Grid,
      73              :                                                         Frame::Inertial>,
      74              :             domain::Tags::FunctionsOfTime,
      75              :             domain::Tags::Coordinates<3, Frame::ElementLogical>,
      76              :             gr::Tags::SpacetimeMetric<DataVector, 3>,
      77              :             gh::gauges::Tags::GaugeCondition,
      78              :             evolution::dg::subcell::Tags::Mesh<3>,
      79              :             evolution::dg::subcell::Tags::Coordinates<3, Frame::ElementLogical>,
      80              :             evolution::dg::subcell::Tags::ActiveGrid>>(
      81              :         [](const gsl::not_null<tnsr::aa<DataVector, 3, Frame::Inertial>*> pi,
      82              :            const gsl::not_null<tnsr::iaa<DataVector, 3, Frame::Inertial>*> phi,
      83              :            const double initial_time, const Mesh<3>& dg_mesh,
      84              :            const ElementMap<3, Frame::Grid>& logical_to_grid_map,
      85              :            const domain::CoordinateMapBase<Frame::Grid, Frame::Inertial, 3>&
      86              :                grid_to_inertial_map,
      87              :            const std::unordered_map<
      88              :                std::string,
      89              :                std::unique_ptr<domain::FunctionsOfTime::FunctionOfTime>>&
      90              :                functions_of_time,
      91              :            const tnsr::I<DataVector, 3, Frame::ElementLogical>&
      92              :                dg_logical_coordinates,
      93              :            const tnsr::aa<DataVector, 3, Frame::Inertial>& spacetime_metric,
      94              :            const gh::gauges::GaugeCondition& gauge_condition,
      95              :            const Mesh<3>& subcell_mesh,
      96              :            const tnsr::I<DataVector, 3, Frame::ElementLogical>&
      97              :                subcell_logical_coordinates,
      98              :            const evolution::dg::subcell::ActiveGrid active_grid,
      99              :            const bool set_pi_and_phi_from_constraints) {
     100              :           if (active_grid == evolution::dg::subcell::ActiveGrid::Dg) {
     101              :             gh::gauges::SetPiAndPhiFromConstraints<
     102              :                 ghmhd::GhValenciaDivClean::InitialData::
     103              :                     analytic_solutions_and_data_list,
     104              :                 3>::impl(pi, phi, initial_time, dg_mesh, logical_to_grid_map,
     105              :                          grid_to_inertial_map, functions_of_time,
     106              :                          dg_logical_coordinates, spacetime_metric,
     107              :                          gauge_condition, set_pi_and_phi_from_constraints);
     108              :           } else {
     109              :             gh::gauges::SetPiAndPhiFromConstraints<
     110              :                 ghmhd::GhValenciaDivClean::InitialData::
     111              :                     analytic_solutions_and_data_list,
     112              :                 3>::impl(pi, phi, initial_time, subcell_mesh,
     113              :                          logical_to_grid_map, grid_to_inertial_map,
     114              :                          functions_of_time, subcell_logical_coordinates,
     115              :                          spacetime_metric, gauge_condition,
     116              :                          set_pi_and_phi_from_constraints);
     117              :           }
     118              :         },
     119              :         make_not_null(&box),
     120              :         Parallel::get<gh::Tags::SetPiAndPhiFromConstraints>(cache));
     121              :     return {Parallel::AlgorithmExecution::Continue, std::nullopt};
     122              :   }
     123              : };
     124              : }  // namespace grmhd::GhValenciaDivClean