Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <limits>
       7              : #include <pup.h>
       8              : 
       9              : #include "Options/Comparator.hpp"
      10              : #include "Options/String.hpp"
      11              : #include "ParallelAlgorithms/EventsAndTriggers/Trigger.hpp"
      12              : #include "Time/TimeStepId.hpp"
      13              : #include "Utilities/Serialization/CharmPupable.hpp"
      14              : #include "Utilities/TMPL.hpp"
      15              : 
      16              : /// \cond
      17              : namespace Tags {
      18              : struct TimeStepId;
      19              : }  // namespace Tags
      20              : /// \endcond
      21              : 
      22              : namespace Triggers {
      23              : /// \ingroup EventsAndTriggersGroup
      24              : /// \ingroup TimeGroup
      25              : /// Trigger based on a comparison with the slab number.
      26            1 : class SlabCompares : public Trigger {
      27              :  public:
      28              :   /// \cond
      29              :   SlabCompares() = default;
      30              :   explicit SlabCompares(CkMigrateMessage* /*unused*/) {}
      31              :   using PUP::able::register_constructor;
      32              :   WRAPPED_PUPable_decl_template(SlabCompares);  // NOLINT
      33              :   /// \endcond
      34              : 
      35            0 :   struct Comparison {
      36            0 :     using type = Options::Comparator;
      37            0 :     constexpr static Options::String help = "Comparison type";
      38              :   };
      39              : 
      40            0 :   struct Value {
      41            0 :     using type = uint64_t;
      42            0 :     constexpr static Options::String help = "Value to compare to";
      43              :   };
      44              : 
      45            0 :   using options = tmpl::list<Comparison, Value>;
      46            0 :   static constexpr Options::String help{
      47              :       "Trigger based on a comparison with the slab number."};
      48              : 
      49            0 :   explicit SlabCompares(const Options::Comparator comparator,
      50              :                         const uint64_t slab_number)
      51              :       : comparator_(comparator), slab_number_(slab_number) {}
      52              : 
      53            0 :   using argument_tags = tmpl::list<Tags::TimeStepId>;
      54              : 
      55            0 :   bool operator()(const TimeStepId& time_step_id) const {
      56              :     return comparator_(static_cast<uint64_t>(time_step_id.slab_number()),
      57              :                        slab_number_);
      58              :   }
      59              : 
      60              :   // NOLINTNEXTLINE(google-runtime-references)
      61            0 :   void pup(PUP::er& p) {
      62              :     p | comparator_;
      63              :     p | slab_number_;
      64              :   }
      65              : 
      66              :  private:
      67            0 :   Options::Comparator comparator_{};
      68            0 :   uint64_t slab_number_{};
      69              : };
      70              : }  // namespace Triggers