Function get_var

core/src/compiler/syms.rs:90–121

Gets the register and prototype of a local or global variable if it already exists.

(
    vref: &VarRef,
    table: &HashMapWithIds<SymbolKey, SymbolPrototype, u8>,
    make_register: MKR,
    scope: RegisterScope,
)

/// Gets the register and prototype of a local or global variable if it already exists.
fn get_var<MKR>(
    vref: &VarRef,
    table: &HashMapWithIds<SymbolKey, SymbolPrototype, u8>,
    make_register: MKR,
    scope: RegisterScope,
) -> Result<(Register, SymbolPrototype)>
where
    MKR: FnOnce(u8) -> std::result::Result<Register, bytecode::OutOfRegistersError>,
{
    let key = SymbolKey::from(&vref.name);
    match table.get(&key) {
        Some((SymbolPrototype::Array(info), reg)) => {
            if !vref.accepts(info.subtype) {
                return Err(Error::IncompatibleTypeAnnotationInReference(vref.clone()));
            }

            let reg = make_register(reg).map_err(|_| Error::OutOfRegisters(scope))?;
            Ok((reg, SymbolPrototype::Array(*info)))
        }

        Some((SymbolPrototype::Scalar(etype), reg)) => {
            if !vref.accepts(*etype) {
                return Err(Error::IncompatibleTypeAnnotationInReference(vref.clone()));
            }

            let reg = make_register(reg).map_err(|_| Error::OutOfRegisters(scope))?;
            Ok((reg, SymbolPrototype::Scalar(*etype)))
        }

        None => Err(Error::UndefinedSymbol(vref.clone(), scope)),
    }
}

/// Defines a new local or global variable (or array) and assigns a register to it.
///