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 55629e6d9d
commit d5df868f10
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> {
use self::parsing::Expression;
use self::parsing::ExpressionType::*;
use self::Type::*; use self::TConst::*;
match expr {
&Expression(ref e, Some(ref anno)) => {
let anno_ty = self.type_from_anno(anno)?;
let expr = Expression(*e, None);
let ty = self.type_infer(&expr)?;
let ty = self.type_infer_exprtype(&e)?;
self.unify(ty, anno_ty)
},
&Expression(ref e, None) => match e {
&IntLiteral(_) => Ok(Const(Int)),
&FloatLiteral(_) => Ok(Const(Float)),
&StringLiteral(_) => Ok(Const(StringT)),
&BoolLiteral(_) => Ok(Const(Bool)),
_ => Err(format!("Type not yet implemented"))
}
&Expression(ref e, None) => self.type_infer_exprtype(e)
}
}
fn type_infer_exprtype(&mut self, expr: &parsing::ExpressionType) -> TypeResult<Type> {
use self::parsing::ExpressionType::*;
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> {