Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <cstdint>
       7              : #include <memory>
       8              : #include <pup.h>
       9              : #include <pup_stl.h>
      10              : #include <utility>
      11              : 
      12              : #include "Options/ParseOptions.hpp"
      13              : #include "Options/String.hpp"
      14              : #include "ParallelAlgorithms/EventsAndTriggers/Trigger.hpp"
      15              : #include "Time/TimeSequence.hpp"
      16              : #include "Time/TimeStepId.hpp"
      17              : #include "Utilities/Serialization/CharmPupable.hpp"
      18              : #include "Utilities/TMPL.hpp"
      19              : 
      20              : /// \cond
      21              : namespace Tags {
      22              : struct TimeStepId;
      23              : }  // namespace Tags
      24              : /// \endcond
      25              : 
      26              : namespace Triggers {
      27              : /// \ingroup EventsAndTriggersGroup
      28              : /// \ingroup TimeGroup
      29              : /// Trigger at specified numbers of slabs after the simulation start.
      30            1 : class Slabs : public Trigger {
      31              :  public:
      32              :   /// \cond
      33              :   Slabs() = default;
      34              :   explicit Slabs(CkMigrateMessage* /*unused*/) {}
      35              :   using PUP::able::register_constructor;
      36              :   WRAPPED_PUPable_decl_template(Slabs);  // NOLINT
      37              :   /// \endcond
      38              : 
      39            0 :   static constexpr Options::String help{
      40              :     "Trigger at specified numbers of slabs after the simulation start."};
      41              : 
      42            0 :   explicit Slabs(std::unique_ptr<TimeSequence<uint64_t>> slabs)
      43              :       : slabs_(std::move(slabs)) {}
      44              : 
      45            0 :   using argument_tags = tmpl::list<Tags::TimeStepId>;
      46              : 
      47            0 :   bool operator()(const TimeStepId& time_id) const {
      48              :     const auto unsigned_slab =
      49              :         static_cast<std::uint64_t>(time_id.slab_number());
      50              :     return slabs_->times_near(unsigned_slab)[1] == unsigned_slab;
      51              :   }
      52              : 
      53              :   // NOLINTNEXTLINE(google-runtime-references)
      54            0 :   void pup(PUP::er& p) override { p | slabs_; }
      55              : 
      56              :  private:
      57            0 :   std::unique_ptr<TimeSequence<uint64_t>> slabs_{};
      58              : };
      59              : }  // namespace Triggers
      60              : 
      61              : template <>
      62            0 : struct Options::create_from_yaml<Triggers::Slabs> {
      63              :   template <typename Metavariables>
      64            0 :   static Triggers::Slabs create(const Option& options) {
      65              :     return Triggers::Slabs(
      66              :         options.parse_as<std::unique_ptr<TimeSequence<uint64_t>>,
      67              :                          Metavariables>());
      68              :   }
      69              : };