diff options
Diffstat (limited to 'tvix/eval/src/upvalues.rs')
-rw-r--r-- | tvix/eval/src/upvalues.rs | 41 |
1 files changed, 11 insertions, 30 deletions
diff --git a/tvix/eval/src/upvalues.rs b/tvix/eval/src/upvalues.rs index 6da9bcf7cc10..450ea130ff24 100644 --- a/tvix/eval/src/upvalues.rs +++ b/tvix/eval/src/upvalues.rs @@ -7,10 +7,7 @@ //! in order to resolve each free variable in the scope to a value. //! "Upvalue" is a term taken from Lua. -use std::{ - cell::{Ref, RefMut}, - ops::Index, -}; +use std::ops::Index; use crate::{opcode::UpvalueIdx, Value}; @@ -68,6 +65,16 @@ impl Upvalues { Some(stack) => stack.len(), } } + + /// Resolve deferred upvalues from the provided stack slice, + /// mutating them in the internal upvalue slots. + pub fn resolve_deferred_upvalues(&mut self, stack: &[Value]) { + for upvalue in self.static_upvalues.iter_mut() { + if let Value::DeferredUpvalue(update_from_idx) = upvalue { + *upvalue = stack[update_from_idx.0].clone(); + } + } + } } impl Index<UpvalueIdx> for Upvalues { @@ -77,29 +84,3 @@ impl Index<UpvalueIdx> for Upvalues { &self.static_upvalues[index.0] } } - -/// `UpvalueCarrier` is implemented by all types that carry upvalues. -pub trait UpvalueCarrier { - fn upvalue_count(&self) -> usize; - - /// Read-only accessor for the stored upvalues. - fn upvalues(&self) -> Ref<'_, Upvalues>; - - /// Mutable accessor for stored upvalues. - fn upvalues_mut(&self) -> RefMut<'_, Upvalues>; - - /// Read an upvalue at the given index. - fn upvalue(&self, idx: UpvalueIdx) -> Ref<'_, Value> { - Ref::map(self.upvalues(), |v| &v.static_upvalues[idx.0]) - } - - /// Resolve deferred upvalues from the provided stack slice, - /// mutating them in the internal upvalue slots. - fn resolve_deferred_upvalues(&self, stack: &[Value]) { - for upvalue in self.upvalues_mut().static_upvalues.iter_mut() { - if let Value::DeferredUpvalue(idx) = upvalue { - *upvalue = stack[idx.0].clone(); - } - } - } -} |