X-Git-Url: https://ruin.nu/git/?p=proglang.git;a=blobdiff_plain;f=CompInt.hs;h=1934447c5e1110079dc94f93cf2867a9cfb52c47;hp=1188466da980b4d1886b4d3deff9d5e199fa08fc;hb=HEAD;hpb=cfe5796efb06251047f77bf9d2295d7093910292

diff --git a/CompInt.hs b/CompInt.hs
index 1188466..1934447 100644
--- a/CompInt.hs
+++ b/CompInt.hs
@@ -14,7 +14,6 @@ import Abssyntax
 import Typecheck
 import Interpret
 import Compile
-import Data.Map as Map hiding (showTree)
 
 import ErrM
 
@@ -22,33 +21,37 @@ type ParseFun a = [Token] -> Err a
 
 myLLexer = myLexer
 
-
-cHeader = "#include <stdio.h>\nint read(){\nint n;\nscanf(\"%d\",&n);\nreturn n;\n}\nint main(void){\n"
-
-cFooter = "return 0;}"
+splitFunStm :: [FuncStm] -> ([Func],[Stm])
+splitFunStm [] = ([],[])
+splitFunStm ((F f):fss) = let (fs,ss) = splitFunStm fss in (f:fs,ss)
+splitFunStm ((S s):fss) = let (fs,ss) = splitFunStm fss in (fs,s:ss)
 
 putStrV :: Int -> String -> IO ()
 putStrV v s = if v > 1 then putStrLn s else return ()
 
-runFile :: ([Stm] -> IO()) -> ParseFun Stms -> FilePath -> IO ()
+runFile :: ([Func] -> [Stm] -> IO()) -> ParseFun Program -> FilePath -> IO ()
 runFile e p f = readFile f >>= run e p
 
-run :: ([Stm] -> IO()) -> ParseFun Stms -> String -> IO ()
+run :: ([Func] -> [Stm] -> IO()) -> ParseFun Program -> String -> IO ()
 run e p s = let ts = myLLexer s in case p ts of
 	Bad s    -> do
 		putStrLn "\nParse              Failed...\n"
 		putStrLn "Tokens:"
 		putStrLn $ show ts
 		putStrLn s
-	Ok (Program s) -> do
-		typeCheck s
-		e s
+	Ok (Program s) -> let (fun,st) = splitFunStm (s) in do
+		typeCheck fun st
+		e fun st
 
 main :: IO ()
 main = do 
 	args <- getArgs
 	case args of
-		[] -> hGetContents stdin >>= run interpret pStms
-		"-c":f:[] -> runFile (writeFile (f++".c") . compile)  pStms f
-		f:[] -> runFile interpret pStms f
-		_ -> print "Too many arguments"
+		[] -> hGetContents stdin >>= run interpret pProgram
+		"-c":f:[] -> let file = (f++".c") in do
+			putStrLn $ "Compiling "++f++" to the C99-compatible file:"++file
+			runFile (\fun st -> writeFile file $ compile fun st)  pProgram f
+		f:[] -> runFile interpret pProgram f
+		_ -> do
+			putStrLn "Usage: ./CompInt [-c] <file>"
+			putStrLn "-c : compile <file> to C99-compatible file"