greg
/
schala
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Finished basic constant type inference

This commit is contained in:
greg 2018-02-21 14:14:24 -08:00
parent 687d482853
commit bb0fb716e4
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/schala_lang/typechecking.rs
View File

@ -50,22 +50,24 @@ impl TypeContext {
} }
fn type_infer(&mut self, expr: &parsing::Expression) -> TypeResult<Type> { fn type_infer(&mut self, expr: &parsing::Expression) -> TypeResult<Type> {
use self::parsing::Expression; use self::parsing::Expression;
use self::parsing::ExpressionType::*;
use self::Type::*; use self::TConst::*;
match expr { match expr {
&Expression(ref e, Some(ref anno)) => { &Expression(ref e, Some(ref anno)) => {
let anno_ty = self.type_from_anno(anno)?; let anno_ty = self.type_from_anno(anno)?;
let expr = Expression(*e, None); let ty = self.type_infer_exprtype(&e)?;
let ty = self.type_infer(&expr)?;
self.unify(ty, anno_ty) self.unify(ty, anno_ty)
}, },
&Expression(ref e, None) => match e { &Expression(ref e, None) => self.type_infer_exprtype(e)
&IntLiteral(_) => Ok(Const(Int)), }
&FloatLiteral(_) => Ok(Const(Float)), }
&StringLiteral(_) => Ok(Const(StringT)), fn type_infer_exprtype(&mut self, expr: &parsing::ExpressionType) -> TypeResult<Type> {
&BoolLiteral(_) => Ok(Const(Bool)), use self::parsing::ExpressionType::*;
_ => Err(format!("Type not yet implemented")) use self::Type::*; use self::TConst::*;
} match expr {
&IntLiteral(_) => Ok(Const(Int)),
&FloatLiteral(_) => Ok(Const(Float)),
&StringLiteral(_) => Ok(Const(StringT)),
&BoolLiteral(_) => Ok(Const(Bool)),
_ => Err(format!("Type not yet implemented"))
} }
} }
fn type_from_anno(&mut self, anno: &parsing::TypeName) -> TypeResult<Type> { fn type_from_anno(&mut self, anno: &parsing::TypeName) -> TypeResult<Type> {