1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef TEST_SUPPORT_COUNTING_PREDICATES_H
10 #define TEST_SUPPORT_COUNTING_PREDICATES_H
11 
12 #include <cstddef>
13 
14 template <typename Predicate, typename Arg>
15 struct unary_counting_predicate {
16 public:
17     typedef Arg argument_type;
18     typedef bool result_type;
19 
unary_counting_predicateunary_counting_predicate20     unary_counting_predicate(Predicate p) : p_(p), count_(0) {}
~unary_counting_predicateunary_counting_predicate21     ~unary_counting_predicate() {}
22 
operatorunary_counting_predicate23     bool operator () (const Arg &a) const { ++count_; return p_(a); }
countunary_counting_predicate24     size_t count() const { return count_; }
resetunary_counting_predicate25     void reset() { count_ = 0; }
26 
27 private:
28     Predicate p_;
29     mutable size_t count_;
30 };
31 
32 
33 template <typename Predicate, typename Arg1, typename Arg2=Arg1>
34 struct binary_counting_predicate {
35 public:
36     typedef Arg1 first_argument_type;
37     typedef Arg2 second_argument_type;
38     typedef bool result_type;
39 
binary_counting_predicatebinary_counting_predicate40     binary_counting_predicate ( Predicate p ) : p_(p), count_(0) {}
~binary_counting_predicatebinary_counting_predicate41     ~binary_counting_predicate() {}
42 
operatorbinary_counting_predicate43     bool operator () (const Arg1 &a1, const Arg2 &a2) const { ++count_; return p_(a1, a2); }
countbinary_counting_predicate44     size_t count() const { return count_; }
resetbinary_counting_predicate45     void reset() { count_ = 0; }
46 
47 private:
48     Predicate p_;
49     mutable size_t count_;
50 };
51 
52 #endif // TEST_SUPPORT_COUNTING_PREDICATES_H
53