Convert to use Pass infrastructure

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@834 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 18 insertions, 5 deletions

diff --git a/include/llvm/Transforms/Instrumentation/TraceValues.h b/include/llvm/Transforms/Instrumentation/TraceValues.h
index af21209..8f4ee1d 100644
--- a/include/llvm/Transforms/Instrumentation/TraceValues.h
+++ b/include/llvm/Transforms/Instrumentation/TraceValues.h
@@ -15,9 +15,8 @@
 #ifndef LLVM_TRANSFORMS_INSTRUMENTATION_TRACEVALUES_H
 #define LLVM_TRANSFORMS_INSTRUMENTATION_TRACEVALUES_H
 
-class Module;
-class Method;
-class BasicBlock;
+#include "llvm/Transforms/Pass.h"
+
 class Instruction;
 class Value;
 class Type;
@@ -70,6 +69,20 @@ void            InsertCodeToTraceValues (Method* method,
                                          bool traceBasicBlockExits,
                                          bool traceMethodExits);
 
-//**************************************************************************/
 
-#endif LLVM_TRANSFORMS_INSTRUMENTATION_TRACEVALUES_H
+class InsertTraceCode : public ConcretePass<InsertTraceCode> {
+  bool TraceBasicBlockExits, TraceMethodExits;
+public:
+  InsertTraceCode(bool traceBasicBlockExits, bool traceMethodExits)
+    : TraceBasicBlockExits(traceBasicBlockExits), 
+      TraceMethodExits(traceMethodExits) {}
+
+  // doPerMethodWork - This method does the work.  Always successful.
+  //
+  bool doPerMethodWorkVirt(Method *M) {
+    InsertCodeToTraceValues(M, TraceBasicBlockExits, TraceMethodExits);
+    return false;
+  }
+};
+
+#endif /*LLVM_TRANSFORMS_INSTRUMENTATION_TRACEVALUES_H*/