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 <memory>
       8              : 
       9              : #include "Parallel/Algorithms/AlgorithmNodegroupDeclarations.hpp"
      10              : #include "Parallel/ArrayCollection/CreateElementCollection.hpp"
      11              : #include "Parallel/ArrayCollection/DgElementArrayMember.hpp"
      12              : #include "Parallel/ArrayCollection/DgElementArrayMemberBase.hpp"
      13              : #include "Parallel/ArrayCollection/Tags/ElementCollection.hpp"
      14              : #include "Parallel/ArrayCollection/TransformPdalForNodegroup.hpp"
      15              : #include "Parallel/GlobalCache.hpp"
      16              : #include "Parallel/Local.hpp"
      17              : #include "Parallel/ParallelComponentHelpers.hpp"
      18              : #include "Parallel/Printf/Printf.hpp"
      19              : #include "ParallelAlgorithms/Actions/TerminatePhase.hpp"
      20              : #include "Utilities/Gsl.hpp"
      21              : #include "Utilities/TMPL.hpp"
      22              : 
      23              : namespace Parallel {
      24              : /*!
      25              :  * \brief A collection of DG elements on a node.
      26              :  *
      27              :  * The `PhaseDepActionList` is the PDAL that was used for the array
      28              :  * approach. Some actions will require updating to support nodegroups if they
      29              :  * haven't already been.
      30              :  */
      31              : template <size_t Dim, class Metavariables, class PhaseDepActionList>
      32            1 : struct DgElementCollection {
      33              :   /// \brief The `DgElementCollection` is currently a nodegroup.
      34              :   ///
      35              :   /// It should be possible to generalize this to work as a group too.
      36            1 :   using chare_type = Parallel::Algorithms::Nodegroup;
      37            0 :   static constexpr bool checkpoint_data = true;
      38              :   /// \brief The metavariables
      39            1 :   using metavariables = Metavariables;
      40              :   /// \brief The simple tags necessary from option parsing.
      41            1 :   using simple_tags_from_options = Parallel::get_simple_tags_from_options<
      42              :       Parallel::get_initialization_actions_list<PhaseDepActionList>>;
      43              :   /// \brief The type of the `Parallel::DgElementArrayMember` with all
      44              :   /// template parameters.
      45            1 :   using dg_element_array_member =
      46              :       Parallel::DgElementArrayMember<Dim, Metavariables, PhaseDepActionList,
      47              :                                      simple_tags_from_options>;
      48              :   /// \brief The tag `Parallel::Tags::ElementCollection` of this
      49              :   /// `DgElementCollection`
      50            1 :   using element_collection_tag =
      51              :       Parallel::Tags::ElementCollection<Dim, Metavariables, PhaseDepActionList,
      52              :                                         simple_tags_from_options>;
      53              : 
      54              :   /// \brief The phase dependent action lists.
      55              :   ///
      56              :   /// These are computed using
      57              :   /// `Parallel::TransformPhaseDependentActionListForNodegroup` from the
      58              :   /// `PhaseDepActionList` template parameter.
      59            1 :   using phase_dependent_action_list = tmpl::append<
      60              :       tmpl::list<Parallel::PhaseActions<
      61              :           Parallel::Phase::Initialization,
      62              :           tmpl::list<Actions::CreateElementCollection<Dim, Metavariables,
      63              :                                                       PhaseDepActionList,
      64              :                                                       simple_tags_from_options>,
      65              :                      Parallel::Actions::TerminatePhase>>>,
      66              :       TransformPhaseDependentActionListForNodegroup<PhaseDepActionList>>;
      67              : 
      68              :   /// @{
      69              :   /// \brief The tags for the global cache.
      70              :   ///
      71              :   /// These are computed from the `PhaseDepActionList` template parameter.
      72            1 :   using const_global_cache_tags = tmpl::remove_duplicates<
      73              :       typename Parallel::detail::get_const_global_cache_tags_from_pdal<
      74              :           PhaseDepActionList>::type>;
      75            1 :   using mutable_global_cache_tags =
      76              :       typename Parallel::detail::get_mutable_global_cache_tags_from_pdal<
      77              :           PhaseDepActionList>::type;
      78              :   /// @}
      79              : 
      80              :   /// \brief Starts the next phase of execution.
      81            1 :   static void execute_next_phase(
      82              :       const Parallel::Phase next_phase,
      83              :       Parallel::CProxy_GlobalCache<Metavariables>& global_cache) {
      84              :     Parallel::printf("%s\n", next_phase);
      85              :     auto& local_cache = *Parallel::local_branch(global_cache);
      86              :     Parallel::get_parallel_component<DgElementCollection>(local_cache)
      87              :         .start_phase(next_phase);
      88              :   }
      89              : };
      90              : }  // namespace Parallel