[project @ 1999-06-22 07:59:54 by simonpj]
[ghc-hetmet.git] / ghc / compiler / typecheck / TcClassDcl.lhs
index e7b7676..794eb83 100644 (file)
@@ -1,72 +1,66 @@
 %
-% (c) The GRASP/AQUA Project, Glasgow University, 1992-1995
+% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
 %
 \section[TcClassDcl]{Typechecking class declarations}
 
 \begin{code}
-module TcClassDcl ( tcClassDecl1, tcClassDecls2, tcMethodBind, badMethodErr ) where
+module TcClassDcl ( kcClassDecl, tcClassDecl1, tcClassDecls2, 
+                   tcMethodBind, checkFromThisClass
+                 ) where
 
 #include "HsVersions.h"
 
-import HsSyn           ( HsDecl(..), ClassDecl(..), Sig(..), MonoBinds(..),
-                         InPat(..), HsBinds(..), GRHSsAndBinds(..), GRHS(..),
+import HsSyn           ( HsDecl(..), TyClDecl(..), Sig(..), MonoBinds(..),
+                         InPat(..), HsBinds(..), GRHSs(..),
                          HsExpr(..), HsLit(..), HsType(..), pprClassAssertion,
-                         unguardedRHS, andMonoBinds, getTyVarName
+                         unguardedRHS, andMonoBinds, andMonoBindList, getTyVarName,
+                         isClassDecl, isClassOpSig, collectMonoBinders
                        )
 import HsPragmas       ( ClassPragmas(..) )
 import BasicTypes      ( NewOrData(..), TopLevelFlag(..), RecFlag(..) )
-import RnHsSyn         ( RenamedClassDecl, RenamedClassPragmas,
+import RnHsSyn         ( RenamedTyClDecl, RenamedClassPragmas,
                          RenamedClassOpSig, RenamedMonoBinds,
                          RenamedContext, RenamedHsDecl, RenamedSig
                        )
 import TcHsSyn         ( TcMonoBinds )
 
-import Inst            ( Inst, InstOrigin(..), LIE, emptyLIE, plusLIE, newDicts, newMethod )
-import TcEnv           ( TcIdOcc(..), GlobalValueEnv, tcAddImportedIdInfo,
-                         tcLookupClass, tcLookupTyVar, 
-                         tcExtendGlobalTyVars, tcExtendLocalValEnv
+import Inst            ( Inst, InstOrigin(..), LIE, emptyLIE, plusLIE, plusLIEs, newDicts, newMethod )
+import TcEnv           ( TcId, ValueEnv, TcTyThing(..), tcAddImportedIdInfo,
+                         tcLookupClass, tcLookupTy, tcExtendTyVarEnvForMeths, tcExtendGlobalTyVars,
+                         tcExtendLocalValEnv
                        )
-import TcBinds         ( tcBindWithSigs, bindInstsOfLocalFuns, 
-                         checkSigTyVars, sigCtxt, tcPragmaSigs, TcSigInfo(..)
-                       )
-import TcKind          ( unifyKinds, TcKind )
+import TcBinds         ( tcBindWithSigs, tcSpecSigs )
+import TcUnify         ( unifyKinds )
 import TcMonad
-import TcMonoType      ( tcHsType, tcContext )
-import TcSimplify      ( tcSimplifyAndCheck )
-import TcType          ( TcType, TcTyVar, TcTyVarSet, tcInstSigTyVars, 
-                         zonkSigTyVar, tcInstSigTcType
+import TcMonoType      ( tcHsType, tcHsTopType, tcExtendTopTyVarScope, 
+                         tcContext, checkSigTyVars, sigCtxt, mkTcSig
                        )
-import PrelVals                ( nO_METHOD_BINDING_ERROR_ID )
+import TcSimplify      ( tcSimplifyAndCheck, bindInstsOfLocalFuns )
+import TcType          ( TcType, TcTyVar, tcInstTyVars, zonkTcTyVarBndr, tcGetTyVar )
+import PrelInfo                ( nO_METHOD_BINDING_ERROR_ID )
 import FieldLabel      ( firstFieldLabelTag )
-import Bag             ( unionManyBags )
+import Bag             ( unionManyBags, bagToList )
 import Class           ( mkClass, classBigSig, Class )
 import CmdLineOpts      ( opt_GlasgowExts, opt_WarnMissingMethods )
-import MkId            ( mkDataCon, mkSuperDictSelId, 
-                         mkMethodSelId, mkDefaultMethodId
-                       )
-import Id              ( Id, StrictnessMark(..),
-                         getIdUnfolding, idType, idName
-                       )
-import CoreUnfold      ( getUnfoldingTemplate )
+import MkId            ( mkDictSelId, mkDataConId, mkDefaultMethodId )
+import DataCon         ( mkDataCon, notMarkedStrict )
+import Id              ( Id, setInlinePragma, getIdUnfolding, idType, idName )
+import CoreUnfold      ( unfoldingTemplate )
 import IdInfo
-import Name            ( Name, isLocallyDefined, OccName, nameOccName,
-                         NamedThing(..) )
+import Name            ( Name, nameOccName, isLocallyDefined, NamedThing(..) )
+import NameSet         ( emptyNameSet )
 import Outputable
-import Type            ( mkFunTy, mkTyVarTy, mkTyVarTys, mkDictTy, splitRhoTy,
-                         mkSigmaTy, mkForAllTys, Type, ThetaType
+import Type            ( mkFunTy, mkTyVarTy, mkTyVarTys, mkDictTy,
+                         mkSigmaTy, mkForAllTys, Type, ThetaType,
+                         boxedTypeKind, mkArrowKind
                        )
-import TyVar           ( mkTyVarSet, tyVarKind, TyVar )
-import TyCon           ( mkDataTyCon )
-import Kind            ( mkBoxedTypeKind, mkArrowKind )
+import Var             ( tyVarKind, TyVar )
+import VarSet          ( mkVarSet )
+import TyCon           ( mkAlgTyCon )
 import Unique          ( Unique, Uniquable(..) )
 import Util
-import Maybes          ( assocMaybe, maybeToBool, seqMaybe )
-
-
--- import TcPragmas    ( tcGenPragmas, tcClassOpPragmas )
-tcGenPragmas ty id ps = returnNF_Tc noIdInfo
-tcClassOpPragmas ty sel def spec ps = returnNF_Tc (spec `setSpecInfo` noIdInfo, 
-                                                  noIdInfo)
+import Maybes          ( seqMaybe )
+import FiniteMap        ( lookupWithDefaultFM )
 \end{code}
 
 
@@ -104,39 +98,69 @@ Now DictTy in Type is just a form of type synomym:
 Death to "ExpandingDicts".
 
 
+%************************************************************************
+%*                                                                     *
+\subsection{Kind checking}
+%*                                                                     *
+%************************************************************************
+
 \begin{code}
-tcClassDecl1 rec_env rec_inst_mapper
-            (ClassDecl context class_name
+kcClassDecl (ClassDecl context class_name
                        tyvar_names class_sigs def_methods pragmas 
-                       tycon_name datacon_name src_loc)
-  = tcAddSrcLoc src_loc        $
-    tcAddErrCtxt (classDeclCtxt class_name) $
-
-        -- CHECK ARITY 1 FOR HASKELL 1.4
+                       tycon_name datacon_name sc_sel_names src_loc)
+  =         -- CHECK ARITY 1 FOR HASKELL 1.4
     checkTc (opt_GlasgowExts || length tyvar_names == 1)
            (classArityErr class_name)          `thenTc_`
 
-       -- LOOK THINGS UP IN THE ENVIRONMENT
-    tcLookupClass class_name                   `thenTc` \ (class_kinds, rec_class) ->
-    mapAndUnzipNF_Tc (tcLookupTyVar . getTyVarName) tyvar_names
-                                               `thenNF_Tc` \ (tyvar_kinds, rec_tyvars) ->
+       -- Get the (mutable) class kind
+    tcLookupTy class_name                      `thenNF_Tc` \ (kind, _, _) ->
 
-       -- FORCE THE CLASS AND ITS TYVAR TO HAVE SAME KIND
-    unifyKinds class_kinds tyvar_kinds `thenTc_`
+       -- Make suitable tyvars and do kind checking
+       -- The net effect is to mutate the class kind
+    tcExtendTopTyVarScope kind tyvar_names     $ \ _ _ ->
+    tcContext context                          `thenTc_`
+    mapTc kc_sig the_class_sigs                        `thenTc_`
 
+    returnTc ()
+  where
+    the_class_sigs = filter isClassOpSig class_sigs
+  
+    kc_sig (ClassOpSig _ _ op_ty loc) = tcAddSrcLoc loc (tcHsType op_ty)
+\end{code}
+
+
+%************************************************************************
+%*                                                                     *
+\subsection{Type checking}
+%*                                                                     *
+%************************************************************************
+
+\begin{code}
+tcClassDecl1 rec_env rec_inst_mapper rec_vrcs
+            (ClassDecl context class_name
+                       tyvar_names class_sigs def_methods pragmas 
+                       tycon_name datacon_name sc_sel_names src_loc)
+  =    -- LOOK THINGS UP IN THE ENVIRONMENT
+    tcLookupTy class_name                              `thenTc` \ (class_kind, _, AClass rec_class) ->
+    tcExtendTopTyVarScope class_kind tyvar_names       $ \ tyvars _ ->
+       -- The class kind is by now immutable
+       
        -- CHECK THE CONTEXT
-    tcClassContext class_name rec_class rec_tyvars context pragmas     
+--  traceTc (text "tcClassCtxt" <+> ppr class_name)    `thenTc_`
+    tcClassContext class_name rec_class tyvars context sc_sel_names
                                                `thenTc` \ (sc_theta, sc_tys, sc_sel_ids) ->
+--  traceTc (text "tcClassCtxt done" <+> ppr class_name)       `thenTc_`
 
        -- CHECK THE CLASS SIGNATURES,
-    mapTc (tcClassSig rec_env rec_class rec_tyvars) class_sigs
+    mapTc (tcClassSig rec_env rec_class tyvars) 
+         (filter isClassOpSig class_sigs)
                                                `thenTc` \ sig_stuff ->
 
        -- MAKE THE CLASS OBJECT ITSELF
     let
        (op_tys, op_sel_ids, defm_ids) = unzip3 sig_stuff
        rec_class_inst_env = rec_inst_mapper rec_class
-       clas = mkClass (getName class_name) rec_tyvars
+       clas = mkClass class_name tyvars
                       sc_theta sc_sel_ids op_sel_ids defm_ids
                       tycon
                       rec_class_inst_env
@@ -146,20 +170,30 @@ tcClassDecl1 rec_env rec_inst_mapper
                        [_]   -> NewType
                        other -> DataType
 
-        dict_con_id = mkDataCon datacon_name
-                          [NotMarkedStrict | _ <- dict_component_tys]
+        dict_con = mkDataCon datacon_name
+                          [notMarkedStrict | _ <- dict_component_tys]
                           [{- No labelled fields -}]
-                          rec_tyvars
+                          tyvars
                           [{-No context-}]
                           [{-No existential tyvars-}] [{-Or context-}]
                           dict_component_tys
-                          tycon
+                          tycon dict_con_id
+
+       -- In general, constructors don't have to be inlined, but this one
+       -- does, because we don't make a top level binding for it.      
+       dict_con_id = mkDataConId dict_con
+                     `setInlinePragma` IMustBeINLINEd
 
-       tycon = mkDataTyCon tycon_name
-                           (foldr (mkArrowKind . tyVarKind) mkBoxedTypeKind rec_tyvars)
-                           rec_tyvars
+        argvrcs = lookupWithDefaultFM rec_vrcs (pprPanic "tcClassDecl1: argvrcs:" $
+                                                         ppr tycon_name)
+                                      tycon_name
+
+       tycon = mkAlgTyCon tycon_name
+                           class_kind
+                           tyvars
                            []                  -- No context
-                           [dict_con_id]       -- Constructors
+                            argvrcs
+                           [dict_con]          -- Constructors
                            []                  -- No derivings
                            (Just clas)         -- Yes!  It's a dictionary 
                            new_or_data
@@ -172,12 +206,12 @@ tcClassDecl1 rec_env rec_inst_mapper
 \begin{code}
 tcClassContext :: Name -> Class -> [TyVar]
               -> RenamedContext        -- class context
-              -> RenamedClassPragmas   -- pragmas for superclasses  
+              -> [Name]                -- Names for superclass selectors
               -> TcM s (ThetaType,     -- the superclass context
                         [Type],        -- types of the superclass dictionaries
                         [Id])          -- superclass selector Ids
 
-tcClassContext class_name rec_class rec_tyvars context pragmas
+tcClassContext class_name rec_class rec_tyvars context sc_sel_names
   =    -- Check the context.
        -- The renamer has already checked that the context mentions
        -- only the type variable of the class decl.
@@ -193,31 +227,19 @@ tcClassContext class_name rec_class rec_tyvars context pragmas
 
     let
        sc_tys = [mkDictTy sc tys | (sc,tys) <- sc_theta]
+       sc_sel_ids = zipWithEqual "tcClassContext" mk_super_id sc_sel_names sc_tys
     in
-
-       -- Make super-class selector ids
-       -- We number them off, 1, 2, 3 etc so that we can construct
-       -- names for the selectors.  Thus
-       --      class (C a, C b) => D a b where ...
-       -- gives superclass selectors
-       --      D_sc1, D_sc2
-       -- (We used to call them D_C, but now we can have two different
-       --  superclasses both called C!)
-    mapTc mk_super_id (sc_theta `zip` [firstFieldLabelTag..])  `thenTc` \ sc_sel_ids ->
-
        -- Done
     returnTc (sc_theta, sc_tys, sc_sel_ids)
 
   where
     rec_tyvar_tys = mkTyVarTys rec_tyvars
 
-    mk_super_id ((super_class, tys), index)
-        = tcGetUnique                  `thenNF_Tc` \ uniq ->
-         let
-               ty = mkForAllTys rec_tyvars $
-                    mkFunTy (mkDictTy rec_class rec_tyvar_tys) (mkDictTy super_class tys)
-         in
-         returnTc (mkSuperDictSelId uniq rec_class index ty)
+    mk_super_id name dict_ty
+        = mkDictSelId name rec_class ty
+        where
+         ty = mkForAllTys rec_tyvars $
+              mkFunTy (mkDictTy rec_class rec_tyvar_tys) dict_ty
 
     check_constraint (c, tys) = checkTc (all is_tyvar tys)
                                        (superClassErr class_name (c, tys))
@@ -226,7 +248,7 @@ tcClassContext class_name rec_class rec_tyvars context pragmas
     is_tyvar other        = False
 
 
-tcClassSig :: GlobalValueEnv           -- Knot tying only!
+tcClassSig :: ValueEnv         -- Knot tying only!
           -> Class                     -- ...ditto...
           -> [TyVar]                   -- The class type variable, used for error check only
           -> RenamedClassOpSig
@@ -245,16 +267,15 @@ tcClassSig rec_env rec_clas rec_clas_tyvars
 
     -- NB: Renamer checks that the class type variable is mentioned in local_ty,
     -- and that it is not constrained by theta
-    tcHsType op_ty                             `thenTc` \ local_ty ->
+--  traceTc (text "tcClassSig" <+> ppr op_name)        `thenTc_`
+    tcHsTopType op_ty                          `thenTc` \ local_ty ->
     let
        global_ty   = mkSigmaTy rec_clas_tyvars 
                                [(rec_clas, mkTyVarTys rec_clas_tyvars)]
                                local_ty
-    in
 
        -- Build the selector id and default method id
-    let
-       sel_id      = mkMethodSelId op_name rec_clas global_ty
+       sel_id      = mkDictSelId op_name rec_clas global_ty
        maybe_dm_id = case maybe_dm_name of
                           Nothing      -> Nothing
                           Just dm_name -> let 
@@ -262,6 +283,7 @@ tcClassSig rec_env rec_clas rec_clas_tyvars
                                           in
                                           Just (tcAddImportedIdInfo rec_env dm_id)
     in
+--  traceTc (text "tcClassSig done" <+> ppr op_name)   `thenTc_`
     returnTc (local_ty, sel_id, maybe_dm_id)
 \end{code}
 
@@ -290,12 +312,12 @@ each local class decl.
 
 \begin{code}
 tcClassDecls2 :: [RenamedHsDecl]
-             -> NF_TcM s (LIE s, TcMonoBinds s)
+             -> NF_TcM s (LIE, TcMonoBinds)
 
 tcClassDecls2 decls
   = foldr combine
          (returnNF_Tc (emptyLIE, EmptyMonoBinds))
-         [tcClassDecl2 cls_decl | ClD cls_decl <- decls]
+         [tcClassDecl2 cls_decl | TyClD cls_decl <- decls, isClassDecl cls_decl]
   where
     combine tc1 tc2 = tc1 `thenNF_Tc` \ (lie1, binds1) ->
                      tc2 `thenNF_Tc` \ (lie2, binds2) ->
@@ -306,11 +328,11 @@ tcClassDecls2 decls
 @tcClassDecl2@ is the business end of things.
 
 \begin{code}
-tcClassDecl2 :: RenamedClassDecl       -- The class declaration
-            -> NF_TcM s (LIE s, TcMonoBinds s)
+tcClassDecl2 :: RenamedTyClDecl                -- The class declaration
+            -> NF_TcM s (LIE, TcMonoBinds)
 
 tcClassDecl2 (ClassDecl context class_name
-                       tyvar_names class_sigs default_binds pragmas _ _ src_loc)
+                       tyvar_names class_sigs default_binds pragmas _ _ _ src_loc)
 
   | not (isLocallyDefined class_name)
   = returnNF_Tc (emptyLIE, EmptyMonoBinds)
@@ -320,23 +342,20 @@ tcClassDecl2 (ClassDecl context class_name
     tcAddSrcLoc src_loc                                          $
 
        -- Get the relevant class
-    tcLookupClass class_name           `thenTc` \ (_, clas) ->
+    tcLookupClass class_name                           `thenNF_Tc` \ clas ->
     let
        (tyvars, sc_theta, sc_sel_ids, op_sel_ids, defm_ids) = classBigSig clas
 
        -- The selector binds are already in the selector Id's unfoldings
-       sel_binds = [ CoreMonoBind (RealId sel_id) (getUnfoldingTemplate (getIdUnfolding sel_id))
-                   | sel_id <- sc_sel_ids ++ op_sel_ids, 
-                     isLocallyDefined sel_id
+       sel_binds = [ CoreMonoBind sel_id (unfoldingTemplate (getIdUnfolding sel_id))
+                   | sel_id <- sc_sel_ids ++ op_sel_ids 
                    ]
-
-       final_sel_binds = andMonoBinds sel_binds
     in
        -- Generate bindings for the default methods
     tcDefaultMethodBinds clas default_binds            `thenTc` \ (const_insts, meth_binds) ->
 
-    returnTc (const_insts, 
-             final_sel_binds `AndMonoBinds` meth_binds)
+    returnTc (const_insts,
+             meth_binds `AndMonoBinds` andMonoBindList sel_binds)
 \end{code}
 
 %************************************************************************
@@ -361,23 +380,11 @@ we get the default methods:
 defm.Foo.op1 :: forall a. Foo a => a -> Bool
 defm.Foo.op1 = /\a -> \dfoo -> \x -> True
 
-====================== OLD ==================
-\begin{verbatim}
-defm.Foo.op2 :: forall a, b. (Foo a, Ord b) => a -> b -> b -> b
-defm.Foo.op2 = /\ a b -> \ dfoo dord -> \x y z ->
-                 if (op1 a dfoo x) && (< b dord y z) then y else z
-\end{verbatim}
-Notice that, like all ids, the foralls of defm.Foo.op2 are at the top.
-====================== END OF OLD ===================
-
-NEW:
-\begin{verbatim}
 defm.Foo.op2 :: forall a. Foo a => forall b. Ord b => a -> b -> b -> b
 defm.Foo.op2 = /\ a -> \ dfoo -> /\ b -> \ dord -> \x y z ->
                  if (op1 a dfoo x) && (< b dord y z) then y else z
 \end{verbatim}
 
-
 When we come across an instance decl, we may need to use the default
 methods:
 \begin{verbatim}
@@ -416,45 +423,18 @@ dfun.Foo.List
 tcDefaultMethodBinds
        :: Class
        -> RenamedMonoBinds
-       -> TcM s (LIE s, TcMonoBinds s)
+       -> TcM s (LIE, TcMonoBinds)
 
 tcDefaultMethodBinds clas default_binds
-  =    -- Construct suitable signatures
-    tcInstSigTyVars tyvars             `thenNF_Tc` \ (clas_tyvars, inst_tys, inst_env) ->
+  =    -- Check that the default bindings come from this class
+    checkFromThisClass clas op_sel_ids default_binds   `thenNF_Tc_`
 
-       -- Typecheck the default bindings
-    let
-       tc_dm sel_id_w_dm@(_, Just dm_id)
-         = tcMethodBind clas origin inst_tys clas_tyvars 
-                        default_binds [{-no prags-}] False
-                        sel_id_w_dm            `thenTc` \ (bind, insts, (_, local_dm_id)) ->
-           returnTc (bind, insts, (clas_tyvars, RealId dm_id, local_dm_id))
-    in    
-    mapAndUnzip3Tc tc_dm sel_ids_w_dms         `thenTc` \ (defm_binds, insts_needed, abs_bind_stuff) ->
-
-       -- Check the context
-    newDicts origin [(clas,inst_tys)]          `thenNF_Tc` \ (this_dict, [this_dict_id]) ->
-    let
-       avail_insts = this_dict
-    in
-    tcAddErrCtxt (classDeclCtxt clas) $
-    mapNF_Tc zonkSigTyVar clas_tyvars          `thenNF_Tc` \ clas_tyvars' ->
-    tcSimplifyAndCheck
-       (ptext SLIT("class") <+> ppr clas)
-       (mkTyVarSet clas_tyvars')
-       avail_insts
-       (unionManyBags insts_needed)            `thenTc` \ (const_lie, dict_binds) ->
-
-    let
-       full_binds = AbsBinds
-                       clas_tyvars'
-                       [this_dict_id]
-                       abs_bind_stuff
-                       (dict_binds `AndMonoBinds` andMonoBinds defm_binds)
-    in
-    returnTc (const_lie, full_binds)
+       -- Do each default method separately
+    mapAndUnzipTc tc_dm sel_ids_w_dms                  `thenTc` \ (defm_binds, const_lies) ->
 
+    returnTc (plusLIEs const_lies, andMonoBindList defm_binds)
   where
+
     (tyvars, sc_theta, sc_sel_ids, op_sel_ids, defm_ids) = classBigSig clas
 
     sel_ids_w_dms = [pair | pair@(_, Just _) <- op_sel_ids `zip` defm_ids]
@@ -462,7 +442,70 @@ tcDefaultMethodBinds clas default_binds
                        -- user default declaration
 
     origin = ClassDeclOrigin
+
+    -- We make a separate binding for each default method.
+    -- At one time I used a single AbsBinds for all of them, thus
+    -- AbsBind [d] [dm1, dm2, dm3] { dm1 = ...; dm2 = ...; dm3 = ... }
+    -- But that desugars into
+    -- ds = \d -> (..., ..., ...)
+    -- dm1 = \d -> case ds d of (a,b,c) -> a
+    -- And since ds is big, it doesn't get inlined, so we don't get good
+    -- default methods.  Better to make separate AbsBinds for each
+    
+    tc_dm sel_id_w_dm@(_, Just dm_id)
+      = tcInstTyVars tyvars            `thenNF_Tc` \ (clas_tyvars, inst_tys, _) ->
+       let
+           theta = [(clas,inst_tys)]
+       in
+       newDicts origin theta                   `thenNF_Tc` \ (this_dict, [this_dict_id]) ->
+       let
+           avail_insts = this_dict
+       in
+       tcExtendTyVarEnvForMeths tyvars clas_tyvars (
+           tcMethodBind clas origin clas_tyvars inst_tys theta
+                        default_binds [{-no prags-}] False
+                        sel_id_w_dm    
+        )                                      `thenTc` \ (defm_bind, insts_needed, (_, local_dm_id)) ->
+    
+       tcAddErrCtxt (defltMethCtxt clas) $
+    
+           -- tcMethodBind has checked that the class_tyvars havn't
+           -- been unified with each other or another type, but we must
+           -- still zonk them before passing them to tcSimplifyAndCheck
+       mapNF_Tc zonkTcTyVarBndr clas_tyvars    `thenNF_Tc` \ clas_tyvars' ->
+    
+           -- Check the context
+       tcSimplifyAndCheck
+           (ptext SLIT("class") <+> ppr clas)
+           (mkVarSet clas_tyvars')
+           avail_insts
+           insts_needed                        `thenTc` \ (const_lie, dict_binds) ->
+    
+       let
+           full_bind = AbsBinds
+                           clas_tyvars'
+                           [this_dict_id]
+                           [(clas_tyvars', dm_id, local_dm_id)]
+                           emptyNameSet        -- No inlines (yet)
+                           (dict_binds `andMonoBinds` defm_bind)
+       in
+       returnTc (full_bind, const_lie)
+\end{code}
+
+\begin{code}
+checkFromThisClass :: Class -> [Id] -> RenamedMonoBinds -> NF_TcM s ()
+checkFromThisClass clas op_sel_ids mono_binds
+  = mapNF_Tc check_from_this_class bndrs       `thenNF_Tc_`
+    returnNF_Tc ()
+  where
+    check_from_this_class (bndr, loc)
+         | nameOccName bndr `elem` sel_names = returnNF_Tc ()
+         | otherwise                         = tcAddSrcLoc loc $
+                                               addErrTc (badMethodErr bndr clas)
+    sel_names = map getOccName op_sel_ids
+    bndrs = bagToList (collectMonoBinders mono_binds)
 \end{code}
+    
 
 @tcMethodBind@ is used to type-check both default-method and
 instance-decl method declarations.  We must type-check methods one at a
@@ -472,96 +515,103 @@ tyvar sets.
 \begin{code}
 tcMethodBind 
        :: Class
-       -> InstOrigin s
-       -> [TcType s]           -- Instance types
-       -> [TcTyVar s]          -- Free variables of those instance types
-                               --  they'll be signature tyvars, and we
-                               --  want to check that they don't bound
+       -> InstOrigin
+       -> [TcTyVar]            -- Instantiated type variables for the
+                               --  enclosing class/instance decl. 
+                               --  They'll be signature tyvars, and we
+                               --  want to check that they don't get bound
+       -> [TcType]             -- Instance types
+       -> TcThetaType          -- Available theta; this could be used to check
+                               --  the method signature, but actually that's done by
+                               --  the caller;  here, it's just used for the error message
        -> RenamedMonoBinds     -- Method binding (pick the right one from in here)
        -> [RenamedSig]         -- Pramgas (just for this one)
        -> Bool                 -- True <=> supply default decl if no explicit decl
                                --              This is true for instance decls, 
                                --              false for class decls
        -> (Id, Maybe Id)       -- The method selector and default-method Id
-       -> TcM s (TcMonoBinds s, LIE s, (LIE s, TcIdOcc s))
+       -> TcM s (TcMonoBinds, LIE, (LIE, TcId))
 
-tcMethodBind clas origin inst_tys inst_tyvars 
+tcMethodBind clas origin inst_tyvars inst_tys inst_theta
             meth_binds prags supply_default_bind
             (sel_id, maybe_dm_id)
- | no_user_bind && not supply_default_bind
- = pprPanic "tcMethodBind" (ppr clas <+> ppr inst_tys)
-
- | otherwise
  = tcGetSrcLoc                 `thenNF_Tc` \ loc -> 
 
-       -- Warn if no method binding, only if -fwarn-missing-methods
-   warnTc (opt_WarnMissingMethods && no_user_bind && no_user_default)
-         (omittedMethodWarn sel_id clas)               `thenNF_Tc_`
+   newMethod origin sel_id inst_tys    `thenNF_Tc` \ meth@(_, meth_id) ->
+   mkTcSig meth_id loc                 `thenNF_Tc` \ sig_info -> 
 
-   newMethod origin (RealId sel_id) inst_tys   `thenNF_Tc` \ meth@(_, TcId meth_id) ->
-   tcInstSigTcType (idType meth_id)    `thenNF_Tc` \ (tyvars', rho_ty') ->
    let
-     (theta', tau') = splitRhoTy rho_ty'
+     meth_name      = idName meth_id
+     maybe_user_bind = find_bind meth_name meth_binds
+
+     no_user_bind    = case maybe_user_bind of {Nothing -> True; other -> False}
+     no_user_default = case maybe_dm_id     of {Nothing -> True; other -> False}
+
+     meth_bind = case maybe_user_bind of
+                       Just bind -> bind
+                       Nothing   -> mk_default_bind meth_name loc
 
-     meth_name = idName meth_id
-     sig_info   = TySigInfo meth_name meth_id tyvars' theta' tau' loc
-     meth_bind = mk_meth_bind meth_name loc
      meth_prags = find_prags meth_name prags
    in
-   tcExtendLocalValEnv [meth_name] [meth_id] (
-       tcPragmaSigs meth_prags
-   )                                           `thenTc` \ (prag_info_fn, prag_binds1, prag_lie) ->
 
-       -- Check that the signatures match
-   tcExtendGlobalTyVars inst_tyvars (
+       -- Warn if no method binding, only if -fwarn-missing-methods
+   if no_user_bind && not supply_default_bind then
+       pprPanic "tcMethodBind" (ppr clas <+> ppr inst_tys)
+   else
+   warnTc (opt_WarnMissingMethods && no_user_bind && no_user_default)
+         (omittedMethodWarn sel_id clas)               `thenNF_Tc_`
+
+       -- Check the bindings; first add inst_tyvars to the envt
+       -- so that we don't quantify over them in nested places
+       -- The *caller* put the class/inst decl tyvars into the envt
+   tcExtendGlobalTyVars (mkVarSet inst_tyvars) (
      tcAddErrCtxt (methodCtxt sel_id)          $
-     tcBindWithSigs NotTopLevel [meth_name] meth_bind [sig_info]
-                   NonRecursive prag_info_fn   
-   )                                                   `thenTc` \ (binds, insts, _) ->
+     tcBindWithSigs NotTopLevel meth_bind 
+                   [sig_info] meth_prags NonRecursive 
+   )                                           `thenTc` \ (binds, insts, _) ->
+
+
+   tcExtendLocalValEnv [(meth_name, meth_id)] (
+       tcSpecSigs meth_prags
+   )                                           `thenTc` \ (prag_binds1, prag_lie) ->
 
        -- The prag_lie for a SPECIALISE pragma will mention the function
        -- itself, so we have to simplify them away right now lest they float
        -- outwards!
    bindInstsOfLocalFuns prag_lie [meth_id]     `thenTc` \ (prag_lie', prag_binds2) ->
 
+
        -- Now check that the instance type variables
        -- (or, in the case of a class decl, the class tyvars)
        -- have not been unified with anything in the environment
-   tcAddErrCtxt (monoCtxt sel_id) (
-     tcAddErrCtxt (sigCtxt sel_id) $
-     checkSigTyVars inst_tyvars (idType meth_id)
-   )                                                   `thenTc_` 
+   tcAddErrCtxtM (sigCtxt sig_msg (mkSigmaTy inst_tyvars inst_theta (idType meth_id))) $
+   checkSigTyVars inst_tyvars                                          `thenTc_` 
 
    returnTc (binds `AndMonoBinds` prag_binds1 `AndMonoBinds` prag_binds2, 
             insts `plusLIE` prag_lie', 
             meth)
  where
-   sel_name = idName sel_id
-
-   maybe_user_bind = find meth_binds
+   sig_msg ty = sep [ptext SLIT("When checking the expected type for"),
+                   nest 4 (ppr sel_name <+> dcolon <+> ppr ty)]
 
-   no_user_bind    = case maybe_user_bind of {Nothing -> True; other -> False}
-   no_user_default = case maybe_dm_id     of {Nothing -> True; other -> False}
-
-   find EmptyMonoBinds                        = Nothing
-   find (AndMonoBinds b1 b2)                  = find b1 `seqMaybe` find b2
-   find b@(FunMonoBind op_name _ _ _)         = if op_name == sel_name then Just b else Nothing
-   find b@(PatMonoBind (VarPatIn op_name) _ _) = if op_name == sel_name then Just b else Nothing
-   find other = panic "Urk! Bad instance method binding"
+   sel_name = idName sel_id
 
        -- The renamer just puts the selector ID as the binder in the method binding
        -- but we must use the method name; so we substitute it here.  Crude but simple.
-   mk_meth_bind meth_name loc
-     = case maybe_user_bind of
-        Just (FunMonoBind _ fix matches loc)    -> FunMonoBind meth_name fix matches loc
-        Just (PatMonoBind (VarPatIn _) rhs loc) -> PatMonoBind (VarPatIn meth_name) rhs loc
-        Nothing                                 -> mk_default_bind meth_name loc
+   find_bind meth_name (FunMonoBind op_name fix matches loc)
+       | op_name == sel_name = Just (FunMonoBind meth_name fix matches loc)
+   find_bind meth_name (PatMonoBind (VarPatIn op_name) grhss loc)
+       | op_name == sel_name = Just (PatMonoBind (VarPatIn meth_name) grhss loc)
+   find_bind meth_name (AndMonoBinds b1 b2)
+                             = find_bind meth_name b1 `seqMaybe` find_bind meth_name b2
+   find_bind meth_name other  = Nothing        -- Default case
+
 
        -- Find the prags for this method, and replace the
        -- selector name with the method name
    find_prags meth_name [] = []
-   find_prags meth_name (SpecSig name ty spec loc : prags)
-       | name == sel_name = SpecSig meth_name ty spec loc : find_prags meth_name prags
+   find_prags meth_name (SpecSig name ty loc : prags)
+       | name == sel_name = SpecSig meth_name ty loc : find_prags meth_name prags
    find_prags meth_name (InlineSig name loc : prags)
        | name == sel_name = InlineSig meth_name loc : find_prags meth_name prags
    find_prags meth_name (NoInlineSig name loc : prags)
@@ -570,7 +620,7 @@ tcMethodBind clas origin inst_tys inst_tyvars
 
    mk_default_bind local_meth_name loc
       = PatMonoBind (VarPatIn local_meth_name)
-                   (GRHSsAndBindsIn (unguardedRHS (default_expr loc) loc) EmptyBinds)
+                   (GRHSs (unguardedRHS (default_expr loc) loc) EmptyBinds Nothing)
                    loc
 
    default_expr loc 
@@ -590,21 +640,16 @@ Contexts and errors
 classArityErr class_name
   = ptext SLIT("Too many parameters for class") <+> quotes (ppr class_name)
 
-classDeclCtxt class_name
-  = ptext SLIT("In the class declaration for") <+> quotes (ppr class_name)
-
 superClassErr class_name sc
   = ptext SLIT("Illegal superclass constraint") <+> quotes (pprClassAssertion sc)
     <+> ptext SLIT("in declaration for class") <+> quotes (ppr class_name)
 
+defltMethCtxt class_name
+  = ptext SLIT("When checking the default methods for class") <+> quotes (ppr class_name)
+
 methodCtxt sel_id
   = ptext SLIT("In the definition for method") <+> quotes (ppr sel_id)
 
-monoCtxt sel_id
-  = sep [ptext SLIT("Probable cause: the right hand side of") <+> quotes (ppr sel_id),
-         nest 4 (ptext SLIT("mentions a top-level variable subject to the dreaded monomorphism restriction"))
-    ]
-
 badMethodErr bndr clas
   = hsep [ptext SLIT("Class"), quotes (ppr clas), 
          ptext SLIT("does not have a method"), quotes (ppr bndr)]