working on adapting the typechecker

This commit is contained in:
sebastianselander 2023-03-22 17:52:39 +01:00
parent 936cb1301f
commit 914855e20f
3 changed files with 658 additions and 652 deletions

View file

@ -1,12 +1,13 @@
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * PROGRAM
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
Program. Program ::= [Def] ;
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * TOP-LEVEL
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
DBind. Def ::= Bind ;
DSig. Def ::= Sig ;
@ -16,9 +17,9 @@ Sig. Sig ::= LIdent ":" Type ;
Bind. Bind ::= LIdent [LIdent] "=" Exp ;
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * TYPES
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
TLit. Type2 ::= UIdent ;
TVar. Type2 ::= TVar ;
@ -30,9 +31,9 @@ internal TEVar. Type1 ::= TEVar ;
MkTVar. TVar ::= LIdent ;
internal MkTEVar. TEVar ::= LIdent ;
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * DATA TYPES
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
Constructor. Constructor ::= UIdent ":" Type ;
@ -40,12 +41,12 @@ Indexed. Indexed ::= UIdent "(" [Type] ")" ;
Data. Data ::= "data" Indexed "where" "{" [Constructor] "}" ;
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * EXPRESSIONS
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
EAnn. Exp5 ::= "(" Exp ":" Type ")" ;
EId. Exp4 ::= LIdent ;
EId. Exp4 ::= Ident ;
ELit. Exp4 ::= Lit ;
EApp. Exp3 ::= Exp3 Exp4 ;
EAdd. Exp1 ::= Exp1 "+" Exp2 ;
@ -53,16 +54,16 @@ ELet. Exp ::= "let" LIdent "=" Exp "in" Exp ;
EAbs. Exp ::= "\\" LIdent "." Exp ;
ECase. Exp ::= "case" Exp "of" "{" [Inj] "}";
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * LITERALS
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
LInt. Lit ::= Integer ;
LChar. Lit ::= Char ;
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * CASE
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
Inj. Inj ::= Init "=>" Exp ;
@ -70,9 +71,9 @@ InitLit. Init ::= Lit ;
InitConstructor. Init ::= UIdent [LIdent] ;
InitCatch. Init ::= "_" ;
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
-- * AUX
--------------------------------------------------------------------------------
-------------------------------------------------------------------------------
separator Def ";" ;
separator nonempty Constructor "" ;
@ -80,6 +81,7 @@ separator Type " " ;
separator nonempty Inj ";" ;
separator Ident " ";
separator LIdent " ";
separator TVar " " ;
coercions Exp 5 ;
coercions Type 2 ;

File diff suppressed because it is too large Load diff

View file

@ -2,178 +2,207 @@
module TypeChecker.TypeCheckerIr where
-- import Control.Monad.Except
-- import Control.Monad.Reader
-- import Control.Monad.State
-- import Data.Functor.Identity (Identity)
-- import Data.Map (Map)
-- import Grammar.Abs (
-- Data (..),
-- Ident (..),
-- Init (..),
-- Literal (..),
-- Type (..),
-- )
-- import Grammar.Print
-- import Prelude
-- import Prelude qualified as C (Eq, Ord, Read, Show)
import Control.Monad.Except
import Control.Monad.Reader
import Control.Monad.State
import Data.Functor.Identity (Identity)
import Data.Map (Map)
import Grammar.Abs (
Data (..),
Ident (..),
Init (..),
Lit (..),
TVar (..),
)
import Grammar.Abs qualified as GA (Type (..))
import Grammar.Print
import Prelude
import Prelude qualified as C (Eq, Ord, Read, Show)
-- -- | A data type representing type variables
-- data Poly = Forall [Ident] Type
-- deriving (Show)
-- | A data type representing type variables
data Poly = Forall [Ident] Type
deriving (Show)
-- newtype Ctx = Ctx {vars :: Map Ident Poly}
-- deriving Show
newtype Ctx = Ctx {vars :: Map Ident Poly}
deriving (Show)
-- data Env = Env
-- { count :: Int
-- , sigs :: Map Ident Type
-- , constructors :: Map Ident Type
-- } deriving Show
data Env = Env
{ count :: Int
, sigs :: Map Ident GA.Type
, constructors :: Map Ident GA.Type
}
deriving (Show)
-- type Error = String
-- type Subst = Map Ident Type
type Error = String
type Subst = Map Ident Type
-- type Infer = StateT Env (ReaderT Ctx (ExceptT Error Identity))
type Infer = StateT Env (ReaderT Ctx (ExceptT Error Identity))
-- newtype Program = Program [Def]
-- deriving (C.Eq, C.Ord, C.Show, C.Read)
newtype Program = Program [Def]
deriving (C.Eq, C.Ord, C.Show, C.Read)
-- data Exp
-- = EId Id
-- | ELit Type Literal
-- | ELet Bind Exp
-- | EApp Type Exp Exp
-- | EAdd Type Exp Exp
-- | EAbs Type Id Exp
-- | ECase Type Exp [Inj]
-- deriving (C.Eq, C.Ord, C.Read, C.Show)
data Type
= TLit Ident
| TVar TVar
| TFun Type Type
| TAll TVar Type
| TIndexed Indexed
deriving (Show, Eq, Ord, Read)
-- data Inj = Inj (Init, Type) Exp
-- deriving (C.Eq, C.Ord, C.Read, C.Show)
data Exp
= EId Id
| ELit Lit
| ELet Bind ExpT
| EApp ExpT ExpT
| EAdd ExpT ExpT
| EAbs Id ExpT
| ECase ExpT [Inj]
deriving (C.Eq, C.Ord, C.Read, C.Show)
-- data Def = DBind Bind | DData Data
-- deriving (C.Eq, C.Ord, C.Read, C.Show)
type ExpT = (Exp, Type)
-- type Id = (Ident, Type)
data Indexed = Indexed Ident [Type]
deriving (Show, Read, Ord, Eq)
-- data Bind = Bind Id Exp
-- deriving (C.Eq, C.Ord, C.Show, C.Read)
data Inj = Inj (Init, Type) ExpT
deriving (C.Eq, C.Ord, C.Read, C.Show)
-- instance Print [Def] where
-- prt _ [] = concatD []
-- prt _ (x : xs) = concatD [prt 0 x, doc (showString "\n"), prt 0 xs]
data Def = DBind Bind | DData Data
deriving (C.Eq, C.Ord, C.Read, C.Show)
-- instance Print Def where
-- prt i (DBind bind) = prt i bind
-- prt i (DData d) = prt i d
type Id = (Ident, Type)
-- instance Print Program where
-- prt i (Program sc) = prPrec i 0 $ prt 0 sc
data Bind = Bind Id [Id] ExpT
deriving (C.Eq, C.Ord, C.Show, C.Read)
-- instance Print Bind where
-- prt i (Bind (t, name) rhs) =
-- prPrec i 0 $
-- concatD
-- [ prt 0 name
-- , doc $ showString ":"
-- , prt 0 t
-- , doc $ showString "\n"
-- , prt 0 name
-- , doc $ showString "="
-- , prt 0 rhs
-- ]
instance Print [Def] where
prt _ [] = concatD []
prt _ (x : xs) = concatD [prt 0 x, doc (showString "\n"), prt 0 xs]
-- instance Print [Bind] where
-- prt _ [] = concatD []
-- prt _ [x] = concatD [prt 0 x]
-- prt _ (x : xs) = concatD [prt 0 x, doc (showString ";"), doc (showString "\n"), prt 0 xs]
instance Print Def where
prt i (DBind bind) = prt i bind
prt i (DData d) = prt i d
-- prtIdPs :: Int -> [Id] -> Doc
-- prtIdPs i = prPrec i 0 . concatD . map (prtIdP i)
instance Print Program where
prt i (Program sc) = prPrec i 0 $ prt 0 sc
-- prtId :: Int -> Id -> Doc
-- prtId i (name, t) =
-- prPrec i 0 $
-- concatD
-- [ prt 0 name
-- , doc $ showString ":"
-- , prt 0 t
-- ]
instance Print Bind where
prt i (Bind (t, name) args rhs) =
prPrec i 0 $
concatD
[ prt 0 name
, doc $ showString ":"
, prt 0 t
, doc $ showString "\n"
, prt 0 name
, doc $ showString "="
, prt 0 rhs
]
-- prtIdP :: Int -> Id -> Doc
-- prtIdP i (name, t) =
-- prPrec i 0 $
-- concatD
-- [ doc $ showString "("
-- , prt 0 name
-- , doc $ showString ":"
-- , prt 0 t
-- , doc $ showString ")"
-- ]
instance Print [Bind] where
prt _ [] = concatD []
prt _ [x] = concatD [prt 0 x]
prt _ (x : xs) = concatD [prt 0 x, doc (showString ";"), doc (showString "\n"), prt 0 xs]
-- instance Print Exp where
-- prt i = \case
-- EId n -> prPrec i 3 $ concatD [prtId 0 n, doc $ showString "\n"]
-- ELit _ (LInt i1) -> prPrec i 3 $ concatD [prt 0 i1, doc $ showString "\n"]
-- ELet bs e ->
-- prPrec i 3 $
-- concatD
-- [ doc $ showString "let"
-- , prt 0 bs
-- , doc $ showString "in"
-- , prt 0 e
-- , doc $ showString "\n"
-- ]
-- EApp _ e1 e2 ->
-- prPrec i 2 $
-- concatD
-- [ prt 2 e1
-- , prt 3 e2
-- ]
-- EAdd t e1 e2 ->
-- prPrec i 1 $
-- concatD
-- [ doc $ showString "@"
-- , prt 0 t
-- , prt 1 e1
-- , doc $ showString "+"
-- , prt 2 e2
-- , doc $ showString "\n"
-- ]
-- EAbs t n e ->
-- prPrec i 0 $
-- concatD
-- [ doc $ showString "@"
-- , prt 0 t
-- , doc $ showString "\\"
-- , prtId 0 n
-- , doc $ showString "."
-- , prt 0 e
-- , doc $ showString "\n"
-- ]
-- ECase t exp injs ->
-- prPrec
-- i
-- 0
-- ( concatD
-- [ doc (showString "case")
-- , prt 0 exp
-- , doc (showString "of")
-- , doc (showString "{")
-- , prt 0 injs
-- , doc (showString "}")
-- , doc (showString ":")
-- , prt 0 t
-- , doc $ showString "\n"
-- ]
-- )
prtIdPs :: Int -> [Id] -> Doc
prtIdPs i = prPrec i 0 . concatD . map (prtIdP i)
-- instance Print Inj where
-- prt i = \case
-- Inj (init, t) exp -> prPrec i 0 (concatD [prt 0 init, doc (showString ":"), prt 0 t, doc (showString "=>"), prt 0 exp])
prtId :: Int -> Id -> Doc
prtId i (name, t) =
prPrec i 0 $
concatD
[ prt 0 name
, doc $ showString ":"
, prt 0 t
]
-- instance Print [Inj] where
-- prt _ [] = concatD []
-- prt _ [x] = concatD [prt 0 x]
-- prt _ (x : xs) = concatD [prt 0 x, doc (showString ";"), prt 0 xs]
prtIdP :: Int -> Id -> Doc
prtIdP i (name, t) =
prPrec i 0 $
concatD
[ doc $ showString "("
, prt 0 name
, doc $ showString ":"
, prt 0 t
, doc $ showString ")"
]
instance Print Exp where
prt i = \case
EId n -> prPrec i 3 $ concatD [prtId 0 n, doc $ showString "\n"]
ELit _ lit -> prPrec i 3 $ concatD [prt 0 lit, doc $ showString "\n"]
ELet bs e ->
prPrec i 3 $
concatD
[ doc $ showString "let"
, prt 0 bs
, doc $ showString "in"
, prt 0 e
, doc $ showString "\n"
]
EApp _ e1 e2 ->
prPrec i 2 $
concatD
[ prt 2 e1
, prt 3 e2
]
EAdd t e1 e2 ->
prPrec i 1 $
concatD
[ doc $ showString "@"
, prt 0 t
, prt 1 e1
, doc $ showString "+"
, prt 2 e2
, doc $ showString "\n"
]
EAbs t n e ->
prPrec i 0 $
concatD
[ doc $ showString "@"
, prt 0 t
, doc $ showString "\\"
, prtId 0 n
, doc $ showString "."
, prt 0 e
, doc $ showString "\n"
]
ECase t exp injs ->
prPrec
i
0
( concatD
[ doc (showString "case")
, prt 0 exp
, doc (showString "of")
, doc (showString "{")
, prt 0 injs
, doc (showString "}")
, doc (showString ":")
, prt 0 t
, doc $ showString "\n"
]
)
instance Print ExpT where
prt i (e, t) = concatD [prt i e, doc (showString ":"), prt i t]
instance Print Inj where
prt i = \case
Inj (init, t) exp -> prPrec i 0 (concatD [prt 0 init, doc (showString ":"), prt 0 t, doc (showString "=>"), prt 0 exp])
instance Print [Inj] where
prt _ [] = concatD []
prt _ [x] = concatD [prt 0 x]
prt _ (x : xs) = concatD [prt 0 x, doc (showString ";"), prt 0 xs]
instance Print Type where
prt i = \case
TLit uident -> prPrec i 2 (concatD [prt 0 uident])
TVar tvar -> prPrec i 2 (concatD [prt 0 tvar])
TAll tvar type_ -> prPrec i 1 (concatD [doc (showString "forall"), prt 0 tvar, doc (showString "."), prt 0 type_])
TIndexed indexed -> prPrec i 1 (concatD [prt 0 indexed])
TFun type_1 type_2 -> prPrec i 0 (concatD [prt 1 type_1, doc (showString "->"), prt 0 type_2])
instance Print Indexed where
prt i (Indexed u ts) = concatD [prt i u, prt i ts]