[project @ 1999-06-23 10:33:03 by simonmar]
[ghc-hetmet.git] / ghc / compiler / main / Main.lhs
index 01c5a55..9a62fae 100644 (file)
@@ -1,5 +1,5 @@
 %
-% (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}
 
@@ -8,57 +8,76 @@ module Main ( main ) where
 
 #include "HsVersions.h"
 
-import IO      ( IOMode(..),
-                 hGetContents, hPutStr, hClose, openFile,
-                 stdin,stderr
-               )
+import IO              ( hPutStr, stderr )
 import HsSyn
-import RdrHsSyn                ( RdrName )
 import BasicTypes      ( NewOrData(..) )
 
-import ReadPrefix      ( rdModule )
+import RdrHsSyn                ( RdrNameHsModule )
+import FastString      ( mkFastCharString, unpackFS )
+import StringBuffer    ( hGetStringBuffer )
+import Parser          ( parse )
+import Lex             ( PState(..), P, ParseResult(..) )
+import SrcLoc          ( mkSrcLoc )
+
 import Rename          ( renameModule )
-import RnMonad         ( ExportEnv )
 
-import MkIface         -- several functions
-import TcModule                ( typecheckModule )
-import Desugar         ( deSugar, pprDsWarnings )
+import MkIface         ( startIface, ifaceDecls, endIface )
+import TcModule                ( TcResults(..), typecheckModule )
+import Desugar         ( deSugar )
 import SimplCore       ( core2core )
+import CoreLint                ( endPass )
+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 CoreUnfold      ( Unfolding )
-import Bag             ( emptyBag, isEmptyBag )
+import Module          ( ModuleName, moduleNameUserString )
+import AbsCSyn         ( absCNop )
 import CmdLineOpts
-import ErrUtils                ( pprBagOfErrors, ghcExit, doIfSet, dumpIfSet )
+import ErrUtils                ( ghcExit, doIfSet, dumpIfSet )
 import Maybes          ( maybeToBool, MaybeErr(..) )
-import Specialise      ( SpecialiseData(..) )
-import StgSyn          ( GenStgBinding )
-import TcInstUtil      ( InstInfo )
 import TyCon           ( isDataTyCon )
 import Class           ( classTyCon )
 import UniqSupply      ( mkSplitUniqSupply )
 
-import PprAbsC         ( dumpRealC, writeRealC )
-import PprCore         ( pprCoreBinding )
 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 =
- _scc_ "main" 
- let
-    cmd_line_info = classifyOpts
- in
- doIt cmd_line_info
+ --  _scc_ "main" 
+ doIt classifyOpts
+\end{code}
+
+\begin{code}
+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}
 
 \begin{code}
@@ -66,176 +85,126 @@ doIt :: ([CoreToDo], [StgToDo]) -> IO ()
 
 doIt (core_cmds, stg_cmds)
   = doIfSet opt_Verbose 
-       (hPutStr stderr ("Glasgow Haskell Compiler, version\ 
-                        \ PROJECTVERSION\ 
-                        \, for Haskell 1.4\n"))                >>
+       (hPutStr stderr "Glasgow Haskell Compiler, version "    >>
+        hPutStr stderr compiler_version                        >>
+        hPutStr stderr ", for Haskell 98\n"                    >>
+        hPutStr stderr "\tcompiled by GHC version "            >>
+        hPutStr stderr booter_version                          >>
+        hPutStr stderr "\n")                                   >>
 
-    -- ******* READER
-    show_pass "Reader" >>
-    _scc_     "Reader"
-    rdModule           >>= \ (mod_name, rdr_module) ->
+       --------------------------  Reader  ----------------
+    show_pass "Parser" >>
+    _scc_     "Parser"
+    parseModule                >>= \ (mod_name, rdr_module) ->
 
-    dumpIfSet opt_D_dump_rdr "Reader" (ppr rdr_module)         >>
+    dumpIfSet opt_D_dump_parsed "Parser" (ppr rdr_module) >>
 
     dumpIfSet opt_D_source_stats "Source Statistics"
-       (ppSourceStats rdr_module)              >>
+       (ppSourceStats False 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 'r'      >>= \ ru_uniqs  -> -- rules
     mkSplitUniqSupply 'c'      >>= \ c2s_uniqs -> -- core-to-stg
---    _scc_     "spl-st"
+    mkSplitUniqSupply 'u'      >>= \ tidy_uniqs -> -- tidy up
     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
+       --------------------------  Rename  ----------------
     show_pass "Renamer"                        >>
     _scc_     "Renamer"
 
-    renameModule rn_uniqs rdr_module           >>=
-       \ maybe_rn_stuff ->
+    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 (rn_mod, iface_file_stuff, rn_name_supply, imported_modules) ->
+       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<blah>")
     -- "endIface" finishes the job.
-    startIface mod_name                                        >>= \ if_handle ->
-    ifaceMain if_handle iface_file_stuff               >>
+    startIface this_mod iface_file_stuff       >>= \ if_handle ->
 
 
-    -- ******* TYPECHECKER
+       --------------------------  Typechecking ----------------
     show_pass "TypeCheck"                              >>
     _scc_     "TypeCheck"
-    typecheckModule tc_uniqs rn_name_supply rn_mod     >>= \ maybe_tc_stuff ->
+    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 (all_binds,
-             local_tycons, local_classes, inst_info, 
-             ddump_deriv) ->
+       Just (tc_results@(TcResults {tc_tycons  = local_tycons, 
+                                    tc_classes = local_classes, 
+                                    tc_insts   = inst_info })) ->
 
 
-    -- ******* DESUGARER
-    show_pass "DeSugar"                        >>
+       --------------------------  Desugaring ----------------
     _scc_     "DeSugar"
-    deSugar ds_uniqs mod_name all_binds                >>= \ desugared ->
+    deSugar this_mod ds_uniqs tc_results       >>= \ (desugared, rules, h_code, c_code) ->
 
 
-    -- ******* CORE-TO-CORE SIMPLIFICATION
-    show_pass "Core2Core"                      >>
+       --------------------------  Main Core-language transformations ----------------
     _scc_     "Core2Core"
-    let
-       local_data_tycons = filter isDataTyCon local_tycons
-    in
-    core2core core_cmds mod_name
-             sm_uniqs local_data_tycons desugared
-                                               >>=
-        \ (simplified, spec_data
-               {- SpecData _ _ _ gen_data_tycons all_tycon_specs _ _ _ -}
-          ) ->
+    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) -> 
 
-    -- ******* STG-TO-STG SIMPLIFICATION
+
+       --------------------------  Convert to STG code -------------------------------
     show_pass "Core2Stg"                       >>
     _scc_     "Core2Stg"
     let
-       stg_binds   = topCoreBindsToStg c2s_uniqs simplified
+       stg_binds   = topCoreBindsToStg c2s_uniqs tidy_binds
     in
 
+       --------------------------  Simplify STG code -------------------------------
     show_pass "Stg2Stg"                        >>
     _scc_     "Stg2Stg"
-    stg2stg stg_cmds mod_name st_uniqs stg_binds
-                                               >>=
-       \ (stg_binds2, cost_centre_info) ->
+    stg2stg stg_cmds this_mod st_uniqs stg_binds >>= \ (stg_binds2, cost_centre_info) ->
 
-    dumpIfSet opt_D_dump_stg "STG syntax:" (pprStgBindings stg_binds2) >>
 
+       --------------------------  Interface file -------------------------------
        -- Dump instance decls and type signatures into the interface file
+    _scc_     "Interface"
     let
-       final_ids = collectFinalStgBinders stg_binds2
+       final_ids = collectFinalStgBinders (map fst stg_binds2)
     in
-    _scc_     "Interface"
-    ifaceDecls if_handle local_tycons local_classes inst_info final_ids simplified     >>
+    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".)
-    
+           -- We are definitely done w/ interface-file stuff at this point:
+           -- (See comments near call to "startIface".)
+
 
-    -- ******* "ABSTRACT", THEN "FLAT", THEN *REAL* C!
+       --------------------------  Code generation -------------------------------
     show_pass "CodeGen"                        >>
     _scc_     "CodeGen"
-    let
-       all_local_data_tycons = filter isDataTyCon (map classTyCon local_classes)
-                               ++ local_data_tycons
-                                       -- Generate info tables  for the data constrs arising
-                                       -- from class decls as well
-
-       all_tycon_specs       = emptyFM -- Not specialising tycons any more
-
-       abstractC      = codeGen mod_name               -- module name for CC labelling
-                                cost_centre_info
-                                imported_modules       -- import names for CC registering
-                                all_local_data_tycons  -- type constructors generated locally
-                                all_tycon_specs        -- tycon specialisations
-                                stg_binds2
-
-       flat_abstractC = flattenAbsC fl_uniqs abstractC
-    in
-    dumpIfSet opt_D_dump_absC "Abstract C"
-       (dumpRealC abstractC)                   >>
+    codeGen this_mod imported_modules
+           cost_centre_info
+           local_tycons local_classes 
+           stg_binds2                          >>= \ abstractC ->
 
-    dumpIfSet opt_D_dump_flatC "Flat Abstract C"
-       (dumpRealC flat_abstractC)              >>
 
-    show_pass "CodeOutput"                     >>
+       --------------------------  Code output -------------------------------
+    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]
-    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
+    codeOutput this_mod c_code h_code abstractC ncg_uniqs      >>
 
-    dumpIfSet opt_D_dump_asm "Asm code" ncg_output_d   >>
-    doOutput opt_ProduceS ncg_output_w                         >>
 
-    dumpIfSet opt_D_dump_realC "Real C" c_output_d     >>
-    doOutput opt_ProduceC c_output_w                   >>
+       --------------------------  Final report -------------------------------
+    reportCompile mod_name (showSDoc (ppSourceStats True rdr_module)) >>
 
     ghcExit 0
     } }
@@ -248,17 +217,9 @@ doIt (core_cmds, stg_cmds)
        then \ what -> hPutStr stderr ("*** "++what++":\n")
        else \ what -> return ()
 
-    doOutput switch io_action
-      = case switch of
-         Nothing -> return ()
-         Just fname ->
-           openFile fname WriteMode    >>= \ handle ->
-           io_action handle            >>
-           hClose handle
-
-
-ppSourceStats (HsModule name version exports imports fixities decls src_loc)
- = vcat (map pp_val
+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),
@@ -292,17 +253,19 @@ ppSourceStats (HsModule name version exports imports fixities decls src_loc)
               ])
   where
     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
+    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]
@@ -320,9 +283,9 @@ ppSourceStats (HsModule name version exports imports fixities decls src_loc)
     (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 class_decls)
+       = foldr add2 (0,0) (map class_info tycl_decls)
     (inst_method_ds, method_specs, method_inlines)
        = foldr add3 (0,0,0) (map inst_info inst_decls)
 
@@ -332,21 +295,21 @@ ppSourceStats (HsModule name version exports imports fixities decls src_loc)
     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_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 (ClassOpSig _ _ _ _) = (0,1,0,0)
-    sig_info (SpecSig _ _ _ _)    = (0,0,1,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
@@ -358,17 +321,27 @@ ppSourceStats (HsModule name version exports imports fixities decls src_loc)
 
     data_info (TyData _ _ _ _ constrs derivs _ _)
        = (length constrs, 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 _ _)
        = case count_sigs inst_sigs of
            (_,_,ss,is) ->
               (addpr (count_monobinds inst_meths), ss, is)
 
+    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
     add2 (x1,x2) (y1,y2) = (x1+y1,x2+y2)
@@ -377,3 +350,60 @@ ppSourceStats (HsModule name version exports imports fixities decls src_loc)
     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}