rust_multistackvm 0.38.0

Stack-based virtual machine on top of rust_multistack crate
Documentation
use crate::multistackvm::{VM, StackOps};
use rust_dynamic::types::LAMBDA;
use easy_error::{Error, bail};

#[derive(Debug, Clone)]
pub enum TypeCond {
    IfTrue,
    IfFalse,
}

#[time_graph::instrument]
fn stdlib_logic_if_base(vm: &mut VM, cond: TypeCond, op: StackOps, err_prefix: String) -> Result<&mut VM, Error> {
    match op {
        StackOps::FromStack => {
            if vm.stack.current_stack_len() < 2 {
                bail!("Stack is too shallow for inline {}", &err_prefix);
            }
        }
        StackOps::FromWorkBench => {
            if vm.stack.current_stack_len() < 1 {
                bail!("Stack is too shallow for inline {}", &err_prefix);
            }
            if vm.stack.workbench.len() < 1 {
                bail!("Workbench is too shallow for inline {}", &err_prefix);
            }
        }
    }
    match vm.stack.pull() {
        Some(lambda_val) => {
            match lambda_val.type_of() {
                LAMBDA => {
                    let cond_v = match op {
                        StackOps::FromStack => vm.stack.pull(),
                        StackOps::FromWorkBench => vm.stack.pull_from_workbench(),
                    };
                    match cond_v {
                        Some(condition_val) => {
                            match condition_val.cast_bool() {
                                Ok(cond_bool) => {
                                    match cond {
                                        TypeCond::IfTrue => {
                                            if cond_bool {
                                                match lambda_val.type_of() {
                                                    LAMBDA => {
                                                        return vm.lambda_eval(lambda_val);
                                                    }
                                                    _ => {
                                                        vm.stack.push(lambda_val);
                                                        return vm.i("execute".to_string());
                                                    }
                                                }
                                            }
                                        }
                                        TypeCond::IfFalse => {
                                            if ! cond_bool {
                                                match lambda_val.type_of() {
                                                    LAMBDA => {
                                                        return vm.lambda_eval(lambda_val);
                                                    }
                                                    _ => {
                                                        vm.stack.push(lambda_val);
                                                        return vm.i("execute".to_string());
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                                Err(err) => {
                                    bail!("{} returns error: {}", &err_prefix, err);
                                }
                            }
                        }
                        None => {
                            bail!("{} returns: NO DATA #2", &err_prefix);
                        }
                    }
                }
                _ => {
                    bail!("{}: #1 parameter must be lambda", &err_prefix);
                }
            }
        }
        None => {
            bail!("{} returns: NO DATA #1", &err_prefix);
        }
    }
    Ok(vm)
}

pub fn stdlib_logic_if(vm: &mut VM) -> Result<&mut VM, Error> {
    stdlib_logic_if_base(vm, TypeCond::IfTrue, StackOps::FromStack, "IF".to_string())
}

pub fn stdlib_logic_if_in_workbench(vm: &mut VM) -> Result<&mut VM, Error> {
    stdlib_logic_if_base(vm, TypeCond::IfTrue, StackOps::FromWorkBench, "IF.".to_string())
}

pub fn stdlib_logic_if_false(vm: &mut VM) -> Result<&mut VM, Error> {
    stdlib_logic_if_base(vm, TypeCond::IfFalse, StackOps::FromStack, "?FALSE".to_string())
}

pub fn stdlib_logic_if_false_in_workbench(vm: &mut VM) -> Result<&mut VM, Error> {
    stdlib_logic_if_base(vm, TypeCond::IfFalse, StackOps::FromStack, "?FALSE.".to_string())
}

pub fn stdlib_logic_if_stack(vm: &mut VM) -> Result<&mut VM, Error> {
    if vm.stack.current_stack_len() < 2 {
        bail!("Stack is too shallow for inline if.stack");
    }
    match vm.stack.pull() {
        Some(lambda_val) => {
            if lambda_val.is_type(LAMBDA) {
                match vm.stack.pull() {
                    Some(condition_val) => {
                        match condition_val.cast_string() {
                            Ok(cond) => {
                                if vm.stack.current_stack_name() == Some(cond) {
                                    return vm.lambda_eval(lambda_val);
                                }
                            }
                            Err(err) => {
                                bail!("IF.stack returns error: {}", err);
                            }
                        }
                    }
                    None => {
                        bail!("IF.stack returns: NO DATA #2");
                    }
                }
            } else {
                bail!("IF.stack: #1 parameter must be lambda");
            }
        }
        None => {
            bail!("IF.stack returns: NO DATA #1");
        }
    }
    Ok(vm)
}



pub fn init_stdlib(vm: &mut VM) {
    let _ = vm.register_inline("if".to_string(), stdlib_logic_if);
    let _ = vm.register_inline("if.in_workbench".to_string(), stdlib_logic_if_in_workbench);
    let _ = vm.register_inline("if.false".to_string(), stdlib_logic_if_false);
    let _ = vm.register_inline("if.false.in_workbench".to_string(), stdlib_logic_if_false_in_workbench);
    let _ = vm.register_inline("if.stack".to_string(), stdlib_logic_if_stack);
}