CodeOutput.lhs 7.76 KB
Newer Older
1 2 3 4 5 6
%
% (c) The GRASP/AQUA Project, Glasgow University, 1993-1998
%
\section{Code output phase}

\begin{code}
7
module CodeOutput( codeOutput, outputForeignStubs ) where
8 9 10

#include "HsVersions.h"

11
#ifndef OMIT_NATIVE_CODEGEN
12
import UniqSupply	( mkSplitUniqSupply )
13 14
import AsmCodeGen	( nativeCodeGen )
#endif
15

16 17 18
#ifdef ILX
import IlxGen		( ilxGen )
#endif
19

20
#ifdef JAVA
21
import JavaGen		( javaGen )
22
import OccurAnal	( occurAnalyseBinds )
23
import qualified PrintJava
24
import OccurAnal	( occurAnalyseBinds )
25
#endif
26

27
import FastString	( unpackFS )
28
import DriverState	( v_HCHeader )
29 30
import Id		( Id )
import StgSyn		( StgBinding )
31
import AbsCSyn		( AbstractC )
32
import PprAbsC		( dumpRealC, writeRealC )
33
import HscTypes		( ModGuts(..), ModGuts, ForeignStubs(..), typeEnvTyCons )
34
import CmdLineOpts
35
import ErrUtils		( dumpIfSet_dyn, showPass )
36
import Outputable
37
import Pretty		( Mode(..), printDoc )
38
import CmdLineOpts	( DynFlags, HscLang(..), dopt_OutName )
39
import DATA_IOREF	( readIORef, writeIORef )
40
import Monad		( when )
41
import IO
42 43 44
\end{code}


45 46 47 48 49 50
%************************************************************************
%*									*
\subsection{Steering}
%*									*
%************************************************************************

51
\begin{code}
52
codeOutput :: DynFlags
53
	   -> ModGuts
54
	   -> [(StgBinding,[Id])]	-- The STG program with SRTs
55
	   -> AbstractC			-- Compiled abstract C
56
	   -> IO (Bool{-stub_h_exists-}, Bool{-stub_c_exists-})
57 58 59 60 61 62 63 64 65 66
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),
67 68 69
    -- but not both.  [Allowing for both gives a space leak on
    -- flat_abstractC.  WDP 94/10]

70
    -- Dunno if the above comment is still meaningful now.  JRS 001024.
71

72 73
    do	{ showPass dflags "CodeOutput"
	; let filenm = dopt_OutName dflags 
74
	; stub_names <- outputForeignStubs dflags foreign_stubs
75 76 77 78
	; case dopt_HscLang dflags of
             HscInterpreted -> return stub_names
             HscAsm         -> outputAsm dflags filenm flat_abstractC
          		       >> return stub_names
79
             HscC           -> outputC dflags filenm flat_abstractC stub_names
80
          		       >> return stub_names
81 82 83
             HscJava        -> 
#ifdef JAVA
			       outputJava dflags filenm mod_name tycons core_binds
84
          		       >> return stub_names
85 86 87
#else
                               panic "Java support not compiled into this ghc"
#endif
88
	     HscILX         -> 
rrt's avatar
rrt committed
89
#ifdef ILX
90
	                       outputIlx dflags filenm mod_name tycons stg_binds
91
			       >> return stub_names
92 93
#else
                               panic "ILX support not compiled into this ghc"
rrt's avatar
rrt committed
94
#endif
95
	}
96

97
doOutput :: String -> (Handle -> IO ()) -> IO ()
98
doOutput filenm io_action = bracket (openFile filenm WriteMode) hClose io_action
99 100 101 102 103 104 105 106 107 108
\end{code}


%************************************************************************
%*									*
\subsection{C}
%*									*
%************************************************************************

\begin{code}
109
outputC dflags filenm flat_absC (stub_h_exists, _)
110
  = do dumpIfSet_dyn dflags Opt_D_dump_realC "Real C" (dumpRealC flat_absC)
111 112 113
       header <- readIORef v_HCHeader
       doOutput filenm $ \ h -> do
	  hPutStr h header
114 115
	  when stub_h_exists $ 
	     hPutStrLn h ("#include \"" ++ (hscStubHOutName dflags) ++ "\"")
116
	  writeRealC h flat_absC
117 118 119 120 121 122 123 124 125 126
\end{code}


%************************************************************************
%*									*
\subsection{Assembler}
%*									*
%************************************************************************

\begin{code}
127
outputAsm dflags filenm flat_absC
128

129
#ifndef OMIT_NATIVE_CODEGEN
130

131
  = do ncg_uniqs <- mkSplitUniqSupply 'n'
132 133
       let (stix_final, ncg_output_d) = _scc_ "NativeCodeGen" 
				        nativeCodeGen flat_absC ncg_uniqs
134
       dumpIfSet_dyn dflags Opt_D_dump_stix "Final stix code" stix_final
135 136 137
       dumpIfSet_dyn dflags Opt_D_dump_asm "Asm code" (docToSDoc ncg_output_d)
       _scc_ "OutputAsm" doOutput filenm $
	   \f -> printDoc LeftMode f ncg_output_d
138 139 140 141
  where

#else /* OMIT_NATIVE_CODEGEN */

142 143
  = pprPanic "This compiler was built without a native code generator"
	     (text "Use -fvia-C instead")
144 145 146

#endif
\end{code}
147 148


149 150 151 152 153
%************************************************************************
%*									*
\subsection{Java}
%*									*
%************************************************************************
154

155
\begin{code}
156
#ifdef JAVA
157
outputJava dflags filenm mod tycons core_binds
158
  = doOutput filenm (\ f -> printForUser f alwaysQualify pp_java)
159
	-- User style printing for now to keep indentation
160
  where
161 162 163
    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
164
    pp_java   = PrintJava.compilationUnit java_code
165
#endif
166
\end{code}
167

168

rrt's avatar
rrt committed
169 170 171 172 173 174 175 176
%************************************************************************
%*									*
\subsection{Ilx}
%*									*
%************************************************************************

\begin{code}
#ifdef ILX
177 178
outputIlx dflags filename mod tycons stg_binds
  =  doOutput filename (\ f -> printForC f pp_ilx)
rrt's avatar
rrt committed
179 180 181 182 183 184
  where
    pp_ilx = ilxGen mod tycons stg_binds
#endif
\end{code}


185 186 187 188 189 190 191
%************************************************************************
%*									*
\subsection{Foreign import/export}
%*									*
%************************************************************************

\begin{code}
192 193 194 195 196 197 198 199 200 201 202 203 204 205
    -- 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 _)
206
  = do
207
	dumpIfSet_dyn dflags Opt_D_dump_foreign
208 209
                      "Foreign export header file" stub_h_output_d

210 211
	stub_h_file_exists
           <- outputForeignStubs_help (hscStubHOutName dflags) stub_h_output_w
212
		("#include \"HsFFI.h\"\n" ++ cplusplus_hdr) cplusplus_ftr
213

214
	dumpIfSet_dyn dflags Opt_D_dump_foreign
215 216
                      "Foreign export stubs" stub_c_output_d

217 218 219 220
	  -- Extend the list of foreign headers (used in outputC)
        fhdrs <- readIORef v_HCHeader
	let new_fhdrs = fhdrs ++ mkForeignHeaders hdrs
        writeIORef v_HCHeader new_fhdrs
221

222 223
	stub_c_file_exists
           <- outputForeignStubs_help (hscStubCOutName dflags) stub_c_output_w
224
		("#define IN_STG_CODE 0\n" ++ 
225
		 new_fhdrs ++
226 227 228
		 "#include \"RtsAPI.h\"\n" ++
		 cplusplus_hdr)
		 cplusplus_ftr
229
	   -- We're adding the default hc_header to the stub file, but this
230 231
	   -- isn't really HC code, so we need to define IN_STG_CODE==0 to
	   -- avoid the register variables etc. being enabled.
232 233

        return (stub_h_file_exists, stub_c_file_exists)
234
  where
235 236 237 238 239 240 241 242
    -- C stubs for "foreign export"ed functions.
    stub_c_output_d = pprCode CStyle c_code
    stub_c_output_w = showSDoc stub_c_output_d

    -- Header file protos for "foreign export"ed functions.
    stub_h_output_d = pprCode CStyle h_code
    stub_h_output_w = showSDoc stub_h_output_d

243 244
cplusplus_hdr = "#ifdef __cplusplus\nextern \"C\" {\n#endif\n"
cplusplus_ftr = "#ifdef __cplusplus\n}\n#endif\n"
245

246 247 248
-- 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.
249 250 251
outputForeignStubs_help fname ""      header footer = return False
outputForeignStubs_help fname doc_str header footer
   = do writeFile fname (header ++ doc_str ++ '\n':footer ++ "\n")
252
        return True
253 254
\end{code}