Make var methods better

This commit is contained in:
greg 2017-12-10 03:35:51 -08:00
parent e243b99d3b
commit 254f2ae4b8
1 changed files with 21 additions and 7 deletions

View File

@ -9,6 +9,25 @@ pub struct EvaluatorState {
vars: HashMap<String, Sexp> vars: HashMap<String, Sexp>
} }
impl EvaluatorState {
fn new() -> EvaluatorState {
EvaluatorState {
vars: HashMap::new(),
}
}
fn set_var(&mut self, var: String, value: Sexp) {
self.vars.insert(var, value);
}
fn get_var(&self, var: &str) -> Option<&Sexp> {
self.vars.get(var)
}
fn push_env(&mut self) {
}
fn pop_env(&mut self) {
}
}
pub struct Rukka { pub struct Rukka {
state: EvaluatorState state: EvaluatorState
} }
@ -48,16 +67,11 @@ impl ProgrammingLanguageInterface for Rukka {
} }
impl EvaluatorState { impl EvaluatorState {
fn new() -> EvaluatorState {
EvaluatorState {
vars: HashMap::new(),
}
}
fn eval(&mut self, expr: Sexp) -> Result<Sexp, String> { fn eval(&mut self, expr: Sexp) -> Result<Sexp, String> {
use self::Sexp::*; use self::Sexp::*;
Ok(match expr { Ok(match expr {
SymbolAtom(ref sym) => { SymbolAtom(ref sym) => {
match self.vars.get(sym) { match self.get_var(sym) {
Some(ref sexp) => { Some(ref sexp) => {
let q: &Sexp = sexp; //WTF? if I delete this line, the copy doesn't work?? let q: &Sexp = sexp; //WTF? if I delete this line, the copy doesn't work??
q.clone() //TODO make this not involve a clone q.clone() //TODO make this not involve a clone
@ -105,7 +119,7 @@ impl EvaluatorState {
"define" => match operands { "define" => match operands {
Cons(box SymbolAtom(sym), box Cons(box expr, box Nil)) => { Cons(box SymbolAtom(sym), box Cons(box expr, box Nil)) => {
let evaluated = self.eval(expr)?; let evaluated = self.eval(expr)?;
self.vars.insert(sym, evaluated); self.set_var(sym, evaluated);
Nil Nil
}, },
_ => return Err(format!("Bad assignment")), _ => return Err(format!("Bad assignment")),