diff -r 850b8dd3e6df -r eae5900fd8a4 tools/pas2c.hs --- a/tools/pas2c.hs Thu Nov 24 16:33:36 2011 +0100 +++ b/tools/pas2c.hs Thu Nov 24 20:59:13 2011 +0300 @@ -4,22 +4,47 @@ import Text.PrettyPrint.HughesPJ import Data.Maybe import Data.Char -import Text.Parsec.String +import Text.Parsec.Prim +import Control.Monad.State +import System.IO +import System.Directory +import Control.Monad.IO.Class +import PascalPreprocessor +import Control.Exception +import System.IO.Error +import qualified Data.Set as Set pas2C :: String -> IO String -pas2C fileName = do - ptree <- parseFromFile pascalUnit fileName - case ptree of - (Left a) -> return (show a) - (Right a) -> (return . render . pascal2C) a - +pas2C = flip evalStateT initState . f + where + printLn = liftIO . hPutStrLn stderr + initState = Set.empty + f :: String -> StateT (Set.Set String) IO String + f fileName = do + liftIO $ setCurrentDirectory "../hedgewars/" + + fc' <- liftIO $ tryJust (guard . isDoesNotExistError) $ preprocess fileName + case fc' of + (Left a) -> return "" + (Right fc) -> do + modify $ Set.insert fileName + printLn $ "Preprocessed " ++ fileName + liftIO $ writeFile "debug.txt" fc + let ptree = parse pascalUnit fileName fc + case ptree of + (Left a) -> return (show a) + (Right a) -> (return . render . pascal2C) a + pascal2C :: PascalUnit -> Doc pascal2C (Unit unitName interface implementation init fin) = interface2C interface $+$ implementation2C implementation - +pascal2C (Program _ implementation mainFunction) = + implementation2C implementation + $+$ + tvar2C (FunctionDeclaration (Identifier "main") (SimpleType $ Identifier "int") (Just (TypesAndVars [], mainFunction))) interface2C :: Interface -> Doc interface2C (Interface uses tvars) = typesAndVars2C tvars @@ -90,8 +115,8 @@ phrase2C (WhileCycle expr phrase) = text "while" <> parens (expr2C expr) $$ (phrase2C $ wrapPhrase phrase) phrase2C (SwitchCase expr cases mphrase) = text "switch" <> parens (expr2C expr) <> text "of" $+$ (nest 4 . vcat . map case2C) cases where - case2C :: (Expression, Phrase) -> Doc - case2C (e, p) = text "case" <+> parens (expr2C e) <> char ':' <> nest 4 (phrase2C p $+$ text "break;") + case2C :: ([Expression], Phrase) -> Doc + case2C (e, p) = text "case" <+> parens (hsep . punctuate (char ',') . map expr2C $ e) <> char ':' <> nest 4 (phrase2C p $+$ text "break;") phrase2C (WithBlock ref p) = text "namespace" <> parens (ref2C ref) $$ (phrase2C $ wrapPhrase p) phrase2C (ForCycle (Identifier i) e1 e2 p) = text "for" <> (parens . hsep . punctuate (char ';') $ [text i <+> text "=" <+> expr2C e1, text i <+> text "<=" <+> expr2C e2, text "++" <> text i])