Include the existential dictionaries in dataConOrigInstPat
[ghc.git] / compiler / ilxGen / IlxGen.lhs
1 %
2 \section{Generate .NET extended IL}
3
4 \begin{code}
5 module IlxGen( ilxGen ) where
6
7 #include "HsVersions.h"
8
9 import Char     ( ord, chr )
10 import StgSyn
11 import Id       ( idType, idName, isDeadBinder, idArity )
12 import Var      ( Var, Id, TyVar, isId, isTyVar, tyVarKind, tyVarName )
13 import VarEnv
14 import VarSet   ( isEmptyVarSet )
15 import TyCon    ( TyCon,  tyConPrimRep, isUnboxedTupleTyCon, tyConDataCons, 
16                   tyConTyVars, isDataTyCon, isAlgTyCon, tyConArity
17                 )
18 import Type     ( liftedTypeKind, openTypeKind, unliftedTypeKind,
19                   isUnLiftedType, isTyVarTy, mkTyVarTy, predTypeRep, pprType,
20                   splitForAllTys, splitFunTys, applyTy, applyTys, eqKind, tyVarsOfTypes
21                 )
22 import TypeRep  ( Type(..) )
23 import DataCon  ( isUnboxedTupleCon, dataConTyCon, dataConRepType, dataConRepArgTys, DataCon(..) )
24 import Literal  ( Literal(..) )
25 import PrelNames        -- Lots of keys
26 import PrimOp           ( PrimOp(..) )
27 import ForeignCall      ( CCallConv(..), ForeignCall(..), CCallSpec(..), CCallTarget(..), DNCallSpec(..) )
28 import TysWiredIn       ( mkTupleTy, tupleCon )
29 import PrimRep          ( PrimRep(..) )
30 import Name             ( nameModule, nameOccName, isExternalName, isInternalName, NamedThing(getName) )
31 import Subst            ( substTyWith )
32
33 import Module           ( Module, PackageName, ModuleName, moduleName, 
34                           modulePackage, basePackage,
35                           isHomeModule, isVanillaModule,
36                           pprModuleName, mkHomeModule, mkModuleName
37                         )
38
39 import UniqFM
40 import BasicTypes       ( Boxity(..) )
41 import CStrings         ( CLabelString, pprCLabelString )
42 import Outputable
43 import Char             ( ord )
44 import List             ( partition, elem, insertBy,any  )
45 import UniqSet
46
47 import TysPrim  ( foreignObjPrimTyCon, weakPrimTyCon, byteArrayPrimTyCon, mutableByteArrayPrimTyCon )
48
49 -- opt_SimplDoEtaReduction is used to help with assembly naming conventions for different
50 -- versions of compiled Haskell code.  We add a ".O" to all assembly and module 
51 -- names when this is set (because that's clue that -O was set).  
52 -- One day this will be configured by the command line.
53 import DynFlags ( opt_InPackage, opt_SimplDoEtaReduction )
54
55 import Util             ( lengthIs, equalLength )
56
57 \end{code}
58
59
60
61 %************************************************************************
62 %*                                                                      *
63 \subsection{Main driver}
64 %*                                                                      *
65 %************************************************************************
66
67 \begin{code}
68 ilxGen :: Module -> [TyCon] -> [(StgBinding,[Id])] -> SDoc
69         -- The TyCons should include those arising from classes
70 ilxGen mod tycons binds_w_srts
71   =  vcat [ text ".module '" <> (ppr (moduleName mod)) <> hscOptionQual <> text "o'",
72             text ".assembly extern 'mscorlib' {}",
73             vcat (map (ilxImportPackage topenv) (uniqSetToList import_packages)),
74             vcat (map (ilxImportModule topenv) (uniqSetToList import_modules)),
75             vcat (map (ilxImportTyCon topenv) (uniqSetToList import_tycons)),
76             vcat (map (ilxImportCCall topenv) (map snd (ufmToList import_ccalls))),
77             vcat (map (ilxTyCon topenv) data_tycons),
78             vcat (map (ilxBindClosures topenv) binds),
79             ilxTopBind mod topenv toppairs
80          ]
81     where
82       binds = map fst binds_w_srts
83       toppairs = ilxPairs binds
84       topenv = extendIlxEnvWithTops (emptyIlxEnv False mod) mod toppairs
85         -- Generate info from class decls as well
86       (import_packages,import_modules,import_tycons,import_ccalls) = importsBinds topenv binds (importsPrelude emptyImpInfo)
87       data_tycons = filter isDataTyCon tycons
88 \end{code}
89
90 %************************************************************************
91 %*                                                                      *
92 \subsection{Find Imports}
93 %*                                                                      *
94 %************************************************************************
95
96 \begin{code}
97
98 importsBinds :: IlxEnv -> [StgBinding] -> ImportsInfo -> ImportsInfo
99 importsBinds env binds = foldR (importsBind env) binds
100
101 importsNone :: ImportsInfo -> ImportsInfo
102 importsNone sofar = sofar
103
104 importsBind :: IlxEnv -> StgBinding -> ImportsInfo -> ImportsInfo
105 importsBind env (StgNonRec _ b rhs) = importsRhs env rhs.importsVar env b
106 importsBind env (StgRec _ pairs) = foldR (\(b,rhs) -> importsRhs env rhs . importsVar env b) pairs
107
108 importsRhs :: IlxEnv -> StgRhs -> ImportsInfo -> ImportsInfo
109 importsRhs env (StgRhsCon _ con args) = importsDataCon env con . importsStgArgs env args
110 importsRhs env (StgRhsClosure _ _ _ _ args body) = importsExpr env body. importsVars env args
111
112 importsExpr :: IlxEnv -> StgExpr -> ImportsInfo -> ImportsInfo
113 importsExpr env (StgLit _) = importsNone
114 importsExpr env (StgApp f args) = importsVar env f.importsStgArgs env args
115 importsExpr env (StgConApp con args) = importsDataCon env con.importsStgArgs env args
116 importsExpr env (StgOpApp (StgFCallOp (CCall (CCallSpec (StaticTarget c) cc _)) _) args rty)
117   = addCCallInfo (c,cc, map stgArgType tm_args, rty) . importsStgArgs env args
118   where 
119     (ty_args,tm_args) = splitTyArgs1 args 
120
121 importsExpr env (StgOpApp _ args res_ty) = importsType env res_ty. importsStgArgs env args
122
123
124 importsExpr env (StgSCC _ expr) = importsExpr env expr
125 importsExpr env (StgCase scrut _ _ bndr _ alts)
126   = importsExpr env scrut. imports_alts alts. importsVar env bndr
127    where
128     imports_alts (StgAlgAlts _ alg_alts deflt)  -- The Maybe TyCon part is dealt with 
129                                                 -- by the case-binder's type
130       = foldR imports_alg_alt alg_alts .  imports_deflt deflt
131        where
132         imports_alg_alt (con, bndrs, _, rhs)
133           = importsExpr env rhs . importsDataCon env con. importsVars env bndrs
134
135     imports_alts (StgPrimAlts _ alg_alts deflt)
136       = foldR imports_prim_alt alg_alts . imports_deflt deflt
137        where
138         imports_prim_alt (_, rhs) = importsExpr env rhs
139     imports_deflt StgNoDefault = importsNone
140     imports_deflt (StgBindDefault rhs) = importsExpr env rhs
141
142
143 importsExpr env (StgLetNoEscape _ _ bind body) = importsExpr env (StgLet bind body)
144 importsExpr env (StgLet bind body)
145   = importsBind env bind .  importsExpr env body
146
147 importsApp env v args = importsVar env v.  importsStgArgs env args
148 importsStgArgs env args = foldR (importsStgArg env) args
149
150 importsStgArg :: IlxEnv -> StgArg -> ImportsInfo -> ImportsInfo
151 importsStgArg env (StgTypeArg ty) = importsType env ty
152 importsStgArg env (StgVarArg v) = importsVar env v
153 importsStgArg env _ = importsNone
154
155 importsVars env vs = foldR (importsVar env) vs
156 importsVar env v = importsName env (idName v). importsType env (idType v)
157
158 importsName env n
159    | isInternalName n = importsNone
160    | ilxEnvModule env == nameModule n = importsNone
161    | isHomeModule (nameModule n) =  addModuleImpInfo (moduleName (nameModule n))
162 -- See HACK below
163    | isVanillaModule (nameModule n)  && not inPrelude =  importsPrelude
164    | isVanillaModule (nameModule n)  && inPrelude =   addModuleImpInfo (moduleName (nameModule n))
165 -- End HACK
166    | otherwise = addPackageImpInfo (modulePackage (nameModule n))
167
168
169 importsPrelude | inPrelude = addModuleImpInfo (mkModuleName "PrelGHC")
170                | otherwise = addPackageImpInfo basePackage
171
172
173 importsType :: IlxEnv -> Type -> ImportsInfo -> ImportsInfo
174 importsType env ty = importsType2 env (deepIlxRepType ty)
175
176 importsType2 :: IlxEnv -> Type -> ImportsInfo -> ImportsInfo
177 importsType2 env (AppTy f x) =  importsType2 env f . importsType2 env x
178 importsType2 env (TyVarTy _) = importsNone
179 importsType2 env (TyConApp tc args) =importsTyCon env tc . importsTypeArgs2 env args
180 importsType2 env (FunTy arg res) =  importsType env arg .  importsType2 env res
181 importsType2 env (ForAllTy tv body_ty) =  importsType2 env body_ty
182 importsType2 env (NoteTy _ ty) = importsType2 env ty
183 importsType2 _ _ = panic "IlxGen.lhs: importsType2 ty"
184 importsTypeArgs2 env tys = foldR (importsType2 env) tys
185
186 importsDataCon env dcon = importsTyCon env (dataConTyCon dcon)
187
188 importsTyCon env tc | (not (isDataTyCon tc) || 
189                    isInternalName (getName tc) || 
190                    ilxEnvModule env == nameModule (getName tc)) = importsNone
191 importsTyCon env tc | otherwise = importsName env (getName tc) . addTyConImpInfo tc .
192                                     foldR (importsTyConDataCon env) (tyConDataCons tc)
193
194
195 importsTyConDataCon :: IlxEnv -> DataCon -> ImportsInfo -> ImportsInfo
196 importsTyConDataCon env dcon = foldR (importsTyConDataConType env) (filter (not . isVoidIlxRepType) (dataConRepArgTys dcon))
197
198 importsTyConDataConType :: IlxEnv -> Type -> ImportsInfo -> ImportsInfo
199 importsTyConDataConType env ty = importsTyConDataConType2 env (deepIlxRepType ty)
200
201 importsTyConDataConType2 :: IlxEnv -> Type -> ImportsInfo -> ImportsInfo
202 importsTyConDataConType2 env (AppTy f x) =  importsTyConDataConType2 env f . importsTyConDataConType2 env x
203 importsTyConDataConType2 env (TyVarTy _) = importsNone
204 importsTyConDataConType2 env (TyConApp tc args) = importsTyConDataConTypeTyCon env tc . importsTyConDataConTypeArgs2 env args
205 importsTyConDataConType2 env (FunTy arg res) = importsTyConDataConType env arg .  importsTyConDataConType2 env res
206 importsTyConDataConType2 env (ForAllTy tv body_ty) = importsTyConDataConType2 env body_ty
207 importsTyConDataConType2 env (NoteTy _ ty) = importsTyConDataConType2 env ty
208 importsTyConDataConType2 _ _ = panic "IlxGen.lhs: importsTyConDataConType2 ty"
209 importsTyConDataConTypeArgs2 env tys = foldR (importsTyConDataConType2 env) tys
210
211 importsTyConDataConTypeTyCon env tc | (not (isDataTyCon tc) || 
212                    isInternalName (getName tc) || 
213                    ilxEnvModule env == nameModule (getName tc)) = importsNone
214 importsTyConDataConTypeTyCon env tc | otherwise = importsName env (getName tc)
215
216
217 type StaticCCallInfo = (CLabelString,CCallConv,[Type],Type)
218 type ImportsInfo = (UniqSet PackageName, UniqSet ModuleName, UniqSet TyCon, UniqFM StaticCCallInfo)
219    -- (Packages, Modules, Datatypes, Imported CCalls)
220
221 emptyImpInfo :: ImportsInfo
222 emptyImpInfo = (emptyUniqSet, emptyUniqSet, emptyUniqSet, emptyUFM)
223 addPackageImpInfo p (w,x,y,z) = (addOneToUniqSet w p, x, y,z)
224 addModuleImpInfo m (w,x,y,z) = (w, addOneToUniqSet x m, y,z)
225 addTyConImpInfo tc (w,x,y,z) = (w, x, addOneToUniqSet y tc,z)
226 addCCallInfo info@(nm,a,b,c) (w,x,y,z) = (w, x, y,addToUFM z nm info)
227
228 ilxImportTyCon :: IlxEnv -> TyCon -> SDoc
229 ilxImportTyCon env tycon | isDataTyCon tycon = ilxTyConDef True env tycon
230 ilxImportTyCon _ _ | otherwise =  empty
231
232 ilxImportPackage :: IlxEnv -> PackageName -> SDoc
233 ilxImportPackage _ p = text ".assembly extern" <+> singleQuotes (ppr p <> hscOptionQual) <+> text "{ }"
234
235 ilxImportModule :: IlxEnv -> ModuleName -> SDoc
236 ilxImportModule _ m = text ".module extern" <+> singleQuotes (ppr m <> hscOptionQual <> text "o")
237
238 -- Emit a P/Invoke declaration for the imported C function
239 -- TODO: emit the right DLL name
240 ilxImportCCall :: IlxEnv -> StaticCCallInfo -> SDoc
241 ilxImportCCall env (c,cc,args,ret) = 
242     text ".method static assembly pinvokeimpl" <+> 
243     parens (doubleQuotes (text "HSstd_cbits.dll") <+> text "cdecl") <+> retdoc <+> singleQuotes (pprCLabelString c) <+> 
244     pprCValArgTys ilxTypeL env (map deepIlxRepType (filter (not. isVoidIlxRepType) args)) <+> 
245     text "unmanaged preservesig { }"
246   where 
247     retdoc = 
248           if isVoidIlxRepType ret then text "void" 
249           else ilxTypeR env (deepIlxRepType ret)
250
251
252 \end{code}
253
254 %************************************************************************
255 %*                                                                      *
256 \subsection{Type declarations}
257 %*                                                                      *
258 %************************************************************************
259
260 \begin{code}
261
262
263 ilxTyCon :: IlxEnv -> TyCon -> SDoc
264 ilxTyCon env tycon =  ilxTyConDef False env tycon
265
266 -- filter to get only dataTyCons?
267 ilxTyConDef importing env tycon = 
268         vcat [empty $$ line,
269               text ".classunion" <+> (if importing then text "import" else empty) <+> tycon_ref <+> tyvars_text <+> super_text   <+> alts_text]
270    where
271      tycon_ref =  nameReference env (getName tycon)  <> (ppr tycon)
272      super_text = if importing then empty else text "extends thunk" <> angleBrackets (text "class" <+> tycon_ref)
273      tyvars = tyConTyVars tycon
274      (ilx_tvs, _) = categorizeTyVars tyvars
275      alts_env = extendIlxEnvWithFormalTyVars env ilx_tvs 
276      tyvars_text = pprTyVarBinders alts_env ilx_tvs 
277      alts = vcat (map (pprIlxDataCon alts_env) (tyConDataCons tycon))
278      alts_text = nest 2 (braces alts)
279
280 pprIlxDataCon env dcon =
281         text ".alternative" <+> pprId dcon <+> 
282         parens (pprSepWithCommas (ilxTypeL env) (map deepIlxRepType (filter (not. isVoidIlxRepType) (dataConRepArgTys dcon))))
283 \end{code}
284
285
286 %************************************************************************
287 %*                                                                      *
288 \subsection{Getting the .closures and literals out}                     *
289 %************************************************************************
290
291 \begin{code}
292
293 ilxBindClosures :: IlxEnv -> StgBinding -> SDoc
294 ilxBindClosures env (StgNonRec _ b rhs) = ilxRhsClosures env (b,rhs)
295 ilxBindClosures env (StgRec _ pairs)  
296   = vcat (map (ilxRhsClosures new_env) pairs)
297   where
298      new_env = extendIlxEnvWithBinds env pairs
299
300 ---------------
301 ilxRhsClosures _ (_, StgRhsCon _ _ _)
302   = empty
303
304 ilxRhsClosures env (bndr, StgRhsClosure _ _ fvs upd args rhs)
305   = vcat [ilxExprClosures next_env rhs,
306
307          empty $$ line,
308          kind_text <+> singleQuotes cloname <+>  free_vs_text,
309          nest 2 (braces (
310             nest 2 (vcat [empty,
311                           vcat [text ".apply" <+> closure_sig_text,
312                                 body_text
313                           ],
314                           empty
315                     ])
316                 ))
317     ]
318   where
319     kind_of_thing = case upd of
320                           Updatable -> ASSERT( null args ) ".thunk"
321                           otherwise -> ".closure"
322     kind_text = text kind_of_thing 
323                 
324     cloname = ilxEnvQualifyByModule env (ppr bndr)
325     next_env = ilxPlaceStgRhsClosure env bndr 
326     (free_vs_text,env_with_fvs) = pprFreeBinders next_env fvs
327
328
329     closure_sig_text =     
330       vcat [ text "()",
331              (case args of 
332                []        -> empty
333                otherwise -> args_text),
334              text "-->" <+>  rty_text]
335
336     (args_text,env_with_args) = pprArgBinders env_with_fvs args
337
338         -- Find the type returned, from the no. of args and the type of "bndr"
339     rty_text = 
340       case retType env_with_fvs (idIlxRepType bndr) args of
341        Just (env,ty) -> 
342           if isVoidIlxRepType ty  then  (text "void")
343           else ilxTypeR env ty 
344        Nothing -> trace "WARNING!  IlxGen.trace could not find return type - see generated ILX for context where this occurs." (text "// Could not find return type:" <+> ilxTypeR env_with_fvs (idIlxRepType bndr)<+> text ", non representation: " <+> ilxTypeR env_with_fvs (idType bndr))
345
346     -- strip off leading ForAll and Fun type constructions
347     -- up to the given number of arguments, extending the environment as
348     -- we go.  
349     retType env ty [] = Just (env, ty)
350     retType env (ForAllTy tv ty) (arg:args) = retType (extendIlxEnvWithTyArgs env [tv]) ty args
351     retType env (FunTy l r) (arg:args) = retType env r args
352     retType _ _ _  = Nothing
353
354         -- Code for the local variables
355     locals = ilxExprLocals env_with_args rhs
356
357     env_with_locals = extendIlxEnvWithLocals env_with_args locals
358
359         -- Code for the body of the main apply method
360     body_code = vcat [empty,
361                       pprIlxLocals env_with_args locals,
362                       ilxExpr (IlxEEnv env_with_locals (mkUniqSet (filter (not.isTyVar) args))) rhs Return,
363                       empty
364                 ]
365
366     body_text = nest 2 (braces (text ".maxstack 100" <+> nest 2 body_code))
367
368
369 pprIlxLocals env [] = empty
370 pprIlxLocals env vs 
371    = text ".locals" <+> parens (pprSepWithCommas (pprIlxLocal env) (filter nonVoidLocal vs))
372   where
373     nonVoidLocal (LocalId v,_) = not (isVoidIlxRepId v)
374     nonVoidLocal _ = True
375
376 pprIlxLocal env (LocalId v,_) = ilxTypeL env (idIlxRepType v) <+> pprId v
377 pprIlxLocal env (LocalSDoc (ty,doc,pin),_) = ilxTypeL env (deepIlxRepType ty) <+> (if pin then text "pinned" else empty) <+> doc
378
379
380 pprFreeBinders env fvs 
381     = (ilx_tvs_text <+> vs_text, env2)
382     where   
383        (free_ilx_tvs, _,free_vs) = categorizeVars fvs
384        real_free_vs = filter (not . isVoidIlxRepId) free_vs
385         -- ignore the higher order type parameters for the moment
386        env1 = extendIlxEnvWithFreeTyVars env free_ilx_tvs 
387        ilx_tvs_text = pprTyVarBinders env1 free_ilx_tvs
388        vs_text = parens (pprSepWithCommas ppr_id real_free_vs)
389        ppr_id v = ilxTypeL env1 (idIlxRepType v) <+> pprId v 
390        env2 = extendIlxEnvWithFreeVars env1 real_free_vs 
391
392 pprIdBinder env v = parens (ilxTypeL env (idIlxRepType v) <+> pprId v)
393
394         -- Declarations for the arguments of the main apply method
395 pprArgBinders env [] = (empty,env)
396 pprArgBinders env (arg:args)
397     = (arg_text <+> rest_text, res_env)
398    where 
399      (arg_text,env') = pprArgBinder env arg
400      (rest_text,res_env) = pprArgBinders env' args 
401
402 -- We could probably omit some void argument binders, but
403 -- don't...
404 pprArgBinder env arg 
405   | isVoidIlxRepId arg = (text "()", extendIlxEnvWithArgs env [arg])
406   | otherwise 
407       = if isTyVar arg then 
408          let env' = extendIlxEnvWithTyArgs env [arg] in 
409          (pprTyVarBinder env' arg, env')
410       else (pprIdBinder env arg,extendIlxEnvWithArgs env [arg])
411
412 --------------
413 -- Compute local variables used by generated method.
414 -- The names of some generated locals are recorded as SDocs.
415
416 data LocalSpec = LocalId Id | LocalSDoc (Type, SDoc, Bool)  -- flag is for pinning
417
418 ilxExprLocals :: IlxEnv -> StgExpr -> [(LocalSpec,Maybe (IlxEnv,StgRhs))]
419 ilxExprLocals env (StgLet bind body)              = ilxBindLocals env bind ++ ilxExprLocals env body
420 ilxExprLocals env (StgLetNoEscape _ _ bind body)  = ilxBindLocals env bind ++ ilxExprLocals env body  -- TO DO????
421 ilxExprLocals env (StgCase scrut _ _ bndr _ alts) 
422      = ilxExprLocals (ilxPlaceStgCaseScrut env) scrut ++ 
423        (if isDeadBinder bndr then [] else [(LocalId bndr,Nothing)]) ++ 
424        ilxAltsLocals env alts
425 ilxExprLocals env (StgOpApp (StgFCallOp fcall _) args _) 
426      = concat (ilxMapPlaceArgs 0 ilxCCallArgLocals env args)
427 ilxExprLocals _ _  = []
428
429 -- Generate locals to use for pinning arguments as we cross the boundary
430 -- to C.
431 ilxCCallArgLocals env (StgVarArg v) | pinCCallArg v = 
432    [(LocalSDoc (idType v, ilxEnvQualifyByExact env (ppr v) <> text "pin", True), Nothing)]
433 ilxCCallArgLocals _ _ | otherwise = []
434
435 ilxBindLocals env (StgNonRec _ b rhs) = [(LocalId b,Just (env, rhs))]
436 ilxBindLocals env (StgRec _ pairs)    = map (\(x,y) -> (LocalId x,Just (env, y))) pairs
437
438 ilxAltsLocals env (StgAlgAlts  _ alts deflt) = ilxDefltLocals env deflt ++ concat (ilxMapPlaceAlts ilxAlgAltLocals env alts)
439 ilxAltsLocals env (StgPrimAlts _ alts deflt) = ilxDefltLocals env deflt ++ concat (ilxMapPlaceAlts ilxPrimAltLocals env alts)
440
441 ilxAlgAltLocals env (_, bndrs, _, rhs) = map (\x -> (LocalId x,Nothing)) (filter (\v -> isId v && not (isDeadBinder v)) bndrs) ++ ilxExprLocals env rhs
442 ilxPrimAltLocals env (_, rhs)          = ilxExprLocals env rhs
443
444 ilxDefltLocals _ StgNoDefault   = []
445 ilxDefltLocals env (StgBindDefault rhs) = ilxExprLocals (ilxPlaceStgBindDefault env) rhs
446
447 --------------
448 ilxExprClosures :: IlxEnv -> StgExpr -> SDoc
449 ilxExprClosures env (StgApp _ args)
450   = vcat (ilxMapPlaceArgs 0 (ilxArgClosures) env args)  -- get strings
451 ilxExprClosures env (StgConApp _ args)
452   = vcat (ilxMapPlaceArgs 0 (ilxArgClosures) env args) -- get strings
453 ilxExprClosures env (StgOpApp _ args _)
454   = vcat (ilxMapPlaceArgs 0 (ilxArgClosures) env args) -- get strings
455 ilxExprClosures env (StgLet bind body)
456   = ilxBindClosures env bind $$ ilxExprClosures (extendIlxEnvWithBinds env (ilxPairs1 bind)) body
457 ilxExprClosures env (StgLetNoEscape _ _ bind body)  -- TO DO????
458   = ilxBindClosures env bind $$ ilxExprClosures (extendIlxEnvWithBinds env (ilxPairs1 bind)) body
459 ilxExprClosures env (StgCase scrut _ _ _ _ alts)
460   = ilxExprClosures (ilxPlaceStgCaseScrut env) scrut $$ ilxAltsClosures env alts 
461 ilxExprClosures env (StgLit lit) 
462   = ilxGenLit env lit 
463 ilxExprClosures _ _ 
464   = empty
465
466 ilxAltsClosures env (StgAlgAlts _ alts deflt)
467   = vcat [ilxExprClosures (ilxPlaceAlt env i) rhs | (i,(_, _, _, rhs))  <- [1..] `zip` alts]
468     $$ 
469     ilxDefltClosures env deflt
470
471 ilxAltsClosures env (StgPrimAlts _ alts deflt)
472   = vcat [ilxExprClosures (ilxPlaceAlt env i) rhs | (i,(_, rhs)) <- [1..] `zip` alts]
473     $$ 
474     vcat [ ilxGenLit (ilxPlacePrimAltLit env i) lit | (i,(lit,_)) <- [1..] `zip` alts]
475     $$ 
476     ilxDefltClosures  env deflt
477
478 ilxDefltClosures env (StgBindDefault rhs) = ilxExprClosures (ilxPlaceStgBindDefault env) rhs
479 ilxDefltClosures _ StgNoDefault   = empty
480
481 ilxArgClosures env (StgLitArg lit) = ilxGenLit env lit 
482 ilxArgClosures _ _ = empty
483
484
485
486 ilxGenLit env (MachStr fs) 
487   = vcat [text ".field static assembly char "  <+> singleQuotes nm <+> text "at" <+> nm <> text "L",
488           text ".data" <+> nm <> text "L" <+> text "= char *("  <> pprFSInILStyle fs  <> text ")"
489          ]
490  where
491    nm = ilxEnvQualifyByExact env (text "string")
492
493 ilxGenLit  _ _ = empty
494
495 \end{code}
496
497
498 %************************************************************************
499 %*                                                                      *
500 \subsection{Generating code}
501 %*                                                                      *
502 %************************************************************************
503
504
505 \begin{code}
506
507 -- Environment when generating expressions
508 data IlxEEnv = IlxEEnv IlxEnv (UniqSet Id)
509
510 data Sequel = Return | Jump IlxLabel
511
512 ilxSequel Return     = text "ret"
513 ilxSequel (Jump lbl) = text "br" <+> pprIlxLabel lbl
514
515 isReturn Return = True
516 isReturn (Jump _) = False
517
518
519 ilxExpr :: IlxEEnv -> StgExpr 
520         -> Sequel       -- What to do at the end
521         -> SDoc
522
523 ilxExpr (IlxEEnv env _) (StgApp fun args) sequel
524   = ilxFunApp env fun args (isReturn sequel) $$ ilxSequel sequel
525
526 -- ilxExpr eenv (StgLit lit) sequel
527 ilxExpr (IlxEEnv env _) (StgLit lit) sequel
528   = pushLit env lit $$ ilxSequel sequel
529
530 -- ilxExpr eenv (StgConApp data_con args) sequel
531 ilxExpr (IlxEEnv env _) (StgConApp data_con args) sequel
532   = text " /* ilxExpr:StgConApp */ " <+>  ilxConApp env data_con args $$ ilxSequel sequel
533
534 -- ilxExpr eenv (StgPrimApp primop args _) sequel
535 ilxExpr (IlxEEnv env _) (StgOpApp (StgFCallOp fcall _) args ret_ty) sequel
536   = ilxFCall env fcall args ret_ty $$ ilxSequel sequel
537
538 ilxExpr (IlxEEnv env _) (StgOpApp (StgPrimOp primop) args ret_ty) sequel
539   = ilxPrimOpTable primop args env $$ ilxSequel sequel
540
541 --BEGIN TEMPORARY
542 -- The following are versions of a peephole optimizations for "let t = \[] t2[fvs] in t"
543 -- I think would be subsumed by a general treatmenet of let-no-rec bindings??
544 ilxExpr eenv@(IlxEEnv env _) (StgLet (StgNonRec _ bndr (StgRhsClosure _ _ _ _ [] rhs)) (StgApp fun [])) sequel 
545               | (bndr == fun && null (ilxExprLocals env rhs)) -- TO DO???
546   = ilxExpr eenv rhs sequel
547 ilxExpr eenv@(IlxEEnv env _) (StgLetNoEscape _ _ (StgNonRec _ bndr (StgRhsClosure _ _ _ _ [] rhs)) (StgApp fun [])) sequel 
548               | (bndr == fun && null (ilxExprLocals env rhs)) -- TO DO???
549   = ilxExpr eenv rhs sequel
550 --END TEMPORARY
551
552 ilxExpr eenv (StgLet bind body) sequel
553   = ilxBind eenv bind $$ ilxExpr eenv body sequel
554
555
556 ilxExpr eenv (StgLetNoEscape _ _ bind body) sequel -- TO DO???
557   = ilxBind eenv bind $$ ilxExpr eenv body sequel
558
559 -- StgCase: Special case 1 to avoid spurious branch.
560 ilxExpr eenv@(IlxEEnv env live) (StgCase (StgApp fun args) live_in_case _live_in_alts bndr _ alts) sequel
561   = vcat [ilxWipe env (uniqSetToList (live `minusUniqSet` live_in_case)),
562           ilxFunApp (ilxPlaceStgCaseScrut env) fun args False,
563           --ilxWipe env (uniqSetToList (live_in_case `minusUniqSet` _live_in_alts)),
564           --ilxAlts (IlxEEnv env _live_in_alts) bndr alts sequel
565           ilxAlts (IlxEEnv env live_in_case) bndr alts sequel
566     ]
567
568 -- StgCase: Special case 2 to avoid spurious branch.
569 ilxExpr eenv@(IlxEEnv env live) (StgCase (StgOpApp (StgPrimOp primop) args ret_ty) live_in_case _live_in_alts bndr _ alts) sequel
570   = vcat [ilxWipe env (uniqSetToList (live `minusUniqSet` live_in_case)),
571           ilxPrimOpTable primop args (ilxPlaceStgCaseScrut env),
572           --ilxWipe env (uniqSetToList (live_in_case `minusUniqSet` _live_in_alts)),
573           --ilxAlts (IlxEEnv env _live_in_alts) bndr alts sequel
574           ilxAlts (IlxEEnv env live_in_case) bndr alts sequel
575     ]
576
577 -- StgCase: Normal case.
578 ilxExpr eenv@(IlxEEnv env live) (StgCase scrut live_in_case _live_in_alts bndr _ alts) sequel
579   = vcat [ilxWipe env (uniqSetToList (live `minusUniqSet` live_in_case)),
580           ilxExpr (IlxEEnv (ilxPlaceStgCaseScrut env) live_in_case) scrut (Jump join_lbl),
581           ilxLabel join_lbl,
582           --ilxWipe env (uniqSetToList (live_in_case `minusUniqSet` _live_in_alts)),
583           --ilxAlts (IlxEEnv env _live_in_alts) bndr alts sequel
584           ilxAlts (IlxEEnv env live_in_case) bndr alts sequel
585     ]
586   where
587     join_lbl = mkJoinLabel bndr
588
589 ilxExpr _ _ _ 
590   = panic "ilxExpr:  Patterns not matched:(IlxEEnv _ _) (StgSCC _ _) _ (IlxEEnv _ _) (StgLam _ _ _) _"
591
592
593 -- Wipe out locals and arguments that are no longer in use, to
594 -- prevent space leaks. If the VM is implemented 100% correctly then
595 -- this should probably not be needed, as the live variable analysis
596 -- in the JIT would tell the GC that these locals and arguments are
597 -- no longer live.  However I'm putting it in here so we can
598 -- check out if it helps.
599 --
600 -- Also, in any case this doesn't capture everything we need.  e.g.
601 -- when making a call:
602 --     case f x of ...
603 -- where x is not used in the alternatives, then the variable x
604 -- is no longer live from the point it is transferred to the call
605 -- onwards.  We should expunge "live_in_case - live_in_alts" right
606 -- before making the call, not after returning from the call....
607 --
608 -- Strictly speaking we also don't need to do this for primitive
609 -- values such as integers and addresses, i.e. things not
610 -- mapped down to GC'able objects.
611 ilxWipe env ids 
612    = vcat (map (ilxWipeOne env) (filter (not.isVoidIlxRepId) ids))
613
614 ilxWipeOne env id
615    = case lookupIlxVarEnv env id of
616           Just Local  -> text "ldloca " <+> pprId id <+> text "initobj.any" <+> (ilxTypeL env (idIlxRepType id))
617           Just Arg   -> text "deadarg " <+> pprId id <+> text "," <+> (ilxTypeL env (idIlxRepType id))
618           Just (CloVar _)  -> ilxComment (text "not yet wiping closure variable" <+> pprId id )
619           _ -> ilxComment (text "cannot wipe non-local/non-argument" <+> pprId id )
620   where 
621       
622
623 ----------------------
624
625 ilxAlts :: IlxEEnv -> Id -> StgCaseAlts -> Sequel -> SDoc
626 ilxAlts (IlxEEnv env live) bndr alts sequel
627         -- At the join label, the result is on top
628         -- of the stack
629   = vcat [store_in_bndr,
630           do_case_analysis alts
631     ]
632   where
633     scrut_rep_ty = deepIlxRepType (idType bndr)
634
635     store_in_bndr | isDeadBinder bndr = empty
636                   | isVoidIlxRepId bndr 
637                         = ilxComment (text "ignoring store of zero-rep value to be analyzed")
638                   | otherwise         = text "dup" $$ (text "stloc" <+> pprId bndr)
639
640     do_case_analysis (StgAlgAlts _ []    deflt)
641         = do_deflt deflt
642
643     do_case_analysis (StgAlgAlts _ args deflt) 
644         = do_alg_alts ([1..] `zip` args) deflt
645
646     do_case_analysis (StgPrimAlts _ alts deflt)
647         = do_prim_alts ([1..] `zip` alts) $$ do_deflt deflt
648
649     do_alg_alts [(i, alt@(data_con,bndrs,used_flags, rhs))] StgNoDefault | isUnboxedTupleCon data_con
650       -- Collapse the analysis of unboxed tuples where 
651       -- some or all elements are zero-sized
652       --
653       -- TO DO: add bndrs to set of live variables
654           = case bndrs' of
655                   [h] -> bind_collapse bndrs used_flags <+> do_rhs_no_pop alt_env rhs
656                   _ -> bind_components alt_env dcon' bndrs 0 used_flags <+> do_rhs alt_env rhs
657            where 
658             bndrs' = filter (not. isVoidIlxRepId) bndrs
659             -- Replacement unboxed tuple type constructor, used if any of the
660             -- arguments have zero-size and more than one remains.
661             dcon'  = tupleCon Unboxed (length bndrs')
662
663             alt_env = IlxEEnv (ilxPlaceAlt env i) live
664             --alt_env = IlxEEnv (ilxPlaceAlt env i) 
665
666             bind_collapse [] _ = panic "bind_collapse: unary element not found"
667             bind_collapse (h:t) (is_used:used_flags) 
668                 | isVoidIlxRepId h = ilxComment (text "zero-rep binding eliminated") <+> (bind_collapse t used_flags)
669                 | not is_used = ilxComment (text "not used") <+> text "pop"
670                 | otherwise = text "stloc" <+> pprId h
671
672
673     do_alg_alts [(i, alt@(data_con,bndrs,used_flags, rhs))] StgNoDefault 
674             = vcat [text "castdata" <+> sep [ilxTypeR env scrut_rep_ty <> comma,
675                                              ilxConRef env data_con],
676                 do_alg_alt (IlxEEnv (ilxPlaceAlt env i) live) alt
677               ]
678
679     do_alg_alts alts deflt
680         = vcat [text "datacase" <+> sep [ilxTypeR env scrut_rep_ty,text ",",
681                                          pprSepWithCommas pp_case labels_w_alts],
682                 do_deflt deflt,
683                 vcat (map do_labelled_alg_alt labels_w_alts)
684           ]
685         where
686           pp_case (i, (lbl, (data_con, _, _, _))) = parens (ilxConRef env data_con <> comma <> pprIlxLabel lbl)
687           labels_w_alts = [(i,(mkAltLabel bndr i, alt)) | (i, alt) <- alts]
688
689     do_prim_alts [] = empty
690     do_prim_alts ((i, (lit,alt)) : alts) 
691         = vcat [text "dup", pushLit (ilxPlacePrimAltLit env i) lit, text "bne.un" <+> pprIlxLabel lbl, 
692                 do_rhs (IlxEEnv (ilxPlaceAlt env i) live) alt, 
693                 ilxLabel lbl, do_prim_alts alts]
694         where
695           lbl = mkAltLabel bndr i
696
697     do_labelled_alg_alt (i,(lbl, alt)) 
698         = ilxLabel lbl $$ do_alg_alt (IlxEEnv (ilxPlaceAlt env i) live) alt
699
700     do_alg_alt alt_eenv (data_con, bndrs, used_flags, rhs) 
701       = vcat [bind_components alt_eenv data_con bndrs 0 used_flags,
702               do_rhs alt_eenv rhs
703              ]
704
705     bind_components alt_eenv data_con [] n _ = empty
706     bind_components alt_eenv data_con (h:t) n (is_used:used_flags) 
707        | isVoidIlxRepId h 
708              -- don't increase the count in this case
709              = ilxComment (text "zero-rep binding eliminated") 
710                <+> bind_components alt_eenv data_con t n used_flags
711        | otherwise 
712              = bind_component alt_eenv data_con h is_used n 
713                <+> bind_components alt_eenv data_con t (n + 1) used_flags
714
715     bind_component alt_eenv@(IlxEEnv alt_env _) data_con bndr is_used reduced_fld_no 
716         | not is_used 
717             = ilxComment (text "not used")
718         | isVoidIlxRepId bndr 
719             = ilxComment (text "ignoring bind of zero-rep variable")
720         | otherwise   = vcat [text "dup",
721                               ld_data alt_env data_con reduced_fld_no bndr,
722                               text "stloc" <+> pprId bndr]
723
724     do_deflt (StgBindDefault rhs) = do_rhs (IlxEEnv (ilxPlaceStgBindDefault env) live) rhs
725     do_deflt StgNoDefault         = empty
726
727     do_rhs alt_eenv rhs  
728         | isVoidIlxRepId bndr = do_rhs_no_pop alt_eenv rhs     -- void on the stack, nothing to pop
729         | otherwise = text "pop" $$ do_rhs_no_pop alt_eenv rhs  -- drop the value
730
731     do_rhs_no_pop alt_env rhs = ilxExpr alt_env rhs sequel
732
733     ld_data alt_env data_con reduced_fld_no bndr
734       | isUnboxedTupleCon data_con
735       = text "ldfld" <+> sep [text "!" <> integer reduced_fld_no,
736                               ilxTypeR alt_env scrut_rep_ty <> text "::fld" <> integer reduced_fld_no]
737       | otherwise 
738       = text "lddata" <+> sep [ilxTypeR alt_env scrut_rep_ty <> comma, 
739                                ilxConRef env data_con <> comma,
740                                integer reduced_fld_no]
741
742
743 -------------------------
744
745 ilxBestTermArity = 3
746 ilxBestTypeArity = 7
747
748
749 -- Constants of unlifted types are represented as
750 -- applications to no arguments.
751 ilxFunApp env fun [] _ | isUnLiftedType (idType fun)
752   = pushId env fun
753
754 ilxFunApp env fun args tail_call 
755   =     -- For example:
756         --      ldloc f         function of type forall a. a->a
757         --      ldloc x         arg of type Int
758         --      .tail callfunc <Int32> (!0) --> !0
759         --
760     vcat [pushId env fun,ilxFunAppAfterPush env fun args tail_call]
761
762 ilxFunAppAfterPush env fun args tail_call 
763   =     -- For example:
764         --      ldloc f         function of type forall a. a->a
765         --      ldloc x         arg of type Int
766         --      .tail callfunc <Int32> (!0) --> !0
767         --
768     vcat [ilxFunAppArgs env 0 (idIlxRepType fun) args tail_call known_clo]
769   where
770     known_clo :: KnownClosure
771     known_clo =
772       case lookupIlxBindEnv env fun of
773           Just (_, StgRhsClosure  _ _ _ Updatable _ _)   -> Nothing 
774           Just (place, StgRhsClosure  _ _ fvs _ args _)  -> Just (place,fun,args,fvs)
775           _ -> Nothing -- trace (show fun ++ " --> " ++ show (idArity fun))
776
777 type KnownClosure = Maybe (  IlxEnv     -- Of the binding site of the function
778                            , Id         -- The function
779                            , [Var]      -- Binders
780                            , [Var])     -- Free vars of the closure
781
782 -- Push as many arguments as ILX allows us to in one go, and call the function
783 -- Recurse until we're done.
784 -- The function is already on the stack
785 ilxFunAppArgs :: IlxEnv
786               -> Int            -- Number of args already pushed (zero is a special case;
787                                 --      otherwise used only for place generation)
788               -> Type           -- Type of the function
789               -> [StgArg]       -- The arguments
790               -> Bool           -- True <=> tail call please
791               -> KnownClosure   -- Information about the function we're calling
792               -> SDoc
793
794 ilxFunAppArgs env num_sofar funty args tail_call known_clo
795  =   vcat [vcat (ilxMapPlaceArgs num_sofar pushArgWithVoids env now_args),
796            call_instr <+> (if num_sofar == 0 then text "() /* first step in every Haskell app. is to a thunk */ " else empty)
797                      <+> now_args_text
798                      <+> text "-->" 
799                      <+> later_ty_text,
800            later
801           ]
802   where
803     now_args_text = 
804       case now_arg_tys of
805         [] -> empty
806         _ -> hsep (map (pprIlxArgInfo env_after_now_tyvs) now_arg_tys)
807
808     later_ty_text
809         | isVoidIlxRepType later_ty = text "void"
810         | otherwise = ilxTypeR env_after_now_tyvs later_ty
811
812     (now_args,now_arg_tys,env_after_now_tyvs,later_args,later_ty) = 
813         case args of
814           (StgTypeArg v:rest) -> get_type_args ilxBestTypeArity args env funty
815           _ -> get_term_args 0 ilxBestTermArity args env funty
816
817      -- Only apply up to maxArity real (non-type) arguments
818      -- at a time.  ILX should, in principle, allow us to apply
819      -- arbitrary numbers, but you will get more succinct 
820      -- (and perhaps more efficient) IL code
821      -- if you apply in clumps according to its maxArity setting.
822      -- This is because it has to unwind the stack and store it away
823      -- in local variables to do the partial applications.
824      --
825      -- Similarly, ILX only allows one type application at a time, at
826      -- least until we implement unwinding the stack for this case.
827      --
828      -- NB: In the future we may have to be more careful 
829      -- all the way through 
830      -- this file to bind type variables as we move through
831      -- type abstractions and "forall" types.  This would apply
832      -- especially if the type variables were ever bound by expressions
833      -- involving the type variables.  
834
835     -- This part strips off at most "max" term applications or one type application
836     get_type_args 0 args env funty = ([],[],env,args,funty)
837     get_type_args max args env (NoteTy _ ty) = 
838           trace "IlxGen Internal Error: non representation type passed to get_args" (get_type_args max args env ty)
839     get_type_args max ((arg@(StgTypeArg v)):rest) env (ForAllTy tv rem_funty) 
840         = if isIlxTyVar tv then 
841             let env2 = extendIlxEnvWithFormalTyVars env [tv] in 
842             let rest_ty = deepIlxRepType (substTyWith [tv] [v] rem_funty) in 
843             let (now,now_tys,env3,later,later_ty) = get_type_args (max - 1) rest env rest_ty in 
844             let arg_ty = mkTyVarTy tv in 
845             (arg:now,(arg,arg_ty):now_tys,env2, later, later_ty)
846           else 
847              get_type_args max rest env rem_funty  -- ? subst??
848     get_type_args _ (StgTypeArg _:_) _ _ = trace "IlxGen Internal Error: get_type_args could not get ForAllTy for corresponding arg" ([],[],env,[],funty)
849     get_type_args _ args env funty = ([],[],env,args,funty)
850
851     get_term_args n max args env (NoteTy _ ty)
852        -- Skip NoteTy types 
853        = trace "IlxGen Internal Error: non representation type passed to get_term_args" (get_term_args n max args env ty)
854     get_term_args n 0 args env funty
855        -- Stop if we've hit the maximum number of ILX arguments to apply n one hit.
856        = ([],[],env,args,funty)
857     get_term_args n max args env funty
858       | (case known_clo of
859            Just (_,_,needed,_) -> needed `lengthIs` n
860            Nothing -> False)
861        -- Stop if we have the optimal number for a direct call
862        = ([],[],env,args,funty)
863     get_term_args _ _ (args@(StgTypeArg _:_)) env funty 
864        -- Stop if we hit a type arg.
865        = ([],[],env,args,funty)
866     get_term_args n max (h:t) env (FunTy dom ran)
867        -- Take an argument.
868        = let (now,now_tys,env2,later,later_ty) = get_term_args (n+1) (max - 1) t env ran in 
869          (h:now, (h,dom):now_tys,env2,later,later_ty)
870     get_term_args _ max (h:t) env funty = trace "IlxGen Internal Error: get_term_args could not get FunTy or ForAllTy for corresponding arg" ([],[],env,[],funty)
871     get_term_args _ max args env funty = ([],[],env,args,funty)
872
873     -- Are there any remaining arguments?
874     done  = case later_args of
875           [] -> True
876           _ -> False
877
878     -- If so, generate the subsequent calls.
879     later = if done then text "// done"  
880             else ilxFunAppArgs env (num_sofar + length now_args) later_ty later_args tail_call Nothing
881
882     -- Work out whether to issue a direct call a known closure (callclo) or
883     -- an indirect call (callfunc).  Basically, see if the identifier has
884     -- been let-bound, and then check we are applying exactly the right 
885     -- number of arguments.  Also check that it's not a thunk (actually, this
886     -- is done up above).
887     -- 
888     -- The nasty "all" check makes sure that 
889     -- the set of type variables in scope at the callsite is a superset 
890     -- of the set of type variables needed for the direct call.  This is
891     -- is needed because not all of the type variables captured by a 
892     -- let-bound binding will get propogated down to the callsite, and 
893     -- the ILX system of polymorphism demands that the free type variables
894     -- get reapplied when we issue the direct "callclo".  The
895     -- type variables are in reality also "bound up" in the closure that is
896     -- passed as the first argument, so when we do an indirect call
897     -- to that closure we're fine, which is why we don't need them in 
898     -- the "callfunc" case.
899     basic_call_instr =
900       case known_clo of
901         Just (known_env,fun,needed,fvs) | (equalLength needed now_args) && 
902                                           all (\x -> elemIlxTyEnv x env) free_ilx_tvs -> 
903            vcat [text "callclo class",
904                  nameReference env (idName fun) <+> singleQuotes (ilxEnvQualifyByModule env (ppr fun)),
905                  pprTypeArgs ilxTypeR env (map mkTyVarTy free_ilx_tvs)]
906            <> text ","
907           where 
908            (free_ilx_tvs, free_non_ilx_tvs,free_vs) = categorizeVars fvs
909         otherwise -> text "callfunc"
910     call_instr =
911            if (tail_call && done) then text "tail." <+> basic_call_instr
912            else basic_call_instr
913
914
915 --------------------------
916 -- Print the arg info at the call site
917 -- For type args we are, at the moment, required to
918 -- give both the actual and the formal (bound).  The formal
919 -- bound is always System.Object at the moment (bounds are
920 -- not properly implemented in ILXASM in any case, and nor do
921 -- we plan on making use og them) For
922 -- non-type args the actuals are on the stack, and we just give the
923 -- formal type.
924 pprIlxArgInfo env (StgTypeArg  arg,ty) =  
925     angleBrackets (ilxTypeR env (deepIlxRepType arg) <+> ilxComment (text "actual for tyvar")) <+> text "<class [mscorlib] System.Object>" 
926 pprIlxArgInfo env (_,ty) =  
927     parens (ilxTypeL env ty)
928
929
930 ----------------------------
931 -- Code for a binding
932 ilxBind :: IlxEEnv -> StgBinding -> SDoc
933 ilxBind eenv@(IlxEEnv env _) bind = 
934     vcat [vcat (map (ilxRhs env rec) pairs), 
935           vcat (map (ilxFixupRec env rec) pairs)]
936        where 
937          rec = ilxRecIds1 bind
938          pairs = ilxPairs1 bind
939
940
941 ----------------------------
942 -- Allocate a closure or constructor.  Fix up recursive definitions.
943 ilxRhs :: IlxEnv -> [Id] -> (Id, StgRhs) -> SDoc
944
945 ilxRhs env rec (bndr, _) | isVoidIlxRepId bndr  
946   = empty
947
948 ilxRhs env rec (bndr, StgRhsCon _ con args)
949   = vcat [text " /* ilxRhs:StgRhsCon */ " <+> ilxConApp env con args,
950            text "stloc" <+> pprId bndr
951           ]
952
953 ilxRhs env rec (bndr, StgRhsClosure _ _ fvs upd args rhs)
954   =     -- Assume .closure v<any A>(int64,!A) { 
955         --              .apply <any B> (int32) (B) { ... }
956         --         }
957         -- Then
958         --    let v = \B (x:int32) (y:B). ... 
959         -- becomes:
960         --    newclo v<int32>(int64,!0)
961         --    stloc v
962     vcat [vcat (map pushFv free_vs),
963           (if null free_non_ilx_tvs then empty else (ilxComment (text "ignored some higher order type arguments in application - code will be non-verifiable"))),
964           text "newclo" <+> clotext,
965           text "stloc" <+> pprId bndr
966     ]
967   where
968     pushFv id = if elem id rec then text "ldnull" else pushId env id
969     (free_ilx_tvs, free_non_ilx_tvs,free_vs) = categorizeVars fvs
970     clotext = pprIlxNamedTyConApp env (ilxEnvQualifyByModule env (ppr bndr)) (map mkTyVarTy free_ilx_tvs)
971
972 ilxFixupRec env rec (bndr, _) | isVoidIlxRepId bndr = ilxComment (text "no recursive fixup for void-rep-id")
973
974 ilxFixupRec env rec (bndr, StgRhsCon _ con args)
975   = text "// no recursive fixup"
976
977 ilxFixupRec env rec (bndr, StgRhsClosure _ _ fvs upd args rhs)
978      = vcat [vcat (map fixFv rec)]
979   where
980     fixFv recid = if elem recid fvs then 
981                     vcat [pushId env bndr,
982                           pushId env recid,
983                           text "stclofld" <+> clotext <> text "," <+> pprId recid] 
984                 else text "//no fixup needed for" <+> pprId recid
985     (free_ilx_tvs, free_non_ilx_tvs,free_vs) = categorizeVars fvs
986     clotext = pprIlxNamedTyConApp env (ilxEnvQualifyByModule env (ppr bndr)) (map mkTyVarTy free_ilx_tvs)
987
988
989
990 ---------------------------------------------
991 -- Code for a top-level binding in a module
992 ilxPairs binds = concat (map ilxPairs1 binds)
993
994 ilxPairs1 (StgNonRec _ bndr rhs) = [(bndr,rhs)]
995 ilxPairs1 (StgRec _ pairs)       = pairs
996
997 ilxRecIds1 (StgNonRec _ bndr rhs) = []
998 ilxRecIds1 (StgRec _ pairs)       = map fst pairs
999
1000 ---------------------------------------------
1001 -- Code for a top-level binding in a module
1002 -- TODO: fix up recursions amongst CAF's
1003 -- e.g. 
1004 --    x = S x
1005 -- for infinity...
1006 -- 
1007 -- For the moment I've put in a completely spurious "reverse"...
1008 --
1009 -- Consider: make fixing up of CAF's part of ILX?  i.e.
1010 -- put static, constant, allocated datastructures into ILX. 
1011
1012 stableSortBy :: (a -> a -> Ordering) -> [a] -> [a]
1013 stableSortBy f (h:t) = insertBy f h (stableSortBy f t)
1014 stableSortBy f [] = []
1015
1016 usedBy :: (Id,StgRhs) -> (Id,StgRhs) -> Ordering
1017 usedBy (m,_) (_,StgRhsCon _ data_con args) | any (isArg m) args = LT
1018 usedBy (m,_) (n,_) | m == n = EQ
1019 usedBy (m,_) (_,_) = GT
1020
1021 isArg m  (StgVarArg n) = (n == m)
1022 isArg m _ = False
1023
1024
1025 ilxTopBind :: Module -> IlxEnv -> [(Id,StgRhs)] -> SDoc
1026 --ilxTopBind mod env (StgNonRec _ bndr rhs) = 
1027 --ilxTopRhs env (bndr,rhs)
1028 ilxTopBind mod env pairs       = 
1029    vcat [text ".class" <+> pprId mod,
1030          nest 2 (braces (nest 2 (vcat [empty,cctor, flds, empty])))]
1031      where
1032        cctor = vcat [text ".method static rtspecialname specialname void .cctor()",
1033                      nest 2 (braces 
1034                       (nest 2 (vcat [text ".maxstack 100",
1035                                      text "ldstr \"LOG: initializing module" <+> pprId mod <+> text "\" call void ['mscorlib']System.Console::WriteLine(class [mscorlib]System.String)",
1036                                      vcat (map (ilxTopRhs mod env) (stableSortBy usedBy pairs)), 
1037                                      text "ldstr \"LOG: initialized module" <+> pprId mod <+> text "\" call void ['mscorlib']System.Console::WriteLine(class [mscorlib]System.String)",
1038                                      text "ret",
1039                                      empty])))]
1040        flds =   vcat (map (ilxTopRhsStorage mod env) pairs)
1041
1042 --ilxTopRhs mod env (bndr, _) | isVoidIlxRepId bndr 
1043 --  = empty
1044
1045 ilxTopRhs mod env (bndr, StgRhsClosure _ _ fvs upd args rhs)
1046   = vcat [vcat (map (pushId env) free_vs),
1047          (if null free_non_ilx_tvs then empty else (ilxComment (text "ignored some higher order type arguments in application - code will be non verifiable...."))),
1048           text "newclo" <+> pprIlxNamedTyConApp env (ilxEnvQualifyByModule env (ppr bndr)) (map mkTyVarTy free_ilx_tvs),
1049           text "stsfld"  <+> pprFieldRef env (mod,bndTy,bndr)
1050     ]
1051   where
1052     (free_ilx_tvs, free_non_ilx_tvs,free_vs) = categorizeVars fvs
1053     bndTy = idIlxRepType bndr
1054
1055 ilxTopRhs mod env (bndr, StgRhsCon _ data_con args)
1056   = vcat [ text " /* ilxTopRhs: StgRhsCon */ " <+> ilxConApp env data_con args, 
1057            text "stsfld" <+> pprFieldRef env (mod,bndTy,bndr)
1058     ]
1059   where
1060     bndTy = idIlxRepType bndr
1061
1062 pprFieldRef env (mod,ty,id) 
1063   =  ilxTypeL env ty <+> moduleReference env mod <+> pprId mod <> text "::" <> pprId id
1064
1065 ilxTopRhsStorage mod env (bndr, StgRhsClosure _ _ _ _ _ _) 
1066   =   text ".field public static " <+> ilxTypeL env bndTy <+> pprId bndr
1067   where
1068     bndTy = idIlxRepType bndr
1069 ilxTopRhsStorage mod env (bndr, StgRhsCon _ _ _) 
1070   =   text ".field public static " <+> ilxTypeL env bndTy <+> pprId bndr
1071   where
1072     bndTy = idIlxRepType bndr
1073
1074 --------------------------------------
1075 -- Push an argument
1076 pushArgWithVoids =  pushArg_aux True
1077 pushArg = pushArg_aux False
1078
1079 pushArg_aux voids env (StgTypeArg ty) = empty
1080 pushArg_aux voids env (StgVarArg var) = pushId_aux voids env var
1081 pushArg_aux voids env (StgLitArg lit) = pushLit env lit
1082
1083
1084 mapi f l = mapi_aux f l 0
1085
1086 mapi_aux f [] n = []
1087 mapi_aux f (h:t) n = f n h : mapi_aux f t (n+1)
1088
1089 --------------------------------------
1090 -- Push an Id
1091 pushId = pushId_aux False
1092
1093 pushId_aux :: Bool -> IlxEnv -> Id -> SDoc
1094 pushId_aux voids _ id | isVoidIlxRepId id =
1095    /* if voids then  text "ldunit" else */ ilxComment (text "pushId: void rep skipped")
1096 pushId_aux _ env var 
1097   = case lookupIlxVarEnv env var of
1098           Just Arg    -> text "ldarg"    <+> pprId var
1099           Just (CloVar n) -> text "ldenv" <+> int n
1100           Just Local  -> text "ldloc"    <+> pprId var
1101           Just (Top m)  -> 
1102              vcat [ilxComment (text "pushId (Top) " <+> pprId m), 
1103                    text "ldsfld" <+> ilxTypeL env (idIlxRepType var)
1104                       <+> moduleReference env m <+> pprId (moduleName m) <> text "::" <> pprId var]
1105
1106           Nothing ->  
1107              vcat [ilxComment (text "pushId (import) " <+> pprIlxTopVar env var), 
1108                    text "ldsfld" <+> ilxTypeL env (idIlxRepType var) 
1109                     <+> pprIlxTopVar env var]
1110
1111 --------------------------------------
1112 -- Push a literal
1113 pushLit env (MachChar c)   = text "ldc.i4" <+> int c
1114 pushLit env (MachStr s)    = text "ldsflda char "  <+> ilxEnvQualifyByExact env (text "string") -- pprFSInILStyle s 
1115 pushLit env (MachInt i)    = text "ldc.i4" <+> integer i
1116 pushLit env (MachInt64 i)  = text "ldc.i8" <+> integer i
1117 pushLit env (MachWord w)   = text "ldc.i4" <+> integer w <+> text "conv.u4"
1118 pushLit env (MachWord64 w) = text "ldc.i8" <+> integer w <+> text "conv.u8"
1119 pushLit env (MachFloat f)  = text "ldc.r4" <+> rational f
1120 pushLit env (MachDouble f) = text "ldc.r8" <+> rational f
1121 pushLit env (MachNullAddr)  = text "ldc.i4 0"
1122 pushLit env (MachLabel l _) = trace "WARNING: Cannot compile MachLabel to ILX in IlxGen.lhs" (text "// MachLabel!!!  Not valid in ILX!!")
1123
1124 pprIlxTopVar env v
1125   | isExternalName n = (nameReference env n) <> pprId (nameModule n) <> text "::" <> singleQuotes (ppr (nameModule n) <> text "_" <> ppr (nameOccName n))
1126   | otherwise      = pprId (nameOccName n)
1127   where
1128     n = idName v
1129
1130 \end{code}
1131
1132
1133 %************************************************************************
1134 %*                                                                      *
1135 \subsection{Printing types}
1136 %*                                                                      *
1137 %************************************************************************
1138
1139
1140 \begin{code}
1141
1142 isVoidIlxRepType (NoteTy   _ ty) = isVoidIlxRepType ty
1143 isVoidIlxRepType (TyConApp tc _) | (tyConPrimRep tc == VoidRep) = True
1144 isVoidIlxRepType (TyConApp tc tys) 
1145   = isUnboxedTupleTyCon tc && null (filter (not. isVoidIlxRepType) tys)
1146 isVoidIlxRepType _ = False
1147
1148 isVoidIlxRepId id = isVoidIlxRepType (idType id)
1149
1150
1151
1152 -- Get rid of all NoteTy and NewTy artifacts
1153 deepIlxRepType :: Type -> Type
1154 deepIlxRepType (FunTy l r)
1155   = FunTy (deepIlxRepType l) (deepIlxRepType r)
1156
1157 deepIlxRepType ty@(TyConApp tc tys) 
1158   =        -- collapse UnboxedTupleTyCon down when it contains VoidRep types.
1159            -- e.g.      (# State#, Int#, Int# #)  ===>   (# Int#, Int# #)
1160             if isUnboxedTupleTyCon tc then 
1161                let tys' = map deepIlxRepType (filter (not. isVoidIlxRepType) tys) in 
1162                case tys' of
1163                   [h] -> h
1164                   _ -> mkTupleTy Unboxed (length tys') tys'
1165             else 
1166               TyConApp tc (map deepIlxRepType tys)
1167 deepIlxRepType (AppTy f x)     = AppTy (deepIlxRepType f) (deepIlxRepType x)
1168 deepIlxRepType (ForAllTy b ty) = ForAllTy b (deepIlxRepType ty)
1169 deepIlxRepType (NoteTy   _ ty) = deepIlxRepType ty
1170 deepIlxRepType (PredTy p)      = deepIlxRepType (predTypeRep p)
1171 deepIlxRepType ty@(TyVarTy tv) = ty
1172
1173 idIlxRepType id = deepIlxRepType (idType id)
1174
1175 --------------------------
1176 -- Some primitive type constructors are not thunkable.
1177 -- Everything else needs to be marked thunkable.
1178 ilxTypeL :: IlxEnv -> Type -> SDoc
1179
1180 ilxTypeL env ty | isUnLiftedType ty ||  isVoidIlxRepType ty = ilxTypeR env ty
1181 ilxTypeL env ty = text "thunk" <> angleBrackets (ilxTypeR env ty)
1182
1183
1184 --------------------------
1185 -- Print non-thunkable version of type.
1186 --
1187
1188 ilxTypeR :: IlxEnv -> Type -> SDoc
1189 ilxTypeR env ty | isVoidIlxRepType ty = text "/* unit skipped */"
1190 ilxTypeR env ty@(AppTy f _) | isTyVarTy f    = ilxComment (text "type app:" <+> pprType ty) <+> (text "class [mscorlib]System.Object")
1191 ilxTypeR env ty@(AppTy f x)     = trace "ilxTypeR: should I be beta reducing types?!" (ilxComment (text "ilxTypeR: should I be beta reducing types?!") <+> ilxTypeR env (applyTy f x))
1192 ilxTypeR env (TyVarTy tv)       = ilxTyVar env tv
1193
1194 -- The following is a special rule for types constructed out of 
1195 -- higher kinds, e.g. Monad f or Functor f.  
1196 --
1197 -- The code below is not as general as it should be, but as I
1198 -- have no idea if this approach will even work, I'm going to
1199 -- just try it out on some simple cases arising from the prelude.
1200 ilxTypeR env ty@(TyConApp tc (h:t)) | isAlgTyCon tc && null (tyConTyVars tc)
1201    = ilxComment (text "what on earth? 2") <+> (ilxTypeR env (TyConApp tc t))
1202 ilxTypeR env ty@(TyConApp tc (h:t)) | isAlgTyCon tc && not (isIlxTyVar (hd (tyConTyVars tc)))
1203    = ilxTypeR env (TyConApp tc t)
1204 ilxTypeR env (TyConApp tc args) = ilxTyConApp env tc args
1205
1206   -- nb. the only legitimate place for VoidIlxRepTypes to occur in normalized IlxRepTypes 
1207   -- is on the left of an arrow
1208   --  We could probably eliminate all but a final occurrence of these.
1209 ilxTypeR env (FunTy arg res)| isVoidIlxRepType res 
1210     = pprIlxFunTy (ilxTypeL env arg) (text "void")
1211 ilxTypeR env (FunTy arg res)
1212     = pprIlxFunTy (ilxTypeL env arg) (ilxTypeR env res)
1213
1214 ilxTypeR env ty@(ForAllTy tv body_ty) | isIlxTyVar tv
1215   = parens (text "forall" <+> pprTyVarBinders env' [tv] <+> nest 2 (ilxTypeR env' body_ty))
1216     where
1217        env' = extendIlxEnvWithFormalTyVars env [tv]
1218
1219 ilxTypeR env ty@(ForAllTy tv body_ty) | otherwise
1220   = ilxComment (text "higher order type var " <+> pprId tv) <+>
1221     pprIlxFunTy (text "class [mscorlib]System.Object") (ilxTypeR env body_ty)
1222
1223 ilxTypeR env (NoteTy _ ty)       
1224    = trace "WARNING! non-representation type given to ilxTypeR: see generated ILX for context where this occurs"
1225      (vcat [text "/* WARNING! non-representation type given to ilxTypeR! */",
1226            ilxTypeR env ty ])
1227
1228 pprIlxFunTy dom ran = parens (hsep [text "func",parens dom,text "-->", ran])
1229
1230 ilxTyConApp env tcon args =
1231    case lookupUFM tyPrimConTable (getUnique tcon) of
1232         Just f  -> f args env
1233         Nothing -> 
1234             (if isUnboxedTupleTyCon tcon then pprIlxUnboxedTupleTyConApp else pprIlxBoxedTyConApp)
1235               env tcon args
1236
1237 pprIlxTyCon env tcon = nameReference env (getName tcon) <> ppr tcon
1238 pprIlxUnboxedTupleTyConApp env tcon args 
1239   = text "/* unboxed */ value class" <+> pprIlxTyCon env tcon' <> pprTypeArgs ilxTypeL env non_void
1240   where 
1241    non_void = filter (not . isVoidIlxRepType) args
1242    tcon' = dataConTyCon (tupleCon Unboxed (length non_void)) 
1243 pprIlxBoxedTyConApp env tcon args 
1244   = pprIlxNamedTyConApp env (pprIlxTyCon env tcon) args
1245 pprIlxNamedTyConApp env tcon_text args 
1246   = text "class" <+> tcon_text <> pprTypeArgs ilxTypeR env args
1247
1248 -- Returns e.g: <Int32, Bool>
1249 -- Void-sized type arguments are _always_ eliminated, everywhere.
1250 -- If the type constructor is an unboxed tuple type then it should already have
1251 -- been adjusted to be the correct constructor.
1252 pprTypeArgs f env tys = pprTypeArgs_aux f env (filter (not . isVoidIlxRepType) tys)
1253
1254 pprTypeArgs_aux f env []  = empty
1255 pprTypeArgs_aux f env tys = angleBrackets (pprSepWithCommas (f env) tys)
1256
1257
1258 pprTyVarBinders :: IlxEnv -> [TyVar] -> SDoc
1259 -- Returns e.g: <class [mscorlib]System.Object> <class [mscorlib]System.Object>
1260 -- plus a new environment with the type variables added.
1261 pprTyVarBinders env [] = empty
1262 pprTyVarBinders env tvs = angleBrackets (pprSepWithCommas (pprTyVarBinder_aux env) tvs)
1263
1264 pprTyVarBinder :: IlxEnv -> TyVar -> SDoc
1265 pprTyVarBinder env tv = 
1266     if isIlxTyVar tv then 
1267        angleBrackets (pprTyVarBinder_aux env tv)
1268     else
1269        ilxComment (text "higher order tyvar" <+> pprId tv <+> 
1270                          text ":" <+> ilxTypeR env (tyVarKind tv)) <+>
1271              ilxComment (text "omitted")
1272              -- parens (text "class [mscorlib]System.Object" <+> pprId tv)
1273
1274
1275 pprTyVarBinder_aux env tv = 
1276    ilxComment (text "tyvar" <+> pprId tv <+> text ":" <+> 
1277                         ilxTypeR env (tyVarKind tv)) <+>
1278              (text "class [mscorlib]System.Object")
1279
1280 -- Only a subset of Haskell types can be generalized using the type quantification
1281 -- of ILX
1282 isIlxForAllKind h = 
1283         ( h `eqKind` liftedTypeKind) ||
1284         ( h `eqKind` unliftedTypeKind) ||
1285         ( h `eqKind` openTypeKind)
1286
1287 isIlxTyVar v = isTyVar v && isIlxForAllKind (tyVarKind v)
1288
1289 categorizeVars fvs = (ilx_tvs, non_ilx_tvs, vs)
1290          where
1291            (tvs, vs) = partition isTyVar fvs
1292            (ilx_tvs, non_ilx_tvs) = categorizeTyVars tvs
1293
1294 categorizeTyVars tyvs = partition isIlxTyVar tyvs
1295
1296 pprValArgTys ppr_ty env tys = parens (pprSepWithCommas (ppr_ty env) tys)
1297
1298 pprId id = singleQuotes (ppr id)
1299
1300 \end{code}                      
1301
1302 %************************************************************************
1303 %*                                                                      *
1304 \subsection{IlxEnv}     
1305 %*                                                                      *
1306 %************************************************************************
1307
1308 \begin{code}
1309 type IlxTyEnv = [TyVar]
1310 emptyIlxTyEnv = []
1311
1312 -- Nb. There is currently no distinction between the kinds of type variables.
1313 -- We may need to add this to print out correct numbers, esp. for
1314 -- "forall" types
1315 extendIlxTyEnvWithFreeTyVars env tyvars = env ++ mkIlxTyEnv tyvars -- bound by .closure x<...> in a closure declared with type parameters
1316 extendIlxTyEnvWithFormalTyVars env tyvars = env ++ mkIlxTyEnv tyvars -- bound by "forall <...>" in a type
1317 extendIlxTyEnvWithTyArgs env tyvars = env ++ mkIlxTyEnv tyvars -- bound by "<...>" in a closure implementing a universal type
1318
1319 formalIlxTyEnv tyvars = mkIlxTyEnv tyvars
1320 mkIlxTyEnv tyvars = [ v | v <- tyvars, isIlxTyVar v ]
1321
1322 data HowBound = Top Module      -- Bound in a modules
1323               | Arg     -- Arguments to the enclosing closure
1324               | CloVar Int -- A free variable of the enclosing closure
1325                            -- The int is the index of the field in the 
1326                            -- environment
1327               | Local   -- Local let binding
1328
1329 -- The SDoc prints a unique name for the syntactic block we're currently processing,
1330 -- e.g. Foo_bar_baz when inside closure baz inside closure bar inside module Foo.
1331 data IlxEnv = IlxEnv (Module, IlxTyEnv, IdEnv HowBound,IdEnv (IlxEnv, StgRhs), Place,Bool)
1332 type Place = (SDoc,SDoc)
1333
1334 ilxTyVar  env tv
1335   = go 0 (ilxEnvTyEnv env)
1336   where
1337     go n []                 
1338       = pprTrace "ilxTyVar" (pprId tv <+> text "tv_env = { "
1339            <+> pprSepWithCommas
1340                  (\x -> pprId x <+> text ":" <+> ilxTypeR env (tyVarKind x)) 
1341                (ilxEnvTyEnv env) <+> text "}") 
1342         (char '!' <> pprId tv) 
1343     go n (x:xs)
1344       = {- pprTrace "go" (ppr (tyVarName tv) <+> ppr (tyVarName x)) -}
1345         (if tyVarName x== tyVarName tv then  char '!' <> int n <+> ilxComment (char '!' <> pprId tv) 
1346          else go (n+1) xs)
1347
1348 emptyIlxEnv :: Bool -> Module -> IlxEnv
1349 emptyIlxEnv trace mod = IlxEnv (mod, emptyIlxTyEnv, emptyVarEnv, emptyVarEnv, (ppr mod,empty),trace)
1350
1351 nextPlace place sdoc = place <> sdoc
1352 usePlace  place sdoc = place <> sdoc
1353
1354 ilxEnvModule (IlxEnv (m, _, _,  _, _,_)) = m
1355 ilxEnvSetPlace (IlxEnv (m, tv_env, id_env,  bind_env, (mod,exact),tr)) sdoc 
1356    = IlxEnv (m, tv_env, id_env,  bind_env, (mod, sdoc),tr)
1357 ilxEnvNextPlace (IlxEnv (m, tv_env, id_env,  bind_env, (mod,exact),tr)) sdoc 
1358    = IlxEnv (m, tv_env, id_env,  bind_env, (mod, nextPlace exact sdoc),tr)
1359 ilxEnvQualifyByModule (IlxEnv (_, _, _, _,(mod,_),_)) sdoc = usePlace mod sdoc
1360 ilxEnvQualifyByExact (IlxEnv (_, _, _, _,(mod,exact),_)) sdoc = usePlace mod sdoc <> usePlace exact sdoc
1361
1362 ilxPlaceStgBindDefault env = ilxEnvNextPlace env (text "D")
1363 ilxPlaceStgRhsClosure env bndr = ilxEnvSetPlace env (ppr bndr) -- binders are already unique
1364 ilxPlaceStgCaseScrut env = ilxEnvNextPlace env (text "S")
1365
1366 ilxPlaceAlt :: IlxEnv -> Int -> IlxEnv
1367 ilxPlaceAlt env i = ilxEnvNextPlace env (text "a" <> int i)
1368 ilxPlacePrimAltLit env i = ilxEnvNextPlace env (text "P" <> int i)
1369 ilxMapPlaceArgs start f env args = [ f (ilxEnvNextPlace env (text "A" <> int i)) a | (i,a) <- [start..] `zip` args ]
1370 ilxMapPlaceAlts f env alts = [ f (ilxPlaceAlt env i) alt | (i,alt) <- [1..] `zip` alts ]
1371
1372 extendIlxEnvWithFreeTyVars (IlxEnv (mod, tv_env, id_env,  bind_env, place,tr)) tyvars 
1373   = IlxEnv (mod, extendIlxTyEnvWithFreeTyVars tv_env tyvars,id_env,  bind_env, place,tr)
1374
1375 extendIlxEnvWithFormalTyVars (IlxEnv (mod, tv_env, id_env,  bind_env, place,tr)) tyvars 
1376   = IlxEnv (mod, extendIlxTyEnvWithFormalTyVars tv_env tyvars,id_env,  bind_env, place,tr)
1377
1378 extendIlxEnvWithTyArgs (IlxEnv (mod, tv_env, id_env,  bind_env, place,tr)) tyvars 
1379   = IlxEnv (mod, extendIlxTyEnvWithTyArgs tv_env tyvars,id_env,  bind_env, place,tr)
1380
1381 extendIlxEnvWithArgs :: IlxEnv -> [Var] -> IlxEnv
1382 extendIlxEnvWithArgs (IlxEnv (mod, tv_env, id_env,  bind_env, place,tr)) args
1383   = IlxEnv (mod, extendIlxTyEnvWithTyArgs tv_env [tv      | tv <- args, isIlxTyVar tv],
1384             extendVarEnvList id_env [(v,Arg) | v  <- args, not (isIlxTyVar v)], 
1385              bind_env, place,tr)
1386
1387 extendIlxEnvWithFreeVars (IlxEnv (mod, tv_env, id_env,  bind_env, place,tr)) args
1388   = IlxEnv (mod, 
1389             extendIlxTyEnvWithFreeTyVars tv_env [tv | tv <- args, isIlxTyVar tv],
1390             extendVarEnvList id_env (clovs 0 args), 
1391             bind_env, 
1392             place,tr)
1393    where
1394      clovs _ [] = []
1395      clovs n (x:xs) = if not (isIlxTyVar x) then (x,CloVar n):clovs (n+1) xs else clovs n xs
1396
1397 extendIlxEnvWithBinds env@(IlxEnv (mod, tv_env, id_env, bind_env, place,tr)) bnds
1398   = IlxEnv (mod, tv_env, id_env, 
1399             extendVarEnvList bind_env [(v,(env,rhs)) | (v,rhs) <- bnds], 
1400             place,tr)
1401
1402 extendIlxEnvWithLocals (IlxEnv (m, tv_env, id_env, bind_env, p,tr)) locals
1403   = IlxEnv (m, tv_env, 
1404             extendVarEnvList id_env [(v,Local) | (LocalId v,_) <- locals],
1405             extendVarEnvList bind_env [(v,(env,rhs)) | (LocalId v,Just (env,rhs)) <- locals], 
1406             p,tr)
1407 extendIlxEnvWithTops env@(IlxEnv (m, tv_env, id_env, bind_env, place,tr)) mod binds
1408   = IlxEnv (m, tv_env, 
1409             extendVarEnvList id_env [(bndr,Top mod) | (bndr,rhs) <- binds], 
1410             extendVarEnvList bind_env [(bndr,(env, rhs)) | (bndr,rhs) <- binds], 
1411             place,tr)
1412
1413 formalIlxEnv (IlxEnv (m, tv_env, id_env, bind_env, place, tr)) tyvars 
1414   = IlxEnv (m, formalIlxTyEnv tyvars, id_env, bind_env, place, tr)
1415
1416 ilxEnvTyEnv :: IlxEnv -> IlxTyEnv
1417 ilxEnvTyEnv (IlxEnv (_, tv_env, _,_,_,_)) = tv_env 
1418 elemIlxTyEnv var env = elem var (ilxEnvTyEnv env )
1419 elemIlxVarEnv var (IlxEnv (_, _, id_env,_,_,_)) = elemVarEnv var id_env 
1420 lookupIlxVarEnv (IlxEnv (_, _, id_env,_,_,_)) var = lookupVarEnv id_env var
1421 lookupIlxBindEnv (IlxEnv (_, _, _, bind_env,_,_)) var = lookupVarEnv bind_env var
1422
1423 \end{code}
1424
1425
1426 \begin{code}
1427 type IlxLabel = SDoc
1428
1429 pprIlxLabel lbl = lbl
1430
1431 mkJoinLabel :: Id -> IlxLabel
1432 mkJoinLabel v = text "J_" <> ppr v
1433
1434 mkAltLabel  :: Id -> Int -> IlxLabel
1435 mkAltLabel v n = text "A" <> int n <> ppr v
1436
1437 ilxLabel :: IlxLabel -> SDoc
1438 ilxLabel lbl =  line $$ (pprIlxLabel lbl <> colon)
1439 \end{code}
1440
1441
1442 %************************************************************************
1443 %*                                                                      *
1444 \subsection{Local pretty helper functions}
1445 %*                                                                      *
1446 %************************************************************************
1447
1448 \begin{code}
1449 pprSepWithCommas :: (a -> SDoc) -> [a] -> SDoc
1450 pprSepWithCommas pp xs = sep (punctuate comma (map pp xs))
1451 ilxComment pp   = text "/*" <+> pp <+> text "*/"
1452 singleQuotes pp = char '\'' <> pp <> char '\''
1453
1454 line = text "// ----------------------------------"
1455
1456 hscOptionQual = text ".i_"
1457
1458 nameReference env n
1459   | isInternalName n = empty
1460   | ilxEnvModule env == nameModule n  = text ""
1461   | isHomeModule (nameModule n)   = moduleNameReference (moduleName (nameModule n))
1462 -- HACK: no Vanilla modules should be around, but they are!!  This
1463 -- gets things working for the scenario "standard library linked as one
1464 -- assembly with multiple modules + a one module program running on top of this"
1465 -- Same applies to all other mentions of Vailla modules in this file
1466   | isVanillaModule (nameModule n)  && not inPrelude =  basePackageReference
1467   | isVanillaModule (nameModule n)  && inPrelude =   moduleNameReference (moduleName (nameModule n))
1468 -- end hack
1469   | otherwise = packageReference (modulePackage (nameModule n))
1470
1471 packageReference p = brackets (singleQuotes (ppr p  <> hscOptionQual))
1472 moduleNameReference m = brackets ((text ".module") <+> (singleQuotes (pprModuleName m <> hscOptionQual <> text "o")))
1473
1474 moduleReference env m
1475   | ilxEnvModule env   == m = text ""
1476   | isHomeModule m = moduleNameReference (moduleName m)
1477   -- See hack above
1478   | isVanillaModule m && not inPrelude =  basePackageReference
1479   | isVanillaModule m && inPrelude =  moduleNameReference (moduleName m)
1480   -- end hack
1481   | otherwise  =  packageReference (modulePackage m)
1482
1483 basePackageReference = packageReference basePackage
1484 inPrelude = basePackage == opt_InPackage
1485
1486 ------------------------------------------------
1487 -- This code is copied from absCSyn/CString.lhs,
1488 -- and modified to do the correct thing!  It's
1489 -- still a mess though.  Also, still have to do the
1490 -- right thing for embedded nulls.
1491
1492 pprFSInILStyle :: FastString -> SDoc
1493 pprFSInILStyle fs = doubleQuotes (text (stringToC (unpackFS fs)))
1494
1495 stringToC   :: String -> String
1496 -- Convert a string to the form required by C in a C literal string
1497 -- Tthe hassle is what to do w/ strings like "ESC 0"...
1498 stringToC ""  = ""
1499 stringToC [c] = charToC c
1500 stringToC (c:cs)
1501     -- if we have something "octifiable" in "c", we'd better "octify"
1502     -- the rest of the string, too.
1503   = if (c < ' ' || c > '~')
1504     then (charToC c) ++ (concat (map char_to_C cs))
1505     else (charToC c) ++ (stringToC cs)
1506   where
1507     char_to_C c | c == '\n' = "\\n"     -- use C escapes when we can
1508                 | c == '\a' = "\\a"
1509                 | c == '\b' = "\\b"     -- ToDo: chk some of these...
1510                 | c == '\r' = "\\r"
1511                 | c == '\t' = "\\t"
1512                 | c == '\f' = "\\f"
1513                 | c == '\v' = "\\v"
1514                 | otherwise = '\\' : (trigraph (ord c))
1515
1516 charToC :: Char -> String
1517 -- Convert a character to the form reqd in a C character literal
1518 charToC c = if (c >= ' ' && c <= '~')   -- non-portable...
1519             then case c of
1520                   '\'' -> "\\'"
1521                   '\\' -> "\\\\"
1522                   '"'  -> "\\\""
1523                   '\n' -> "\\n"
1524                   '\a' -> "\\a"
1525                   '\b' -> "\\b"
1526                   '\r' -> "\\r"
1527                   '\t' -> "\\t"
1528                   '\f' -> "\\f"
1529                   '\v' -> "\\v"
1530                   _    -> [c]
1531             else '\\' : (trigraph (ord c))
1532
1533 trigraph :: Int -> String
1534 trigraph n
1535   = [chr ((n `div` 100) `rem` 10 + ord '0'),
1536      chr ((n `div` 10) `rem` 10 + ord '0'),
1537      chr (n `rem` 10 + ord '0')]
1538
1539
1540 \end{code}
1541
1542 %************************************************************************
1543 %*                                                                      *
1544 \subsection{PrimOps and Constructors}
1545 %*                                                                      *
1546 %************************************************************************
1547
1548 \begin{code}
1549 ----------------------------
1550 -- Allocate a fresh constructor
1551
1552 ilxConApp env data_con args
1553   | isUnboxedTupleCon data_con
1554      = let tm_args' = filter (not. isVoidIlxRepType . stgArgType) tm_args in 
1555        case tm_args' of
1556         [h] -> 
1557           -- Collapse the construction of an unboxed tuple type where
1558           -- every element is zero-sized
1559             vcat (ilxMapPlaceArgs 0 pushArg env tm_args')
1560         _ -> 
1561           -- Minimize the construction of an unboxed tuple type, which
1562           -- may contain zero-sized elements.  Recompute all the 
1563           -- bits and pieces from the simpler case below for the new data
1564           -- type constructor....
1565            let data_con' = tupleCon Unboxed (length tm_args') in 
1566            let rep_ty_args' = filter (not . isVoidIlxRepType) rep_ty_args in 
1567
1568            let tycon' = dataConTyCon data_con' in
1569            let (formal_tyvars', formal_tau_ty') = splitForAllTys (dataConRepType data_con') in 
1570            let (formal_arg_tys', _)     = splitFunTys formal_tau_ty' in
1571            let formal_env'           = formalIlxEnv env formal_tyvars' in 
1572
1573            vcat [vcat (ilxMapPlaceArgs 0 pushArg env tm_args'),
1574                    sep [text "newobj void ",
1575                         ilxTyConApp env tycon' rep_ty_args',
1576                         text "::.ctor",
1577                         pprValArgTys ilxTypeR formal_env' (map deepIlxRepType formal_arg_tys')
1578                    ]
1579              ]
1580  | otherwise
1581     -- Now all other constructions
1582      =  --  Assume C :: forall a. a -> T a -> T a
1583         --      ldloc x         arg of type Int
1584         --      ldloc y         arg of type T Int
1585         --      newdata classunion T<Int32>, C(!0, T <!0>)
1586         --
1587         let tycon   = dataConTyCon data_con in 
1588         let (formal_tyvars, formal_tau_ty) = splitForAllTys (dataConRepType data_con) in
1589         let (formal_arg_tys, _)     = splitFunTys formal_tau_ty in 
1590
1591        vcat [vcat (ilxMapPlaceArgs 0 pushArg env tm_args),
1592           sep [ text "newdata",
1593                 nest 2 (ilxTyConApp env tycon rep_ty_args <> comma),
1594                 nest 2 (ilxConRef env data_con)
1595           ]
1596         ]
1597  where
1598    tycon   = dataConTyCon data_con 
1599    rep_ty_args = map deepIlxRepType ty_args
1600    (ty_args,tm_args) = if isAlgTyCon tycon then splitTyArgs (tyConTyVars tycon) args  else splitTyArgs1 args
1601
1602 -- Split some type arguments off, throwing away the higher kinded ones for the moment.
1603 -- Base the higher-kinded checks off a corresponding list of formals.
1604 splitTyArgs :: [Var]            -- Formals
1605             -> [StgArg]         -- Actuals
1606             -> ([Type], [StgArg])
1607 splitTyArgs (htv:ttv) (StgTypeArg h:t) 
1608    | isIlxTyVar htv = ((h:l), r) 
1609    | otherwise = trace "splitTyArgs: threw away higher kinded type arg" (l, r) 
1610    where (l,r) = splitTyArgs ttv t 
1611 splitTyArgs _ l = ([],l)
1612  
1613 -- Split some type arguments off, where none should be higher kinded
1614 splitTyArgs1 :: [StgArg] -> ([Type], [StgArg])
1615 splitTyArgs1 (StgTypeArg ty : args) = (ty:tys, args')
1616                                     where
1617                                       (tys, args') = splitTyArgs1 args
1618 splitTyArgs1 args                   = ([], args)
1619
1620 ilxConRef env data_con
1621  | isUnboxedTupleCon data_con
1622     = let data_con' = tupleCon Unboxed (length non_void_args)in 
1623       pprId data_con' <> arg_text
1624  | otherwise 
1625     = pprId data_con <> arg_text
1626   where
1627     arg_text = pprValArgTys ilxTypeL env' (map deepIlxRepType non_void_args)
1628     non_void_args = filter (not . isVoidIlxRepType) arg_tys
1629     (tyvars, tau_ty) = splitForAllTys (dataConRepType data_con)
1630     (arg_tys, _)     = splitFunTys tau_ty
1631     env'             = formalIlxEnv env tyvars
1632
1633
1634
1635
1636 \end{code}
1637
1638
1639 %************************************************************************
1640 %*                                                                      *
1641 \subsection{PrimOps and Prim Representations}                           *
1642 %************************************************************************
1643
1644 \begin{code}
1645
1646 ilxPrimApp env op              args ret_ty = ilxPrimOpTable op args env
1647
1648
1649 type IlxTyFrag = IlxEnv -> SDoc
1650 ilxType s env = text s
1651
1652 ilxLift ty env = text "thunk" <> angleBrackets (ty env)
1653
1654 ilxTypeSeq :: [IlxTyFrag] -> IlxTyFrag
1655 ilxTypeSeq ops env = hsep (map (\x -> x env) ops)
1656
1657 tyPrimConTable :: UniqFM ([Type] -> IlxTyFrag)
1658 tyPrimConTable = 
1659   listToUFM [(addrPrimTyConKey,         (\_ -> repAddr)),
1660 --           (fileStreamPrimTyConKey,   (\_ -> repFileStream)),
1661              (foreignObjPrimTyConKey,   (\_ -> repForeign)),
1662              (stablePtrPrimTyConKey,    (\[ty] -> repStablePtr {- (ilxTypeL2 ty) -})),
1663              (stableNamePrimTyConKey,   (\[ty] -> repStableName {- (ilxTypeL2 ty) -} )),
1664              (charPrimTyConKey,         (\_ -> repChar)),
1665              (wordPrimTyConKey,         (\_ -> repWord)),
1666              (byteArrayPrimTyConKey,    (\_ -> repByteArray)),
1667              (intPrimTyConKey,          (\_ -> repInt)),
1668              (int64PrimTyConKey,        (\_ -> repInt64)),
1669              (word64PrimTyConKey,       (\_ -> repWord64)),
1670              (floatPrimTyConKey,        (\_ -> repFloat)),
1671              (doublePrimTyConKey,       (\_ -> repDouble)),
1672               -- These can all also accept unlifted parameter types so we explicitly lift.
1673              (arrayPrimTyConKey,        (\[ty] -> repArray (ilxTypeL2 ty))),
1674              (mutableArrayPrimTyConKey,         (\[_, ty] -> repMutArray (ilxTypeL2 ty))),
1675              (weakPrimTyConKey,         (\[ty] -> repWeak (ilxTypeL2 ty))),
1676              (mVarPrimTyConKey,         (\[_, ty] -> repMVar (ilxTypeL2 ty))),
1677              (mutVarPrimTyConKey,       (\[ty1, ty2] -> repMutVar (ilxTypeL2 ty1) (ilxTypeL2 ty2))),
1678              (mutableByteArrayPrimTyConKey,     (\_ -> repByteArray)),
1679              (threadIdPrimTyConKey,     (\_ -> repThread)),
1680              (bcoPrimTyConKey,  (\_ -> repBCO))
1681              ]
1682
1683 ilxTypeL2 :: Type -> IlxTyFrag
1684 ilxTypeL2 ty env = ilxTypeL env ty
1685 ilxTypeR2 :: Type -> IlxTyFrag
1686 ilxTypeR2 ty env = ilxTypeR env ty
1687
1688 ilxMethTyVarA = ilxType "!!0"
1689 ilxMethTyVarB = ilxType "!!1"
1690 prelGHCReference :: IlxTyFrag
1691 prelGHCReference env =
1692    if ilxEnvModule env == mkHomeModule (mkModuleName "PrelGHC") then empty
1693    else if inPrelude then moduleNameReference (mkModuleName "PrelGHC")
1694    else basePackageReference
1695
1696 prelBaseReference :: IlxTyFrag
1697 prelBaseReference env =
1698    if ilxEnvModule env == mkHomeModule (mkModuleName "PrelBase") then empty
1699    else if inPrelude then moduleNameReference (mkModuleName "PrelBase")
1700    else basePackageReference
1701
1702 repThread = ilxType "class [mscorlib]System.Threading.Thread /* ThreadId# */ "
1703 repByteArray = ilxType "unsigned int8[] /* ByteArr# */ "
1704 --repFileStream = text "void * /* FileStream# */ "  -- text "class [mscorlib]System.IO.FileStream"
1705 repInt = ilxType "int32"
1706 repWord = ilxType "unsigned int32"
1707 repAddr =ilxType "/* Addr */ void *"
1708 repInt64 = ilxType "int64"
1709 repWord64 = ilxType "unsigned int64"
1710 repFloat = ilxType "float32"
1711 repDouble = ilxType "float64"
1712 repChar = ilxType "/* Char */ unsigned int8"
1713 repForeign = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_Foreignzh"]
1714 repInteger = ilxUnboxedPairRep repInt repByteArray
1715 repIntegerPair = ilxUnboxedQuadRep repInt repByteArray repInt repByteArray
1716 repArray ty = ilxTypeSeq [ty,ilxType "[]"]
1717 repMutArray ty = ilxTypeSeq [ty,ilxType "[]"]
1718 repMVar ty = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_MVarzh",ilxTyParams [ty]]
1719 repMutVar _ ty2 = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_MutVarzh",ilxTyParams [ty2]]
1720 repWeak ty1 = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_Weakzh",ilxTyParams [ty1]]
1721 repStablePtr {- ty1 -} = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_StablePtrzh" {- ,ilxTyParams [ty1] -} ]
1722 repStableName {- ty1 -}  = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_StableNamezh" {- ,ilxTyParams [ty1] -} ]
1723 classWeak = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_Weakzh"]
1724 repBCO = ilxTypeSeq [ilxType "class ",prelGHCReference,ilxType "PrelGHC_BCOzh"]
1725
1726 ilxTyPair l r = ilxTyParams [l,r]
1727 ilxTyTriple l m r = ilxTyParams [l,m,r]
1728 ilxTyQuad l m1 m2 r = ilxTyParams [l,m1,m2,r]
1729 ilxUnboxedEmptyRep = ilxTypeSeq [ilxType "value class",prelGHCReference,ilxType "PrelGHC_Z1H"]
1730 ilxUnboxedPairRep l r = ilxTypeSeq [ilxType "value class",prelGHCReference,ilxType "PrelGHC_Z2H",ilxTyPair l r]
1731 ilxUnboxedTripleRep l m r = ilxTypeSeq [ilxType "value class",prelGHCReference,ilxType "PrelGHC_Z3H",ilxTyTriple l m r]
1732 ilxUnboxedQuadRep l m1 m2 r = ilxTypeSeq [ilxType "value class",prelGHCReference,ilxType "PrelGHC_Z4H",ilxTyQuad l m1 m2 r]
1733
1734 ilxTyIO b = ilxTypeSeq [ilxType "(func ( /* unit skipped */ ) --> ", b, ilxType ")"]
1735
1736 ilxTyParams :: [IlxTyFrag] -> IlxTyFrag
1737 ilxTyParams [] env = empty
1738 ilxTyParams l env = angleBrackets (ilxTyParamsAux l env)
1739   where
1740    ilxTyParamsAux [] env = empty
1741    ilxTyParamsAux [h] env = h env
1742    ilxTyParamsAux (h:t) env = h env <> text "," <+> ilxTyParamsAux t env
1743    ilxTyParams [] env = empty
1744
1745
1746 type IlxOpFrag = IlxEnv -> SDoc
1747 ilxOp :: String -> IlxOpFrag
1748 ilxOp s env = text s
1749 ilxOpSeq :: [IlxOpFrag] -> IlxOpFrag
1750 ilxOpSeq ops env = hsep (map (\x -> x env) ops)
1751
1752 ilxParams :: [IlxOpFrag] -> IlxOpFrag
1753 ilxParams l env = parens (ilxParamsAux l env)
1754   where
1755    ilxParamsAux [] env = empty
1756    ilxParamsAux [h] env = h env
1757    ilxParamsAux (h:t) env = h env <> text "," <+> ilxParamsAux t env
1758
1759
1760 ilxMethodRef rty cls nm tyargs args = 
1761     ilxOpSeq [rty,cls,ilxOp "::",ilxOp nm,
1762               ilxTyParams tyargs,ilxParams args]
1763
1764 ilxCall m = ilxOpSeq [ilxOp "call", m]
1765
1766 ilxSupportClass = ilxOpSeq [prelGHCReference, ilxOp "'GHC.support'"]
1767 ilxSuppMeth rty nm tyargs args = ilxMethodRef rty ilxSupportClass nm tyargs args
1768
1769 ilxCallSuppMeth rty nm tyargs args  = ilxCall (ilxSuppMeth rty nm tyargs args)
1770
1771 ilxMkBool :: IlxOpFrag
1772 ilxMkBool =  ilxOpSeq [ilxOp "call class",prelBaseReference,
1773                        ilxOp "PrelBase_Bool",
1774                        prelGHCReference,ilxOp "GHC.support::mkBool(bool)"]
1775 ilxCgt = ilxOpSeq [ilxOp "cgt",ilxMkBool]
1776 ilxCge = ilxOpSeq [ilxOp "clt ldc.i4 0 ceq ",ilxMkBool]
1777 ilxClt = ilxOpSeq [ilxOp "clt ",ilxMkBool]
1778 ilxCle = ilxOpSeq [ilxOp "cgt ldc.i4 0 ceq ",ilxMkBool]
1779 ilxCeq = ilxOpSeq [ilxOp "ceq ",ilxMkBool]
1780 ilxCne = ilxOpSeq [ilxOp "ceq ldc.i4 0 ceq " ,ilxMkBool]
1781 ilxCgtUn = ilxOpSeq [ilxOp "cgt.un ",ilxMkBool]
1782 ilxCgeUn  = ilxOpSeq [ilxOp "clt.un ldc.i4 0 ceq ",ilxMkBool]
1783 ilxCltUn = ilxOpSeq [ilxOp "clt.un ",ilxMkBool]
1784 ilxCleUn = ilxOpSeq [ilxOp "cgt.un ldc.i4 0 ceq ",ilxMkBool]
1785
1786 ilxAddrOfForeignOp = ilxOpSeq [ilxOp "ldfld void *" , repForeign, ilxOp "::contents"]
1787 ilxAddrOfByteArrOp = ilxOp "ldc.i4 0 ldelema unsigned int8"
1788
1789 ilxPrimOpTable :: PrimOp -> [StgArg] -> IlxOpFrag
1790 ilxPrimOpTable op
1791   = case op of
1792         CharGtOp    -> simp_op ilxCgt
1793         CharGeOp    -> simp_op ilxCge
1794         CharEqOp    -> simp_op ilxCeq
1795         CharNeOp    -> simp_op ilxCne
1796         CharLtOp    -> simp_op ilxClt
1797         CharLeOp    -> simp_op ilxCle
1798
1799         OrdOp       -> simp_op (ilxOp "conv.i4") -- chars represented by UInt32 (u4)
1800         ChrOp       -> simp_op (ilxOp "conv.u4")
1801
1802         IntGtOp     -> simp_op ilxCgt
1803         IntGeOp     -> simp_op ilxCge
1804         IntEqOp     -> simp_op ilxCeq
1805         IntNeOp     -> simp_op ilxCne
1806         IntLtOp     -> simp_op ilxClt
1807         IntLeOp     -> simp_op ilxCle
1808
1809         Narrow8IntOp   -> simp_op  (ilxOp"conv.i1")
1810         Narrow16IntOp  -> simp_op (ilxOp "conv.i2")
1811         Narrow32IntOp  -> simp_op (ilxOp "conv.i4")
1812         Narrow8WordOp  -> simp_op (ilxOp "conv.u1")
1813         Narrow16WordOp -> simp_op (ilxOp "conv.u2")
1814         Narrow32WordOp -> simp_op (ilxOp "conv.u4")
1815
1816         WordGtOp     -> simp_op ilxCgtUn
1817         WordGeOp     -> simp_op ilxCgeUn
1818         WordEqOp     -> simp_op ilxCeq
1819         WordNeOp     -> simp_op ilxCne
1820         WordLtOp     -> simp_op ilxCltUn
1821         WordLeOp     -> simp_op ilxCleUn
1822
1823         AddrGtOp     -> simp_op ilxCgt
1824         AddrGeOp     -> simp_op ilxCge
1825         AddrEqOp     -> simp_op ilxCeq
1826         AddrNeOp     -> simp_op ilxCne
1827         AddrLtOp     -> simp_op ilxClt
1828         AddrLeOp     -> simp_op ilxCle
1829
1830         FloatGtOp     -> simp_op ilxCgt
1831         FloatGeOp     -> simp_op ilxCge
1832         FloatEqOp     -> simp_op ilxCeq
1833         FloatNeOp     -> simp_op ilxCne
1834         FloatLtOp     -> simp_op ilxClt
1835         FloatLeOp     -> simp_op ilxCle
1836
1837         DoubleGtOp     -> simp_op ilxCgt
1838         DoubleGeOp     -> simp_op ilxCge
1839         DoubleEqOp     -> simp_op ilxCeq
1840         DoubleNeOp     -> simp_op ilxCne
1841         DoubleLtOp     -> simp_op ilxClt
1842         DoubleLeOp     -> simp_op ilxCle
1843
1844     -- Int#-related ops:
1845         IntAddOp    -> simp_op (ilxOp "add")
1846         IntSubOp    -> simp_op (ilxOp "sub")
1847         IntMulOp    -> simp_op (ilxOp "mul")
1848         IntQuotOp   -> simp_op (ilxOp "div")
1849         IntNegOp    -> simp_op (ilxOp "neg")
1850         IntRemOp    -> simp_op (ilxOp "rem")
1851
1852     -- Addr# ops:
1853         AddrAddOp  -> simp_op (ilxOp "add")
1854         AddrSubOp  -> simp_op (ilxOp "sub")
1855         AddrRemOp  -> simp_op (ilxOp "rem")
1856         Int2AddrOp -> warn_op "int2Addr" (simp_op (ilxOp "/* PrimOp int2Addr */ "))
1857         Addr2IntOp -> warn_op "addr2Int" (simp_op (ilxOp "/* PrimOp addr2Int */ "))
1858
1859     -- Word#-related ops:
1860         WordAddOp    -> simp_op (ilxOp "add")
1861         WordSubOp    -> simp_op (ilxOp "sub")
1862         WordMulOp    -> simp_op (ilxOp "mul")
1863         WordQuotOp   -> simp_op (ilxOp "div")
1864         WordRemOp    -> simp_op (ilxOp "rem")
1865
1866         ISllOp      -> simp_op (ilxOp "shl")
1867         ISraOp      -> simp_op (ilxOp "shr")
1868         ISrlOp      -> simp_op (ilxOp "shr.un")
1869         IntAddCOp   -> simp_op (ilxCallSuppMeth (ilxUnboxedPairRep repInt repInt) "IntAddCOp" [] [repInt, repInt])
1870         IntSubCOp   -> simp_op (ilxCallSuppMeth (ilxUnboxedPairRep repInt repInt) "IntSubCOp" [] [repInt, repInt])
1871         IntGcdOp    -> simp_op (ilxCallSuppMeth repInt "IntGcdOp" [] [repInt, repInt])
1872
1873
1874     -- Word#-related ops:
1875         AndOp       -> simp_op (ilxOp "and") 
1876         OrOp        -> simp_op (ilxOp "or") 
1877         NotOp       -> simp_op (ilxOp "not") 
1878         XorOp       -> simp_op (ilxOp "xor") 
1879         SllOp       -> simp_op (ilxOp "shl") 
1880         SrlOp       -> simp_op (ilxOp "shr") 
1881         Word2IntOp  -> simp_op (ilxOp "conv.i4")
1882         Int2WordOp  -> simp_op (ilxOp "conv.u4")
1883
1884     -- Float#-related ops:
1885         FloatAddOp   -> simp_op (ilxOp "add")
1886         FloatSubOp   -> simp_op (ilxOp "sub")
1887         FloatMulOp   -> simp_op (ilxOp "mul")
1888         FloatDivOp   -> simp_op (ilxOp "div")
1889         FloatNegOp   -> simp_op (ilxOp "neg")
1890         Float2IntOp  -> simp_op (ilxOp "conv.i4")
1891         Int2FloatOp  -> simp_op (ilxOp "conv.r4")
1892
1893         DoubleAddOp     -> simp_op (ilxOp "add")
1894         DoubleSubOp     -> simp_op (ilxOp "sub")
1895         DoubleMulOp     -> simp_op (ilxOp "mul")
1896         DoubleDivOp     -> simp_op (ilxOp "div")
1897         DoubleNegOp     -> simp_op (ilxOp "neg")
1898         Double2IntOp    -> simp_op (ilxOp "conv.i4")
1899         Int2DoubleOp    -> simp_op (ilxOp "conv.r4")
1900         Double2FloatOp  -> simp_op (ilxOp "conv.r4")
1901         Float2DoubleOp  -> simp_op (ilxOp "conv.r8")
1902         DoubleDecodeOp  -> simp_op (ilxCallSuppMeth (ilxUnboxedTripleRep repInt repInt repByteArray) "decodeDouble" [] [ilxType "float64"])
1903         FloatDecodeOp   -> simp_op (ilxCallSuppMeth (ilxUnboxedTripleRep repInt repInt repByteArray) "decodeFloat" [] [ilxType "float32"])
1904
1905         FloatExpOp   -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Exp(float64) conv.r4")
1906         FloatLogOp   -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Log(float64) conv.r4")
1907         FloatSqrtOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Sqrt(float64) conv.r4")
1908         FloatSinOp   -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Sin(float64) conv.r4")
1909         FloatCosOp   -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Cos(float64) conv.r4")
1910         FloatTanOp   -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Tan(float64) conv.r4")
1911         FloatAsinOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Asin(float64) conv.r4")
1912         FloatAcosOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Acos(float64) conv.r4")
1913         FloatAtanOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Atan(float64) conv.r4")
1914         FloatSinhOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Sinh(float64) conv.r4")
1915         FloatCoshOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Cosh(float64) conv.r4")
1916         FloatTanhOp  -> simp_op (ilxOp "conv.r8 call float64 [mscorlib]System.Math::Tanh(float64) conv.r4")
1917         FloatPowerOp -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Pow(float64, float64) conv.r4") --  ** op, make use of implicit cast to r8...
1918
1919         DoubleExpOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Exp(float64)")
1920         DoubleLogOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Log(float64)")
1921         DoubleSqrtOp  -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Sqrt(float64)")
1922           
1923         DoubleSinOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Sin(float64)")
1924         DoubleCosOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Cos(float64)")
1925         DoubleTanOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Tan(float64)")
1926           
1927         DoubleAsinOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Asin(float64)")
1928         DoubleAcosOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Acos(float64)")
1929         DoubleAtanOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Atan(float64)")
1930           
1931         DoubleSinhOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Sinh(float64)")
1932         DoubleCoshOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Cosh(float64)")
1933         DoubleTanhOp   -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Tanh(float64)")
1934           
1935         DoublePowerOp  -> simp_op (ilxOp "call float64 [mscorlib]System.Math::Pow(float64, float64)")
1936
1937     -- Integer (and related...) ops: bail out to support routines
1938         IntegerAndOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerAndOp" [] [repInt, repByteArray, repInt, repByteArray])
1939         IntegerOrOp        -> simp_op (ilxCallSuppMeth repInteger "IntegerOrOp" [] [repInt, repByteArray, repInt, repByteArray])
1940         IntegerXorOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerXorOp" [] [repInt, repByteArray, repInt, repByteArray])
1941         IntegerComplementOp -> simp_op (ilxCallSuppMeth repInteger "IntegerComplementOp" [] [repInt, repByteArray])
1942         IntegerAddOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerAddOp" [] [repInt, repByteArray, repInt, repByteArray])
1943         IntegerSubOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerSubOp" [] [repInt, repByteArray, repInt, repByteArray])
1944         IntegerMulOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerMulOp" [] [repInt, repByteArray, repInt, repByteArray])
1945         IntegerGcdOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerGcdOp" [] [repInt, repByteArray, repInt, repByteArray])
1946         IntegerQuotRemOp   -> simp_op (ilxCallSuppMeth repIntegerPair "IntegerQuotRemOp" [] [repInt, repByteArray, repInt, repByteArray])
1947         IntegerDivModOp    -> simp_op (ilxCallSuppMeth repIntegerPair "IntegerDivModOp" [] [repInt, repByteArray, repInt, repByteArray])
1948         IntegerIntGcdOp    -> simp_op (ilxCallSuppMeth repInt "IntegerIntGcdOp" [] [repInt, repByteArray, repInt])
1949         IntegerDivExactOp  -> simp_op (ilxCallSuppMeth repInteger "IntegerDivExactOp" [] [repInt, repByteArray, repInt, repByteArray])
1950         IntegerQuotOp      -> simp_op (ilxCallSuppMeth repInteger "IntegerQuotOp" [] [repInt, repByteArray, repInt, repByteArray])
1951         IntegerRemOp       -> simp_op (ilxCallSuppMeth repInteger "IntegerRemOp" [] [repInt, repByteArray, repInt, repByteArray])
1952         IntegerCmpOp       -> simp_op (ilxCallSuppMeth repInt "IntegerCmpOp" [] [repInt, repByteArray, repInt, repByteArray])
1953         IntegerCmpIntOp    -> simp_op (ilxCallSuppMeth repInt "IntegerCmpIntOp" [] [repInt, repByteArray, repInt])
1954         Integer2IntOp      -> simp_op (ilxCallSuppMeth repInt "Integer2IntOp" [] [repInt, repByteArray])
1955         Integer2WordOp     -> simp_op (ilxCallSuppMeth repWord "Integer2WordOp" [] [repInt, repByteArray])
1956         Int2IntegerOp      -> simp_op (ilxCallSuppMeth repInteger "Int2IntegerOp" [] [repInt])
1957         Word2IntegerOp     -> simp_op (ilxCallSuppMeth repInteger "Word2IntegerOp" [] [repWord])
1958 --      IntegerToInt64Op   -> simp_op (ilxCallSuppMeth repInt64 "IntegerToInt64Op" [] [repInt,repByteArray])
1959         Int64ToIntegerOp   -> simp_op (ilxCallSuppMeth repInteger "Int64ToIntegerOp" [] [repInt64])
1960 --      IntegerToWord64Op  -> simp_op (ilxCallSuppMeth repWord64 "IntegerToWord64Op" [] [repInt,repByteArray])
1961         Word64ToIntegerOp  -> simp_op (ilxCallSuppMeth repInteger "Word64ToIntegerOp" [] [repWord64])
1962
1963
1964
1965         IndexByteArrayOp_Char      -> simp_op (ilxOp "ldelem.u1")
1966         IndexByteArrayOp_WideChar  -> simp_op (ilxOp "ldelem.u4")
1967         IndexByteArrayOp_Int       -> simp_op (ilxOp "ldelem.i4")
1968         IndexByteArrayOp_Word      -> simp_op (ilxOp "ldelem.u4")
1969         IndexByteArrayOp_Addr      -> simp_op (ilxOp "ldelem.u")
1970         IndexByteArrayOp_Float     -> simp_op (ilxOp "ldelem.r4")
1971         IndexByteArrayOp_Double    -> simp_op (ilxOp "ldelem.r8")
1972         IndexByteArrayOp_StablePtr -> simp_op (ilxOp "ldelem.ref")
1973         IndexByteArrayOp_Int8     -> simp_op (ilxOp "ldelem.i1")
1974         IndexByteArrayOp_Int16     -> simp_op (ilxOp "ldelem.i2")
1975         IndexByteArrayOp_Int32     -> simp_op (ilxOp "ldelem.i4")
1976         IndexByteArrayOp_Int64     -> simp_op (ilxOp "ldelem.i8")
1977         IndexByteArrayOp_Word8    -> simp_op (ilxOp "ldelem.u1")
1978         IndexByteArrayOp_Word16    -> simp_op (ilxOp "ldelem.u2")
1979         IndexByteArrayOp_Word32    -> simp_op (ilxOp "ldelem.u4")
1980         IndexByteArrayOp_Word64    -> simp_op (ilxOp "ldelem.u8")
1981
1982             {- should be monadic??? -}
1983         ReadByteArrayOp_Char      -> simp_op (ilxOp "ldelem.u1")
1984         ReadByteArrayOp_WideChar  -> simp_op (ilxOp "ldelem.u4")
1985         ReadByteArrayOp_Int       -> simp_op (ilxOp "ldelem.i4")
1986         ReadByteArrayOp_Word      -> simp_op (ilxOp "ldelem.u4")
1987         ReadByteArrayOp_Addr      -> simp_op (ilxOp "ldelem.u")
1988         ReadByteArrayOp_Float     -> simp_op (ilxOp "ldelem.r4")
1989         ReadByteArrayOp_Double    -> simp_op (ilxOp "ldelem.r8")
1990         ReadByteArrayOp_StablePtr -> simp_op (ilxOp "ldelem.ref")
1991         ReadByteArrayOp_Int8     -> simp_op (ilxOp "ldelem.i1")
1992         ReadByteArrayOp_Int16     -> simp_op (ilxOp "ldelem.i2")
1993         ReadByteArrayOp_Int32     -> simp_op (ilxOp "ldelem.i4")
1994         ReadByteArrayOp_Int64     -> simp_op (ilxOp "ldelem.i8")
1995         ReadByteArrayOp_Word8    -> simp_op (ilxOp "ldelem.u1")
1996         ReadByteArrayOp_Word16    -> simp_op (ilxOp "ldelem.u2")
1997         ReadByteArrayOp_Word32    -> simp_op (ilxOp "ldelem.u4")
1998         ReadByteArrayOp_Word64    -> simp_op (ilxOp "ldelem.u8")
1999                  {-   MutByteArr# s -> Int# -> State# s -> (# State# s, Char# #) -}
2000                  {- ByteArr# -> Int# -> Char# -}
2001
2002
2003         WriteByteArrayOp_Char      -> simp_op (ilxOp "stelem.u1")
2004         WriteByteArrayOp_WideChar   -> simp_op (ilxOp "stelem.u4")
2005         WriteByteArrayOp_Int       -> simp_op (ilxOp "stelem.i4")
2006         WriteByteArrayOp_Word      -> simp_op (ilxOp "stelem.u4")
2007         WriteByteArrayOp_Addr      -> simp_op (ilxOp "stelem.u")
2008         WriteByteArrayOp_Float     -> simp_op (ilxOp "stelem.r4")
2009         WriteByteArrayOp_Double    -> simp_op (ilxOp "stelem.r8")
2010         WriteByteArrayOp_StablePtr -> simp_op (ilxOp "stelem.ref")
2011         WriteByteArrayOp_Int8     -> simp_op (ilxOp "stelem.i1")
2012         WriteByteArrayOp_Int16     -> simp_op (ilxOp "stelem.i2")
2013         WriteByteArrayOp_Int32     -> simp_op (ilxOp "stelem.i4")
2014         WriteByteArrayOp_Int64     -> simp_op (ilxOp "stelem.i8")
2015         WriteByteArrayOp_Word8    -> simp_op (ilxOp "stelem.u1")
2016         WriteByteArrayOp_Word16    -> simp_op (ilxOp "stelem.u2")
2017         WriteByteArrayOp_Word32    -> simp_op (ilxOp "stelem.u4")
2018         WriteByteArrayOp_Word64    -> simp_op (ilxOp "stelem.i8 /* nb. no stelem.u8 */")
2019                  {- MutByteArr# s -> Int# -> Char# -> State# s -> State# s -}
2020
2021         IndexOffAddrOp_Char    -> simp_op (ilxOp "sizeof unsigned int8 mul add ldind.u1")
2022         IndexOffAddrOp_WideChar    -> simp_op (ilxOp "sizeof int32 mul add ldind.u4")
2023         IndexOffAddrOp_Int     -> simp_op (ilxOp "sizeof int32 mul add ldind.i4")
2024         IndexOffAddrOp_Word    -> simp_op (ilxOp "sizeof int32 mul add ldind.u4")
2025         IndexOffAddrOp_Addr    -> simp_op (ilxOp "sizeof native unsigned int mul add ldind.i")
2026         IndexOffAddrOp_StablePtr   -> simp_op (ilxOp "sizeof native unsigned int mul add ldind.ref")
2027         IndexOffAddrOp_Float   -> simp_op (ilxOp "sizeof float32 mul add ldind.r4")
2028         IndexOffAddrOp_Double  -> simp_op (ilxOp "sizeof float64 mul add ldind.r8")
2029         IndexOffAddrOp_Int8   -> simp_op (ilxOp "sizeof int8 mul add ldind.i1")
2030         IndexOffAddrOp_Int16   -> simp_op (ilxOp "sizeof int16 mul add ldind.i2")
2031         IndexOffAddrOp_Int32   -> simp_op (ilxOp "sizeof int32 mul add ldind.i4")
2032         IndexOffAddrOp_Int64   -> simp_op (ilxOp "sizeof int64 mul add ldind.i8")
2033         IndexOffAddrOp_Word8  -> simp_op (ilxOp "sizeof unsigned int8 mul add ldind.u1")
2034         IndexOffAddrOp_Word16 -> simp_op (ilxOp "sizeof unsigned int16 mul add ldind.u2")
2035         IndexOffAddrOp_Word32  -> simp_op (ilxOp "sizeof unsigned int32 mul add ldind.u4")
2036         IndexOffAddrOp_Word64  -> simp_op (ilxOp "sizeof int64 mul add ldind.u8")
2037
2038         -- ForeignObj: load the address inside the object first
2039         -- TODO: is this remotely right?
2040         EqForeignObj                 -> warn_op "eqForeignObj" (simp_op (ilxOp "pop /* PrimOp eqForeignObj */ "))
2041         IndexOffForeignObjOp_Char    -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof unsigned int8 mul add ldind.u1"])
2042         IndexOffForeignObjOp_WideChar    -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof int32 mul add ldind.u4"])
2043         IndexOffForeignObjOp_Int     -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof int32 mul add ldind.i4"])
2044         IndexOffForeignObjOp_Word    -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof unsigned int32 mul add ldind.u4"])
2045         IndexOffForeignObjOp_Addr    ->  arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof native unsigned int mul add ldind.i  "])
2046         IndexOffForeignObjOp_StablePtr    ->  ty1_arg2_op (\ty fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof native unsigned int mul add ldind.ref  "])
2047         IndexOffForeignObjOp_Float   -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof float32 mul add ldind.r4"])
2048         IndexOffForeignObjOp_Double  -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof float64 mul add ldind.r8"])
2049         IndexOffForeignObjOp_Int8   -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof int8 mul add ldind.i1"])
2050         IndexOffForeignObjOp_Int16   -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof int16 mul add ldind.i2"])
2051         IndexOffForeignObjOp_Int32   -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof int32 mul add ldind.i4"])
2052         IndexOffForeignObjOp_Int64   -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof int64 mul add ldind.i8"])
2053         IndexOffForeignObjOp_Word8  -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof unsigned int8 mul add ldind.u1"])
2054         IndexOffForeignObjOp_Word16  -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof unsigned int16 mul add ldind.u2"])
2055         IndexOffForeignObjOp_Word32  -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof unsigned int32 mul add ldind.u4"])
2056         IndexOffForeignObjOp_Word64  -> arg2_op (\fobj n -> ilxOpSeq [fobj, ilxAddrOfForeignOp, n, ilxOp "sizeof unsigned int64 mul add ldind.u8"])
2057
2058         ReadOffAddrOp_Char   -> simp_op (ilxOp "sizeof unsigned int8 mul add ldind.u1")
2059         ReadOffAddrOp_WideChar -> simp_op (ilxOp "sizeof int32 mul add ldind.u4")
2060         ReadOffAddrOp_Int    -> simp_op (ilxOp "sizeof int32 mul add ldind.i4")
2061         ReadOffAddrOp_Word   -> simp_op (ilxOp "sizeof unsigned int32 mul add ldind.u4")
2062         ReadOffAddrOp_Addr   -> simp_op (ilxOp "sizeof native unsigned int mul add ldind.i")
2063         ReadOffAddrOp_Float  -> simp_op (ilxOp "sizeof float32 mul add ldind.r4")
2064         ReadOffAddrOp_Double -> simp_op (ilxOp "sizeof float64 mul add ldind.r8")
2065         ReadOffAddrOp_StablePtr  -> simp_op (ilxOp "sizeof native unsigned int mul add ldind.ref")
2066         ReadOffAddrOp_Int8  -> simp_op (ilxOp "sizeof int8 mul add ldind.i1")
2067         ReadOffAddrOp_Int16  -> simp_op (ilxOp "sizeof int16 mul add ldind.i2")
2068         ReadOffAddrOp_Int32  -> simp_op (ilxOp "sizeof int32 mul add ldind.i4")
2069         ReadOffAddrOp_Int64  -> simp_op (ilxOp "sizeof int64 mul add ldind.i8")
2070         ReadOffAddrOp_Word8 -> simp_op (ilxOp "sizeof unsigned int8 mul add ldind.u1")
2071         ReadOffAddrOp_Word16 -> simp_op (ilxOp "sizeof unsigned int16 mul add ldind.u2")
2072         ReadOffAddrOp_Word32 -> simp_op (ilxOp "sizeof unsigned int32 mul add ldind.u4")
2073         ReadOffAddrOp_Word64 -> simp_op (ilxOp "sizeof unsigned int64 mul add ldind.u8")
2074                   {-    Addr# -> Int# -> Char# -> State# s -> State# s -} 
2075
2076         WriteOffAddrOp_Char   -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "add", v, ilxOp "stind.u1"])
2077         WriteOffAddrOp_WideChar   -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof int32 mul add", v, ilxOp "stind.u4"])
2078         WriteOffAddrOp_Int    -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof int32 mul add", v, ilxOp "stind.i4"])
2079         WriteOffAddrOp_Word   -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof int32 mul add", v, ilxOp "stind.u4"])
2080         WriteOffAddrOp_Addr   -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof native unsigned int mul add", v, ilxOp "stind.i"])
2081         WriteOffAddrOp_ForeignObj   -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof native unsigned int mul add", v, ilxOp "stind.ref"])
2082         WriteOffAddrOp_Float  -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof float32 mul add", v,ilxOp "stind.r4"])
2083         WriteOffAddrOp_StablePtr   -> ty2_arg4_op (\ty1 sty addr n v s -> ilxOpSeq [addr, n, ilxOp "sizeof native unsigned int mul add", v, ilxOp "stind.ref"])
2084         WriteOffAddrOp_Double -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof float64 mul add",v,ilxOp "stind.r8"])
2085         WriteOffAddrOp_Int8  -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof int8 mul add",v,ilxOp "stind.i1"])
2086         WriteOffAddrOp_Int16  -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof int16 mul add",v,ilxOp "stind.i2"])
2087         WriteOffAddrOp_Int32  -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof int32 mul add",v,ilxOp "stind.i4"])
2088         WriteOffAddrOp_Int64  -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof int64 mul add",v,ilxOp "stind.i8"])
2089         WriteOffAddrOp_Word8 -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof unsigned int8 mul add",v,ilxOp "stind.u1"])
2090         WriteOffAddrOp_Word16 -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof unsigned int16 mul add",v,ilxOp "stind.u2"])
2091         WriteOffAddrOp_Word32 -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof unsigned int32 mul add",v,ilxOp "stind.u4"])
2092         WriteOffAddrOp_Word64 -> ty1_arg4_op (\sty addr n v s -> ilxOpSeq [addr,n,ilxOp "sizeof unsigned int64 mul add",v,ilxOp "stind.u8"])
2093                   {-    Addr# -> Int# -> Char# -> State# s -> State# s -} 
2094
2095             {- should be monadic??? -}
2096         NewPinnedByteArrayOp_Char -> warn_op "newPinnedByteArray" (simp_op (ilxOp "newarr [mscorlib]System.Byte "))
2097         NewByteArrayOp_Char      -> simp_op (ilxOp "newarr [mscorlib]System.Byte")
2098 --      NewByteArrayOp_Int       -> simp_op (ilxOp "newarr [mscorlib]System.Int32")
2099 --      NewByteArrayOp_Word      -> simp_op (ilxOp "newarr [mscorlib]System.UInt32")
2100 --      NewByteArrayOp_Addr      -> simp_op (ilxOp "newarr [mscorlib]System.UInt64")
2101 --      NewByteArrayOp_Float     -> simp_op (ilxOp "newarr [mscorlib]System.Single")
2102 --      NewByteArrayOp_Double    -> simp_op (ilxOp "newarr [mscorlib]System.Double")
2103 --      NewByteArrayOp_StablePtr -> simp_op (ilxOp "newarr [mscorlib]System.UInt32")
2104 --      NewByteArrayOp_Int64     -> simp_op (ilxOp "newarr [mscorlib]System.Int64")  TODO: there is no unique for this one -}
2105 --      NewByteArrayOp_Word64    -> simp_op (ilxOp "newarr  [mscorlib]System.UInt64") -}
2106                   {- Int# -> State# s -> (# State# s, MutByteArr# s #) -}
2107         ByteArrayContents_Char   -> warn_op "byteArrayContents" (simp_op ilxAddrOfByteArrOp)
2108
2109         UnsafeFreezeByteArrayOp ->   ty1_op (\ty1  -> ilxOp "nop ")
2110                   {- MutByteArr# s -> State# s -> (# State# s, ByteArr# #) -}
2111         SizeofByteArrayOp  -> simp_op (ilxOp "ldlen")
2112                   {- ByteArr# -> Int# -}
2113
2114         SameMutableByteArrayOp -> ty1_op (\ty1  -> ilxCeq)
2115                  {- MutByteArr# s -> MutByteArr# s -> Bool -}
2116         SizeofMutableByteArrayOp -> ty1_op (\ty1  -> ilxOp "ldlen")
2117                  {- MutByteArr# s -> Int# -}
2118
2119         SameMutVarOp -> ty2_op (\ty1 ty2 -> ilxCeq)
2120                  {- MutVar# s a -> MutVar# s a -> Bool -}
2121         NewMutVarOp -> ty2_op (\ty1 ty2 -> ilxOpSeq [ilxOp "newobj void" , repMutVar ty1 ty2 , ilxOp "::.ctor(!0)"])
2122                  {- a -> State# s -> (# State# s, MutVar# s a #) -}
2123         ReadMutVarOp -> ty2_op (\ty1 ty2 ->  ilxOpSeq [ilxOp "ldfld !0" , repMutVar ty1 ty2 , ilxOp "::contents"])
2124                  {-  MutVar# s a -> State# s -> (# State# s, a #) -}
2125         WriteMutVarOp -> ty2_op (\ty1 ty2 -> ilxOpSeq [ilxOp "stfld !0" , repMutVar ty1 ty2 , ilxOp "::contents"])
2126                  {- MutVar# s a -> a -> State# s -> State# s -}
2127
2128         NewArrayOp -> ty2_op (\ty1 ty2 -> ilxCallSuppMeth (ilxType "!!0[]") "newArray" [ty1] [repInt,ilxMethTyVarA])
2129                  {- Int# -> a -> State# s -> (# State# s, MutArr# s a #) -}
2130         IndexArrayOp -> ty1_op (\ty1 -> ilxOp "ldelem.ref")
2131                  {- Array# a -> Int# -> (# a #) -}
2132         WriteArrayOp -> ty2_op (\ty1 ty2 -> ilxOp "stelem.ref")
2133                  {- MutArr# s a -> Int# -> a -> State# s -> State# s -}
2134         ReadArrayOp -> ty2_op (\ty1 ty2 -> ilxOp "ldelem.ref")
2135                  {- MutArr# s a -> Int# -> State# s -> (# State# s, a #) -}
2136         UnsafeFreezeArrayOp -> ty2_op (\ty1 ty2 -> ilxOp "nop")
2137                  {-   MutArr# s a -> State# s -> (# State# s, Array# a #) -}
2138         UnsafeThawArrayOp -> ty2_op (\ty1 ty2 -> ilxOp "nop")
2139                  {-  Array# a -> State# s -> (# State# s, MutArr# s a #) -}
2140
2141         SameMutableArrayOp -> ty2_op (\ty1 ty2 -> ilxCeq)
2142                  {- MutArr# s a -> MutArr# s a -> Bool -}
2143
2144
2145         RaiseOp -> ty2_op (\ty1 ty2 -> ilxOp "throw")
2146         CatchOp -> ty2_op (\ty1 ty2 -> 
2147                 ilxCallSuppMeth ilxMethTyVarA "'catch'" [ty1,ty2] [ilxLift (ilxTyIO (ilxType "!!0")), 
2148                                                               ilxOp "thunk<(func (!!1) --> (func ( /* unit skipped */ ) --> !!0))>"])
2149                             {-        (State# RealWorld -> (# State# RealWorld, a #) )
2150                                    -> (b -> State# RealWorld -> (# State# RealWorld, a #) ) 
2151                                    -> State# RealWorld
2152                                    -> (# State# RealWorld, a #) 
2153                              -} 
2154
2155         BlockAsyncExceptionsOp -> ty1_op (\ty1 -> 
2156                 ilxCallSuppMeth ilxMethTyVarA "blockAsyncExceptions" [ty1] [ilxLift (ilxTyIO (ilxType "!!0"))])
2157
2158                 {-     (State# RealWorld -> (# State# RealWorld, a #))
2159                     -> (State# RealWorld -> (# State# RealWorld, a #))
2160                 -}
2161
2162         UnblockAsyncExceptionsOp -> ty1_op (\ty1 -> 
2163                 ilxCallSuppMeth ilxMethTyVarA "unblockAsyncExceptions" [ty1] [ilxLift (ilxTyIO (ilxType "!!0"))])
2164
2165                 {-
2166                     State# RealWorld -> (# State# RealWorld, a #))
2167                     -> (State# RealWorld -> (# State# RealWorld, a #))
2168                 -}
2169  
2170         NewMVarOp -> ty2_op (\sty ty -> 
2171                 ilxOpSeq [ilxOp "newobj void " , repMVar ty , ilxOp "::.ctor()"])
2172                  {- State# s -> (# State# s, MVar# s a #) -}
2173
2174         TakeMVarOp -> ty2_op (\sty ty -> 
2175                 ilxCallSuppMeth ilxMethTyVarA "takeMVar" [ty] [repMVar ilxMethTyVarA])
2176                   {-  MVar# s a -> State# s -> (# State# s, a #) -}
2177
2178         -- These aren't yet right
2179         TryTakeMVarOp -> ty2_op (\sty ty -> 
2180                 ilxCallSuppMeth (ilxUnboxedPairRep repInt ilxMethTyVarA) "tryTakeMVar" [ty] [repMVar ilxMethTyVarA])
2181                   {-  MVar# s a -> State# s -> (# State# s, a #) -}
2182
2183         TryPutMVarOp -> ty2_op (\sty ty -> 
2184                 ilxCallSuppMeth repInt "tryPutMVar" [ty] [repMVar ilxMethTyVarA,ilxMethTyVarA])
2185                   {-  MVar# s a -> State# s -> (# State# s, a #) -}
2186
2187         PutMVarOp -> ty2_op (\sty ty -> 
2188                 ilxCallSuppMeth (ilxOp "void") "putMVar" [ty] [repMVar ilxMethTyVarA, ilxMethTyVarA])
2189                    {- MVar# s a -> a -> State# s -> State# s -}
2190
2191         SameMVarOp -> ty2_op (\sty ty -> ilxCeq)
2192                    {- MVar# s a -> MVar# s a -> Bool -}
2193
2194 --      TakeMaybeMVarOp -> ty2_op (\sty ty -> 
2195 --              (ilxCallSuppMeth (ilxUnboxedPairRep repInt ilxMethTyVarA) "tryTakeMVar" [ty] [repMVar ilxMethTyVarA]))
2196 --              {- MVar# s a -> State# s -> (# State# s, Int#, a #) -}
2197
2198         IsEmptyMVarOp -> ty2_op (\sty ty -> 
2199                 ilxCallSuppMeth repInt "isEmptyMVar" [ty] [repMVar ilxMethTyVarA])
2200                {- MVar# s a -> State# s -> (# State# s, Int# #) -}
2201
2202         TouchOp -> warn_op "touch" (ty1_op (\ty1 -> ilxOp "pop /* PrimOp touch */ "))
2203
2204                {- a -> Int# -}
2205         DataToTagOp -> ty1_op (\ty1 -> 
2206                 ilxCallSuppMeth repInt "dataToTag" [ty1] [ilxMethTyVarA])
2207                {- a -> Int# -}
2208
2209         TagToEnumOp -> ty1_op (\ty1 -> 
2210                 ilxCallSuppMeth ilxMethTyVarA "tagToEnum" [ty1] [repInt])
2211                {- Int# -> a -}
2212
2213         MakeStablePtrOp -> ty1_op (\ty1 -> ilxOpSeq [ilxOp "box", ty1, ilxOp "newobj void", repStablePtr {- ty1 -}, ilxOp "::.ctor(class [mscorlib]System.Object)"])
2214                  {-   a -> State# RealWorld -> (# State# RealWorld, StablePtr# a #) -}
2215         MakeStableNameOp -> ty1_op (\ty1 -> ilxOpSeq [ilxOp "pop newobj void", repStableName {- ty1 -}, ilxOp "::.ctor()"])
2216                         -- primOpInfo MakeStableNameOp = mkGenPrimOp SLIT("makeStableName#")  [alphaTyVar] [alphaTy, mkStatePrimTy realWorldTy] ((mkTupleTy Unboxed 2 [mkStatePrimTy realWorldTy, mkStableNamePrimTy alphaTy]))
2217
2218         EqStableNameOp -> ty1_op (\ty1 -> ilxOp "ceq")
2219                -- [alphaTyVar] [mkStableNamePrimTy alphaTy, mkStableNamePrimTy alphaTy] (intPrimTy)
2220         StableNameToIntOp -> warn_op "StableNameToIntOp" (ty1_op (\ty1 -> ilxOp "pop ldc.i4 0"))
2221                -- [alphaTyVar] [mkStableNamePrimTy alphaTy] (intPrimTy)
2222
2223         DeRefStablePtrOp -> ty1_op (\ty1 -> ilxOpSeq [ilxOp "ldfld class [mscorlib]System.Object", repStablePtr {- ty1 -}, ilxOp "::contents"])
2224                  {-  StablePtr# a -> State# RealWorld -> (# State# RealWorld, a #) -}
2225
2226         EqStablePtrOp -> ty1_op (\ty1 -> ilxOp "ceq")
2227                  {-  StablePtr# a -> StablePtr# a -> Int# -}
2228
2229             -- The 3rd argument to MkWeakOp is always a IO Monad action, i.e. passed as () --> ()
2230         MkWeakOp -> ty3_op (\ty1 ty2 ty3 ->  ilxCall (ilxMethodRef (repWeak ilxMethTyVarB) classWeak "bake" [ilxLift ty1,ilxLift ty2] [ilxMethTyVarA, ilxMethTyVarB, ilxLift (ilxTyIO ilxUnboxedEmptyRep)]))
2231                  {- o -> b -> c -> State# RealWorld -> (# State# RealWorld, Weak# b #) -}
2232
2233         DeRefWeakOp -> ty1_op (\ty1 ->  ilxCall (ilxMethodRef (ilxUnboxedPairRep repInt ilxMethTyVarA) classWeak "deref" [ty1] [repWeak ilxMethTyVarA]))
2234         FinalizeWeakOp -> ty1_op (\ty1 ->  ilxCall (ilxMethodRef (ilxUnboxedPairRep repInt (ilxTyIO ilxUnboxedEmptyRep)) classWeak "finalizer" [ty1] [repWeak ilxMethTyVarA]))
2235                    {-    Weak# a -> State# RealWorld -> (# State# RealWorld, Int#, 
2236         State# RealWorld -> (# State# RealWorld, Unit #)) #) -}
2237
2238         MkForeignObjOp -> simp_op (ilxOpSeq [ilxOp "newobj void", repForeign, ilxOp "::.ctor(void *)"])
2239         WriteForeignObjOp -> ty1_op (\sty -> ilxOpSeq [ilxOp "stfld void *", repForeign, ilxOp "::contents"])
2240         ForeignObjToAddrOp -> simp_op ilxAddrOfForeignOp
2241         YieldOp -> simp_op (ilxOpSeq [ilxOp "call class [mscorlib]System.Threading.Thread class [mscorlib]System.Threading.Thread::get_CurrentThread() 
2242                                 call instance void class [mscorlib]System.Threading.Thread::Suspend()"])
2243         MyThreadIdOp -> simp_op (ilxOpSeq [ilxOp "call default  class [mscorlib]System.Threading.Thread class [mscorlib]System.Threading.Thread::get_CurrentThread() "])
2244         -- This pushes a THUNK across as the exception value.
2245         -- This is the correct Haskell semantics...  TODO: we should probably
2246         -- push across an HaskellThreadAbortException object that wraps this
2247         -- thunk, but which is still actually an exception of
2248         -- an appropriate type.
2249         KillThreadOp -> ty1_op (\ty -> ilxOpSeq [ilxOp "call instance void class [mscorlib]System.Threading.Thread::Abort(class [mscorlib]System.Object) "])
2250               {-   ThreadId# -> a -> State# RealWorld -> State# RealWorld -}
2251
2252         ForkOp -> warn_op "ForkOp" (simp_op (ilxOp "/* ForkOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2253         ParOp ->  warn_op "ParOp" (simp_op (ilxOp "/* ParOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2254         DelayOp -> simp_op (ilxOp "call void class [mscorlib]System.Threading.Thread::Sleep(int32) ")
2255                  {-    Int# -> State# s -> State# s -}
2256
2257         WaitReadOp  -> warn_op "WaitReadOp" (simp_op (ilxOp "/* WaitReadOp skipped... */ pop"))
2258         WaitWriteOp -> warn_op "WaitWriteOp" (simp_op (ilxOp " /* WaitWriteOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2259         ParAtForNowOp -> warn_op "ParAtForNowOp" (simp_op (ilxOp " /* ParAtForNowOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2260         ParAtRelOp -> warn_op "ParAtRelOp" (simp_op (ilxOp " /* ParAtRelOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2261         ParAtAbsOp -> warn_op "ParAtAbsOp" (simp_op (ilxOp " /* ParAtAbsOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2262         ParAtOp -> warn_op "ParAtOp" (simp_op (ilxOp " /* ParAtOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2263         ParLocalOp -> warn_op "ParLocalOp" (simp_op (ilxOp " /* ParLocalOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2264         ParGlobalOp -> warn_op "ParGlobalOp" (simp_op (ilxOp " /* ParGlobalOp skipped... */ newobj void [mscorlib]System.Object::.ctor() throw"))
2265         SeqOp -> warn_op "SeqOp" (simp_op (ilxOp " newobj void [mscorlib]System.Object::.ctor() throw "))
2266         AddrToHValueOp -> warn_op "AddrToHValueOp" (simp_op (ilxOp "newobj void [mscorlib]System.Object::.ctor() throw"))
2267 --      ReallyUnsafePtrEqualityOp -> simp_op (ilxOp "ceq")
2268
2269         MkApUpd0_Op ->  warn_op "MkApUpd0_Op" (simp_op (ilxOp " newobj void [mscorlib]System.Object::.ctor() throw"))
2270         NewBCOOp ->  warn_op "NewBCOOp" (simp_op (ilxOp " newobj void [mscorlib]System.Object::.ctor() throw"))
2271                   -- ("newBCO#")  [alphaTyVar, deltaTyVar] [byteArrayPrimTy, byteArrayPrimTy, mkArrayPrimTy alphaTy, byteArrayPrimTy, mkStatePrimTy deltaTy] ((mkTupleTy Unboxed 2 [mkStatePrimTy deltaTy, bcoPrimTy]))
2272         _        -> pprPanic "Unimplemented primop" (ppr op)
2273
2274
2275 ty1_op :: (IlxTyFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2276 ty1_op  op ((StgTypeArg ty1):rest)  = 
2277       ilxOpSeq [getArgsStartingAt 1 rest, 
2278                 op (ilxTypeR2 (deepIlxRepType ty1))]
2279
2280 ty2_op :: (IlxTyFrag -> IlxTyFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2281 ty2_op  op ((StgTypeArg ty1):(StgTypeArg ty2):rest)  = 
2282       ilxOpSeq [getArgsStartingAt 2 rest, 
2283                 op (ilxTypeR2 (deepIlxRepType ty1)) 
2284                    (ilxTypeR2 (deepIlxRepType ty2))]
2285
2286 ty3_op :: (IlxTyFrag -> IlxTyFrag -> IlxTyFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2287 ty3_op  op ((StgTypeArg ty1):(StgTypeArg ty2):(StgTypeArg ty3):rest) = 
2288       ilxOpSeq [getArgsStartingAt 3 rest, 
2289                 op (ilxTypeR2 (deepIlxRepType ty1)) 
2290                    (ilxTypeR2 (deepIlxRepType ty2))
2291                    (ilxTypeR2 (deepIlxRepType ty3))]
2292
2293 arg2_op :: (IlxTyFrag -> IlxOpFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2294 arg2_op  op [a1, a2] = 
2295        op (getAsArg 1 a1)
2296           (getAsArg 2 a2)
2297
2298 ty1_arg2_op :: (IlxTyFrag -> IlxOpFrag ->  IlxOpFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2299 ty1_arg2_op  op [(StgTypeArg ty1), a1, a2] = 
2300        op (ilxTypeR2 (deepIlxRepType ty1)) 
2301           (getAsArg 1 a1)
2302           (getAsArg 2 a2)
2303
2304 ty1_arg4_op :: (IlxTyFrag -> IlxOpFrag -> IlxOpFrag -> IlxOpFrag -> IlxOpFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2305 ty1_arg4_op  op [(StgTypeArg ty1), a1, a2, a3, a4] = 
2306        op (ilxTypeR2 (deepIlxRepType ty1)) 
2307           (getAsArg 1 a1)
2308           (getAsArg 2 a2)
2309           (getAsArg 3 a3)
2310           (getAsArg 4 a4)
2311
2312 ty2_arg4_op :: (IlxTyFrag -> IlxTyFrag -> IlxOpFrag -> IlxOpFrag -> IlxOpFrag -> IlxOpFrag -> IlxOpFrag) -> [StgArg] ->  IlxOpFrag 
2313 ty2_arg4_op  op [(StgTypeArg ty1), (StgTypeArg ty2),a1, a2, a3, a4] = 
2314        op (ilxTypeR2 (deepIlxRepType ty1)) 
2315           (ilxTypeR2 (deepIlxRepType ty2)) 
2316           (getAsArg 2 a1)
2317           (getAsArg 3 a2)
2318           (getAsArg 4 a3)
2319           (getAsArg 5 a4)
2320
2321 hd (h:t) = h
2322
2323 getAsArg n a env = hd (ilxMapPlaceArgs n pushArg env [a])
2324 getArgsStartingAt n a env = vcat (ilxMapPlaceArgs n pushArg env a)
2325
2326 simp_op :: IlxOpFrag -> [StgArg] -> IlxOpFrag
2327 simp_op  op args env    = vcat (ilxMapPlaceArgs 0 pushArg env args) $$ op env
2328 warn_op  warning f args = trace ("WARNING! IlxGen cannot translate primop " ++ warning) (f args)
2329 \end{code}
2330
2331 %************************************************************************
2332 %*                                                                      *
2333 \subsection{C Calls}
2334 %*                                                                      *
2335 %************************************************************************
2336
2337 \begin{code}
2338 -- Call the P/Invoke stub wrapper generated in the import section.
2339 -- We eliminate voids in and around an IL C Call.  
2340 -- We also do some type-directed translation for pinning Haskell-managed blobs
2341 -- of data as we throw them across the boundary.
2342 ilxFCall env (CCall (CCallSpec (StaticTarget c) cconv gc)) args ret_ty
2343  = ilxComment ((text "C call") <+> pprCLabelString c) <+> 
2344         vcat [vcat (ilxMapPlaceArgs 0 pushCArg env args),
2345               text "call" <+> retdoc <+> pprCLabelString c <+> tyarg_doc
2346                     <+> pprCValArgTys ilxTypeL env (map deepIlxRepType (filter (not. isVoidIlxRepType) (map stgArgType tm_args))) ]
2347   where 
2348     retdoc | isVoidIlxRepType ret_ty = text "void"
2349            | otherwise               = ilxTypeR env (deepIlxRepType ret_ty)
2350     (ty_args,tm_args) = splitTyArgs1 args
2351     tyarg_doc | not (isEmptyVarSet (tyVarsOfTypes ty_args)) = text "/* type variable found */"
2352               | otherwise = pprTypeArgs ilxTypeR env ty_args
2353
2354 ilxFCall env (DNCall (DNCallSpec call_instr)) args ret_ty
2355   = ilxComment (text "IL call") <+> 
2356     vcat [vcat (ilxMapPlaceArgs 0 pushEvalArg env tm_args), 
2357           ptext call_instr
2358                 -- In due course we'll need to pass the type arguments
2359                 -- and to do that we'll need to have more than just a string
2360                 -- for call_instr
2361     ]
2362   where
2363     (ty_args,tm_args) = splitTyArgs1 args 
2364
2365 -- Push and argument and force its evaluation if necessary.
2366 pushEvalArg _ (StgTypeArg _) = empty
2367 pushEvalArg env (StgVarArg arg) = ilxFunApp env arg [] False
2368 pushEvalArg env (StgLitArg lit) = pushLit env lit
2369
2370
2371 hasTyCon (TyConApp tc _) tc2 = tc == tc2
2372 hasTyCon _  _ = False
2373
2374 isByteArrayCArgTy ty = hasTyCon ty byteArrayPrimTyCon || hasTyCon ty mutableByteArrayPrimTyCon
2375 isByteArrayCArg v = isByteArrayCArgTy (deepIlxRepType (idType v))
2376
2377 isForeignObjCArgTy ty = hasTyCon ty foreignObjPrimTyCon
2378 isForeignObjCArg v = isForeignObjCArgTy (deepIlxRepType (idType v))
2379
2380 pinCCallArg v = isByteArrayCArg v || isForeignObjCArg v  
2381
2382 pinCArg  env arg v = pushArg env arg <+> text "dup stloc" <+> singleQuotes (ilxEnvQualifyByExact env (ppr v) <> text "pin") 
2383 pushCArg  env arg@(StgVarArg v) | isByteArrayCArg v = pinCArg env arg v <+> ilxAddrOfByteArrOp env
2384 pushCArg env arg@(StgVarArg v) | isForeignObjCArg v = pinCArg env arg v <+> ilxAddrOfForeignOp env
2385 pushCArg env arg | otherwise = pushArg env arg
2386
2387 pprCValArgTys f env tys = parens (pprSepWithCommas (pprCValArgTy f env) tys)
2388 pprCValArgTy f env ty | isByteArrayCArgTy ty = text "void *" <+> ilxComment (text "interior pointer into ByteArr#")
2389 pprCValArgTy f env ty | isForeignObjCArgTy ty = text "void *" <+> ilxComment (text "foreign object")
2390 pprCValArgTy f env ty | otherwise = f env ty
2391
2392
2393 foldR            :: (a -> b -> b) -> [a] -> b -> b
2394 -- foldR _ [] z     =  z
2395 -- foldR f (x:xs) z =  f x (foldR f xs z) 
2396 {-# INLINE foldR #-}
2397 foldR k xs z = go xs
2398              where
2399                go []     = z
2400                go (y:ys) = y `k` go ys
2401
2402 \end{code}
2403