aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--include/llvm/PassManager.h23
-rw-r--r--lib/VMCore/PassManager.cpp56
2 files changed, 79 insertions, 0 deletions
diff --git a/include/llvm/PassManager.h b/include/llvm/PassManager.h
index 7d4a2df6d1..2769a4dbf2 100644
--- a/include/llvm/PassManager.h
+++ b/include/llvm/PassManager.h
@@ -17,6 +17,9 @@
#ifndef LLVM_PASSMANAGER_H
#define LLVM_PASSMANAGER_H
+#include "llvm/Pass.h"
+#include <vector>
+
namespace llvm {
class Pass;
@@ -85,6 +88,26 @@ public:
bool doFinalization();
};
+/// BasicBlockpassManager_New manages BasicBlockPass. It batches all the
+/// pass together and sequence them to process one basic block before
+/// processing next basic block.
+class BasicBlockPassManager_New: public Pass {
+
+public:
+ BasicBlockPassManager_New() { }
+
+ /// Add a pass into a passmanager queue.
+ bool addPass(Pass *p);
+
+ /// Execute all of the passes scheduled for execution. Keep track of
+ /// whether any of the passes modifies the function, and if so, return true.
+ bool runOnFunction(Function &F);
+
+private:
+ // Collection of pass that are not yet scheduled
+ std::vector<Pass *> PassVector;
+};
+
} // End llvm namespace
#endif
diff --git a/lib/VMCore/PassManager.cpp b/lib/VMCore/PassManager.cpp
new file mode 100644
index 0000000000..bd8faac60a
--- /dev/null
+++ b/lib/VMCore/PassManager.cpp
@@ -0,0 +1,56 @@
+//===- PassManager.cpp - LLVM Pass Infrastructure Implementation ----------===//
+//
+// 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 implements the LLVM Pass Manager infrastructure.
+//
+//===----------------------------------------------------------------------===//
+
+
+#include "llvm/PassManager.h"
+#include "llvm/Function.h"
+#include "llvm/Module.h"
+
+using namespace llvm;
+
+/// BasicBlockPassManager implementation
+
+/// Add pass P into PassVector and return TRUE. If this pass is not
+/// manageable by this manager then return FALSE.
+bool
+BasicBlockPassManager_New::addPass (Pass *P) {
+
+ BasicBlockPass *BP = dynamic_cast<BasicBlockPass*>(P);
+ if (!BP)
+ return false;
+
+ // TODO: Check if it suitable to manage P using this BasicBlockPassManager
+ // or we need another instance of BasicBlockPassManager
+
+ // Add pass
+ PassVector.push_back(BP);
+ return true;
+}
+
+/// Execute all of the passes scheduled for execution by invoking
+/// runOnBasicBlock method. Keep track of whether any of the passes modifies
+/// the function, and if so, return true.
+bool
+BasicBlockPassManager_New::runOnFunction(Function &F) {
+
+ bool Changed = false;
+ for (Function::iterator I = F.begin(), E = F.end(); I != E; ++I)
+ for (std::vector<Pass *>::iterator itr = PassVector.begin(),
+ e = PassVector.end(); itr != e; ++itr) {
+ Pass *P = *itr;
+ BasicBlockPass *BP = dynamic_cast<BasicBlockPass*>(P);
+ Changed |= BP->runOnBasicBlock(*I);
+ }
+ return Changed;
+}
+