2 % (c) The University of Glasgow, 2006
4 \section[HscTypes]{Types for the per-module compiler}
8 -- * Sessions and compilation state
9 Session(..), HscEnv(..), hscEPS,
10 FinderCache, FindResult(..), ModLocationCache,
11 Target(..), TargetId(..), pprTarget, pprTargetId,
14 ModDetails(..), emptyModDetails,
15 ModGuts(..), CgGuts(..), ModImports(..), ForeignStubs(..),
17 ModSummary(..), showModMsg, isBootSummary,
18 msHsFilePath, msHiFilePath, msObjFilePath,
20 HscSource(..), isHsBoot, hscSourceString, -- Re-exported from DriverPhases
22 HomePackageTable, HomeModInfo(..), emptyHomePackageTable,
23 hptInstances, hptRules,
25 ExternalPackageState(..), EpsStats(..), addEpsInStats,
26 PackageTypeEnv, PackageIfaceTable, emptyPackageIfaceTable,
27 lookupIfaceByModule, emptyModIface,
29 InteractiveContext(..), emptyInteractiveContext,
30 icPrintUnqual, mkPrintUnqualified,
32 ModIface(..), mkIfaceDepCache, mkIfaceVerCache, mkIfaceFixCache,
35 Deprecs(..), IfaceDeprecs,
37 FixityEnv, FixItem(..), lookupFixity, emptyFixityEnv,
39 implicitTyThings, isImplicitTyThing,
41 TyThing(..), tyThingClass, tyThingTyCon, tyThingDataCon, tyThingId,
42 TypeEnv, lookupType, mkTypeEnv, emptyTypeEnv,
43 extendTypeEnv, extendTypeEnvList, extendTypeEnvWithIds, lookupTypeEnv,
44 typeEnvElts, typeEnvClasses, typeEnvTyCons, typeEnvIds,
47 WhetherHasOrphans, IsBootInterface, Usage(..),
48 Dependencies(..), noDependencies,
49 NameCache(..), OrigNameCache, OrigIParamCache,
50 Avails, availsToNameSet, availsToNameEnv, availName, availNames,
51 GenAvailInfo(..), AvailInfo, RdrAvailInfo,
54 Deprecations, DeprecTxt, plusDeprecs,
56 PackageInstEnv, PackageRuleBase,
59 Linkable(..), isObjectLinkable,
60 Unlinked(..), CompiledByteCode,
61 isObject, nameOfObject, isInterpretable, byteCodeOfObject,
65 #include "HsVersions.h"
67 import Breakpoints ( SiteNumber, Coord, noDbgSites )
69 import ByteCodeAsm ( CompiledByteCode )
72 import RdrName ( GlobalRdrEnv, emptyGlobalRdrEnv,
73 LocalRdrEnv, emptyLocalRdrEnv, GlobalRdrElt(..),
74 unQualOK, ImpDeclSpec(..), Provenance(..),
75 ImportSpec(..), lookupGlobalRdrEnv )
76 import Name ( Name, NamedThing, getName, nameOccName, nameModule )
79 import OccName ( OccName, OccEnv, lookupOccEnv, mkOccEnv, emptyOccEnv,
82 import InstEnv ( InstEnv, Instance )
83 import FamInstEnv ( FamInstEnv, FamInst )
84 import Rules ( RuleBase )
85 import CoreSyn ( CoreBind )
86 import Id ( Id, isImplicitId )
87 import Type ( TyThing(..) )
89 import Class ( Class, classSelIds, classATs, classTyCon )
91 import DataCon ( DataCon, dataConImplicitIds )
92 import PrelNames ( gHC_PRIM )
93 import Packages ( PackageId )
94 import DynFlags ( DynFlags(..), DynFlag(..), isOneShot, HscTarget (..) )
95 import DriverPhases ( HscSource(..), isHsBoot, hscSourceString, Phase )
96 import BasicTypes ( Version, initialVersion, IPName,
97 Fixity, defaultFixity, DeprecTxt )
99 import FiniteMap ( FiniteMap )
100 import CoreSyn ( CoreRule )
101 import Maybes ( orElse, expectJust, catMaybes, seqMaybe )
103 import SrcLoc ( SrcSpan, Located )
104 import UniqFM ( lookupUFM, eltsUFM, emptyUFM )
105 import UniqSupply ( UniqSupply )
106 import FastString ( FastString )
108 import StringBuffer ( StringBuffer )
110 import System.Time ( ClockTime )
111 import Data.IORef ( IORef, readIORef )
115 %************************************************************************
117 \subsection{Compilation environment}
119 %************************************************************************
123 -- | The Session is a handle to the complete state of a compilation
124 -- session. A compilation session consists of a set of modules
125 -- constituting the current program or library, the context for
126 -- interactive evaluation, and various caches.
127 newtype Session = Session (IORef HscEnv)
130 HscEnv is like Session, except that some of the fields are immutable.
131 An HscEnv is used to compile a single module from plain Haskell source
132 code (after preprocessing) to either C, assembly or C--. Things like
133 the module graph don't change during a single compilation.
135 Historical note: "hsc" used to be the name of the compiler binary,
136 when there was a separate driver and compiler. To compile a single
137 module, the driver would invoke hsc on the source code... so nowadays
138 we think of hsc as the layer of the compiler that deals with compiling
144 hsc_dflags :: DynFlags,
145 -- The dynamic flag settings
147 hsc_targets :: [Target],
148 -- The targets (or roots) of the current session
150 hsc_mod_graph :: ModuleGraph,
151 -- The module graph of the current session
153 hsc_IC :: InteractiveContext,
154 -- The context for evaluating interactive statements
156 hsc_HPT :: HomePackageTable,
157 -- The home package table describes already-compiled
158 -- home-packge modules, *excluding* the module we
159 -- are compiling right now.
160 -- (In one-shot mode the current module is the only
161 -- home-package module, so hsc_HPT is empty. All other
162 -- modules count as "external-package" modules.
163 -- However, even in GHCi mode, hi-boot interfaces are
164 -- demand-loadeded into the external-package table.)
166 -- hsc_HPT is not mutable because we only demand-load
167 -- external packages; the home package is eagerly
168 -- loaded, module by module, by the compilation manager.
170 -- The HPT may contain modules compiled earlier by --make
171 -- but not actually below the current module in the dependency
172 -- graph. (This changes a previous invariant: changed Jan 05.)
174 hsc_EPS :: {-# UNPACK #-} !(IORef ExternalPackageState),
175 hsc_NC :: {-# UNPACK #-} !(IORef NameCache),
176 -- These are side-effected by compiling to reflect
177 -- sucking in interface files. They cache the state of
178 -- external interface files, in effect.
180 hsc_FC :: {-# UNPACK #-} !(IORef FinderCache),
181 hsc_MLC :: {-# UNPACK #-} !(IORef ModLocationCache),
182 -- The finder's cache. This caches the location of modules,
183 -- so we don't have to search the filesystem multiple times.
185 hsc_global_rdr_env :: GlobalRdrEnv,
186 hsc_global_type_env :: TypeEnv
189 hscEPS :: HscEnv -> IO ExternalPackageState
190 hscEPS hsc_env = readIORef (hsc_EPS hsc_env)
192 -- | A compilation target.
194 -- A target may be supplied with the actual text of the
195 -- module. If so, use this instead of the file contents (this
196 -- is for use in an IDE where the file hasn't been saved by
198 data Target = Target TargetId (Maybe (StringBuffer,ClockTime))
201 = TargetModule ModuleName
202 -- ^ A module name: search for the file
203 | TargetFile FilePath (Maybe Phase)
204 -- ^ A filename: preprocess & parse it to find the module name.
205 -- If specified, the Phase indicates how to compile this file
206 -- (which phase to start from). Nothing indicates the starting phase
207 -- should be determined from the suffix of the filename.
210 pprTarget :: Target -> SDoc
211 pprTarget (Target id _) = pprTargetId id
213 instance Outputable Target where
216 pprTargetId (TargetModule m) = ppr m
217 pprTargetId (TargetFile f _) = text f
219 instance Outputable TargetId where
222 type HomePackageTable = ModuleNameEnv HomeModInfo
223 -- Domain = modules in the home package
224 -- "home" package name cached here for convenience
225 type PackageIfaceTable = ModuleEnv ModIface
226 -- Domain = modules in the imported packages
228 emptyHomePackageTable = emptyUFM
229 emptyPackageIfaceTable = emptyModuleEnv
232 = HomeModInfo { hm_iface :: !ModIface,
233 hm_details :: !ModDetails,
234 hm_linkable :: !(Maybe Linkable) }
235 -- hm_linkable might be Nothing if:
236 -- a) this is an .hs-boot module
237 -- b) temporarily during compilation if we pruned away
238 -- the old linkable because it was out of date.
239 -- after a complete compilation (GHC.load), all hm_linkable
240 -- fields in the HPT will be Just.
242 -- When re-linking a module (hscNoRecomp), we construct
243 -- the HomModInfo by building a new ModDetails from the
244 -- old ModIface (only).
246 -- | Find the 'ModIface' for a 'Module'
253 lookupIfaceByModule dflags hpt pit mod
254 | modulePackageId mod == thisPackage dflags
255 = -- The module comes from the home package, so look first
256 -- in the HPT. If it's not from the home package it's wrong to look
257 -- in the HPT, because the HPT is indexed by *ModuleName* not Module
258 fmap hm_iface (lookupUFM hpt (moduleName mod))
259 `seqMaybe` lookupModuleEnv pit mod
261 | otherwise = lookupModuleEnv pit mod -- Look in PIT only
263 -- If the module does come from the home package, why do we look in the PIT as well?
264 -- (a) In OneShot mode, even home-package modules accumulate in the PIT
265 -- (b) Even in Batch (--make) mode, there is *one* case where a home-package
266 -- module is in the PIT, namely GHC.Prim when compiling the base package.
267 -- We could eliminate (b) if we wanted, by making GHC.Prim belong to a packake
268 -- of its own, but it doesn't seem worth the bother.
273 hptInstances :: HscEnv -> (ModuleName -> Bool) -> [Instance]
274 -- Find all the instance declarations that are in modules imported
275 -- by this one, directly or indirectly, and are in the Home Package Table
276 -- This ensures that we don't see instances from modules --make compiled
277 -- before this one, but which are not below this one
278 hptInstances hsc_env want_this_module
280 | mod_info <- eltsUFM (hsc_HPT hsc_env)
281 , want_this_module (moduleName (mi_module (hm_iface mod_info)))
282 , ispec <- md_insts (hm_details mod_info) ]
284 hptRules :: HscEnv -> [(ModuleName, IsBootInterface)] -> [CoreRule]
285 -- Get rules from modules "below" this one (in the dependency sense)
286 -- C.f Inst.hptInstances
287 hptRules hsc_env deps
288 | isOneShot (ghcMode (hsc_dflags hsc_env)) = []
291 hpt = hsc_HPT hsc_env
294 | -- Find each non-hi-boot module below me
297 -- unsavoury: when compiling the base package with --make, we
298 -- sometimes try to look up RULES for GHC.Prim. GHC.Prim won't
299 -- be in the HPT, because we never compile it; it's in the EPT
300 -- instead. ToDo: clean up, and remove this slightly bogus
302 , mod /= moduleName gHC_PRIM
304 -- Look it up in the HPT
305 , let rules = case lookupUFM hpt mod of
306 Just info -> md_rules (hm_details info)
307 Nothing -> pprTrace "WARNING in hptRules" msg []
308 msg = vcat [ptext SLIT("missing module") <+> ppr mod,
309 ptext SLIT("Probable cause: out-of-date interface files")]
310 -- This really shouldn't happen, but see Trac #962
316 %************************************************************************
318 \subsection{The Finder cache}
320 %************************************************************************
323 -- | The 'FinderCache' maps home module names to the result of
324 -- searching for that module. It records the results of searching for
325 -- modules along the search path. On @:load@, we flush the entire
326 -- contents of this cache.
328 -- Although the @FinderCache@ range is 'FindResult' for convenience ,
329 -- in fact it will only ever contain 'Found' or 'NotFound' entries.
331 type FinderCache = ModuleNameEnv FindResult
333 -- | The result of searching for an imported module.
335 = Found ModLocation Module
336 -- the module was found
337 | NoPackage PackageId
338 -- the requested package was not found
339 | FoundMultiple [PackageId]
340 -- *error*: both in multiple packages
341 | PackageHidden PackageId
342 -- for an explicit source import: the package containing the module is
344 | ModuleHidden PackageId
345 -- for an explicit source import: the package containing the module is
346 -- exposed, but the module itself is hidden.
347 | NotFound [FilePath] (Maybe PackageId)
348 -- the module was not found, the specified places were searched
349 | NotFoundInPackage PackageId
350 -- the module was not found in this package
352 -- | Cache that remembers where we found a particular module. Contains both
353 -- home modules and package modules. On @:load@, only home modules are
354 -- purged from this cache.
355 type ModLocationCache = ModuleEnv ModLocation
358 %************************************************************************
360 \subsection{Symbol tables and Module details}
362 %************************************************************************
364 A @ModIface@ plus a @ModDetails@ summarises everything we know
365 about a compiled module. The @ModIface@ is the stuff *before* linking,
366 and can be written out to an interface file. (The @ModDetails@ is after
367 linking; it is the "linked" form of the mi_decls field.)
369 When we *read* an interface file, we also construct a @ModIface@ from it,
370 except that the mi_decls part is empty; when reading we consolidate
371 the declarations into a single indexed map in the @PersistentRenamerState@.
376 mi_module :: !Module,
377 mi_mod_vers :: !Version, -- Module version: changes when anything changes
379 mi_orphan :: !WhetherHasOrphans, -- Whether this module has orphans
380 mi_finsts :: !WhetherHasFamInst, -- Whether module has family insts
381 mi_boot :: !IsBootInterface, -- Read from an hi-boot file?
383 mi_deps :: Dependencies,
384 -- This is consulted for directly-imported modules,
385 -- but not for anything else (hence lazy)
387 -- Usages; kept sorted so that it's easy to decide
388 -- whether to write a new iface file (changing usages
389 -- doesn't affect the version of this module)
390 mi_usages :: [Usage],
391 -- NOT STRICT! we read this field lazily from the interface file
392 -- It is *only* consulted by the recompilation checker
395 -- Kept sorted by (mod,occ), to make version comparisons easier
396 mi_exports :: ![IfaceExport],
397 mi_exp_vers :: !Version, -- Version number of export list
400 mi_fixities :: [(OccName,Fixity)],
401 -- NOT STRICT! we read this field lazily from the interface file
404 mi_deprecs :: IfaceDeprecs,
405 -- NOT STRICT! we read this field lazily from the interface file
407 -- Type, class and variable declarations
408 -- The version of an Id changes if its fixity or deprecations change
409 -- (as well as its type of course)
410 -- Ditto data constructors, class operations, except that
411 -- the version of the parent class/tycon changes
412 mi_decls :: [(Version,IfaceDecl)], -- Sorted
414 mi_globals :: !(Maybe GlobalRdrEnv),
415 -- Binds all the things defined at the top level in
416 -- the *original source* code for this module. which
417 -- is NOT the same as mi_exports, nor mi_decls (which
418 -- may contains declarations for things not actually
419 -- defined by the user). Used for GHCi and for inspecting
420 -- the contents of modules via the GHC API only.
422 -- (We need the source file to figure out the
423 -- top-level environment, if we didn't compile this module
424 -- from source then this field contains Nothing).
426 -- Strictly speaking this field should live in the
427 -- HomeModInfo, but that leads to more plumbing.
429 -- Instance declarations and rules
430 mi_insts :: [IfaceInst], -- Sorted
431 mi_fam_insts :: [IfaceFamInst], -- Sorted
432 mi_rules :: [IfaceRule], -- Sorted
433 mi_rule_vers :: !Version, -- Version number for rules and
434 -- instances (for classes and families)
437 -- Cached environments for easy lookup
438 -- These are computed (lazily) from other fields
439 -- and are not put into the interface file
440 mi_dep_fn :: Name -> Maybe DeprecTxt, -- Cached lookup for mi_deprecs
441 mi_fix_fn :: OccName -> Fixity, -- Cached lookup for mi_fixities
442 mi_ver_fn :: OccName -> Maybe (OccName, Version)
443 -- Cached lookup for mi_decls
444 -- The Nothing in mi_ver_fn means that the thing
445 -- isn't in decls. It's useful to know that when
446 -- seeing if we are up to date wrt the old interface
447 -- The 'OccName' is the parent of the name, if it has one.
450 -- Should be able to construct ModDetails from mi_decls in ModIface
453 -- The next two fields are created by the typechecker
454 md_exports :: [AvailInfo],
455 md_types :: !TypeEnv,
456 md_insts :: ![Instance], -- Dfun-ids for the instances in this module
457 md_fam_insts :: ![FamInst],
458 md_rules :: ![CoreRule], -- Domain may include Ids from other modules
459 md_dbg_sites :: ![(SiteNumber, Coord)] -- Breakpoint sites inserted by the renamer
462 emptyModDetails = ModDetails { md_types = emptyTypeEnv,
467 md_dbg_sites = noDbgSites}
469 -- A ModGuts is carried through the compiler, accumulating stuff as it goes
470 -- There is only one ModGuts at any time, the one for the module
471 -- being compiled right now. Once it is compiled, a ModIface and
472 -- ModDetails are extracted and the ModGuts is dicarded.
476 mg_module :: !Module,
477 mg_boot :: IsBootInterface, -- Whether it's an hs-boot module
478 mg_exports :: ![AvailInfo], -- What it exports
479 mg_deps :: !Dependencies, -- What is below it, directly or
481 mg_dir_imps :: ![Module], -- Directly-imported modules; used to
482 -- generate initialisation code
483 mg_usages :: ![Usage], -- Version info for what it needed
485 mg_rdr_env :: !GlobalRdrEnv, -- Top-level lexical environment
486 mg_fix_env :: !FixityEnv, -- Fixity env, for things declared in
489 mg_fam_inst_env :: FamInstEnv, -- Type-family instance enviroment
490 -- for *home-package* modules (including
491 -- this one). c.f. tcg_fam_inst_env
493 mg_types :: !TypeEnv,
494 mg_insts :: ![Instance], -- Instances
495 mg_fam_insts :: ![FamInst], -- Instances
496 mg_rules :: ![CoreRule], -- Rules from this module
497 mg_binds :: ![CoreBind], -- Bindings for this module
498 mg_foreign :: !ForeignStubs,
499 mg_deprecs :: !Deprecations, -- Deprecations declared in the module
500 mg_hpc_info :: !HpcInfo, -- info about coverage tick boxes
501 mg_dbg_sites :: ![(SiteNumber, Coord)] -- Bkpts inserted by the renamer
504 -- The ModGuts takes on several slightly different forms:
506 -- After simplification, the following fields change slightly:
507 -- mg_rules Orphan rules only (local ones now attached to binds)
508 -- mg_binds With rules attached
511 ---------------------------------------------------------
512 -- The Tidy pass forks the information about this module:
513 -- * one lot goes to interface file generation (ModIface)
514 -- and later compilations (ModDetails)
515 -- * the other lot goes to code generation (CgGuts)
518 cg_module :: !Module,
520 cg_tycons :: [TyCon],
521 -- Algebraic data types (including ones that started
522 -- life as classes); generate constructors and info
523 -- tables Includes newtypes, just for the benefit of
526 cg_binds :: [CoreBind],
527 -- The tidied main bindings, including
528 -- previously-implicit bindings for record and class
529 -- selectors, and data construtor wrappers. But *not*
530 -- data constructor workers; reason: we we regard them
531 -- as part of the code-gen of tycons
533 cg_dir_imps :: ![Module],
534 -- Directly-imported modules; used to generate
535 -- initialisation code
537 cg_foreign :: !ForeignStubs,
538 cg_dep_pkgs :: ![PackageId], -- Used to generate #includes for C code gen
539 cg_hpc_info :: !HpcInfo -- info about coverage tick boxes
542 -----------------------------------
545 imp_direct :: ![(Module,Bool)], -- Explicitly-imported modules
546 -- Boolean is true if we imported the whole
547 -- module (apart, perhaps, from hiding some)
548 imp_pkg_mods :: !ModuleSet, -- Non-home-package modules on which we depend,
549 -- directly or indirectly
550 imp_home_names :: !NameSet -- Home package things on which we depend,
551 -- directly or indirectly
554 -----------------------------------
555 data ForeignStubs = NoStubs
557 SDoc -- Header file prototypes for
558 -- "foreign exported" functions
559 SDoc -- C stubs to use when calling
560 -- "foreign exported" functions
561 [FastString] -- Headers that need to be included
562 -- into C code generated for this module
563 [Id] -- Foreign-exported binders
564 -- we have to generate code to register these
569 emptyModIface :: Module -> ModIface
571 = ModIface { mi_module = mod,
572 mi_mod_vers = initialVersion,
576 mi_deps = noDependencies,
579 mi_exp_vers = initialVersion,
581 mi_deprecs = NoDeprecs,
586 mi_globals = Nothing,
587 mi_rule_vers = initialVersion,
588 mi_dep_fn = emptyIfaceDepCache,
589 mi_fix_fn = emptyIfaceFixCache,
590 mi_ver_fn = emptyIfaceVerCache
595 %************************************************************************
597 \subsection{The interactive context}
599 %************************************************************************
602 data InteractiveContext
603 = InteractiveContext {
604 ic_toplev_scope :: [Module], -- Include the "top-level" scope of
607 ic_exports :: [Module], -- Include just the exports of these
610 ic_rn_gbl_env :: GlobalRdrEnv, -- The cached GlobalRdrEnv, built from
611 -- ic_toplev_scope and ic_exports
613 ic_rn_local_env :: LocalRdrEnv, -- Lexical context for variables bound
614 -- during interaction
616 ic_type_env :: TypeEnv -- Ditto for types
619 emptyInteractiveContext
620 = InteractiveContext { ic_toplev_scope = [],
622 ic_rn_gbl_env = emptyGlobalRdrEnv,
623 ic_rn_local_env = emptyLocalRdrEnv,
624 ic_type_env = emptyTypeEnv }
626 icPrintUnqual :: InteractiveContext -> PrintUnqualified
627 icPrintUnqual ictxt = mkPrintUnqualified (ic_rn_gbl_env ictxt)
630 %************************************************************************
632 Building a PrintUnqualified
634 %************************************************************************
637 mkPrintUnqualified :: GlobalRdrEnv -> PrintUnqualified
638 mkPrintUnqualified env = (qual_name, qual_mod)
641 | null gres = Just (moduleName mod)
642 -- it isn't in scope at all, this probably shouldn't happen,
643 -- but we'll qualify it by the original module anyway.
644 | any unQualOK gres = Nothing
645 | (Imported is : _) <- map gre_prov gres, (idecl : _) <- is
646 = Just (is_as (is_decl idecl))
647 | otherwise = panic "mkPrintUnqualified"
649 gres = [ gre | gre <- lookupGlobalRdrEnv env occ,
650 nameModule (gre_name gre) == mod ]
652 qual_mod mod = Nothing -- For now...
656 %************************************************************************
660 %************************************************************************
663 implicitTyThings :: TyThing -> [TyThing]
664 -- If you change this, make sure you change LoadIface.ifaceDeclSubBndrs in sync
666 implicitTyThings (AnId id) = []
668 -- For type constructors, add the data cons (and their extras),
669 -- and the selectors and generic-programming Ids too
671 -- Newtypes don't have a worker Id, so don't generate that?
672 implicitTyThings (ATyCon tc) = implicitCoTyCon tc ++
673 map AnId (tyConSelIds tc) ++
674 concatMap (extras_plus . ADataCon)
677 -- For classes, add the class selector Ids, and assoicated TyCons
678 -- and the class TyCon too (and its extras)
679 implicitTyThings (AClass cl)
680 = map AnId (classSelIds cl) ++
681 map ATyCon (classATs cl) ++
682 -- No extras_plus for the classATs, because they
683 -- are only the family decls; they have no implicit things
684 extras_plus (ATyCon (classTyCon cl))
686 -- For data cons add the worker and wrapper (if any)
687 implicitTyThings (ADataCon dc) = map AnId (dataConImplicitIds dc)
689 -- | returns 'True' if there should be no interface-file declaration
690 -- for this thing on its own: either it is built-in, or it is part
691 -- of some other declaration, or it is generated implicitly by some
692 -- other declaration.
693 isImplicitTyThing :: TyThing -> Bool
694 isImplicitTyThing (ADataCon _) = True
695 isImplicitTyThing (AnId id) = isImplicitId id
696 isImplicitTyThing (AClass _) = False
697 isImplicitTyThing (ATyCon tc) = isImplicitTyCon tc
699 -- For newtypes and indexed data types, add the implicit coercion tycon
701 = map ATyCon . catMaybes $ [newTyConCo_maybe tc,
702 tyConFamilyCoercion_maybe tc]
704 extras_plus thing = thing : implicitTyThings thing
706 extendTypeEnvWithIds :: TypeEnv -> [Id] -> TypeEnv
707 extendTypeEnvWithIds env ids
708 = extendNameEnvList env [(getName id, AnId id) | id <- ids]
711 %************************************************************************
715 %************************************************************************
718 type TypeEnv = NameEnv TyThing
720 emptyTypeEnv :: TypeEnv
721 typeEnvElts :: TypeEnv -> [TyThing]
722 typeEnvClasses :: TypeEnv -> [Class]
723 typeEnvTyCons :: TypeEnv -> [TyCon]
724 typeEnvIds :: TypeEnv -> [Id]
725 typeEnvDataCons :: TypeEnv -> [DataCon]
726 lookupTypeEnv :: TypeEnv -> Name -> Maybe TyThing
728 emptyTypeEnv = emptyNameEnv
729 typeEnvElts env = nameEnvElts env
730 typeEnvClasses env = [cl | AClass cl <- typeEnvElts env]
731 typeEnvTyCons env = [tc | ATyCon tc <- typeEnvElts env]
732 typeEnvIds env = [id | AnId id <- typeEnvElts env]
733 typeEnvDataCons env = [dc | ADataCon dc <- typeEnvElts env]
735 mkTypeEnv :: [TyThing] -> TypeEnv
736 mkTypeEnv things = extendTypeEnvList emptyTypeEnv things
738 lookupTypeEnv = lookupNameEnv
740 -- Extend the type environment
741 extendTypeEnv :: TypeEnv -> TyThing -> TypeEnv
742 extendTypeEnv env thing = extendNameEnv env (getName thing) thing
744 extendTypeEnvList :: TypeEnv -> [TyThing] -> TypeEnv
745 extendTypeEnvList env things = foldl extendTypeEnv env things
749 lookupType :: DynFlags
755 lookupType dflags hpt pte name
756 -- in one-shot, we don't use the HPT
757 | not (isOneShot (ghcMode dflags)) && modulePackageId mod == this_pkg
758 = do hm <- lookupUFM hpt (moduleName mod) -- Maybe monad
759 lookupNameEnv (md_types (hm_details hm)) name
761 = lookupNameEnv pte name
762 where mod = nameModule name
763 this_pkg = thisPackage dflags
767 tyThingTyCon (ATyCon tc) = tc
768 tyThingTyCon other = pprPanic "tyThingTyCon" (ppr other)
770 tyThingClass (AClass cls) = cls
771 tyThingClass other = pprPanic "tyThingClass" (ppr other)
773 tyThingDataCon (ADataCon dc) = dc
774 tyThingDataCon other = pprPanic "tyThingDataCon" (ppr other)
776 tyThingId (AnId id) = id
777 tyThingId other = pprPanic "tyThingId" (ppr other)
780 %************************************************************************
782 \subsection{Auxiliary types}
784 %************************************************************************
786 These types are defined here because they are mentioned in ModDetails,
787 but they are mostly elaborated elsewhere
790 mkIfaceVerCache :: [(Version,IfaceDecl)]
791 -> (OccName -> Maybe (OccName, Version))
792 mkIfaceVerCache pairs
793 = \occ -> lookupOccEnv env occ
795 env = foldr add_decl emptyOccEnv pairs
796 add_decl (v,d) env0 = foldr add_imp env1 (ifaceDeclSubBndrs d)
799 env1 = extendOccEnv env0 decl_name (decl_name, v)
800 add_imp bndr env = extendOccEnv env bndr (decl_name, v)
802 emptyIfaceVerCache :: OccName -> Maybe (OccName, Version)
803 emptyIfaceVerCache occ = Nothing
805 ------------------ Deprecations -------------------------
808 | DeprecAll DeprecTxt -- Whole module deprecated
809 | DeprecSome a -- Some specific things deprecated
812 type IfaceDeprecs = Deprecs [(OccName,DeprecTxt)]
813 type Deprecations = Deprecs (NameEnv (OccName,DeprecTxt))
814 -- Keep the OccName so we can flatten the NameEnv to
815 -- get an IfaceDeprecs from a Deprecations
816 -- Only an OccName is needed, because a deprecation always
817 -- applies to things defined in the module in which the
818 -- deprecation appears.
820 mkIfaceDepCache:: IfaceDeprecs -> Name -> Maybe DeprecTxt
821 mkIfaceDepCache NoDeprecs = \n -> Nothing
822 mkIfaceDepCache (DeprecAll t) = \n -> Just t
823 mkIfaceDepCache (DeprecSome pairs) = lookupOccEnv (mkOccEnv pairs) . nameOccName
825 emptyIfaceDepCache :: Name -> Maybe DeprecTxt
826 emptyIfaceDepCache n = Nothing
828 plusDeprecs :: Deprecations -> Deprecations -> Deprecations
829 plusDeprecs d NoDeprecs = d
830 plusDeprecs NoDeprecs d = d
831 plusDeprecs d (DeprecAll t) = DeprecAll t
832 plusDeprecs (DeprecAll t) d = DeprecAll t
833 plusDeprecs (DeprecSome v1) (DeprecSome v2) = DeprecSome (v1 `plusNameEnv` v2)
838 type Avails = [AvailInfo]
839 type AvailInfo = GenAvailInfo Name
840 type RdrAvailInfo = GenAvailInfo OccName
842 data GenAvailInfo name = Avail name -- An ordinary identifier
843 | AvailTC name -- The name of the type or class
844 [name] -- The available pieces of type/class.
845 -- NB: If the type or class is itself
846 -- to be in scope, it must be in this list.
847 -- Thus, typically: AvailTC Eq [Eq, ==, /=]
849 -- Equality used when deciding if the interface has changed
851 type IfaceExport = (Module, [GenAvailInfo OccName])
853 availsToNameSet :: [AvailInfo] -> NameSet
854 availsToNameSet avails = foldr add emptyNameSet avails
855 where add avail set = addListToNameSet set (availNames avail)
857 availsToNameEnv :: [AvailInfo] -> NameEnv AvailInfo
858 availsToNameEnv avails = foldr add emptyNameEnv avails
859 where add avail env = extendNameEnvList env
860 (zip (availNames avail) (repeat avail))
862 availName :: GenAvailInfo name -> name
863 availName (Avail n) = n
864 availName (AvailTC n _) = n
866 availNames :: GenAvailInfo name -> [name]
867 availNames (Avail n) = [n]
868 availNames (AvailTC n ns) = ns
870 instance Outputable n => Outputable (GenAvailInfo n) where
873 pprAvail :: Outputable n => GenAvailInfo n -> SDoc
874 pprAvail (Avail n) = ppr n
875 pprAvail (AvailTC n ns) = ppr n <> braces (hsep (punctuate comma (map ppr ns)))
879 mkIfaceFixCache :: [(OccName, Fixity)] -> OccName -> Fixity
880 mkIfaceFixCache pairs
881 = \n -> lookupOccEnv env n `orElse` defaultFixity
885 emptyIfaceFixCache :: OccName -> Fixity
886 emptyIfaceFixCache n = defaultFixity
888 -- This fixity environment is for source code only
889 type FixityEnv = NameEnv FixItem
891 -- We keep the OccName in the range so that we can generate an interface from it
892 data FixItem = FixItem OccName Fixity SrcSpan
894 instance Outputable FixItem where
895 ppr (FixItem occ fix loc) = ppr fix <+> ppr occ <+> parens (ppr loc)
897 emptyFixityEnv :: FixityEnv
898 emptyFixityEnv = emptyNameEnv
900 lookupFixity :: FixityEnv -> Name -> Fixity
901 lookupFixity env n = case lookupNameEnv env n of
902 Just (FixItem _ fix _) -> fix
903 Nothing -> defaultFixity
907 %************************************************************************
909 \subsection{WhatsImported}
911 %************************************************************************
914 type WhetherHasOrphans = Bool
916 -- * an instance decl in a module other than the defn module for
917 -- one of the tycons or classes in the instance head
918 -- * a transformation rule in a module other than the one defining
919 -- the function in the head of the rule.
921 type WhetherHasFamInst = Bool -- This module defines family instances?
923 type IsBootInterface = Bool
925 -- Dependency info about modules and packages below this one
926 -- in the import hierarchy. See TcRnTypes.ImportAvails for details.
927 -- The orphan modules in `dep_orphs' do *not* include family instance orphans,
928 -- as they are anyway included in `dep_finsts'.
930 -- Invariant: the dependencies of a module M never includes M
931 -- Invariant: the lists are unordered, with no duplicates
933 = Deps { dep_mods :: [(ModuleName, -- Home-package module dependencies
935 , dep_pkgs :: [PackageId] -- External package dependencies
936 , dep_orphs :: [Module] -- Orphan modules (whether home or
938 , dep_finsts :: [Module] -- Modules that contain family
939 -- instances (whether home or
943 -- Equality used only for old/new comparison in MkIface.addVersionInfo
945 noDependencies :: Dependencies
946 noDependencies = Deps [] [] [] []
949 = Usage { usg_name :: ModuleName, -- Name of the module
950 usg_mod :: Version, -- Module version
951 usg_entities :: [(OccName,Version)], -- Sorted by occurrence name
952 -- NB. usages are for parent names only, eg. tycon but not constructors.
953 usg_exports :: Maybe Version, -- Export-list version, if we depend on it
954 usg_rules :: Version -- Orphan-rules version (for non-orphan
955 -- modules this will always be initialVersion)
957 -- This type doesn't let you say "I imported f but none of the rules in
958 -- the module". If you use anything in the module you get its rule version
959 -- So if the rules change, you'll recompile, even if you don't use them.
960 -- This is easy to implement, and it's safer: you might not have used the rules last
961 -- time round, but if someone has added a new rule you might need it this time
963 -- The export list field is (Just v) if we depend on the export list:
964 -- i.e. we imported the module directly, whether or not we
965 -- enumerated the things we imported, or just imported everything
966 -- We need to recompile if M's exports change, because
967 -- if the import was import M, we might now have a name clash in the
969 -- if the import was import M(x) M might no longer export x
970 -- The only way we don't depend on the export list is if we have
972 -- And of course, for modules that aren't imported directly we don't
973 -- depend on their export lists
977 %************************************************************************
979 The External Package State
981 %************************************************************************
984 type PackageTypeEnv = TypeEnv
985 type PackageRuleBase = RuleBase
986 type PackageInstEnv = InstEnv
987 type PackageFamInstEnv = FamInstEnv
989 data ExternalPackageState
991 eps_is_boot :: !(ModuleNameEnv (ModuleName, IsBootInterface)),
992 -- In OneShot mode (only), home-package modules
993 -- accumulate in the external package state, and are
994 -- sucked in lazily. For these home-pkg modules
995 -- (only) we need to record which are boot modules.
996 -- We set this field after loading all the
997 -- explicitly-imported interfaces, but before doing
1000 -- The ModuleName part is not necessary, but it's useful for
1001 -- debug prints, and it's convenient because this field comes
1002 -- direct from TcRnTypes.ImportAvails.imp_dep_mods
1004 eps_PIT :: !PackageIfaceTable,
1005 -- The ModuleIFaces for modules in external packages
1006 -- whose interfaces we have opened
1007 -- The declarations in these interface files are held in
1008 -- eps_decls, eps_inst_env, eps_fam_inst_env, eps_rules
1009 -- (below), not in the mi_decls fields of the iPIT.
1010 -- What _is_ in the iPIT is:
1017 eps_PTE :: !PackageTypeEnv, -- Domain = external-package modules
1019 eps_inst_env :: !PackageInstEnv, -- The total InstEnv accumulated
1020 -- from all the external-package
1022 eps_fam_inst_env :: !PackageFamInstEnv,-- Ditto FamInstEnv
1023 eps_rule_base :: !PackageRuleBase, -- Ditto RuleEnv
1025 eps_mod_fam_inst_env :: !(ModuleEnv FamInstEnv), -- identifies family
1026 -- instances of each mod
1028 eps_stats :: !EpsStats
1031 -- "In" means read from iface files
1032 -- "Out" means actually sucked in and type-checked
1033 data EpsStats = EpsStats { n_ifaces_in
1034 , n_decls_in, n_decls_out
1035 , n_rules_in, n_rules_out
1036 , n_insts_in, n_insts_out :: !Int }
1038 addEpsInStats :: EpsStats -> Int -> Int -> Int -> EpsStats
1039 -- Add stats for one newly-read interface
1040 addEpsInStats stats n_decls n_insts n_rules
1041 = stats { n_ifaces_in = n_ifaces_in stats + 1
1042 , n_decls_in = n_decls_in stats + n_decls
1043 , n_insts_in = n_insts_in stats + n_insts
1044 , n_rules_in = n_rules_in stats + n_rules }
1047 The NameCache makes sure that there is just one Unique assigned for
1048 each original name; i.e. (module-name, occ-name) pair. The Name is
1049 always stored as a Global, and has the SrcLoc of its binding location.
1050 Actually that's not quite right. When we first encounter the original
1051 name, we might not be at its binding site (e.g. we are reading an
1052 interface file); so we give it 'noSrcLoc' then. Later, when we find
1053 its binding site, we fix it up.
1057 = NameCache { nsUniqs :: UniqSupply,
1058 -- Supply of uniques
1059 nsNames :: OrigNameCache,
1060 -- Ensures that one original name gets one unique
1061 nsIPs :: OrigIParamCache
1062 -- Ensures that one implicit parameter name gets one unique
1065 type OrigNameCache = ModuleEnv (OccEnv Name)
1066 type OrigIParamCache = FiniteMap (IPName OccName) (IPName Name)
1071 %************************************************************************
1073 The module graph and ModSummary type
1074 A ModSummary is a node in the compilation manager's
1075 dependency graph, and it's also passed to hscMain
1077 %************************************************************************
1079 A ModuleGraph contains all the nodes from the home package (only).
1080 There will be a node for each source module, plus a node for each hi-boot
1084 type ModuleGraph = [ModSummary] -- The module graph,
1085 -- NOT NECESSARILY IN TOPOLOGICAL ORDER
1087 emptyMG :: ModuleGraph
1090 -- The nodes of the module graph are
1091 -- EITHER a regular Haskell source module
1092 -- OR a hi-boot source module
1096 ms_mod :: Module, -- Identity of the module
1097 ms_hsc_src :: HscSource, -- Source is Haskell, hs-boot, external core
1098 ms_location :: ModLocation, -- Location
1099 ms_hs_date :: ClockTime, -- Timestamp of source file
1100 ms_obj_date :: Maybe ClockTime, -- Timestamp of object, maybe
1101 ms_srcimps :: [Located ModuleName], -- Source imports
1102 ms_imps :: [Located ModuleName], -- Non-source imports
1103 ms_hspp_file :: FilePath, -- Filename of preprocessed source.
1104 ms_hspp_opts :: DynFlags, -- Cached flags from OPTIONS, INCLUDE
1105 -- and LANGUAGE pragmas.
1106 ms_hspp_buf :: Maybe StringBuffer -- The actual preprocessed source, maybe.
1109 -- The ModLocation contains both the original source filename and the
1110 -- filename of the cleaned-up source file after all preprocessing has been
1111 -- done. The point is that the summariser will have to cpp/unlit/whatever
1112 -- all files anyway, and there's no point in doing this twice -- just
1113 -- park the result in a temp file, put the name of it in the location,
1114 -- and let @compile@ read from that file on the way back up.
1116 -- The ModLocation is stable over successive up-sweeps in GHCi, wheres
1117 -- the ms_hs_date and imports can, of course, change
1119 msHsFilePath, msHiFilePath, msObjFilePath :: ModSummary -> FilePath
1120 msHsFilePath ms = expectJust "msHsFilePath" (ml_hs_file (ms_location ms))
1121 msHiFilePath ms = ml_hi_file (ms_location ms)
1122 msObjFilePath ms = ml_obj_file (ms_location ms)
1124 isBootSummary :: ModSummary -> Bool
1125 isBootSummary ms = isHsBoot (ms_hsc_src ms)
1127 instance Outputable ModSummary where
1129 = sep [text "ModSummary {",
1130 nest 3 (sep [text "ms_hs_date = " <> text (show (ms_hs_date ms)),
1131 text "ms_mod =" <+> ppr (ms_mod ms)
1132 <> text (hscSourceString (ms_hsc_src ms)) <> comma,
1133 text "ms_imps =" <+> ppr (ms_imps ms),
1134 text "ms_srcimps =" <+> ppr (ms_srcimps ms)]),
1138 showModMsg :: HscTarget -> Bool -> ModSummary -> String
1139 showModMsg target recomp mod_summary
1140 = showSDoc (hsep [text (mod_str ++ replicate (max 0 (16 - length mod_str)) ' '),
1141 char '(', text (msHsFilePath mod_summary) <> comma,
1143 #if defined(GHCI) && defined(DEBUGGER)
1144 HscInterpreted | recomp &&
1145 Opt_Debugging `elem` modflags
1146 -> text "interpreted(debugging)"
1148 HscInterpreted | recomp
1149 -> text "interpreted"
1150 HscNothing -> text "nothing"
1151 _other -> text (msObjFilePath mod_summary),
1154 mod = moduleName (ms_mod mod_summary)
1155 mod_str = showSDoc (ppr mod) ++ hscSourceString (ms_hsc_src mod_summary)
1156 modflags= flags(ms_hspp_opts mod_summary)
1160 %************************************************************************
1162 \subsection{Hpc Support}
1164 %************************************************************************
1167 type HpcInfo = Int -- just the number of ticks in a module
1169 noHpcInfo :: HpcInfo
1170 noHpcInfo = 0 -- default = 0
1173 %************************************************************************
1175 \subsection{Linkable stuff}
1177 %************************************************************************
1179 This stuff is in here, rather than (say) in Linker.lhs, because the Linker.lhs
1180 stuff is the *dynamic* linker, and isn't present in a stage-1 compiler
1183 data Linkable = LM {
1184 linkableTime :: ClockTime, -- Time at which this linkable was built
1185 -- (i.e. when the bytecodes were produced,
1186 -- or the mod date on the files)
1187 linkableModule :: Module, -- Should be Module, but see below
1188 linkableUnlinked :: [Unlinked]
1191 isObjectLinkable :: Linkable -> Bool
1192 isObjectLinkable l = not (null unlinked) && all isObject unlinked
1193 where unlinked = linkableUnlinked l
1194 -- A linkable with no Unlinked's is treated as a BCO. We can
1195 -- generate a linkable with no Unlinked's as a result of
1196 -- compiling a module in HscNothing mode, and this choice
1197 -- happens to work well with checkStability in module GHC.
1199 instance Outputable Linkable where
1200 ppr (LM when_made mod unlinkeds)
1201 = (text "LinkableM" <+> parens (text (show when_made)) <+> ppr mod)
1202 $$ nest 3 (ppr unlinkeds)
1204 -------------------------------------------
1209 | BCOs CompiledByteCode
1212 data CompiledByteCode = NoByteCode
1215 instance Outputable Unlinked where
1216 ppr (DotO path) = text "DotO" <+> text path
1217 ppr (DotA path) = text "DotA" <+> text path
1218 ppr (DotDLL path) = text "DotDLL" <+> text path
1220 ppr (BCOs bcos) = text "BCOs" <+> ppr bcos
1222 ppr (BCOs bcos) = text "No byte code"
1225 isObject (DotO _) = True
1226 isObject (DotA _) = True
1227 isObject (DotDLL _) = True
1230 isInterpretable = not . isObject
1232 nameOfObject (DotO fn) = fn
1233 nameOfObject (DotA fn) = fn
1234 nameOfObject (DotDLL fn) = fn
1235 nameOfObject other = pprPanic "nameOfObject" (ppr other)
1237 byteCodeOfObject (BCOs bc) = bc
1238 byteCodeOfObject other = pprPanic "byteCodeOfObject" (ppr other)