------------------------------------------------------------------------------
--- $Id: Main.hs,v 1.24 2001/03/04 11:18:03 qrczak Exp $
---
--- (originally "GlueHsc.hs" by Marcin 'Qrczak' Kowalczyk)
+------------------------------------------------------------------------
+-- $Id: Main.hs,v 1.35 2002/01/17 08:37:57 sof 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.
-- See the documentation in the Users' Guide for more details.
import GetOpt
-import System (getProgName, getArgs, ExitCode(..), exitWith, exitFailure)
-import KludgedSystem (system, defaultCompiler)
+import Config
+import System (getProgName, getArgs, ExitCode(..), exitWith, exitFailure, system)
import Directory (removeFile)
-import Parsec
-import ParsecError
-import Monad (liftM, liftM2, when)
-import Char (ord, intToDigit, isSpace, isAlpha, isAlphaNum, toUpper)
+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_TARGET_OS
+import Win32DLL
+#endif
+
version :: String
version = "hsc2hs-0.65"
data Flag
= Help
| Version
- | Template String
- | Compiler String
- | Linker String
- | CompFlag String
- | LinkFlag String
- | Keep
- | 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 "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 "" ["keep"] (NoArg Keep) "don't delete *.hs_make.c",
- Option "i" ["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_TARGET_OS
+ h <- getModuleHandle Nothing
+ n <- getModuleFileName h
+ 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
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
+ s <- readFile name
+ case parser of
+ Parser p -> case p (SourcePos name 1) s of
+ Success _ _ _ toks -> output flags name toks
+ Failure (SourcePos name' line) msg -> do
+ putStrLn (name'++":"++show line++": "++msg)
+ exitFailure
+
+------------------------------------------------------------------------
+-- 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")
- keyArg pos pzero <|> do
- char '{'
- skipMany (oneOf " \t")
- sp <- keyArg pos (string "\n")
- char '}'
- return sp
-
-keyArg :: SourcePos -> Parser String -> Parser Token
-keyArg pos eol = do
- key <- liftM2 (:) (letter <|> char '_') (many (alphaNum <|> char '_'))
- <?> "hsc directive"
- skipMany (oneOf " \t")
- arg <- argument eol
+ 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++"_make.c"
- oProgName = baseName++"_make.o"
- progName = baseName++"_make"
- 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
+ outHName = outDir++outBase++"_hsc.h"
+ 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 defaultCompiler
+ [] -> 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"++
case compilerStatus of
e@(ExitFailure _) -> exitWith e
_ -> return ()
- when (null [() | Keep <- flags]) $ removeFile cProgName
+ removeFile cProgName
linkerStatus <- system $
linker++
_ -> return ()
removeFile oProgName
- system (execProgName++" >"++outHsName)
+ system (execProgName++" >"++outName)
removeFile progName
when needsH $ writeFile outHName $
- "#ifndef "++includeGuard++"\n\
- \#define "++includeGuard++"\n\
- \#if __GLASGOW_HASKELL__ && __GLASGOW_HASKELL__ < 409\n\
- \#include <Rts.h>\n\
- \#endif\n\
- \#include <HsFFI.h>\n\
- \#if __NHC__\n\
- \#undef HsChar\n\
- \#define HsChar int\n\
+ "#ifndef "++includeGuard++"\n\
+ \#define "++includeGuard++"\n\
+ \#if " ++
+ "__GLASGOW_HASKELL__ && __GLASGOW_HASKELL__ < 409\n\
+ \#include <Rts.h>\n\
+ \#endif\n\
+ \#include <HsFFI.h>\n\
+ \#if __NHC__\n\
+ \#undef HsChar\n\
+ \#define HsChar int\n\
\#endif\n"++
- concat ["#include "++n++"\n" | Include n <- flags]++
+ concatMap outFlagH flags++
concatMap outTokenH specials++
"#endif\n"
putStrLn ("Only one "++what++" may be specified")
exitFailure
-outHeaderCProg :: [(SourcePos, String, String)] -> String
-outHeaderCProg =
- concatMap $ \(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"
+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
+ 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" -> case inH of
- Nothing -> outOption ("-#include "++arg)
- Just _ -> ""
- "define" -> case inH of
- Nothing | goodForOptD arg -> outOption ("-optc-D"++toOptD arg)
- _ -> ""
- _ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n"
- _ -> ""
+ "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
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"
(enum, rest) -> let
this = case break (== '=') $ dropWhile isSpace enum of
(name, []) ->
- " hsc_enum ("++t++", "++f++", \
+ " hsc_enum ("++t++", "++f++", \
\hsc_haskellize (\""++name++"\"), "++
name++");\n"
(hsName, _:cName) ->
- " hsc_enum ("++t++", "++f++", \
+ " 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) =
case key 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"
"def" -> case arg of
's':'t':'r':'u':'c':'t':' ':_ -> ""
't':'y':'p':'e':'d':'e':'f':' ':_ -> ""
- 'i':'n':'l':'i':'n':'e':' ':_ ->
- outCLine pos++
- "#ifndef __GNUC__\n\
- \extern\n\
- \#endif\n"++
- header++
- "\n#ifndef __GNUC__\n\
- \;\n\
- \#else\n"++
- body++
- "\n#endif\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"
- where (header, body) = span (\c -> c /= '{' && c /= '=') arg
_ | conditional key -> outCLine pos++"#"++key++" "++arg++"\n"
_ -> ""
conditional "warning" = True
conditional _ = False
-sourceFileName :: SourcePos -> String
-sourceFileName pos = fileName (sourceName pos)
- where
- fileName s = case break (== '/') s of
- (name, []) -> name
- (_, _:rest) -> fileName rest
-
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