author | unc0rr |
Tue, 06 Dec 2011 17:59:10 +0400 | |
changeset 6509 | 648caa66991b |
parent 6499 | 33180b479efa |
child 6511 | bc6e67598dde |
permissions | -rw-r--r-- |
6273 | 1 |
module Pas2C where |
2 |
||
3 |
import Text.PrettyPrint.HughesPJ |
|
4 |
import Data.Maybe |
|
6277 | 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 | 16 |
|
6467 | 17 |
import PascalParser |
18 |
import PascalUnitSyntaxTree |
|
6273 | 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 | 23 |
s <- flip execStateT initState $ f fn |
6467 | 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 | 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 | 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 | 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 | 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 | 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 | 53 |
|
6467 | 54 |
toCFiles :: (String, PascalUnit) -> IO () |
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 | 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 | 86 |
|
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 | 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 | 93 |
|
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 | 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 | 102 |
uses2List (Uses ids) = map (\(Identifier i _) -> i) ids |
6273 | 103 |
|
6509 | 104 |
id2C :: Bool -> Identifier -> Reader a Doc |
105 |
id2C isDecl (Identifier i _) = return $ text i |
|
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 | 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 | 111 |
n <- id2C True name |
112 |
return $ t <+> n <> parens p <> text ";" |
|
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 | 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 | 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 | 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 | 133 |
tvar2C _ (VarDeclaration isConst (ids, t) mInitExpr) = do |
134 |
t' <- type2C t |
|
135 |
i <- mapM (id2C True) ids |
|
136 |
ie <- initExpr mInitExpr |
|
137 |
return $ if isConst then text "const" else empty |
|
138 |
<+> t' |
|
139 |
<+> (hsep . punctuate (char ',') $ i) |
|
140 |
<+> ie |
|
141 |
<> text ";" |
|
6355 | 142 |
where |
6509 | 143 |
initExpr Nothing = return $ empty |
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 | 146 |
tvar2C f (FunctionDeclaration (Identifier ("<op " ++ op ++ ">") Unknown) ret params body) |
6355 | 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 | 149 |
initExpr2C (InitBinOp op expr1 expr2) = do |
150 |
e1 <- initExpr2C expr1 |
|
151 |
e2 <- initExpr2C expr2 |
|
152 |
o <- op2C op |
|
153 |
return $ parens $ e1 <+> o <+> e2 |
|
154 |
initExpr2C (InitNumber s) = return $ text s |
|
155 |
initExpr2C (InitFloat s) = return $ text s |
|
156 |
initExpr2C (InitHexNumber s) = return $ text "0x" <> (text . map toLower $ s) |
|
157 |
initExpr2C (InitString s) = return $ doubleQuotes $ text s |
|
158 |
initExpr2C (InitReference i) = id2C False i |
|
159 |
initExpr2C _ = return $ text "<<expression>>" |
|
6391 | 160 |
|
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 | 163 |
type2C UnknownType = return $ text "void" |
164 |
type2C (String l) = return $ text $ "string" ++ show l |
|
165 |
type2C (SimpleType i) = id2C True i |
|
166 |
type2C (PointerTo t) = liftM (<> text "*") $ type2C t |
|
167 |
type2C (RecordType tvs union) = do |
|
168 |
t <- mapM (tvar2C False) tvs |
|
169 |
return $ text "{" $+$ (nest 4 . vcat $ t) $+$ text "}" |
|
170 |
type2C (RangeType r) = return $ text "<<range type>>" |
|
171 |
type2C (Sequence ids) = return $ text "<<sequence type>>" |
|
172 |
type2C (ArrayDecl r t) = return $ text "<<array type>>" |
|
173 |
type2C (Set t) = return $ text "<<set>>" |
|
174 |
type2C (FunctionType returnType params) = return $ text "<<function>>" |
|
6273 | 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 | 177 |
phrase2C (Phrases p) = do |
178 |
ps <- mapM phrase2C p |
|
179 |
return $ text "{" $+$ (nest 4 . vcat $ ps) $+$ text "}" |
|
180 |
phrase2C (ProcCall f@(FunCall {}) []) = liftM (<> semi) $ ref2C f |
|
181 |
phrase2C (ProcCall ref params) = do |
|
182 |
r <- ref2C ref |
|
183 |
ps <- mapM expr2C params |
|
184 |
return $ r <> parens (hsep . punctuate (char ',') $ ps) <> semi |
|
185 |
phrase2C (IfThenElse (expr) phrase1 mphrase2) = do |
|
186 |
e <- expr2C expr |
|
187 |
p1 <- (phrase2C . wrapPhrase) phrase1 |
|
188 |
el <- elsePart |
|
189 |
return $ |
|
190 |
text "if" <> parens e $+$ p1 $+$ el |
|
6273 | 191 |
where |
6509 | 192 |
elsePart | isNothing mphrase2 = return $ empty |
193 |
| otherwise = liftM (text "else" $$) $ (phrase2C . wrapPhrase) (fromJust mphrase2) |
|
194 |
phrase2C (Assignment ref expr) = do |
|
195 |
r <- ref2C ref |
|
196 |
e <- expr2C expr |
|
197 |
return $ |
|
198 |
r <> text " = " <> e <> semi |
|
199 |
phrase2C (WhileCycle expr phrase) = do |
|
200 |
e <- expr2C expr |
|
201 |
p <- phrase2C $ wrapPhrase phrase |
|
202 |
return $ text "while" <> parens e $$ p |
|
203 |
phrase2C (SwitchCase expr cases mphrase) = do |
|
204 |
e <- expr2C expr |
|
205 |
cs <- mapM case2C cases |
|
206 |
return $ |
|
207 |
text "switch" <> parens e <> text "of" $+$ (nest 4 . vcat) cs |
|
6273 | 208 |
where |
6509 | 209 |
case2C :: ([InitExpression], Phrase) -> Reader a Doc |
210 |
case2C (e, p) = do |
|
211 |
ie <- mapM initExpr2C e |
|
212 |
ph <- phrase2C p |
|
213 |
return $ |
|
214 |
text "case" <+> parens (hsep . punctuate (char ',') $ ie) <> char ':' <> nest 4 (ph $+$ text "break;") |
|
215 |
phrase2C (WithBlock ref p) = do |
|
216 |
r <- ref2C ref |
|
217 |
ph <- phrase2C $ wrapPhrase p |
|
218 |
return $ text "namespace" <> parens r $$ ph |
|
219 |
phrase2C (ForCycle i' e1' e2' p) = do |
|
220 |
i <- id2C False i' |
|
221 |
e1 <- expr2C e1' |
|
222 |
e2 <- expr2C e2' |
|
223 |
ph <- phrase2C (wrapPhrase p) |
|
224 |
return $ |
|
225 |
text "for" <> (parens . hsep . punctuate (char ';') $ [i <+> text "=" <+> e1, i <+> text "<=" <+> e2, text "++" <> i]) |
|
226 |
$$ |
|
227 |
ph |
|
228 |
phrase2C (RepeatCycle e' p') = do |
|
229 |
e <- expr2C e' |
|
230 |
p <- phrase2C (Phrases p') |
|
231 |
return $ text "do" <+> p <+> text "while" <> parens (text "!" <> parens e) |
|
232 |
phrase2C NOP = return $ text ";" |
|
6355 | 233 |
|
6273 | 234 |
|
6307 | 235 |
wrapPhrase p@(Phrases _) = p |
236 |
wrapPhrase p = Phrases [p] |
|
6273 | 237 |
|
6355 | 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 | 240 |
expr2C (Expression s) = return $ text s |
241 |
expr2C (BinOp op expr1 expr2) = do |
|
242 |
e1 <- expr2C expr1 |
|
243 |
e2 <- expr2C expr2 |
|
244 |
o <- op2C op |
|
245 |
return $ parens $ e1 <+> o <+> e2 |
|
246 |
expr2C (NumberLiteral s) = return $ text s |
|
247 |
expr2C (FloatLiteral s) = return $ text s |
|
248 |
expr2C (HexNumber s) = return $ text "0x" <> (text . map toLower $ s) |
|
249 |
expr2C (StringLiteral s) = return $ doubleQuotes $ text s |
|
6277 | 250 |
expr2C (Reference ref) = ref2C ref |
6509 | 251 |
expr2C (PrefixOp op expr) = liftM2 (<+>) (op2C op) (expr2C expr) |
252 |
expr2C Null = return $ text "NULL" |
|
253 |
expr2C (BuiltInFunCall params ref) = do |
|
254 |
r <- ref2C ref |
|
255 |
ps <- mapM expr2C params |
|
256 |
return $ |
|
257 |
r <> parens (hsep . punctuate (char ',') $ ps) |
|
258 |
expr2C _ = return $ text "<<expression>>" |
|
6273 | 259 |
|
6307 | 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 | 262 |
ref2C (ArrayElement exprs ref) = do |
263 |
r <- ref2C ref |
|
264 |
es <- mapM expr2C exprs |
|
265 |
return $ r <> (brackets . hcat) (punctuate comma es) |
|
266 |
ref2C (SimpleReference name) = id2C False name |
|
267 |
ref2C (RecordField (Dereference ref1) ref2) = do |
|
268 |
r1 <- ref2C ref1 |
|
269 |
r2 <- ref2C ref2 |
|
270 |
return $ |
|
271 |
r1 <> text "->" <> r2 |
|
272 |
ref2C (RecordField ref1 ref2) = do |
|
273 |
r1 <- ref2C ref1 |
|
274 |
r2 <- ref2C ref2 |
|
275 |
return $ |
|
276 |
r1 <> text "." <> r2 |
|
277 |
ref2C (Dereference ref) = liftM ((parens $ text "*") <>) $ ref2C ref |
|
278 |
ref2C (FunCall params ref) = do |
|
279 |
r <- ref2C ref |
|
280 |
ps <- mapM expr2C params |
|
281 |
return $ |
|
282 |
r <> parens (hsep . punctuate (char ',') $ ps) |
|
283 |
ref2C (Address ref) = do |
|
284 |
r <- ref2C ref |
|
285 |
return $ text "&" <> parens r |
|
286 |
ref2C (TypeCast t' expr) = do |
|
287 |
t <- id2C False t' |
|
288 |
e <- expr2C expr |
|
289 |
return $ parens t <> e |
|
6467 | 290 |
ref2C (RefExpression expr) = expr2C expr |
6355 | 291 |
|
6509 | 292 |
|
293 |
op2C :: String -> Reader a Doc |
|
294 |
op2C "or" = return $ text "|" |
|
295 |
op2C "and" = return $ text "&" |
|
296 |
op2C "not" = return $ text "!" |
|
297 |
op2C "xor" = return $ text "^" |
|
298 |
op2C "div" = return $ text "/" |
|
299 |
op2C "mod" = return $ text "%" |
|
300 |
op2C "shl" = return $ text "<<" |
|
301 |
op2C "shr" = return $ text ">>" |
|
302 |
op2C "<>" = return $ text "!=" |
|
303 |
op2C "=" = return $ text "==" |
|
304 |
op2C a = return $ text a |
|
6273 | 305 |
|
306 |
maybeVoid "" = "void" |
|
307 |
maybeVoid a = a |