Refactoring: mainly rename ic_env_tvs to ic_untch
[ghc-hetmet.git] / compiler / typecheck / TcRnMonad.lhs
index 8d92737..f171336 100644 (file)
@@ -35,7 +35,7 @@ import Bag
 import Outputable
 import UniqSupply
 import Unique
-import LazyUniqFM
+import UniqFM
 import DynFlags
 import StaticFlags
 import FastString
@@ -72,14 +72,16 @@ initTc hsc_env hsc_src keep_rn_syntax mod do_this
        tvs_var      <- newIORef emptyVarSet ;
        dfuns_var    <- newIORef emptyNameSet ;
        keep_var     <- newIORef emptyNameSet ;
-    used_rdrnames_var <- newIORef Set.empty ;
+        used_rdr_var <- newIORef Set.empty ;
        th_var       <- newIORef False ;
+       lie_var      <- newIORef emptyBag ;
        dfun_n_var   <- newIORef emptyOccSet ;
        type_env_var <- case hsc_type_env_var hsc_env of {
                            Just (_mod, te_var) -> return te_var ;
                            Nothing             -> newIORef emptyNameEnv } ;
        let {
-            maybe_rn_syntax empty_val
+             maybe_rn_syntax :: forall a. a -> Maybe a ;
+             maybe_rn_syntax empty_val
                | keep_rn_syntax = Just empty_val
                | otherwise      = Nothing ;
                        
@@ -98,7 +100,7 @@ initTc hsc_env hsc_src keep_rn_syntax mod do_this
                tcg_th_used   = th_var,
                tcg_exports  = [],
                tcg_imports  = emptyImportAvails,
-        tcg_used_rdrnames = used_rdrnames_var,
+                tcg_used_rdrnames = used_rdr_var,
                tcg_dus      = emptyDUs,
 
                 tcg_rn_imports = [],
@@ -106,6 +108,8 @@ initTc hsc_env hsc_src keep_rn_syntax mod do_this
                tcg_rn_decls   = maybe_rn_syntax emptyRnGroup,
 
                tcg_binds    = emptyLHsBinds,
+               tcg_sigs     = emptyNameSet,
+               tcg_ev_binds = emptyBag,
                tcg_warns    = NoWarnings,
                tcg_anns     = [],
                tcg_insts    = [],
@@ -114,9 +118,9 @@ initTc hsc_env hsc_src keep_rn_syntax mod do_this
                tcg_fords    = [],
                tcg_dfun_n   = dfun_n_var,
                tcg_keep     = keep_var,
-               tcg_doc      = Nothing,
-               tcg_hmi      = HaddockModInfo Nothing Nothing Nothing Nothing,
-                tcg_hpc      = False
+               tcg_doc_hdr  = Nothing,
+                tcg_hpc      = False,
+                tcg_main     = Nothing
             } ;
             lcl_env = TcLclEnv {
                tcl_errs       = errs_var,
@@ -127,9 +131,8 @@ initTc hsc_env hsc_src keep_rn_syntax mod do_this
                tcl_arrow_ctxt = NoArrowCtxt,
                tcl_env        = emptyNameEnv,
                tcl_tyvars     = tvs_var,
-               tcl_lie        = panic "initTc:LIE", -- only valid inside getLIE
-               tcl_tybinds    = panic "initTc:tybinds" 
-                                               -- only valid inside a getTyBinds
+               tcl_lie        = lie_var,
+               tcl_untch      = emptyVarSet
             } ;
        } ;
    
@@ -140,6 +143,13 @@ initTc hsc_env hsc_src keep_rn_syntax mod do_this
                          Right res -> return (Just res)
                          Left _    -> return Nothing } ;
 
+        -- Check for unsolved constraints
+       lie <- readIORef lie_var ;
+        if isEmptyBag lie 
+           then return ()
+           else pprPanic "initTc: unsolved constraints" 
+                         (pprWantedsWithLocs lie) ;
+
        -- Collect any error messages
        msgs <- readIORef errs_var ;
 
@@ -230,20 +240,29 @@ Command-line flags
 getDOpts :: TcRnIf gbl lcl DynFlags
 getDOpts = do { env <- getTopEnv; return (hsc_dflags env) }
 
+xoptM :: ExtensionFlag -> TcRnIf gbl lcl Bool
+xoptM flag = do { dflags <- getDOpts; return (xopt flag dflags) }
+
 doptM :: DynFlag -> TcRnIf gbl lcl Bool
 doptM flag = do { dflags <- getDOpts; return (dopt flag dflags) }
 
-setOptM :: DynFlag -> TcRnIf gbl lcl a -> TcRnIf gbl lcl a
+-- XXX setOptM and unsetOptM operate on different types. One should be renamed.
+
+setOptM :: ExtensionFlag -> TcRnIf gbl lcl a -> TcRnIf gbl lcl a
 setOptM flag = updEnv (\ env@(Env { env_top = top }) ->
-                        env { env_top = top { hsc_dflags = dopt_set (hsc_dflags top) flag}} )
+                        env { env_top = top { hsc_dflags = xopt_set_flattened (hsc_dflags top) flag}} )
 
 unsetOptM :: DynFlag -> TcRnIf gbl lcl a -> TcRnIf gbl lcl a
 unsetOptM flag = updEnv (\ env@(Env { env_top = top }) ->
                         env { env_top = top { hsc_dflags = dopt_unset (hsc_dflags top) flag}} )
 
 -- | Do it flag is true
-ifOptM :: DynFlag -> TcRnIf gbl lcl () -> TcRnIf gbl lcl ()
-ifOptM flag thing_inside = do { b <- doptM flag; 
+ifDOptM :: DynFlag -> TcRnIf gbl lcl () -> TcRnIf gbl lcl ()
+ifDOptM flag thing_inside = do { b <- doptM flag; 
+                               if b then thing_inside else return () }
+
+ifXOptM :: ExtensionFlag -> TcRnIf gbl lcl () -> TcRnIf gbl lcl ()
+ifXOptM flag thing_inside = do { b <- xoptM flag; 
                                if b then thing_inside else return () }
 
 getGhcMode :: TcRnIf gbl lcl GhcMode
@@ -342,9 +361,37 @@ instance MonadUnique (IOEnv (Env gbl lcl)) where
 %************************************************************************
 
 \begin{code}
-traceTc, traceRn, traceSplice :: SDoc -> TcRn ()
+newTcRef :: a -> TcRnIf gbl lcl (TcRef a)
+newTcRef = newMutVar 
+
+readTcRef :: TcRef a -> TcRnIf gbl lcl a
+readTcRef = readMutVar
+
+writeTcRef :: TcRef a -> a -> TcRnIf gbl lcl ()
+writeTcRef = writeMutVar
+
+updTcRef :: TcRef a -> (a -> a) -> TcRnIf gbl lcl ()
+updTcRef = updMutVar
+\end{code}
+
+%************************************************************************
+%*                                                                     *
+               Debugging
+%*                                                                     *
+%************************************************************************
+
+\begin{code}
+traceTc :: String -> SDoc -> TcRn () 
+traceTc = traceTcN 1
+
+traceTcN :: Int -> String -> SDoc -> TcRn () 
+traceTcN level herald doc
+  | level <= opt_TraceLevel = traceOptTcRn Opt_D_dump_tc_trace $
+                              hang (text herald) 2 doc
+  | otherwise               = return ()
+
+traceRn, traceSplice :: SDoc -> TcRn ()
 traceRn      = traceOptTcRn Opt_D_dump_rn_trace
-traceTc      = traceOptTcRn Opt_D_dump_tc_trace
 traceSplice  = traceOptTcRn Opt_D_dump_splices
 
 
@@ -354,16 +401,17 @@ traceHiDiffs = traceOptIf Opt_D_dump_hi_diffs
 
 
 traceOptIf :: DynFlag -> SDoc -> TcRnIf m n ()  -- No RdrEnv available, so qualify everything
-traceOptIf flag doc = ifOptM flag $
+traceOptIf flag doc = ifDOptM flag $
                      liftIO (printForUser stderr alwaysQualify doc)
 
 traceOptTcRn :: DynFlag -> SDoc -> TcRn ()
-traceOptTcRn flag doc = ifOptM flag $ do
-                       { ctxt <- getErrCtxt
-                       ; loc  <- getSrcSpanM
-                       ; env0 <- tcInitTidyEnv
-                       ; err_info <- mkErrInfo env0 ctxt 
-                       ; let real_doc = mkLocMessage loc (doc $$ err_info)
+-- Output the message, with current location if opt_PprStyle_Debug
+traceOptTcRn flag doc = ifDOptM flag $ do
+                       { loc  <- getSrcSpanM
+                       ; let real_doc 
+                                | opt_PprStyle_Debug = mkLocMessage loc doc
+                                | otherwise = doc   -- The full location is 
+                                                   -- usually way too much
                        ; dumpTcRn real_doc }
 
 dumpTcRn :: SDoc -> TcRn ()
@@ -376,7 +424,7 @@ debugDumpTcRn doc | opt_NoDebugOutput = return ()
                   | otherwise         = dumpTcRn doc
 
 dumpOptTcRn :: DynFlag -> SDoc -> TcRn ()
-dumpOptTcRn flag doc = ifOptM flag (dumpTcRn doc)
+dumpOptTcRn flag doc = ifDOptM flag (dumpTcRn doc)
 \end{code}
 
 
@@ -455,6 +503,7 @@ wrapLocSndM fn (L loc a) =
     return (b, L loc c)
 \end{code}
 
+Reporting errors
 
 \begin{code}
 getErrsVar :: TcRn (TcRef Messages)
@@ -469,47 +518,25 @@ addErr msg = do { loc <- getSrcSpanM ; addErrAt loc msg }
 failWith :: Message -> TcRn a
 failWith msg = addErr msg >> failM
 
-addLocErr :: Located e -> (e -> Message) -> TcRn ()
-addLocErr (L loc e) fn = addErrAt loc (fn e)
-
 addErrAt :: SrcSpan -> Message -> TcRn ()
-addErrAt loc msg = addLongErrAt loc msg empty
-
-addLongErrAt :: SrcSpan -> Message -> Message -> TcRn ()
-addLongErrAt loc msg extra
-  = do { traceTc (ptext (sLit "Adding error:") <+> (mkLocMessage loc (msg $$ extra))) ;        
-        errs_var <- getErrsVar ;
-        rdr_env <- getGlobalRdrEnv ;
-         dflags <- getDOpts ;
-        let { err = mkLongErrMsg loc (mkPrintUnqualified dflags rdr_env) msg extra } ;
-        (warns, errs) <- readMutVar errs_var ;
-        writeMutVar errs_var (warns, errs `snocBag` err) }
+-- addErrAt is mainly (exclusively?) used by the renamer, where
+-- tidying is not an issue, but it's all lazy so the extra
+-- work doesn't matter
+addErrAt loc msg = do { ctxt <- getErrCtxt 
+                     ; tidy_env <- tcInitTidyEnv
+                      ; err_info <- mkErrInfo tidy_env ctxt
+                     ; addLongErrAt loc msg err_info }
 
 addErrs :: [(SrcSpan,Message)] -> TcRn ()
 addErrs msgs = mapM_ add msgs
             where
               add (loc,msg) = addErrAt loc msg
 
-addReport :: Message -> TcRn ()
-addReport msg = do loc <- getSrcSpanM; addReportAt loc msg
-
-addReportAt :: SrcSpan -> Message -> TcRn ()
-addReportAt loc msg
-  = do { errs_var <- getErrsVar ;
-        rdr_env <- getGlobalRdrEnv ;
-         dflags <- getDOpts ;
-        let { warn = mkWarnMsg loc (mkPrintUnqualified dflags rdr_env) msg } ;
-        (warns, errs) <- readMutVar errs_var ;
-        writeMutVar errs_var (warns `snocBag` warn, errs) }
-
 addWarn :: Message -> TcRn ()
-addWarn msg = addReport (ptext (sLit "Warning:") <+> msg)
+addWarn msg = addReport (ptext (sLit "Warning:") <+> msg) empty
 
 addWarnAt :: SrcSpan -> Message -> TcRn ()
-addWarnAt loc msg = addReportAt loc (ptext (sLit "Warning:") <+> msg)
-
-addLocWarn :: Located e -> (e -> Message) -> TcRn ()
-addLocWarn (L loc e) fn = addReportAt loc (fn e)
+addWarnAt loc msg = addReportAt loc (ptext (sLit "Warning:") <+> msg) empty
 
 checkErr :: Bool -> Message -> TcRn ()
 -- Add the error if the bool is False
@@ -522,8 +549,8 @@ warnIf False _   = return ()
 addMessages :: Messages -> TcRn ()
 addMessages (m_warns, m_errs)
   = do { errs_var <- getErrsVar ;
-        (warns, errs) <- readMutVar errs_var ;
-        writeMutVar errs_var (warns `unionBags` m_warns,
+        (warns, errs) <- readTcRef errs_var ;
+        writeTcRef errs_var (warns `unionBags` m_warns,
                               errs  `unionBags` m_errs) }
 
 discardWarnings :: TcRn a -> TcRn a
@@ -534,24 +561,56 @@ discardWarnings :: TcRn a -> TcRn a
 discardWarnings thing_inside
   | opt_PprStyle_Debug = thing_inside
   | otherwise
-  = do { errs_var <- newMutVar emptyMessages
+  = do { errs_var <- newTcRef emptyMessages
        ; result <- setErrsVar errs_var thing_inside
-       ; (_warns, errs) <- readMutVar errs_var
+       ; (_warns, errs) <- readTcRef errs_var
        ; addMessages (emptyBag, errs)
        ; return result }
 \end{code}
 
 
+%************************************************************************
+%*                                                                     *
+       Shared error message stuff: renamer and typechecker
+%*                                                                     *
+%************************************************************************
+
+\begin{code}
+addReport :: Message -> Message -> TcRn ()
+addReport msg extra_info = do loc <- getSrcSpanM; addReportAt loc msg extra_info
+
+addReportAt :: SrcSpan -> Message -> Message -> TcRn ()
+addReportAt loc msg extra_info
+  = do { errs_var <- getErrsVar ;
+        rdr_env <- getGlobalRdrEnv ;
+         dflags <- getDOpts ;
+        let { warn = mkLongWarnMsg loc (mkPrintUnqualified dflags rdr_env)
+                                   msg extra_info } ;
+        (warns, errs) <- readTcRef errs_var ;
+        writeTcRef errs_var (warns `snocBag` warn, errs) }
+
+addLongErrAt :: SrcSpan -> Message -> Message -> TcRn ()
+addLongErrAt loc msg extra
+  = do { traceTc "Adding error:" (mkLocMessage loc (msg $$ extra)) ;   
+        errs_var <- getErrsVar ;
+        rdr_env <- getGlobalRdrEnv ;
+         dflags <- getDOpts ;
+        let { err = mkLongErrMsg loc (mkPrintUnqualified dflags rdr_env) msg extra } ;
+        (warns, errs) <- readTcRef errs_var ;
+        writeTcRef errs_var (warns, errs `snocBag` err) }
+\end{code}
+
+
 \begin{code}
 try_m :: TcRn r -> TcRn (Either IOEnvFailure r)
 -- Does try_m, with a debug-trace on failure
 try_m thing 
   = do { mb_r <- tryM thing ;
         case mb_r of 
-            Left exn -> do { traceTc (exn_msg exn); return mb_r }
+            Left exn -> do { traceTc "tryTc/recoverM recovering from" $
+                                      text (showException exn)
+                            ; return mb_r }
             Right _  -> return mb_r }
-  where
-    exn_msg exn = text "tryTc/recoverM recovering from" <+> text (showException exn)
 
 -----------------------
 recoverM :: TcRn r     -- Recovery action; do this if the main one fails
@@ -585,9 +644,9 @@ tryTc :: TcRn a -> TcRn (Messages, Maybe a)
 -- It also returns all the errors and warnings accumulated by m
 -- It always succeeds (never raises an exception)
 tryTc m 
- = do {        errs_var <- newMutVar emptyMessages ;
+ = do {        errs_var <- newTcRef emptyMessages ;
        res  <- try_m (setErrsVar errs_var m) ; 
-       msgs <- readMutVar errs_var ;
+       msgs <- readTcRef errs_var ;
        return (msgs, case res of
                            Left _  -> Nothing
                            Right val -> Just val)
@@ -618,10 +677,10 @@ tryTcLIE :: TcM a -> TcM (Messages, Maybe a)
 -- for the thing is propagated only if there are no errors
 -- Hence it's restricted to the type-check monad
 tryTcLIE thing_inside
-  = do  { ((msgs, mb_res), lie) <- getLIE (tryTcErrs thing_inside) ;
+  = do  { ((msgs, mb_res), lie) <- getConstraints (tryTcErrs thing_inside) ;
        ; case mb_res of
            Nothing  -> return (msgs, Nothing)
-           Just val -> do { extendLIEs lie; return (msgs, Just val) }
+           Just val -> do { emitConstraints lie; return (msgs, Just val) }
        }
 
 -----------------------
@@ -659,7 +718,7 @@ ifErrsM :: TcRn r -> TcRn r -> TcRn r
 -- otherwise does 'main'
 ifErrsM bale_out normal
  = do { errs_var <- getErrsVar ;
-       msgs <- readMutVar errs_var ;
+       msgs <- readTcRef errs_var ;
        dflags <- getDOpts ;
        if errorsFound dflags msgs then
           bale_out
@@ -674,8 +733,7 @@ failIfErrsM = ifErrsM failM (return ())
 
 %************************************************************************
 %*                                                                     *
-       Context management and error message generation
-                   for the type checker
+       Context management for the type checker
 %*                                                                     *
 %************************************************************************
 
@@ -708,18 +766,22 @@ maybeAddErrCtxt Nothing    thing_inside = thing_inside
 popErrCtxt :: TcM a -> TcM a
 popErrCtxt = updCtxt (\ msgs -> case msgs of { [] -> []; (_ : ms) -> ms })
 
-getInstLoc :: InstOrigin -> TcM InstLoc
-getInstLoc origin
+getCtLoc :: orig -> TcM (CtLoc orig)
+getCtLoc origin
   = do { loc <- getSrcSpanM ; env <- getLclEnv ;
-        return (InstLoc origin loc (tcl_ctxt env)) }
+        return (CtLoc origin loc (tcl_ctxt env)) }
 
-setInstCtxt :: InstLoc -> TcM a -> TcM a
--- Add the SrcSpan and context from the first Inst in the list
---     (they all have similar locations)
-setInstCtxt (InstLoc _ src_loc ctxt) thing_inside
+setCtLoc :: CtLoc orig -> TcM a -> TcM a
+setCtLoc (CtLoc _ src_loc ctxt) thing_inside
   = setSrcSpan src_loc (setErrCtxt ctxt thing_inside)
 \end{code}
 
+%************************************************************************
+%*                                                                     *
+            Error message generation (type checker)
+%*                                                                     *
+%************************************************************************
+
     The addErrTc functions add an error message, but do not cause failure.
     The 'M' variants pass a TidyEnv that has already been used to
     tidy up the message; we then use it to tidy the context messages
@@ -766,7 +828,7 @@ addWarnTcM :: (TidyEnv, Message) -> TcM ()
 addWarnTcM (env0, msg)
  = do { ctxt <- getErrCtxt ;
        err_info <- mkErrInfo env0 ctxt ;
-       addReport (vcat [ptext (sLit "Warning:") <+> msg, err_info]) }
+       addReport (ptext (sLit "Warning:") <+> msg) err_info }
 
 warnTc :: Bool -> Message -> TcM ()
 warnTc warn_if_true warn_msg
@@ -817,7 +879,7 @@ mkErrInfo env ctxts
    go :: Int -> TidyEnv -> [ErrCtxt] -> TcM SDoc
    go _ _   [] = return empty
    go n env ((is_landmark, ctxt) : ctxts)
-     | is_landmark || opt_PprStyle_Debug || n < mAX_CONTEXTS
+     | is_landmark || n < mAX_CONTEXTS -- Too verbose || opt_PprStyle_Debug 
      = do { (env', msg) <- ctxt env
          ; let n' = if is_landmark then n else n+1
           ; rest <- go n' env' ctxts
@@ -840,52 +902,77 @@ debugTc thing
 
 %************************************************************************
 %*                                                                     *
-            Type constraints (the so-called LIE)
+            Type constraints
 %*                                                                     *
 %************************************************************************
 
 \begin{code}
+newTcEvBinds :: TcM EvBindsVar
+newTcEvBinds = do { ref <- newTcRef emptyEvBindMap
+                         ; uniq <- newUnique
+                         ; return (EvBindsVar ref uniq) }
+
+extendTcEvBinds :: TcEvBinds -> EvVar -> EvTerm -> TcM TcEvBinds
+extendTcEvBinds binds@(TcEvBinds binds_var) var rhs 
+  = do { addTcEvBind binds_var var rhs
+       ; return binds }
+extendTcEvBinds (EvBinds bnds) var rhs
+  = return (EvBinds (bnds `snocBag` EvBind var rhs))
+
+addTcEvBind :: EvBindsVar -> EvVar -> EvTerm -> TcM ()
+-- Add a binding to the TcEvBinds by side effect
+addTcEvBind (EvBindsVar ev_ref _) var rhs
+  = do { bnds <- readTcRef ev_ref
+       ; writeTcRef ev_ref (extendEvBinds bnds var rhs) }
+
 chooseUniqueOccTc :: (OccSet -> OccName) -> TcM OccName
 chooseUniqueOccTc fn =
   do { env <- getGblEnv
      ; let dfun_n_var = tcg_dfun_n env
-     ; set <- readMutVar dfun_n_var
+     ; set <- readTcRef dfun_n_var
      ; let occ = fn set
-     ; writeMutVar dfun_n_var (extendOccSet set occ)
-     ; return occ
-     }
+     ; writeTcRef dfun_n_var (extendOccSet set occ)
+     ; return occ }
+
+getConstraintVar :: TcM (TcRef WantedConstraints)
+getConstraintVar = do { env <- getLclEnv; return (tcl_lie env) }
 
-getLIEVar :: TcM (TcRef LIE)
-getLIEVar = do { env <- getLclEnv; return (tcl_lie env) }
+setConstraintVar :: TcRef WantedConstraints -> TcM a -> TcM a
+setConstraintVar lie_var = updLclEnv (\ env -> env { tcl_lie = lie_var })
 
-setLIEVar :: TcRef LIE -> TcM a -> TcM a
-setLIEVar lie_var = updLclEnv (\ env -> env { tcl_lie = lie_var })
+emitConstraints :: WantedConstraints -> TcM ()
+emitConstraints ct
+  = do { lie_var <- getConstraintVar ;
+        updTcRef lie_var (`andWanteds` ct) }
 
-getLIE :: TcM a -> TcM (a, [Inst])
--- (getLIE m) runs m, and returns the type constraints it generates
-getLIE thing_inside
-  = do { lie_var <- newMutVar emptyLIE ;
+emitConstraint :: WantedConstraint -> TcM ()
+emitConstraint ct
+  = do { lie_var <- getConstraintVar ;
+        updTcRef lie_var (`extendWanteds` ct) }
+
+getConstraints :: TcM a -> TcM (a, WantedConstraints)
+-- (getConstraints m) runs m, and returns the type constraints it generates
+getConstraints thing_inside
+  = do { lie_var <- newTcRef emptyWanteds ;
         res <- updLclEnv (\ env -> env { tcl_lie = lie_var }) 
                          thing_inside ;
-        lie <- readMutVar lie_var ;
-        return (res, lieToList lie) }
-
-extendLIE :: Inst -> TcM ()
-extendLIE inst
-  = do { lie_var <- getLIEVar ;
-        lie <- readMutVar lie_var ;
-        writeMutVar lie_var (inst `consLIE` lie) }
-
-extendLIEs :: [Inst] -> TcM ()
-extendLIEs [] 
-  = return ()
-extendLIEs insts
-  = do { lie_var <- getLIEVar ;
-        lie <- readMutVar lie_var ;
-        writeMutVar lie_var (mkLIE insts `plusLIE` lie) }
-\end{code}
+        lie <- readTcRef lie_var ;
+        return (res, lie) }
+
+setUntouchables :: TcTyVarSet -> TcM a -> TcM a
+setUntouchables untch_tvs thing_inside
+  = updLclEnv (\ env -> env { tcl_untch = untch_tvs }) thing_inside 
+
+getUntouchables :: TcM TcTyVarSet 
+getUntouchables = do { env <- getLclEnv; return (tcl_untch env) } 
+   -- NB: no need to zonk this TcTyVarSet: they are, after all, untouchable!
+
+isUntouchable :: TcTyVar -> TcM Bool
+isUntouchable tv = do { env <- getLclEnv; return (tv `elemVarSet` tcl_untch env) }
+
+getLclTypeEnv :: TcM (NameEnv TcTyThing)
+getLclTypeEnv = do { env <- getLclEnv; return (tcl_env env) }
 
-\begin{code}
 setLclTypeEnv :: TcLclEnv -> TcM a -> TcM a
 -- Set the local type envt, but do *not* disturb other fields,
 -- notably the lie_var
@@ -899,61 +986,23 @@ setLclTypeEnv lcl_env thing_inside
 
 %************************************************************************
 %*                                                                     *
-            Meta type variable bindings
-%*                                                                     *
-%************************************************************************
-
-\begin{code}
-getTcTyVarBindsVar :: TcM (TcRef TcTyVarBinds)
-getTcTyVarBindsVar = do { env <- getLclEnv; return (tcl_tybinds env) }
-
-getTcTyVarBinds :: TcM a -> TcM (a, TcTyVarBinds)
-getTcTyVarBinds thing_inside
-  = do { tybinds_var <- newMutVar emptyBag
-       ; res <- updLclEnv (\ env -> env { tcl_tybinds = tybinds_var }) 
-                         thing_inside
-       ; tybinds <- readMutVar tybinds_var
-       ; return (res, tybinds) 
-       }
-
-bindMetaTyVar :: TcTyVar -> TcType -> TcM ()
-bindMetaTyVar tv ty
-  = do { ASSERTM2( do { details <- readMutVar (metaTvRef tv)
-                      ; return (isFlexi details) }, ppr tv )
-       ; tybinds_var <- getTcTyVarBindsVar
-       ; tybinds <- readMutVar tybinds_var
-       ; writeMutVar tybinds_var (tybinds `snocBag` TcTyVarBind tv ty) 
-       }
-
-getTcTyVarBindsRelation :: TcM [(TcTyVar, TcTyVarSet)]
-getTcTyVarBindsRelation
-  = do { tybinds_var <- getTcTyVarBindsVar
-       ; tybinds <- readMutVar tybinds_var
-       ; return $ map freeTvs (bagToList tybinds)
-       }
-  where
-    freeTvs (TcTyVarBind tv ty) = (tv, tyVarsOfType ty)
-\end{code}
-
-%************************************************************************
-%*                                                                     *
             Template Haskell context
 %*                                                                     *
 %************************************************************************
 
 \begin{code}
 recordThUse :: TcM ()
-recordThUse = do { env <- getGblEnv; writeMutVar (tcg_th_used env) True }
+recordThUse = do { env <- getGblEnv; writeTcRef (tcg_th_used env) True }
 
 keepAliveTc :: Id -> TcM ()    -- Record the name in the keep-alive set
 keepAliveTc id 
   | isLocalId id = do { env <- getGblEnv; 
-                     ; updMutVar (tcg_keep env) (`addOneToNameSet` idName id) }
+                     ; updTcRef (tcg_keep env) (`addOneToNameSet` idName id) }
   | otherwise = return ()
 
 keepAliveSetTc :: NameSet -> TcM ()    -- Record the name in the keep-alive set
 keepAliveSetTc ns = do { env <- getGblEnv; 
-                      ; updMutVar (tcg_keep env) (`unionNameSets` ns) }
+                      ; updTcRef (tcg_keep env) (`unionNameSets` ns) }
 
 getStage :: TcM ThStage
 getStage = do { env <- getLclEnv; return (tcl_th_ctxt env) }
@@ -996,7 +1045,7 @@ initIfaceTcRn :: IfG a -> TcRn a
 initIfaceTcRn thing_inside
   = do  { tcg_env <- getGblEnv 
        ; let { if_env = IfGblEnv { if_rec_types = Just (tcg_mod tcg_env, get_type_env) }
-             ; get_type_env = readMutVar (tcg_type_env_var tcg_env) }
+             ; get_type_env = readTcRef (tcg_type_env_var tcg_env) }
        ; setEnvs (if_env, ()) thing_inside }
 
 initIfaceExtCore :: IfL a -> TcRn a
@@ -1015,7 +1064,7 @@ initIfaceCheck :: HscEnv -> IfG a -> IO a
 -- Initialise the environment with no useful info at all
 initIfaceCheck hsc_env do_this
  = do let rec_types = case hsc_type_env_var hsc_env of
-                         Just (mod,var) -> Just (mod, readMutVar var)
+                         Just (mod,var) -> Just (mod, readTcRef var)
                          Nothing        -> Nothing
           gbl_env = IfGblEnv { if_rec_types = rec_types }
       initTcRnIf 'i' hsc_env gbl_env () do_this
@@ -1025,8 +1074,8 @@ initIfaceTc :: ModIface
 -- Used when type-checking checking an up-to-date interface file
 -- No type envt from the current module, but we do know the module dependencies
 initIfaceTc iface do_this
- = do  { tc_env_var <- newMutVar emptyTypeEnv
-       ; let { gbl_env = IfGblEnv { if_rec_types = Just (mod, readMutVar tc_env_var) } ;
+ = do  { tc_env_var <- newTcRef emptyTypeEnv
+       ; let { gbl_env = IfGblEnv { if_rec_types = Just (mod, readTcRef tc_env_var) } ;
              ; if_lenv = mkIfLclEnv mod doc
           }
        ; setEnvs (gbl_env, if_lenv) (do_this tc_env_var)
@@ -1090,9 +1139,9 @@ forkM_maybe doc thing_inside
                    -- Bleat about errors in the forked thread, if -ddump-if-trace is on
                    -- Otherwise we silently discard errors. Errors can legitimately
                    -- happen when compiling interface signatures (see tcInterfaceSigs)
-                     ifOptM Opt_D_dump_if_trace 
+                     ifDOptM Opt_D_dump_if_trace 
                             (print_errs (hang (text "forkM failed:" <+> doc)
-                                            4 (text (show exn))))
+                                            2 (text (show exn))))
 
                    ; traceIf (text "} ending fork (badly)" <+> doc)
                    ; return Nothing }