Factor mir::[Mut]Visitor implementations into a common macro.

Author: michaelwoerister

src/librustc/mir/repr.rs

@@ -20,6 +20,7 @@ use syntax::codemap::Span;
 use std::borrow::{Cow, IntoCow};
 use std::fmt::{self, Debug, Formatter, Write};
 use std::{iter, u32};
+use std::ops::{Index, IndexMut};
 
 /// Lowered representation of a single function.
 #[derive(RustcEncodable, RustcDecodable)]
@@ -67,6 +68,22 @@ impl<'tcx> Mir<'tcx> {
     }
 }
 
+impl<'tcx> Index<BasicBlock> for Mir<'tcx> {
+    type Output = BasicBlockData<'tcx>;
+
+    #[inline]
+    fn index(&self, index: BasicBlock) -> &BasicBlockData<'tcx> {
+        self.basic_block_data(index)
+    }
+}
+
+impl<'tcx> IndexMut<BasicBlock> for Mir<'tcx> {
+    #[inline]
+    fn index_mut(&mut self, index: BasicBlock) -> &mut BasicBlockData<'tcx> {
+        self.basic_block_data_mut(index)
+    }
+}
+
 ///////////////////////////////////////////////////////////////////////////
 // Mutability and borrow kinds