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 <optional>
       8              : #include <utility>
       9              : 
      10              : #include "DataStructures/DataBox/DataBox.hpp"
      11              : #include "DataStructures/DataBox/PrefixHelpers.hpp"
      12              : #include "DataStructures/DataBox/Prefixes.hpp"
      13              : #include "DataStructures/Tensor/Tensor.hpp"
      14              : #include "DataStructures/Variables.hpp"
      15              : #include "Domain/Tags.hpp"
      16              : #include "Elliptic/Utilities/GetAnalyticData.hpp"
      17              : #include "NumericalAlgorithms/Spectral/Mesh.hpp"
      18              : #include "Parallel/AlgorithmExecution.hpp"
      19              : #include "Parallel/GlobalCache.hpp"
      20              : #include "ParallelAlgorithms/Initialization/MutateAssign.hpp"
      21              : #include "ParallelAlgorithms/LinearSolver/Tags.hpp"
      22              : #include "Utilities/TMPL.hpp"
      23              : 
      24              : namespace elliptic::Actions {
      25              : 
      26              : /*!
      27              :  * \brief Initialize the dynamic fields of the elliptic system, i.e. those we
      28              :  * solve for.
      29              :  *
      30              :  * Uses:
      31              :  * - System:
      32              :  *   - `primal_fields`
      33              :  * - DataBox:
      34              :  *   - `InitialGuessTag`
      35              :  *   - `Tags::Coordinates<Dim, Frame::Inertial>`
      36              :  *
      37              :  * DataBox:
      38              :  * - Adds:
      39              :  *   - `primal_fields`
      40              :  */
      41              : template <typename System, typename InitialGuessTag>
      42            1 : struct InitializeFields {
      43              :  private:
      44            0 :   using fields_tag = ::Tags::Variables<typename System::primal_fields>;
      45              : 
      46              :  public:
      47            0 :   using simple_tags = tmpl::list<fields_tag>;
      48            0 :   using compute_tags = tmpl::list<>;
      49            0 :   using const_global_cache_tags = tmpl::list<InitialGuessTag>;
      50              : 
      51              :   template <typename DbTagsList, typename... InboxTags, typename Metavariables,
      52              :             size_t Dim, typename ActionList, typename ParallelComponent>
      53            0 :   static Parallel::iterable_action_return_t apply(
      54              :       db::DataBox<DbTagsList>& box,
      55              :       const tuples::TaggedTuple<InboxTags...>& /*inboxes*/,
      56              :       const Parallel::GlobalCache<Metavariables>& /*cache*/,
      57              :       const ElementId<Dim>& /*array_index*/, const ActionList /*meta*/,
      58              :       const ParallelComponent* const /*meta*/) {
      59              :     const auto& inertial_coords =
      60              :         get<domain::Tags::Coordinates<Dim, Frame::Inertial>>(box);
      61              :     const auto& initial_guess = db::get<InitialGuessTag>(box);
      62              :     auto initial_fields =
      63              :         elliptic::util::get_analytic_data<typename fields_tag::tags_list>(
      64              :             initial_guess, box, inertial_coords);
      65              :     ::Initialization::mutate_assign<simple_tags>(make_not_null(&box),
      66              :                                                  std::move(initial_fields));
      67              :     return {Parallel::AlgorithmExecution::Continue, std::nullopt};
      68              :   }
      69              : };
      70              : 
      71              : }  // namespace elliptic::Actions