commit fbc8dd306aa7699a866db278db08d842762b2dc2
author Lang Hames <lhames@gmail.com> Fri Feb 17 21:29:41 2012 +0000
committer Lang Hames <lhames@gmail.com> Fri Feb 17 21:29:41 2012 +0000
tree fd8c3e05b0e085f0c2c7b40101bf65ad3d34c1e1
parent d9165eb02fadbef35bc75490854c11dcfd28a920

Add support for regmask slots to HMEditor. Also fixes a comment error.

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

lib/CodeGen/LiveIntervalAnalysis.cpp

diff --git a/lib/CodeGen/LiveIntervalAnalysis.cpp b/lib/CodeGen/LiveIntervalAnalysis.cpp
index f1f66f2..4e66bf6 100644
--- a/lib/CodeGen/LiveIntervalAnalysis.cpp
+++ b/lib/CodeGen/LiveIntervalAnalysis.cpp
@@ -1069,7 +1069,6 @@
     assert(intervalRangesSane(li) && "Broke live interval moving use.");
   }
 
-  // Returns the new
   void moveUseUp(SlotIndex oldIdx, LiveRange& lr, LiveInterval& li) {
     bool liveThrough = lr.end > oldIdx.getRegSlot();
     if (liveThrough)
@@ -1098,7 +1097,7 @@
   }
 
   // Update intervals for all operands of mi from oldIndex to newIndex.
-  void moveAllOperands(MachineInstr* mi, SlotIndex oldIdx) {
+  void moveAllOperandsFrom(MachineInstr* mi, SlotIndex oldIdx) {
     // Figure out the direction we're moving.
     bool movingUp = newIdx < oldIdx;
 
@@ -1109,6 +1108,11 @@
          mopItr != mopEnd; ++mopItr) {
       const MachineOperand& mop = *mopItr;
 
+      if (mop.isRegMask()) {
+        updateRegMaskSlots(oldIdx);
+        continue;
+      }
+
       if (!mop.isReg() || mop.getReg() == 0)
         continue;
 
@@ -1249,6 +1253,16 @@
       moveUseDown(oldIdx, *lr, li, mbb);
     }
   }
+
+  void updateRegMaskSlots(SlotIndex oldIdx) {
+    SmallVectorImpl<SlotIndex>::iterator rmItr =
+      std::lower_bound(lis.RegMaskSlots.begin(), lis.RegMaskSlots.end(),
+                       oldIdx);
+    assert(*rmItr == oldIdx && "No RegMask at oldIdx.");
+    *rmItr = newIdx;
+    assert(*prior(rmItr) < *rmItr && *rmItr < *next(rmItr) &&
+           "RegSlots out of order. Did you move one call across another?");
+  }
 };
 
 void LiveIntervals::handleMove(MachineInstr* mi) {
@@ -1263,5 +1277,5 @@
   assert(!mi->isBundled() && "Can't handle bundled instructions yet.");
 
   HMEditor hme(*this, newIndex);
-  hme.moveAllOperands(mi, oldIndex);
+  hme.moveAllOperandsFrom(mi, oldIndex);
 }