[project @ 2001-05-08 10:58:48 by simonmar]
[ghc-hetmet.git] / ghc / compiler / main / MkIface.lhs
index 4720cb0..45828c7 100644 (file)
@@ -1,12 +1,13 @@
 %
 % (c) The GRASP/AQUA Project, Glasgow University, 1993-1998
 %
+
 \section[MkIface]{Print an interface for a module}
 
 \begin{code}
 module MkIface ( 
-       mkModDetails, mkModDetailsFromIface, completeIface, 
-       writeIface, pprIface, pprUsage
+       mkFinalIface,
+       pprModDetails, pprIface, pprUsage
   ) where
 
 #include "HsVersions.h"
@@ -19,166 +20,121 @@ import BasicTypes ( Fixity(..), NewOrData(..),
                        )
 import RnMonad
 import RnHsSyn         ( RenamedInstDecl, RenamedTyClDecl )
-import TcHsSyn         ( TypecheckedRuleDecl )
 import HscTypes                ( VersionInfo(..), ModIface(..), ModDetails(..),
+                         ModuleLocation(..), 
                          IfaceDecls, mkIfaceDecls, dcl_tycl, dcl_rules, dcl_insts,
-                         TyThing(..), DFunId, TypeEnv, Avails,
+                         TyThing(..), DFunId, Avails,
                          WhatsImported(..), GenAvailInfo(..), 
                          ImportVersion, AvailInfo, Deprecations(..),
-                         extendTypeEnvList, lookupVersion,
+                         lookupVersion,
                        )
+import CmStaticInfo    ( GhciMode(..) )
 
 import CmdLineOpts
-import Id              ( idType, idInfo, isImplicitId, isDictFunId,
-                         idSpecialisation, isLocalId, idName, hasNoBinding
+import Id              ( idType, idInfo, isImplicitId, idCgInfo,
+                         isLocalId, idName,
                        )
-import Var             ( isId )
-import VarSet
 import DataCon         ( StrictnessMark(..), dataConId, dataConSig, dataConFieldLabels, dataConStrictMarks )
 import IdInfo          -- Lots
-import CoreSyn         ( CoreBind, CoreRule(..), IdCoreRule, 
-                         isBuiltinRule, rulesRules, 
-                         bindersOf, bindersOfBinds
-                       )
-import CoreFVs         ( ruleSomeLhsFreeVars )
+import CoreSyn         ( CoreRule(..) )
+import CoreFVs         ( ruleLhsFreeNames )
 import CoreUnfold      ( neverUnfold, unfoldingTemplate )
-import Name            ( getName, nameModule, Name, NamedThing(..) )
-import Name    -- Env
+import PprCore         ( pprIdCoreRule )
+import Name            ( getName, nameModule, toRdrName, isGlobalName, 
+                         nameIsLocalOrFrom, Name, NamedThing(..) )
+import NameEnv
+import NameSet
 import OccName         ( pprOccName )
 import TyCon           ( TyCon, getSynTyConDefn, isSynTyCon, isNewTyCon, isAlgTyCon, tyConGenIds,
                          tyConTheta, tyConTyVars, tyConDataCons, tyConFamilySize, isClassTyCon
                        )
 import Class           ( classExtraBigSig, classTyCon, DefMeth(..) )
 import FieldLabel      ( fieldLabelType )
-import Type            ( splitSigmaTy, tidyTopType, deNoteType )
+import Type            ( splitSigmaTy, tidyTopType, deNoteType, namesOfDFunHead )
 import SrcLoc          ( noSrcLoc )
 import Outputable
 import Module          ( ModuleName )
-import Maybes          ( orElse )
+import Util            ( sortLt, unJust )
+import ErrUtils                ( dumpIfSet_dyn )
 
+import Monad           ( when )
 import IO              ( IOMode(..), openFile, hClose )
 \end{code}
 
 
 %************************************************************************
 %*                                                                     *
-\subsection{Write a new interface file}
+\subsection{Completing an interface}
 %*                                                                     *
 %************************************************************************
 
 \begin{code}
-mkModDetails :: TypeEnv                -- From typechecker
-            -> [CoreBind]      -- Final bindings
-                               -- they have authoritative arity info
-            -> [IdCoreRule]    -- Tidy orphan rules
-            -> ModDetails
-mkModDetails type_env tidy_binds orphan_rules
-  = ModDetails { md_types = new_type_env,
-                md_rules = rule_dcls,
-                md_insts = filter isDictFunId final_ids }
-  where
-       -- The competed type environment is gotten from
-       --      a) keeping the types and classes
-       --      b) removing all Ids, 
-       --      c) adding Ids with correct IdInfo, including unfoldings,
-       --              gotten from the bindings
-       -- From (c) we keep only those Ids with Global names;
-       --          the CoreTidy pass makes sure these are all and only
-       --          the externally-accessible ones
-       -- This truncates the type environment to include only the 
-       -- exported Ids and things needed from them, which saves space
-       --
-       -- However, we do keep things like constructors, which should not appear 
-       -- in interface files, because they are needed by importing modules when
-       -- using the compilation manager
-    new_type_env = extendTypeEnvList (filterNameEnv keep_it type_env)
-                                    (map AnId final_ids)
-
-       -- We keep constructor workers, because they won't appear
-       -- in the bindings from which final_ids are derived!
-    keep_it (AnId id) = hasNoBinding id
-    keep_it other     = True
-
-    final_ids  = [id | bind <- tidy_binds
-                    , id <- bindersOf bind
-                    , isGlobalName (idName id)]
-
-       -- The complete rules are gotten by combining
-       --      a) the orphan rules
-       --      b) rules embedded in the top-level Ids
-    rule_dcls | opt_OmitInterfacePragmas = []
-             | otherwise                = getRules orphan_rules tidy_binds (mkVarSet final_ids)
-
--- This version is used when we are re-linking a module
--- so we've only run the type checker on its previous interface 
-mkModDetailsFromIface :: TypeEnv 
-                     -> [TypecheckedRuleDecl]
-                     -> ModDetails
-mkModDetailsFromIface type_env rules
-  = ModDetails { md_types = type_env,
-                md_rules = rule_dcls,
-                md_insts = dfun_ids }
-  where
-    dfun_ids  = [dfun_id | AnId dfun_id <- nameEnvElts type_env, isDictFunId dfun_id]
-    rule_dcls = [(id,rule) | IfaceRuleOut id rule <- rules]
-       -- All the rules from an interface are of the IfaceRuleOut form
-\end{code}
 
-\begin{code}
-getRules :: [IdCoreRule]       -- Orphan rules
-        -> [CoreBind]          -- Bindings, with rules in the top-level Ids
-        -> IdSet               -- Ids that are exported, so we need their rules
-        -> [IdCoreRule]
-getRules orphan_rules binds emitted
-  = orphan_rules ++ local_rules
-  where
-    local_rules  = [ (fn, rule)
-                  | fn <- bindersOfBinds binds,
-                    fn `elemVarSet` emitted,
-                    rule <- rulesRules (idSpecialisation fn),
-                    not (isBuiltinRule rule),
-                               -- We can't print builtin rules in interface files
-                               -- Since they are built in, an importing module
-                               -- will have access to them anyway
-
-                       -- Sept 00: I've disabled this test.  It doesn't stop many, if any, rules
-                       -- from coming out, and to make it work properly we need to add ????
-                       --      (put it back in for now)
-                    all (`elemVarSet` emitted) (varSetElems (ruleSomeLhsFreeVars interestingId rule))
-                               -- Spit out a rule only if all its lhs free vars are emitted
-                               -- This is a good reason not to do it when we emit the Id itself
-                  ]
-
-interestingId id = isId id && isLocalId id
-\end{code}
 
 
-%************************************************************************
-%*                                                                     *
-\subsection{Completing an interface}
-%*                                                                     *
-%************************************************************************
+mkFinalIface :: GhciMode
+            -> DynFlags
+            -> ModuleLocation
+            -> Maybe ModIface          -- The old interface, if we have it
+            -> ModIface                -- The new one, minus the decls and versions
+            -> ModDetails              -- The ModDetails for this module
+            -> IO ModIface             -- The new one, complete with decls and versions
+-- mkFinalIface 
+--     a) completes the interface
+--     b) writes it out to a file if necessary
+
+mkFinalIface ghci_mode dflags location 
+            maybe_old_iface new_iface new_details
+  = do { 
+               -- Add the new declarations, and the is-orphan flag
+         let iface_w_decls = new_iface { mi_decls = new_decls,
+                                         mi_orphan = orphan_mod }
+
+               -- Add version information
+       ; let (final_iface, maybe_diffs) = addVersionInfo maybe_old_iface iface_w_decls
+
+               -- Write the interface file, if necessary
+       ; when (must_write_hi_file maybe_diffs)
+              (writeIface hi_file_path final_iface)
+
+               -- Debug printing
+       ; write_diffs dflags final_iface maybe_diffs
+
+       ; return final_iface }
 
-\begin{code}
-completeIface :: Maybe ModIface                -- The old interface, if we have it
-             -> ModIface               -- The new one, minus the decls and versions
-             -> ModDetails             -- The ModDetails for this module
-             -> (ModIface, Maybe SDoc) -- The new one, complete with decls and versions
-                                       -- The SDoc is a debug document giving differences
-                                       -- Nothing => no change
-
-       -- NB: 'Nothing' means that even the usages havn't changed, so there's no
-       --     need to write a new interface file.  But even if the usages have
-       --     changed, the module version may not have.
-completeIface maybe_old_iface new_iface mod_details 
-  = addVersionInfo maybe_old_iface (new_iface { mi_decls = new_decls })
   where
-     new_decls   = mkIfaceDecls ty_cls_dcls rule_dcls inst_dcls
-     inst_dcls   = map ifaceInstance (md_insts mod_details)
-     ty_cls_dcls = foldNameEnv ifaceTyCls [] (md_types mod_details)
-     rule_dcls   = map ifaceRule (md_rules mod_details)
+     must_write_hi_file Nothing      = False
+     must_write_hi_file (Just diffs) = ghci_mode /= Interactive
+               -- We must write a new .hi file if there are some changes
+               -- and we're not in interactive mode
+               -- maybe_diffs = 'Nothing' means that even the usages havn't changed, 
+               --     so there's no need to write a new interface file.  But even if 
+               --     the usages have changed, the module version may not have.
+
+     hi_file_path = unJust "mkFinalIface" (ml_hi_file location)
+     new_decls    = mkIfaceDecls ty_cls_dcls rule_dcls inst_dcls
+     inst_dcls    = map ifaceInstance (md_insts new_details)
+     ty_cls_dcls  = foldNameEnv ifaceTyCls [] (md_types new_details)
+     rule_dcls    = map ifaceRule (md_rules new_details)
+     orphan_mod   = isOrphanModule (mi_module new_iface) new_details
+
+write_diffs dflags new_iface Nothing
+  = do when (dopt Opt_D_dump_hi_diffs dflags) (printDump (text "INTERFACE UNCHANGED"))
+       dumpIfSet_dyn dflags Opt_D_dump_hi "UNCHANGED FINAL INTERFACE" (pprIface new_iface)
+
+write_diffs dflags new_iface (Just sdoc_diffs)
+  = do dumpIfSet_dyn dflags Opt_D_dump_hi_diffs "INTERFACE HAS CHANGED" sdoc_diffs
+       dumpIfSet_dyn dflags Opt_D_dump_hi "NEW FINAL INTERFACE" (pprIface new_iface)
 \end{code}
 
+\begin{code}
+isOrphanModule this_mod (ModDetails {md_insts = insts, md_rules = rules})
+  = any orphan_inst insts || any orphan_rule rules
+  where
+    orphan_inst dfun_id = no_locals (namesOfDFunHead (idType dfun_id))
+    orphan_rule rule    = no_locals (ruleLhsFreeNames rule)
+    no_locals names     = isEmptyNameSet (filterNameSet (nameIsLocalOrFrom this_mod) names)
+\end{code}
 
 \begin{code}
 ifaceTyCls :: TyThing -> [RenamedTyClDecl] -> [RenamedTyClDecl]
@@ -272,18 +228,20 @@ ifaceTyCls (AnId id) so_far
 
     id_type = idType id
     id_info = idInfo id
+    cg_info = idCgInfo id
+    arity_info = cgArity cg_info
+    caf_info   = cgCafInfo cg_info
 
     hs_idinfo | opt_OmitInterfacePragmas = []
              | otherwise                = arity_hsinfo  ++ caf_hsinfo  ++ cpr_hsinfo ++ 
                                           strict_hsinfo ++ wrkr_hsinfo ++ unfold_hsinfo
 
     ------------  Arity  --------------
-    arity_hsinfo = case arityInfo id_info of
-                       a@(ArityExactly n) -> [HsArity a]
-                       other              -> []
+    arity_hsinfo | arity_info == 0 = []
+                | otherwise       = [HsArity arity_info]
 
     ------------ Caf Info --------------
-    caf_hsinfo = case cafInfo id_info of
+    caf_hsinfo = case caf_info of
                   NoCafRefs -> [HsNoCafRefs]
                   otherwise -> []
 
@@ -301,8 +259,9 @@ ifaceTyCls (AnId id) so_far
     work_info   = workerInfo id_info
     has_worker  = case work_info of { HasWorker _ _ -> True; other -> False }
     wrkr_hsinfo = case work_info of
-                   HasWorker work_id wrap_arity -> [HsWorker (getName work_id)]
-                   NoWorker                     -> []
+                   HasWorker work_id wrap_arity -> 
+                       [HsWorker (getName work_id) wrap_arity]
+                   NoWorker -> []
 
     ------------  Unfolding  --------------
        -- The unfolding is redundant if there is a worker
@@ -456,6 +415,53 @@ diffDecls (VersionInfo { vers_module = old_mod_vers, vers_decls = old_decls_vers
 
 %************************************************************************
 %*                                                                     *
+\subsection{Writing ModDetails}
+%*                                                                     *
+%************************************************************************
+
+\begin{code}
+pprModDetails :: ModDetails -> SDoc
+pprModDetails (ModDetails { md_types = type_env, md_insts = dfun_ids, md_rules = rules })
+  = vcat [ dump_types dfun_ids type_env
+        , dump_insts dfun_ids
+        , dump_rules rules]
+         
+dump_types dfun_ids type_env
+  = text "TYPE SIGNATURES" $$ nest 4 (dump_sigs ids)
+  where
+    ids = [id | AnId id <- nameEnvElts type_env, want_sig id]
+    want_sig id | opt_PprStyle_Debug = True
+               | otherwise          = isLocalId id && 
+                                      isGlobalName (idName id) && 
+                                      not (id `elem` dfun_ids)
+       -- isLocalId ignores data constructors, records selectors etc
+       -- The isGlobalName ignores local dictionary and method bindings
+       -- that the type checker has invented.  User-defined things have
+       -- Global names.
+
+dump_insts []       = empty
+dump_insts dfun_ids = text "INSTANCES" $$ nest 4 (dump_sigs dfun_ids)
+
+dump_sigs ids
+       -- Print type signatures
+       -- Convert to HsType so that we get source-language style printing
+       -- And sort by RdrName
+  = vcat $ map ppr_sig $ sortLt lt_sig $
+    [ (toRdrName id, toHsType (idType id))
+    | id <- ids ]
+  where
+    lt_sig (n1,_) (n2,_) = n1 < n2
+    ppr_sig (n,t)        = ppr n <+> dcolon <+> ppr t
+
+dump_rules [] = empty
+dump_rules rs = vcat [ptext SLIT("{-# RULES"),
+                     nest 4 (vcat (map pprIdCoreRule rs)),
+                     ptext SLIT("#-}")]
+\end{code}
+
+
+%************************************************************************
+%*                                                                     *
 \subsection{Writing an interface file}
 %*                                                                     *
 %************************************************************************
@@ -487,7 +493,7 @@ pprIface iface
 
        , pprFixities (mi_fixities iface) (dcl_tycl decls)
        , pprIfaceDecls (vers_decls version_info) decls
-       , pprDeprecs (mi_deprecs iface)
+       , pprRulesAndDeprecs (dcl_rules decls) (mi_deprecs iface)
        ]
   where
     version_info = mi_version iface
@@ -550,7 +556,6 @@ pprUsage (m, has_orphans, is_boot, whats_imported)
 pprIfaceDecls version_map decls
   = vcat [ vcat [ppr i <+> semi | i <- dcl_insts decls]
         , vcat (map ppr_decl (dcl_tycl decls))
-        , pprRules (dcl_rules decls)
         ]
   where
     ppr_decl d  = ppr_vers d <+> ppr d <> semi
@@ -568,17 +573,23 @@ pprFixities fixity_map decls
           (n,_) <- tyClDeclNames d, 
           Just fix <- [lookupNameEnv fixity_map n]] <> semi
 
-pprRules []    = empty
-pprRules rules = hsep [ptext SLIT("{-## __R"), vcat (map ppr rules), ptext SLIT("##-}")]
-
-pprDeprecs NoDeprecs = empty
-pprDeprecs deprecs   = ptext SLIT("{-## __D") <+> guts <+> ptext SLIT("##-}")
-                    where
-                      guts = case deprecs of
-                               DeprecAll txt  -> doubleQuotes (ptext txt)
-                               DeprecSome env -> pp_deprecs env
-
-pp_deprecs env = vcat (punctuate semi (map pp_deprec (nameEnvElts env)))
-              where
-                pp_deprec (name, txt) = pprOcc name <+> doubleQuotes (ptext txt)
+-- Disgusting to print these two together, but that's 
+-- the way the interface parser currently expects them.
+pprRulesAndDeprecs [] NoDeprecs = empty
+pprRulesAndDeprecs rules deprecs
+  = ptext SLIT("{-##") <+> (pp_rules rules $$ pp_deprecs deprecs) <+> ptext SLIT("##-}")
+  where
+    pp_rules []    = empty
+    pp_rules rules = ptext SLIT("__R") <+> vcat (map ppr rules)
+
+    pp_deprecs NoDeprecs = empty
+    pp_deprecs deprecs   = ptext SLIT("__D") <+> guts
+                         where
+                           guts = case deprecs of
+                                       DeprecAll txt  -> doubleQuotes (ptext txt)
+                                       DeprecSome env -> ppr_deprec_env env
+
+ppr_deprec_env env = vcat (punctuate semi (map pp_deprec (nameEnvElts env)))
+                  where
+                    pp_deprec (name, txt) = pprOcc name <+> doubleQuotes (ptext txt)
 \end{code}