author | unc0rr |
Wed, 23 Nov 2011 20:12:16 +0300 | |
changeset 6414 | 8474b7fa84d6 |
parent 6399 | a904c735979c |
child 6417 | eae5900fd8a4 |
permissions | -rw-r--r-- |
6273 | 1 |
module Pas2C where |
2 |
||
3 |
import PascalParser |
|
4 |
import Text.PrettyPrint.HughesPJ |
|
5 |
import Data.Maybe |
|
6277 | 6 |
import Data.Char |
6355 | 7 |
import Text.Parsec.String |
6273 | 8 |
|
9 |
||
6355 | 10 |
pas2C :: String -> IO String |
11 |
pas2C fileName = do |
|
12 |
ptree <- parseFromFile pascalUnit fileName |
|
13 |
case ptree of |
|
14 |
(Left a) -> return (show a) |
|
15 |
(Right a) -> (return . render . pascal2C) a |
|
16 |
||
6273 | 17 |
pascal2C :: PascalUnit -> Doc |
6391 | 18 |
pascal2C (Unit unitName interface implementation init fin) = |
19 |
interface2C interface |
|
20 |
$+$ |
|
21 |
implementation2C implementation |
|
6273 | 22 |
|
6391 | 23 |
interface2C :: Interface -> Doc |
24 |
interface2C (Interface uses tvars) = typesAndVars2C tvars |
|
6273 | 25 |
|
26 |
implementation2C :: Implementation -> Doc |
|
27 |
implementation2C (Implementation uses tvars) = typesAndVars2C tvars |
|
28 |
||
29 |
||
30 |
typesAndVars2C :: TypesAndVars -> Doc |
|
31 |
typesAndVars2C (TypesAndVars ts) = vcat $ map tvar2C ts |
|
32 |
||
33 |
||
34 |
tvar2C :: TypeVarDeclaration -> Doc |
|
6307 | 35 |
tvar2C (FunctionDeclaration (Identifier name) returnType Nothing) = |
36 |
type2C returnType <+> text (name ++ "();") |
|
6399 | 37 |
tvar2C (FunctionDeclaration (Identifier name) returnType (Just (tvars, phrase))) = |
6307 | 38 |
type2C returnType <+> text (name ++ "()") |
6273 | 39 |
$$ |
6399 | 40 |
text "{" $+$ (nest 4 $ typesAndVars2C tvars) |
41 |
$+$ |
|
6273 | 42 |
phrase2C phrase |
6399 | 43 |
$+$ |
44 |
text "}" |
|
6355 | 45 |
tvar2C (TypeDeclaration (Identifier i) t) = text "type" <+> text i <+> type2C t <> text ";" |
46 |
tvar2C (VarDeclaration isConst (ids, t) mInitExpr) = |
|
47 |
if isConst then text "const" else empty |
|
48 |
<+> |
|
49 |
type2C t |
|
50 |
<+> |
|
51 |
(hsep . punctuate (char ',') . map (\(Identifier i) -> text i) $ ids) |
|
52 |
<+> |
|
53 |
initExpr mInitExpr |
|
54 |
<> |
|
55 |
text ";" |
|
56 |
where |
|
57 |
initExpr Nothing = empty |
|
58 |
initExpr (Just e) = text "=" <+> initExpr2C e |
|
59 |
||
6391 | 60 |
initExpr2C :: InitExpression -> Doc |
61 |
initExpr2C (InitBinOp op expr1 expr2) = parens $ (initExpr2C expr1) <+> op2C op <+> (initExpr2C expr2) |
|
62 |
initExpr2C (InitNumber s) = text s |
|
63 |
initExpr2C (InitFloat s) = text s |
|
64 |
initExpr2C (InitHexNumber s) = text "0x" <> (text . map toLower $ s) |
|
65 |
initExpr2C (InitString s) = doubleQuotes $ text s |
|
66 |
initExpr2C (InitReference (Identifier i)) = text i |
|
67 |
||
68 |
||
6355 | 69 |
initExpr2C _ = text "<<expression>>" |
6273 | 70 |
|
6307 | 71 |
type2C :: TypeDecl -> Doc |
72 |
type2C UnknownType = text "void" |
|
6399 | 73 |
type2C (String l) = text $ "string" ++ show l |
6355 | 74 |
type2C (SimpleType (Identifier i)) = text i |
75 |
type2C (PointerTo t) = type2C t <> text "*" |
|
76 |
type2C (RecordType tvs) = text "{" $+$ (nest 4 . vcat . map tvar2C $ tvs) $+$ text "}" |
|
77 |
type2C (RangeType r) = text "<<range type>>" |
|
78 |
type2C (Sequence ids) = text "<<sequence type>>" |
|
79 |
type2C (ArrayDecl r t) = text "<<array type>>" |
|
80 |
||
6273 | 81 |
|
82 |
phrase2C :: Phrase -> Doc |
|
6307 | 83 |
phrase2C (Phrases p) = text "{" $+$ (nest 4 . vcat . map phrase2C $ p) $+$ text "}" |
6273 | 84 |
phrase2C (ProcCall (Identifier name) params) = text name <> parens (hsep . punctuate (char ',') . map expr2C $ params) <> semi |
6307 | 85 |
phrase2C (IfThenElse (expr) phrase1 mphrase2) = text "if" <> parens (expr2C expr) $+$ (phrase2C . wrapPhrase) phrase1 $+$ elsePart |
6273 | 86 |
where |
87 |
elsePart | isNothing mphrase2 = empty |
|
6307 | 88 |
| otherwise = text "else" $$ (phrase2C . wrapPhrase) (fromJust mphrase2) |
6277 | 89 |
phrase2C (Assignment ref expr) = ref2C ref <> text " = " <> expr2C expr <> semi |
6307 | 90 |
phrase2C (WhileCycle expr phrase) = text "while" <> parens (expr2C expr) $$ (phrase2C $ wrapPhrase phrase) |
91 |
phrase2C (SwitchCase expr cases mphrase) = text "switch" <> parens (expr2C expr) <> text "of" $+$ (nest 4 . vcat . map case2C) cases |
|
6273 | 92 |
where |
93 |
case2C :: (Expression, Phrase) -> Doc |
|
6307 | 94 |
case2C (e, p) = text "case" <+> parens (expr2C e) <> char ':' <> nest 4 (phrase2C p $+$ text "break;") |
6355 | 95 |
phrase2C (WithBlock ref p) = text "namespace" <> parens (ref2C ref) $$ (phrase2C $ wrapPhrase p) |
96 |
phrase2C (ForCycle (Identifier i) e1 e2 p) = |
|
97 |
text "for" <> (parens . hsep . punctuate (char ';') $ [text i <+> text "=" <+> expr2C e1, text i <+> text "<=" <+> expr2C e2, text "++" <> text i]) |
|
98 |
$$ |
|
99 |
phrase2C (wrapPhrase p) |
|
100 |
phrase2C (RepeatCycle e p) = text "do" <+> phrase2C (Phrases p) <+> text "while" <> parens (text "!" <> parens (expr2C e)) |
|
101 |
||
6273 | 102 |
|
6307 | 103 |
wrapPhrase p@(Phrases _) = p |
104 |
wrapPhrase p = Phrases [p] |
|
6273 | 105 |
|
6355 | 106 |
|
6273 | 107 |
expr2C :: Expression -> Doc |
108 |
expr2C (Expression s) = text s |
|
6307 | 109 |
expr2C (BinOp op expr1 expr2) = parens $ (expr2C expr1) <+> op2C op <+> (expr2C expr2) |
6277 | 110 |
expr2C (NumberLiteral s) = text s |
111 |
expr2C (HexNumber s) = text "0x" <> (text . map toLower $ s) |
|
112 |
expr2C (StringLiteral s) = doubleQuotes $ text s |
|
113 |
expr2C (Reference ref) = ref2C ref |
|
6307 | 114 |
expr2C (PrefixOp op expr) = op2C op <+> expr2C expr |
115 |
{- |
|
6277 | 116 |
| PostfixOp String Expression |
117 |
| CharCode String |
|
6273 | 118 |
-} |
119 |
expr2C _ = empty |
|
120 |
||
6307 | 121 |
|
122 |
ref2C :: Reference -> Doc |
|
6317
83b93a2d2741
Improve parsing of complex references like "a^[b[c], d]"
unc0rr
parents:
6307
diff
changeset
|
123 |
ref2C (ArrayElement exprs ref) = ref2C ref <> (brackets . hcat) (punctuate comma $ map expr2C exprs) |
6307 | 124 |
ref2C (SimpleReference (Identifier name)) = text name |
125 |
ref2C (RecordField (Dereference ref1) ref2) = ref2C ref1 <> text "->" <> ref2C ref2 |
|
126 |
ref2C (RecordField ref1 ref2) = ref2C ref1 <> text "." <> ref2C ref2 |
|
127 |
ref2C (Dereference ref) = parens $ text "*" <> ref2C ref |
|
6317
83b93a2d2741
Improve parsing of complex references like "a^[b[c], d]"
unc0rr
parents:
6307
diff
changeset
|
128 |
ref2C (FunCall params ref) = ref2C ref <> parens (hsep . punctuate (char ',') . map expr2C $ params) |
83b93a2d2741
Improve parsing of complex references like "a^[b[c], d]"
unc0rr
parents:
6307
diff
changeset
|
129 |
ref2C (Address ref) = text "&" <> ref2C ref |
6307 | 130 |
|
6355 | 131 |
|
6275 | 132 |
op2C "or" = text "|" |
133 |
op2C "and" = text "&" |
|
6307 | 134 |
op2C "not" = text "!" |
135 |
op2C "xor" = text "^" |
|
6275 | 136 |
op2C "div" = text "/" |
137 |
op2C "mod" = text "%" |
|
6307 | 138 |
op2C "shl" = text "<<" |
139 |
op2C "shr" = text ">>" |
|
6275 | 140 |
op2C "<>" = text "!=" |
6277 | 141 |
op2C "=" = text "==" |
6275 | 142 |
op2C a = text a |
6273 | 143 |
|
144 |
maybeVoid "" = "void" |
|
145 |
maybeVoid a = a |