make traits more flexible by splitting out node-related fragment

[oota-llvm.git] / include / llvm / ADT / Statistic.h

diff --git a/include/llvm/ADT/Statistic.h b/include/llvm/ADT/Statistic.h
index 74abe6442b6fa528651b3f0c3c43f55a0e114a52..537f86637953bf76251761912b6cd66b83273e97 100644 (file)
--- a/include/llvm/ADT/Statistic.h
+++ b/include/llvm/ADT/Statistic.h
@@ -2,8 +2,8 @@
 //
 //                     The LLVM Compiler Infrastructure
 //
-// This file was developed by the LLVM research group and is distributed under
-// the University of Illinois Open Source License. See LICENSE.TXT for details.
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 //
@@ -29,20 +29,22 @@
 namespace llvm {
 
 class Statistic {
+public:
   const char *Name;
   const char *Desc;
   unsigned Value : 31;
   bool Initialized : 1;
-public:
-  // Normal constructor, default initialize data item...
-  Statistic(const char *name, const char *desc)
-    : Name(name), Desc(desc), Value(0), Initialized(0) {
-  }
 
   unsigned getValue() const { return Value; }
   const char *getName() const { return Name; }
   const char *getDesc() const { return Desc; }
-  
+
+  /// construct - This should only be called for non-global statistics.
+  void construct(const char *name, const char *desc) {
+    Name = name; Desc = desc;
+    Value = 0; Initialized = 0;
+  }
+
   // Allow use of this class as the value itself.
   operator unsigned() const { return Value; }
   const Statistic &operator=(unsigned Val) { Value = Val; return init(); }
@@ -54,8 +56,8 @@ public:
   const Statistic &operator-=(const unsigned &V) { Value -= V; return init(); }
   const Statistic &operator*=(const unsigned &V) { Value *= V; return init(); }
   const Statistic &operator/=(const unsigned &V) { Value /= V; return init(); }
-  
-private:
+
+protected:
   Statistic &init() {
     if (!Initialized) RegisterStatistic();
     return *this;
@@ -63,6 +65,11 @@ private:
   void RegisterStatistic();
 };
 
+// STATISTIC - A macro to make definition of statistics really simple.  This
+// automatically passes the DEBUG_TYPE of the file into the statistic.
+#define STATISTIC(VARNAME, DESC) \
+  static llvm::Statistic VARNAME = { DEBUG_TYPE, DESC, 0, 0 }
+
 } // End llvm namespace
 
 #endif