[project @ 2003-06-27 21:17:24 by simonpj]

[ghc-hetmet.git] / ghc / compiler / typecheck / TcDefaults.lhs

diff --git a/ghc/compiler/typecheck/TcDefaults.lhs b/ghc/compiler/typecheck/TcDefaults.lhs
index 49f9421..f107451 100644 (file)
--- a/ghc/compiler/typecheck/TcDefaults.lhs
+++ b/ghc/compiler/typecheck/TcDefaults.lhs
@@ -1,77 +1,89 @@
 %
-% (c) The AQUA Project, Glasgow University, 1993-1995
+% (c) The AQUA Project, Glasgow University, 1993-1998
 %
 \section[TcDefaults]{Typechecking \tr{default} declarations}
 
 \begin{code}
-#include "HsVersions.h"
-
 module TcDefaults ( tcDefaults ) where
 
-IMP_Ubiq()
-
-import HsSyn           ( HsDecl(..), TyDecl, ClassDecl, InstDecl, HsBinds,
-                         DefaultDecl(..), HsType, IfaceSig,
-                         HsExpr, HsLit, ArithSeqInfo, Fake, InPat)
-import RnHsSyn         ( RenamedHsDecl(..), RenamedDefaultDecl(..) )
+#include "HsVersions.h"
 
-import TcMonad
-import Inst            ( InstOrigin(..) )
-import TcEnv           ( tcLookupClassByKey )
-import SpecEnv         ( SpecEnv )
+import HsSyn           ( DefaultDecl(..) )
+import Name            ( Name )
+import TcRnMonad
+import TcEnv           ( tcLookupGlobal_maybe )
 import TcMonoType      ( tcHsType )
-import TcSimplify      ( tcSimplifyCheckThetas )
-import TcType          ( TcIdOcc )
-
-import TysWiredIn      ( intTy, doubleTy, unitTy )
-import Type             ( SYN_IE(Type) )
-import Unique          ( numClassKey )
-import Pretty          ( ptext, vcat )
-import ErrUtils                ( addShortErrLocLine )
-import Util
+import TcSimplify      ( tcSimplifyDefault )
+import TcType           ( Type, mkClassPred, isTauTy )
+import PrelNames       ( numClassName )
+import Outputable
+import HscTypes                ( TyThing(..) )
 \end{code}
 
 \begin{code}
-default_default = [intTy, doubleTy]        -- language-specified default `default'
-
-tcDefaults :: [RenamedHsDecl]
-          -> TcM s [Type]          -- defaulting types to heave
+tcDefaults :: [DefaultDecl Name]
+          -> TcM [Type]            -- Defaulting types to heave
                                    -- into Tc monad for later use
                                    -- in Disambig.
-tcDefaults decls = tc_defaults [default_decl | DefD default_decl <- decls]
-
-tc_defaults [] = returnTc default_default
-
-tc_defaults [DefaultDecl mono_tys locn]
-  = tcAddSrcLoc locn $
-    mapTc tcHsType mono_tys    `thenTc` \ tau_tys ->
-
-    case tau_tys of
-      [] -> returnTc []                -- no defaults
 
-      _  ->
-           -- Check that all the types are instances of Num
-           -- We only care about whether it worked or not
-
-       tcLookupClassByKey numClassKey                  `thenNF_Tc` \ num ->
-       tcSimplifyCheckThetas
-               [ (num, ty) | ty <- tau_tys ]           `thenTc_`
-
-       returnTc tau_tys
-
-tc_defaults decls
-  = failTc (dupDefaultDeclErr decls)
-
-
-dupDefaultDeclErr (DefaultDecl _ locn1 : dup_things) sty
-  = vcat (item1 : map dup_item dup_things)
+tcDefaults [] 
+  = getDefaultTys              -- No default declaration, so get the
+                               -- default types from the envt; 
+                               -- i.e. use the curent ones
+                               -- (the caller will put them back there)
+       -- It's important not to return defaultDefaultTys here (which
+       -- we used to do) because in a TH program, tcDefaults [] is called
+       -- repeatedly, once for each group of declarations between top-level
+       -- splices.  We don't want to carefully set the default types in
+       -- one group, only for the next group to ignore them and install
+       -- defaultDefaultTys
+
+tcDefaults [DefaultDecl [] locn]
+  = returnM []                 -- Default declaration specifying no types
+
+tcDefaults [DefaultDecl mono_tys locn]
+  = tcLookupGlobal_maybe numClassName  `thenM` \ maybe_num ->
+    case maybe_num of
+       Just (AClass num_class) -> common_case num_class
+       other                   -> returnM []
+               -- In the Nothing case, Num has not been sucked in, so the 
+               -- defaults will never be used; so simply discard the default decl.
+               -- This slightly benefits modules that don't use any
+               -- numeric stuff at all, by avoid the necessity of
+               -- always sucking in Num
   where
-    item1
-      = addShortErrLocLine locn1 (\ sty ->
-       ptext SLIT("multiple default declarations")) sty
-
-    dup_item (DefaultDecl _ locn)
-      = addShortErrLocLine locn (\ sty ->
-       ptext SLIT("here was another default declaration")) sty
+    common_case num_class
+      = addSrcLoc locn                 $
+       addErrCtxt defaultDeclCtxt      $
+       mappM tc_default_ty mono_tys    `thenM` \ tau_tys ->
+    
+               -- Check that all the types are instances of Num
+               -- We only care about whether it worked or not
+       tcSimplifyDefault [mkClassPred num_class [ty] | ty <- tau_tys]  `thenM_`
+    
+       returnM tau_tys
+
+tcDefaults decls@(DefaultDecl _ loc : _) =
+    addSrcLoc loc $
+    failWithTc (dupDefaultDeclErr decls)
+
+
+tc_default_ty hs_ty 
+ = tcHsType hs_ty                              `thenM` \ ty ->
+   checkTc (isTauTy ty) (polyDefErr hs_ty)     `thenM_`
+   returnM ty
+
+defaultDeclCtxt =  ptext SLIT("when checking that each type in a default declaration")
+                   $$ ptext SLIT("is an instance of class Num")
+
+
+dupDefaultDeclErr (DefaultDecl _ locn1 : dup_things)
+  = hang (ptext SLIT("Multiple default declarations"))
+      4  (vcat (map pp dup_things))
+  where
+    pp (DefaultDecl _ locn) = ptext SLIT("here was another default declaration") <+> ppr locn
 
+polyDefErr ty 
+  = hang (ptext SLIT("Illegal polymorphic type in default declaration") <> colon) 4 (ppr ty) 
 \end{code}
+