Finished initial BinOp/PrefixOp
This commit is contained in:
parent
0dd8861f83
commit
d229a57837
40
src/schala_lang/builtin.rs
Normal file
40
src/schala_lang/builtin.rs
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
use std::rc::Rc;
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
|
pub struct BinOp {
|
||||||
|
pub sigil: Rc<String>
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
|
pub struct PrefixOp {
|
||||||
|
pub sigil: Rc<String>
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BinOp {
|
||||||
|
pub fn from_sigil(sigil: Rc<String>) -> BinOp {
|
||||||
|
BinOp { sigil }
|
||||||
|
}
|
||||||
|
pub fn min_precedence() -> i32 {
|
||||||
|
i32::min_value()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_precedence(op: &str) -> i32 {
|
||||||
|
match op {
|
||||||
|
"+" | "-" => 10,
|
||||||
|
"*" | "/" | "%" => 20,
|
||||||
|
_ => 30,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PrefixOp {
|
||||||
|
pub fn from_sigil(sigil: Rc<String>) -> PrefixOp {
|
||||||
|
PrefixOp { sigil }
|
||||||
|
}
|
||||||
|
pub fn is_prefix(op: &str) -> bool {
|
||||||
|
match op {
|
||||||
|
"+" | "-" | "!" | "~" => true,
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -1,4 +1,5 @@
|
|||||||
use schala_lang::parsing::{AST, Statement, Declaration, Expression, Variant, ExpressionType, Operation};
|
use schala_lang::parsing::{AST, Statement, Declaration, Expression, Variant, ExpressionType};
|
||||||
|
use schala_lang::builtin::{BinOp, PrefixOp};
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
|
|
||||||
@ -162,12 +163,12 @@ impl State {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn eval_binexp(&mut self, op: Operation, lhs: Box<Expression>, rhs: Box<Expression>) -> EvalResult<FullyEvaluatedExpr> {
|
fn eval_binexp(&mut self, op: BinOp, lhs: Box<Expression>, rhs: Box<Expression>) -> EvalResult<FullyEvaluatedExpr> {
|
||||||
use self::FullyEvaluatedExpr::*;
|
use self::FullyEvaluatedExpr::*;
|
||||||
let evaled_lhs = self.eval_expr(*lhs)?;
|
let evaled_lhs = self.eval_expr(*lhs)?;
|
||||||
let evaled_rhs = self.eval_expr(*rhs)?;
|
let evaled_rhs = self.eval_expr(*rhs)?;
|
||||||
let opstr: &str = &op.0;
|
let sigil: &str = op.sigil.as_ref().as_str();
|
||||||
Ok(match (opstr, evaled_lhs, evaled_rhs) {
|
Ok(match (sigil, evaled_lhs, evaled_rhs) {
|
||||||
("+", UnsignedInt(l), UnsignedInt(r)) => UnsignedInt(l + r),
|
("+", UnsignedInt(l), UnsignedInt(r)) => UnsignedInt(l + r),
|
||||||
("++", Str(s1), Str(s2)) => Str(format!("{}{}", s1, s2)),
|
("++", Str(s1), Str(s2)) => Str(format!("{}{}", s1, s2)),
|
||||||
("-", UnsignedInt(l), UnsignedInt(r)) => UnsignedInt(l - r),
|
("-", UnsignedInt(l), UnsignedInt(r)) => UnsignedInt(l - r),
|
||||||
@ -178,12 +179,13 @@ impl State {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn eval_prefix_exp(&mut self, op: Operation, expr: Box<Expression>) -> EvalResult<FullyEvaluatedExpr> {
|
fn eval_prefix_exp(&mut self, op: PrefixOp, expr: Box<Expression>) -> EvalResult<FullyEvaluatedExpr> {
|
||||||
use self::FullyEvaluatedExpr::*;
|
use self::FullyEvaluatedExpr::*;
|
||||||
let evaled_expr = self.eval_expr(*expr)?;
|
let evaled_expr = self.eval_expr(*expr)?;
|
||||||
let opstr: &str = &op.0;
|
|
||||||
|
|
||||||
Ok(match (opstr, evaled_expr) {
|
let sigil: &str = op.sigil.as_ref().as_str();
|
||||||
|
|
||||||
|
Ok(match (sigil, evaled_expr) {
|
||||||
("!", Bool(true)) => Bool(false),
|
("!", Bool(true)) => Bool(false),
|
||||||
("!", Bool(false)) => Bool(true),
|
("!", Bool(false)) => Bool(true),
|
||||||
("-", UnsignedInt(n)) => SignedInt(-1*(n as i64)),
|
("-", UnsignedInt(n)) => SignedInt(-1*(n as i64)),
|
||||||
|
@ -2,6 +2,7 @@ use std::rc::Rc;
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use schala_lang::parsing;
|
use schala_lang::parsing;
|
||||||
|
use schala_lang::builtin;
|
||||||
|
|
||||||
pub struct TypeContext {
|
pub struct TypeContext {
|
||||||
bindings: HashMap<Rc<String>, Type>
|
bindings: HashMap<Rc<String>, Type>
|
||||||
@ -129,7 +130,7 @@ impl TypeContext {
|
|||||||
_ => Err(format!("Type not yet implemented"))
|
_ => Err(format!("Type not yet implemented"))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn infer_optype(&mut self, _op: &parsing::Operation) -> TypeResult<Type> {
|
fn infer_optype(&mut self, _op: &builtin::BinOp) -> TypeResult<Type> {
|
||||||
use self::Type::*; use self::TConst::*;
|
use self::Type::*; use self::TConst::*;
|
||||||
//this is a shim; not all ops are binops from int -> int -> int
|
//this is a shim; not all ops are binops from int -> int -> int
|
||||||
Ok(Func(bx!(Const(Int)), bx!(Func(bx!(Const(Int)), bx!(Const(Int))))))
|
Ok(Func(bx!(Const(Int)), bx!(Func(bx!(Const(Int)), bx!(Const(Int))))))
|
||||||
|
Loading…
Reference in New Issue
Block a user