added public methods to iterate all blocks of a local function

Author: reuvenpo

src/module/functions/local_function/mod.rs

@@ -10,6 +10,7 @@ use crate::ir::*;
 use crate::map::{IdHashMap, IdHashSet};
 use crate::parse::IndicesToIds;
 use crate::{Data, DataId, FunctionBuilder, FunctionId, MemoryId, Module, Result, TypeId, ValType};
+use id_arena::Id;
 use std::collections::BTreeMap;
 use wasmparser::{FuncValidator, Operator, ValidatorResources};
 
@@ -104,6 +105,16 @@ impl LocalFunction {
         &self.builder.arena[id]
     }
 
+    /// Iterate over all the blocks in the function
+    pub fn blocks(&self) -> impl Iterator<Item = (Id<InstrSeq>, &InstrSeq)> {
+        self.builder.arena.iter()
+    }
+
+    /// Mutably iterate over all the blocks in the function
+    pub fn blocks_mut(&mut self) -> impl Iterator<Item = (Id<InstrSeq>, &mut InstrSeq)> {
+        self.builder.arena.iter_mut()
+    }
+
     /// Get the block associated with the given id.
     pub fn block_mut(&mut self, id: InstrSeqId) -> &mut InstrSeq {
         &mut self.builder.arena[id]