wip typing

lib-ltcore/Cargo.toml

@@ -4,7 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-laddertypes = { path = "../../lib-laddertypes" }
+laddertypes = { path = "../../lib-laddertypes", features = ["pretty"] }
 tisc = { path = "../../lib-tisc" }
 serde = { version = "1.0", features = ["derive"] }
-
+tiny-ansi = "0.1.0"

lib-ltcore/src/expr.rs

@@ -4,11 +4,13 @@ use {
         sync::{Arc, RwLock}
     },
     crate::{
-        lexer::InputRegionTag
-    }
+        lexer::InputRegionTag,
+        symbols::Scope
+    },
+    tiny_ansi::TinyAnsi
 };
 
-#[derive(Clone, Debug, PartialEq)]
+#[derive(Clone, Debug)]
 pub enum Statement {
     Assignment {
         name_region: InputRegionTag,
@@ -16,33 +18,75 @@ pub enum Statement {
         val_expr: LTExpr,
     },
     LetAssign {
-        typ: Option<TypeTag>,
+        name_region: InputRegionTag,
+        typ: Option<laddertypes::TypeTerm>,
         var_id: String,
         val_expr: LTExpr,
     },
-    WhileLoop {
-        condition: LTExpr,
-        body: Vec<Statement>,
-    },
     Return(LTExpr),
     Expr(LTExpr),
 }
 
 #[derive(Clone, Debug, PartialEq)]
-pub enum TypeError {
-    ParseError(laddertypes::parser::ParseError),
-    Mismatch {
+pub enum TypeErrorKind {
+//    ParseError(laddertypes::parser::ParseError),
+    AssignMismatch {
         expected: laddertypes::TypeTerm,
         received: laddertypes::TypeTerm,
     },
+    ArgTypeMismatch {
+        expected: laddertypes::TypeTerm,
+        received: laddertypes::TypeTerm,
+    },
+    BranchMismatch {
+        if_branch: laddertypes::TypeTerm,
+        else_branch: laddertypes::TypeTerm
+    },
+    SuperfluousArgument,
     NoSymbol,
-    SuperflousArgument,
     Todo
 }
 
-pub type TypeTag = Result<laddertypes::TypeTerm, TypeError>;
+#[derive(Clone, Debug)]
+pub struct TypeError {
+    pub region: InputRegionTag,
+    pub kind: TypeErrorKind
+}
 
-#[derive(Clone, Debug, PartialEq)]
+impl TypeErrorKind {
+    pub fn fmt(&self, dict: &mut impl laddertypes::TypeDict) -> String {
+        match self {
+            TypeErrorKind::BranchMismatch { if_branch, else_branch } => {
+                format!("Type Mismatch\nif branch\n:::{}\nelse branch\n:::{}",
+                    if_branch.clone().sugar(dict).pretty(dict, 1),
+                    else_branch.clone().sugar(dict).pretty(dict, 1)
+                )                
+            },
+            TypeErrorKind::AssignMismatch { expected, received } |
+            TypeErrorKind::ArgTypeMismatch { expected, received } => {
+                format!("Type Mismatch\n{}{}\n{}{}",
+                    "expected\n ::: ".green(),
+                    expected.clone().sugar(dict).pretty(dict, 1),
+                    "received\n ::: ".green(),
+                    received.clone().sugar(dict).pretty(dict, 1)
+                )
+            }
+            TypeErrorKind::SuperfluousArgument => {
+                format!("Superfluous Argument")
+            }
+            TypeErrorKind::NoSymbol => {
+                format!("Unknown Symbol")
+            }
+            TypeErrorKind::Todo => {
+                format!("TODO")
+            }
+        }
+    }
+}
+
+pub type TypeTag = Result< laddertypes::TypeTerm, Vec<TypeError> >;
+
+#[derive(Clone, Debug)]
 pub enum LTExpr {
     WordLiteral {
         region: InputRegionTag,
@@ -59,12 +103,12 @@ pub enum LTExpr {
     },
     Ascend {
         region: InputRegionTag,
-        typ: TypeTag,
+        typ: laddertypes::TypeTerm,
         expr: Box<LTExpr>
     },
     Descend {
         region: InputRegionTag,
-        typ: TypeTag,
+        typ: laddertypes::TypeTerm,
         expr: Box<LTExpr>
     },
     Application {
@@ -75,6 +119,7 @@ pub enum LTExpr {
     },
     Abstraction {
         region: InputRegionTag,
+        scope: Arc<RwLock<Scope>>,
         args: Vec<(InputRegionTag, String, Option<TypeTag>)>,
         body: Box<LTExpr>,
     },
@@ -84,58 +129,86 @@ pub enum LTExpr {
         if_expr: Box<LTExpr>,
         else_expr: Box<LTExpr>,
     },
+    WhileLoop {
+        region: InputRegionTag,
+        condition: Box<LTExpr>,
+        body: Box<LTExpr>,
+    },
     Block {
         region: InputRegionTag,
+        scope: Arc<RwLock<Scope>>,
         statements: Vec<Statement>,
     },
     ExportBlock {
         region: InputRegionTag,
+        scope: Arc<RwLock<Scope>>,
         statements: Vec<Statement>,
     }
 }
 
-impl LTExpr {   
+impl LTExpr {
     pub fn get_region(&self) -> InputRegionTag {
         match self {
-            LTExpr::WordLiteral { region, val } => region,
-            LTExpr::StringLiteral { region, value } => region,
-            LTExpr::Symbol { region, typ, symbol } => region,
-            LTExpr::Ascend { region, typ, expr } => region,
+            LTExpr::WordLiteral{ region, val } => region,
+            LTExpr::StringLiteral{ region, value } => region,
+            LTExpr::Symbol{ region, typ, symbol } => region,
+            LTExpr::Ascend{ region, typ, expr } => region,
             LTExpr::Descend{ region, typ, expr } => region,
             LTExpr::Application{ region, typ, head, body } => region,
-            LTExpr::Abstraction{ region, args, body } => region,
+            LTExpr::Abstraction{ region, scope, args, body } => region,
             LTExpr::Branch{ region, condition, if_expr, else_expr } => region,
-            LTExpr::Block{ region, statements } => region,
-            LTExpr::ExportBlock{ region, statements } => region
+            LTExpr::WhileLoop{ region, condition, body } => region,
+            LTExpr::Block{ region, scope, statements } => region,
+            LTExpr::ExportBlock{ region, scope, statements } => region
         }.clone()
     }
 
-    pub fn lit_uint(val: u64) -> Self {
-        LTExpr::WordLiteral {
-            region: InputRegionTag::default(),
-            val: val as tisc::VM_Word,
-        }
-    }
-
-    pub fn application(head: LTExpr, body: Vec<LTExpr>) -> Self {
-        LTExpr::Application {
-            region: InputRegionTag::default(),
-            typ: None,
-            head: Box::new(head),
-            body: body,
-        }
-    }
-
-    pub fn block(body: Vec<Statement>) -> Self {
-        LTExpr::Block { region: InputRegionTag::default(), statements: body }
-    }
-}
-
-impl Statement {
-    pub fn while_loop(cond: LTExpr, body: Vec<Statement>) -> Self {
-        Statement::WhileLoop {
-            condition: cond,
-            body,
+    pub fn get(&self, addr: Vec<usize>) -> Option<LTExpr> {
+        if addr.len() == 0 {
+            Some(self.clone())
+        } else {
+            let mut sub_addr = addr.clone();
+            let top_idx = sub_addr.remove(0);
+            match self {
+                LTExpr::Ascend{ region, typ, expr } => expr.get(addr),
+                LTExpr::Descend{ region, typ, expr } => expr.get(addr),
+                LTExpr::Application{ region, typ, head, body } => {
+                    match top_idx {
+                        0 => head.get(sub_addr),
+                        i => {
+                            if let Some(b) = body.get(i - 1) {
+                                b.get(sub_addr)
+                            } else {
+                                None
+                            }
+                        }
+                    }
+                }
+                LTExpr::Abstraction{ region, scope, args, body } => {
+                    body.get(addr)
+                }
+                LTExpr::Branch{ region, condition, if_expr, else_expr } => {
+                    match top_idx {
+                        0 => condition.get(sub_addr),
+                        1 => if_expr.get(sub_addr),
+                        2 => else_expr.get(sub_addr),
+                        _ => None
+                    }
+                }
+                LTExpr::WhileLoop{ region, condition, body } => {
+                    match top_idx {
+                        0 => condition.get(sub_addr),
+                        1 => body.get(sub_addr),
+                        _ => None
+                    }
+                }
+                LTExpr::Block{ region, scope, statements } |
+                LTExpr::ExportBlock{ region, scope, statements } => {
+//                    statements.get(top_idx)?.get(sub_addr)
+                    None
+                }
+                _ => None
+            }
         }
     }
 }

lib-ltcore/src/lib.rs

@@ -2,6 +2,7 @@
 pub mod expr;
 pub mod lexer;
 pub mod parser;
+pub mod typing;
 pub mod procedure_compiler;
 pub mod runtime;
 pub mod symbols;

lib-ltcore/src/parser.rs

@@ -2,6 +2,12 @@ use {
     crate::{
         expr::{LTExpr, Statement, TypeError, TypeTag},
         lexer::{LTIRLexer, LTIRToken, LexError, InputRegionTag},
+        symbols::{Scope}
+    },
+    laddertypes::{
+        dict::TypeDict,
+        parser::ParseLadderType,
+        unparser::UnparseLadderType
     },
     std::{
         iter::Peekable,
@@ -54,7 +60,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
  *  `: T`
  */
 pub fn parse_type_tag<It>(
-    typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+    typectx: &mut impl TypeDict,
     tokens: &mut Peekable<It>,
 ) -> Result<Option<(InputRegionTag, laddertypes::TypeTerm)>, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
@@ -64,7 +70,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
         match peektok {
             Ok(LTIRToken::AssignType(typeterm_str)) => {
                 tokens.next();
-                match typectx.write().unwrap().parse(typeterm_str.as_str()) {
+                match typectx.parse(typeterm_str.as_str()) {
                     Ok(typeterm) => Ok(Some((region, typeterm))),
                     Err(parse_error) => Err((region, ParseError::TypeParseError(parse_error))),
                 }
@@ -109,7 +115,7 @@ impl VariableBinding {
  * or  `x : T`
  */
 pub fn parse_binding_expr<It>(
-    typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+    typectx: &mut impl TypeDict,
     tokens: &mut Peekable<It>,
 ) -> Result< VariableBinding, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
@@ -142,7 +148,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
  * `{ x:T; y:U; ... }`
  */
 pub fn parse_binding_block<It>(
-    typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+    typectx: &mut impl TypeDict,
     tokens: &mut Peekable<It>,
 ) -> Result< Vec<VariableBinding>, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
@@ -175,7 +181,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
 }
 
 pub fn parse_statement<It>(
-    typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+    super_scope: &Arc<RwLock<Scope>>,
     tokens: &mut Peekable<It>,
 ) -> Result<crate::expr::Statement, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
@@ -188,7 +194,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                         tokens.next();
                         // todo accept address-expression instead of symbol
                         let (name_region, name) = parse_symbol(tokens)?;
-                        let val_expr = parse_expr(typectx, tokens)?;
+                        let val_expr = parse_expr(super_scope, tokens)?;
                         let _ = parse_expect(tokens, LTIRToken::StatementSep)?;
 
                         Ok(Statement::Assignment {
@@ -200,48 +206,40 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                     "let" => {
                         tokens.next();
                         let (name_region, name) = parse_symbol(tokens)?;
-                        let typ = parse_type_tag(typectx, tokens)?;
+                        let typ = parse_type_tag(&mut *super_scope.write().unwrap(), tokens)?;
+
                         /* todo
                         let mut variable_bindings = parse_binding_expr(typectx, tokens)?;
                         */
                         let _ = parse_expect(tokens, LTIRToken::AssignValue);
-                        let val_expr = parse_expr(typectx, tokens)?;
+                        let val_expr = parse_expr(super_scope, tokens)?;
                         let _ = parse_expect(tokens, LTIRToken::StatementSep)?;
 
                         Ok(Statement::LetAssign {
+                            name_region,
                             typ: match typ {
-                                Some((r,t)) => Some(Ok(t)),
+                                Some((r,t)) => Some(t),
                                 None => None
                             },
                             var_id: name,
                             val_expr,
                         })
                     }
-                    "while" => {
-                        tokens.next();
-                        let _ = parse_expect(tokens, LTIRToken::ExprOpen)?;
-                        let cond = parse_expr(typectx, tokens)?;
-                        let _ = parse_expect(tokens, LTIRToken::ExprClose)?;
-                        Ok(Statement::WhileLoop {
-                            condition: cond,
-                            body: parse_statement_block(typectx, tokens)?,
-                        })
-                    }
                     "return" => {
                         tokens.next();
-                        let expr = parse_expr(typectx, tokens)?;
+                        let expr = parse_expr(super_scope, tokens)?;
                         let _ = parse_expect(tokens, LTIRToken::StatementSep)?;
-                        Ok(Statement::Return(parse_expr(typectx, tokens)?))
+                        Ok(Statement::Return(parse_expr(super_scope, tokens)?))
                     }
                     _ => {
-                        let expr = parse_expr(typectx, tokens)?;
+                        let expr = parse_expr(super_scope, tokens)?;
                         let _ = parse_expect(tokens, LTIRToken::StatementSep)?;
                         Ok(Statement::Expr(expr))
                     }
                 }
             }
             Ok(_) => {
-                let expr = parse_expr(typectx, tokens)?;
+                let expr = parse_expr(super_scope, tokens)?;
                 let _ = parse_expect(tokens, LTIRToken::StatementSep)?;
                 Ok(Statement::Expr(expr))
             }
@@ -253,7 +251,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
 }
 
 pub fn parse_statement_block<It>(
-    typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+    scope: &Arc<RwLock<Scope>>,
     tokens: &mut Peekable<It>,
 ) -> Result<Vec<Statement>, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
@@ -268,7 +266,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                 return Ok(statements);
             }
             Ok(_) => {
-                statements.push(parse_statement(typectx, tokens)?);
+                statements.push(parse_statement(scope, tokens)?);
             }
             Err(err) => {
                 return Err((*region, ParseError::LexError(err.clone())));
@@ -280,14 +278,21 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
 }
 
 pub fn parse_atom<It>(
+    typectx: &mut impl TypeDict,
     tokens: &mut Peekable<It>,
 ) -> Result<crate::expr::LTExpr, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
 {
     match tokens.next() {
         Some((region, Ok(LTIRToken::Symbol(sym)))) => Ok(LTExpr::Symbol{ region, symbol: sym, typ: None }),
-        Some((region, Ok(LTIRToken::Char(c)))) => Ok(LTExpr::lit_uint(c as u64)),
-        Some((region, Ok(LTIRToken::Num(n)))) => Ok(LTExpr::lit_uint(n as u64)),
+        Some((region, Ok(LTIRToken::Char(c)))) => Ok(
+            LTExpr::Ascend {
+                region: region.clone(),
+                typ: typectx.parse("Char ~ Unicode ~ ℤ_2^32").unwrap(),
+                expr: Box::new(LTExpr::WordLiteral{ region, val: c as tisc::VM_Word })
+            }
+        ),
+        Some((region, Ok(LTIRToken::Num(n)))) => Ok(LTExpr::WordLiteral{ region, val: n as tisc::VM_Word }),
         Some((region, Ok(_))) => Err((region, ParseError::UnexpectedToken)),
         Some((region, Err(err))) => Err((region, ParseError::LexError(err))),
         None => Err((InputRegionTag::default(), ParseError::UnexpectedEnd)),
@@ -295,7 +300,7 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
 }
 
 pub fn parse_expr<It>(
-    typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+    super_scope: &Arc<RwLock<Scope>>,
     tokens: &mut Peekable<It>,
 ) -> Result<crate::expr::LTExpr, (InputRegionTag, ParseError)>
 where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
@@ -309,12 +314,15 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                     let region = region.clone();
                     tokens.next();
 
-                    let mut variable_bindings = parse_binding_expr(typectx, tokens)?;
+                    let scope = Scope::with_parent(super_scope);
+
+                    let mut variable_bindings = parse_binding_expr(&mut *scope.write().unwrap(), tokens)?;
                     let _ = parse_expect(tokens, LTIRToken::MapsTo);
-                    let body = parse_expr(typectx, tokens)?;
+                    let body = parse_expr(&scope, tokens)?;
 
                     return Ok(LTExpr::Abstraction {
                         region,
+                        scope,
                         args: variable_bindings.flatten().into_iter().map(|(r,s,t)| (r,s,t.map(|t|Ok(t))) ).collect(),
                         body: Box::new(body),
                     });
@@ -332,14 +340,23 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                         }
                         _ => {}
                     }
-                    children.push(parse_expr(typectx, tokens)?);
+                    children.push(parse_expr(super_scope, tokens)?);
                 }
             }
             Ok(LTIRToken::ExprClose) => {
                 break;
             }
             Ok(LTIRToken::BlockOpen) => {
-                children.push(LTExpr::block(parse_statement_block(typectx, tokens)?));
+                let region = region.clone();
+                let scope = Scope::with_parent(super_scope);
+                let statements = parse_statement_block(&scope, tokens)?;
+
+                children.push(
+                    LTExpr::Block {
+                        region,
+                        scope,
+                        statements
+                    });
             }
             Ok(LTIRToken::BlockClose) => {
                 break;
@@ -354,37 +371,45 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                 children.push(LTExpr::StringLiteral{ region, value });
             }
             Ok(LTIRToken::Ascend(type_str)) => {
-                let region = region.clone();
-                let typ =
-                    match typectx.write().unwrap().parse(type_str) {
-                        Ok(t) => Ok(t),
-                        Err(e) => Err(TypeError::ParseError(e))
-                    };
+                let mut region = region.clone();
+                let typ = super_scope.write().unwrap().parse(type_str);
 
                 if let Some(expr) = children.pop() {
-                    children.push(LTExpr::Ascend {
-                        region: region.clone(),
-                        typ,
-                        expr: Box::new(expr)
-                    });
+                    region.begin = expr.get_region().begin;
+
+                    match typ {
+                        Ok(typ) => {
+                            children.push(LTExpr::Ascend {
+                                region: region.clone(),
+                                typ,
+                                expr: Box::new(expr)
+                            });
+                        },
+                        Err(e) => {
+                            return Err((region, ParseError::TypeParseError(e)));
+                        }
+                    }
                 }
 
                 tokens.next();
             }
             Ok(LTIRToken::Descend(type_str)) => {
                 let region = region.clone();
-                let typ =
-                    match typectx.write().unwrap().parse(type_str) {
-                        Ok(t) => Ok(t),
-                        Err(e) => Err(TypeError::ParseError(e))
-                    };
+                let typ = super_scope.write().unwrap().parse(type_str);
 
                 if let Some(expr) = children.pop() {
-                    children.push(LTExpr::Descend {
-                        region,
-                        typ,
-                        expr: Box::new(expr)
-                    });
+                    match typ {
+                        Ok(typ) => {
+                            children.push(LTExpr::Descend {
+                                region,
+                                typ,
+                                expr: Box::new(expr)
+                            });
+                        }
+                        Err(e) => {
+                            return Err((region, ParseError::TypeParseError(e)));
+                        }
+                    }
                 }
 
                 tokens.next();
@@ -394,16 +419,21 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                     let region = region.clone();
                     tokens.next();
                     let _ = parse_expect(tokens, LTIRToken::ExprOpen)?;
-                    let cond = parse_expr(typectx, tokens)?;
+                    let cond = parse_expr(super_scope, tokens)?;
                     let _ = parse_expect(tokens, LTIRToken::ExprClose)?;
-                    let if_expr = LTExpr::block(parse_statement_block(typectx, tokens)?);
-                    let mut else_expr = LTExpr::block(vec![]);
+
+                    let if_statements = parse_statement_block(super_scope, tokens)?;
+
+                    let scope = super_scope.clone();
+                    let if_expr = LTExpr::Block{ region: region.clone(), scope, statements: if_statements };
+                    let scope = super_scope.clone();
+                    let mut else_expr = LTExpr::Block{ region: InputRegionTag::default(), scope, statements: vec![] };
 
                     if let Some((region, peektok)) = tokens.peek() {
                         if let Ok(LTIRToken::Symbol(name)) = peektok {
                             if name == "else" {
                                 tokens.next();
-                                else_expr = parse_expr(typectx, tokens)?;
+                                else_expr = parse_expr(super_scope, tokens)?;
                             }
                         }
                     }
@@ -415,21 +445,35 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
                         else_expr: Box::new(else_expr),
                     });
                 },
+                "while" => {
+                    let region = region.clone();
+                    tokens.next();
+                    let _ = parse_expect(tokens, LTIRToken::ExprOpen)?;
+                    let cond = parse_expr(super_scope, tokens)?;
+                    let _ = parse_expect(tokens, LTIRToken::ExprClose)?;
+                    children.push(LTExpr::WhileLoop {
+                        region,
+                        condition: Box::new(cond),
+                        body: Box::new(parse_expr(super_scope, tokens)?),
+                    });
+                }
                 "export" => {
                     let region = region.clone();
                     tokens.next();
-                    let block = parse_statement_block(typectx, tokens)?;
+                    let scope = Scope::with_parent(super_scope);
+                    let block = parse_statement_block(&scope, tokens)?;
                     children.push(LTExpr::ExportBlock {
                         region,
+                        scope,
                         statements: block
                     });
                 },
                 name => {
-                    children.push(parse_atom(tokens)?);
+                    children.push(parse_atom(&mut *super_scope.write().unwrap(), tokens)?);
                 }
             },
             Ok(atom) => {
-                children.push(parse_atom(tokens)?);
+                children.push(parse_atom(&mut *super_scope.write().unwrap(), tokens)?);
             }
             Err(err) => {
                 return Err((*region, ParseError::LexError(err.clone())));
@@ -439,8 +483,16 @@ where It: Iterator<Item = (InputRegionTag, Result<LTIRToken, LexError>)>
 
     if children.len() > 1 {
         let head = children.remove(0);
+
+        let mut region = head.get_region();
+        for c in children.iter() {
+            let cr = c.get_region();
+            region.begin = usize::min( region.begin, cr.begin );
+            region.end   = usize::max( region.end, cr.end );
+        }
+        
         Ok(LTExpr::Application {
-            region: InputRegionTag::default(),
+            region,
             typ: None,
             head: Box::new(head),
             body: children,

lib-ltcore/src/procedure_compiler.rs

@@ -8,107 +8,90 @@ use {
         ops::Deref,
         sync::{Arc, RwLock},
     },
+    laddertypes::{
+        parser::ParseLadderType,
+        unparser::UnparseLadderType
+    },
     tisc::{assembler::AssemblyWord, linker::LinkAddr},
+    tiny_ansi::TinyAnsi
 };
 
 pub struct ProcedureCompiler {
-    pub symbols: Arc<RwLock<Scope>>,
+    proc_symbol: String,
+    scope: Arc<RwLock<Scope>>,
     asm: tisc::Assembler,
-    linker: tisc::Linker,
-    result_size: usize,
-
+    subroutines: Vec<tisc::assembler::AssemblyWord>,
+    pub linker: tisc::Linker,
     pub diagnostics: Vec<( InputRegionTag, String )>
 }
 
 impl ProcedureCompiler {
-    pub fn new(parent_scope: &Arc<RwLock<Scope>>) -> Self {
+    pub fn new(proc_symbol: String, scope: Arc<RwLock<Scope>>) -> Self {
         ProcedureCompiler {
-            symbols: Scope::with_parent(parent_scope),
+            proc_symbol,
+            scope,
+            subroutines: Vec::new(),
             asm: tisc::Assembler::new(),
             linker: tisc::Linker::new(),
-            result_size: 0,
-
             diagnostics: Vec::new()
         }
     }
+/*
+    pub fn export_symbols(&self) -> Vec<(String, SymbolDef)> {
+        let mut scope = self.scope.write().unwrap();
+        scope.update_link_addresses(&self.proc_symbol, &self.linker);
+        scope.export()
+    }
+*/
+    pub fn get_bytecode(mut self, ret: bool) -> (
+        Vec<(String, SymbolDef)>,
+        Vec<tisc::assembler::AssemblyWord>
+    ) {
+        let frame_size = self.scope.read().unwrap().get_frame_size();
+        if frame_size > 0 {
+            let alloc_asm = tisc::Assembler::new()
+                .lit(frame_size as tisc::VM_Word).call("data-frame-alloc");
+            let drop_asm = tisc::Assembler::new()
+                .lit(frame_size as tisc::VM_Word).call("data-frame-drop");
 
-    pub fn into_asm(mut self, proc_symbol: &String) -> (Vec<(String, SymbolDef)>, Vec<(InputRegionTag, String)>, Vec<tisc::assembler::AssemblyWord>) {
-        let mut symbols =
-            Arc::try_unwrap(self.symbols).ok().unwrap()
-                .into_inner().unwrap();
+            self.asm = alloc_asm.join( self.asm ).join( drop_asm );
+        }
 
-        symbols.update_link_addresses(
-            proc_symbol,
+        let main_section = self.asm.build();
+
+        //self.linker.add_procedure( &self.proc_symbol, main_section );
+        // ^--- this would insert the asm section at the end,
+        //     we however need it an the beginning of the bytecode
+
+        // insert section at front
+        self.linker.next_addr += main_section.len() as i64;
+        for (name,section) in self.linker.symbols.iter_mut() {
+            section.addr += main_section.len() as i64;
+        }
+        self.linker.symbols.insert(
+            self.proc_symbol.clone(),
+            tisc::linker::Section { addr: 0, data: main_section }
+        );
+
+        // replace all symbol definitions from subroutines
+        // with relative LinkAddr`s
+        self.scope.write().unwrap().update_link_addresses(
+            &self.proc_symbol,
             &self.linker
         );
 
-        let data_frame_size = symbols.get_frame_size() as i64;
-
-        let body = self.asm.build();
-        self.linker.add_procedure("__procedure_body__", body);
-        let body_addr = self
-            .linker
-            .get_link_addr(&"__procedure_body__".into())
-            .unwrap();
-
-        let subroutines = self
-            .linker
-            .link_relative(&"__subroutines__".into())
-            .expect("link error");
-
-        let mut entry = tisc::Assembler::new();
-        if data_frame_size > 0 {
-            entry = entry.lit(data_frame_size).call("data-frame-alloc");
-        }
-        entry = entry.call_symbol(LinkAddr::Relative {
-            symbol: "__subroutines__".into(),
-            offset: body_addr,
-        });
-
-        if data_frame_size > 0 {
-            entry = entry.lit(data_frame_size).call("data-frame-drop");
-        }
-
-        let mut superlink = tisc::Linker::new();
-        superlink.add_procedure("", entry.build());
-        superlink.add_procedure("__subroutines__", subroutines);
-
-        symbols.update_link_addresses(
-            &proc_symbol,
-            &superlink
-        );
-
-        let mut symbol_exports = symbols.export();
-        let subroutines_addr = superlink.get_link_addr(&"__subroutines__".into()).unwrap();
-        for (name, def) in symbol_exports.iter_mut() {
-            match def {
-                SymbolDef::Procedure{ in_types:_, out_types:_, link_addr, export:_ } => {
-                    match link_addr {
-                        LinkAddr::Relative{ symbol, offset } => {
-                            *offset += subroutines_addr;
-                        }
-                        LinkAddr::Absolute(addr) => {
-                            *addr += subroutines_addr;
-                        }
-                    }
-                }
-                _ => {}
-            }
-        }
-        let bytecode = superlink.link_relative(proc_symbol).expect("link error");
-        (symbol_exports, self.diagnostics, bytecode)
-    }
-
-    pub fn verify(&self) {
-        // todo
+        (
+            self.scope.read().unwrap().export(),
+            self.linker.link_relative( &self.proc_symbol ).expect("link error")
+        )
     }
 
     pub fn compile_statement(mut self, statement: &Statement, enable_export: bool) -> Self {
         match statement {
             Statement::Assignment { name_region, var_id, val_expr } => {
-                self = self.compile(val_expr);
+                self = self.compile_expr(val_expr);
 
-                match self.symbols.read().unwrap().get(var_id) {
+                match self.scope.read().unwrap().get(var_id) {
                     Some(SymbolDef::FrameRef { typ, stack_ref }) => {
                         self.asm = self.asm.lit(stack_ref).call("data-frame-set");
                     }
@@ -123,9 +106,8 @@ impl ProcedureCompiler {
                         out_types,
                         link_addr,
                         export
-                    }) => {
-                        self.asm = self
-                            .asm
+                    }) =>  {
+                        self.asm = self.asm
                             .call(var_id.as_str())
                             .inst(tisc::VM_Instruction::Store);
                     }
@@ -138,73 +120,66 @@ impl ProcedureCompiler {
                 }
             }
             Statement::LetAssign {
+                name_region,
                 typ,
                 var_id,
                 val_expr,
-            } => match val_expr {
-                LTExpr::Abstraction { region:_, args: _, body: _ } => {
-                    self.symbols
-                        .write()
-                        .unwrap()
-                        .declare_proc(var_id.clone(), vec![], vec![], enable_export);
+            } => {
+                let val_type = self.scope.read().unwrap()
+                    .get(var_id).unwrap()
+                    .get_type(&mut self.scope.clone());
+                let val_type = val_type.sugar(&mut self.scope.clone());
+                match val_type {
+                    laddertypes::SugaredTypeTerm::Func(mut f_types) => {
+                        let mut c = ProcedureCompiler::new(
+                            var_id.clone(),
+                            self.scope.clone()
+                        );
+                        c = c.compile_expr( val_expr );
+                        self.diagnostics.append(&mut c.diagnostics);
 
-                    let (exports, mut diagnostics, lambda_procedure) = ProcedureCompiler::new(&self.symbols)
-                        .compile(val_expr)
-                        .into_asm(var_id);
+                        let (symbols,code) = c.get_bytecode( true );
+                        eprintln!("LET assign compiled {}", var_id);
+                        for (i,l) in tisc::assembler::disassemble( &code ).iter().enumerate() {
+                            eprintln!("{}+{} ... {}", var_id, i, l);
+                        }
+                        self.linker.add_procedure(var_id, code);
 
-                    self.diagnostics.append(&mut diagnostics);
+                        /*
+                        let out_types = vec![ f_types.pop().unwrap().desugar(&mut self.scope.clone()) ];
+                        let in_types = f_types.into_iter().map(|t| t.desugar(&mut self.scope.clone())).collect();
 
-                    self.linker.add_procedure(var_id, lambda_procedure);
+                        self.scope.write().unwrap().declare_proc(
+                            var_id.clone(),
+                            in_types,
+                            out_types,
+                            true
+                        );
+                        */
+                    }
 
-                    let offset = self.linker.get_link_addr(var_id).unwrap();
-
-                    // forward already exported symbols
-                    if enable_export {
-                        self.symbols.write().unwrap().import( exports );
+                    _ => {
+                        self = self.compile_statement(&Statement::Assignment {
+                            name_region: *name_region,
+                            var_id: var_id.clone(),
+                            val_expr: val_expr.clone(),
+                        }, false);
                     }
                 }
-                _ => {
-                    self.symbols
-                        .write()
-                        .unwrap()
-                        .declare_var(var_id.clone(), laddertypes::TypeTerm::unit());
-
-                    self = self.compile_statement(&Statement::Assignment {
-                        name_region: InputRegionTag::default(),
-                        var_id: var_id.clone(),
-                        val_expr: val_expr.clone(),
-                    }, false);
-                }
-            },
-            Statement::WhileLoop { condition, body } => {
-                let asm = self.asm;
-
-                self.asm = tisc::Assembler::new();
-                self = self.compile(condition);
-                let cond_asm = self.asm;
-
-                self.asm = tisc::Assembler::new();
-                for statement in body.into_iter() {
-                    self = self.compile_statement(statement, false);
-                }
-                let body_asm = self.asm;
-
-                self.asm = asm;
-                self.asm = self.asm.while_loop(cond_asm, body_asm);
             }
             Statement::Expr(expr) => {
-                self = self.compile(expr);
+                self = self.compile_expr(expr);
             }
             Statement::Return(expr) => {
-                self = self.compile(expr);
+                self = self.compile_expr(expr);
             }
         }
         self
     }
 
-    pub fn compile(mut self, expr: &LTExpr) -> Self {
+    pub fn compile_expr(mut self, expr: &LTExpr) -> Self {
         match expr {
-            LTExpr::Symbol { region, typ, symbol } => match self.symbols.read().unwrap().get(symbol) {
+            LTExpr::Symbol { region, typ, symbol } => match self.scope.read().unwrap().get(symbol) {
                 Some(SymbolDef::FrameRef { typ, stack_ref }) => {
                     self.asm = self.asm.lit(stack_ref).call("data-frame-get");
                 }
@@ -235,34 +210,10 @@ impl ProcedureCompiler {
                 self.asm = self.asm.lit(*val);
             }
             LTExpr::Ascend { region, typ, expr } => {
-                self = self.compile(expr);
+                self = self.compile_expr(expr);
             }
             LTExpr::Descend { region, typ, expr } => {
-                self = self.compile(expr);
-            }
-            LTExpr::Application { region, typ, head, body } => {                
-                for arg in body.iter().rev() {
-                    self = self.compile(arg);
-                }
-                self = self.compile(head);
-            }
-            LTExpr::Abstraction { region, args, body } => {
-                for (region, arg_name, arg_type) in args.iter() {
-                    if let Some(Ok(typeterm)) = arg_type {
-                        let id = self
-                            .symbols
-                            .write()
-                            .unwrap()
-                            .declare_var(arg_name.clone(), typeterm.clone());
-                        self.asm = self.asm.lit(id).call("data-frame-set");
-                    } else {
-                        self.diagnostics.push((
-                            region.clone(),
-                            format!("invalid type {:?} for argument {}", arg_type, arg_name)
-                        ));
-                    }
-                }
-                self = self.compile(body);
+                self = self.compile_expr(expr);
             }
             LTExpr::Branch {
                 region,
@@ -270,29 +221,112 @@ impl ProcedureCompiler {
                 if_expr,
                 else_expr,
             } => {
-                self = self.compile(condition);
+                self = self.compile_expr(condition);
 
                 let asm = self.asm;
                 self.asm = tisc::Assembler::new();
-                self = self.compile(if_expr);
+                self = self.compile_expr(if_expr);
                 let if_asm = self.asm;
                 self.asm = tisc::Assembler::new();
-                self = self.compile(else_expr);
+                self = self.compile_expr(else_expr);
                 let else_asm = self.asm;
                 self.asm = asm;
                 self.asm = self.asm.branch(if_asm, else_asm);
             }
-            LTExpr::Block { region, statements } => {
-                for s in statements.iter() {
-                    self = self.compile_statement(s, false);
-                }
+            LTExpr::WhileLoop { region, condition, body } => {
+                let asm = self.asm;
+
+                self.asm = tisc::Assembler::new();
+                self = self.compile_expr(condition);
+                let cond_asm = self.asm;
+
+                self.asm = tisc::Assembler::new();
+                self = self.compile_expr(body);
+                let body_asm = self.asm;
+
+                self.asm = asm;
+                self.asm = self.asm.while_loop(cond_asm, body_asm);
             }
-            LTExpr::ExportBlock{ region, statements } => {
-                for s in statements.iter() {
-                    self = self.compile_statement(s, true);
+            LTExpr::Application { region, typ, head, body } => {                
+                for arg in body.iter().rev() {
+                    self = self.compile_expr(arg);
                 }
+                self = self.compile_expr(head);
+            }
+            LTExpr::Abstraction { region, scope, args, body } => {
+                let mut abs_compiler = ProcedureCompiler::new("__abs__".into(), scope.clone());
+
+                for (region, arg_name, arg_type) in args.iter() {
+                    match scope.read().unwrap().get(arg_name) {
+                        Some(SymbolDef::FrameRef{ typ, stack_ref }) => {
+                            eprintln!("Arg {} stack ref = {}", arg_name, stack_ref);
+
+                            // TODO: aknowledge actual size of arguments
+                            // let arg_size = typ.get_size()
+                            let arg_size = 1;
+
+                            for i in 0..arg_size {
+                                abs_compiler.asm = abs_compiler.asm
+                                    .lit(stack_ref + i)
+                                    .call("data-frame-set");
+                            }
+                        }
+                        _ => {
+                            self.diagnostics.push(
+                                (region.clone(),
+                                format!("argument variable is not a frame-ref"))
+                            );
+                        }
+                    }
+                }
+
+                abs_compiler = abs_compiler.compile_expr( body );
+                let (abs_symbols, mut abs_code) = abs_compiler.get_bytecode( false );
+
+                for (s,def) in abs_symbols.iter() {
+                    eprintln!("{} = {:?}", s, def);
+                }
+                for (i, l) in tisc::assembler::disassemble(&abs_code).into_iter().enumerate() {
+                    eprintln!("__abs__+{} .. {}", i, l);
+                }
+
+                self.asm.words.append( &mut abs_code );
+/*
+                self.linker.add_procedure(
+                    "__abs__".into(),
+                    abs_code
+                );*/
+            }
+            LTExpr::Block { region, scope, statements } => {
+                let mut block_compiler = ProcedureCompiler::new(
+                    "__block__".into(),
+                    scope.clone()
+                );
+
+                for stmnt in statements.iter() {
+                    block_compiler = block_compiler.compile_statement( stmnt, true );
+                }
+
+                let (block_symbols, mut block_code) = block_compiler.get_bytecode( true );
+
+                eprintln!("BLOCK compiler:");
+                for (s,def) in block_symbols.iter() {
+                    eprintln!("{} = {:?}", s, def);
+                }
+                for (i,l) in tisc::assembler::disassemble( &block_code ).into_iter().enumerate() {
+                    eprintln!("block+{} .. {}", i, l);
+                }
+
+                self.linker.
+                self.scope.write().unwrap().import(
+                      block_symbols
+                );
+                self.asm.words.append(&mut block_code);
+            }
+            LTExpr::ExportBlock{ region, scope, statements } => {
             }
         }
         self
     }
 }
+

lib-ltcore/src/runtime.rs

@@ -6,7 +6,6 @@ use {
 
 pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
     let symbols = Scope::new();
-    let typectx = symbols.read().unwrap().typectx.clone();
 
     /* Duplicate the top item on the stack,
      * and whatever type this word has is preserved
@@ -14,8 +13,8 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
     symbols.write().unwrap().declare_proc_parse(
         "dup",
         vec!["T"],
-        vec!["T~machine::Word"],
-        vec!["T~machine::Word", "T~machine::Word"],
+        vec!["T~machine.Word"],
+        vec!["T~machine.Word", "T~machine.Word"],
     );
 
     /* drop topmost element
@@ -23,7 +22,7 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
     symbols.write().unwrap().declare_proc_parse(
         "drop",
         vec!["T"],
-        vec!["T~machine::Word"],
+        vec!["T~machine.Word"],
         vec![],
     );
     /* Put a single Ascii character on stdout
@@ -31,14 +30,14 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
     symbols.write().unwrap().declare_proc_parse(
         "emit",
         vec![],
-        vec!["Char~Ascii~machine::Word"],
+        vec!["Char~Unicode~ℤ_2^32~ℤ_2^64~machine.UInt64~machine.Word"],
         vec![],
     );
     symbols.write().unwrap().declare_proc_parse(
         "accept",
         vec![],
         vec![],
-        vec!["Char~Ascii~machine::Word"],
+        vec!["Char~Unicode~ℤ_2^32~ℤ_2^64~machine.UInt64~machine.Word"],
     );
 
     linker.add_procedure("dup", tisc::Assembler::new().inst(tisc::VM_Instruction::Dup).build());
@@ -58,7 +57,7 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
             "ℤ_2^64~machine.UInt64~machine.Word",
             "ℤ_2^64~machine.UInt64~machine.Word",
         ],
-        vec!["ℤ_2^64~machine::UInt64~machine::Word"],
+        vec!["ℤ_2^64~machine.UInt64~machine.Word"],
     );
     symbols.write().unwrap().declare_proc_parse(
         "i-",
@@ -67,7 +66,7 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
             "ℤ_2^64~machine.UInt64~machine.Word",
             "ℤ_2^64~machine.UInt64~machine.Word",
         ],
-        vec!["ℤ_2^64~machine::UInt64~machine::Word"],
+        vec!["ℤ_2^64~machine::UInt64~machine.Word"],
     );
     symbols.write().unwrap().declare_proc_parse(
         "i*",
@@ -85,7 +84,7 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
             "ℤ_2^64~machine.UInt64~machine.Word",
             "ℤ_2^64~machine.UInt64~machine.Word",
         ],
-        vec!["ℤ_2^64~machine::UInt64~machine::Word"],
+        vec!["ℤ_2^64~machine.UInt64~machine.Word"],
     );
     symbols.write().unwrap().declare_proc_parse(
         "i%",
@@ -163,27 +162,27 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
 
     symbols.write().unwrap().declare_proc_parse(
         "bit-neg",
-        vec![], vec!["machine::Word", "machine::Word"], vec!["machine::Word"],
+        vec![], vec!["machine.Word", "machine.Word"], vec!["machine.Word"],
     );
     symbols.write().unwrap().declare_proc_parse(
         "bit-and",
-        vec![], vec!["machine::Word", "machine::Word"], vec!["machine::Word"],
+        vec![], vec!["machine.Word", "machine.Word"], vec!["machine.Word"],
     );    
     symbols.write().unwrap().declare_proc_parse(
         "bit-or",
-        vec![], vec!["machine::Word", "machine::Word"], vec!["machine::Word"],
+        vec![], vec!["machine.Word", "machine.Word"], vec!["machine.Word"],
     );    
     symbols.write().unwrap().declare_proc_parse(
         "bit-xor",
-        vec![], vec!["machine::Word", "machine::Word"], vec!["machine::Word"],
+        vec![], vec!["machine.Word", "machine.Word"], vec!["machine.Word"],
     );
      symbols.write().unwrap().declare_proc_parse(
         "bit-shl",
-        vec![], vec!["machine::Word", "machine::Word"], vec!["machine::Word"],
+        vec![], vec!["machine.Word", "machine.Word"], vec!["machine.Word"],
     );    
     symbols.write().unwrap().declare_proc_parse(
         "bit-shr",
-        vec![], vec!["machine::Word", "machine::Word"], vec!["machine::Word"],
+        vec![], vec!["machine.Word", "machine.Word"], vec!["machine.Word"],
     );
 
     linker.add_procedure("bit-neg", tisc::Assembler::new().inst(tisc::VM_Instruction::BitNeg).build());
@@ -221,7 +220,7 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
 
     symbols.write().unwrap().declare_static_parse(
         "data-frame-ptr",
-        "<MutRef <Seq machine::Word>>~machine::Address~machine::Word",
+        "<MutRef <Seq machine.Word>>~machine.Address~machine.Word",
     );
     linker.add_static("data-frame-ptr", vec![0x1000]);
 
@@ -230,7 +229,7 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
         vec!["T"],
         vec![
             "T~machine::Word",
-            "<RefMut T~machine::Word>~LocalVariableId~machine::UInt64~machine::Word",
+            "<RefMut T~machine.Word>~LocalVariableId~machine.UInt64~machine.Word",
         ],
         vec![],
     );
@@ -248,8 +247,8 @@ pub fn init_runtime(linker: &mut Linker) -> Arc<RwLock<Scope>> {
     symbols.write().unwrap().declare_proc_parse(
         "data-frame-get",
         vec!["T"],
-        vec!["<Ref T~machine::Word>~DataFrameRef~machine::UInt64~machine::Word"],
-        vec!["T~machine::Word"],
+        vec!["<Ref T~machine.Word>~DataFrameRef~machine.UInt64~machine.Word"],
+        vec!["T~machine.Word"],
     );
     linker.add_procedure(
         "data-frame-get",

lib-ltcore/src/symbols.rs

@@ -1,5 +1,9 @@
 use {
     crate::expr::LTExpr,
+    laddertypes::{
+        TypeDict, TypeID,
+        parser::ParseLadderType
+    },
     std::{
         collections::HashMap,
         sync::{Arc, RwLock},
@@ -28,7 +32,7 @@ pub enum SymbolDef {
 impl SymbolDef {
     pub fn get_type(
         &self,
-        typectx: &Arc<RwLock<laddertypes::dict::TypeDict>>,
+        typedict: &mut impl laddertypes::dict::TypeDict,
     ) -> laddertypes::TypeTerm {
         match self {
             SymbolDef::FrameRef { typ, stack_ref: _ } => typ.clone(),
@@ -38,29 +42,38 @@ impl SymbolDef {
                 out_types,
                 link_addr: _,
                 export: _,
-            } => laddertypes::TypeTerm::App(
-                std::iter::once(
-                    typectx
-                    .write()
-                    .unwrap()
-                    .parse("Func")
-                    .expect("parse typeterm")
-                ).chain(
-                    in_types.clone().into_iter()
-                ).chain(
+            } => {
+                let mut out_types = out_types.clone();
+                let out_type =
+                    if out_types.len() == 1 {
+                        out_types.pop().unwrap()
+                    } else {
+                        laddertypes::TypeTerm::App(
+                            std::iter::once(
+                                typedict.parse("Struct").unwrap()
+                            ).chain(
+                                out_types.into_iter()
+                            ).collect()
+                        )
+                    };
+
+                laddertypes::TypeTerm::App(
                     std::iter::once(
-                        typectx.write().unwrap().parse("Struct").expect("parse typeterm")
+                        typedict.parse("Func").expect("parse typeterm")
                     ).chain(
-                        out_types.clone().into_iter()
-                    )
-                ).collect()
-            ),
+                        in_types.clone().into_iter()
+                    ).chain(
+                        std::iter::once(out_type)
+                    ).collect()
+                )
+            },
         }
     }
 }
 
 /* Describes a lexical scope of symbols
  */
+#[derive(Clone, Debug)]
 pub struct Scope {
     /* definition of runtime symbols
      */
@@ -68,7 +81,7 @@ pub struct Scope {
 
     /* type symbols
      */
-    pub typectx: Arc<RwLock<laddertypes::TypeDict>>,
+    typedict: Arc<RwLock<laddertypes::BimapTypeDict>>,
 
     /* number of words required for
      * the stack frame of this scope
@@ -81,39 +94,70 @@ pub struct Scope {
     parent: Option<Arc<RwLock<Scope>>>,
 }
 
+impl TypeDict for Scope {
+    fn insert(&mut self, name: String, id: TypeID) {
+        self.typedict.write().unwrap().insert(name,id)
+    }
+    fn add_varname(&mut self, vn: String) -> TypeID {
+        self.typedict.add_varname(vn)
+    }
+    fn add_typename(&mut self, tn: String) -> TypeID {
+        if let Some(parent) = self.parent.as_mut() {
+            parent.add_typename(tn)
+        } else {
+            self.typedict.add_typename(tn)
+        }
+    }
+    fn get_typeid(&self, tn: &String) -> Option<TypeID> {
+        if let Some(id) = self.typedict.get_typeid(tn) {
+            Some(id)
+        } else {
+            if let Some(parent) = self.parent.as_ref() {
+                parent.get_typeid(tn)
+            } else {
+                None
+            }
+        }
+    }
+    fn get_typename(&self, tid: &TypeID) -> Option<String> {
+        if let Some(name) = self.typedict.get_typename(tid) {
+            Some(name)
+        } else {
+            if let Some(parent) = self.parent.as_ref() {
+                parent.get_typename(tid)
+            } else {
+                None
+            }
+        }
+    }
+}
+
 impl Scope {
     pub fn new() -> Arc<RwLock<Self>> {
         Arc::new(RwLock::new(Scope {
             symbols: HashMap::new(),
-            typectx: Arc::new(RwLock::new(laddertypes::dict::TypeDict::new())),
+            typedict: Arc::new(RwLock::new(laddertypes::dict::BimapTypeDict::new())),
             frame_size: 0,
             parent: None,
         }))
     }
 
     pub fn with_parent(parent: &Arc<RwLock<Scope>>) -> Arc<RwLock<Self>> {
-        let s = Scope {
-            symbols: HashMap::new(),
-
-            // todo: create proper child scope
-            typectx: parent.read().unwrap().typectx.clone(),
-
-            frame_size: 0,
-            parent: Some(parent.clone()),
-        };
-
-        Arc::new(RwLock::new(s))
+        let mut s = Scope::new();
+        s.write().unwrap().parent = Some(parent.clone());
+        s
     }
 
-    pub fn export(self) -> Vec<(String, SymbolDef)> {
+    pub fn export(&self) -> Vec<(String, SymbolDef)> {
         self.symbols
-            .into_iter()
+            .iter()
             .filter(|(name, def)|
                 match def {
                     SymbolDef::Procedure { in_types:_, out_types:_, link_addr:_, export } => *export,
                     _ => false
                 }
             )
+            .map(|(n,d)| (n.clone(), d.clone()))
             .collect()
     }
 
@@ -146,6 +190,14 @@ impl Scope {
         }
     }
 
+    pub fn get_type(&mut self, name: &str) -> Option<laddertypes::TypeTerm> {
+        if let Some(sdef) = self.get(name) {
+            Some(sdef.get_type( &mut self.typedict ))
+        } else {
+            None
+        }
+    }
+
     /// takes the link-addresses from a Linker
     /// and updates the symbol table to relative addresses
     /// based on the next super-label
@@ -157,7 +209,7 @@ impl Scope {
         for (name, def) in self.symbols.iter_mut() {
             if let Some(offset) = linker.get_link_addr( name ) {
                 match def {
-                   SymbolDef::Procedure {
+                    SymbolDef::Procedure {
                         in_types:_,out_types:_,
                         link_addr,
                         export:_
@@ -184,29 +236,26 @@ impl Scope {
         out_types: Vec<&str>,
     ) {
         for v in type_vars {
-            self.typectx.write().unwrap().add_varname(v.into());
+            self.add_varname(v.into());
         }
 
+        let mut td = self.typedict.clone();
+
         self.declare_proc(
             String::from(name),
             in_types
                 .into_iter()
-                .map(|t| {
-                    self.typectx
-                        .write()
-                        .unwrap()
-                        .parse(t)
-                        .expect("parse typeterm")
+                .map(move |t| {
+                    td.parse(t).expect("parse typeterm")
                 })
                 .collect(),
             out_types
                 .into_iter()
-                .map(|t| {
-                    self.typectx
-                        .write()
-                        .unwrap()
-                        .parse(t)
-                        .expect("parse typeterm")
+                .map({
+                    let mut td = self.typedict.clone();
+                    move |t| {
+                        td.parse(t).expect("parse typeterm")
+                    }
                 })
                 .collect(),
             false
@@ -235,9 +284,6 @@ impl Scope {
 
     pub fn declare_var_parse(&mut self, name: &str, typ: &str) {
         let typ = self
-            .typectx
-            .write()
-            .unwrap()
             .parse(typ)
             .expect("parse typeterm");
         self.declare_var(String::from(name), typ);
@@ -257,9 +303,6 @@ impl Scope {
 
     pub fn declare_static_parse(&mut self, name: &str, typ: &str) {
         let typ = self
-            .typectx
-            .write()
-            .unwrap()
             .parse(typ)
             .expect("parse typeterm");
         self.declare_static(String::from(name), typ);

lib-ltcore/src/typing.rs

@@ -0,0 +1,319 @@
+use {
+    crate::{
+        lexer::InputRegionTag,
+        expr::{LTExpr, Statement, TypeTag, TypeError, TypeErrorKind},
+        symbols::{Scope, SymbolDef},
+    },
+    std::{
+        ops::Deref,
+        sync::{Arc, RwLock},
+    },
+    laddertypes::{
+        parser::ParseLadderType,
+        unparser::UnparseLadderType,
+        dict::TypeDict
+    },
+    tisc::{assembler::AssemblyWord, linker::LinkAddr},
+    tiny_ansi::TinyAnsi
+};
+
+impl LTExpr {
+    /*
+    pub fn get_type(&self) -> TypeTag {
+        Err(TypeError::Todo)
+    }*/
+
+    pub fn infer_type(&self, scope: &Arc<RwLock<Scope>>) -> TypeTag
+    {
+        match self {
+            LTExpr::WordLiteral{ region, val } => {
+                Ok(scope.write().unwrap().parse(
+                    "ℤ_2^64 ~ machine.UInt64 ~ machine.Word"
+                ).unwrap())
+            }
+
+            LTExpr::StringLiteral{ region, value } => {
+                Ok(scope.write().unwrap().parse(
+                    "<Seq Char ~ Unicode ~ ℤ_2^32 ~ ℤ_2^64 ~ machine.UInt64>
+                    ~ <TermArray 0 machine.UInt64 ~ machine.Word>"
+                ).unwrap())
+            }
+
+            LTExpr::Symbol { region, typ, symbol } => {
+                let mut s = scope.write().unwrap();
+                if let Some(sdef) = s.get(symbol) {
+                    Ok(sdef.get_type(&mut *s))
+                } else {
+                    let region = region.clone();
+                    Err(vec![ TypeError{ region, kind: TypeErrorKind::NoSymbol } ])
+                }
+            }
+
+            LTExpr::Ascend { region, typ, expr } => {
+                let expr_type = expr.infer_type( scope )?;
+                let sub_type = typ.clone();
+
+                /*
+                 * todo: check potential overlap of typ with expr_type
+                 */
+                if let Ok(i) = sub_type.is_syntactic_subtype_of(&expr_type) {
+                    let mut lnf = expr_type.get_lnf_vec();
+                    let mut sub_lnf = sub_type.get_lnf_vec();
+
+                    for x in 0..i {
+                        lnf.insert(x, sub_lnf.remove(0));
+                    }
+                    let result_type = laddertypes::TypeTerm::Ladder(lnf);
+                    Ok(result_type)
+                } else {
+                    Ok(laddertypes::TypeTerm::Ladder(vec![
+                        sub_type,
+                        expr_type
+                    ]))
+                }
+            }
+
+            LTExpr::Descend { region, typ, expr } => {
+                let expr_type = expr.infer_type(scope)?;
+                let super_type = typ.clone();
+
+                if let Ok(i) = expr_type.is_syntactic_subtype_of(&super_type) {
+                    let lnf = expr_type.get_lnf_vec();
+                    let result_type = laddertypes::TypeTerm::Ladder(lnf[i..].into_iter().cloned().collect());
+                    Ok(result_type)
+                } else {
+                    return Err(vec![ TypeError{
+                        region: region.clone(),
+                        kind: TypeErrorKind::ArgTypeMismatch {
+                            expected: expr_type,
+                            received: super_type
+                        }
+                    } ]);
+                }
+            }
+
+            LTExpr::Abstraction { region, scope, args, body } => {
+                let mut f = Vec::new();
+
+                for (region, name, typ) in args {
+                    if let Some(typ) = typ {
+                        let typ = typ.clone()?;
+                        let sugar_typ = typ.clone().sugar(&mut *scope.write().unwrap());
+                        f.push( sugar_typ );
+
+                        scope.write().unwrap().declare_var(name.clone(), typ.clone());
+                    }
+                }
+
+                let body_type = body.infer_type( scope )?;
+                f.push( body_type.sugar(&mut *scope.write().unwrap()) );
+
+                Ok(laddertypes::SugaredTypeTerm::Func(f).desugar( &mut *scope.write().unwrap() ))
+            }
+
+            LTExpr::Application{ region, typ, head, body } => {
+                let mut head_type = head.infer_type(scope)?;
+                let mut args = body.into_iter();
+
+                let mut result_type = head_type;
+                let mut sugared_result_type = result_type.sugar(&mut *scope.write().unwrap());
+
+                let mut errors = Vec::new();
+
+                while let laddertypes::SugaredTypeTerm::Func(mut f_types) = sugared_result_type {
+                    sugared_result_type = f_types.pop().unwrap();
+
+                    for (argi, expected_arg_type) in f_types.iter().enumerate() {
+                        if let Some(arg) = args.next() {
+
+                            let expected_arg_type = expected_arg_type.clone().desugar(&mut *scope.write().unwrap());
+
+                            // check subtype
+                            let received_arg_type = arg.infer_type(scope)?;
+                            if ! received_arg_type.is_syntactic_subtype_of(&expected_arg_type).is_ok() {
+                                errors.push(TypeError{
+                                    region: arg.get_region(),
+                                    kind: TypeErrorKind::ArgTypeMismatch {
+                                        expected: expected_arg_type,
+                                        received: received_arg_type
+                                    }
+                                });
+                            }
+
+                        } else {
+                            // partial application.
+                            f_types.push(sugared_result_type);
+                            sugared_result_type = laddertypes::SugaredTypeTerm::Func(
+                                f_types[argi .. ].into_iter().cloned().collect()
+                            );
+
+                            // todo examine stack ..
+                            
+                            return
+                                if errors.len() == 0 {
+                                    result_type = sugared_result_type.desugar(&mut *scope.write().unwrap());
+                                    Ok(result_type)
+                                } else {
+                                    Err(errors)
+                                };
+                        }
+                    }
+                }
+
+                while let Some(arg) = args.next() {
+                    errors.push(TypeError{
+                        region: arg.get_region(),
+                        kind: TypeErrorKind::SuperfluousArgument
+                    });
+                }
+
+                if errors.len() == 0 {
+                    result_type = sugared_result_type.desugar(&mut *scope.write().unwrap());
+                    Ok(result_type)
+                } else {
+                    Err(errors)
+                }
+            }
+
+            LTExpr::Branch { region, condition, if_expr, else_expr } => {
+                let received_cond_type = condition.infer_type(scope)?;
+                let expected_cond_type = scope.write().unwrap().parse("Bool ~ machine.Word").unwrap();
+
+                if received_cond_type.is_syntactic_subtype_of(&expected_cond_type).is_ok() {
+
+                    let if_expr_type = if_expr.infer_type(scope)?;
+                    let else_expr_type = else_expr.infer_type(scope)?;
+
+                    if if_expr_type.is_syntactic_subtype_of(&else_expr_type).is_ok() {
+                        Ok(else_expr_type)
+                    } else if else_expr_type.is_syntactic_subtype_of(&if_expr_type).is_ok() {
+                        Ok(if_expr_type)
+                    } else {
+                        Err(vec![TypeError{
+                           region: region.clone(),
+                           kind: TypeErrorKind::BranchMismatch {
+                               if_branch: if_expr_type,
+                               else_branch: else_expr_type
+                           }
+                        }])
+                    }
+                } else {
+                    Err(vec![ TypeError{
+                        region: condition.get_region(),
+                        kind: TypeErrorKind::ArgTypeMismatch {
+                            expected: expected_cond_type,
+                            received: received_cond_type
+                        }
+                    }])
+                }
+            }
+            LTExpr::WhileLoop { region, condition, body } => {
+                let received_cond_type = condition.infer_type(scope)?;
+                let expected_cond_type = scope.write().unwrap().parse("Bool ~ machine.Word").unwrap();
+
+                if received_cond_type.is_syntactic_subtype_of(&expected_cond_type).is_ok() {
+                    let body_type = body.infer_type(scope)?;
+                    let body_type = body_type.sugar(&mut scope.clone());
+                    let loop_type = laddertypes::SugaredTypeTerm::Seq(vec![ body_type ]);
+                    Ok(loop_type.desugar(&mut scope.clone()))
+                } else {
+                    return Err(vec![ TypeError{
+                        region: condition.get_region(),
+                        kind: TypeErrorKind::ArgTypeMismatch {
+                            expected: expected_cond_type,
+                            received: received_cond_type
+                        }
+                    }]);
+                }
+            }
+            LTExpr::ExportBlock{ region, scope, statements } |
+            LTExpr::Block{ region, scope, statements } => {
+                let mut types = Vec::new();
+
+                for s in statements {
+                    match s.infer_type(scope) {
+                        Ok(Some(t)) => {
+                            if !t.is_empty() {
+                                types.insert(0, t);
+                            }
+                        }
+                        Ok(None) => {}
+                        Err(e) => {
+                            return Err(e);
+                        }
+                    }
+                }
+
+                Ok(
+                    if types.len() == 1 { types.pop().unwrap() }
+                    else { laddertypes::SugaredTypeTerm::Struct(types) }
+                    .desugar(&mut scope.clone())
+                )
+            }
+        }
+    }
+}
+
+impl Statement {
+    pub fn infer_type(&self, scope: &Arc<RwLock<Scope>>) -> Result< Option<laddertypes::SugaredTypeTerm> , Vec<TypeError> > {
+        match self {
+            Statement::LetAssign{ name_region, typ, var_id, val_expr } => {
+                let typ = val_expr.infer_type( scope )?;
+
+                match typ.clone().sugar( &mut scope.clone() ) {
+                    laddertypes::SugaredTypeTerm::Func(mut args) => {
+                        let out_type = args.pop().unwrap();
+                        let out_types =
+                            match out_type.clone() {
+                                laddertypes::SugaredTypeTerm::Struct(oa) => oa.into_iter().map(|t|t.desugar(&mut scope.clone())).collect(),
+                                _ => vec![ out_type.desugar(&mut scope.clone()) ]
+                            };
+                         let in_types = args.into_iter().map(|t| t.desugar(&mut scope.clone())).collect();
+
+                         scope.write().unwrap()
+                             .declare_proc(
+                                 var_id.clone(),
+                                 in_types,
+                                 out_types,
+                                 true
+                            );
+
+                        return Ok(None);
+                    }
+                    _ => {
+                       let id = scope.write().unwrap().declare_var(var_id.clone(), typ);
+                       eprintln!("TYPING declare var = {}", id);
+                       Ok(None)
+                    }
+                }
+            },
+            Statement::Return(expr) |
+            Statement::Expr(expr) => {
+                let t = expr.infer_type(scope)?;
+
+                if t != laddertypes::TypeTerm::App(vec![]) {
+                    let st = t.sugar(&mut scope.clone());
+                    Ok(Some(st))
+                } else {
+                    Ok(None)
+                }
+            }
+            Statement::Assignment { name_region, var_id, val_expr } => {
+                let received_type = val_expr.infer_type(scope)?;
+                let expected_type = scope.write().unwrap().get_type(var_id).unwrap();
+                if ! received_type.is_syntactic_subtype_of(&expected_type).is_ok() {
+                    return Err(vec![ TypeError{
+                        region: val_expr.get_region(),
+                        kind: TypeErrorKind::AssignMismatch {
+                            expected: expected_type,
+                            received: received_type
+                        }
+                    }]);
+                } else {
+                    Ok(None)
+                }
+            }
+        }
+    }
+}
+

ltcc/Cargo.toml

@@ -4,6 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+laddertypes = { path = "../../lib-laddertypes" }
 ltcore = { path = "../lib-ltcore" }
 tisc = { path = "../../lib-tisc" }
 clap = { version = "4.5.15", features = ["derive"] }

ltcc/src/diagnostic.rs

@@ -20,8 +20,8 @@ pub fn print_diagnostic(
 
     let mut line_region = InputRegionTag::default();
 
-    let n_before = 3;
-    let n_after  = 3;
+    let n_before = 5;
+    let n_after  = 5;
 
     let mut last_lines = Vec::new();
     let mut next_lines = 0;
@@ -67,11 +67,11 @@ pub fn print_diagnostic(
             }
 
             print!("\t{}", "|".bright_magenta());
-            for _ in 0..column_begin { print!("{}", ".".magenta().bold()); }
-            for _ in column_begin..column_end { print!("{}", "^".magenta().bold()); }
+            for _ in 0..column_begin_c { print!("{}", ".".magenta().bold()); }
+            for _ in column_begin_c..column_end_c { print!("{}", "^".magenta().bold()); }
             print!("\n");
 
-            print!("{} [{}-{}]: {}\n", "error".bright_red(), column_begin, column_end, message.white());
+            print!("{} [{}-{}]: {}\n", "error".bright_red(), column_begin, column_end, message.yellow());
         }
         else if next_lines > 0 {
             next_lines -= 1;

ltcc/src/main.rs

@@ -6,6 +6,7 @@ use {
     std::{boxed::Box, ops::Deref},
     std::io::Write,
     tiny_ansi::TinyAnsi,
+    laddertypes::dict::TypeDict,
     ltcore::{
         lexer::InputRegionTag,
         expr::{LTExpr, Statement},
@@ -32,8 +33,7 @@ fn main() {
 
     let mut linker = tisc::Linker::new();
     let root_scope = ltcore::runtime::init_runtime(&mut linker);
-    let main_scope = Scope::with_parent(&root_scope);
-    let typectx = main_scope.read().unwrap().typectx.clone();
+    let mut main_scope = Scope::with_parent(&root_scope);
 
     for path in args.sources {
         let iter_chars = iterate_text::file::characters::IterateFileCharacters::new(path.clone());
@@ -49,23 +49,55 @@ fn main() {
                 })
                 .peekable();
 
-        match ltcore::parser::parse_expr( &typectx, &mut program_tokens ) {
-            Ok( ast ) => {
-                let (exports, diagnostics, proc_code) = ProcedureCompiler::new(&main_scope)
-                    .compile(&ast)
-                    .into_asm(&path);
+        match ltcore::parser::parse_expr( &mut main_scope, &mut program_tokens ) {
+            Ok( mut ast ) => {
+                let mut compiler = ProcedureCompiler::new(path.clone(), main_scope.clone());
 
-                for (region, message) in diagnostics {                
+                match ast.infer_type(&main_scope) {
+                    Ok(mut t) => {
+                        eprintln!("Typecheck {}", "OK".green().bold());
+                        t = t.normalize();
+                        t = t.param_normalize();
+                        let mut tc = main_scope.clone();
+                        eprintln!( "{}", t.sugar(&mut tc).pretty(&tc,0) );
+
+                    }
+                    Err(type_errs) => {
+                        for e in type_errs.iter() {
+                            crate::diagnostic::print_diagnostic(
+                                path.as_str(),
+                                e.region,
+                                e.kind.fmt(&mut main_scope.clone())
+                            );
+                        }
+
+                        eprintln!("----------------------------------");
+                        eprintln!("{} ({} errors)", "Typecheck failed".bright_red().bold(), type_errs.len());
+                        return;
+                    }
+                }
+
+                compiler = compiler.compile_expr(&ast);
+                let diagnostics = compiler.diagnostics.clone();
+                let (exports, proc_code) = compiler.get_bytecode(false);
+
+                for (region, message) in diagnostics {
                     crate::diagnostic::print_diagnostic(
                         path.as_str(),
                         region,
                         format!("{}", message)
                     );
                 }
- 
+
+
                 eprintln!("{} {}", "Compiled".green(), path.bold());
                 for (name, def) in exports.iter() {
-                    eprintln!("export {}: {:?}", name.yellow().bold(), def);
+                    eprintln!("export {}:", name.yellow().bold());
+                    let mut t = def.get_type(&mut main_scope);
+                    t = t.normalize();
+                    t = t.param_normalize();
+                    let mut tc = main_scope.clone();
+                    eprintln!( "{}", t.sugar(&mut tc).pretty(&tc,0) );
                 }
 
                 main_scope.write().unwrap().import(
@@ -74,6 +106,10 @@ fn main() {
 
                 /* link assembly-program to symbols
                  */
+                eprintln!("generated bytecode ({})", proc_code.len() );
+                for (i,l) in tisc::assembler::disassemble(&proc_code).iter().enumerate() {
+                    eprintln!("{} .... {}", i,l);
+                }
                 linker.add_procedure(path.as_str(), proc_code);
             }
             Err( (region, parse_error) ) => {

ltcc/test.lt

@@ -0,0 +1 @@
+

ltvm/src/main.rs

@@ -35,8 +35,10 @@ fn main() {
         linker.import( source_path, bincode::deserialize_from( input ).expect("") );
     }
 
-    let entry_addr = linker.get_link_addr(&args.entry)
+    let entry_addr = linker.get_link_addr(&args.entry).unwrap_or(0);
+        /*
         .expect(&format!("cant find entry symbol '{}'", args.entry));
+        */
     let bytecode = linker.link_total().expect("Link error:");
 
     eprintln!("{} ({} bytes)", "Loaded bytecode.".green(), bytecode.len());