commit 3b7261b18efa2ffd058def98b6bfccbe3ee398b6
author Chris Lattner <sabre@nondot.org> Fri Nov 17 01:52:23 2006 +0000
committer Chris Lattner <sabre@nondot.org> Fri Nov 17 01:52:23 2006 +0000
tree 7f60d816d992320532158b8cfdf43c46db2d38ae
parent 71adba6dce685db4a67597174aa425a838481687

implement a todo: change a map into a vector

llvm-svn: 31805

llvm/lib/Target/PowerPC/PPCBranchSelector.cpp

diff --git a/llvm/lib/Target/PowerPC/PPCBranchSelector.cpp b/llvm/lib/Target/PowerPC/PPCBranchSelector.cpp
index dcbcfbd..7efa2ef 100644
--- a/llvm/lib/Target/PowerPC/PPCBranchSelector.cpp
+++ b/llvm/lib/Target/PowerPC/PPCBranchSelector.cpp
@@ -23,7 +23,6 @@
 #include "llvm/Target/TargetAsmInfo.h"
 #include "llvm/ADT/Statistic.h"
 #include "llvm/Support/Compiler.h"
-#include <map>
 using namespace llvm;
 
 static Statistic<> NumExpanded("ppc-branch-select",
@@ -31,9 +30,8 @@
 
 namespace {
   struct VISIBILITY_HIDDEN PPCBSel : public MachineFunctionPass {
-    /// OffsetMap - Mapping between BB and byte offset from start of function.
-    /// TODO: replace this with a vector, using the MBB idx as the key.
-    std::map<MachineBasicBlock*, unsigned> OffsetMap;
+    /// OffsetMap - Mapping between BB # and byte offset from start of function.
+    std::vector<unsigned> OffsetMap;
 
     virtual bool runOnMachineFunction(MachineFunction &Fn);
 
@@ -81,12 +79,14 @@
   // Running total of instructions encountered since beginning of function
   unsigned ByteCount = 0;
   
+  OffsetMap.resize(Fn.getNumBlockIDs());
+  
   // For each MBB, add its offset to the offset map, and count up its
   // instructions
   for (MachineFunction::iterator MFI = Fn.begin(), E = Fn.end(); MFI != E;
        ++MFI) {
     MachineBasicBlock *MBB = MFI;
-    OffsetMap[MBB] = ByteCount;
+    OffsetMap[MBB->getNumber()] = ByteCount;
     
     for (MachineBasicBlock::iterator MBBI = MBB->begin(), EE = MBB->end();
          MBBI != EE; ++MBBI)
@@ -128,7 +128,7 @@
       unsigned Opcode = MBBI->getOperand(1).getImmedValue();
       unsigned CRReg = MBBI->getOperand(0).getReg();
       
-      int Displacement = OffsetMap[DestMBB] - ByteCount;
+      int Displacement = OffsetMap[DestMBB->getNumber()] - ByteCount;
       unsigned Inverted = PPCInstrInfo::invertPPCBranchOpcode(Opcode);
       
       MachineBasicBlock::iterator MBBJ;