test/support/Counter.h

   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 COUNTER_H
  11 #define COUNTER_H
  12
  13 #include <functional> // for std::hash
  14
  15 #include "test_macros.h"
  16
  17 struct Counter_base { static int gConstructed; };
  18
  19 template <typename T>
  20 class Counter : public Counter_base
  21 {
  22 public:
  23     Counter() : data_()                             { ++gConstructed; }
  24     Counter(const T &data) : data_(data)            { ++gConstructed; }
  25     Counter(const Counter& rhs) : data_(rhs.data_)  { ++gConstructed; }
  26     Counter& operator=(const Counter& rhs)          { ++gConstructed; data_ = rhs.data_; return *this; }
  27 #if TEST_STD_VER >= 11
  28     Counter(Counter&& rhs) : data_(std::move(rhs.data_))  { ++gConstructed; }
  29     Counter& operator=(Counter&& rhs) { ++gConstructed; data_ = std::move(rhs.data_); return *this; }
  30 #endif
  31     ~Counter() { --gConstructed; }
  32
  33     const T& get() const {return data_;}
  34
  35     bool operator==(const Counter& x) const {return data_ == x.data_;}
  36     bool operator< (const Counter& x) const {return data_ <  x.data_;}
  37
  38 private:
  39     T data_;
  40 };
  41
  42 int Counter_base::gConstructed = 0;
  43
  44 namespace std {
  45
  46 template <class T>
  47 struct hash<Counter<T> >
  48     : public std::unary_function<Counter<T>, std::size_t>
  49 {
  50     std::size_t operator()(const Counter<T>& x) const {return std::hash<T>(x.get());}
  51 };
  52 }
  53
  54 #endif