From 2317142c753bbe95e63a6e64f3e313fd9de1e089 Mon Sep 17 00:00:00 2001
From: Jakob Stoklund Olesen <jolesen@mozilla.com>
Date: Tue, 14 Feb 2017 15:52:44 -0800
Subject: [PATCH] Add a Layout::next_ebb() method.

This lets us iterate over the blocks in a function without holding a
reference to the layout.
---
 lib/cretonne/src/ir/layout.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lib/cretonne/src/ir/layout.rs b/lib/cretonne/src/ir/layout.rs
index 50dfa61f4e..d30f336780 100644
--- a/lib/cretonne/src/ir/layout.rs
+++ b/lib/cretonne/src/ir/layout.rs
@@ -348,6 +348,11 @@ impl Layout {
     pub fn entry_block(&self) -> Option<Ebb> {
         self.first_ebb
     }
+
+    /// Get the block following `ebb` in the layout order.
+    pub fn next_ebb(&self, ebb: Ebb) -> Option<Ebb> {
+        self.ebbs[ebb].next.expand()
+    }
 }
 
 #[derive(Clone, Debug, Default)]