debug: use #ifdef guards

[mate.git] / Mate.hs

diff --git a/Mate.hs b/Mate.hs
index 8e63fbf578132bd2c3afeeeae57700583c3fccd8..5e6df74f9c77f6925d62ded17039f13676af621b 100644 (file)
--- a/Mate.hs
+++ b/Mate.hs
@@ -1,17 +1,18 @@
+{-# LANGUAGE CPP #-}
 {-# LANGUAGE OverloadedStrings #-}
 module Main where
 
 import System.Environment
 import Data.Char
-import Data.String.Utils
 import Data.List
 import qualified Data.ByteString.Lazy as B
 
+#ifdef DEBUG
 import Text.Printf
 
-import JVM.ClassFile
-import JVM.Converter
 import JVM.Dump
+#endif
+import JVM.ClassFile
 
 import Mate.BasicBlocks
 import Mate.X86CodeGen
@@ -28,17 +29,22 @@ main = do
     [clspath] -> do
       let bclspath = B.pack $ map (fromIntegral . ord) clspath
       cls <- getClassFile bclspath
+#ifdef DEBUG
       dumpClass cls
+#endif
       hmap <- parseMethod cls "main"
-      printMapBB hmap
       case hmap of
         Just hmap' -> do
           let methods = classMethods cls; methods :: [Method Resolved]
           let method = find (\x -> (methodName x) == "main") methods
           case method of
             Just m -> do
-              entry <- compileBB hmap' (MethodInfo "main" bclspath (methodSignature m))
+              let mi = (MethodInfo "main" bclspath (methodSignature m))
+              entry <- compileBB hmap' mi
+              addMethodRef entry mi [bclspath]
+#ifdef DEBUG
               printf "executing `main' now:\n"
+#endif
               executeFuncPtr entry
             Nothing -> error "main not found"
         Nothing -> error "main not found"