Use size_t instead of long to represent memory usage. long is 32 bits

[oota-llvm.git] / include / llvm / Support / InstVisitor.h

diff --git a/include/llvm/Support/InstVisitor.h b/include/llvm/Support/InstVisitor.h
index 36ee361affd080acb6afc2cf921b6751b531720a..6eef0f9a1060a7cc00103eecb52457f851a1e271 100644 (file)
--- a/include/llvm/Support/InstVisitor.h
+++ b/include/llvm/Support/InstVisitor.h
@@ -1,4 +1,11 @@
 //===- llvm/Support/InstVisitor.h - Define instruction visitors -*- C++ -*-===//
+// 
+//                     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 template class is used to define instruction visitors in a typesafe
 // manner without having to use lots of casts and a big switch statement (in
@@ -43,9 +50,10 @@
 #ifndef LLVM_SUPPORT_INSTVISITOR_H
 #define LLVM_SUPPORT_INSTVISITOR_H
 
-#include "llvm/Instruction.h"
+#include "llvm/Function.h"
+#include "llvm/Module.h"
 
-class Module;
+namespace llvm {
 
 // We operate on opaque instruction classes, so forward declare all instruction
 // types now...
@@ -57,20 +65,18 @@ class Module;
 class TerminatorInst; class BinaryOperator;
 class AllocationInst;
 
-
 #define DELEGATE(CLASS_TO_VISIT) \
   return ((SubClass*)this)->visit##CLASS_TO_VISIT((CLASS_TO_VISIT&)I)
 
 
 template<typename SubClass, typename RetTy=void>
-struct InstVisitor {
-  virtual ~InstVisitor() {}           // We are meant to be derived from
-
+class InstVisitor {
   //===--------------------------------------------------------------------===//
   // Interface code - This is the public interface of the InstVisitor that you
   // use to visit instructions...
   //
 
+public:
   // Generic visit method - Allow visitation to all instructions in a range
   template<class Iterator>
   void visit(Iterator Start, Iterator End) {
@@ -80,6 +86,10 @@ struct InstVisitor {
 
   // Define visitors for functions and basic blocks...
   //
+  void visit(Module &M) {
+    ((SubClass*)this)->visitModule(M);
+    visit(M.begin(), M.end());
+  }
   void visit(Function &F) {
     ((SubClass*)this)->visitFunction(F);
     visit(F.begin(), F.end());
@@ -119,6 +129,7 @@ struct InstVisitor {
   // When visiting a module, function or basic block directly, these methods get
   // called to indicate when transitioning into a new unit.
   //
+  void visitModule    (Module &M) {}
   void visitFunction  (Function &F) {}
   void visitBasicBlock(BasicBlock &BB) {}
 
@@ -146,6 +157,7 @@ struct InstVisitor {
   RetTy visitSwitchInst(SwitchInst &I)              { DELEGATE(TerminatorInst);}
   RetTy visitInvokeInst(InvokeInst &I)              { DELEGATE(TerminatorInst);}
   RetTy visitUnwindInst(UnwindInst &I)              { DELEGATE(TerminatorInst);}
+  RetTy visitUnreachableInst(UnreachableInst &I)    { DELEGATE(TerminatorInst);}
   RetTy visitSetCondInst(SetCondInst &I)            { DELEGATE(BinaryOperator);}
   RetTy visitMallocInst(MallocInst &I)              { DELEGATE(AllocationInst);}
   RetTy visitAllocaInst(AllocaInst &I)              { DELEGATE(AllocationInst);}
@@ -155,6 +167,7 @@ struct InstVisitor {
   RetTy visitGetElementPtrInst(GetElementPtrInst &I){ DELEGATE(Instruction); }
   RetTy visitPHINode(PHINode       &I)              { DELEGATE(Instruction); }
   RetTy visitCastInst(CastInst     &I)              { DELEGATE(Instruction); }
+  RetTy visitSelectInst(SelectInst &I)              { DELEGATE(Instruction); }
   RetTy visitCallInst(CallInst     &I)              { DELEGATE(Instruction); }
   RetTy visitShiftInst(ShiftInst   &I)              { DELEGATE(Instruction); }
   RetTy visitVANextInst(VANextInst &I)              { DELEGATE(Instruction); }
@@ -179,4 +192,6 @@ struct InstVisitor {
 
 #undef DELEGATE
 
+} // End llvm namespace
+
 #endif