From bf47bd7b3d4f70bc3a481761b8e9ff074313fb44 Mon Sep 17 00:00:00 2001 From: Eduardo Julian Date: Wed, 3 May 2017 18:17:00 -0400 Subject: - WIP: Implemented more functionality. - Lots of refactorings. --- new-luxc/source/luxc/analyser.lux | 393 +++++--------------------------------- 1 file changed, 45 insertions(+), 348 deletions(-) (limited to 'new-luxc/source/luxc/analyser.lux') diff --git a/new-luxc/source/luxc/analyser.lux b/new-luxc/source/luxc/analyser.lux index 84719738e..2a77d8bb5 100644 --- a/new-luxc/source/luxc/analyser.lux +++ b/new-luxc/source/luxc/analyser.lux @@ -15,354 +15,51 @@ [type] (type ["TC" check])) (luxc ["&" base] + [lang #*] ["&;" module] - ["&;" env] - (module ["&;" def]))) + ["&;" env]) + (. ["&&;" lux])) + +(def: #export (analyse eval ast) + Analyser + (case ast + (^template [ ] + [cursor ( value)] + ( cursor value)) + ([#;BoolS &&lux;analyse-bool] + [#;NatS &&lux;analyse-nat] + [#;IntS &&lux;analyse-int] + [#;DegS &&lux;analyse-deg] + [#;RealS &&lux;analyse-real] + [#;CharS &&lux;analyse-char] + [#;TextS &&lux;analyse-text]) + + (^ [cursor (#;TupleS (list))]) + (&&lux;analyse-unit cursor) + + (^ [cursor (#;TupleS (list singleton))]) + (analyse eval singleton) + + (^ [cursor (#;TupleS elems)]) + (&&lux;analyse-tuple (analyse eval) cursor elems) + + [cursor (#;SymbolS reference)] + (&&lux;analyse-reference cursor reference) + + (^ [cursor (#;FormS (list [_ (#;SymbolS ["" "_lux_check"])] + type + value))]) + (&&lux;analyse-check analyse eval cursor type value) + + (^ [cursor (#;FormS (list [_ (#;SymbolS ["" "_lux_coerce"])] + type + value))]) + (&&lux;analyse-coerce analyse eval cursor type value) + + (^ [cursor (#;FormS (list [_ (#;NatS tag)] + value))]) + (&&lux;analyse-variant (analyse eval) cursor tag value) -(type: #export Pattern Void) - -(type: #export (Analysis' Analysis) - (#Bool Bool) - (#Nat Nat) - (#Int Int) - (#Deg Deg) - (#Real Real) - (#Char Char) - (#Text Text) - (#Variant Nat Bool Analysis) - #Unit - (#Tuple (List Analysis)) - (#Call Analysis (List Analysis)) - (#Case (List [Pattern Analysis])) - (#Function Scope Analysis) - (#Var Ref) - (#Def Ident) - (#Procedure Text (List Analysis)) - ) - -(type: #export #rec Analysis - (Meta [Type Cursor] - (Analysis' Analysis))) - -(def: (with-expected-type expected action) - (All [a] (-> Type (Lux a) (Lux a))) - (function [compiler] - (case (action (set@ #;expected (#;Some expected) compiler)) - (#E;Success [compiler' output]) - (let [old-expected (get@ #;expected compiler)] - (#E;Success [(set@ #;expected old-expected compiler') - output])) - - (#E;Error error) - (#E;Error error)))) - -(def: (analyse-typed-tuple analyse cursor members) - (-> (-> AST (Lux Analysis)) Cursor - (List AST) - (Lux Analysis)) - (do Monad - [expected macro;expected-type] - (let [member-types (type;flatten-tuple expected) - num-types (list;size member-types) - num-members (list;size members)] - (cond (n.= num-types num-members) - (do @ - [=tuple (: (Lux (List Analysis)) - (mapM @ - (function [[expected member]] - (with-expected-type expected - (analyse member))) - (list;zip2 member-types members)))] - (wrap [[expected cursor] - (#Tuple =tuple)])) - - (n.< num-types num-members) - (do @ - [#let [[head-ts tail-ts] (list;split (n.- +2 num-members) - member-types)] - =prevs (mapM @ - (function [[expected member]] - (with-expected-type expected - (analyse member))) - (list;zip2 head-ts members)) - =last (with-expected-type (type;tuple tail-ts) - (analyse (default (undefined) - (list;last members))))] - (wrap [[expected cursor] - (#Tuple (L/append =prevs (list =last)))])) - - ## (n.> num-types num-members) - (do @ - [#let [[head-xs tail-xs] (list;split (n.- +2 num-types) - members)] - =prevs (mapM @ - (function [[expected member]] - (with-expected-type expected - (analyse member))) - (list;zip2 member-types head-xs)) - =last (with-expected-type (default (undefined) - (list;last member-types)) - (analyse-typed-tuple analyse cursor tail-xs))] - (wrap [[expected cursor] - (#Tuple (L/append =prevs (list =last)))])) - )))) - -(def: (within-type-env action) - (All [a] (-> (TC;Check a) (Lux a))) - (function [compiler] - (case (action (get@ #;type-context compiler)) - (#E;Error error) - (#E;Error error) - - (#E;Success [context' output]) - (#E;Success [(set@ #;type-context context' compiler) - output])))) - -(def: get-type - (-> Analysis Type) - (|>. product;left - product;left)) - -(def: (replace-type replacement analysis) - (-> Type Analysis Analysis) - (let [[[_type _cursor] _analysis] analysis] - (: Analysis - [[(: Type replacement) - (: Cursor _cursor)] - (: (Analysis' Analysis) - _analysis)]))) - -(def: (clean-analysis type analysis) - (-> Type Analysis (Lux Analysis)) - (case type - (#;VarT id) - (do Monad - [=type (within-type-env - (TC;clean id type))] - (wrap (replace-type =type analysis))) - - (#;ExT id) - (undefined) - _ - (&;fail (format "Cannot clean type: " (%type type))))) - -(def: (with-unknown-type action) - (All [a] (-> (Lux Analysis) (Lux Analysis))) - (do Monad - [[var-id var-type] (within-type-env - TC;create-var) - analysis (|> (wrap action) - (%> @ - [(with-expected-type var-type)] - [(clean-analysis var-type)])) - _ (within-type-env - (TC;delete-var var-id))] - (wrap analysis))) - -(def: (tuple cursor members) - (-> Cursor (List Analysis) Analysis) - (let [tuple-type (type;tuple (L/map get-type members))] - [[tuple-type cursor] - (#Tuple members)])) - -(def: (realize expected) - (-> Type (TC;Check [(List Type) Type])) - (case expected - (#;NamedT [module name] _expected) - (realize _expected) - - (#;UnivQ env body) - (do TC;Monad - [[var-id var-type] TC;create-var - [tail =expected] (realize (default (undefined) - (type;apply-type expected var-type)))] - (wrap [(list& var-type tail) - =expected])) - - (#;ExQ env body) - (do TC;Monad - [[ex-id ex-type] TC;existential - [tail =expected] (realize (default (undefined) - (type;apply-type expected ex-type)))] - (wrap [(list& ex-type tail) - =expected])) - - _ - (:: TC;Monad wrap [(list) expected]))) - -(def: (analyse-tuple analyse cursor members) - (-> (-> AST (Lux Analysis)) Cursor - (List AST) - (Lux Analysis)) - (do Monad - [expected macro;expected-type] - (case expected - (#;ProdT _) - (analyse-typed-tuple analyse cursor members) - - (#;VarT id) - (do @ - [bound? (within-type-env - (TC;bound? id))] - (if bound? - (do @ - [expected' (within-type-env - (TC;read-var id)) - =tuple (with-expected-type expected' - (analyse-tuple analyse cursor members))] - (wrap (replace-type expected =tuple))) - (do @ - [=members (mapM @ (<|. with-unknown-type - analyse) - members) - #let [=tuple (tuple cursor =members)] - _ (within-type-env - (TC;check expected (get-type =tuple)))] - (wrap (replace-type expected =tuple))))) - - _ - (if (type;quantified? expected) - (do @ - [[bindings expected'] (within-type-env - (realize expected)) - =tuple (with-expected-type expected' - (analyse-tuple analyse cursor members)) - =tuple (foldM @ clean-analysis =tuple bindings) - _ (within-type-env - (TC;check expected (get-type =tuple)))] - (wrap (replace-type expected =tuple))) - (&;fail (format "Invalid type for tuple: " (%type expected)))) - ))) - -(def: (analyse-variant analyse cursor tag value) - (-> (-> AST (Lux Analysis)) Cursor - Nat AST - (Lux Analysis)) - (do Monad - [expected macro;expected-type] - (case expected - (#;SumT _) - (let [flat (type;flatten-variant expected) - type-size (list;size flat)] - (if (n.< type-size tag) - (do @ - [#let [last? (n.= tag (n.dec type-size)) - variant-type (default (undefined) - (list;nth tag flat))] - =value (with-expected-type variant-type - (analyse value))] - (wrap [[expected cursor] - (#Variant tag last? =value)])) - (&;fail (format "Trying to create variant with tag beyond type's limitations." "\n" - "Tag: " (%n tag) "\n" - "Type size: " (%n type-size) "\n" - "Type: " (%type expected) "\n")))) - - _ - (if (type;quantified? expected) - (do @ - [[bindings expected'] (within-type-env - (realize expected)) - =variant (with-expected-type expected' - (analyse-variant analyse cursor tag value)) - =variant (foldM @ clean-analysis =variant bindings) - _ (within-type-env - (TC;check expected (get-type =variant)))] - (wrap (replace-type expected =variant))) - (&;fail (format "Invalid type for variant: " (%type expected))))))) - -(def: (analyse eval ast) - (-> (-> Type AST (Lux Top)) AST (Lux Analysis)) - (do Monad - [] - (case ast - (^template [ ] - [cursor ( value)] - (do @ - [expected macro;expected-type - _ (within-type-env - (TC;check expected ))] - (wrap [[ cursor] - ( value)]))) - ([#;BoolS #Bool Bool] - [#;NatS #Nat Nat] - [#;IntS #Int Int] - [#;DegS #Deg Deg] - [#;RealS #Real Real] - [#;CharS #Char Char] - [#;TextS #Text Text]) - - (^ [cursor (#;TupleS (list))]) - (do @ - [expected macro;expected-type - _ (within-type-env - (TC;check expected Unit))] - (wrap [[Unit cursor] - #Unit])) - - (^ [cursor (#;TupleS (list singleton))]) - (analyse eval singleton) - - (^ [cursor (#;TupleS elems)]) - (do @ - [expected macro;expected-type] - (with-expected-type expected - (analyse-tuple (analyse eval) cursor elems))) - - [cursor (#;SymbolS ["" local-name])] - (do @ - [?local (&env;find local-name)] - (case ?local - (#;Some [actual index]) - (do @ - [expected macro;expected-type - _ (within-type-env - (TC;check expected actual))] - (wrap [[expected cursor] - (#Var index)])) - - #;None - (do @ - [this-module macro;current-module-name] - (analyse eval [cursor (#;SymbolS [this-module local-name])])))) - - [cursor (#;SymbolS def-name)] - (do @ - [expected macro;expected-type - actual (&def;find def-name) - _ (within-type-env - (TC;check expected actual))] - (wrap [[expected cursor] - (#Def def-name)])) - - (^ [cursor (#;FormS (list [_ (#;SymbolS ["" "_lux_check"])] - type - value))]) - (do @ - [expected macro;expected-type - actual (eval Type type) - _ (within-type-env - (TC;check expected actual))] - (with-expected-type actual - (analyse eval value))) - - (^ [cursor (#;FormS (list [_ (#;SymbolS ["" "_lux_coerce"])] - type - value))]) - (do @ - [expected macro;expected-type - actual (eval Type type) - _ (within-type-env - (TC;check expected actual)) - =value (with-expected-type Top - (analyse eval value))] - (wrap (replace-type actual =value))) - - (^ [cursor (#;FormS (list [_ (#;NatS tag)] - value))]) - (analyse-variant (analyse eval) cursor tag value) - - _ - (&;fail (format "Unrecognized syntax: " (%ast ast))) - ))) + (&;fail (format "Unrecognized syntax: " (%ast ast))) + )) -- cgit v1.2.3