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              : #include <optional>
       9              : 
      10              : #include "DataStructures/DataBox/Tag.hpp"
      11              : #include "Domain/Domain.hpp"
      12              : #include "Domain/ElementDistribution.hpp"
      13              : #include "Options/Auto.hpp"
      14              : #include "Options/String.hpp"
      15              : #include "Parallel/Tags/Parallelization.hpp"
      16              : #include "Utilities/TMPL.hpp"
      17              : 
      18              : namespace domain {
      19              : namespace OptionTags {
      20              : /// \ingroup OptionTagsGroup
      21              : /// \ingroup ComputationalDomainGroup
      22            0 : struct ElementDistribution {
      23            0 :   struct RoundRobin {};
      24            0 :   using type = Options::Auto<ElementWeight, RoundRobin>;
      25            0 :   static constexpr Options::String help = {
      26              :       "Weighting pattern to use for ZCurve element distribution. Specify "
      27              :       "RoundRobin to just place each element on the next core."};
      28            0 :   using group = Parallel::OptionTags::Parallelization;
      29              : };
      30              : }  // namespace OptionTags
      31              : 
      32              : namespace Tags {
      33              : /// \ingroup DataBoxTagsGroup
      34              : /// \ingroup ComputationalDomainGroup
      35              : /// Tag that holds method for how to distribute the elements on the given
      36              : /// resources.
      37            1 : struct ElementDistribution : db::SimpleTag {
      38            0 :   using type = std::optional<ElementWeight>;
      39            0 :   using option_tags = tmpl::list<OptionTags::ElementDistribution>;
      40              : 
      41            0 :   static constexpr bool pass_metavariables = false;
      42            0 :   static type create_from_options(const type& element_distribution) {
      43              :     return element_distribution;
      44              :   }
      45              : };
      46              : }  // namespace Tags
      47              : }  // namespace domain