Keep track of family instance modules
[ghc-hetmet.git] / compiler / deSugar / Desugar.lhs
index dc0e124..ab4ee74 100644 (file)
@@ -1,48 +1,49 @@
 %
+% (c) The University of Glasgow 2006
 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
 %
-\section[Desugar]{@deSugar@: the main function}
+
+The Desugarer: turning HsSyn into Core.
 
 \begin{code}
 module Desugar ( deSugar, deSugarExpr ) where
 
 #include "HsVersions.h"
 
-import DynFlags                ( DynFlag(..), DynFlags(..), dopt, GhcMode(..) )
-import StaticFlags     ( opt_SccProfilingOn )
-import DriverPhases    ( isHsBoot )
-import HscTypes                ( ModGuts(..), HscEnv(..), 
-                         Dependencies(..), ForeignStubs(..), TypeEnv, IsBootInterface )
-import HsSyn           ( RuleDecl(..), RuleBndr(..), LHsExpr, LRuleDecl )
-import TcRnTypes       ( TcGblEnv(..), ImportAvails(..) )
-import MkIface         ( mkUsageInfo )
-import Id              ( Id, setIdExported, idName )
-import Name            ( Name, isExternalName, nameIsLocalOrFrom, nameOccName )
+import DynFlags
+import StaticFlags
+import HscTypes
+import HsSyn
+import TcRnTypes
+import MkIface
+import Id
+import Name
 import CoreSyn
-import PprCore         ( pprRules, pprCoreExpr )
+import PprCore
 import DsMonad
-import DsExpr          ( dsLExpr )
-import DsBinds         ( dsTopLHsBinds, decomposeRuleLhs, AutoScc(..) )
-import DsForeign       ( dsForeigns )
+import DsExpr
+import DsBinds
+import DsForeign
 import DsExpr          ()      -- Forces DsExpr to be compiled; DsBinds only
                                -- depends on DsExpr.hi-boot.
 import Module
-import UniqFM          ( eltsUFM, delFromUFM )
-import PackageConfig   ( thPackageId )
-import RdrName         ( GlobalRdrEnv )
+import UniqFM
+import PackageConfig
+import RdrName
 import NameSet
 import VarSet
-import Rules           ( roughTopNames )
-import CoreLint                ( showPass, endPass )
-import CoreFVs         ( ruleRhsFreeVars, exprsFreeNames )
-import ErrUtils                ( doIfSet, dumpIfSet_dyn )
-import ListSetOps      ( insertList )
+import Rules
+import CoreLint
+import CoreFVs
+import ErrUtils
+import ListSetOps
 import Outputable
-import SrcLoc          ( Located(..) )
-import DATA_IOREF      ( readIORef )
-import Maybes          ( catMaybes )
+import SrcLoc
+import Maybes
 import FastString
-import Util            ( sortLe )
+import Util
+
+import Data.IORef
 \end{code}
 
 %************************************************************************
@@ -71,10 +72,14 @@ deSugar hsc_env
                            tcg_binds     = binds,
                            tcg_fords     = fords,
                            tcg_rules     = rules,
-                           tcg_insts     = insts })
+                           tcg_insts     = insts,
+                           tcg_fam_insts = fam_insts })
   = do { showPass dflags "Desugar"
 
        -- Desugar the program
+        ; let export_set = availsToNameSet exports
+       ; let auto_scc = mkAutoScc mod export_set
+
        ; mb_res <- case ghcMode dflags of
                     JustTypecheck -> return (Just ([], [], NoStubs))
                     _             -> initDs hsc_env mod rdr_env type_env $ do
@@ -91,8 +96,8 @@ deSugar hsc_env
 
        {       -- Add export flags to bindings
          keep_alive <- readIORef keep_var
-       ; let final_prs = addExportFlags ghci_mode exports keep_alive 
-                                        all_prs ds_rules
+       ; let final_prs = addExportFlags ghci_mode export_set
+                                 keep_alive all_prs ds_rules
              ds_binds  = [Rec final_prs]
        -- Notice that we put the whole lot in a big Rec, even the foreign binds
        -- When compiling PrelFloat, which defines data Float = F# Float#
@@ -134,26 +139,28 @@ deSugar hsc_env
             le_dep_mod :: (ModuleName, IsBootInterface) -> (ModuleName, IsBootInterface) -> Bool        
             le_dep_mod (m1,_) (m2,_) = moduleNameFS m1 <= moduleNameFS m2
 
-            deps = Deps { dep_mods  = sortLe le_dep_mod dep_mods,
-                          dep_pkgs  = sortLe (<=)   pkgs,      
-                          dep_orphs = sortLe le_mod (imp_orphs imports) }
+            deps = Deps { dep_mods   = sortLe le_dep_mod dep_mods,
+                          dep_pkgs   = sortLe (<=)   pkgs,     
+                          dep_orphs  = sortLe le_mod (imp_orphs  imports),
+                          dep_finsts = sortLe le_mod (imp_finsts imports) }
                -- sort to get into canonical order
 
             mod_guts = ModGuts {       
-               mg_module   = mod,
-               mg_boot     = isHsBoot hsc_src,
-               mg_exports  = exports,
-               mg_deps     = deps,
-               mg_usages   = usages,
-               mg_dir_imps = [m | (m,_,_) <- moduleEnvElts dir_imp_mods],
-               mg_rdr_env  = rdr_env,
-               mg_fix_env  = fix_env,
-               mg_deprecs  = deprecs,
-               mg_types    = type_env,
-               mg_insts    = insts,
-               mg_rules    = ds_rules,
-               mg_binds    = ds_binds,
-               mg_foreign  = ds_fords }
+               mg_module    = mod,
+               mg_boot      = isHsBoot hsc_src,
+               mg_exports   = exports,
+               mg_deps      = deps,
+               mg_usages    = usages,
+               mg_dir_imps  = [m | (m,_,_) <- moduleEnvElts dir_imp_mods],
+               mg_rdr_env   = rdr_env,
+               mg_fix_env   = fix_env,
+               mg_deprecs   = deprecs,
+               mg_types     = type_env,
+               mg_insts     = insts,
+               mg_fam_insts = fam_insts,
+               mg_rules     = ds_rules,
+               mg_binds     = ds_binds,
+               mg_foreign   = ds_fords }
        
         ; return (Just mod_guts)
        }}}
@@ -161,8 +168,18 @@ deSugar hsc_env
   where
     dflags    = hsc_dflags hsc_env
     ghci_mode = ghcMode (hsc_dflags hsc_env)
-    auto_scc | opt_SccProfilingOn = TopLevel
-            | otherwise          = NoSccs
+
+mkAutoScc :: Module -> NameSet -> AutoScc
+mkAutoScc mod exports
+  | not opt_SccProfilingOn     -- No profiling
+  = NoSccs             
+  | opt_AutoSccsOnAllToplevs   -- Add auto-scc on all top-level things
+  = AddSccs mod (\id -> True)
+  | opt_AutoSccsOnExportedToplevs      -- Only on exported things
+  = AddSccs mod (\id -> idName id `elemNameSet` exports)
+  | otherwise
+  = NoSccs
+
 
 deSugarExpr :: HscEnv
            -> Module -> GlobalRdrEnv -> TypeEnv 
@@ -270,6 +287,9 @@ dsRule mod in_scope (L loc (HsRule name act vars lhs tv_lhs rhs fv_rhs))
              lhs_names = fn_name : nameSetToList (exprsFreeNames args)
                -- No need to delete bndrs, because
                -- exprsFreeNames finds only External names
+
+               -- A rule is an orphan only if none of the variables
+               -- mentioned on its left-hand side are locally defined
              orph = case filter (nameIsLocalOrFrom mod) lhs_names of
                        (n:ns) -> Just (nameOccName n)
                        []     -> Nothing