schala/schala-lang/language/src/eval.rs

747 lines
22 KiB
Rust
Raw Normal View History

2018-08-14 21:17:43 -07:00
use std::cell::RefCell;
2017-10-13 03:05:18 -07:00
use std::rc::Rc;
2018-03-03 13:26:22 -08:00
use std::fmt::Write;
2018-05-12 02:27:54 -07:00
use std::io;
2018-03-03 13:26:22 -08:00
use itertools::Itertools;
2019-01-07 13:00:37 -08:00
use crate::util::ScopeStack;
use crate::reduced_ast::{BoundVars, ReducedAST, Stmt, Expr, Lit, Func, Alternative, Subpattern};
use crate::symbol_table::{SymbolSpec, Symbol, SymbolTable};
2017-09-30 23:30:02 -07:00
pub struct State<'a> {
2018-08-19 21:31:28 -07:00
values: ScopeStack<'a, Rc<String>, ValueEntry>,
symbol_table_handle: Rc<RefCell<SymbolTable>>,
2017-10-13 03:05:18 -07:00
}
2018-05-11 01:56:12 -07:00
2018-05-12 01:18:44 -07:00
macro_rules! builtin_binding {
($name:expr, $values:expr) => {
2018-08-13 23:28:03 -07:00
$values.insert(Rc::new(format!($name)), ValueEntry::Binding { constant: true, val: Node::Expr(Expr::Func(Func::BuiltIn(Rc::new(format!($name))))) });
2018-05-12 01:18:44 -07:00
}
}
2018-05-11 01:56:12 -07:00
impl<'a> State<'a> {
pub fn new(symbol_table_handle: Rc<RefCell<SymbolTable>>) -> State<'a> {
2018-08-19 21:31:28 -07:00
let mut values = ScopeStack::new(Some(format!("global")));
2018-05-12 01:18:44 -07:00
builtin_binding!("print", values);
builtin_binding!("println", values);
2018-05-12 02:27:54 -07:00
builtin_binding!("getline", values);
State { values, symbol_table_handle }
2018-05-11 01:56:12 -07:00
}
2018-05-11 02:33:19 -07:00
pub fn debug_print(&self) -> String {
format!("Values: {:?}", self.values)
}
fn new_frame(&'a self, items: &'a Vec<Node>, bound_vars: &BoundVars) -> State<'a> {
let mut inner_state = State {
values: self.values.new_scope(None),
symbol_table_handle: self.symbol_table_handle.clone(),
};
for (bound_var, val) in bound_vars.iter().zip(items.iter()) {
if let Some(bv) = bound_var.as_ref() {
inner_state.values.insert(bv.clone(), ValueEntry::Binding { constant: true, val: val.clone() });
}
}
inner_state
}
2018-05-11 01:56:12 -07:00
}
2018-08-13 23:28:03 -07:00
#[derive(Debug, Clone)]
enum Node {
Expr(Expr),
PrimObject {
name: Rc<String>,
tag: usize,
items: Vec<Node>,
2018-08-14 02:03:05 -07:00
},
PrimTuple {
items: Vec<Node>
2018-08-13 23:28:03 -07:00
}
}
fn paren_wrapped_vec(terms: impl Iterator<Item=String>) -> String {
let mut buf = String::new();
write!(buf, "(").unwrap();
for term in terms.map(|e| Some(e)).intersperse(None) {
match term {
Some(e) => write!(buf, "{}", e).unwrap(),
None => write!(buf, ", ").unwrap(),
};
}
write!(buf, ")").unwrap();
buf
}
impl Node {
fn to_repl(&self) -> String {
match self {
Node::Expr(e) => e.to_repl(),
2018-08-14 23:19:27 -07:00
Node::PrimObject { name, items, .. } if items.len() == 0 => format!("{}", name),
Node::PrimObject { name, items, .. } => format!("{}{}", name, paren_wrapped_vec(items.iter().map(|x| x.to_repl()))),
2018-08-14 02:03:05 -07:00
Node::PrimTuple { items } => format!("{}", paren_wrapped_vec(items.iter().map(|x| x.to_repl()))),
2018-08-13 23:28:03 -07:00
}
}
fn is_true(&self) -> bool {
match self {
2019-01-07 13:00:37 -08:00
Node::Expr(Expr::Lit(crate::reduced_ast::Lit::Bool(true))) => true,
_ => false,
}
}
2018-08-13 23:28:03 -07:00
}
2018-05-11 01:57:29 -07:00
#[derive(Debug)]
enum ValueEntry {
Binding {
2018-05-11 02:24:38 -07:00
constant: bool,
2018-08-13 23:28:03 -07:00
val: /*FullyEvaluatedExpr*/ Node, //TODO make this use a subtype to represent fully evaluatedness
2018-05-11 01:57:29 -07:00
}
}
type EvalResult<T> = Result<T, String>;
2018-05-11 00:25:43 -07:00
impl Expr {
2018-08-13 23:28:03 -07:00
fn to_node(self) -> Node {
Node::Expr(self)
}
2018-05-11 00:25:43 -07:00
fn to_repl(&self) -> String {
2018-05-11 01:07:18 -07:00
use self::Lit::*;
2018-05-11 23:23:54 -07:00
use self::Func::*;
2018-06-04 18:17:03 -07:00
2018-05-11 01:07:18 -07:00
match self {
Expr::Lit(ref l) => match l {
Nat(n) => format!("{}", n),
Int(i) => format!("{}", i),
Float(f) => format!("{}", f),
Bool(b) => format!("{}", b),
2018-05-12 12:58:57 -07:00
StringLit(s) => format!("\"{}\"", s),
2018-05-11 01:07:18 -07:00
},
2018-05-11 23:23:54 -07:00
Expr::Func(f) => match f {
BuiltIn(name) => format!("<built-in function '{}'>", name),
2018-05-11 23:23:54 -07:00
UserDefined { name: None, .. } => format!("<function>"),
UserDefined { name: Some(name), .. } => format!("<function '{}'>", name),
2018-05-11 23:23:54 -07:00
},
Expr::Constructor { type_name, arity, .. } => {
format!("<constructor for `{}` arity {}>", type_name, arity)
},
2018-08-13 23:28:03 -07:00
Expr::Tuple(exprs) => paren_wrapped_vec(exprs.iter().map(|x| x.to_repl())),
2018-05-11 01:07:18 -07:00
_ => format!("{:?}", self),
}
2018-05-11 00:25:43 -07:00
}
fn replace_conditional_target_sigil(self, replacement: &Expr) -> Expr {
use self::Expr::*;
match self {
ConditionalTargetSigilValue => replacement.clone(),
Unit | Lit(_) | Func(_) | Val(_) | Constructor { .. } |
2019-08-12 11:40:16 -07:00
CaseMatch { .. } | UnimplementedSigilValue | ReductionError(_) => self,
Tuple(exprs) => Tuple(exprs.into_iter().map(|e| e.replace_conditional_target_sigil(replacement)).collect()),
Call { f, args } => {
let new_args = args.into_iter().map(|e| e.replace_conditional_target_sigil(replacement)).collect();
Call { f, args: new_args }
},
Conditional { .. } => panic!("Dunno if I need this, but if so implement"),
Assign { .. } => panic!("I'm pretty sure I don't need this"),
}
}
2018-05-11 00:25:43 -07:00
}
2018-05-09 02:27:57 -07:00
impl<'a> State<'a> {
2018-05-12 02:20:50 -07:00
pub fn evaluate(&mut self, ast: ReducedAST, repl: bool) -> Vec<Result<String, String>> {
2018-05-09 03:38:02 -07:00
let mut acc = vec![];
// handle prebindings
for statement in ast.0.iter() {
self.prebinding(statement);
}
2018-05-09 03:38:02 -07:00
for statement in ast.0 {
2018-05-11 00:25:43 -07:00
match self.statement(statement) {
Ok(Some(ref output)) if repl => acc.push(Ok(output.to_repl())),
Ok(_) => (),
2018-05-09 03:38:02 -07:00
Err(error) => {
2018-05-12 12:37:05 -07:00
acc.push(Err(format!("Runtime error: {}", error)));
2018-05-09 03:38:02 -07:00
return acc;
},
}
}
acc
}
fn prebinding(&mut self, stmt: &Stmt) {
match stmt {
Stmt::PreBinding { name, func } => {
2018-08-13 23:28:03 -07:00
let v_entry = ValueEntry::Binding { constant: true, val: Node::Expr(Expr::Func(func.clone())) };
self.values.insert(name.clone(), v_entry);
},
Stmt::Expr(_expr) => {
//TODO have this support things like nested function defs
},
_ => ()
}
}
2018-08-13 23:28:03 -07:00
fn statement(&mut self, stmt: Stmt) -> EvalResult<Option<Node>> {
2018-05-11 01:07:18 -07:00
match stmt {
2018-05-11 02:24:38 -07:00
Stmt::Binding { name, constant, expr } => {
2018-08-13 23:28:03 -07:00
let val = self.expression(Node::Expr(expr))?;
2018-05-11 02:24:38 -07:00
self.values.insert(name.clone(), ValueEntry::Binding { constant, val });
2018-05-11 01:07:18 -07:00
Ok(None)
},
2018-08-13 23:28:03 -07:00
Stmt::Expr(expr) => Ok(Some(self.expression(expr.to_node())?)),
Stmt::PreBinding {..} | Stmt::Noop => Ok(None),
2018-05-11 01:07:18 -07:00
}
}
2018-08-13 23:28:03 -07:00
fn block(&mut self, stmts: Vec<Stmt>) -> EvalResult<Node> {
2018-05-12 13:51:12 -07:00
let mut ret = None;
for stmt in stmts {
ret = self.statement(stmt)?;
}
2018-08-13 23:28:03 -07:00
Ok(ret.unwrap_or(Node::Expr(Expr::Unit)))
2018-05-12 13:51:12 -07:00
}
2018-08-13 23:28:03 -07:00
fn expression(&mut self, node: Node) -> EvalResult<Node> {
2018-05-11 01:07:18 -07:00
use self::Expr::*;
2018-08-13 23:28:03 -07:00
match node {
2018-08-14 02:03:05 -07:00
t @ Node::PrimTuple { .. } => Ok(t),
2018-08-13 23:28:03 -07:00
obj @ Node::PrimObject { .. } => Ok(obj),
Node::Expr(expr) => match expr {
literal @ Lit(_) => Ok(Node::Expr(literal)),
Call { box f, args } => self.call_expression(f, args),
2018-08-13 23:28:03 -07:00
Val(v) => self.value(v),
2018-08-14 21:17:43 -07:00
Constructor { arity, ref name, tag, .. } if arity == 0 => Ok(Node::PrimObject { name: name.clone(), tag, items: vec![] }),
2018-08-13 23:28:03 -07:00
constructor @ Constructor { .. } => Ok(Node::Expr(constructor)),
func @ Func(_) => Ok(Node::Expr(func)),
Tuple(exprs) => {
2018-08-14 02:03:05 -07:00
let nodes = exprs.into_iter().map(|expr| self.expression(Node::Expr(expr))).collect::<Result<Vec<Node>,_>>()?;
Ok(Node::PrimTuple { items: nodes })
2018-08-13 23:28:03 -07:00
},
Conditional { box cond, then_clause, else_clause } => self.conditional(cond, then_clause, else_clause),
2018-10-18 01:46:30 -07:00
Assign { box val, box expr } => self.assign_expression(val, expr),
2018-08-14 12:43:06 -07:00
Unit => Ok(Node::Expr(Unit)),
CaseMatch { box cond, alternatives } => self.case_match_expression(cond, alternatives),
ConditionalTargetSigilValue => Ok(Node::Expr(ConditionalTargetSigilValue)),
UnimplementedSigilValue => Err(format!("Sigil value eval not implemented")),
2019-08-12 11:40:16 -07:00
ReductionError(err) => Err(format!("Reduction error: {}", err)),
2018-08-13 23:28:03 -07:00
}
2018-05-11 01:07:18 -07:00
}
2018-05-09 02:27:57 -07:00
}
2018-05-11 01:30:02 -07:00
fn call_expression(&mut self, f: Expr, args: Vec<Expr>) -> EvalResult<Node> {
use self::Expr::*;
match self.expression(Node::Expr(f))? {
Node::Expr(Constructor { type_name, name, tag, arity }) => self.apply_data_constructor(type_name, name, tag, arity, args),
Node::Expr(Func(f)) => self.apply_function(f, args),
other => return Err(format!("Tried to call {:?} which is not a function or data constructor", other)),
}
}
2018-10-20 00:55:37 -07:00
fn apply_data_constructor(&mut self, _type_name: Rc<String>, name: Rc<String>, tag: usize, arity: usize, args: Vec<Expr>) -> EvalResult<Node> {
if arity != args.len() {
2019-08-05 01:11:01 -07:00
return Err(format!("Data constructor {} requires {} arg(s)", name, arity));
2018-06-04 18:17:03 -07:00
}
2018-08-13 23:28:03 -07:00
let evaled_args = args.into_iter().map(|expr| self.expression(Node::Expr(expr))).collect::<Result<Vec<Node>,_>>()?;
2018-06-12 19:37:53 -07:00
//let evaled_args = vec![];
2018-08-13 23:28:03 -07:00
Ok(Node::PrimObject {
name: name.clone(),
items: evaled_args,
2018-08-14 02:03:05 -07:00
tag
2018-08-13 23:28:03 -07:00
})
2018-06-04 18:17:03 -07:00
}
2018-08-13 23:28:03 -07:00
fn apply_function(&mut self, f: Func, args: Vec<Expr>) -> EvalResult<Node> {
2018-05-11 01:30:02 -07:00
match f {
2018-08-13 23:28:03 -07:00
Func::BuiltIn(sigil) => Ok(Node::Expr(self.apply_builtin(sigil, args)?)),
2018-05-12 00:59:50 -07:00
Func::UserDefined { params, body, name } => {
if params.len() != args.len() {
2018-05-12 12:37:05 -07:00
return Err(format!("calling a {}-argument function with {} args", params.len(), args.len()))
2018-05-12 00:59:50 -07:00
}
let mut func_state = State {
2018-08-14 21:45:45 -07:00
values: self.values.new_scope(name.map(|n| format!("{}", n))),
symbol_table_handle: self.symbol_table_handle.clone(),
};
2018-05-12 00:59:50 -07:00
for (param, val) in params.into_iter().zip(args.into_iter()) {
2018-08-13 23:28:03 -07:00
let val = func_state.expression(Node::Expr(val))?;
2018-05-12 00:59:50 -07:00
func_state.values.insert(param, ValueEntry::Binding { constant: true, val });
}
// TODO figure out function return semantics
2018-05-12 13:51:12 -07:00
func_state.block(body)
2018-05-11 01:30:02 -07:00
}
}
}
2018-05-11 01:57:29 -07:00
fn apply_builtin(&mut self, name: Rc<String>, args: Vec<Expr>) -> EvalResult<Expr> {
2018-05-11 01:30:02 -07:00
use self::Expr::*;
use self::Lit::*;
2018-08-13 23:28:03 -07:00
let evaled_args: Result<Vec<Expr>, String> = args.into_iter().map(|arg| {
match self.expression(Node::Expr(arg)) {
Ok(Node::Expr(e)) => Ok(e),
2018-08-19 15:06:01 -07:00
Ok(Node::PrimTuple { .. }) => Err(format!("Trying to apply a builtin to a tuple")),
2018-08-13 23:28:03 -07:00
Ok(Node::PrimObject { .. }) => Err(format!("Trying to apply a builtin to a primitive object")),
Err(e) => Err(e)
}
}).collect();
2018-05-11 01:30:02 -07:00
let evaled_args = evaled_args?;
Ok(match (name.as_str(), evaled_args.as_slice()) {
2018-05-11 01:46:42 -07:00
/* binops */
2018-05-11 01:30:02 -07:00
("+", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l + r)),
2018-05-11 01:35:39 -07:00
("++", &[Lit(StringLit(ref s1)), Lit(StringLit(ref s2))]) => Lit(StringLit(Rc::new(format!("{}{}", s1, s2)))),
("-", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l - r)),
("*", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l * r)),
("/", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Float((l as f64)/ (r as f64))),
2019-01-10 00:31:37 -08:00
("quot", &[Lit(Nat(l)), Lit(Nat(r))]) => if r == 0 {
2018-05-12 12:37:05 -07:00
return Err(format!("divide by zero"));
2018-05-11 01:30:02 -07:00
} else {
2018-05-11 01:35:39 -07:00
Lit(Nat(l / r))
2018-05-11 01:30:02 -07:00
},
2018-05-11 01:35:39 -07:00
("%", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l % r)),
("^", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l ^ r)),
("&", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l & r)),
("|", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Nat(l | r)),
2018-05-11 01:46:42 -07:00
/* comparisons */
2018-05-12 13:58:21 -07:00
("==", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Bool(l == r)),
("==", &[Lit(Int(l)), Lit(Int(r))]) => Lit(Bool(l == r)),
("==", &[Lit(Float(l)), Lit(Float(r))]) => Lit(Bool(l == r)),
("==", &[Lit(Bool(l)), Lit(Bool(r))]) => Lit(Bool(l == r)),
("==", &[Lit(StringLit(ref l)), Lit(StringLit(ref r))]) => Lit(Bool(l == r)),
("<", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Bool(l < r)),
("<", &[Lit(Int(l)), Lit(Int(r))]) => Lit(Bool(l < r)),
("<", &[Lit(Float(l)), Lit(Float(r))]) => Lit(Bool(l < r)),
("<=", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Bool(l <= r)),
("<=", &[Lit(Int(l)), Lit(Int(r))]) => Lit(Bool(l <= r)),
("<=", &[Lit(Float(l)), Lit(Float(r))]) => Lit(Bool(l <= r)),
(">", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Bool(l > r)),
(">", &[Lit(Int(l)), Lit(Int(r))]) => Lit(Bool(l > r)),
(">", &[Lit(Float(l)), Lit(Float(r))]) => Lit(Bool(l > r)),
(">=", &[Lit(Nat(l)), Lit(Nat(r))]) => Lit(Bool(l >= r)),
(">=", &[Lit(Int(l)), Lit(Int(r))]) => Lit(Bool(l >= r)),
(">=", &[Lit(Float(l)), Lit(Float(r))]) => Lit(Bool(l >= r)),
2018-05-11 01:46:42 -07:00
/* prefix ops */
("!", &[Lit(Bool(true))]) => Lit(Bool(false)),
("!", &[Lit(Bool(false))]) => Lit(Bool(true)),
("-", &[Lit(Nat(n))]) => Lit(Int(-1*(n as i64))),
("-", &[Lit(Int(n))]) => Lit(Int(-1*(n as i64))),
("+", &[Lit(Int(n))]) => Lit(Int(n)),
("+", &[Lit(Nat(n))]) => Lit(Nat(n)),
2018-05-12 01:18:44 -07:00
2018-05-12 03:51:42 -07:00
/* builtin functions */
2018-05-12 01:18:44 -07:00
("print", &[ref anything]) => {
print!("{}", anything.to_repl());
Expr::Unit
},
("println", &[ref anything]) => {
println!("{}", anything.to_repl());
Expr::Unit
},
2018-05-12 02:27:54 -07:00
("getline", &[]) => {
let mut buf = String::new();
io::stdin().read_line(&mut buf).expect("Error readling line in 'getline'");
Lit(StringLit(Rc::new(buf.trim().to_string())))
2018-05-12 02:27:54 -07:00
},
2018-05-12 12:37:05 -07:00
(x, args) => return Err(format!("bad or unimplemented builtin {:?} | {:?}", x, args)),
2018-05-11 01:30:02 -07:00
})
}
2018-05-11 17:24:11 -07:00
2018-08-13 23:28:03 -07:00
fn conditional(&mut self, cond: Expr, then_clause: Vec<Stmt>, else_clause: Vec<Stmt>) -> EvalResult<Node> {
let cond = self.expression(Node::Expr(cond))?;
2018-05-12 13:51:12 -07:00
Ok(match cond {
2018-08-13 23:28:03 -07:00
Node::Expr(Expr::Lit(Lit::Bool(true))) => self.block(then_clause)?,
Node::Expr(Expr::Lit(Lit::Bool(false))) => self.block(else_clause)?,
2018-05-12 13:51:12 -07:00
_ => return Err(format!("Conditional with non-boolean condition"))
})
}
2018-10-18 01:46:30 -07:00
fn assign_expression(&mut self, val: Expr, expr: Expr) -> EvalResult<Node> {
let name = match val {
Expr::Val(name) => name,
_ => return Err(format!("Trying to assign to a non-value")),
};
let constant = match self.values.lookup(&name) {
None => return Err(format!("Constant {} is undefined", name)),
Some(ValueEntry::Binding { constant, .. }) => constant.clone(),
};
if constant {
return Err(format!("trying to update {}, a non-mutable binding", name));
}
let val = self.expression(Node::Expr(expr))?;
self.values.insert(name.clone(), ValueEntry::Binding { constant: false, val });
Ok(Node::Expr(Expr::Unit))
}
2018-11-05 04:02:04 -08:00
fn guard_passes(&mut self, guard: &Option<Expr>, cond: &Node) -> EvalResult<bool> {
if let Some(ref guard_expr) = guard {
let guard_expr = match cond {
Node::Expr(ref e) => guard_expr.clone().replace_conditional_target_sigil(e),
_ => guard_expr.clone()
};
Ok(self.expression(guard_expr.to_node())?.is_true())
} else {
Ok(true)
}
}
fn case_match_expression(&mut self, cond: Expr, alternatives: Vec<Alternative>) -> EvalResult<Node> {
2018-11-05 04:02:04 -08:00
//TODO need to handle recursive subpatterns
let all_subpatterns_pass = |state: &mut State, subpatterns: &Vec<Option<Subpattern>>, items: &Vec<Node>| -> EvalResult<bool> {
2018-11-05 14:01:14 -08:00
if subpatterns.len() == 0 {
return Ok(true)
}
if items.len() != subpatterns.len() {
return Err(format!("Subpattern length isn't correct items {} subpatterns {}", items.len(), subpatterns.len()));
}
2018-11-05 04:02:04 -08:00
for (maybe_subp, cond) in subpatterns.iter().zip(items.iter()) {
if let Some(subp) = maybe_subp {
if !state.guard_passes(&subp.guard, &cond)? {
return Ok(false)
}
}
}
2018-11-05 04:02:04 -08:00
Ok(true)
};
2018-11-05 04:02:04 -08:00
let cond = self.expression(Node::Expr(cond))?;
for alt in alternatives {
// no matter what type of condition we have, ignore alternative if the guard evaluates false
if !self.guard_passes(&alt.matchable.guard, &cond)? {
2018-11-05 04:02:04 -08:00
continue;
}
match cond {
Node::PrimObject { ref tag, ref items, .. } => {
if alt.matchable.tag.map(|t| t == *tag).unwrap_or(true) {
let mut inner_state = self.new_frame(items, &alt.matchable.bound_vars);
if all_subpatterns_pass(&mut inner_state, &alt.matchable.subpatterns, items)? {
2018-11-05 14:01:14 -08:00
return inner_state.block(alt.item);
} else {
continue;
}
}
},
2018-11-03 12:53:09 -07:00
Node::PrimTuple { ref items } => {
let mut inner_state = self.new_frame(items, &alt.matchable.bound_vars);
if all_subpatterns_pass(&mut inner_state, &alt.matchable.subpatterns, items)? {
2018-11-05 14:01:14 -08:00
return inner_state.block(alt.item);
} else {
continue;
}
},
2018-11-02 19:54:04 -07:00
Node::Expr(ref _e) => {
if let None = alt.matchable.tag {
return self.block(alt.item)
}
}
}
}
Err(format!("{:?} failed pattern match", cond))
}
2019-03-11 19:36:10 -07:00
//TODO if I don't need to lookup by name here...
2018-08-13 23:28:03 -07:00
fn value(&mut self, name: Rc<String>) -> EvalResult<Node> {
2018-05-11 17:24:11 -07:00
use self::ValueEntry::*;
2018-05-14 22:55:53 -07:00
use self::Func::*;
//TODO add a layer of indirection here to talk to the symbol table first, and only then look up
//in the values table
2018-05-14 09:46:25 -07:00
let symbol_table = self.symbol_table_handle.borrow();
2018-08-05 18:19:48 -07:00
let value = symbol_table.lookup_by_name(&name);
2018-06-12 19:37:53 -07:00
Ok(match value {
Some(Symbol { name, spec, .. }) => match spec {
2018-08-14 23:19:27 -07:00
//TODO I'll need this type_name later to do a table lookup
SymbolSpec::DataConstructor { type_name: _type_name, type_args, .. } => {
2018-05-30 23:54:24 -07:00
if type_args.len() == 0 {
2018-08-13 23:28:03 -07:00
Node::PrimObject { name: name.clone(), tag: 0, items: vec![] }
2018-05-30 23:54:24 -07:00
} else {
return Err(format!("This data constructor thing not done"))
}
2018-05-14 22:55:53 -07:00
},
2018-06-03 02:39:49 -07:00
SymbolSpec::Func(_) => match self.values.lookup(&name) {
2018-08-13 23:28:03 -07:00
Some(Binding { val: Node::Expr(Expr::Func(UserDefined { name, params, body })), .. }) => {
Node::Expr(Expr::Func(UserDefined { name: name.clone(), params: params.clone(), body: body.clone() }))
2018-05-14 22:55:53 -07:00
},
_ => unreachable!(),
},
2019-01-25 00:57:01 -08:00
SymbolSpec::RecordConstructor { .. } => return Err(format!("This shouldn't be a record!")),
SymbolSpec::Binding => match self.values.lookup(&name) {
Some(Binding { val, .. }) => val.clone(),
None => return Err(format!("Symbol {} exists in symbol table but not in evaluator table", name))
}
2018-05-14 22:55:53 -07:00
},
//TODO ideally this should be returning a runtime error if this is ever None, but it's not
//handling all bindings correctly yet
//None => return Err(format!("Couldn't find value {}", name)),
2018-05-14 22:55:53 -07:00
None => match self.values.lookup(&name) {
Some(Binding { val, .. }) => val.clone(),
None => return Err(format!("Couldn't find value {}", name)),
2018-05-11 17:24:11 -07:00
}
2018-05-14 22:55:53 -07:00
})
2018-05-11 23:23:54 -07:00
}
2018-05-09 02:27:57 -07:00
}
2018-05-12 02:18:34 -07:00
#[cfg(test)]
mod eval_tests {
2018-05-14 23:25:28 -07:00
use std::cell::RefCell;
use std::rc::Rc;
2018-10-20 14:27:00 -07:00
2019-01-07 13:00:37 -08:00
use crate::symbol_table::SymbolTable;
use crate::eval::State;
2018-05-12 02:18:34 -07:00
fn evaluate_all_outputs(input: &str) -> Vec<Result<String, String>> {
let symbol_table = Rc::new(RefCell::new(SymbolTable::new()));
let mut state = State::new(symbol_table);
2019-03-07 23:51:31 -08:00
let ast = crate::util::quick_ast(input);
state.symbol_table_handle.borrow_mut().add_top_level_symbols(&ast).unwrap();
let reduced = ast.reduce(&state.symbol_table_handle.borrow());
let all_output = state.evaluate(reduced, true);
all_output
2018-08-14 21:39:33 -07:00
}
2018-10-16 04:11:18 -07:00
macro_rules! test_in_fresh_env {
2018-05-12 02:18:34 -07:00
($string:expr, $correct:expr) => {
2018-08-14 21:39:33 -07:00
{
let all_output = evaluate_all_outputs($string);
2018-08-14 21:39:33 -07:00
let ref output = all_output.last().unwrap();
assert_eq!(**output, Ok($correct.to_string()));
}
2018-05-12 02:18:34 -07:00
}
}
#[test]
fn test_basic_eval() {
2018-10-16 04:11:18 -07:00
test_in_fresh_env!("1 + 2", "3");
test_in_fresh_env!("let mut a = 1; a = 2", "Unit");
2019-06-16 16:07:27 -07:00
/*
2018-10-16 04:11:18 -07:00
test_in_fresh_env!("let mut a = 1; a = 2; a", "2");
test_in_fresh_env!(r#"("a", 1 + 2)"#, r#"("a", 3)"#);
2019-06-16 16:07:27 -07:00
*/
2018-05-12 02:18:34 -07:00
}
2018-05-14 23:33:11 -07:00
#[test]
fn function_eval() {
2018-10-16 04:11:18 -07:00
test_in_fresh_env!("fn oi(x) { x + 1 }; oi(4)", "5");
test_in_fresh_env!("fn oi(x) { x + 1 }; oi(1+2)", "4");
}
2018-05-14 23:33:11 -07:00
#[test]
fn scopes() {
let scope_ok = r#"
2018-07-11 16:44:15 -07:00
let a = 20
2018-05-14 23:33:11 -07:00
fn haha() {
2018-07-11 16:44:15 -07:00
let a = 10
2018-05-14 23:33:11 -07:00
a
}
haha()
"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(scope_ok, "10");
2018-05-14 23:33:11 -07:00
let scope_ok = r#"
2018-07-11 16:44:15 -07:00
let a = 20
2018-05-14 23:33:11 -07:00
fn haha() {
2018-07-11 16:44:15 -07:00
let a = 10
2018-05-14 23:33:11 -07:00
a
}
a
"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(scope_ok, "20");
2018-05-14 23:33:11 -07:00
}
2018-08-14 21:39:33 -07:00
#[test]
fn if_is_patterns() {
2018-08-14 21:39:33 -07:00
let source = r#"
type Option<T> = Some(T) | None
let x = Some(9); if x is Some(q) then { q } else { 0 }"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(source, "9");
2018-08-14 21:39:33 -07:00
let source = r#"
type Option<T> = Some(T) | None
let x = None; if x is Some(q) then { q } else { 0 }"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(source, "0");
2018-08-14 21:39:33 -07:00
}
#[test]
fn full_if_matching() {
let source = r#"
type Option<T> = Some(T) | None
let a = None
if a { is None -> 4, is Some(x) -> x }
"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(source, "4");
let source = r#"
type Option<T> = Some(T) | None
let a = Some(99)
if a { is None -> 4, is Some(x) -> x }
"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(source, "99");
2018-10-16 04:10:28 -07:00
let source = r#"
let a = 10
if a { is 10 -> "x", is 4 -> "y" }
"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(source, "\"x\"");
2018-10-16 04:10:28 -07:00
let source = r#"
let a = 10
if a { is 15 -> "x", is 10 -> "y" }
"#;
2018-10-16 04:11:18 -07:00
test_in_fresh_env!(source, "\"y\"");
}
2018-10-17 12:43:09 -07:00
2018-10-31 01:45:16 -07:00
#[test]
fn string_pattern() {
let source = r#"
let a = "foo"
if a { is "foo" -> "x", is _ -> "y" }
"#;
test_in_fresh_env!(source, "\"x\"");
}
2018-10-17 12:43:09 -07:00
#[test]
fn boolean_pattern() {
let source = r#"
let a = true
if a {
is true -> "x",
is false -> "y"
}
2018-10-17 12:43:09 -07:00
"#;
test_in_fresh_env!(source, "\"x\"");
}
#[test]
fn boolean_pattern_2() {
let source = r#"
let a = false
if a { is true -> "x", is false -> "y" }
"#;
test_in_fresh_env!(source, "\"y\"");
}
2018-10-18 15:55:24 -07:00
#[test]
fn ignore_pattern() {
let source = r#"
type Option<T> = Some(T) | None
if Some(10) {
is _ -> "hella"
}
"#;
test_in_fresh_env!(source, "\"hella\"");
}
#[test]
fn tuple_pattern() {
let source = r#"
if (1, 2) {
is (1, x) -> x,
is _ -> 99
}
"#;
test_in_fresh_env!(source, 2);
}
#[test]
fn tuple_pattern_2() {
let source = r#"
if (1, 2) {
is (10, x) -> x,
is (y, x) -> x + y
}
"#;
test_in_fresh_env!(source, 3);
}
#[test]
fn tuple_pattern_3() {
let source = r#"
if (1, 5) {
is (10, x) -> x,
is (1, x) -> x
}
"#;
test_in_fresh_env!(source, 5);
}
#[test]
fn tuple_pattern_4() {
let source = r#"
if (1, 5) {
is (10, x) -> x,
is (1, x) -> x,
}
"#;
test_in_fresh_env!(source, 5);
}
2018-11-05 14:11:49 -08:00
#[test]
fn prim_obj_pattern() {
let source = r#"
type Stuff = Mulch(Nat) | Jugs(Nat, String) | Mardok
let a = Mulch(20)
let b = Jugs(1, "haha")
let c = Mardok
let x = if a {
is Mulch(20) -> "x",
is _ -> "ERR"
}
let y = if b {
is Mulch(n) -> "ERR",
is Jugs(2, _) -> "ERR",
is Jugs(1, s) -> s,
is _ -> "ERR",
}
let z = if c {
is Jugs(_, _) -> "ERR",
is Mardok -> "NIGH",
is _ -> "ERR",
}
(x, y, z)
"#;
test_in_fresh_env!(source, r#"("x", "haha", "NIGH")"#);
}
2018-11-06 01:19:16 -08:00
#[test]
fn basic_lambda_syntax() {
let source = r#"
let q = \(x, y) { x * y }
let x = q(5,2)
let y = \(m, n, o) { m + n + o }(1,2,3)
(x, y)
"#;
test_in_fresh_env!(source, r"(10, 6)");
}
2018-11-06 02:42:28 -08:00
#[test]
fn lambda_syntax_2() {
let source = r#"
fn milta() {
\(x) { x + 33 }
}
milta()(10)
"#;
test_in_fresh_env!(source, "43");
}
2018-05-12 02:18:34 -07:00
}