\section{Code output phase}
\begin{code}
-module CodeOutput( codeOutput ) where
+module CodeOutput( codeOutput, outputForeignStubs ) where
#include "HsVersions.h"
import IlxGen ( ilxGen )
#endif
+#ifdef JAVA
import JavaGen ( javaGen )
+import OccurAnal ( occurAnalyseBinds )
import qualified PrintJava
+import OccurAnal ( occurAnalyseBinds )
+#endif
+import FastString ( unpackFS )
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 HscTypes ( ModGuts(..), ModGuts, ForeignStubs(..), typeEnvTyCons )
import CmdLineOpts
import ErrUtils ( dumpIfSet_dyn, showPass )
import Outputable
+import Pretty ( Mode(..), printDoc )
import CmdLineOpts ( DynFlags, HscLang(..), dopt_OutName )
-
-import IOExts
+import DATA_IOREF ( readIORef, writeIORef )
import Monad ( when )
import IO
\end{code}
\begin{code}
codeOutput :: DynFlags
- -> Module
- -> [TyCon] -- Local tycons
- -> [CoreBind] -- Core bindings
+ -> ModGuts
-> [(StgBinding,[Id])] -- The STG program with SRTs
- -> SDoc -- C stubs for foreign exported functions
- -> SDoc -- Header file prototype for foreign exported functions
- -> AbstractC -- Compiled abstract C
+ -> AbstractC -- Compiled abstract C
-> 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),
+codeOutput dflags
+ (ModGuts {mg_module = mod_name,
+ mg_types = type_env,
+ mg_foreign = foreign_stubs,
+ mg_binds = core_binds})
+ stg_binds flat_abstractC
+ = let
+ tycons = typeEnvTyCons type_env
+ in
+ -- You can have C (c_output) or assembly-language (ncg_output),
-- but not both. [Allowing for both gives a space leak on
-- flat_abstractC. WDP 94/10]
do { showPass dflags "CodeOutput"
; let filenm = dopt_OutName dflags
- ; stub_names <- outputForeignStubs dflags c_code h_code
+ ; stub_names <- outputForeignStubs dflags foreign_stubs
; 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 -> outputJava dflags filenm mod_name tycons core_binds
+ 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
}
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}
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
- _scc_ "OutputAsm" 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 */
%************************************************************************
\begin{code}
+#ifdef JAVA
outputJava dflags filenm mod tycons core_binds
= doOutput filenm (\ f -> printForUser f alwaysQualify pp_java)
-- User style printing for now to keep indentation
-- 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}
%************************************************************************
\begin{code}
-outputForeignStubs dflags c_code h_code
+ -- Turn the list of headers requested in foreign import
+ -- declarations into a string suitable for emission into generated
+ -- C code...
+mkForeignHeaders headers
+ = unlines
+ . map (\fname -> "#include \"" ++ unpackFS fname ++ "\"")
+ . reverse
+ $ headers
+
+outputForeignStubs :: DynFlags -> ForeignStubs
+ -> IO (Bool, -- Header file created
+ Bool) -- C file created
+outputForeignStubs dflags NoStubs = return (False, False)
+outputForeignStubs dflags (ForeignStubs h_code c_code hdrs _)
= do
dumpIfSet_dyn dflags Opt_D_dump_foreign
"Foreign export header file" stub_h_output_d
dumpIfSet_dyn dflags Opt_D_dump_foreign
"Foreign export stubs" stub_c_output_d
- hc_header <- readIORef v_HCHeader
+ -- Extend the list of foreign headers (used in outputC)
+ fhdrs <- readIORef v_HCHeader
+ let new_fhdrs = fhdrs ++ mkForeignHeaders hdrs
+ writeIORef v_HCHeader new_fhdrs
stub_c_file_exists
<- outputForeignStubs_help (hscStubCOutName dflags) stub_c_output_w
("#define IN_STG_CODE 0\n" ++
- hc_header ++
+ new_fhdrs ++
"#include \"RtsAPI.h\"\n" ++
cplusplus_hdr)
cplusplus_ftr
- -- we're adding the default hc_header to the stub file, but this
+ -- 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.