[llvm-commits] CVS: llvm/lib/Transforms/IPO/StripDeadPrototypes.cpp

[Chris Lattner]

Changes in directory llvm/lib/Transforms/IPO:

StripDeadPrototypes.cpp updated: 1.3 -> 1.4
---
Log message:

simplify pass, delete dead gvar protos as well.


---
Diffs of the changes:  (+21 -16)

 StripDeadPrototypes.cpp |   37 +++++++++++++++++++++----------------
 1 files changed, 21 insertions(+), 16 deletions(-)


Index: llvm/lib/Transforms/IPO/StripDeadPrototypes.cpp
diff -u llvm/lib/Transforms/IPO/StripDeadPrototypes.cpp:1.3 llvm/lib/Transforms/IPO/StripDeadPrototypes.cpp:1.4
--- llvm/lib/Transforms/IPO/StripDeadPrototypes.cpp:1.3	Mon Feb  5 15:47:39 2007
+++ llvm/lib/Transforms/IPO/StripDeadPrototypes.cpp	Sun Feb 18 16:10:34 2007
@@ -12,13 +12,12 @@
 //
 //===----------------------------------------------------------------------===//
 
+#define DEBUG_TYPE "strip-dead-prototypes"
 #include "llvm/Transforms/IPO.h"
 #include "llvm/Pass.h"
 #include "llvm/Module.h"
 #include "llvm/ADT/Statistic.h"
-#include "llvm/Support/Debug.h"
 #include "llvm/Support/Compiler.h"
-#include <vector>
 using namespace llvm;
 
 STATISTIC(NumDeadPrototypes, "Number of dead prototypes removed");
@@ -37,24 +36,30 @@
 } // end anonymous namespace
 
 bool StripDeadPrototypesPass::runOnModule(Module &M) {
-  // Collect all the functions we want to erase
-  std::vector<Function*> FuncsToErase;
-  for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I)
-    if (I->isDeclaration() &&         // Function must be only a prototype
-        I->use_empty()) {             // Function must not be used
-      FuncsToErase.push_back(&(*I));
+  bool MadeChange = false;
+  
+  // Erase dead function prototypes.
+  for (Module::iterator I = M.begin(), E = M.end(); I != E; ) {
+    Function *F = I++;
+    // Function must be a prototype and unused.
+    if (F->isDeclaration() && F->use_empty()) {
+      F->eraseFromParent();
+      ++NumDeadPrototypes;
+      MadeChange = true;
     }
+  }
 
-  // Erase the functions
-  for (std::vector<Function*>::iterator I = FuncsToErase.begin(), 
-       E = FuncsToErase.end(); I != E; ++I )
-    (*I)->eraseFromParent();
+  // Erase dead function prototypes.
+  for (Module::global_iterator I = M.global_begin(), E = M.global_end();
+       I != E; ) {
+    GlobalVariable *GV = I++;
+    // Global must be a prototype and unused.
+    if (GV->isDeclaration() && GV->use_empty())
+      GV->eraseFromParent();
+  }
   
-  // Increment the statistic
-  NumDeadPrototypes += FuncsToErase.size();
-
   // Return an indication of whether we changed anything or not.
-  return !FuncsToErase.empty();
+  return MadeChange;
 }
 
 ModulePass *llvm::createStripDeadPrototypesPass() {