[project @ 2003-04-08 13:06:48 by simonpj]
[ghc-hetmet.git] / ghc / compiler / typecheck / TcExpr.lhs
index fcf9376..8230d2e 100644 (file)
@@ -13,18 +13,16 @@ import {-# SOURCE #-}       TcSplice( tcSpliceExpr, tcBracket )
 import HsSyn           ( HsReify(..), ReifyFlavour(..) )
 import TcType          ( isTauTy )
 import TcEnv           ( bracketOK, tcMetaTy, checkWellStaged, metaLevel )
-import TcSimplify      ( tcSimplifyBracket )
 import Name            ( isExternalName )
 import qualified DsMeta
 #endif
 
 import HsSyn           ( HsExpr(..), HsLit(..), ArithSeqInfo(..), recBindFields )
 import RnHsSyn         ( RenamedHsExpr, RenamedRecordBinds )
-import TcHsSyn         ( TcExpr, TcRecordBinds, hsLitType, mkHsDictApp, mkHsTyApp, mkHsLet )
+import TcHsSyn         ( TcExpr, TcRecordBinds, hsLitType, mkHsDictApp, mkHsTyApp, mkHsLet, (<$>) )
 import TcRnMonad
-import TcUnify         ( tcSubExp, tcGen, (<$>),
-                         unifyTauTy, unifyFunTy, unifyListTy, unifyPArrTy,
-                         unifyTupleTy )
+import TcUnify         ( tcSubExp, tcGen,
+                         unifyTauTy, unifyFunTy, unifyListTy, unifyPArrTy, unifyTupleTy )
 import BasicTypes      ( isMarkedStrict )
 import Inst            ( InstOrigin(..), 
                          newOverloadedLit, newMethodFromName, newIPDict,
@@ -35,7 +33,7 @@ import TcBinds                ( tcBindsAndThen )
 import TcEnv           ( tcLookupClass, tcLookupGlobal_maybe, tcLookupIdLvl,
                          tcLookupTyCon, tcLookupDataCon, tcLookupId
                        )
-import TcMatches       ( tcMatchesCase, tcMatchLambda, tcDoStmts )
+import TcMatches       ( tcMatchesCase, tcMatchLambda, tcDoStmts, tcThingWithSig )
 import TcMonoType      ( tcHsSigType, UserTypeCtxt(..) )
 import TcPat           ( badFieldCon )
 import TcMType         ( tcInstTyVars, tcInstType, newHoleTyVarTy, zapToType,
@@ -43,17 +41,16 @@ import TcMType              ( tcInstTyVars, tcInstType, newHoleTyVarTy, zapToType,
 import TcType          ( TcType, TcSigmaType, TcRhoType, TyVarDetails(VanillaTv),
                          tcSplitFunTys, tcSplitTyConApp, mkTyVarTys,
                          isSigmaTy, mkFunTy, mkFunTys,
-                         mkTyConApp, mkClassPred, tcFunArgTy,
+                         mkTyConApp, mkClassPred, 
                          tyVarsOfTypes, isLinearPred,
                          liftedTypeKind, openTypeKind, 
-                         tcSplitSigmaTy, tcTyConAppTyCon,
-                         tidyOpenType
+                         tcSplitSigmaTy, tidyOpenType
                        )
 import FieldLabel      ( FieldLabel, fieldLabelName, fieldLabelType, fieldLabelTyCon )
 import Id              ( Id, idType, recordSelectorFieldLabel, isRecordSelector )
 import DataCon         ( DataCon, dataConFieldLabels, dataConSig, dataConStrictMarks, dataConWrapId )
 import Name            ( Name )
-import TyCon           ( TyCon, tyConTyVars, tyConTheta, isAlgTyCon, tyConDataCons, isClassTyCon )
+import TyCon           ( TyCon, tyConTyVars, tyConTheta, isAlgTyCon, tyConDataCons )
 import Subst           ( mkTopTyVarSubst, substTheta, substTy )
 import VarSet          ( emptyVarSet, elemVarSet )
 import TysWiredIn      ( boolTy )
@@ -138,17 +135,10 @@ tcMonoExpr (HsIPVar ip) res_ty
 
 \begin{code}
 tcMonoExpr in_expr@(ExprWithTySig expr poly_ty) res_ty
- = addErrCtxt (exprSigCtxt in_expr)    $
-   tcHsSigType ExprSigCtxt poly_ty     `thenM` \ sig_tc_ty ->
-   tcExpr expr sig_tc_ty               `thenM` \ expr' ->
-
-       -- Must instantiate the outer for-alls of sig_tc_ty
-       -- else we risk instantiating a ? res_ty to a forall-type
-       -- which breaks the invariant that tcMonoExpr only returns phi-types
-   tcInstCall SignatureOrigin sig_tc_ty        `thenM` \ (inst_fn, inst_sig_ty) ->
-   tcSubExp res_ty inst_sig_ty         `thenM` \ co_fn ->
-
-   returnM (co_fn <$> inst_fn expr')
+ = addErrCtxt (exprSigCtxt in_expr)                    $
+   tcHsSigType ExprSigCtxt poly_ty                     `thenM` \ sig_tc_ty ->
+   tcThingWithSig sig_tc_ty (tcMonoExpr expr) res_ty   `thenM` \ (co_fn, expr') ->
+   returnM (co_fn <$> expr')
 
 tcMonoExpr (HsType ty) res_ty
   = failWithTc (text "Can't handle type argument:" <+> ppr ty)
@@ -174,7 +164,8 @@ tcMonoExpr (HsPar expr)    res_ty  = tcMonoExpr expr res_ty `thenM` \ expr' ->
 tcMonoExpr (HsSCC lbl expr) res_ty = tcMonoExpr expr res_ty    `thenM` \ expr' ->
                                     returnM (HsSCC lbl expr')
 
-
+tcMonoExpr (HsCoreAnn lbl expr) res_ty = tcMonoExpr expr res_ty `thenM` \ expr' ->  -- hdaume: core annotation
+                                         returnM (HsCoreAnn lbl expr')
 tcMonoExpr (NegApp expr neg_name) res_ty
   = tcMonoExpr (HsApp (HsVar neg_name) expr) res_ty
        -- ToDo: use tcSyntaxName
@@ -445,12 +436,8 @@ tcMonoExpr expr@(RecordUpd record_expr rbinds) res_ty
                   | (field_name, maybe_sel_id) <- field_names `zip` maybe_sel_ids,
                     not (is_selector maybe_sel_id)
                   ]
-       is_selector (Just (AnId sel_id))
-          = isRecordSelector sel_id &&         -- At the moment, class ops are
-                                               -- treated as record selectors, but
-                                               -- we want to exclude that case here
-            not (isClassTyCon (fieldLabelTyCon (recordSelectorFieldLabel sel_id)))
-       is_selector other = False
+       is_selector (Just (AnId sel_id)) = isRecordSelector sel_id      -- Excludes class ops
+       is_selector other                = False        
     in
     checkM (null bad_guys) (sequenceM bad_guys `thenM_` failM) `thenM_`
     
@@ -620,31 +607,7 @@ tcMonoExpr (PArrSeqIn _) _
        -- Rename excludes these cases otherwise
 
 tcMonoExpr (HsSplice n expr loc) res_ty = addSrcLoc loc (tcSpliceExpr n expr res_ty)
-  
-tcMonoExpr (HsBracket brack loc) res_ty
-  = addSrcLoc loc                      $
-    getStage                           `thenM` \ level ->
-    case bracketOK level of {
-       Nothing         -> failWithTc (illegalBracket level) ;
-       Just next_level ->
-
-       -- Typecheck expr to make sure it is valid,
-       -- but throw away the results.  We'll type check
-       -- it again when we actually use it.
-    newMutVar []                       `thenM` \ pending_splices ->
-    getLIEVar                          `thenM` \ lie_var ->
-
-    setStage (Brack next_level pending_splices lie_var) (
-       getLIE (tcBracket brack)
-    )                                  `thenM` \ (meta_ty, lie) ->
-    tcSimplifyBracket lie              `thenM_`  
-
-    unifyTauTy res_ty meta_ty          `thenM_`
-
-       -- Return the original expression, not the type-decorated one
-    readMutVar pending_splices         `thenM` \ pendings ->
-    returnM (HsBracketOut brack pendings)
-    }
+tcMonoExpr (HsBracket brack loc) res_ty = addSrcLoc loc (tcBracket brack res_ty)
 
 tcMonoExpr (HsReify (Reify flavour name)) res_ty
   = addErrCtxt (ptext SLIT("At the reification of") <+> ppr name)      $
@@ -695,17 +658,33 @@ tcApp fun args res_ty
        split_fun_ty fun_ty (length args)
     )                                          `thenM` \ (expected_arg_tys, actual_result_ty) ->
 
-       -- Now typecheck the args
-    mappM (tcArg fun)
-         (zip3 args expected_arg_tys [1..])    `thenM` \ args' ->
-
-       -- Unify with expected result after type-checking the args
-       -- so that the info from args percolates to actual_result_ty.
+       -- Unify with expected result before (was: after) type-checking the args
+       -- so that the info from res_ty (was: args) percolates to args (was actual_result_ty).
        -- This is when we might detect a too-few args situation.
        -- (One can think of cases when the opposite order would give
        -- a better error message.)
+       -- [March 2003: I'm experimenting with putting this first.  Here's an 
+       --              example where it actually makes a real difference
+       --    class C t a b | t a -> b
+       --    instance C Char a Bool
+       --
+       --    data P t a = forall b. (C t a b) => MkP b
+       --    data Q t   = MkQ (forall a. P t a)
+    
+       --    f1, f2 :: Q Char;
+       --    f1 = MkQ (MkP True)
+       --    f2 = MkQ (MkP True :: forall a. P Char a)
+       --
+       -- With the change, f1 will type-check, because the 'Char' info from
+       -- the signature is propagated into MkQ's argument. With the check
+       -- in the other order, the extra signature in f2 is reqd.]
+
     addErrCtxtM (checkArgsCtxt fun args res_ty actual_result_ty)
-                 (tcSubExp res_ty actual_result_ty)    `thenM` \ co_fn ->
+               (tcSubExp res_ty actual_result_ty)      `thenM` \ co_fn ->
+
+       -- Now typecheck the args
+    mappM (tcArg fun)
+         (zip3 args expected_arg_tys [1..])    `thenM` \ args' ->
 
     returnM (co_fn <$> foldl HsApp fun' args') 
 
@@ -861,7 +840,7 @@ tcId name   -- Look up the Id and instantiate its type
     loop fun fun_ty 
        | isSigmaTy fun_ty
        = tcInstCall orig fun_ty        `thenM` \ (inst_fn, tau) ->
-         loop (inst_fn fun) tau
+         loop (inst_fn <$> fun) tau
 
        | otherwise
        = returnM (fun, fun_ty)
@@ -1093,9 +1072,6 @@ parrCtxt expr
 predCtxt expr
   = hang (ptext SLIT("In the predicate expression:")) 4 (ppr expr)
 
-illegalBracket level
-  = ptext SLIT("Illegal bracket at level") <+> ppr level
-
 appCtxt fun args
   = ptext SLIT("In the application") <+> quotes (ppr the_app)
   where