| //===- llvm/CodeGen/SlotIndexes.h - Slot indexes representation -*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file implements SlotIndex and related classes. The purpose of SlotIndex |
| // is to describe a position at which a register can become live, or cease to |
| // be live. |
| // |
| // SlotIndex is mostly a proxy for entries of the SlotIndexList, a class which |
| // is held is LiveIntervals and provides the real numbering. This allows |
| // LiveIntervals to perform largely transparent renumbering. |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef LLVM_CODEGEN_SLOTINDEXES_H |
| #define LLVM_CODEGEN_SLOTINDEXES_H |
| |
| #include "llvm/CodeGen/MachineBasicBlock.h" |
| #include "llvm/CodeGen/MachineFunction.h" |
| #include "llvm/CodeGen/MachineFunctionPass.h" |
| #include "llvm/ADT/PointerIntPair.h" |
| #include "llvm/ADT/SmallVector.h" |
| #include "llvm/ADT/DenseMap.h" |
| #include "llvm/Support/Allocator.h" |
| |
| namespace llvm { |
| |
| /// This class represents an entry in the slot index list held in the |
| /// SlotIndexes pass. It should not be used directly. See the |
| /// SlotIndex & SlotIndexes classes for the public interface to this |
| /// information. |
| class IndexListEntry { |
| IndexListEntry *next, *prev; |
| MachineInstr *mi; |
| unsigned index; |
| |
| public: |
| |
| IndexListEntry(MachineInstr *mi, unsigned index) : mi(mi), index(index) {} |
| |
| MachineInstr* getInstr() const { return mi; } |
| void setInstr(MachineInstr *mi) { |
| this->mi = mi; |
| } |
| |
| unsigned getIndex() const { return index; } |
| void setIndex(unsigned index) { |
| this->index = index; |
| } |
| |
| IndexListEntry* getNext() { return next; } |
| const IndexListEntry* getNext() const { return next; } |
| void setNext(IndexListEntry *next) { |
| this->next = next; |
| } |
| |
| IndexListEntry* getPrev() { return prev; } |
| const IndexListEntry* getPrev() const { return prev; } |
| void setPrev(IndexListEntry *prev) { |
| this->prev = prev; |
| } |
| }; |
| |
| // Specialize PointerLikeTypeTraits for IndexListEntry. |
| template <> |
| class PointerLikeTypeTraits<IndexListEntry*> { |
| public: |
| static inline void* getAsVoidPointer(IndexListEntry *p) { |
| return p; |
| } |
| static inline IndexListEntry* getFromVoidPointer(void *p) { |
| return static_cast<IndexListEntry*>(p); |
| } |
| enum { NumLowBitsAvailable = 3 }; |
| }; |
| |
| /// SlotIndex - An opaque wrapper around machine indexes. |
| class SlotIndex { |
| friend class SlotIndexes; |
| friend struct DenseMapInfo<SlotIndex>; |
| |
| enum Slot { LOAD, USE, DEF, STORE, NUM }; |
| |
| PointerIntPair<IndexListEntry*, 2, unsigned> lie; |
| |
| SlotIndex(IndexListEntry *entry, unsigned slot) |
| : lie(entry, slot) {} |
| |
| IndexListEntry& entry() const { |
| assert(isValid() && "Attempt to compare reserved index."); |
| return *lie.getPointer(); |
| } |
| |
| int getIndex() const { |
| return entry().getIndex() | getSlot(); |
| } |
| |
| /// Returns the slot for this SlotIndex. |
| Slot getSlot() const { |
| return static_cast<Slot>(lie.getInt()); |
| } |
| |
| static inline unsigned getHashValue(const SlotIndex &v) { |
| void *ptrVal = v.lie.getOpaqueValue(); |
| return (unsigned((intptr_t)ptrVal)) ^ (unsigned((intptr_t)ptrVal) >> 9); |
| } |
| |
| public: |
| enum { |
| /// The default distance between instructions as returned by distance(). |
| /// This may vary as instructions are inserted and removed. |
| InstrDist = 4*NUM |
| }; |
| |
| static inline SlotIndex getEmptyKey() { |
| return SlotIndex(0, 1); |
| } |
| |
| static inline SlotIndex getTombstoneKey() { |
| return SlotIndex(0, 2); |
| } |
| |
| /// Construct an invalid index. |
| SlotIndex() : lie(0, 0) {} |
| |
| // Construct a new slot index from the given one, and set the slot. |
| SlotIndex(const SlotIndex &li, Slot s) |
| : lie(&li.entry(), unsigned(s)) { |
| assert(lie.getPointer() != 0 && |
| "Attempt to construct index with 0 pointer."); |
| } |
| |
| /// Returns true if this is a valid index. Invalid indicies do |
| /// not point into an index table, and cannot be compared. |
| bool isValid() const { |
| return lie.getPointer(); |
| } |
| |
| /// Return true for a valid index. |
| operator bool() const { return isValid(); } |
| |
| /// Print this index to the given raw_ostream. |
| void print(raw_ostream &os) const; |
| |
| /// Dump this index to stderr. |
| void dump() const; |
| |
| /// Compare two SlotIndex objects for equality. |
| bool operator==(SlotIndex other) const { |
| return lie == other.lie; |
| } |
| /// Compare two SlotIndex objects for inequality. |
| bool operator!=(SlotIndex other) const { |
| return lie != other.lie; |
| } |
| |
| /// Compare two SlotIndex objects. Return true if the first index |
| /// is strictly lower than the second. |
| bool operator<(SlotIndex other) const { |
| return getIndex() < other.getIndex(); |
| } |
| /// Compare two SlotIndex objects. Return true if the first index |
| /// is lower than, or equal to, the second. |
| bool operator<=(SlotIndex other) const { |
| return getIndex() <= other.getIndex(); |
| } |
| |
| /// Compare two SlotIndex objects. Return true if the first index |
| /// is greater than the second. |
| bool operator>(SlotIndex other) const { |
| return getIndex() > other.getIndex(); |
| } |
| |
| /// Compare two SlotIndex objects. Return true if the first index |
| /// is greater than, or equal to, the second. |
| bool operator>=(SlotIndex other) const { |
| return getIndex() >= other.getIndex(); |
| } |
| |
| /// isSameInstr - Return true if A and B refer to the same instruction. |
| static bool isSameInstr(SlotIndex A, SlotIndex B) { |
| return A.lie.getPointer() == B.lie.getPointer(); |
| } |
| |
| /// Return the distance from this index to the given one. |
| int distance(SlotIndex other) const { |
| return other.getIndex() - getIndex(); |
| } |
| |
| /// isLoad - Return true if this is a LOAD slot. |
| bool isLoad() const { |
| return getSlot() == LOAD; |
| } |
| |
| /// isDef - Return true if this is a DEF slot. |
| bool isDef() const { |
| return getSlot() == DEF; |
| } |
| |
| /// isUse - Return true if this is a USE slot. |
| bool isUse() const { |
| return getSlot() == USE; |
| } |
| |
| /// isStore - Return true if this is a STORE slot. |
| bool isStore() const { |
| return getSlot() == STORE; |
| } |
| |
| /// Returns the base index for associated with this index. The base index |
| /// is the one associated with the LOAD slot for the instruction pointed to |
| /// by this index. |
| SlotIndex getBaseIndex() const { |
| return getLoadIndex(); |
| } |
| |
| /// Returns the boundary index for associated with this index. The boundary |
| /// index is the one associated with the LOAD slot for the instruction |
| /// pointed to by this index. |
| SlotIndex getBoundaryIndex() const { |
| return getStoreIndex(); |
| } |
| |
| /// Returns the index of the LOAD slot for the instruction pointed to by |
| /// this index. |
| SlotIndex getLoadIndex() const { |
| return SlotIndex(&entry(), SlotIndex::LOAD); |
| } |
| |
| /// Returns the index of the USE slot for the instruction pointed to by |
| /// this index. |
| SlotIndex getUseIndex() const { |
| return SlotIndex(&entry(), SlotIndex::USE); |
| } |
| |
| /// Returns the index of the DEF slot for the instruction pointed to by |
| /// this index. |
| SlotIndex getDefIndex() const { |
| return SlotIndex(&entry(), SlotIndex::DEF); |
| } |
| |
| /// Returns the index of the STORE slot for the instruction pointed to by |
| /// this index. |
| SlotIndex getStoreIndex() const { |
| return SlotIndex(&entry(), SlotIndex::STORE); |
| } |
| |
| /// Returns the next slot in the index list. This could be either the |
| /// next slot for the instruction pointed to by this index or, if this |
| /// index is a STORE, the first slot for the next instruction. |
| /// WARNING: This method is considerably more expensive than the methods |
| /// that return specific slots (getUseIndex(), etc). If you can - please |
| /// use one of those methods. |
| SlotIndex getNextSlot() const { |
| Slot s = getSlot(); |
| if (s == SlotIndex::STORE) { |
| return SlotIndex(entry().getNext(), SlotIndex::LOAD); |
| } |
| return SlotIndex(&entry(), s + 1); |
| } |
| |
| /// Returns the next index. This is the index corresponding to the this |
| /// index's slot, but for the next instruction. |
| SlotIndex getNextIndex() const { |
| return SlotIndex(entry().getNext(), getSlot()); |
| } |
| |
| /// Returns the previous slot in the index list. This could be either the |
| /// previous slot for the instruction pointed to by this index or, if this |
| /// index is a LOAD, the last slot for the previous instruction. |
| /// WARNING: This method is considerably more expensive than the methods |
| /// that return specific slots (getUseIndex(), etc). If you can - please |
| /// use one of those methods. |
| SlotIndex getPrevSlot() const { |
| Slot s = getSlot(); |
| if (s == SlotIndex::LOAD) { |
| return SlotIndex(entry().getPrev(), SlotIndex::STORE); |
| } |
| return SlotIndex(&entry(), s - 1); |
| } |
| |
| /// Returns the previous index. This is the index corresponding to this |
| /// index's slot, but for the previous instruction. |
| SlotIndex getPrevIndex() const { |
| return SlotIndex(entry().getPrev(), getSlot()); |
| } |
| |
| }; |
| |
| /// DenseMapInfo specialization for SlotIndex. |
| template <> |
| struct DenseMapInfo<SlotIndex> { |
| static inline SlotIndex getEmptyKey() { |
| return SlotIndex::getEmptyKey(); |
| } |
| static inline SlotIndex getTombstoneKey() { |
| return SlotIndex::getTombstoneKey(); |
| } |
| static inline unsigned getHashValue(const SlotIndex &v) { |
| return SlotIndex::getHashValue(v); |
| } |
| static inline bool isEqual(const SlotIndex &LHS, const SlotIndex &RHS) { |
| return (LHS == RHS); |
| } |
| }; |
| |
| template <> struct isPodLike<SlotIndex> { static const bool value = true; }; |
| |
| |
| inline raw_ostream& operator<<(raw_ostream &os, SlotIndex li) { |
| li.print(os); |
| return os; |
| } |
| |
| typedef std::pair<SlotIndex, MachineBasicBlock*> IdxMBBPair; |
| |
| inline bool operator<(SlotIndex V, const IdxMBBPair &IM) { |
| return V < IM.first; |
| } |
| |
| inline bool operator<(const IdxMBBPair &IM, SlotIndex V) { |
| return IM.first < V; |
| } |
| |
| struct Idx2MBBCompare { |
| bool operator()(const IdxMBBPair &LHS, const IdxMBBPair &RHS) const { |
| return LHS.first < RHS.first; |
| } |
| }; |
| |
| /// SlotIndexes pass. |
| /// |
| /// This pass assigns indexes to each instruction. |
| class SlotIndexes : public MachineFunctionPass { |
| private: |
| |
| MachineFunction *mf; |
| IndexListEntry *indexListHead; |
| unsigned functionSize; |
| |
| typedef DenseMap<const MachineInstr*, SlotIndex> Mi2IndexMap; |
| Mi2IndexMap mi2iMap; |
| |
| /// MBBRanges - Map MBB number to (start, stop) indexes. |
| SmallVector<std::pair<SlotIndex, SlotIndex>, 8> MBBRanges; |
| |
| /// Idx2MBBMap - Sorted list of pairs of index of first instruction |
| /// and MBB id. |
| SmallVector<IdxMBBPair, 8> idx2MBBMap; |
| |
| // IndexListEntry allocator. |
| BumpPtrAllocator ileAllocator; |
| |
| IndexListEntry* createEntry(MachineInstr *mi, unsigned index) { |
| IndexListEntry *entry = |
| static_cast<IndexListEntry*>( |
| ileAllocator.Allocate(sizeof(IndexListEntry), |
| alignOf<IndexListEntry>())); |
| |
| new (entry) IndexListEntry(mi, index); |
| |
| return entry; |
| } |
| |
| void initList() { |
| assert(indexListHead == 0 && "Zero entry non-null at initialisation."); |
| indexListHead = createEntry(0, ~0U); |
| indexListHead->setNext(0); |
| indexListHead->setPrev(indexListHead); |
| } |
| |
| void clearList() { |
| indexListHead = 0; |
| ileAllocator.Reset(); |
| } |
| |
| IndexListEntry* getTail() { |
| assert(indexListHead != 0 && "Call to getTail on uninitialized list."); |
| return indexListHead->getPrev(); |
| } |
| |
| const IndexListEntry* getTail() const { |
| assert(indexListHead != 0 && "Call to getTail on uninitialized list."); |
| return indexListHead->getPrev(); |
| } |
| |
| // Returns true if the index list is empty. |
| bool empty() const { return (indexListHead == getTail()); } |
| |
| IndexListEntry* front() { |
| assert(!empty() && "front() called on empty index list."); |
| return indexListHead; |
| } |
| |
| const IndexListEntry* front() const { |
| assert(!empty() && "front() called on empty index list."); |
| return indexListHead; |
| } |
| |
| IndexListEntry* back() { |
| assert(!empty() && "back() called on empty index list."); |
| return getTail()->getPrev(); |
| } |
| |
| const IndexListEntry* back() const { |
| assert(!empty() && "back() called on empty index list."); |
| return getTail()->getPrev(); |
| } |
| |
| /// Insert a new entry before itr. |
| void insert(IndexListEntry *itr, IndexListEntry *val) { |
| assert(itr != 0 && "itr should not be null."); |
| IndexListEntry *prev = itr->getPrev(); |
| val->setNext(itr); |
| val->setPrev(prev); |
| |
| if (itr != indexListHead) { |
| prev->setNext(val); |
| } |
| else { |
| indexListHead = val; |
| } |
| itr->setPrev(val); |
| } |
| |
| /// Push a new entry on to the end of the list. |
| void push_back(IndexListEntry *val) { |
| insert(getTail(), val); |
| } |
| |
| /// Renumber locally after inserting newEntry. |
| void renumberIndexes(IndexListEntry *newEntry); |
| |
| public: |
| static char ID; |
| |
| SlotIndexes() : MachineFunctionPass(ID), indexListHead(0) { |
| initializeSlotIndexesPass(*PassRegistry::getPassRegistry()); |
| } |
| |
| virtual void getAnalysisUsage(AnalysisUsage &au) const; |
| virtual void releaseMemory(); |
| |
| virtual bool runOnMachineFunction(MachineFunction &fn); |
| |
| /// Dump the indexes. |
| void dump() const; |
| |
| /// Renumber the index list, providing space for new instructions. |
| void renumberIndexes(); |
| |
| /// Returns the zero index for this analysis. |
| SlotIndex getZeroIndex() { |
| assert(front()->getIndex() == 0 && "First index is not 0?"); |
| return SlotIndex(front(), 0); |
| } |
| |
| /// Returns the base index of the last slot in this analysis. |
| SlotIndex getLastIndex() { |
| return SlotIndex(back(), 0); |
| } |
| |
| /// Returns the invalid index marker for this analysis. |
| SlotIndex getInvalidIndex() { |
| return getZeroIndex(); |
| } |
| |
| /// Returns the distance between the highest and lowest indexes allocated |
| /// so far. |
| unsigned getIndexesLength() const { |
| assert(front()->getIndex() == 0 && |
| "Initial index isn't zero?"); |
| |
| return back()->getIndex(); |
| } |
| |
| /// Returns the number of instructions in the function. |
| unsigned getFunctionSize() const { |
| return functionSize; |
| } |
| |
| /// Returns true if the given machine instr is mapped to an index, |
| /// otherwise returns false. |
| bool hasIndex(const MachineInstr *instr) const { |
| return (mi2iMap.find(instr) != mi2iMap.end()); |
| } |
| |
| /// Returns the base index for the given instruction. |
| SlotIndex getInstructionIndex(const MachineInstr *instr) const { |
| Mi2IndexMap::const_iterator itr = mi2iMap.find(instr); |
| assert(itr != mi2iMap.end() && "Instruction not found in maps."); |
| return itr->second; |
| } |
| |
| /// Returns the instruction for the given index, or null if the given |
| /// index has no instruction associated with it. |
| MachineInstr* getInstructionFromIndex(SlotIndex index) const { |
| return index.isValid() ? index.entry().getInstr() : 0; |
| } |
| |
| /// Returns the next non-null index. |
| SlotIndex getNextNonNullIndex(SlotIndex index) { |
| SlotIndex nextNonNull = index.getNextIndex(); |
| |
| while (&nextNonNull.entry() != getTail() && |
| getInstructionFromIndex(nextNonNull) == 0) { |
| nextNonNull = nextNonNull.getNextIndex(); |
| } |
| |
| return nextNonNull; |
| } |
| |
| /// getIndexBefore - Returns the index of the last indexed instruction |
| /// before MI, or the the start index of its basic block. |
| /// MI is not required to have an index. |
| SlotIndex getIndexBefore(const MachineInstr *MI) const { |
| const MachineBasicBlock *MBB = MI->getParent(); |
| assert(MBB && "MI must be inserted inna basic block"); |
| MachineBasicBlock::const_iterator I = MI, B = MBB->begin(); |
| for (;;) { |
| if (I == B) |
| return getMBBStartIdx(MBB); |
| --I; |
| Mi2IndexMap::const_iterator MapItr = mi2iMap.find(I); |
| if (MapItr != mi2iMap.end()) |
| return MapItr->second; |
| } |
| } |
| |
| /// getIndexAfter - Returns the index of the first indexed instruction |
| /// after MI, or the end index of its basic block. |
| /// MI is not required to have an index. |
| SlotIndex getIndexAfter(const MachineInstr *MI) const { |
| const MachineBasicBlock *MBB = MI->getParent(); |
| assert(MBB && "MI must be inserted inna basic block"); |
| MachineBasicBlock::const_iterator I = MI, E = MBB->end(); |
| for (;;) { |
| ++I; |
| if (I == E) |
| return getMBBEndIdx(MBB); |
| Mi2IndexMap::const_iterator MapItr = mi2iMap.find(I); |
| if (MapItr != mi2iMap.end()) |
| return MapItr->second; |
| } |
| } |
| |
| /// Return the (start,end) range of the given basic block number. |
| const std::pair<SlotIndex, SlotIndex> & |
| getMBBRange(unsigned Num) const { |
| return MBBRanges[Num]; |
| } |
| |
| /// Return the (start,end) range of the given basic block. |
| const std::pair<SlotIndex, SlotIndex> & |
| getMBBRange(const MachineBasicBlock *MBB) const { |
| return getMBBRange(MBB->getNumber()); |
| } |
| |
| /// Returns the first index in the given basic block number. |
| SlotIndex getMBBStartIdx(unsigned Num) const { |
| return getMBBRange(Num).first; |
| } |
| |
| /// Returns the first index in the given basic block. |
| SlotIndex getMBBStartIdx(const MachineBasicBlock *mbb) const { |
| return getMBBRange(mbb).first; |
| } |
| |
| /// Returns the last index in the given basic block number. |
| SlotIndex getMBBEndIdx(unsigned Num) const { |
| return getMBBRange(Num).second; |
| } |
| |
| /// Returns the last index in the given basic block. |
| SlotIndex getMBBEndIdx(const MachineBasicBlock *mbb) const { |
| return getMBBRange(mbb).second; |
| } |
| |
| /// Returns the basic block which the given index falls in. |
| MachineBasicBlock* getMBBFromIndex(SlotIndex index) const { |
| if (MachineInstr *MI = getInstructionFromIndex(index)) |
| return MI->getParent(); |
| SmallVectorImpl<IdxMBBPair>::const_iterator I = |
| std::lower_bound(idx2MBBMap.begin(), idx2MBBMap.end(), index); |
| // Take the pair containing the index |
| SmallVectorImpl<IdxMBBPair>::const_iterator J = |
| ((I != idx2MBBMap.end() && I->first > index) || |
| (I == idx2MBBMap.end() && idx2MBBMap.size()>0)) ? (I-1): I; |
| |
| assert(J != idx2MBBMap.end() && J->first <= index && |
| index < getMBBEndIdx(J->second) && |
| "index does not correspond to an MBB"); |
| return J->second; |
| } |
| |
| bool findLiveInMBBs(SlotIndex start, SlotIndex end, |
| SmallVectorImpl<MachineBasicBlock*> &mbbs) const { |
| SmallVectorImpl<IdxMBBPair>::const_iterator itr = |
| std::lower_bound(idx2MBBMap.begin(), idx2MBBMap.end(), start); |
| bool resVal = false; |
| |
| while (itr != idx2MBBMap.end()) { |
| if (itr->first >= end) |
| break; |
| mbbs.push_back(itr->second); |
| resVal = true; |
| ++itr; |
| } |
| return resVal; |
| } |
| |
| /// Returns the MBB covering the given range, or null if the range covers |
| /// more than one basic block. |
| MachineBasicBlock* getMBBCoveringRange(SlotIndex start, SlotIndex end) const { |
| |
| assert(start < end && "Backwards ranges not allowed."); |
| |
| SmallVectorImpl<IdxMBBPair>::const_iterator itr = |
| std::lower_bound(idx2MBBMap.begin(), idx2MBBMap.end(), start); |
| |
| if (itr == idx2MBBMap.end()) { |
| itr = prior(itr); |
| return itr->second; |
| } |
| |
| // Check that we don't cross the boundary into this block. |
| if (itr->first < end) |
| return 0; |
| |
| itr = prior(itr); |
| |
| if (itr->first <= start) |
| return itr->second; |
| |
| return 0; |
| } |
| |
| /// Insert the given machine instruction into the mapping. Returns the |
| /// assigned index. |
| /// If Late is set and there are null indexes between mi's neighboring |
| /// instructions, create the new index after the null indexes instead of |
| /// before them. |
| SlotIndex insertMachineInstrInMaps(MachineInstr *mi, bool Late = false) { |
| assert(mi2iMap.find(mi) == mi2iMap.end() && "Instr already indexed."); |
| // Numbering DBG_VALUE instructions could cause code generation to be |
| // affected by debug information. |
| assert(!mi->isDebugValue() && "Cannot number DBG_VALUE instructions."); |
| |
| assert(mi->getParent() != 0 && "Instr must be added to function."); |
| |
| // Get the entries where mi should be inserted. |
| IndexListEntry *prevEntry, *nextEntry; |
| if (Late) { |
| // Insert mi's index immediately before the following instruction. |
| nextEntry = &getIndexAfter(mi).entry(); |
| prevEntry = nextEntry->getPrev(); |
| } else { |
| // Insert mi's index immediately after the preceeding instruction. |
| prevEntry = &getIndexBefore(mi).entry(); |
| nextEntry = prevEntry->getNext(); |
| } |
| |
| // Get a number for the new instr, or 0 if there's no room currently. |
| // In the latter case we'll force a renumber later. |
| unsigned dist = ((nextEntry->getIndex() - prevEntry->getIndex())/2) & ~3u; |
| unsigned newNumber = prevEntry->getIndex() + dist; |
| |
| // Insert a new list entry for mi. |
| IndexListEntry *newEntry = createEntry(mi, newNumber); |
| insert(nextEntry, newEntry); |
| |
| // Renumber locally if we need to. |
| if (dist == 0) |
| renumberIndexes(newEntry); |
| |
| SlotIndex newIndex(newEntry, SlotIndex::LOAD); |
| mi2iMap.insert(std::make_pair(mi, newIndex)); |
| return newIndex; |
| } |
| |
| /// Remove the given machine instruction from the mapping. |
| void removeMachineInstrFromMaps(MachineInstr *mi) { |
| // remove index -> MachineInstr and |
| // MachineInstr -> index mappings |
| Mi2IndexMap::iterator mi2iItr = mi2iMap.find(mi); |
| if (mi2iItr != mi2iMap.end()) { |
| IndexListEntry *miEntry(&mi2iItr->second.entry()); |
| assert(miEntry->getInstr() == mi && "Instruction indexes broken."); |
| // FIXME: Eventually we want to actually delete these indexes. |
| miEntry->setInstr(0); |
| mi2iMap.erase(mi2iItr); |
| } |
| } |
| |
| /// ReplaceMachineInstrInMaps - Replacing a machine instr with a new one in |
| /// maps used by register allocator. |
| void replaceMachineInstrInMaps(MachineInstr *mi, MachineInstr *newMI) { |
| Mi2IndexMap::iterator mi2iItr = mi2iMap.find(mi); |
| if (mi2iItr == mi2iMap.end()) |
| return; |
| SlotIndex replaceBaseIndex = mi2iItr->second; |
| IndexListEntry *miEntry(&replaceBaseIndex.entry()); |
| assert(miEntry->getInstr() == mi && |
| "Mismatched instruction in index tables."); |
| miEntry->setInstr(newMI); |
| mi2iMap.erase(mi2iItr); |
| mi2iMap.insert(std::make_pair(newMI, replaceBaseIndex)); |
| } |
| |
| /// Add the given MachineBasicBlock into the maps. |
| void insertMBBInMaps(MachineBasicBlock *mbb) { |
| MachineFunction::iterator nextMBB = |
| llvm::next(MachineFunction::iterator(mbb)); |
| IndexListEntry *startEntry = createEntry(0, 0); |
| IndexListEntry *stopEntry = createEntry(0, 0); |
| IndexListEntry *nextEntry = 0; |
| |
| if (nextMBB == mbb->getParent()->end()) { |
| nextEntry = getTail(); |
| } else { |
| nextEntry = &getMBBStartIdx(nextMBB).entry(); |
| } |
| |
| insert(nextEntry, startEntry); |
| insert(nextEntry, stopEntry); |
| |
| SlotIndex startIdx(startEntry, SlotIndex::LOAD); |
| SlotIndex endIdx(nextEntry, SlotIndex::LOAD); |
| |
| assert(unsigned(mbb->getNumber()) == MBBRanges.size() && |
| "Blocks must be added in order"); |
| MBBRanges.push_back(std::make_pair(startIdx, endIdx)); |
| |
| idx2MBBMap.push_back(IdxMBBPair(startIdx, mbb)); |
| |
| renumberIndexes(); |
| std::sort(idx2MBBMap.begin(), idx2MBBMap.end(), Idx2MBBCompare()); |
| } |
| |
| }; |
| |
| |
| // Specialize IntervalMapInfo for half-open slot index intervals. |
| template <typename> struct IntervalMapInfo; |
| template <> struct IntervalMapInfo<SlotIndex> { |
| static inline bool startLess(const SlotIndex &x, const SlotIndex &a) { |
| return x < a; |
| } |
| static inline bool stopLess(const SlotIndex &b, const SlotIndex &x) { |
| return b <= x; |
| } |
| static inline bool adjacent(const SlotIndex &a, const SlotIndex &b) { |
| return a == b; |
| } |
| }; |
| |
| } |
| |
| #endif // LLVM_CODEGEN_LIVEINDEX_H |