X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Fmain%2FCodeOutput.lhs;h=166c0990e694cb91596bbd26ddf2786efcc4de92;hb=d254a44b8392ff0a4327f1916ef921887ce78769;hp=a8a1a0a3b354097b6794cc4ac5eb15440ab69976;hpb=bca9dd54c2b39638cb4638aaccf6015a104a1df5;p=ghc-hetmet.git diff --git a/ghc/compiler/main/CodeOutput.lhs b/ghc/compiler/main/CodeOutput.lhs index a8a1a0a..166c099 100644 --- a/ghc/compiler/main/CodeOutput.lhs +++ b/ghc/compiler/main/CodeOutput.lhs @@ -4,11 +4,12 @@ \section{Code output phase} \begin{code} -module CodeOutput( codeOutput ) where +module CodeOutput( codeOutput, outputForeignStubs ) where #include "HsVersions.h" #ifndef OMIT_NATIVE_CODEGEN +import UniqSupply ( mkSplitUniqSupply ) import AsmCodeGen ( nativeCodeGen ) #endif @@ -16,24 +17,29 @@ import AsmCodeGen ( nativeCodeGen ) import IlxGen ( ilxGen ) #endif +#ifdef JAVA import JavaGen ( javaGen ) import qualified PrintJava +#endif +import DriverState ( v_HCHeader ) import TyCon ( TyCon ) import Id ( Id ) import CoreSyn ( CoreBind ) +import OccurAnal ( occurAnalyseBinds ) import StgSyn ( StgBinding ) import AbsCSyn ( AbstractC ) import PprAbsC ( dumpRealC, writeRealC ) import Module ( Module ) import CmdLineOpts -import ErrUtils ( dumpIfSet_dyn ) +import ErrUtils ( dumpIfSet_dyn, showPass ) import Outputable +import Pretty ( Mode(..), printDoc ) import CmdLineOpts ( DynFlags, HscLang(..), dopt_OutName ) -import TmpFiles ( newTempName ) -import UniqSupply ( mkSplitUniqSupply ) -import IO ( IOMode(..), hClose, openFile, Handle ) +import IOExts +import Monad ( when ) +import IO \end{code} @@ -52,7 +58,7 @@ codeOutput :: DynFlags -> SDoc -- C stubs for foreign exported functions -> SDoc -- Header file prototype for foreign exported functions -> AbstractC -- Compiled abstract C - -> IO (Maybe FilePath, Maybe FilePath) + -> IO (Bool{-stub_h_exists-}, Bool{-stub_c_exists-}) codeOutput dflags mod_name tycons core_binds stg_binds c_code h_code flat_abstractC = -- You can have C (c_output) or assembly-language (ncg_output), @@ -61,24 +67,33 @@ codeOutput dflags mod_name tycons core_binds stg_binds -- Dunno if the above comment is still meaningful now. JRS 001024. - do let filenm = dopt_OutName dflags - stub_names <- outputForeignStubs dflags c_code h_code - case dopt_HscLang dflags of - HscInterpreted -> return stub_names - HscAsm -> outputAsm dflags filenm flat_abstractC - >> return stub_names - HscC -> outputC dflags filenm flat_abstractC - >> return stub_names - HscJava -> outputJava dflags filenm mod_name tycons core_binds - >> return stub_names + do { showPass dflags "CodeOutput" + ; let filenm = dopt_OutName dflags + ; stub_names <- outputForeignStubs dflags c_code h_code + ; case dopt_HscLang dflags of + HscInterpreted -> return stub_names + HscAsm -> outputAsm dflags filenm flat_abstractC + >> return stub_names + HscC -> outputC dflags filenm flat_abstractC stub_names + >> return stub_names + HscJava -> +#ifdef JAVA + outputJava dflags filenm mod_name tycons core_binds + >> return stub_names +#else + panic "Java support not compiled into this ghc" +#endif + HscILX -> +#ifdef ILX + outputIlx dflags filenm mod_name tycons stg_binds + >> return stub_names +#else + panic "ILX support not compiled into this ghc" +#endif + } doOutput :: String -> (Handle -> IO ()) -> IO () -doOutput filenm io_action - = (do handle <- openFile filenm WriteMode - io_action handle - hClose handle) - `catch` (\err -> pprPanic "Failed to open or write code output file" - (text filenm)) +doOutput filenm io_action = bracket (openFile filenm WriteMode) hClose io_action \end{code} @@ -89,9 +104,14 @@ doOutput filenm io_action %************************************************************************ \begin{code} -outputC dflags filenm flat_absC +outputC dflags filenm flat_absC (stub_h_exists, _) = do dumpIfSet_dyn dflags Opt_D_dump_realC "Real C" (dumpRealC flat_absC) - doOutput filenm (\ h -> writeRealC h flat_absC) + header <- readIORef v_HCHeader + doOutput filenm $ \ h -> do + hPutStr h header + when stub_h_exists $ + hPutStrLn h ("#include \"" ++ (hscStubHOutName dflags) ++ "\"") + writeRealC h flat_absC \end{code} @@ -107,10 +127,12 @@ outputAsm dflags filenm flat_absC #ifndef OMIT_NATIVE_CODEGEN = do ncg_uniqs <- mkSplitUniqSupply 'n' - let (stix_final, ncg_output_d) = nativeCodeGen flat_absC ncg_uniqs + let (stix_final, ncg_output_d) = _scc_ "NativeCodeGen" + nativeCodeGen flat_absC ncg_uniqs dumpIfSet_dyn dflags Opt_D_dump_stix "Final stix code" stix_final - dumpIfSet_dyn dflags Opt_D_dump_asm "Asm code" ncg_output_d - doOutput filenm ( \f -> printForAsm f ncg_output_d) + dumpIfSet_dyn dflags Opt_D_dump_asm "Asm code" (docToSDoc ncg_output_d) + _scc_ "OutputAsm" doOutput filenm $ + \f -> printDoc LeftMode f ncg_output_d where #else /* OMIT_NATIVE_CODEGEN */ @@ -129,12 +151,32 @@ outputAsm dflags filenm flat_absC %************************************************************************ \begin{code} +#ifdef JAVA outputJava dflags filenm mod tycons core_binds - = doOutput filenm (\ f -> printForUser f pp_java) + = doOutput filenm (\ f -> printForUser f alwaysQualify pp_java) -- User style printing for now to keep indentation where - java_code = javaGen mod [{- Should be imports-}] tycons core_binds + occ_anal_binds = occurAnalyseBinds core_binds + -- Make sure we have up to date dead-var information + java_code = javaGen mod [{- Should be imports-}] tycons occ_anal_binds pp_java = PrintJava.compilationUnit java_code +#endif +\end{code} + + +%************************************************************************ +%* * +\subsection{Ilx} +%* * +%************************************************************************ + +\begin{code} +#ifdef ILX +outputIlx dflags filename mod tycons stg_binds + = doOutput filename (\ f -> printForC f pp_ilx) + where + pp_ilx = ilxGen mod tycons stg_binds +#endif \end{code} @@ -150,16 +192,27 @@ outputForeignStubs dflags c_code h_code dumpIfSet_dyn dflags Opt_D_dump_foreign "Foreign export header file" stub_h_output_d - maybe_stub_h_file - <- outputForeignStubs_help True{-.h output-} stub_h_output_w + stub_h_file_exists + <- outputForeignStubs_help (hscStubHOutName dflags) stub_h_output_w + ("#include \"HsFFI.h\"\n" ++ cplusplus_hdr) cplusplus_ftr dumpIfSet_dyn dflags Opt_D_dump_foreign "Foreign export stubs" stub_c_output_d - maybe_stub_c_file - <- outputForeignStubs_help False{-not .h-} stub_c_output_w + hc_header <- readIORef v_HCHeader - return (maybe_stub_h_file, maybe_stub_c_file) + stub_c_file_exists + <- outputForeignStubs_help (hscStubCOutName dflags) stub_c_output_w + ("#define IN_STG_CODE 0\n" ++ + hc_header ++ + "#include \"RtsAPI.h\"\n" ++ + cplusplus_hdr) + cplusplus_ftr + -- we're adding the default hc_header to the stub file, but this + -- isn't really HC code, so we need to define IN_STG_CODE==0 to + -- avoid the register variables etc. being enabled. + + return (stub_h_file_exists, stub_c_file_exists) where -- C stubs for "foreign export"ed functions. stub_c_output_d = pprCode CStyle c_code @@ -169,21 +222,15 @@ outputForeignStubs dflags c_code h_code stub_h_output_d = pprCode CStyle h_code stub_h_output_w = showSDoc stub_h_output_d +cplusplus_hdr = "#ifdef __cplusplus\nextern \"C\" {\n#endif\n" +cplusplus_ftr = "#ifdef __cplusplus\n}\n#endif\n" -- Don't use doOutput for dumping the f. export stubs -- since it is more than likely that the stubs file will -- turn out to be empty, in which case no file should be created. -outputForeignStubs_help is_header "" = return Nothing -outputForeignStubs_help is_header doc_str - = newTempName suffix >>= \ fname -> - writeFile fname (include_prefix ++ doc_str) >> - return (Just suffix) - where - suffix - | is_header = "h_stub" - | otherwise = "c_stub" - include_prefix - | is_header = "#include \"HsFFI.h\"\n" - | otherwise = "#include \"RtsAPI.h\"\n" +outputForeignStubs_help fname "" header footer = return False +outputForeignStubs_help fname doc_str header footer + = do writeFile fname (header ++ doc_str ++ '\n':footer ++ "\n") + return True \end{code}