1 //===----------------------------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef __COUNTING_PREDICATES_H
11 #define __COUNTING_PREDICATES_H
12 
13 
14 template <typename Predicate, typename Arg>
15 struct unary_counting_predicate : public std::unary_function<Arg, bool>  {
16 public:
unary_counting_predicateunary_counting_predicate17     unary_counting_predicate(Predicate p) : p_(p), count_(0) {}
~unary_counting_predicateunary_counting_predicate18     ~unary_counting_predicate() {}
19 
operator ()unary_counting_predicate20     bool operator () (const Arg &a) const { ++count_; return p_(a); }
countunary_counting_predicate21     size_t count() const { return count_; }
resetunary_counting_predicate22     void reset() { count_ = 0; }
23 
24 private:
25     Predicate p_;
26     mutable size_t count_;
27     };
28 
29 
30 template <typename Predicate, typename Arg1, typename Arg2=Arg1>
31 struct binary_counting_predicate : public std::binary_function<Arg1, Arg2, bool> {
32 public:
33 
binary_counting_predicatebinary_counting_predicate34     binary_counting_predicate ( Predicate p ) : p_(p), count_(0) {}
~binary_counting_predicatebinary_counting_predicate35     ~binary_counting_predicate() {}
36 
operator ()binary_counting_predicate37     bool operator () (const Arg1 &a1, const Arg2 &a2) const { ++count_; return p_(a1, a2); }
countbinary_counting_predicate38     size_t count() const { return count_; }
resetbinary_counting_predicate39     void reset() { count_ = 0; }
40 
41 private:
42     Predicate p_;
43     mutable size_t count_;
44     };
45 
46 #endif // __COUNTING_PREDICATES_H
47