[PM] Make the LoopInfoBase and LoopInfo objects movable so that they can

be used as results in the new pass manager.

llvm-svn: 226559

1 files changed, 24 insertions, 0 deletions

diff --git a/llvm/include/llvm/Analysis/LoopInfo.h b/llvm/include/llvm/Analysis/LoopInfo.h
index 18e572a1904..0b78983380f 100644
--- a/llvm/include/llvm/Analysis/LoopInfo.h
+++ b/llvm/include/llvm/Analysis/LoopInfo.h
@@ -502,6 +502,24 @@ public:
   LoopInfoBase() { }
   ~LoopInfoBase() { releaseMemory(); }
 
+  LoopInfoBase(LoopInfoBase &&Arg)
+      : BBMap(std::move(Arg.BBMap)),
+        TopLevelLoops(std::move(Arg.TopLevelLoops)) {
+    // We have to clear the arguments top level loops as we've taken ownership.
+    Arg.TopLevelLoops.clear();
+  }
+  LoopInfoBase &operator=(LoopInfoBase &&RHS) {
+    if (&RHS != this) {
+      BBMap = std::move(RHS.BBMap);
+
+      for (auto *L : TopLevelLoops)
+        delete L;
+      TopLevelLoops = std::move(RHS.TopLevelLoops);
+      RHS.TopLevelLoops.clear();
+    }
+    return *this;
+  }
+
   void releaseMemory() {
     BBMap.clear();
 
@@ -635,6 +653,12 @@ class LoopInfo : public LoopInfoBase<BasicBlock, Loop> {
 public:
   LoopInfo() {}
 
+  LoopInfo(LoopInfo &&Arg) : BaseT(std::move(static_cast<BaseT &>(Arg))) {}
+  LoopInfo &operator=(LoopInfo &&RHS) {
+    BaseT::operator=(std::move(static_cast<BaseT &>(RHS)));
+    return *this;
+  }
+
   // Most of the public interface is provided via LoopInfoBase.
 
   /// updateUnloop - Update LoopInfo after removing the last backedge from a