diff -r 9e724f4863a3 -r 6af78154dc62 tools/pas2c.hs --- a/tools/pas2c.hs Sun Apr 01 15:23:34 2012 +0200 +++ b/tools/pas2c.hs Wed May 02 23:53:45 2012 +0200 @@ -1,3 +1,4 @@ +{-# LANGUAGE ScopedTypeVariables #-} module Pas2C where import Text.PrettyPrint.HughesPJ @@ -13,6 +14,7 @@ import System.IO.Error import qualified Data.Map as Map import Data.List (find) +import Numeric import PascalParser import PascalUnitSyntaxTree @@ -29,10 +31,47 @@ currentScope :: [Record], lastIdentifier :: String, lastType :: BaseType, + stringConsts :: [(String, String)], + uniqCounter :: Int, namespaces :: Map.Map String [Record] } -emptyState = RenderState [] "" BTUnknown +emptyState = RenderState [] "" BTUnknown [] 0 + +getUniq :: State RenderState Int +getUniq = do + i <- gets uniqCounter + modify(\s -> s{uniqCounter = uniqCounter s + 1}) + return i + +addStringConst :: String -> State RenderState Doc +addStringConst str = do + strs <- gets stringConsts + let a = find ((==) str . snd) strs + if isJust a then + do + modify (\s -> s{lastType = BTString}) + return . text . fst . fromJust $ a + else + do + i <- getUniq + let sn = "__str" ++ show i + modify (\s -> s{lastType = BTString, stringConsts = (sn, str) : strs}) + return $ text sn + +escapeStr :: String -> String +escapeStr = foldr escapeChar [] + +escapeChar :: Char -> ShowS +escapeChar '"' s = "\\\"" ++ s +escapeChar a s = a : s + +strInit :: String -> Doc +strInit a = text "STRINIT" <> parens (doubleQuotes (text $ escapeStr a)) + +renderStringConsts :: State RenderState Doc +renderStringConsts = liftM (vcat . map (\(a, b) -> text "const string255" <+> (text a) <+> text "=" <+> strInit b <> semi)) + $ gets stringConsts docToLower :: Doc -> Doc docToLower = text . map toLower . render @@ -76,6 +115,8 @@ renderCFiles units = do let u = Map.toList units let nss = Map.map (toNamespace nss) units + hPutStrLn stderr $ "Units: " ++ (show . Map.keys . Map.filter (not . null) $ nss) + --writeFile "pas2c.log" $ unlines . map (\t -> show (fst t) ++ "\n" ++ (unlines . map ((:) '\t' . show) . snd $ t)) . Map.toList $ nss mapM_ (toCFiles nss) u where toNamespace :: Map.Map String [Record] -> PascalUnit -> [Record] @@ -86,10 +127,16 @@ currentScope $ execState (interface2C interface) (emptyState nss) -withState' :: (a -> a) -> State a b -> State a b -withState' f s = do +withState' :: (RenderState -> RenderState) -> State RenderState a -> State RenderState a +withState' f sf = do st <- liftM f get - return $ evalState s st + let (a, s) = runState sf st + modify(\st -> st{ + lastType = lastType s + , uniqCounter = uniqCounter s + , stringConsts = stringConsts s + }) + return a withLastIdNamespace :: State RenderState Doc -> State RenderState Doc withLastIdNamespace f = do @@ -97,11 +144,12 @@ nss <- gets namespaces withState' (\st -> st{currentScope = fromMaybe [] $ Map.lookup li (namespaces st)}) f -withRecordNamespace :: [(String, BaseType)] -> State RenderState Doc -> State RenderState Doc -withRecordNamespace recs = withState' f +withRecordNamespace :: String -> [(String, BaseType)] -> State RenderState Doc -> State RenderState Doc +withRecordNamespace _ [] = error "withRecordNamespace: empty record" +withRecordNamespace prefix recs = withState' f where f st = st{currentScope = records ++ currentScope st} - records = map (\(a, b) -> (map toLower a, (a, b))) recs + records = map (\(a, b) -> (map toLower a, (prefix ++ a, b))) recs toCFiles :: Map.Map String [Record] -> (String, PascalUnit) -> IO () toCFiles _ (_, System _) = return () @@ -110,14 +158,14 @@ toCFiles' p where toCFiles' (fn, p@(Program {})) = writeFile (fn ++ ".c") $ (render2C initialState . pascal2C) p - toCFiles' (fn, (Unit _ interface implementation _ _)) = do - let (a, s) = runState (interface2C interface) initialState - writeFile (fn ++ ".h") $ "#pragma once\n" ++ (render a) - writeFile (fn ++ ".c") $ (render2C s . implementation2C) implementation + toCFiles' (fn, (Unit unitId interface implementation _ _)) = do + let (a, s) = runState (id2C IOInsert (setBaseType BTUnit unitId) >> interface2C interface) initialState + writeFile (fn ++ ".h") $ "#pragma once\n\n#include \"pas2c.h\"\n\n" ++ (render (a $+$ text "")) + writeFile (fn ++ ".c") $ "#include \"" ++ fn ++ ".h\"\n" ++ (render2C s . implementation2C) implementation initialState = emptyState ns render2C :: RenderState -> State RenderState Doc -> String - render2C a = render . flip evalState a + render2C a = render . ($+$ empty) . flip evalState a usesFiles :: PascalUnit -> [String] usesFiles (Program _ (Implementation uses _) _) = "pas2cSystem" : uses2List uses @@ -131,21 +179,29 @@ pascal2C (Program _ implementation mainFunction) = do impl <- implementation2C implementation - main <- tvar2C True + [main] <- tvar2C True (FunctionDeclaration (Identifier "main" BTInt) (SimpleType $ Identifier "int" BTInt) [] (Just (TypesAndVars [], mainFunction))) return $ impl $+$ main interface2C :: Interface -> State RenderState Doc -interface2C (Interface uses tvars) = liftM2 ($+$) (uses2C uses) (typesAndVars2C True tvars) - +interface2C (Interface uses tvars) = do + u <- uses2C uses + tv <- typesAndVars2C True tvars + r <- renderStringConsts + return (u $+$ r $+$ tv) + implementation2C :: Implementation -> State RenderState Doc -implementation2C (Implementation uses tvars) = liftM2 ($+$) (uses2C uses) (typesAndVars2C True tvars) +implementation2C (Implementation uses tvars) = do + u <- uses2C uses + tv <- typesAndVars2C True tvars + r <- renderStringConsts + return (u $+$ r $+$ tv) typesAndVars2C :: Bool -> TypesAndVars -> State RenderState Doc -typesAndVars2C b (TypesAndVars ts) = liftM vcat $ mapM (tvar2C b) ts +typesAndVars2C b (TypesAndVars ts) = liftM (vcat . map (<> semi) . concat) $ mapM (tvar2C b) ts setBaseType :: BaseType -> Identifier -> Identifier setBaseType bt (Identifier i _) = Identifier i bt @@ -158,7 +214,7 @@ where injectNamespace (Identifier i _) = do getNS <- gets (flip Map.lookup . namespaces) - let f = flip (foldl (\a b -> b:a)) (fromMaybe [] (getNS i)) + let f = flip (foldl (flip (:))) (fromMaybe [] (getNS i)) modify (\s -> s{currentScope = f $ currentScope s}) uses2List :: Uses -> [String] @@ -167,6 +223,12 @@ id2C :: InsertOption -> Identifier -> State RenderState Doc id2C IOInsert (Identifier i t) = do + ns <- gets currentScope +{-- case t of + BTUnknown -> do + ns <- gets currentScope + error $ "id2C IOInsert: type BTUnknown for " ++ show i ++ "\nnamespace: " ++ show (take 100 ns) + _ -> do --} modify (\s -> s{currentScope = (n, (i, t)) : currentScope s, lastIdentifier = n}) return $ text i where @@ -175,16 +237,15 @@ let i' = map toLower i v <- gets $ find (\(a, _) -> a == i') . currentScope ns <- gets currentScope + lt <- gets lastType if isNothing v then - error $ "Not defined: '" ++ i' ++ "'\n" -- ++ show ns + error $ "Not defined: '" ++ i' ++ "'\n" ++ show lt ++ "\n" ++ show (take 100 ns) else let vv = snd $ fromJust v in modify (\s -> s{lastType = snd vv, lastIdentifier = fst vv}) >> (return . text . fst $ vv) id2C IODeferred (Identifier i t) = do let i' = map toLower i v <- gets $ find (\(a, _) -> a == i') . currentScope if (isNothing v) then - do - modify (\s -> s{currentScope = (i', (i, t)) : currentScope s}) return $ text i else return . text . fst . snd . fromJust $ v @@ -197,7 +258,8 @@ BTUnknown -> do ns <- gets currentScope error $ "id2CTyped: type BTUnknown for " ++ show i ++ "\ntype: " ++ show t ++ "\nnamespace: " ++ show (take 100 ns) - _ -> id2C IOInsert (Identifier i tb) + _ -> return () + id2C IOInsert (Identifier i tb) resolveType :: TypeDecl -> State RenderState BaseType @@ -221,8 +283,10 @@ where f :: TypeVarDeclaration -> State RenderState [(String, BaseType)] f (VarDeclaration _ (ids, td) _) = mapM (\(Identifier i _) -> liftM ((,) i) $ resolveType td) ids -resolveType (ArrayDecl (Just _) t) = liftM (BTArray BTInt) $ resolveType t -resolveType (ArrayDecl Nothing t) = liftM (BTArray BTInt) $ resolveType t +resolveType (ArrayDecl (Just i) t) = do + t' <- resolveType t + return $ BTArray i BTInt t' +resolveType (ArrayDecl Nothing t) = liftM (BTArray RangeInfinite BTInt) $ resolveType t resolveType (FunctionType t _) = liftM BTFunction $ resolveType t resolveType (DeriveType (InitHexNumber _)) = return BTInt resolveType (DeriveType (InitNumber _)) = return BTInt @@ -236,127 +300,249 @@ resolveType (String _) = return BTString resolveType VoidType = return BTVoid resolveType (Sequence ids) = return $ BTEnum $ map (\(Identifier i _) -> map toLower i) ids -resolveType (RangeType _) = return $ BTUnknown +resolveType (RangeType _) = return $ BTVoid resolveType (Set t) = liftM BTSet $ resolveType t ---resolveType UnknownType = return BTUnknown -resolveType a = error $ "resolveType: " ++ show a - + + +resolve :: String -> BaseType -> State RenderState BaseType +resolve s (BTUnresolved t) = do + v <- gets $ find (\(a, _) -> a == t) . currentScope + if isJust v then + resolve s . snd . snd . fromJust $ v + else + error $ "Unknown type " ++ show t ++ "\n" ++ s +resolve _ t = return t -fromPointer :: BaseType -> State RenderState BaseType -fromPointer (BTPointerTo t) = f t - where - f (BTUnresolved s) = do - v <- gets $ find (\(a, _) -> a == s) . currentScope - if isJust v then - f . snd . snd . fromJust $ v - else - error $ "Unknown type " ++ show t - f t = return t -fromPointer t = error $ "Dereferencing from non-pointer type " ++ show t +fromPointer :: String -> BaseType -> State RenderState BaseType +fromPointer s (BTPointerTo t) = resolve s t +fromPointer s (BTFunctionReturn _ (BTPointerTo t)) = resolve s t +fromPointer s t = do + ns <- gets currentScope + error $ "Dereferencing from non-pointer type " ++ show t ++ "\n" ++ s ++ "\n\n" ++ show (take 100 ns) + + +functionParams2C params = liftM (hcat . punctuate comma . concat) $ mapM (tvar2C False) params - -tvar2C :: Bool -> TypeVarDeclaration -> State RenderState Doc -tvar2C _ (FunctionDeclaration name returnType params Nothing) = do +fun2C :: Bool -> String -> TypeVarDeclaration -> State RenderState [Doc] +fun2C _ _ (FunctionDeclaration name returnType params Nothing) = do t <- type2C returnType - p <- withState' id $ liftM hcat $ mapM (tvar2C False) params - n <- id2C IOInsert name - return $ t <+> n <> parens p <> text ";" + t'<- gets lastType + p <- withState' id $ functionParams2C params + n <- id2C IOInsert $ setBaseType (BTFunction t') name + return [t empty <+> n <> parens p] -tvar2C True (FunctionDeclaration (Identifier i _) returnType params (Just (tvars, phrase))) = do +fun2C True rv (FunctionDeclaration name returnType params (Just (tvars, phrase))) = do + let res = docToLower $ text rv <> text "_result" t <- type2C returnType t'<- gets lastType - n <- id2C IOInsert (Identifier i (BTFunction t')) - (p, ph) <- withState' (\st -> st{currentScope = (lastIdentifier st, (lastIdentifier st ++ "_result", t')) : currentScope st}) $ do - p <- liftM hcat $ mapM (tvar2C False) params + n <- id2C IOInsert $ setBaseType (BTFunction t') name + (p, ph) <- withState' (\st -> st{currentScope = (map toLower rv, (render res, BTFunctionReturn (render n) t')) : currentScope st}) $ do + p <- functionParams2C params ph <- liftM2 ($+$) (typesAndVars2C False tvars) (phrase2C' phrase) return (p, ph) - let res = docToLower $ n <> text "_result" let phrasesBlock = case returnType of VoidType -> ph - _ -> t <+> res <> semi $+$ ph $+$ text "return" <+> res <> semi - return $ - t <+> n <> parens p + _ -> t empty <+> res <> semi $+$ ph $+$ text "return" <+> res <> semi + return [ + t empty <+> n <> parens p $+$ text "{" $+$ nest 4 phrasesBlock $+$ - text "}" + text "}"] where phrase2C' (Phrases p) = liftM vcat $ mapM phrase2C p phrase2C' p = phrase2C p -tvar2C False (FunctionDeclaration (Identifier name _) _ _ _) = error $ "nested functions not allowed: " ++ name +fun2C False _ (FunctionDeclaration (Identifier name _) _ _ _) = error $ "nested functions not allowed: " ++ name +fun2C _ tv _ = error $ "fun2C: I don't render " ++ show tv +tvar2C :: Bool -> TypeVarDeclaration -> State RenderState [Doc] +tvar2C b f@(FunctionDeclaration (Identifier name _) _ _ _) = + fun2C b name f tvar2C _ td@(TypeDeclaration i' t) = do i <- id2CTyped t i' - tp <- type2C t - return $ text "type" <+> i <+> tp <> semi + tp <- case t of + FunctionType {} -> type2C (PointerTo t) + _ -> type2C t + return [text "typedef" <+> tp i] tvar2C _ (VarDeclaration isConst (ids, t) mInitExpr) = do - t' <- type2C t - i <- mapM (id2CTyped t) ids + t' <- liftM (((if isConst then text "const" else empty) <+>) . ) $ type2C t ie <- initExpr mInitExpr - return $ if isConst then text "const" else empty - <+> t' - <+> (hsep . punctuate (char ',') $ i) - <+> ie - <> text ";" + lt <- gets lastType + case (isConst, lt, ids, mInitExpr) of + (True, BTInt, [i], Just _) -> do + i' <- id2CTyped t i + return [text "enum" <> braces (i' <+> ie)] + (True, BTFloat, [i], Just e) -> do + i' <- id2CTyped t i + ie <- initExpr2C e + return [text "#define" <+> i' <+> parens ie <> text "\n"] + _ -> liftM (map(\i -> t' i <+> ie)) $ mapM (id2CTyped t) ids where initExpr Nothing = return $ empty initExpr (Just e) = liftM (text "=" <+>) (initExpr2C e) -tvar2C f (OperatorDeclaration op i ret params body) = - tvar2C f (FunctionDeclaration i ret params body) +tvar2C f (OperatorDeclaration op (Identifier i _) ret params body) = do + r <- op2CTyped op (extractTypes params) + fun2C f i (FunctionDeclaration r ret params body) +op2CTyped :: String -> [TypeDecl] -> State RenderState Identifier +op2CTyped op t = do + t' <- liftM (render . hcat . punctuate (char '_') . map (\t -> t empty)) $ mapM type2C t + bt <- gets lastType + return $ case bt of + BTRecord {} -> Identifier (t' ++ "_op_" ++ opStr) bt + _ -> Identifier t' bt + where + opStr = case op of + "+" -> "add" + "-" -> "sub" + "*" -> "mul" + "/" -> "div" + "=" -> "eq" + "<" -> "lt" + ">" -> "gt" + _ -> error $ "op2CTyped: unknown op '" ++ op ++ "'" + +extractTypes :: [TypeVarDeclaration] -> [TypeDecl] +extractTypes = concatMap f + where + f (VarDeclaration _ (ids, t) _) = replicate (length ids) t + f a = error $ "extractTypes: can't extract from " ++ show a + initExpr2C :: InitExpression -> State RenderState Doc +initExpr2C InitNull = return $ text "NULL" +initExpr2C (InitAddress expr) = liftM ((<>) (text "&")) (initExpr2C expr) +initExpr2C (InitPrefixOp op expr) = liftM (text (op2C op) <>) (initExpr2C expr) initExpr2C (InitBinOp op expr1 expr2) = do e1 <- initExpr2C expr1 e2 <- initExpr2C expr2 - o <- op2C op - return $ parens $ e1 <+> o <+> e2 + return $ parens $ e1 <+> text (op2C op) <+> e2 initExpr2C (InitNumber s) = return $ text s initExpr2C (InitFloat s) = return $ text s initExpr2C (InitHexNumber s) = return $ text "0x" <> (text . map toLower $ s) -initExpr2C (InitString s) = return $ doubleQuotes $ text s +initExpr2C (InitString [a]) = return . quotes $ text [a] +initExpr2C (InitString s) = return $ strInit s +initExpr2C (InitChar a) = return $ quotes $ text "\\x" <> text (showHex (read a) "") initExpr2C (InitReference i) = id2C IOLookup i -initExpr2C _ = return $ text "<>" +initExpr2C (InitRecord fields) = do + (fs :: [Doc]) <- mapM (\(Identifier a _, b) -> liftM (text "." <> text a <+> equals <+>) $ initExpr2C b) fields + return $ lbrace $+$ (nest 4 . vcat . punctuate comma $ fs) $+$ rbrace +initExpr2C (InitArray [value]) = initExpr2C value +initExpr2C (InitArray values) = liftM (braces . vcat . punctuate comma) $ mapM initExpr2C values +initExpr2C r@(InitRange (Range i@(Identifier i' _))) = do + id2C IOLookup i + t <- gets lastType + case t of + BTEnum s -> return . int $ length s + BTInt -> case i' of + "byte" -> return $ int 256 + _ -> error $ "InitRange identifier: " ++ i' + _ -> error $ "InitRange: " ++ show r +initExpr2C (InitRange (RangeFromTo (InitNumber "0") r)) = initExpr2C $ BuiltInFunction "succ" [r] +initExpr2C (InitRange (RangeFromTo (InitChar "0") (InitChar r))) = initExpr2C $ BuiltInFunction "succ" [InitNumber r] +initExpr2C (InitRange a) = error $ show a --return $ text "<>" +initExpr2C (InitSet []) = return $ text "0" +initExpr2C (InitSet a) = return $ text "<>" +initExpr2C (BuiltInFunction "low" [InitReference e]) = return $ + case e of + (Identifier "LongInt" _) -> int (-2^31) + (Identifier "SmallInt" _) -> int (-2^15) + _ -> error $ "BuiltInFunction 'low': " ++ show e +initExpr2C (BuiltInFunction "high" [e]) = do + initExpr2C e + t <- gets lastType + case t of + (BTArray i _ _) -> initExpr2C $ BuiltInFunction "pred" [InitRange i] + a -> error $ "BuiltInFunction 'high': " ++ show a +initExpr2C (BuiltInFunction "succ" [BuiltInFunction "pred" [e]]) = initExpr2C e +initExpr2C (BuiltInFunction "pred" [BuiltInFunction "succ" [e]]) = initExpr2C e +initExpr2C (BuiltInFunction "succ" [e]) = liftM (<> text " + 1") $ initExpr2C e +initExpr2C (BuiltInFunction "pred" [e]) = liftM (<> text " - 1") $ initExpr2C e +initExpr2C b@(BuiltInFunction _ _) = error $ show b +initExpr2C a = error $ "initExpr2C: don't know how to render " ++ show a -type2C :: TypeDecl -> State RenderState Doc -type2C (SimpleType i) = id2C IOLookup i +range2C :: InitExpression -> State RenderState [Doc] +range2C (InitString [a]) = return [quotes $ text [a]] +range2C (InitRange (Range i)) = liftM (flip (:) []) $ id2C IOLookup i +range2C (InitRange (RangeFromTo (InitString [a]) (InitString [b]))) = return $ map (\i -> quotes $ text [i]) [a..b] +range2C a = liftM (flip (:) []) $ initExpr2C a + +baseType2C :: String -> BaseType -> Doc +baseType2C _ BTFloat = text "float" +baseType2C _ BTBool = text "bool" +baseType2C _ BTString = text "string255" +baseType2C s a = error $ "baseType2C: " ++ show a ++ "\n" ++ s + +type2C :: TypeDecl -> State RenderState (Doc -> Doc) +type2C (SimpleType i) = liftM (\i a -> i <+> a) $ id2C IOLookup i type2C t = do r <- type2C' t rt <- resolveType t modify (\st -> st{lastType = rt}) return r where - type2C' VoidType = return $ text "void" - type2C' (String l) = return $ text $ "string" ++ show l - type2C' (PointerTo (SimpleType i)) = liftM (<> text "*") $ id2C IODeferred i - type2C' (PointerTo t) = liftM (<> text "*") $ type2C t + type2C' VoidType = return (text "void" <+>) + type2C' (String l) = return (text "string255" <+>)--return (text ("string" ++ show l) <+>) + type2C' (PointerTo (SimpleType i)) = liftM (\i a -> text "struct __" <> i <+> text "*" <+> a) $ id2C IODeferred i + type2C' (PointerTo t) = liftM (\t a -> t (parens $ text "*" <> a)) $ type2C t type2C' (RecordType tvs union) = do - t <- mapM (tvar2C False) tvs - return $ text "{" $+$ (nest 4 . vcat $ t) $+$ text "}" - type2C' (RangeType r) = return $ text "<>" + t <- withState' id $ mapM (tvar2C False) tvs + u <- unions + return $ \i -> text "struct __" <> i <+> lbrace $+$ nest 4 ((vcat . map (<> semi) . concat $ t) $$ u) $+$ rbrace <+> i + where + unions = case union of + Nothing -> return empty + Just a -> do + structs <- mapM struct2C a + return $ text "union" $+$ braces (nest 4 $ vcat structs) <> semi + struct2C tvs = do + t <- withState' id $ mapM (tvar2C False) tvs + return $ text "struct" $+$ braces (nest 4 (vcat . map (<> semi) . concat $ t)) <> semi + type2C' (RangeType r) = return (text "int" <+>) type2C' (Sequence ids) = do - mapM_ (id2C IOInsert) ids - return $ text "<>" - type2C' (ArrayDecl r t) = return $ text "<>" - type2C' (Set t) = return $ text "<>" - type2C' (FunctionType returnType params) = return $ text "<>" - type2C' (DeriveType _) = return $ text "<>" + is <- mapM (id2C IOInsert . setBaseType bt) ids + return (text "enum" <+> (braces . vcat . punctuate comma . map (\(a, b) -> a <+> equals <+> text "0x" <> text (showHex b "")) $ zip is [1..]) <+>) + where + bt = BTEnum $ map (\(Identifier i _) -> map toLower i) ids + type2C' (ArrayDecl Nothing t) = type2C (PointerTo t) + type2C' (ArrayDecl (Just r) t) = do + t' <- type2C t + r' <- initExpr2C (InitRange r) + return $ \i -> t' i <> brackets r' + type2C' (Set t) = return (text "<>" <+>) + type2C' (FunctionType returnType params) = do + t <- type2C returnType + p <- withState' id $ functionParams2C params + return (\i -> t empty <+> i <> parens p) + type2C' (DeriveType (InitBinOp _ _ i)) = type2C' (DeriveType i) + type2C' (DeriveType (InitPrefixOp _ i)) = type2C' (DeriveType i) + type2C' (DeriveType (InitNumber _)) = return (text "int" <+>) + type2C' (DeriveType (InitHexNumber _)) = return (text "int" <+>) + type2C' (DeriveType (InitFloat _)) = return (text "float" <+>) + type2C' (DeriveType (BuiltInFunction {})) = return (text "int" <+>) + type2C' (DeriveType (InitString {})) = return (text "string255" <+>) + type2C' (DeriveType r@(InitReference {})) = do + initExpr2C r + t <- gets lastType + return (baseType2C (show r) t <+>) + type2C' (DeriveType a) = error $ "Can't derive type from " ++ show a phrase2C :: Phrase -> State RenderState Doc phrase2C (Phrases p) = do ps <- mapM phrase2C p return $ text "{" $+$ (nest 4 . vcat $ ps) $+$ text "}" phrase2C (ProcCall f@(FunCall {}) []) = liftM (<> semi) $ ref2C f -phrase2C (ProcCall ref params) = do +phrase2C (ProcCall ref []) = liftM (<> semi) $ ref2CF ref +phrase2C (ProcCall ref params) = error $ "ProcCall"{-do r <- ref2C ref ps <- mapM expr2C params - return $ r <> parens (hsep . punctuate (char ',') $ ps) <> semi + return $ r <> parens (hsep . punctuate (char ',') $ ps) <> semi -} phrase2C (IfThenElse (expr) phrase1 mphrase2) = do e <- expr2C expr p1 <- (phrase2C . wrapPhrase) phrase1 @@ -367,10 +553,12 @@ elsePart | isNothing mphrase2 = return $ empty | otherwise = liftM (text "else" $$) $ (phrase2C . wrapPhrase) (fromJust mphrase2) phrase2C (Assignment ref expr) = do - r <- ref2C ref - e <- expr2C expr - return $ - r <> text " = " <> e <> semi + r <- ref2C ref + t <- gets lastType + e <- case (t, expr) of + (BTFunction _, (Reference r')) -> ref2C r' + _ -> expr2C expr + return $ r <+> text "=" <+> e <> semi phrase2C (WhileCycle expr phrase) = do e <- expr2C expr p <- phrase2C $ wrapPhrase phrase @@ -378,19 +566,29 @@ phrase2C (SwitchCase expr cases mphrase) = do e <- expr2C expr cs <- mapM case2C cases + d <- dflt return $ - text "switch" <> parens e <> text "of" $+$ (nest 4 . vcat) cs + text "switch" <> parens e $+$ braces (nest 4 . vcat $ cs ++ d) where case2C :: ([InitExpression], Phrase) -> State RenderState Doc case2C (e, p) = do - ie <- mapM initExpr2C e + ies <- mapM range2C e ph <- phrase2C p return $ - text "case" <+> parens (hsep . punctuate (char ',') $ ie) <> char ':' <> nest 4 (ph $+$ text "break;") -phrase2C (WithBlock ref p) = do + vcat (map (\i -> text "case" <+> i <> colon) . concat $ ies) <> nest 4 (ph $+$ text "break;") + dflt | isNothing mphrase = return [] + | otherwise = do + ph <- mapM phrase2C $ fromJust mphrase + return [text "default:" <+> nest 4 (vcat ph)] + +phrase2C wb@(WithBlock ref p) = do r <- ref2C ref - ph <- phrase2C $ wrapPhrase p - return $ text "namespace" <> parens r $$ ph + t <- gets lastType + case t of + (BTRecord rs) -> withRecordNamespace (render r ++ ".") rs $ phrase2C $ wrapPhrase p + a -> do + ns <- gets currentScope + error $ "'with' block referencing non-record type " ++ show a ++ "\n" ++ show wb ++ "\nnamespace: " ++ show (take 100 ns) phrase2C (ForCycle i' e1' e2' p) = do i <- id2C IOLookup i' e1 <- expr2C e1' @@ -403,52 +601,117 @@ phrase2C (RepeatCycle e' p') = do e <- expr2C e' p <- phrase2C (Phrases p') - return $ text "do" <+> p <+> text "while" <> parens (text "!" <> parens e) + return $ text "do" <+> p <+> text "while" <> parens (text "!" <> parens e) <> semi phrase2C NOP = return $ text ";" +phrase2C (BuiltInFunctionCall [] (SimpleReference (Identifier "exit" BTUnknown))) = return $ text "return" <> semi +phrase2C (BuiltInFunctionCall [e] (SimpleReference (Identifier "exit" BTUnknown))) = liftM (\e -> text "return" <> e <> semi) $ expr2C e +phrase2C (BuiltInFunctionCall [e] (SimpleReference (Identifier "dec" BTUnknown))) = liftM (\e -> text "--" <> e <> semi) $ expr2C e +phrase2C (BuiltInFunctionCall [e1, e2] (SimpleReference (Identifier "dec" BTUnknown))) = liftM2 (\a b -> a <> text " -= " <> b <> semi) (expr2C e1) (expr2C e2) +phrase2C (BuiltInFunctionCall [e] (SimpleReference (Identifier "inc" BTUnknown))) = liftM (\e -> text "++" <> e <> semi) $ expr2C e +phrase2C (BuiltInFunctionCall [e1, e2] (SimpleReference (Identifier "inc" BTUnknown))) = liftM2 (\a b -> a <+> text "+=" <+> b <> semi) (expr2C e1) (expr2C e2) +phrase2C a = error $ "phrase2C: " ++ show a wrapPhrase p@(Phrases _) = p wrapPhrase p = Phrases [p] - expr2C :: Expression -> State RenderState Doc expr2C (Expression s) = return $ text s expr2C (BinOp op expr1 expr2) = do e1 <- expr2C expr1 + t1 <- gets lastType e2 <- expr2C expr2 - o <- op2C op - return $ parens $ e1 <+> o <+> e2 + t2 <- gets lastType + case (op2C op, t1, t2) of + ("+", BTString, BTString) -> expr2C $ BuiltInFunCall [expr1, expr2] (SimpleReference $ Identifier "_strconcat" (BTFunction BTString)) + ("+", BTString, BTChar) -> expr2C $ BuiltInFunCall [expr1, expr2] (SimpleReference $ Identifier "_strappend" (BTFunction BTString)) + ("+", BTChar, BTString) -> expr2C $ BuiltInFunCall [expr1, expr2] (SimpleReference $ Identifier "_strprepend" (BTFunction BTString)) + ("==", BTString, _) -> expr2C $ BuiltInFunCall [expr1, expr2] (SimpleReference $ Identifier "_strcompare" (BTFunction BTBool)) + ("!=", BTString, _) -> expr2C $ BuiltInFunCall [expr1, expr2] (SimpleReference $ Identifier "_strncompare" (BTFunction BTBool)) + ("&", BTBool, _) -> return $ parens e1 <+> text "&&" <+> parens e2 + ("|", BTBool, _) -> return $ parens e1 <+> text "||" <+> parens e2 + (o, _, _) | o `elem` boolOps -> do + modify(\s -> s{lastType = BTBool}) + return $ parens e1 <+> text o <+> parens e2 + | otherwise -> return $ parens e1 <+> text o <+> parens e2 + where + boolOps = ["==", "!=", "<", ">", "<=", ">="] expr2C (NumberLiteral s) = return $ text s expr2C (FloatLiteral s) = return $ text s expr2C (HexNumber s) = return $ text "0x" <> (text . map toLower $ s) -expr2C (StringLiteral s) = return $ doubleQuotes $ text s -expr2C (Reference ref) = ref2C ref -expr2C (PrefixOp op expr) = liftM2 (<+>) (op2C op) (expr2C expr) +expr2C (StringLiteral [a]) = do + modify(\s -> s{lastType = BTChar}) + return . quotes $ text [a] +expr2C (StringLiteral s) = addStringConst s +expr2C (Reference ref) = ref2CF ref +expr2C (PrefixOp op expr) = liftM (text (op2C op) <>) (expr2C expr) expr2C Null = return $ text "NULL" +expr2C (CharCode a) = do + modify(\s -> s{lastType = BTChar}) + return $ quotes $ text "\\x" <> text (showHex (read a) "") +expr2C (HexCharCode a) = return $ quotes $ text "\\x" <> text (map toLower a) +expr2C (SetExpression ids) = mapM (id2C IOLookup) ids >>= return . parens . hcat . punctuate (text " | ") + +expr2C (BuiltInFunCall [e] (SimpleReference (Identifier "ord" _))) = liftM parens $ expr2C e +expr2C (BuiltInFunCall [e] (SimpleReference (Identifier "succ" _))) = liftM (<> text " + 1") $ expr2C e +expr2C (BuiltInFunCall [e] (SimpleReference (Identifier "pred" _))) = liftM (<> text " - 1") $ expr2C e expr2C (BuiltInFunCall params ref) = do r <- ref2C ref + t <- gets lastType ps <- mapM expr2C params + case t of + BTFunction t' -> do + modify (\s -> s{lastType = t'}) + _ -> error $ "BuiltInFunCall lastType: " ++ show t return $ r <> parens (hsep . punctuate (char ',') $ ps) -expr2C _ = return $ text "<>" +expr2C a = error $ "Don't know how to render " ++ show a +ref2CF :: Reference -> State RenderState Doc +ref2CF (SimpleReference name) = do + i <- id2C IOLookup name + t <- gets lastType + case t of + BTFunction _ -> return $ i <> parens empty + _ -> return $ i +ref2CF r = ref2C r ref2C :: Reference -> State RenderState Doc -ref2C ae@(ArrayElement exprs ref) = do - es <- mapM expr2C exprs +-- rewrite into proper form +ref2C (RecordField ref1 (ArrayElement exprs ref2)) = ref2C $ ArrayElement exprs (RecordField ref1 ref2) +ref2C (RecordField ref1 (Dereference ref2)) = ref2C $ Dereference (RecordField ref1 ref2) +ref2C (RecordField ref1 (RecordField ref2 ref3)) = ref2C $ RecordField (RecordField ref1 ref2) ref3 +ref2C (RecordField ref1 (FunCall params ref2)) = ref2C $ FunCall params (RecordField ref1 ref2) +ref2C (ArrayElement (a:b:xs) ref) = ref2C $ ArrayElement (b:xs) (ArrayElement [a] ref) +-- conversion routines +ref2C ae@(ArrayElement [expr] ref) = do + e <- expr2C expr r <- ref2C ref t <- gets lastType ns <- gets currentScope case t of - (BTArray _ (BTArray _ t')) -> modify (\st -> st{lastType = t'}) - (BTArray _ t') -> modify (\st -> st{lastType = t'}) + (BTArray _ _ t') -> modify (\st -> st{lastType = t'}) + (BTFunctionReturn _ (BTArray _ _ t')) -> modify (\st -> st{lastType = t'}) + (BTFunctionReturn _ (BTString)) -> modify (\st -> st{lastType = BTChar}) (BTString) -> modify (\st -> st{lastType = BTChar}) + (BTPointerTo t) -> do + t'' <- fromPointer (show t) =<< gets lastType + case t'' of + BTChar -> modify (\st -> st{lastType = BTChar}) + a -> error $ "Getting element of " ++ show a ++ "\nReference: " ++ show ae ++ "\n" ++ show (take 100 ns) a -> error $ "Getting element of " ++ show a ++ "\nReference: " ++ show ae ++ "\n" ++ show (take 100 ns) - return $ r <> (brackets . hcat) (punctuate comma es) + case t of + BTString -> return $ r <> text ".s" <> brackets e + _ -> return $ r <> brackets e ref2C (SimpleReference name) = id2C IOLookup name -ref2C (RecordField (Dereference ref1) ref2) = do +ref2C rf@(RecordField (Dereference ref1) ref2) = do r1 <- ref2C ref1 - r2 <- ref2C ref2 + t <- fromPointer (show ref1) =<< gets lastType + ns <- gets currentScope + r2 <- case t of + BTRecord rs -> withRecordNamespace "" rs $ ref2C ref2 + BTUnit -> withLastIdNamespace $ ref2C ref2 + a -> error $ "dereferencing from " ++ show a ++ "\n" ++ show rf ++ "\n" ++ show (take 100 ns) return $ r1 <> text "->" <> r2 ref2C rf@(RecordField ref1 ref2) = do @@ -456,48 +719,56 @@ t <- gets lastType ns <- gets currentScope r2 <- case t of - BTRecord rs -> withRecordNamespace rs $ ref2C ref2 - BTUnit -> withLastIdNamespace $ ref2C ref2 + BTFunctionReturn s (BTRecord rs) -> withRecordNamespace "" rs $ ref2C ref2 + BTRecord rs -> withRecordNamespace "" rs $ ref2C ref2 + BTUnit -> withLastIdNamespace $ ref2C ref2 a -> error $ "dereferencing from " ++ show a ++ "\n" ++ show rf ++ "\n" ++ show (take 100 ns) return $ r1 <> text "." <> r2 -ref2C (Dereference ref) = do +ref2C d@(Dereference ref) = do r <- ref2C ref - t <- fromPointer =<< gets lastType + t <- fromPointer (show d) =<< gets lastType modify (\st -> st{lastType = t}) - return $ (parens $ text "*") <> r -ref2C (FunCall params ref) = do - ps <- liftM (parens . hsep . punctuate (char ',')) $ mapM expr2C params + return $ (parens $ text "*" <> r) +ref2C f@(FunCall params ref) = do r <- ref2C ref t <- gets lastType case t of - BTFunction t -> do - modify (\s -> s{lastType = t}) + BTFunction t' -> do + ps <- liftM (parens . hsep . punctuate (char ',')) $ mapM expr2C params + modify (\s -> s{lastType = t'}) return $ r <> ps - _ -> return $ parens r <> ps + BTFunctionReturn r t' -> do + ps <- liftM (parens . hsep . punctuate (char ',')) $ mapM expr2C params + modify (\s -> s{lastType = t'}) + return $ text r <> ps + _ -> case (ref, params) of + (SimpleReference i, [p]) -> ref2C $ TypeCast i p + _ -> error $ "ref2C FunCall erroneous type cast detected: " ++ show f ++ "\nType detected: " ++ show t ref2C (Address ref) = do r <- ref2C ref return $ text "&" <> parens r -ref2C (TypeCast t' expr) = do - t <- id2C IOLookup t' - e <- expr2C expr - return $ parens t <> e +ref2C (TypeCast t'@(Identifier i _) expr) = do + case map toLower i of + "pchar" -> ref2C $ FunCall [expr] (SimpleReference (Identifier "_pchar" $ BTPointerTo BTChar)) + a -> do + e <- expr2C expr + t <- id2C IOLookup t' + return $ parens t <> e ref2C (RefExpression expr) = expr2C expr -op2C :: String -> State RenderState Doc -op2C "or" = return $ text "|" -op2C "and" = return $ text "&" -op2C "not" = return $ text "!" -op2C "xor" = return $ text "^" -op2C "div" = return $ text "/" -op2C "mod" = return $ text "%" -op2C "shl" = return $ text "<<" -op2C "shr" = return $ text ">>" -op2C "<>" = return $ text "!=" -op2C "=" = return $ text "==" -op2C a = return $ text a +op2C :: String -> String +op2C "or" = "|" +op2C "and" = "&" +op2C "not" = "!" +op2C "xor" = "^" +op2C "div" = "/" +op2C "mod" = "%" +op2C "shl" = "<<" +op2C "shr" = ">>" +op2C "<>" = "!=" +op2C "=" = "==" +op2C a = a -maybeVoid "" = "void" -maybeVoid a = a