X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Futils%2Fhsc2hs%2FMain.hs;h=ee30d09ad455df6865790eb5dbc295662bd238fd;hb=c63aaee42e3e95276861121286eefce511cd1343;hp=be87ec6869005308d27ca98cfb63805837c45c13;hpb=978119ba717afa02126f7b78d0452d236db7513a;p=ghc-hetmet.git diff --git a/ghc/utils/hsc2hs/Main.hs b/ghc/utils/hsc2hs/Main.hs index be87ec6..ee30d09 100644 --- a/ghc/utils/hsc2hs/Main.hs +++ b/ghc/utils/hsc2hs/Main.hs @@ -1,7 +1,5 @@ ------------------------------------------------------------------------------ --- $Id: Main.hs,v 1.11 2001/01/13 20:33:51 qrczak Exp $ --- --- (originally "GlueHsc.hs" by Marcin 'Qrczak' Kowalczyk) +------------------------------------------------------------------------ +-- $Id: Main.hs,v 1.41 2002/10/27 10:38:33 mthomas Exp $ -- -- Program for converting .hsc files to .hs files, by converting the -- file into a C program which is run to generate the Haskell source. @@ -11,52 +9,96 @@ -- -- See the documentation in the Users' Guide for more details. +#if __GLASGOW_HASKELL__ >= 504 +import System.Console.GetOpt +#else import GetOpt -import System (getProgName, getArgs, ExitCode(..), system, exitWith, exitFailure) -import Directory (removeFile) -import Parsec -import ParsecError -import Monad (liftM, liftM2, when) -import Char (ord, intToDigit, isSpace, isAlpha, isAlphaNum, toUpper) -import List (intersperse) +#endif + +import Config +import System (getProgName, getArgs, ExitCode(..), exitWith, exitFailure, system) +import Directory (removeFile) +import Monad (MonadPlus(..), liftM, liftM2, when, unless) +import Char (isAlpha, isAlphaNum, isSpace, isDigit, toUpper, intToDigit, ord) +import List (intersperse) + +#include "../../includes/config.h" + +#ifdef mingw32_HOST_OS +-- import Win32DLL +import Foreign.C.String (CString, peekCString) +import Foreign.C.Types +import Foreign.Ptr (nullPtr) +import Foreign.Marshal.Alloc (mallocBytes, free) + +foreign import stdcall "GetModuleHandle" c_GetModuleHandle :: CString -> IO CUInt +foreign import stdcall "GetModuleFileName" c_GetModuleFilename :: CUInt -> CString -> CUInt -> IO CUInt + +ourName :: IO String +ourName = do h <- c_GetModuleHandle nullPtr + cstr <- mallocBytes cstr_len + rv <- c_GetModuleFilename h cstr (CUInt (fromIntegral cstr_len)) + str <- peekCString cstr + free cstr + return str + where cstr_len = 512 +#endif version :: String -version = "hsc2hs-0.64" +version = "hsc2hs-0.65" data Flag = Help | Version - | Template String - | Compiler String - | Linker String - | CompFlag String - | LinkFlag String - | Include String + | Template String + | Compiler String + | Linker String + | CompFlag String + | LinkFlag String + | NoCompile + | Include String + | Define String (Maybe String) + | Output String include :: String -> Flag include s@('\"':_) = Include s include s@('<' :_) = Include s include s = Include ("\""++s++"\"") +define :: String -> Flag +define s = case break (== '=') s of + (name, []) -> Define name Nothing + (name, _:value) -> Define name (Just value) + options :: [OptDescr Flag] options = [ - Option "t" ["template"] (ReqArg Template "FILE") "template file", - Option "" ["cc"] (ReqArg Compiler "PROG") "C compiler to use", - Option "" ["ld"] (ReqArg Linker "PROG") "linker to use", - Option "" ["cflag"] (ReqArg CompFlag "FLAG") "flag to pass to the C compiler", - Option "I" [] (ReqArg (CompFlag . ("-I"++)) - "DIR") "passed to the C compiler", - Option "" ["lflag"] (ReqArg LinkFlag "FLAG") "flag to pass to the linker", - Option "" ["include"] (ReqArg include "FILE") "as if placed in the source", - Option "" ["help"] (NoArg Help) "display this help and exit", - Option "" ["version"] (NoArg Version) "output version information and exit"] + Option "t" ["template"] (ReqArg Template "FILE") "template file", + Option "c" ["cc"] (ReqArg Compiler "PROG") "C compiler to use", + Option "l" ["ld"] (ReqArg Linker "PROG") "linker to use", + Option "C" ["cflag"] (ReqArg CompFlag "FLAG") "flag to pass to the C compiler", + Option "I" [] (ReqArg (CompFlag . ("-I"++)) + "DIR") "passed to the C compiler", + Option "L" ["lflag"] (ReqArg LinkFlag "FLAG") "flag to pass to the linker", + Option "i" ["include"] (ReqArg include "FILE") "as if placed in the source", + Option "D" ["define"] (ReqArg define "NAME[=VALUE]") "as if placed in the source", + Option "o" ["output"] (ReqArg Output "FILE") "name of main output file", + Option "" ["help"] (NoArg Help) "display this help and exit", + Option "" ["version"] (NoArg Version) "output version information and exit", + Option "" ["no-compile"] (NoArg NoCompile) "stop after writing *_hsc_make.c"] main :: IO () main = do prog <- getProgName let header = "Usage: "++prog++" [OPTIONS] INPUT.hsc [...]" args <- getArgs - case getOpt Permute options args of + let opts@(flags, files, errs) = getOpt Permute options args +#ifdef mingw32_HOST_OS + n <- ourName + let tempName = reverse (drop (length "\\bin\\hsc2hs.exe") (reverse n)) ++ "\\template-hsc.h" + let fflags = if [t | Template t <- flags] /= [] then flags else (Template tempName) : flags + let opts = (fflags, files, errs) +#endif + case opts of (flags, _, _) | any isHelp flags -> putStrLn (usageInfo header options) | any isVersion flags -> putStrLn version @@ -71,161 +113,408 @@ main = do exitFailure processFile :: [Flag] -> String -> IO () -processFile flags name = do - parsed <- parseFromFile parser name - case parsed of - Left err -> do print err; exitFailure - Right toks -> output flags name toks +processFile flags name + = do let file_name = dosifyPath name + s <- readFile file_name + case parser of + Parser p -> case p (SourcePos file_name 1) s of + Success _ _ _ toks -> output flags file_name toks + Failure (SourcePos name' line) msg -> do + putStrLn (name'++":"++show line++": "++msg) + exitFailure + +------------------------------------------------------------------------ +-- Convert paths foo/baz to foo\baz on Windows + +#if defined(mingw32_HOST_OS) +subst a b ls = map (\ x -> if x == a then b else x) ls +dosifyPath xs = subst '/' '\\' xs +#else +dosifyPath xs = xs +#endif + +------------------------------------------------------------------------ +-- A deterministic parser which remembers the text which has been parsed. + +newtype Parser a = Parser (SourcePos -> String -> ParseResult a) + +data ParseResult a = Success !SourcePos String String a + | Failure !SourcePos String + +data SourcePos = SourcePos String !Int + +updatePos :: SourcePos -> Char -> SourcePos +updatePos pos@(SourcePos name line) ch = case ch of + '\n' -> SourcePos name (line + 1) + _ -> pos + +instance Monad Parser where + return a = Parser $ \pos s -> Success pos [] s a + Parser m >>= k = + Parser $ \pos s -> case m pos s of + Success pos' out1 s' a -> case k a of + Parser k' -> case k' pos' s' of + Success pos'' out2 imp'' b -> + Success pos'' (out1++out2) imp'' b + Failure pos'' msg -> Failure pos'' msg + Failure pos' msg -> Failure pos' msg + fail msg = Parser $ \pos _ -> Failure pos msg + +instance MonadPlus Parser where + mzero = fail "mzero" + Parser m `mplus` Parser n = + Parser $ \pos s -> case m pos s of + success@(Success _ _ _ _) -> success + Failure _ _ -> n pos s + +getPos :: Parser SourcePos +getPos = Parser $ \pos s -> Success pos [] s pos + +setPos :: SourcePos -> Parser () +setPos pos = Parser $ \_ s -> Success pos [] s () + +message :: Parser a -> String -> Parser a +Parser m `message` msg = + Parser $ \pos s -> case m pos s of + success@(Success _ _ _ _) -> success + Failure pos' _ -> Failure pos' msg + +catchOutput_ :: Parser a -> Parser String +catchOutput_ (Parser m) = + Parser $ \pos s -> case m pos s of + Success pos' out s' _ -> Success pos' [] s' out + Failure pos' msg -> Failure pos' msg + +fakeOutput :: Parser a -> String -> Parser a +Parser m `fakeOutput` out = + Parser $ \pos s -> case m pos s of + Success pos' _ s' a -> Success pos' out s' a + Failure pos' msg -> Failure pos' msg + +lookAhead :: Parser String +lookAhead = Parser $ \pos s -> Success pos [] s s + +satisfy :: (Char -> Bool) -> Parser Char +satisfy p = + Parser $ \pos s -> case s of + c:cs | p c -> Success (updatePos pos c) [c] cs c + _ -> Failure pos "Bad character" + +char_ :: Char -> Parser () +char_ c = do + satisfy (== c) `message` (show c++" expected") + return () + +anyChar_ :: Parser () +anyChar_ = do + satisfy (const True) `message` "Unexpected end of file" + return () + +any2Chars_ :: Parser () +any2Chars_ = anyChar_ >> anyChar_ + +many :: Parser a -> Parser [a] +many p = many1 p `mplus` return [] + +many1 :: Parser a -> Parser [a] +many1 p = liftM2 (:) p (many p) + +many_ :: Parser a -> Parser () +many_ p = many1_ p `mplus` return () + +many1_ :: Parser a -> Parser () +many1_ p = p >> many_ p + +manySatisfy, manySatisfy1 :: (Char -> Bool) -> Parser String +manySatisfy = many . satisfy +manySatisfy1 = many1 . satisfy + +manySatisfy_, manySatisfy1_ :: (Char -> Bool) -> Parser () +manySatisfy_ = many_ . satisfy +manySatisfy1_ = many1_ . satisfy + +------------------------------------------------------------------------ +-- Parser of hsc syntax. data Token = Text SourcePos String | Special SourcePos String String parser :: Parser [Token] -parser = many (text <|> special) +parser = do + pos <- getPos + t <- catchOutput_ text + s <- lookAhead + rest <- case s of + [] -> return [] + _:_ -> liftM2 (:) (special `fakeOutput` []) parser + return (if null t then rest else Text pos t : rest) -text :: Parser Token +text :: Parser () text = do - pos <- getPosition - liftM (Text pos . concat) $ many1 - ( many1 (satisfy (\ch -> not (isAlpha ch || ch `elem` "\"#'-_{"))) - <|> (do a <- satisfy (\ch -> isAlpha ch || ch == '_') - b <- many (satisfy (\ch -> isAlphaNum ch || ch == '_' || ch == '\'')) - return (a:b)) - <|> (do char '\"'; a <- hsString '\"'; char '\"'; return ("\""++a++"\"")) - <|> (do try (string "##"); return "#") - <|> (do char '\''; a <- hsString '\''; char '\''; return ("\'"++a++"\'")) - <|> (do try (string "--"); a <- many (satisfy (/= '\n')); return ("--"++a)) - <|> string "-" - <|> (do try (string "{-#"); optional (try linePragma); a <- hsComment; return ("{-#"++a)) - <|> (do try (string "{-"); a <- hsComment; return ("{-"++a)) - <|> string "{" - "Haskell source") + s <- lookAhead + case s of + [] -> return () + c:_ | isAlpha c || c == '_' -> do + anyChar_ + manySatisfy_ (\c' -> isAlphaNum c' || c' == '_' || c' == '\'') + text + c:_ | isHsSymbol c -> do + symb <- catchOutput_ (manySatisfy_ isHsSymbol) + case symb of + "#" -> return () + '-':'-':symb' | all (== '-') symb' -> do + return () `fakeOutput` symb + manySatisfy_ (/= '\n') + text + _ -> do + return () `fakeOutput` unescapeHashes symb + text + '\"':_ -> do anyChar_; hsString '\"'; text + '\'':_ -> do anyChar_; hsString '\''; text + '{':'-':_ -> do any2Chars_; linePragma `mplus` hsComment; text + _:_ -> do anyChar_; text + +hsString :: Char -> Parser () +hsString quote = do + s <- lookAhead + case s of + [] -> return () + c:_ | c == quote -> anyChar_ + '\\':c:_ + | isSpace c -> do + anyChar_ + manySatisfy_ isSpace + char_ '\\' `mplus` return () + hsString quote + | otherwise -> do any2Chars_; hsString quote + _:_ -> do anyChar_; hsString quote + +hsComment :: Parser () +hsComment = do + s <- lookAhead + case s of + [] -> return () + '-':'}':_ -> any2Chars_ + '{':'-':_ -> do any2Chars_; hsComment; hsComment + _:_ -> do anyChar_; hsComment linePragma :: Parser () linePragma = do - state <- getState - spaces - string "LINE" - skipMany1 space - line <- many1 digit - skipMany1 space - char '\"' - file <- many (satisfy (/= '\"')) - char '\"' - spaces - string "#-}" - setState state - setPosition (newPos file (read line - 1) 1) - -hsComment :: Parser String -hsComment = - ( (do a <- many1 (noneOf "-{"); b <- hsComment; return (a++b)) - <|> try (string "-}") - <|> (do char '-'; b <- hsComment; return ('-':b)) - <|> (do try (string "{-"); a <- hsComment; b <- hsComment; return ("{-"++a++b)) - <|> (do char '{'; b <- hsComment; return ('{':b)) - "Haskell comment") - -hsString :: Char -> Parser String -hsString quote = - liftM concat $ many - ( many1 (noneOf (quote:"\n\\")) - <|> (do char '\\'; a <- escape; return ('\\':a)) - "Haskell character or string") + char_ '#' + manySatisfy_ isSpace + satisfy (\c -> c == 'L' || c == 'l') + satisfy (\c -> c == 'I' || c == 'i') + satisfy (\c -> c == 'N' || c == 'n') + satisfy (\c -> c == 'E' || c == 'e') + manySatisfy1_ isSpace + line <- liftM read $ manySatisfy1 isDigit + manySatisfy1_ isSpace + char_ '\"' + name <- manySatisfy (/= '\"') + char_ '\"' + manySatisfy_ isSpace + char_ '#' + char_ '-' + char_ '}' + setPos (SourcePos name (line - 1)) + +isHsSymbol :: Char -> Bool +isHsSymbol '!' = True; isHsSymbol '#' = True; isHsSymbol '$' = True +isHsSymbol '%' = True; isHsSymbol '&' = True; isHsSymbol '*' = True +isHsSymbol '+' = True; isHsSymbol '.' = True; isHsSymbol '/' = True +isHsSymbol '<' = True; isHsSymbol '=' = True; isHsSymbol '>' = True +isHsSymbol '?' = True; isHsSymbol '@' = True; isHsSymbol '\\' = True +isHsSymbol '^' = True; isHsSymbol '|' = True; isHsSymbol '-' = True +isHsSymbol '~' = True +isHsSymbol _ = False + +unescapeHashes :: String -> String +unescapeHashes [] = [] +unescapeHashes ('#':'#':s) = '#' : unescapeHashes s +unescapeHashes (c:s) = c : unescapeHashes s + +lookAheadC :: Parser String +lookAheadC = liftM joinLines lookAhead where - escape = (do a <- many1 (satisfy isSpace); char '\\'; return (a++"\\")) - <|> (do a <- anyChar; return [a]) + joinLines [] = [] + joinLines ('\\':'\n':s) = joinLines s + joinLines (c:s) = c : joinLines s + +satisfyC :: (Char -> Bool) -> Parser Char +satisfyC p = do + s <- lookAhead + case s of + '\\':'\n':_ -> do any2Chars_ `fakeOutput` []; satisfyC p + _ -> satisfy p + +charC_ :: Char -> Parser () +charC_ c = do + satisfyC (== c) `message` (show c++" expected") + return () + +anyCharC_ :: Parser () +anyCharC_ = do + satisfyC (const True) `message` "Unexpected end of file" + return () + +any2CharsC_ :: Parser () +any2CharsC_ = anyCharC_ >> anyCharC_ + +manySatisfyC :: (Char -> Bool) -> Parser String +manySatisfyC = many . satisfyC + +manySatisfyC_ :: (Char -> Bool) -> Parser () +manySatisfyC_ = many_ . satisfyC special :: Parser Token special = do - pos <- getPosition - char '#' - skipMany (oneOf " \t") - key <- liftM2 (:) (letter <|> char '_') (many (alphaNum <|> char '_')) - "hsc directive" - skipMany (oneOf " \t") - arg <- argument pzero + manySatisfyC_ (\c -> isSpace c && c /= '\n') + s <- lookAheadC + case s of + '{':_ -> do + anyCharC_ + manySatisfyC_ isSpace + sp <- keyArg (== '\n') + charC_ '}' + return sp + _ -> keyArg (const False) + +keyArg :: (Char -> Bool) -> Parser Token +keyArg eol = do + pos <- getPos + key <- keyword `message` "hsc keyword or '{' expected" + manySatisfyC_ (\c' -> isSpace c' && c' /= '\n' || eol c') + arg <- catchOutput_ (argument eol) return (Special pos key arg) -argument :: Parser String -> Parser String -argument eol = - liftM concat $ many - ( many1 (noneOf "\n\"\'()/[\\]{}") - <|> eol - <|> (do char '\"'; a <- cString '\"'; char '\"'; return ("\""++a++"\"")) - <|> (do char '\''; a <- cString '\''; char '\''; return ("\'"++a++"\'")) - <|> (do char '('; a <- nested; char ')'; return ("("++a++")")) - <|> (do try (string "/*"); cComment; return " ") - <|> (do try (string "//"); skipMany (satisfy (/= '\n')); return " ") - <|> string "/" - <|> (do char '['; a <- nested; char ']'; return ("["++a++"]")) - <|> (do char '\\'; a <- anyChar; return $ if a == '\n' then [] else ['\\',a]) - <|> (do char '{'; a <- nested; char '}'; return ("{"++a++"}")) - "C expression") - where nested = argument (string "\n") +keyword :: Parser String +keyword = do + c <- satisfyC (\c' -> isAlpha c' || c' == '_') + cs <- manySatisfyC (\c' -> isAlphaNum c' || c' == '_') + return (c:cs) + +argument :: (Char -> Bool) -> Parser () +argument eol = do + s <- lookAheadC + case s of + [] -> return () + c:_ | eol c -> do anyCharC_; argument eol + '\n':_ -> return () + '\"':_ -> do anyCharC_; cString '\"'; argument eol + '\'':_ -> do anyCharC_; cString '\''; argument eol + '(':_ -> do anyCharC_; nested ')'; argument eol + ')':_ -> return () + '/':'*':_ -> do any2CharsC_; cComment; argument eol + '/':'/':_ -> do + any2CharsC_; manySatisfyC_ (/= '\n'); argument eol + '[':_ -> do anyCharC_; nested ']'; argument eol + ']':_ -> return () + '{':_ -> do anyCharC_; nested '}'; argument eol + '}':_ -> return () + _:_ -> do anyCharC_; argument eol + +nested :: Char -> Parser () +nested c = do argument (== '\n'); charC_ c cComment :: Parser () -cComment = - ( (do skipMany1 (noneOf "*"); cComment) - <|> (do try (string "*/"); return ()) - <|> (do char '*'; cComment) - "C comment") - -cString :: Char -> Parser String -cString quote = - liftM concat $ many - ( many1 (noneOf (quote:"\n\\")) - <|> (do char '\\'; a <- anyChar; return ['\\',a]) - "C character or string") +cComment = do + s <- lookAheadC + case s of + [] -> return () + '*':'/':_ -> do any2CharsC_ + _:_ -> do anyCharC_; cComment + +cString :: Char -> Parser () +cString quote = do + s <- lookAheadC + case s of + [] -> return () + c:_ | c == quote -> anyCharC_ + '\\':_:_ -> do any2CharsC_; cString quote + _:_ -> do anyCharC_; cString quote + +------------------------------------------------------------------------ +-- Write the output files. + +splitName :: String -> (String, String) +splitName name = + case break (== '/') name of + (file, []) -> ([], file) + (dir, sep:rest) -> (dir++sep:restDir, restFile) + where + (restDir, restFile) = splitName rest + +splitExt :: String -> (String, String) +splitExt name = + case break (== '.') name of + (base, []) -> (base, []) + (base, sepRest@(sep:rest)) + | null restExt -> (base, sepRest) + | otherwise -> (base++sep:restBase, restExt) + where + (restBase, restExt) = splitExt rest output :: [Flag] -> String -> [Token] -> IO () -output flags name toks = let - baseName = case reverse name of - 'c':base -> reverse base - _ -> name++".hs" - cProgName = baseName++"c_make_hs.c" - oProgName = baseName++"c_make_hs.o" - progName = baseName++"c_make_hs" - outHsName = baseName - outHName = baseName++".h" - outCName = baseName++".c" +output flags name toks = do - execProgName = case progName of - '/':_ -> progName - _ -> "./"++progName + (outName, outDir, outBase) <- case [f | Output f <- flags] of + [] + | not (null ext) && + last ext == 'c' -> return (dir++base++init ext, dir, base) + | ext == ".hs" -> return (dir++base++"_out.hs", dir, base) + | otherwise -> return (dir++base++".hs", dir, base) + where + (dir, file) = splitName name + (base, ext) = splitExt file + [f] -> let + (dir, file) = splitName f + (base, _) = splitExt file + in return (f, dir, base) + _ -> onlyOne "output file" - specials = [(pos, key, arg) | Special pos key arg <- toks] + let cProgName = outDir++outBase++"_hsc_make.c" + oProgName = outDir++outBase++"_hsc_make.o" + progName = outDir++outBase++"_hsc_make" ++ progNameSuffix + outHFile = outBase++"_hsc.h" + outHName = outDir++outHFile + outCName = outDir++outBase++"_hsc.c" + + let execProgName + | null outDir = '.':pathSep:progName + | otherwise = progName - needsC = any (\(_, key, _) -> key == "def") specials - needsH = needsC + let specials = [(pos, key, arg) | Special pos key arg <- toks] - includeGuard = map fixChar outHName - where - fixChar c | isAlphaNum c = toUpper c - | otherwise = '_' + let needsC = any (\(_, key, _) -> key == "def") specials + needsH = needsC - in do + let includeGuard = map fixChar outHName + where + fixChar c | isAlphaNum c = toUpper c + | otherwise = '_' compiler <- case [c | Compiler c <- flags] of [] -> return "ghc" [c] -> return c _ -> onlyOne "compiler" + linker <- case [l | Linker l <- flags] of - [] -> return "gcc" + [] -> return cGCC [l] -> return l _ -> onlyOne "linker" - + writeFile cProgName $ - concat ["#include \""++t++"\"\n" | Template t <- flags]++ - concat ["#include "++f++"\n" | Include f <- flags]++ - outHeaderCProg specials++ + concatMap outFlagHeaderCProg flags++ + concatMap outHeaderCProg specials++ "\nint main (void)\n{\n"++ outHeaderHs flags (if needsH then Just outHName else Nothing) specials++ - outHsLine (newPos name 0 1)++ + outHsLine (SourcePos name 0)++ concatMap outTokenHs toks++ " return 0;\n}\n" + unless (null [() | NoCompile <- flags]) $ exitWith ExitSuccess + compilerStatus <- system $ compiler++ " -c"++ @@ -247,68 +536,89 @@ output flags name toks = let _ -> return () removeFile oProgName - system (execProgName++" >"++outHsName) + progStatus <- system (execProgName++" >"++outName) removeFile progName + case progStatus of + e@(ExitFailure _) -> exitWith e + _ -> return () when needsH $ writeFile outHName $ - "#ifndef "++includeGuard++"\n\ - \#define "++includeGuard++"\n\ - \#if __GLASGOW_HASKELL__ && __GLASGOW_HASKELL__ < 409\n\ - \#include \n\ - \#endif\n\ - \#include \n"++ - concat ["#include "++n++"\n" | Include n <- flags]++ + "#ifndef "++includeGuard++"\n\ + \#define "++includeGuard++"\n\ + \#if " ++ + "__GLASGOW_HASKELL__ && __GLASGOW_HASKELL__ < 409\n\ + \#include \n\ + \#endif\n\ + \#include \n\ + \#if __NHC__\n\ + \#undef HsChar\n\ + \#define HsChar int\n\ + \#endif\n"++ + concatMap outFlagH flags++ concatMap outTokenH specials++ "#endif\n" when needsC $ writeFile outCName $ - "#include \""++outHName++"\"\n"++ + "#include \""++outHFile++"\"\n"++ concatMap outTokenC specials + -- NB. outHFile not outHName; works better when processed + -- by gcc or mkdependC. onlyOne :: String -> IO a onlyOne what = do putStrLn ("Only one "++what++" may be specified") exitFailure -outHeaderCProg :: [(SourcePos, String, String)] -> String -outHeaderCProg = - concatMap $ \(pos, key, arg) -> outCLine pos ++ case key of - "include" -> "#include "++arg++"\n" - "define" -> "#define "++arg++"\n" - "undef" -> "#undef "++arg++"\n" - "def" -> case arg of - 's':'t':'r':'u':'c':'t':' ':_ -> arg++"\n" - 't':'y':'p':'e':'d':'e':'f':' ':_ -> arg++"\n" - _ -> "" - _ | conditional key -> "#"++key++" "++arg++"\n" - "let" -> case break (== '=') arg of - (_, "") -> "" - (header, _:body) -> case break isSpace header of - (name, args) -> - "#define hsc_"++name++"("++dropWhile isSpace args++") \ - \printf ("++joinLines body++");\n" +outFlagHeaderCProg :: Flag -> String +outFlagHeaderCProg (Template t) = "#include \""++t++"\"\n" +outFlagHeaderCProg (Include f) = "#include "++f++"\n" +outFlagHeaderCProg (Define n Nothing) = "#define "++n++"\n" +outFlagHeaderCProg (Define n (Just v)) = "#define "++n++" "++v++"\n" +outFlagHeaderCProg _ = "" + +outHeaderCProg :: (SourcePos, String, String) -> String +outHeaderCProg (pos, key, arg) = case key of + "include" -> outCLine pos++"#include "++arg++"\n" + "define" -> outCLine pos++"#define "++arg++"\n" + "undef" -> outCLine pos++"#undef "++arg++"\n" + "def" -> case arg of + 's':'t':'r':'u':'c':'t':' ':_ -> outCLine pos++arg++"\n" + 't':'y':'p':'e':'d':'e':'f':' ':_ -> outCLine pos++arg++"\n" _ -> "" + _ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n" + "let" -> case break (== '=') arg of + (_, "") -> "" + (header, _:body) -> case break isSpace header of + (name, args) -> + outCLine pos++ + "#define hsc_"++name++"("++dropWhile isSpace args++") \ + \printf ("++joinLines body++");\n" + _ -> "" where joinLines = concat . intersperse " \\\n" . lines outHeaderHs :: [Flag] -> Maybe String -> [(SourcePos, String, String)] -> String outHeaderHs flags inH toks = - "#if __GLASGOW_HASKELL__ && __GLASGOW_HASKELL__ < 409\n\ - \ printf (\"{-# OPTIONS -optc-D__GLASGOW_HASKELL__=%d #-}\\n\", \ - \__GLASGOW_HASKELL__);\n\ + "#if " ++ + "__GLASGOW_HASKELL__ && __GLASGOW_HASKELL__ < 409\n\ + \ printf (\"{-# OPTIONS -optc-D" ++ + "__GLASGOW_HASKELL__=%d #-}\\n\", \ + \__GLASGOW_HASKELL__);\n\ \#endif\n"++ - includeH++ - concatMap outSpecial toks + case inH of + Nothing -> concatMap outFlag flags++concatMap outSpecial toks + Just f -> outOption ("-#include \""++f++"\"") where - outSpecial (pos, key, arg) = outCLine pos ++ case key of - "include" -> case inH of - Nothing -> outOption ("-#include "++arg) - Just _ -> "" - "define" -> case inH of - Nothing | goodForOptD arg -> outOption ("-optc-D"++toOptD arg) - _ -> "" - _ | conditional key -> "#"++key++" "++arg++"\n" - _ -> "" + outFlag (Include f) = outOption ("-#include "++f) + outFlag (Define n Nothing) = outOption ("-optc-D"++n) + outFlag (Define n (Just v)) = outOption ("-optc-D"++n++"="++v) + outFlag _ = "" + outSpecial (pos, key, arg) = case key of + "include" -> outOption ("-#include "++arg) + "define" | goodForOptD arg -> outOption ("-optc-D"++toOptD arg) + | otherwise -> "" + _ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n" + _ -> "" goodForOptD arg = case arg of "" -> True c:_ | isSpace c -> True @@ -317,11 +627,6 @@ outHeaderHs flags inH toks = toOptD arg = case break isSpace arg of (name, "") -> name (name, _:value) -> name++'=':dropWhile isSpace value - includeH = concat [ - outOption ("-#include "++name++"") - | name <- case inH of - Nothing -> [name | Include name <- flags] - Just name -> ["\""++name++"\""]] outOption s = " printf (\"{-# OPTIONS %s #-}\\n\", \""++ showCString s++"\");\n" @@ -336,82 +641,104 @@ outTokenHs (Text pos text) = where outText s = " fputs (\""++showCString s++"\", stdout);\n" outTokenHs (Special pos key arg) = - outCLine pos ++ case key of + case key of "include" -> "" "define" -> "" "undef" -> "" "def" -> "" - _ | conditional key -> "#"++key++" "++arg++"\n" + _ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n" "let" -> "" - _ -> " hsc_"++key++" ("++arg++");\n" + "enum" -> outCLine pos++outEnum arg + _ -> outCLine pos++" hsc_"++key++" ("++arg++");\n" + +outEnum :: String -> String +outEnum arg = + case break (== ',') arg of + (_, []) -> "" + (t, _:afterT) -> case break (== ',') afterT of + (f, afterF) -> let + enums [] = "" + enums (_:s) = case break (== ',') s of + (enum, rest) -> let + this = case break (== '=') $ dropWhile isSpace enum of + (name, []) -> + " hsc_enum ("++t++", "++f++", \ + \hsc_haskellize (\""++name++"\"), "++ + name++");\n" + (hsName, _:cName) -> + " hsc_enum ("++t++", "++f++", \ + \printf (\"%s\", \""++hsName++"\"), "++ + cName++");\n" + in this++enums rest + in enums afterF + +outFlagH :: Flag -> String +outFlagH (Include f) = "#include "++f++"\n" +outFlagH (Define n Nothing) = "#define "++n++"\n" +outFlagH (Define n (Just v)) = "#define "++n++" "++v++"\n" +outFlagH _ = "" outTokenH :: (SourcePos, String, String) -> String outTokenH (pos, key, arg) = - outCLine pos ++ case key of - "include" -> "#include "++arg++"\n" - "define" -> "#define " ++arg++"\n" - "undef" -> "#undef " ++arg++"\n" - "def" -> case arg of + case key of + "include" -> outCLine pos++"#include "++arg++"\n" + "define" -> outCLine pos++"#define " ++arg++"\n" + "undef" -> outCLine pos++"#undef " ++arg++"\n" + "def" -> outCLine pos++case arg of 's':'t':'r':'u':'c':'t':' ':_ -> arg++"\n" 't':'y':'p':'e':'d':'e':'f':' ':_ -> arg++"\n" 'i':'n':'l':'i':'n':'e':' ':_ -> - "#ifdef __GNUC__\n\ - \extern\n\ + "#ifdef __GNUC__\n\ + \extern\n\ \#endif\n"++ arg++"\n" _ -> "extern "++header++";\n" - where header = takeWhile (\c -> c/='{' && c/='=') arg - _ | conditional key -> "#"++key++" "++arg++"\n" + where header = takeWhile (\c -> c /= '{' && c /= '=') arg + _ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n" _ -> "" outTokenC :: (SourcePos, String, String) -> String outTokenC (pos, key, arg) = - outCLine pos ++ case key of + case key of "def" -> case arg of 's':'t':'r':'u':'c':'t':' ':_ -> "" 't':'y':'p':'e':'d':'e':'f':' ':_ -> "" - 'i':'n':'l':'i':'n':'e':' ':_ -> - "#ifndef __GNUC__\n\ - \extern\n\ - \#endif\n"++ - header++ - "\n#ifndef __GNUC__\n\ - \;\n\ - \#else\n"++ - body++ - "\n#endif\n" - _ -> arg++"\n" - where (header, body) = span (\c -> c/='{' && c/='=') arg - _ | conditional key -> "#"++key++" "++arg++"\n" + 'i':'n':'l':'i':'n':'e':' ':arg' -> + case span (\c -> c /= '{' && c /= '=') arg' of + (header, body) -> + outCLine pos++ + "#ifndef __GNUC__\n\ + \extern inline\n\ + \#endif\n"++ + header++ + "\n#ifndef __GNUC__\n\ + \;\n\ + \#else\n"++ + body++ + "\n#endif\n" + _ -> outCLine pos++arg++"\n" + _ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n" _ -> "" conditional :: String -> Bool -conditional "if" = True -conditional "ifdef" = True -conditional "ifndef" = True -conditional "elif" = True -conditional "else" = True -conditional "endif" = True -conditional "error" = True -conditional _ = False - -sourceFileName :: SourcePos -> String -sourceFileName pos = fileName (sourceName pos) - where - fileName s = case break (== '/') s of - (name, []) -> name - (_, _:rest) -> fileName rest +conditional "if" = True +conditional "ifdef" = True +conditional "ifndef" = True +conditional "elif" = True +conditional "else" = True +conditional "endif" = True +conditional "error" = True +conditional "warning" = True +conditional _ = False outCLine :: SourcePos -> String -outCLine pos = - "# "++show (sourceLine pos)++ - " \""++showCString (sourceFileName pos)++"\"\n" +outCLine (SourcePos name line) = + "# "++show line++" \""++showCString (snd (splitName name))++"\"\n" outHsLine :: SourcePos -> String -outHsLine pos = - " hsc_line ("++ - show (sourceLine pos + 1)++", \""++ - showCString (sourceFileName pos)++"\");\n" +outHsLine (SourcePos name line) = + " hsc_line ("++show (line + 1)++", \""++ + showCString (snd (splitName name))++"\");\n" showCString :: String -> String showCString = concatMap showCChar @@ -432,3 +759,4 @@ showCString = concatMap showCChar intToDigit (ord c `quot` 64), intToDigit (ord c `quot` 8 `mod` 8), intToDigit (ord c `mod` 8)] +