1 files changed, 8 insertions, 1 deletions
diff --git a/include/llvm/CodeGen/MachineBasicBlock.h b/include/llvm/CodeGen/MachineBasicBlock.h
index 5faf2a21a9..f12c20008c 100644
--- a/include/llvm/CodeGen/MachineBasicBlock.h
+++ b/include/llvm/CodeGen/MachineBasicBlock.h
@@ -63,9 +63,11 @@ public:
   const BasicBlock *BB;
   std::vector<MachineBasicBlock *> Predecessors;
   std::vector<MachineBasicBlock *> Successors;
+  int Number;
 
 public:
-  MachineBasicBlock(const BasicBlock *bb = 0) : Prev(0), Next(0), BB(bb) {
+  MachineBasicBlock(const BasicBlock *bb = 0) : Prev(0), Next(0), BB(bb),
+                                                Number(-1) {
     Insts.parent = this;
   }
   ~MachineBasicBlock() {}
@@ -78,6 +80,7 @@ public:
   /// getParent - Return the MachineFunction containing this basic block.
   ///
   const MachineFunction *getParent() const;
+        MachineFunction *getParent();
 
   typedef ilist<MachineInstr>::iterator                       iterator;
   typedef ilist<MachineInstr>::const_iterator           const_iterator;
@@ -158,6 +161,10 @@ public:
   void dump() const;
   void print(std::ostream &OS) const;
 
+  // MachineBasicBlocks are uniquely numbered at the function level
+  // (unless they're not in a MachineFunction yet)
+  int getNumber() const { return Number; }
+
 private:   // Methods used to maintain doubly linked list of blocks...
   friend class ilist_traits<MachineBasicBlock>;