mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add public API to retrieve internal locals

Browse Source
This commit is contained in:
Kirby Linvill 2023-10-25 16:51:18 +01:00
parent 93d1b3e92a
commit f4d80a5f09
No known key found for this signature in database
GPG Key ID: E304CE3F028E6E3F
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
compiler/stable_mir/src/mir/body.rs
View File

@ -33,15 +33,21 @@ impl Body {
Self { blocks, locals, arg_count } Self { blocks, locals, arg_count }
} }
/// Gets the function's return local. /// Return local that holds this function's return value.
pub fn ret_local(&self) -> &LocalDecl { pub fn ret_local(&self) -> &LocalDecl {
&self.locals[0] &self.locals[0]
} }
/// Gets the locals in `self` that correspond to the function's arguments. /// Locals in `self` that correspond to this function's arguments.
pub fn arg_locals(&self) -> &[LocalDecl] { pub fn arg_locals(&self) -> &[LocalDecl] {
&self.locals[1..self.arg_count + 1] &self.locals[1..self.arg_count + 1]
} }
/// Internal locals for this function. These are the locals that are
/// neither the return local nor the argument locals.
pub fn internal_locals(&self) -> &[LocalDecl] {
&self.locals[self.arg_count + 1..]
}
} }
type LocalDecls = Vec<LocalDecl>; type LocalDecls = Vec<LocalDecl>;