tools/pas2c.hs
author unc0rr
Tue, 06 Dec 2011 17:59:10 +0400
changeset 6509 648caa66991b
parent 6499 33180b479efa
child 6511 bc6e67598dde
permissions -rw-r--r--
Convert into Reader monad
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
     1
module Pas2C where
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
     2
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
     3
import Text.PrettyPrint.HughesPJ
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
     4
import Data.Maybe
6277
627b5752733a A try to improve parser move (has regressions)
unc0rr
parents: 6275
diff changeset
     5
import Data.Char
6417
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
     6
import Text.Parsec.Prim
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
     7
import Control.Monad.State
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
     8
import System.IO
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
     9
import System.Directory
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    10
import Control.Monad.IO.Class
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    11
import PascalPreprocessor
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    12
import Control.Exception
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    13
import System.IO.Error
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    14
import qualified Data.Map as Map
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    15
import Control.Monad.Reader
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
    16
6467
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    17
import PascalParser
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    18
import PascalUnitSyntaxTree
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
    19
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    20
pas2C :: String -> IO ()
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    21
pas2C fn = do
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    22
    setCurrentDirectory "../hedgewars/"
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    23
    s <- flip execStateT initState $ f fn
6467
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    24
    mapM_ toCFiles (Map.toList s)
6417
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    25
    where
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    26
    printLn = liftIO . hPutStrLn stderr
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    27
    print = liftIO . hPutStr stderr
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    28
    initState = Map.empty
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    29
    f :: String -> StateT (Map.Map String PascalUnit) IO ()
6417
eae5900fd8a4 Improve parser a bit, preparation to parsing whole program at once and compiling it into single C file
unc0rr
parents: 6399
diff changeset
    30
    f fileName = do
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    31
        processed <- gets $ Map.member fileName
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    32
        unless processed $ do
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    33
            print ("Preprocessing '" ++ fileName ++ ".pas'... ")
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    34
            fc' <- liftIO 
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    35
                $ tryJust (guard . isDoesNotExistError) 
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    36
                $ preprocess (fileName ++ ".pas")
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    37
            case fc' of
6453
11c578d30bd3 Countless imporvements to the parser and countless help to the parser in sources.
unc0rr
parents: 6450
diff changeset
    38
                (Left a) -> do
11c578d30bd3 Countless imporvements to the parser and countless help to the parser in sources.
unc0rr
parents: 6450
diff changeset
    39
                    modify (Map.insert fileName System)
11c578d30bd3 Countless imporvements to the parser and countless help to the parser in sources.
unc0rr
parents: 6450
diff changeset
    40
                    printLn "doesn't exist"
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    41
                (Right fc) -> do
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    42
                    print "ok, parsing... "
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    43
                    let ptree = parse pascalUnit fileName fc
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    44
                    case ptree of
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    45
                         (Left a) -> do
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    46
                            liftIO $ writeFile "preprocess.out" fc
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    47
                            printLn $ show a ++ "\nsee preprocess.out for preprocessed source"
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    48
                            fail "stop"
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    49
                         (Right a) -> do
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    50
                            printLn "ok"
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    51
                            modify (Map.insert fileName a)
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    52
                            mapM_ f (usesFiles a)
6455
d2b13364eddd More verbose progress log, dump the result
unc0rr
parents: 6453
diff changeset
    53
6467
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    54
toCFiles :: (String, PascalUnit) -> IO ()
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    55
toCFiles (_, System) = return ()
6474
42e9773eedfd - Improve renderer a bit, disallow nested functions
unc0rr
parents: 6467
diff changeset
    56
toCFiles p@(fn, pu) = do
42e9773eedfd - Improve renderer a bit, disallow nested functions
unc0rr
parents: 6467
diff changeset
    57
    hPutStrLn stderr $ "Rendering '" ++ fn ++ "'..."
42e9773eedfd - Improve renderer a bit, disallow nested functions
unc0rr
parents: 6467
diff changeset
    58
    toCFiles' p
42e9773eedfd - Improve renderer a bit, disallow nested functions
unc0rr
parents: 6467
diff changeset
    59
    where
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    60
    toCFiles' (fn, p@(Program {})) = writeFile (fn ++ ".c") $ (render2C . pascal2C) p
6474
42e9773eedfd - Improve renderer a bit, disallow nested functions
unc0rr
parents: 6467
diff changeset
    61
    toCFiles' (fn, (Unit _ interface implementation _ _)) = do
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    62
        writeFile (fn ++ ".h") $ "#pragma once\n" ++ (render2C . interface2C $ interface)
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    63
        writeFile (fn ++ ".c") $ (render2C . implementation2C) implementation
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    64
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    65
system :: [(String, String)]
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    66
system = []
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    67
        
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    68
render2C = render . flip runReader system
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    69
6467
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    70
usesFiles :: PascalUnit -> [String]
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    71
usesFiles (Program _ (Implementation uses _) _) = uses2List uses
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    72
usesFiles (Unit _ (Interface uses1 _) (Implementation uses2 _) _ _) = uses2List uses1 ++ uses2List uses2
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    73
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    74
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
    75
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    76
pascal2C :: PascalUnit -> Reader a Doc
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    77
pascal2C (Unit _ interface implementation init fin) =
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    78
    liftM2 ($+$) (interface2C interface) (implementation2C implementation)
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    79
    
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    80
pascal2C (Program _ implementation mainFunction) = do
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    81
    impl <- implementation2C implementation
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    82
    main <- tvar2C True 
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    83
        (FunctionDeclaration (Identifier "main" BTInt) (SimpleType $ Identifier "int" BTInt) [] (Just (TypesAndVars [], mainFunction)))
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    84
    return $ impl $+$ main
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    85
6467
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    86
    
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
    87
    
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    88
interface2C :: Interface -> Reader a Doc
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    89
interface2C (Interface uses tvars) = liftM2 ($+$) (uses2C uses) (typesAndVars2C True tvars)
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
    90
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    91
implementation2C :: Implementation -> Reader a Doc
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    92
implementation2C (Implementation uses tvars) = liftM2 ($+$) (uses2C uses) (typesAndVars2C True tvars)
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
    93
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
    94
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    95
typesAndVars2C :: Bool -> TypesAndVars -> Reader a Doc
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    96
typesAndVars2C b (TypesAndVars ts) = liftM vcat $ mapM (tvar2C b) ts
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
    97
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    98
uses2C :: Uses -> Reader a Doc
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
    99
uses2C uses = return $ vcat . map (\i -> text $ "#include \"" ++ i ++ ".h\"") $ uses2List uses
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
   100
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
   101
uses2List :: Uses -> [String]
6489
e1f0058cfedd Add base type tags to identifiers
unc0rr
parents: 6474
diff changeset
   102
uses2List (Uses ids) = map (\(Identifier i _) -> i) ids
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   103
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   104
id2C :: Bool -> Identifier -> Reader a Doc
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   105
id2C isDecl (Identifier i _) = return $ text i
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   106
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   107
tvar2C :: Bool -> TypeVarDeclaration -> Reader a Doc
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   108
tvar2C _ (FunctionDeclaration name returnType params Nothing) = do
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   109
    t <- type2C returnType 
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   110
    p <- liftM hcat $ mapM (tvar2C False) params
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   111
    n <- id2C True name
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   112
    return $ t <+> n <> parens p <> text ";"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   113
tvar2C True (FunctionDeclaration name returnType params (Just (tvars, phrase))) = do
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   114
    t <- type2C returnType 
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   115
    p <- liftM hcat $ mapM (tvar2C False) params
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   116
    ph <- liftM2 ($+$) (typesAndVars2C False tvars) (phrase2C' phrase)
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   117
    n <- id2C True name
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   118
    return $ 
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   119
        t <+> n <> parens p
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
   120
        $+$
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   121
        text "{" 
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   122
        $+$ 
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   123
        nest 4 ph
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   124
        $+$
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   125
        text "}"
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
   126
    where
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   127
    phrase2C' (Phrases p) = liftM vcat $ mapM phrase2C p
6425
1ef4192aa80d - Parse unions, sets, function type, packed arrays and some more imporvements to the parser. Now it parses uVariable, uConsts and even SDLh.pas
unc0rr
parents: 6417
diff changeset
   128
    phrase2C' p = phrase2C p
6489
e1f0058cfedd Add base type tags to identifiers
unc0rr
parents: 6474
diff changeset
   129
tvar2C False (FunctionDeclaration (Identifier name _) _ _ _) = error $ "nested functions not allowed: " ++ name
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   130
tvar2C _ (TypeDeclaration (Identifier i _) t) = do
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   131
    tp <- type2C t
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   132
    return $ text "type" <+> text i <+> tp <> text ";"
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   133
tvar2C _ (VarDeclaration isConst (ids, t) mInitExpr) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   134
    t' <- type2C t
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   135
    i <- mapM (id2C True) ids
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   136
    ie <- initExpr mInitExpr
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   137
    return $ if isConst then text "const" else empty
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   138
        <+> t'
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   139
        <+> (hsep . punctuate (char ',') $ i)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   140
        <+> ie
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   141
        <> text ";"
6355
734fed7aefd3 Introduce initialization expressions
unc0rr
parents: 6317
diff changeset
   142
    where
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   143
    initExpr Nothing = return $ empty
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   144
    initExpr (Just e) = liftM (text "=" <+>) (initExpr2C e)
6474
42e9773eedfd - Improve renderer a bit, disallow nested functions
unc0rr
parents: 6467
diff changeset
   145
tvar2C f (OperatorDeclaration op _ ret params body) = 
6489
e1f0058cfedd Add base type tags to identifiers
unc0rr
parents: 6474
diff changeset
   146
    tvar2C f (FunctionDeclaration (Identifier ("<op " ++ op ++ ">") Unknown) ret params body)
6355
734fed7aefd3 Introduce initialization expressions
unc0rr
parents: 6317
diff changeset
   147
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   148
initExpr2C :: InitExpression -> Reader a Doc
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   149
initExpr2C (InitBinOp op expr1 expr2) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   150
    e1 <- initExpr2C expr1
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   151
    e2 <- initExpr2C expr2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   152
    o <- op2C op
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   153
    return $ parens $ e1 <+> o <+> e2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   154
initExpr2C (InitNumber s) = return $ text s
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   155
initExpr2C (InitFloat s) = return $ text s
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   156
initExpr2C (InitHexNumber s) = return $ text "0x" <> (text . map toLower $ s)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   157
initExpr2C (InitString s) = return $ doubleQuotes $ text s 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   158
initExpr2C (InitReference i) = id2C False i
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   159
initExpr2C _ = return $ text "<<expression>>"
6391
bd5851ab3157 - Parse sets initialization
unc0rr
parents: 6355
diff changeset
   160
bd5851ab3157 - Parse sets initialization
unc0rr
parents: 6355
diff changeset
   161
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   162
type2C :: TypeDecl -> Reader a Doc
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   163
type2C UnknownType = return $ text "void"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   164
type2C (String l) = return $ text $ "string" ++ show l
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   165
type2C (SimpleType i) = id2C True i
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   166
type2C (PointerTo t) = liftM (<> text "*") $ type2C t
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   167
type2C (RecordType tvs union) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   168
    t <- mapM (tvar2C False) tvs
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   169
    return $ text "{" $+$ (nest 4 . vcat $ t) $+$ text "}"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   170
type2C (RangeType r) = return $ text "<<range type>>"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   171
type2C (Sequence ids) = return $ text "<<sequence type>>"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   172
type2C (ArrayDecl r t) = return $ text "<<array type>>"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   173
type2C (Set t) = return $ text "<<set>>"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   174
type2C (FunctionType returnType params) = return $ text "<<function>>"
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   175
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   176
phrase2C :: Phrase -> Reader a Doc
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   177
phrase2C (Phrases p) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   178
    ps <- mapM phrase2C p
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   179
    return $ text "{" $+$ (nest 4 . vcat $ ps) $+$ text "}"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   180
phrase2C (ProcCall f@(FunCall {}) []) = liftM (<> semi) $ ref2C f
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   181
phrase2C (ProcCall ref params) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   182
    r <- ref2C ref
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   183
    ps <- mapM expr2C params
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   184
    return $ r <> parens (hsep . punctuate (char ',') $ ps) <> semi
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   185
phrase2C (IfThenElse (expr) phrase1 mphrase2) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   186
    e <- expr2C expr
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   187
    p1 <- (phrase2C . wrapPhrase) phrase1
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   188
    el <- elsePart
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   189
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   190
        text "if" <> parens e $+$ p1 $+$ el
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   191
    where
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   192
    elsePart | isNothing mphrase2 = return $ empty
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   193
             | otherwise = liftM (text "else" $$) $ (phrase2C . wrapPhrase) (fromJust mphrase2)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   194
phrase2C (Assignment ref expr) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   195
    r <- ref2C ref 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   196
    e <- expr2C expr
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   197
    return $
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   198
        r <> text " = " <> e <> semi
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   199
phrase2C (WhileCycle expr phrase) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   200
    e <- expr2C expr
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   201
    p <- phrase2C $ wrapPhrase phrase
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   202
    return $ text "while" <> parens e $$ p
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   203
phrase2C (SwitchCase expr cases mphrase) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   204
    e <- expr2C expr
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   205
    cs <- mapM case2C cases
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   206
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   207
        text "switch" <> parens e <> text "of" $+$ (nest 4 . vcat) cs
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   208
    where
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   209
    case2C :: ([InitExpression], Phrase) -> Reader a Doc
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   210
    case2C (e, p) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   211
        ie <- mapM initExpr2C e
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   212
        ph <- phrase2C p
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   213
        return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   214
            text "case" <+> parens (hsep . punctuate (char ',') $ ie) <> char ':' <> nest 4 (ph $+$ text "break;")
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   215
phrase2C (WithBlock ref p) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   216
    r <- ref2C ref 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   217
    ph <- phrase2C $ wrapPhrase p
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   218
    return $ text "namespace" <> parens r $$ ph
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   219
phrase2C (ForCycle i' e1' e2' p) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   220
    i <- id2C False i'
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   221
    e1 <- expr2C e1'
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   222
    e2 <- expr2C e2'
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   223
    ph <- phrase2C (wrapPhrase p)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   224
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   225
        text "for" <> (parens . hsep . punctuate (char ';') $ [i <+> text "=" <+> e1, i <+> text "<=" <+> e2, text "++" <> i])
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   226
        $$
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   227
        ph
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   228
phrase2C (RepeatCycle e' p') = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   229
    e <- expr2C e'
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   230
    p <- phrase2C (Phrases p')
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   231
    return $ text "do" <+> p <+> text "while" <> parens (text "!" <> parens e)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   232
phrase2C NOP = return $ text ";"
6355
734fed7aefd3 Introduce initialization expressions
unc0rr
parents: 6317
diff changeset
   233
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   234
6307
25cfd9f4a567 Even more improvements to the parser and converter
unc0rr
parents: 6277
diff changeset
   235
wrapPhrase p@(Phrases _) = p
25cfd9f4a567 Even more improvements to the parser and converter
unc0rr
parents: 6277
diff changeset
   236
wrapPhrase p = Phrases [p]
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   237
6355
734fed7aefd3 Introduce initialization expressions
unc0rr
parents: 6317
diff changeset
   238
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   239
expr2C :: Expression -> Reader a Doc
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   240
expr2C (Expression s) = return $ text s
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   241
expr2C (BinOp op expr1 expr2) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   242
    e1 <- expr2C expr1
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   243
    e2 <- expr2C expr2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   244
    o <- op2C op
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   245
    return $ parens $ e1 <+> o <+> e2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   246
expr2C (NumberLiteral s) = return $ text s
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   247
expr2C (FloatLiteral s) = return $ text s
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   248
expr2C (HexNumber s) = return $ text "0x" <> (text . map toLower $ s)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   249
expr2C (StringLiteral s) = return $ doubleQuotes $ text s 
6277
627b5752733a A try to improve parser move (has regressions)
unc0rr
parents: 6275
diff changeset
   250
expr2C (Reference ref) = ref2C ref
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   251
expr2C (PrefixOp op expr) = liftM2 (<+>) (op2C op) (expr2C expr)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   252
expr2C Null = return $ text "NULL"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   253
expr2C (BuiltInFunCall params ref) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   254
    r <- ref2C ref 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   255
    ps <- mapM expr2C params
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   256
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   257
        r <> parens (hsep . punctuate (char ',') $ ps)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   258
expr2C _ = return $ text "<<expression>>"
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   259
6307
25cfd9f4a567 Even more improvements to the parser and converter
unc0rr
parents: 6277
diff changeset
   260
6499
33180b479efa Start converting into monadic code using Reader monad (will be used to store information about namespace)
unc0rr
parents: 6489
diff changeset
   261
ref2C :: Reference -> Reader a Doc
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   262
ref2C (ArrayElement exprs ref) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   263
    r <- ref2C ref 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   264
    es <- mapM expr2C exprs
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   265
    return $ r <> (brackets . hcat) (punctuate comma es)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   266
ref2C (SimpleReference name) = id2C False name
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   267
ref2C (RecordField (Dereference ref1) ref2) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   268
    r1 <- ref2C ref1 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   269
    r2 <- ref2C ref2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   270
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   271
        r1 <> text "->" <> r2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   272
ref2C (RecordField ref1 ref2) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   273
    r1 <- ref2C ref1 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   274
    r2 <- ref2C ref2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   275
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   276
        r1 <> text "." <> r2
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   277
ref2C (Dereference ref) = liftM ((parens $ text "*") <>) $ ref2C ref
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   278
ref2C (FunCall params ref) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   279
    r <- ref2C ref
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   280
    ps <- mapM expr2C params
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   281
    return $ 
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   282
        r <> parens (hsep . punctuate (char ',') $ ps)
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   283
ref2C (Address ref) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   284
    r <- ref2C ref
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   285
    return $ text "&" <> parens r
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   286
ref2C (TypeCast t' expr) = do
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   287
    t <- id2C False t'
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   288
    e <- expr2C expr
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   289
    return $ parens t <> e
6467
090269e528df - Improve parsing of prefix operators
unc0rr
parents: 6455
diff changeset
   290
ref2C (RefExpression expr) = expr2C expr
6355
734fed7aefd3 Introduce initialization expressions
unc0rr
parents: 6317
diff changeset
   291
6509
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   292
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   293
op2C :: String -> Reader a Doc
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   294
op2C "or" = return $ text "|"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   295
op2C "and" = return $ text "&"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   296
op2C "not" = return $ text "!"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   297
op2C "xor" = return $ text "^"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   298
op2C "div" = return $ text "/"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   299
op2C "mod" = return $ text "%"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   300
op2C "shl" = return $ text "<<"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   301
op2C "shr" = return $ text ">>"
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   302
op2C "<>" = return $ text "!="
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   303
op2C "=" = return $ text "=="
648caa66991b Convert into Reader monad
unc0rr
parents: 6499
diff changeset
   304
op2C a = return $ text a
6273
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   305
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   306
maybeVoid "" = "void"
13262c6e5027 Starting pas2C using library called 'pretty'
unc0rr
parents:
diff changeset
   307
maybeVoid a = a