module Main ( main ) where
IMP_Ubiq(){-uitous-}
+IMPORT_1_3(IO(stderr,hPutStr,hClose,openFile,IOMode(..)))
import HsSyn
+import RdrHsSyn ( RdrName )
+import BasicTypes ( NewOrData(..) )
import ReadPrefix ( rdModule )
import Rename ( renameModule )
+import RnMonad ( ExportEnv )
+
import MkIface -- several functions
import TcModule ( typecheckModule )
-import Desugar ( deSugar, DsMatchContext, pprDsWarnings )
+import Desugar ( deSugar, pprDsWarnings
+#if __GLASGOW_HASKELL__ <= 200
+ , DsMatchContext
+#endif
+ )
import SimplCore ( core2core )
import CoreToStg ( topCoreBindsToStg )
+import StgSyn ( collectFinalStgBinders, pprStgBindings )
import SimplStg ( stg2stg )
import CodeGen ( codeGen )
#if ! OMIT_NATIVE_CODEGEN
import AbsCSyn ( absCNop, AbstractC )
import AbsCUtils ( flattenAbsC )
+import CoreUnfold ( Unfolding )
import Bag ( emptyBag, isEmptyBag )
import CmdLineOpts
-import ErrUtils ( pprBagOfErrors, ghcExit )
+import ErrUtils ( pprBagOfErrors, ghcExit, doIfSet, dumpIfSet )
import Maybes ( maybeToBool, MaybeErr(..) )
-import RdrHsSyn ( getRawExportees )
import Specialise ( SpecialiseData(..) )
-import StgSyn ( pprPlainStgBinding, GenStgBinding )
+import StgSyn ( GenStgBinding )
import TcInstUtil ( InstInfo )
import TyCon ( isDataTyCon )
import UniqSupply ( mkSplitUniqSupply )
import PprAbsC ( dumpRealC, writeRealC )
import PprCore ( pprCoreBinding )
-import PprStyle ( PprStyle(..) )
import Pretty
import Id ( GenId ) -- instances
-import Name ( Name, RdrName ) -- instances
+import Name ( Name ) -- instances
import PprType ( GenType, GenTyVar ) -- instances
-import RnHsSyn ( RnName ) -- instances
import TyVar ( GenTyVar ) -- instances
import Unique ( Unique ) -- instances
+
+import Outputable ( PprStyle(..), Outputable(..), pprDumpStyle, pprErrorsStyle )
+
\end{code}
\begin{code}
-main
- = hGetContents stdin >>= \ input_pgm ->
- let
- cmd_line_info = classifyOpts
- in
- doIt cmd_line_info input_pgm
+main =
+ _scc_ "main"
+ let
+ cmd_line_info = classifyOpts
+ in
+ doIt cmd_line_info
\end{code}
\begin{code}
-doIt :: ([CoreToDo], [StgToDo]) -> String -> IO ()
+doIt :: ([CoreToDo], [StgToDo]) -> IO ()
-doIt (core_cmds, stg_cmds) input_pgm
- = doDump opt_Verbose "Glasgow Haskell Compiler, version 2.01, for Haskell 1.3" "" >>
+doIt (core_cmds, stg_cmds)
+ = doIfSet opt_Verbose
+ (hPutStr stderr ("Glasgow Haskell Compiler, version " ++
+ show PROJECTVERSION ++
+ ", for Haskell 1.4\n")) >>
-- ******* READER
show_pass "Reader" >>
_scc_ "Reader"
rdModule >>= \ (mod_name, rdr_module) ->
- doDump opt_D_dump_rdr "Reader:"
- (pp_show (ppr pprStyle rdr_module)) >>
+ dumpIfSet opt_D_dump_rdr "Reader"
+ (ppr pprDumpStyle rdr_module) >>
- doDump opt_D_source_stats "\nSource Statistics:"
- (pp_show (ppSourceStats rdr_module)) >>
+ dumpIfSet opt_D_source_stats "Source Statistics"
+ (ppSourceStats rdr_module) >>
-- UniqueSupplies for later use (these are the only lower case uniques)
+-- _scc_ "spl-rn"
mkSplitUniqSupply 'r' >>= \ rn_uniqs -> -- renamer
+-- _scc_ "spl-tc"
mkSplitUniqSupply 'a' >>= \ tc_uniqs -> -- typechecker
+-- _scc_ "spl-ds"
mkSplitUniqSupply 'd' >>= \ ds_uniqs -> -- desugarer
+-- _scc_ "spl-sm"
mkSplitUniqSupply 's' >>= \ sm_uniqs -> -- core-to-core simplifier
+-- _scc_ "spl-c2s"
mkSplitUniqSupply 'c' >>= \ c2s_uniqs -> -- core-to-stg
+-- _scc_ "spl-st"
mkSplitUniqSupply 'g' >>= \ st_uniqs -> -- stg-to-stg passes
+-- _scc_ "spl-absc"
mkSplitUniqSupply 'f' >>= \ fl_uniqs -> -- absC flattener
+-- _scc_ "spl-ncg"
mkSplitUniqSupply 'n' >>= \ ncg_uniqs -> -- native-code generator
-- ******* RENAMER
show_pass "Renamer" >>
_scc_ "Renamer"
- renameModule rn_uniqs rdr_module >>=
- \ (rn_mod, rn_env, import_names,
- usage_stuff,
- rn_errs_bag, rn_warns_bag) ->
-
- if (not (isEmptyBag rn_errs_bag)) then
- hPutStr stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle rn_errs_bag))
- >> hPutStr stderr "\n" >>
- hPutStr stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle rn_warns_bag))
- >> hPutStr stderr "\n" >>
- ghcExit 1
-
- else -- No renaming errors ...
+ renameModule rn_uniqs rdr_module >>=
+ \ maybe_rn_stuff ->
+ case maybe_rn_stuff of {
+ Nothing -> -- Hurrah! Renamer reckons that there's no need to
+ -- go any further
+ return ();
+
+ Just (rn_mod, iface_file_stuff, rn_name_supply, imported_modules) ->
+ -- Oh well, we've got to recompile for real
- (if (isEmptyBag rn_warns_bag) then
- return ()
- else
- hPutStr stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle rn_warns_bag))
- >> hPutStr stderr "\n"
- ) >>
-
- doDump opt_D_dump_rn "Renamer:"
- (pp_show (ppr pprStyle rn_mod)) >>
-- Safely past renaming: we can start the interface file:
-- (the iface file is produced incrementally, as we have
-- the information that we need...; we use "iface<blah>")
-- "endIface" finishes the job.
- let
- (usages_map, version_info, instance_modules) = usage_stuff
- in
- startIface mod_name >>= \ if_handle ->
- ifaceUsages if_handle usages_map >>
- ifaceVersions if_handle version_info >>
- ifaceExportList if_handle rn_mod >>
- ifaceFixities if_handle rn_mod >>
- ifaceInstanceModules if_handle instance_modules >>
+ startIface mod_name >>= \ if_handle ->
+ ifaceMain if_handle iface_file_stuff >>
+
-- ******* TYPECHECKER
- show_pass "TypeCheck" >>
+ show_pass "TypeCheck" >>
_scc_ "TypeCheck"
- case (case (typecheckModule tc_uniqs {-idinfo_fm-} rn_env rn_mod) of
- Succeeded (stuff, warns)
- -> (emptyBag, warns, stuff)
- Failed (errs, warns)
- -> (errs, warns, error "tc_results"))
-
- of { (tc_errs_bag, tc_warns_bag, tc_results) ->
-
- if (not (isEmptyBag tc_errs_bag)) then
- hPutStr stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle tc_errs_bag))
- >> hPutStr stderr "\n" >>
- hPutStr stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle tc_warns_bag))
- >> hPutStr stderr "\n" >>
- ghcExit 1
-
- else ( -- No typechecking errors ...
-
- (if (isEmptyBag tc_warns_bag) then
- return ()
- else
- hPutStr stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle tc_warns_bag))
- >> hPutStr stderr "\n"
- ) >>
-
- case tc_results
- of { (typechecked_quint@(recsel_binds, class_binds, inst_binds, val_binds, const_binds),
- interface_stuff@(_,local_tycons,_,_),
- pragma_tycon_specs, ddump_deriv) ->
-
- doDump opt_D_dump_tc "Typechecked:"
- (pp_show (ppAboves [
- ppr pprStyle recsel_binds,
- ppr pprStyle class_binds,
- ppr pprStyle inst_binds,
- ppAboves (map (\ (i,e) -> ppr pprStyle (VarMonoBind i e)) const_binds),
- ppr pprStyle val_binds])) >>
-
- doDump opt_D_dump_deriv "Derived instances:"
- (pp_show (ddump_deriv pprStyle)) >>
-
- -- OK, now do the interface stuff that relies on typechecker output:
- ifaceDecls if_handle interface_stuff >>
- ifaceInstances if_handle interface_stuff >>
+ typecheckModule tc_uniqs rn_name_supply rn_mod >>= \ maybe_tc_stuff ->
+ case maybe_tc_stuff of {
+ Nothing -> ghcExit 1; -- Type checker failed
+
+ Just (all_binds,
+ local_tycons, local_classes, inst_info, pragma_tycon_specs,
+ ddump_deriv) ->
+
-- ******* DESUGARER
show_pass "DeSugar" >>
_scc_ "DeSugar"
- let
- (desugared,ds_warnings)
- = deSugar ds_uniqs mod_name typechecked_quint
- in
- (if isEmptyBag ds_warnings then
- return ()
- else
- hPutStr stderr (ppShow pprCols (pprDsWarnings pprErrorsStyle ds_warnings))
- >> hPutStr stderr "\n"
- ) >>
-
- doDump opt_D_dump_ds "Desugared:" (pp_show (ppAboves
- (map (pprCoreBinding pprStyle) desugared)))
- >>
+ deSugar ds_uniqs mod_name all_binds >>= \ desugared ->
+
- -- ******* CORE-TO-CORE SIMPLIFICATION (NB: I/O op)
+ -- ******* CORE-TO-CORE SIMPLIFICATION
show_pass "Core2Core" >>
_scc_ "Core2Core"
let
local_data_tycons = filter isDataTyCon local_tycons
in
- core2core core_cmds mod_name pprStyle
+ core2core core_cmds mod_name
sm_uniqs local_data_tycons pragma_tycon_specs desugared
>>=
+ \ (simplified,
+ SpecData _ _ _ gen_data_tycons all_tycon_specs _ _ _) ->
- \ (simplified, inlinings_env,
- SpecData _ _ _ gen_tycons all_tycon_specs _ _ _) ->
-
- doDump opt_D_dump_simpl "Simplified:" (pp_show (ppAboves
- (map (pprCoreBinding pprStyle) simplified)))
- >>
-- ******* STG-TO-STG SIMPLIFICATION
show_pass "Core2Stg" >>
show_pass "Stg2Stg" >>
_scc_ "Stg2Stg"
- stg2stg stg_cmds mod_name pprStyle st_uniqs stg_binds
+ stg2stg stg_cmds mod_name st_uniqs stg_binds
>>=
-
\ (stg_binds2, cost_centre_info) ->
- doDump opt_D_dump_stg "STG syntax:"
- (pp_show (ppAboves (map (pprPlainStgBinding pprStyle) stg_binds2)))
+ dumpIfSet opt_D_dump_stg "STG syntax:"
+ (pprStgBindings pprDumpStyle stg_binds2)
>>
+ -- Dump instance decls and type signatures into the interface file
+ let
+ final_ids = collectFinalStgBinders stg_binds2
+ in
+ _scc_ "Interface"
+ ifaceDecls if_handle local_tycons local_classes inst_info final_ids simplified >>
+ endIface if_handle >>
-- We are definitely done w/ interface-file stuff at this point:
-- (See comments near call to "startIface".)
- endIface if_handle >>
+
-- ******* "ABSTRACT", THEN "FLAT", THEN *REAL* C!
show_pass "CodeGen" >>
_scc_ "CodeGen"
let
- abstractC = codeGen mod_name -- module name for CC labelling
+ abstractC = codeGen mod_name -- module name for CC labelling
cost_centre_info
- import_names -- import names for CC registering
- gen_tycons -- type constructors generated locally
- all_tycon_specs -- tycon specialisations
+ imported_modules -- import names for CC registering
+ gen_data_tycons -- type constructors generated locally
+ all_tycon_specs -- tycon specialisations
stg_binds2
flat_abstractC = flattenAbsC fl_uniqs abstractC
in
- doDump opt_D_dump_absC "Abstract C:"
+ dumpIfSet opt_D_dump_absC "Abstract C"
(dumpRealC abstractC) >>
- doDump opt_D_dump_flatC "Flat Abstract C:"
+ dumpIfSet opt_D_dump_flatC "Flat Abstract C"
(dumpRealC flat_abstractC) >>
+ show_pass "CodeOutput" >>
+ _scc_ "CodeOutput"
-- You can have C (c_output) or assembly-language (ncg_output),
-- but not both. [Allowing for both gives a space leak on
-- flat_abstractC. WDP 94/10]
#endif
in
- doDump opt_D_dump_asm "" ncg_output_d >>
- doOutput opt_ProduceS ncg_output_w >>
+ dumpIfSet opt_D_dump_asm "Asm code" ncg_output_d >>
+ doOutput opt_ProduceS ncg_output_w >>
- doDump opt_D_dump_realC "" c_output_d >>
- doOutput opt_ProduceC c_output_w >>
+ dumpIfSet opt_D_dump_realC "Real C" c_output_d >>
+ doOutput opt_ProduceC c_output_w >>
ghcExit 0
- } ) }
+ } }
where
-------------------------------------------------------------
- -- ****** printing styles and column width:
-
- pprCols = (80 :: Int) -- could make configurable
-
- (pprStyle, pprErrorsStyle)
- = if opt_PprStyle_All then
- (PprShowAll, PprShowAll)
- else if opt_PprStyle_Debug then
- (PprDebug, PprDebug)
- else if opt_PprStyle_User then
- (PprForUser, PprForUser)
- else -- defaults...
- (PprDebug, PprForUser)
-
- pp_show p = ppShow {-WAS:pprCols-}10000{-random-} p
-
- -------------------------------------------------------------
-- ****** help functions:
show_pass
io_action handle >>
hClose handle
- doDump switch hdr string
- = if switch
- then hPutStr stderr hdr >>
- hPutStr stderr ('\n': string) >>
- hPutStr stderr "\n"
- else return ()
-
-ppSourceStats (HsModule name version exports imports fixities typedecls typesigs
- classdecls instdecls instsigs defdecls binds
- [{-no sigs-}] src_loc)
- = ppAboves (map pp_val
+ppSourceStats (HsModule name version exports imports fixities decls src_loc)
+ = vcat (map pp_val
[("ExportAll ", export_all), -- 1 if no export list
("ExportDecls ", export_ds),
("ExportModules ", export_ms),
(" ImpPartial ", import_partial),
(" ImpHiding ", import_hiding),
("FixityDecls ", fixity_ds),
- ("DefaultDecls ", defalut_ds),
+ ("DefaultDecls ", default_ds),
("TypeDecls ", type_ds),
("DataDecls ", data_ds),
("NewTypeDecls ", newt_ds),
("FunBinds ", fn_bind_ds),
("InlineMeths ", method_inlines),
("InlineBinds ", bind_inlines),
- ("SpecialisedData ", data_specs),
- ("SpecialisedInsts ", inst_specs),
+-- ("SpecialisedData ", data_specs),
+-- ("SpecialisedInsts ", inst_specs),
("SpecialisedMeths ", method_specs),
("SpecialisedBinds ", bind_specs)
])
where
- pp_val (str, 0) = ppNil
- pp_val (str, n) = ppBesides [ppStr str, ppInt n]
-
- (export_decls, export_mods) = getRawExportees exports
- type_decls = filter is_type_decl typedecls
- data_decls = filter is_data_decl typedecls
- newt_decls = filter is_newt_decl typedecls
-
- export_ds = length export_decls
- export_ms = length export_mods
- export_all = if export_ds == 0 && export_ms == 0 then 1 else 0
-
- fixity_ds = length fixities
- defalut_ds = length defdecls
- type_ds = length type_decls
- data_ds = length data_decls
- newt_ds = length newt_decls
- class_ds = length classdecls
- inst_ds = length instdecls
+ pp_val (str, 0) = empty
+ pp_val (str, n) = hcat [text str, int n]
+
+ fixity_ds = length fixities
+ type_decls = [d | TyD d@(TySynonym _ _ _ _) <- decls]
+ data_decls = [d | TyD d@(TyData DataType _ _ _ _ _ _ _) <- decls]
+ newt_decls = [d | TyD d@(TyData NewType _ _ _ _ _ _ _) <- decls]
+ type_ds = length type_decls
+ data_ds = length data_decls
+ newt_ds = length newt_decls
+ class_decls = [d | ClD d <- decls]
+ class_ds = length class_decls
+ inst_decls = [d | InstD d <- decls]
+ inst_ds = length inst_decls
+ default_ds = length [() | DefD _ <- decls]
+ val_decls = [d | ValD d <- decls]
+
+ real_exports = case exports of { Nothing -> []; Just es -> es }
+ n_exports = length real_exports
+ export_ms = length [() | IEModuleContents _ <- real_exports]
+ export_ds = n_exports - export_ms
+ export_all = case exports of { Nothing -> 1; other -> 0 }
(val_bind_ds, fn_bind_ds, bind_tys, bind_specs, bind_inlines)
- = count_binds binds
+ = count_binds (foldr ThenBinds EmptyBinds val_decls)
(import_no, import_qual, import_as, import_all, import_partial, import_hiding)
= foldr add6 (0,0,0,0,0,0) (map import_info imports)
(data_constrs, data_derivs)
= foldr add2 (0,0) (map data_info (newt_decls ++ data_decls))
(class_method_ds, default_method_ds)
- = foldr add2 (0,0) (map class_info classdecls)
+ = foldr add2 (0,0) (map class_info class_decls)
(inst_method_ds, method_specs, method_inlines)
- = foldr add3 (0,0,0) (map inst_info instdecls)
+ = foldr add3 (0,0,0) (map inst_info inst_decls)
- data_specs = length typesigs
- inst_specs = length instsigs
count_binds EmptyBinds = (0,0,0,0,0)
count_binds (ThenBinds b1 b2) = count_binds b1 `add5` count_binds b2
- count_binds (SingleBind b) = case count_bind b of
- (vs,fs) -> (vs,fs,0,0,0)
- count_binds (BindWith b sigs) = case (count_bind b, count_sigs sigs) of
- ((vs,fs),(ts,_,ss,is)) -> (vs,fs,ts,ss,is)
-
- count_bind EmptyBind = (0,0)
- count_bind (NonRecBind b) = count_monobinds b
- count_bind (RecBind b) = count_monobinds b
+ count_binds (MonoBind b sigs _) = case (count_monobinds b, count_sigs sigs) of
+ ((vs,fs),(ts,_,ss,is)) -> (vs,fs,ts,ss,is)
count_monobinds EmptyMonoBinds = (0,0)
count_monobinds (AndMonoBinds b1 b2) = count_monobinds b1 `add2` count_monobinds b2
count_sigs sigs = foldr add4 (0,0,0,0) (map sig_info sigs)
- sig_info (Sig _ _ _ _) = (1,0,0,0)
+ sig_info (Sig _ _ _) = (1,0,0,0)
sig_info (ClassOpSig _ _ _ _) = (0,1,0,0)
sig_info (SpecSig _ _ _ _) = (0,0,1,0)
sig_info (InlineSig _ _) = (0,0,0,1)
sig_info _ = (0,0,0,0)
- import_info (ImportDecl _ qual as spec _)
+ import_info (ImportDecl _ qual _ as spec _)
= add6 (1, qual_info qual, as_info as, 0,0,0) (spec_info spec)
qual_info False = 0
qual_info True = 1
spec_info (Just (False, _)) = (0,0,0,0,1,0)
spec_info (Just (True, _)) = (0,0,0,0,0,1)
- data_info (TyData _ _ _ constrs derivs _ _)
+ data_info (TyData _ _ _ _ constrs derivs _ _)
= (length constrs, case derivs of {Nothing -> 0; Just ds -> length ds})
- data_info (TyNew _ _ _ constr derivs _ _)
- = (length constr, case derivs of {Nothing -> 0; Just ds -> length ds})
class_info (ClassDecl _ _ _ meth_sigs def_meths _ _)
= case count_sigs meth_sigs of
(_,classops,_,_) ->
(classops, addpr (count_monobinds def_meths))
- inst_info (InstDecl _ _ inst_meths _ _ inst_sigs _ _)
+ inst_info (InstDecl _ inst_meths inst_sigs _ _)
= case count_sigs inst_sigs of
(_,_,ss,is) ->
(addpr (count_monobinds inst_meths), ss, is)
- is_type_decl (TySynonym _ _ _ _) = True
- is_type_decl _ = False
- is_data_decl (TyData _ _ _ _ _ _ _) = True
- is_data_decl _ = False
- is_newt_decl (TyNew _ _ _ _ _ _ _) = True
- is_newt_decl _ = False
-
addpr (x,y) = x+y
add1 x1 y1 = x1+y1
add2 (x1,x2) (y1,y2) = (x1+y1,x2+y2)