import AST as K -- K for Kaleidoscope
import Utils
+import Control.Monad.Trans.Class
import Control.Monad.Trans.Reader
import Control.Monad.IO.Class
import Data.String
import LLVM.AST.FloatingPointPredicate hiding (False, True)
import LLVM.AST.Operand
import LLVM.AST.Type as Type
+import LLVM.Context
import LLVM.IRBuilder
+import LLVM.Module
+import LLVM.PassManager
import LLVM.Pretty
+import LLVM.Target
import System.IO
import System.IO.Error
import Text.Read (readMaybe)
main :: IO ()
-main = buildModuleT "main" repl >>= Text.hPutStrLn stderr . ("\n" <>) . ppll
+main = do
+ withContext $ \ctx -> withHostTargetMachine $ \tm -> do
+ ast <- runReaderT (buildModuleT "main" repl) ctx
+ return ()
-repl :: ModuleBuilderT IO ()
+repl :: ModuleBuilderT (ReaderT Context IO) ()
repl = do
liftIO $ hPutStr stderr "ready> "
mline <- liftIO $ catchIOError (Just <$> getLine) eofHandler
Just ast -> do
hoist $ buildAST ast
mostRecentDef >>= liftIO . Text.hPutStrLn stderr . ppll
+
+ ast <- moduleSoFar "main"
+ ctx <- lift ask
+ liftIO $ withModuleFromAST ctx ast $ \mdl -> do
+ let spec = defaultCuratedPassSetSpec { optLevel = Just 3 }
+ -- this returns true if the module was modified
+ withPassManager spec $ flip runPassManager mdl
+ Text.hPutStrLn stderr . ("\n" <>) . ppllvm =<< moduleAST mdl
repl
where
eofHandler e