X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Fmain%2FMain.lhs;h=432a2f2fa57de23f2dd5c970513c758fe695c336;hb=c837a7f2f12d3f340454b9525be188f3ec27c8cf;hp=9d2071362e5b7a8910b37fb6248d534e185ddf31;hpb=f9120c200bcf613b58d742802172fb4c08171f0d;p=ghc-hetmet.git diff --git a/ghc/compiler/main/Main.lhs b/ghc/compiler/main/Main.lhs index 9d20713..432a2f2 100644 --- a/ghc/compiler/main/Main.lhs +++ b/ghc/compiler/main/Main.lhs @@ -1,360 +1,230 @@ % -% (c) The GRASP/AQUA Project, Glasgow University, 1993-1996 +% (c) The GRASP/AQUA Project, Glasgow University, 1993-1998 % \section[GHC_Main]{Main driver for Glasgow Haskell compiler} \begin{code} -#include "HsVersions.h" - module Main ( main ) where -import Ubiq{-uitous-} - -import PreludeGlaST ( thenPrimIO, _FILE{-instances-} ) -- ToDo: STOP using this... +#include "HsVersions.h" -import MainMonad +import IO ( hPutStr, stderr ) import HsSyn +import BasicTypes ( NewOrData(..) ) + +import RdrHsSyn ( RdrNameHsModule ) +import FastString ( mkFastCharString, unpackFS ) +import StringBuffer ( hGetStringBuffer ) +import Parser ( parse ) +import Lex ( PState(..), P, ParseResult(..) ) +import SrcLoc ( mkSrcLoc ) -import ReadPrefix ( rdModule ) import Rename ( renameModule ) -import Typecheck ( typecheckModule, InstInfo ) -import Desugar ( deSugar, DsMatchContext, pprDsWarnings ) + +import MkIface ( startIface, ifaceDecls, endIface ) +import TcModule ( TcResults(..), typecheckModule ) +import Desugar ( deSugar ) import SimplCore ( core2core ) +import CoreLint ( endPass ) +import CoreSyn ( coreBindsSize ) +import CoreTidy ( tidyCorePgm ) import CoreToStg ( topCoreBindsToStg ) +import StgSyn ( collectFinalStgBinders, pprStgBindings ) import SimplStg ( stg2stg ) import CodeGen ( codeGen ) -#if ! OMIT_NATIVE_CODEGEN -import AsmCodeGen ( dumpRealAsm, writeRealAsm ) -#endif +import CodeOutput ( codeOutput ) -import AbsCSyn ( absCNop, AbstractC ) -import AbsCUtils ( flattenAbsC ) -import Bag ( emptyBag, isEmptyBag ) +import Module ( ModuleName, moduleNameUserString ) +import AbsCSyn ( absCNop ) import CmdLineOpts -import ErrUtils ( pprBagOfErrors ) +import ErrUtils ( ghcExit, doIfSet, dumpIfSet ) import Maybes ( maybeToBool, MaybeErr(..) ) -import PrelInfo ( builtinNameInfo ) -import RdrHsSyn ( getRawExportees ) -import Specialise ( SpecialiseData(..) ) -import StgSyn ( pprPlainStgBinding, GenStgBinding ) - -import PprAbsC ( dumpRealC, writeRealC ) -import PprCore ( pprCoreBinding ) -import PprStyle ( PprStyle(..) ) -import Pretty - -import Id ( GenId ) -- instances -import Name ( Name, RdrName ) -- instances -import PprType ( GenType, GenTyVar ) -- instances -import RnHsSyn ( RnName ) -- instances -import TyVar ( GenTyVar ) -- instances -import Unique ( Unique ) -- instances - -{- ---import MkIface ( mkInterface ) --} +import TyCon ( isDataTyCon ) +import Class ( classTyCon ) +import UniqSupply ( mkSplitUniqSupply ) + +import FiniteMap ( emptyFM ) +import Outputable +import Char ( isSpace ) +#if REPORT_TO_MOTHERLODE && __GLASGOW_HASKELL__ >= 303 +import SocketPrim +import BSD +import IOExts ( unsafePerformIO ) +import NativeInfo ( os, arch ) +#endif \end{code} \begin{code} -main - = readMn stdin `thenMn` \ input_pgm -> - let - cmd_line_info = classifyOpts - in - doIt cmd_line_info input_pgm +main = + -- _scc_ "main" + doIt classifyOpts \end{code} \begin{code} -doIt :: ([CoreToDo], [StgToDo]) -> String -> MainIO () +parseModule :: IO (ModuleName, RdrNameHsModule) +parseModule = do + buf <- hGetStringBuffer True{-expand tabs-} (unpackFS src_filename) + case parse buf PState{ bol = 0#, atbol = 1#, + context = [], glasgow_exts = glaexts, + loc = mkSrcLoc src_filename 1 } of + + PFailed err -> do + printErrs err + ghcExit 1 + return (error "parseModule") -- just to get the types right + + POk _ m@(HsModule mod _ _ _ _ _) -> + return (mod, m) + where + glaexts | opt_GlasgowExts = 1# + | otherwise = 0# +\end{code} -doIt (core_cmds, stg_cmds) input_pgm - = doDump opt_Verbose "Glasgow Haskell Compiler, version 1.3-xx" "" - `thenMn_` +\begin{code} +doIt :: ([CoreToDo], [StgToDo]) -> IO () - -- ******* READER - show_pass "Reader" `thenMn_` - rdModule `thenMn` +doIt (core_cmds, stg_cmds) + = doIfSet opt_Verbose + (hPutStr stderr "Glasgow Haskell Compiler, version " >> + hPutStr stderr compiler_version >> + hPutStr stderr ", for Haskell 98, compiled by GHC version " >> + hPutStr stderr booter_version >> + hPutStr stderr "\n") >> - \ (mod_name, rdr_module) -> + -------------------------- Reader ---------------- + show_pass "Parser" >> + _scc_ "Parser" + parseModule >>= \ (mod_name, rdr_module) -> - let - -- reader things used much later - ds_mod_name = mod_name - if_mod_name = mod_name - co_mod_name = mod_name - st_mod_name = mod_name - cc_mod_name = mod_name - in - doDump opt_D_dump_rdr "Reader:" - (pp_show (ppr pprStyle rdr_module)) `thenMn_` + dumpIfSet opt_D_dump_parsed "Parser" (ppr rdr_module) >> - doDump opt_D_source_stats "\nSource Statistics:" - (pp_show (ppSourceStats rdr_module)) `thenMn_` + dumpIfSet opt_D_source_stats "Source Statistics" + (ppSourceStats False rdr_module) >> -- UniqueSupplies for later use (these are the only lower case uniques) - getSplitUniqSupplyMn 'r' `thenMn` \ rn_uniqs -> -- renamer - getSplitUniqSupplyMn 't' `thenMn` \ tc_uniqs -> -- typechecker - getSplitUniqSupplyMn 'd' `thenMn` \ ds_uniqs -> -- desugarer - getSplitUniqSupplyMn 's' `thenMn` \ sm_uniqs -> -- core-to-core simplifier - getSplitUniqSupplyMn 'c' `thenMn` \ c2s_uniqs -> -- core-to-stg - getSplitUniqSupplyMn 'g' `thenMn` \ st_uniqs -> -- stg-to-stg passes - getSplitUniqSupplyMn 'f' `thenMn` \ fl_uniqs -> -- absC flattener - getSplitUniqSupplyMn 'n' `thenMn` \ ncg_uniqs -> -- native-code generator - - -- ******* RENAMER - show_pass "Renamer" `thenMn_` - - case builtinNameInfo - of { (wiredin_fm, key_fm, idinfo_fm) -> - - renameModule wiredin_fm key_fm rn_uniqs rdr_module `thenMn` - \ (rn_mod, import_names, - version_info, instance_modules, - rn_errs_bag, rn_warns_bag) -> - - if (not (isEmptyBag rn_errs_bag)) then - writeMn stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle rn_errs_bag)) - `thenMn_` writeMn stderr "\n" `thenMn_` - writeMn stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle rn_warns_bag)) - `thenMn_` writeMn stderr "\n" `thenMn_` - exitMn 1 - - else -- No renaming errors ... - - (if (isEmptyBag rn_warns_bag) then - returnMn () - else - writeMn stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle rn_warns_bag)) - `thenMn_` writeMn stderr "\n" - ) `thenMn_` - - doDump opt_D_dump_rn "Renamer:" - (pp_show (ppr pprStyle rn_mod)) `thenMn_` - - exitMn 0 -{- LATER ... - - -- ******* TYPECHECKER - show_pass "TypeCheck" `thenMn_` - case (case (typecheckModule tc_uniqs idinfo_fm rn_info 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 - writeMn stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle tc_errs_bag)) - `thenMn_` writeMn stderr "\n" `thenMn_` - writeMn stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle tc_warns_bag)) - `thenMn_` writeMn stderr "\n" `thenMn_` - exitMn 1 - - else ( -- No typechecking errors ... - - (if (isEmptyBag tc_warns_bag) then - returnMn () - else - writeMn stderr (ppShow pprCols (pprBagOfErrors pprErrorsStyle tc_warns_bag)) - `thenMn_` writeMn stderr "\n" - ) `thenMn_` - - case tc_results - of { (typechecked_quint@(recsel_binds, class_binds, inst_binds, val_binds, const_binds), - interface_stuff@(_,_,_,_,_), -- @-pat just for strictness... - (local_tycons,local_classes), 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])) `thenMn_` - - doDump opt_D_dump_deriv "Derived instances:" - (pp_show (ddump_deriv pprStyle)) `thenMn_` - - -- ******* DESUGARER - show_pass "DeSugar" `thenMn_` - let - (desugared,ds_warnings) - = deSugar ds_uniqs ds_mod_name typechecked_quint - in - (if isEmptyBag ds_warnings then - returnMn () - else - writeMn stderr (ppShow pprCols (pprDsWarnings pprErrorsStyle ds_warnings)) - `thenMn_` writeMn stderr "\n" - ) `thenMn_` - - doDump opt_D_dump_ds "Desugared:" (pp_show (ppAboves - (map (pprCoreBinding pprStyle) desugared))) - `thenMn_` - - -- ******* CORE-TO-CORE SIMPLIFICATION (NB: I/O op) - core2core core_cmds co_mod_name pprStyle - sm_uniqs local_tycons pragma_tycon_specs desugared - `thenMn` - - \ (simplified, inlinings_env, - SpecData _ _ _ gen_tycons all_tycon_specs _ _ _) -> - - doDump opt_D_dump_simpl "Simplified:" (pp_show (ppAboves - (map (pprCoreBinding pprStyle) simplified))) - `thenMn_` - - -- ******* STG-TO-STG SIMPLIFICATION - show_pass "Core2Stg" `thenMn_` + mkSplitUniqSupply 'r' >>= \ rn_uniqs -> -- renamer + mkSplitUniqSupply 'a' >>= \ tc_uniqs -> -- typechecker + mkSplitUniqSupply 'd' >>= \ ds_uniqs -> -- desugarer + mkSplitUniqSupply 'r' >>= \ ru_uniqs -> -- rules + mkSplitUniqSupply 'c' >>= \ c2s_uniqs -> -- core-to-stg + mkSplitUniqSupply 'u' >>= \ tidy_uniqs -> -- tidy up + mkSplitUniqSupply 'g' >>= \ st_uniqs -> -- stg-to-stg passes + mkSplitUniqSupply 'n' >>= \ ncg_uniqs -> -- native-code generator + + -------------------------- Rename ---------------- + show_pass "Renamer" >> + _scc_ "Renamer" + + 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 + reportCompile mod_name "Compilation NOT required!" >> + return (); + + Just (this_mod, rn_mod, iface_file_stuff, rn_name_supply, imported_modules) -> + -- Oh well, we've got to recompile for real + + + -------------------------- Start interface file ---------------- + -- 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") + -- "endIface" finishes the job. + startIface this_mod iface_file_stuff >>= \ if_handle -> + + + -------------------------- Typechecking ---------------- + show_pass "TypeCheck" >> + _scc_ "TypeCheck" + typecheckModule tc_uniqs rn_name_supply + iface_file_stuff rn_mod >>= \ maybe_tc_stuff -> + case maybe_tc_stuff of { + Nothing -> ghcExit 1; -- Type checker failed + + Just (tc_results@(TcResults {tc_tycons = local_tycons, + tc_classes = local_classes, + tc_insts = inst_info })) -> + + + -------------------------- Desugaring ---------------- + _scc_ "DeSugar" + deSugar this_mod ds_uniqs tc_results >>= \ (desugared, rules, h_code, c_code) -> + + + -------------------------- Main Core-language transformations ---------------- + _scc_ "Core2Core" + core2core core_cmds desugared rules >>= \ (simplified, imp_rule_ids) -> + + -- Do the final tidy-up + tidyCorePgm tidy_uniqs this_mod + simplified imp_rule_ids >>= \ (tidy_binds, tidy_imp_rule_ids) -> + + + -------------------------- Convert to STG code ------------------------------- + show_pass "Core2Stg" >> + _scc_ "Core2Stg" let - stg_binds = topCoreBindsToStg c2s_uniqs simplified + stg_binds = topCoreBindsToStg c2s_uniqs tidy_binds in - show_pass "Stg2Stg" `thenMn_` - stg2stg stg_cmds st_mod_name pprStyle st_uniqs stg_binds - `thenMn` - - \ (stg_binds2, cost_centre_info) -> - - doDump opt_D_dump_stg "STG syntax:" - (pp_show (ppAboves (map (pprPlainStgBinding pprStyle) stg_binds2))) - `thenMn_` - -{- LATER ... - -- ******* INTERFACE GENERATION (needs STG output) -{- let - mod_name = "_TestName_" - export_list_fns = (\ x -> False, \ x -> False) - inlinings_env = nullIdEnv - fixities = [] - if_global_ids = [] - if_ce = nullCE - if_tce = nullTCE - if_inst_info = emptyBag - in --} + -------------------------- Simplify STG code ------------------------------- + show_pass "Stg2Stg" >> + _scc_ "Stg2Stg" + stg2stg stg_cmds this_mod st_uniqs stg_binds >>= \ (stg_binds2, cost_centre_info) -> - show_pass "Interface" `thenMn_` - let - mod_interface - = mkInterface if_mod_name export_list_fns - inlinings_env all_tycon_specs - interface_stuff - stg_binds2 - in - doOutput opt_ProduceHi ( \ file -> - ppAppendFile file 1000{-pprCols-} mod_interface ) - `thenMn_` --} - -- ******* "ABSTRACT", THEN "FLAT", THEN *REAL* C! - show_pass "CodeGen" `thenMn_` + -------------------------- Interface file ------------------------------- + -- Dump instance decls and type signatures into the interface file + _scc_ "Interface" let - abstractC = codeGen cc_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 - stg_binds2 - - flat_abstractC = flattenAbsC fl_uniqs abstractC + final_ids = collectFinalStgBinders (map fst stg_binds2) in - doDump opt_D_dump_absC "Abstract C:" - (dumpRealC abstractC) `thenMn_` + coreBindsSize tidy_binds `seq` +-- TEMP: the above call zaps some space usage allocated by the +-- simplifier, which for reasons I don't understand, persists +-- thoroughout code generation - doDump opt_D_dump_flatC "Flat Abstract C:" - (dumpRealC flat_abstractC) `thenMn_` + ifaceDecls if_handle local_tycons local_classes + inst_info final_ids tidy_binds imp_rule_ids >> + endIface if_handle >> + -- We are definitely done w/ interface-file stuff at this point: + -- (See comments near call to "startIface".) - -- 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] - let - (flat_absC_c, flat_absC_ncg) = - case (maybeToBool opt_ProduceC || opt_D_dump_realC, - maybeToBool opt_ProduceS || opt_D_dump_asm) of - (True, False) -> (flat_abstractC, absCNop) - (False, True) -> (absCNop, flat_abstractC) - (False, False) -> (absCNop, absCNop) - (True, True) -> error "ERROR: Can't do both .hc and .s at the same time" - - c_output_d = dumpRealC flat_absC_c - c_output_w = (\ f -> writeRealC f flat_absC_c) - -#if OMIT_NATIVE_CODEGEN - ncg_output_d = error "*** GHC not built with a native-code generator ***" - ncg_output_w = ncg_output_d -#else - ncg_output_d = dumpRealAsm flat_absC_ncg ncg_uniqs - ncg_output_w = (\ f -> writeRealAsm f flat_absC_ncg ncg_uniqs) -#endif - in - - doDump opt_D_dump_asm "" ncg_output_d `thenMn_` - doOutput opt_ProduceS ncg_output_w `thenMn_` - - doDump opt_D_dump_realC "" c_output_d `thenMn_` - doOutput opt_ProduceC c_output_w `thenMn_` - exitMn 0 - } ) } + -------------------------- Code generation ------------------------------- + show_pass "CodeGen" >> + _scc_ "CodeGen" + codeGen this_mod imported_modules + cost_centre_info + local_tycons local_classes + stg_binds2 >>= \ abstractC -> -LATER -} - } - where - ------------------------------------------------------------- - -- ****** printing styles and column width: - - pprCols = (80 :: Int) -- could make configurable + -------------------------- Code output ------------------------------- + show_pass "CodeOutput" >> + _scc_ "CodeOutput" + codeOutput this_mod c_code h_code abstractC ncg_uniqs >> - (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 + -------------------------- Final report ------------------------------- + reportCompile mod_name (showSDoc (ppSourceStats True rdr_module)) >> + ghcExit 0 + } } + where ------------------------------------------------------------- -- ****** help functions: show_pass = if opt_D_show_passes - then \ what -> writeMn stderr ("*** "++what++":\n") - else \ what -> returnMn () - - doOutput switch io_action - = case switch of - Nothing -> returnMn () - Just fn -> let fname = _UNPK_ fn in - fopen fname "a+" `thenPrimIO` \ file -> - if (file == ``NULL'') then - error ("doOutput: failed to open:"++fname) - else - io_action file `thenMn` \ () -> - fclose file `thenPrimIO` \ status -> - if status == 0 - then returnMn () - else error ("doOutput: closed failed: "{-++show status++" "-}++fname) - - doDump switch hdr string - = if switch - then writeMn stderr hdr `thenMn_` - writeMn stderr ('\n': string) `thenMn_` - writeMn stderr "\n" - else returnMn () - - -ppSourceStats (HsModule name version exports imports fixities typedecls typesigs - classdecls instdecls instsigs defdecls binds - [{-no sigs-}] src_loc) - = ppAboves (map pp_val + then \ what -> hPutStr stderr ("*** "++what++":\n") + else \ what -> return () + +ppSourceStats short (HsModule name version exports imports decls src_loc) + = (if short then hcat else vcat) + (map pp_val [("ExportAll ", export_all), -- 1 if no export list ("ExportDecls ", export_ds), ("ExportModules ", export_ms), @@ -365,7 +235,7 @@ ppSourceStats (HsModule name version exports imports fixities typedecls typesigs (" ImpPartial ", import_partial), (" ImpHiding ", import_hiding), ("FixityDecls ", fixity_ds), - ("DefaultDecls ", defalut_ds), + ("DefaultDecls ", default_ds), ("TypeDecls ", type_ds), ("DataDecls ", data_ds), ("NewTypeDecls ", newt_ds), @@ -381,73 +251,71 @@ ppSourceStats (HsModule name version exports imports fixities typedecls typesigs ("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) + | not short = hcat [text str, int n] + | otherwise = hcat [text (trim str), equals, int n, semi] + + trim ls = takeWhile (not.isSpace) (dropWhile isSpace ls) + + fixity_ds = length [() | FixD d <- decls] + -- NB: this omits fixity decls on local bindings and + -- in class decls. ToDo + + tycl_decls = [d | TyClD d <- decls] + (class_ds, data_ds, newt_ds, type_ds) = countTyClDecls tycl_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)) + = foldr add2 (0,0) (map data_info tycl_decls) (class_method_ds, default_method_ds) - = foldr add2 (0,0) (map class_info classdecls) + = foldr add2 (0,0) (map class_info tycl_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_binds (MonoBind b sigs _) = case (count_monobinds 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_monobinds EmptyMonoBinds = (0,0) - count_monobinds (AndMonoBinds b1 b2) = count_monobinds b1 `add2` count_monobinds b2 + count_monobinds EmptyMonoBinds = (0,0) + count_monobinds (AndMonoBinds b1 b2) = count_monobinds b1 `add2` count_monobinds b2 count_monobinds (PatMonoBind (VarPatIn n) r _) = (1,0) - count_monobinds (PatMonoBind p r _) = (0,1) - count_monobinds (FunMonoBind f m _) = (0,1) + count_monobinds (PatMonoBind p r _) = (0,1) + count_monobinds (FunMonoBind f _ m _) = (0,1) 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 (SpecSig _ _ _) = (0,0,1,0) + sig_info (InlineSig _ _ _) = (0,0,0,1) + sig_info (NoInlineSig _ _ _) = (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 @@ -457,27 +325,28 @@ ppSourceStats (HsModule name version exports imports fixities typedecls typesigs 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}) + data_info other = (0,0) - class_info (ClassDecl _ _ _ meth_sigs def_meths _ _) + class_info (ClassDecl _ _ _ meth_sigs def_meths _ _ _ _ _) = case count_sigs meth_sigs of (_,classops,_,_) -> (classops, addpr (count_monobinds def_meths)) + class_info other = (0,0) - 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 :: (Int,Int) -> Int + add1 :: Int -> Int -> Int + add2 :: (Int,Int) -> (Int,Int) -> (Int, Int) + add3 :: (Int,Int,Int) -> (Int,Int,Int) -> (Int, Int, Int) + add4 :: (Int,Int,Int,Int) -> (Int,Int,Int,Int) -> (Int, Int, Int, Int) + add5 :: (Int,Int,Int,Int,Int) -> (Int,Int,Int,Int,Int) -> (Int, Int, Int, Int, Int) + add6 :: (Int,Int,Int,Int,Int,Int) -> (Int,Int,Int,Int,Int,Int) -> (Int, Int, Int, Int, Int, Int) addpr (x,y) = x+y add1 x1 y1 = x1+y1 @@ -487,3 +356,60 @@ ppSourceStats (HsModule name version exports imports fixities typedecls typesigs add5 (x1,x2,x3,x4,x5) (y1,y2,y3,y4,y5) = (x1+y1,x2+y2,x3+y3,x4+y4,x5+y5) add6 (x1,x2,x3,x4,x5,x6) (y1,y2,y3,y4,y5,y6) = (x1+y1,x2+y2,x3+y3,x4+y4,x5+y5,x6+y6) \end{code} + +\begin{code} +compiler_version :: String +compiler_version = + case (show opt_HiVersion) of + [x] -> ['0','.',x] + ls@[x,y] -> "0." ++ ls + ls -> go ls + where + -- 10232353 => 10232.53 + go ls@[x,y] = '.':ls + go (x:xs) = x:go xs + +booter_version + = case "\ + \ __GLASGOW_HASKELL__" of + ' ':n:ns -> n:'.':ns + ' ':m -> m +\end{code} + +\begin{code} +reportCompile :: ModuleName -> String -> IO () +#if REPORT_TO_MOTHERLODE && __GLASGOW_HASKELL__ >= 303 +reportCompile mod_name info + | not opt_ReportCompile = return () + | otherwise = (do + sock <- udpSocket 0 + addr <- motherShip + sendTo sock (moduleNameUserString mod_name ++ ';': compiler_version ++ + ';': os ++ ';':arch ++ '\n':' ':info ++ "\n") addr + return ()) `catch` (\ _ -> return ()) + +motherShip :: IO SockAddr +motherShip = do + he <- getHostByName "laysan.dcs.gla.ac.uk" + case (hostAddresses he) of + [] -> IOERROR (userError "No address!") + (x:_) -> return (SockAddrInet motherShipPort x) + +--magick +motherShipPort :: PortNumber +motherShipPort = mkPortNumber 12345 + +-- creates a socket capable of sending datagrams, +-- binding it to a port +-- ( 0 => have the system pick next available port no.) +udpSocket :: Int -> IO Socket +udpSocket p = do + pr <- getProtocolNumber "udp" + s <- socket AF_INET Datagram pr + bindSocket s (SockAddrInet (mkPortNumber p) iNADDR_ANY) + return s +#else +reportCompile _ _ = return () +#endif + +\end{code}