Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include "Utilities/ErrorHandling/Error.hpp"
       7              : 
       8              : /// \cond
       9              : namespace Options {
      10              : class Option;
      11              : template <typename T>
      12              : struct create_from_yaml;
      13              : }  // namespace Options
      14              : namespace PUP {
      15              : class er;
      16              : }  // namespace PUP
      17              : /// \endcond
      18              : 
      19              : namespace Options {
      20              : /// An option-creatable mathematical comparison.
      21            1 : class Comparator {
      22              :  public:
      23            0 :   enum class Comparison {
      24              :     EqualTo,
      25              :     NotEqualTo,
      26              :     LessThan,
      27              :     GreaterThan,
      28              :     LessThanOrEqualTo,
      29              :     GreaterThanOrEqualTo
      30              :   };
      31              : 
      32            0 :   Comparator() = default;
      33            0 :   Comparator(Comparison comparison);
      34              : 
      35              :   template <typename T1, typename T2>
      36            0 :   bool operator()(const T1& t1, const T2& t2) const {
      37              :     switch (comparison_) {
      38              :       case Comparison::EqualTo:
      39              :         return t1 == t2;
      40              :       case Comparison::NotEqualTo:
      41              :         return t1 != t2;
      42              :       case Comparison::LessThan:
      43              :         return t1 < t2;
      44              :       case Comparison::GreaterThan:
      45              :         return t1 > t2;
      46              :       case Comparison::LessThanOrEqualTo:
      47              :         return t1 <= t2;
      48              :       case Comparison::GreaterThanOrEqualTo:
      49              :         return t1 >= t2;
      50              :       default:
      51              :         ERROR("Invalid comparison");
      52              :     }
      53              :   }
      54              : 
      55              :   // NOLINTNEXTLINE(google-runtime-references)
      56            0 :   void pup(PUP::er& p);
      57              : 
      58              :  private:
      59            0 :   Comparison comparison_;
      60              : };
      61              : 
      62              : template <>
      63            0 : struct create_from_yaml<Comparator> {
      64              :   template <typename Metavariables>
      65            0 :   static Comparator create(const Option& options) {
      66              :     return create_impl(options);
      67              :   }
      68              : 
      69              :  private:
      70            0 :   static Comparator create_impl(const Option& options);
      71              : };
      72              : }  // namespace Options