orca_wasm/subiterator/
module_subiterator.rs

1//! SubIterator for a Module
2
3use crate::ir::id::FunctionID;
4use crate::ir::types::Location;
5use crate::subiterator::function_subiterator::FuncSubIterator;
6
7/// Sub-iterator for a Module. Keeps track of current location in a Module.
8pub struct ModuleSubIterator {
9    /// The current function the SubIterator is at
10    pub(crate) curr_idx: usize,
11    /// Metadata containing a functions index and number_of_instructions
12    metadata: Vec<(FunctionID, usize)>,
13    /// The function iterator used to keep track of the location in the function.
14    pub(crate) func_iterator: FuncSubIterator,
15    /// Functions to skip. Provide an empty vector if no functions are to be skipped.
16    skip_funcs: Vec<FunctionID>,
17}
18
19impl ModuleSubIterator {
20    /// Creates a new ModuleSubIterator
21    pub fn new(metadata: Vec<(FunctionID, usize)>, skip_funcs: Vec<FunctionID>) -> Self {
22        let curr_idx = 0;
23
24        let (_curr_fid, curr_num_instrs) = metadata[curr_idx];
25        let mut mod_it = ModuleSubIterator {
26            curr_idx,
27            metadata,
28            func_iterator: FuncSubIterator::new(curr_num_instrs),
29            skip_funcs,
30        };
31        mod_it.handle_skips();
32
33        mod_it
34    }
35
36    pub fn get_curr_func(&self) -> (FunctionID, usize) {
37        self.metadata[self.curr_idx]
38    }
39
40    /// Returns the current Location in the Module as a Location
41    /// and a bool value that says whether the location is at the end of the function.
42    pub fn curr_loc(&self) -> (Location, bool) {
43        let curr_instr = self.func_iterator.curr_instr;
44        (
45            Location::Module {
46                func_idx: self.get_curr_func().0,
47                instr_idx: curr_instr,
48            },
49            self.func_iterator.is_end(curr_instr),
50        )
51    }
52
53    /// Resets the ModuleSubIterator when it is a Child SubIterator of a ComponentSubIterator
54    pub(crate) fn reset_from_comp_iterator(&mut self, metadata: Vec<(FunctionID, usize)>) {
55        self.metadata = metadata;
56        self.reset();
57    }
58
59    /// Resets the ModuleSubIterator when it is not a Child SubIterator
60    pub fn reset(&mut self) {
61        self.curr_idx = 0;
62        self.handle_skips();
63        self.func_iterator.reset(self.get_curr_func().1);
64    }
65
66    fn handle_skips(&mut self) {
67        let mut curr_fid = self.get_curr_func().0;
68        while self.skip_funcs.contains(&curr_fid) {
69            self.curr_idx += 1;
70            if self.curr_idx >= self.metadata.len() {
71                break;
72            }
73            curr_fid = self.get_curr_func().0;
74        }
75    }
76
77    /// Goes to the next function in the module
78    fn next_function(&mut self) -> bool {
79        if !self.has_next_function() {
80            return false;
81        }
82        self.curr_idx += 1;
83
84        // skip over configured funcs
85        self.handle_skips();
86        if self.curr_idx < self.metadata.len() {
87            self.func_iterator = FuncSubIterator::new(self.get_curr_func().1);
88            true
89        } else {
90            false
91        }
92    }
93
94    /// Checks if there are functions left to visit
95    pub fn has_next_function(&self) -> bool {
96        self.curr_idx + 1 < self.metadata.len()
97    }
98
99    /// Checks if there are functions left to visit
100    pub(crate) fn has_next(&self) -> bool {
101        self.func_iterator.has_next() || self.has_next_function()
102    }
103
104    /// Goes to the next function
105    pub(crate) fn next(&mut self) -> bool {
106        if self.func_iterator.has_next() {
107            self.func_iterator.next()
108        } else {
109            self.next_function()
110        }
111    }
112}
orca_wasm/subiterator/module_subiterator.rs

orca_wasm/subiterator/
module_subiterator.rs
