Don't leak all of the Loop objects created...

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

1 files changed, 8 insertions, 1 deletions

diff --git a/include/llvm/Analysis/LoopInfo.h b/include/llvm/Analysis/LoopInfo.h
index 13535bc7..5b6000d 100644
--- a/include/llvm/Analysis/LoopInfo.h
+++ b/include/llvm/Analysis/LoopInfo.h
@@ -46,10 +46,14 @@ public:
 private:
   friend class LoopInfo;
   inline Loop(const BasicBlock *BB) { Blocks.push_back(BB); LoopDepth = 0; }
+  ~Loop() {
+    for (unsigned i = 0, e = SubLoops.size(); i != e; ++i)
+      delete SubLoops[i];
+  }
   
   void setLoopDepth(unsigned Level) {
     LoopDepth = Level;
-    for (unsigned i = 0; i < SubLoops.size(); ++i)
+    for (unsigned i = 0, e = SubLoops.size(); i != e; ++i)
       SubLoops[i]->setLoopDepth(Level+1);
   }
 };
@@ -69,6 +73,7 @@ public:
 
   // LoopInfo ctor - Calculate the natural loop information for a CFG
   LoopInfo(AnalysisID id) { assert(id == ID); }
+  ~LoopInfo() { releaseMemory(); }
 
   const std::vector<Loop*> &getTopLevelLoops() const { return TopLevelLoops; }
 
@@ -103,6 +108,8 @@ public:
   // runOnMethod - Pass framework implementation
   virtual bool runOnMethod(Function *F);
 
+  virtual void releaseMemory();
+
   // getAnalysisUsageInfo - Provide loop info, require dominator set
   //
   virtual void getAnalysisUsageInfo(Pass::AnalysisSet &Requires,