Changeset 22244 for v6

Timestamp:

09/14/08 21:54:19 (3 months ago)

Author:

pmurias

Message:

[m0ld] added a --print-bytecode to print bytecode

Files:

• v6/smop/m0ld/M0ld.hs (modified) (5 diffs)

v6/smop/m0ld/M0ld.hs

@@ -1 +1 @@
 import Text.ParserCombinators.Parsec hiding (label)
 import qualified Data.Map as Map
-import System.IO hiding (getContents,putStrLn,print)
-import Prelude hiding (getContents,putStrLn,print)
+import System.IO hiding (getContents,putStrLn,print,putStr)
+import Prelude hiding (getContents,putStrLn,print,putStr)
 import System.IO.UTF8
 import Debug.Trace
+import System.Console.GetOpt
+import System.Environment
 
 type Register = [Char]
@@ -240 +242 @@
 emit stmts regMap labelsMap = concatMap (\op -> toBytecode op regMap labelsMap) stmts ++ [0]
 
+joinStr sep [] = ""
 joinStr sep list = foldl (\a b -> a ++ sep ++ b) (head list) (tail list)
 
@@ -269 +272 @@
         ++ "})"
 
+prettyPrintBytecode stmts =
+    let labelsMap = mapLabels stmts
+        regMap    = mapRegisters stmts
+        freeRegs  = countRegister stmts
+        prettyPrintOp (Decl _ _) = ""
+        prettyPrintOp op = (joinStr " " $ ( map show (toBytecode op regMap labelsMap))) ++ "\n"
+        in concat $ map prettyPrintOp stmts
+
 type ImplicitDecls = Map.Map Value [Char]
 
@@ -274 +285 @@
 
 main = do
+    args <- getArgs
+    let (options,nonoptions,errors) =  getOpt RequireOrder [Option [] ["print-bytecode"] (NoArg "print-bytecode") "print resulting mold bytecode in a human readable form"] args 
+    mapM putStr errors
     hFlush stdout
     line <- getContents
@@ -279 +293 @@
         Left err      -> error  $ show err
         Right (stmts,constants) -> do 
-            putStrLn $ dumpToC $ (implicitDecls constants) ++ stmts
+            if elem "print-bytecode" options then putStrLn $ prettyPrintBytecode $ (implicitDecls constants) ++ stmts
+                else putStrLn $ dumpToC $ (implicitDecls constants) ++ stmts