TcInteract.lhs 97.7 KB
Newer Older
1
\begin{code}
Ian Lynagh's avatar
Ian Lynagh committed
2 3 4 5 6 7 8
{-# OPTIONS -fno-warn-tabs #-}
-- The above warning supression flag is a temporary kludge.
-- While working on this module you are encouraged to remove it and
-- detab the module (please do the detabbing in a separate patch). See
--     http://hackage.haskell.org/trac/ghc/wiki/Commentary/CodingStyle#TabsvsSpaces
-- for details

9
module TcInteract ( 
10 11 12
     solveInteract, solveInteractGiven, solveInteractWanted,
     AtomicInert, tyVarsOfInert, 
     InertSet, emptyInert, updInertSet, extractUnsolved, solveOne,
13 14 15 16
  ) where  

#include "HsVersions.h"

17

18 19 20 21
import BasicTypes 
import TcCanonical
import VarSet
import Type
dimitris's avatar
dimitris committed
22
import Unify
23 24 25 26 27

import Id 
import Var

import TcType
28
import HsBinds
29

30
import Inst( tyVarsOfEvVar )
31 32
import Class
import TyCon
33 34 35 36 37 38 39
import Name

import FunDeps

import Coercion
import Outputable

40
import TcRnTypes
41
import TcErrors
42
import TcSMonad
43
import Maybes( orElse )
44
import Bag
45

46
import Control.Monad( when )
47 48
import Unique
import UniqFM
49 50 51 52
import FastString ( sLit ) 
import DynFlags
\end{code}

53
Note [InertSet invariants]
54 55 56 57 58 59 60 61
~~~~~~~~~~~~~~~~~~~~~~~~~~~
An InertSet is a bag of canonical constraints, with the following invariants:

  1 No two constraints react with each other. 
    
    A tricky case is when there exists a given (solved) dictionary 
    constraint and a wanted identical constraint in the inert set, but do 
    not react because reaction would create loopy dictionary evidence for 
batterseapower's avatar
batterseapower committed
62
    the wanted. See note [Recursive instances and superclases]
63 64 65 66 67

  2 Given equalities form an idempotent substitution [none of the
    given LHS's occur in any of the given RHS's or reactant parts]

  3 Wanted equalities also form an idempotent substitution
68

69 70 71 72 73 74
  4 The entire set of equalities is acyclic.

  5 Wanted dictionaries are inert with the top-level axiom set 

  6 Equalities of the form tv1 ~ tv2 always have a touchable variable
    on the left (if possible).
75 76

  7 No wanted constraints tv1 ~ tv2 with tv1 touchable. Such constraints
77 78
    will be marked as solved right before being pushed into the inert set. 
    See note [Touchables and givens].
79

dimitris's avatar
dimitris committed
80 81 82 83 84
  8 No Given constraint mentions a touchable unification variable, but 
    Given/Solved may do so. 

  9 Given constraints will also have their superclasses in the inert set, 
    but Given/Solved will not. 
85 86 87 88 89 90 91 92 93 94 95 96 97 98
 
Note that 6 and 7 are /not/ enforced by canonicalization but rather by 
insertion in the inert list, ie by TcInteract. 

During the process of solving, the inert set will contain some
previously given constraints, some wanted constraints, and some given
constraints which have arisen from solving wanted constraints. For
now we do not distinguish between given and solved constraints.

Note that we must switch wanted inert items to given when going under an
implication constraint (when in top-level inference mode).

\begin{code}

99
data CCanMap a = CCanMap { cts_given   :: UniqFM CanonicalCts
100
                                          -- Invariant: all Given
101
                         , cts_derived :: UniqFM CanonicalCts 
102
                                          -- Invariant: all Derived
103
                         , cts_wanted  :: UniqFM CanonicalCts } 
104
                                          -- Invariant: all Wanted
105

106 107 108 109
cCanMapToBag :: CCanMap a -> CanonicalCts 
cCanMapToBag cmap = foldUFM unionBags rest_wder (cts_given cmap)
  where rest_wder = foldUFM unionBags rest_der  (cts_wanted cmap) 
        rest_der  = foldUFM unionBags emptyCCan (cts_derived cmap)
110 111

emptyCCanMap :: CCanMap a 
112
emptyCCanMap = CCanMap { cts_given = emptyUFM, cts_derived = emptyUFM, cts_wanted = emptyUFM } 
113

114
updCCanMap:: Uniquable a => (a,CanonicalCt) -> CCanMap a -> CCanMap a 
115 116
updCCanMap (a,ct) cmap 
  = case cc_flavor ct of 
117 118 119 120
      Wanted {}  -> cmap { cts_wanted  = insert_into (cts_wanted cmap)  } 
      Given {}   -> cmap { cts_given   = insert_into (cts_given cmap)   }
      Derived {} -> cmap { cts_derived = insert_into (cts_derived cmap) }
  where 
121
    insert_into m = addToUFM_C unionBags m a (singleCCan ct)
122 123

getRelevantCts :: Uniquable a => a -> CCanMap a -> (CanonicalCts, CCanMap a) 
124 125
-- Gets the relevant constraints and returns the rest of the CCanMap
getRelevantCts a cmap 
126 127 128 129 130 131
    = let relevant = lookup (cts_wanted cmap) `unionBags`
                     lookup (cts_given cmap)  `unionBags`
                     lookup (cts_derived cmap) 
          residual_map = cmap { cts_wanted  = delFromUFM (cts_wanted cmap) a
                              , cts_given   = delFromUFM (cts_given cmap) a
                              , cts_derived = delFromUFM (cts_derived cmap) a }
132
      in (relevant, residual_map) 
133 134
  where
    lookup map = lookupUFM map a `orElse` emptyCCan
135

136
extractUnsolvedCMap :: CCanMap a -> (CanonicalCts, CCanMap a)
137 138 139
-- Gets the wanted or derived constraints and returns a residual
-- CCanMap with only givens.
extractUnsolvedCMap cmap =
140 141
  let wntd = foldUFM unionBags emptyCCan (cts_wanted cmap)
      derd = foldUFM unionBags emptyCCan (cts_derived cmap)
142
  in (wntd `unionBags` derd, 
143
      cmap { cts_wanted = emptyUFM, cts_derived = emptyUFM })
144

145

146
-- See Note [InertSet invariants]
147
data InertSet 
148
  = IS { inert_eqs          :: CanonicalCts               -- Equalities only (CTyEqCan)
149
       , inert_dicts        :: CCanMap Class              -- Dictionaries only
batterseapower's avatar
batterseapower committed
150 151
       , inert_ips          :: CCanMap (IPName Name)      -- Implicit parameters
       , inert_irreds       :: CanonicalCts               -- Irreducible predicates
152 153
       , inert_frozen       :: CanonicalCts
       , inert_funeqs       :: CCanMap TyCon              -- Type family equalities only
154 155
               -- This representation allows us to quickly get to the relevant 
               -- inert constraints when interacting a work item with the inert set.
156
       }
157

158 159 160 161
tyVarsOfInert :: InertSet -> TcTyVarSet 
tyVarsOfInert (IS { inert_eqs    = eqs
                  , inert_dicts  = dictmap
                  , inert_ips    = ipmap
batterseapower's avatar
batterseapower committed
162
                  , inert_irreds = irreds
163 164 165
                  , inert_frozen = frozen
                  , inert_funeqs = funeqmap }) = tyVarsOfCanonicals cts
  where
batterseapower's avatar
batterseapower committed
166
    cts = eqs `andCCan` frozen `andCCan` irreds `andCCan` cCanMapToBag dictmap
167
              `andCCan` cCanMapToBag ipmap `andCCan` cCanMapToBag funeqmap
168

169
instance Outputable InertSet where
170
  ppr is = vcat [ vcat (map ppr (Bag.bagToList $ inert_eqs is))
batterseapower's avatar
batterseapower committed
171
                , vcat (map ppr (Bag.bagToList $ inert_irreds is)) 
172
                , vcat (map ppr (Bag.bagToList $ cCanMapToBag (inert_dicts is)))
173 174
                , vcat (map ppr (Bag.bagToList $ cCanMapToBag (inert_ips is))) 
                , vcat (map ppr (Bag.bagToList $ cCanMapToBag (inert_funeqs is)))
175 176
                , text "Frozen errors =" <+> -- Clearly print frozen errors
                    vcat (map ppr (Bag.bagToList $ inert_frozen is))
177 178
                ]
                       
179
emptyInert :: InertSet
180
emptyInert = IS { inert_eqs    = Bag.emptyBag
181
                , inert_frozen = Bag.emptyBag
batterseapower's avatar
batterseapower committed
182
                , inert_irreds = Bag.emptyBag
183 184
                , inert_dicts  = emptyCCanMap
                , inert_ips    = emptyCCanMap
185
                , inert_funeqs = emptyCCanMap }
186 187

updInertSet :: InertSet -> AtomicInert -> InertSet 
188 189
updInertSet is item 
  | isCTyEqCan item                     -- Other equality 
batterseapower's avatar
batterseapower committed
190
  = let eqs' = inert_eqs is `Bag.snocBag` item
191 192 193 194 195
    in is { inert_eqs = eqs' } 
  | Just cls <- isCDictCan_Maybe item   -- Dictionary 
  = is { inert_dicts = updCCanMap (cls,item) (inert_dicts is) } 
  | Just x  <- isCIPCan_Maybe item      -- IP 
  = is { inert_ips   = updCCanMap (x,item) (inert_ips is) }  
batterseapower's avatar
batterseapower committed
196 197
  | isCIrredEvCan item                     -- Presently-irreducible evidence
  = is { inert_irreds = inert_irreds is `Bag.snocBag` item }
198 199
  | Just tc <- isCFunEqCan_Maybe item   -- Function equality 
  = is { inert_funeqs = updCCanMap (tc,item) (inert_funeqs is) }
200
  | otherwise 
201
  = is { inert_frozen = inert_frozen is `Bag.snocBag` item }
202

203
extractUnsolved :: InertSet -> (InertSet, CanonicalCts)
204
-- Postcondition: the returned canonical cts are either Derived, or Wanted.
batterseapower's avatar
batterseapower committed
205
extractUnsolved is@(IS {inert_eqs = eqs, inert_irreds = irreds}) 
206 207 208
  = let is_solved  = is { inert_eqs    = solved_eqs
                        , inert_dicts  = solved_dicts
                        , inert_ips    = solved_ips
batterseapower's avatar
batterseapower committed
209
                        , inert_irreds = solved_irreds
210 211
                        , inert_frozen = emptyCCan
                        , inert_funeqs = solved_funeqs }
212
    in (is_solved, unsolved)
213

dimitris's avatar
dimitris committed
214
  where (unsolved_eqs, solved_eqs)       = Bag.partitionBag (not.isGivenOrSolvedCt) eqs
batterseapower's avatar
batterseapower committed
215
        (unsolved_irreds, solved_irreds) = Bag.partitionBag (not.isGivenOrSolvedCt) irreds
216 217 218
        (unsolved_ips, solved_ips)       = extractUnsolvedCMap (inert_ips is) 
        (unsolved_dicts, solved_dicts)   = extractUnsolvedCMap (inert_dicts is) 
        (unsolved_funeqs, solved_funeqs) = extractUnsolvedCMap (inert_funeqs is) 
219

batterseapower's avatar
batterseapower committed
220
        unsolved = unsolved_eqs `unionBags` inert_frozen is `unionBags` unsolved_irreds `unionBags`
221
                   unsolved_ips `unionBags` unsolved_dicts `unionBags` unsolved_funeqs
222 223 224 225 226 227 228 229 230 231 232 233 234 235
\end{code}

%*********************************************************************
%*                                                                   * 
*                      Main Interaction Solver                       *
*                                                                    *
**********************************************************************

Note [Basic plan] 
~~~~~~~~~~~~~~~~~
1. Canonicalise (unary)
2. Pairwise interaction (binary)
    * Take one from work list 
    * Try all pair-wise interactions with each constraint in inert
236 237 238 239
   
   As an optimisation, we prioritize the equalities both in the 
   worklist and in the inerts. 

240 241
3. Try to solve spontaneously for equalities involving touchables 
4. Top-level interaction (binary wrt top-level)
batterseapower's avatar
batterseapower committed
242
   Superclass decomposition belongs in (1), see note [Adding superclasses]
243 244 245 246 247

\begin{code}
type AtomicInert = CanonicalCt     -- constraint pulled from InertSet
type WorkItem    = CanonicalCt     -- constraint pulled from WorkList

248
------------------------
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275
data StopOrContinue 
  = Stop			-- Work item is consumed
  | ContinueWith WorkItem	-- Not consumed

instance Outputable StopOrContinue where
  ppr Stop             = ptext (sLit "Stop")
  ppr (ContinueWith w) = ptext (sLit "ContinueWith") <+> ppr w

-- Results after interacting a WorkItem as far as possible with an InertSet
data StageResult
  = SR { sr_inerts     :: InertSet
           -- The new InertSet to use (REPLACES the old InertSet)
       , sr_new_work   :: WorkList
           -- Any new work items generated (should be ADDED to the old WorkList)
           -- Invariant: 
           --    sr_stop = Just workitem => workitem is *not* in sr_inerts and
           --                               workitem is inert wrt to sr_inerts
       , sr_stop       :: StopOrContinue
       }

instance Outputable StageResult where
  ppr (SR { sr_inerts = inerts, sr_new_work = work, sr_stop = stop })
    = ptext (sLit "SR") <+> 
      braces (sep [ ptext (sLit "inerts =") <+> ppr inerts <> comma
             	  , ptext (sLit "new work =") <+> ppr work <> comma
             	  , ptext (sLit "stop =") <+> ppr stop])

276 277 278
type SubGoalDepth = Int	  -- Starts at zero; used to limit infinite
     		    	  -- recursion of sub-goals
type SimplifierStage = SubGoalDepth -> WorkItem -> InertSet -> TcS StageResult 
279 280

-- Combine a sequence of simplifier 'stages' to create a pipeline 
281 282 283
runSolverPipeline :: SubGoalDepth
                  -> [(String, SimplifierStage)]
		  -> InertSet -> WorkItem 
284 285
                  -> TcS (InertSet, WorkList)
-- Precondition: non-empty list of stages 
286
runSolverPipeline depth pipeline inerts workItem
287 288 289 290 291
  = do { traceTcS "Start solver pipeline" $ 
            vcat [ ptext (sLit "work item =") <+> ppr workItem
                 , ptext (sLit "inerts    =") <+> ppr inerts]

       ; let itr_in = SR { sr_inerts = inerts
292 293
                         , sr_new_work = emptyWorkList
                         , sr_stop = ContinueWith workItem }
294 295 296 297
       ; itr_out <- run_pipeline pipeline itr_in
       ; let new_inert 
              = case sr_stop itr_out of 
       	          Stop              -> sr_inerts itr_out
298
                  ContinueWith item -> sr_inerts itr_out `updInertSet` item
299 300 301 302 303 304 305 306 307 308 309
       ; return (new_inert, sr_new_work itr_out) }
  where 
    run_pipeline :: [(String, SimplifierStage)]
                 -> StageResult -> TcS StageResult
    run_pipeline [] itr                         = return itr
    run_pipeline _  itr@(SR { sr_stop = Stop }) = return itr

    run_pipeline ((name,stage):stages) 
                 (SR { sr_new_work = accum_work
                     , sr_inerts   = inerts
                     , sr_stop     = ContinueWith work_item })
310
      = do { itr <- stage depth work_item inerts 
311
           ; traceTcS ("Stage result (" ++ name ++ ")") (ppr itr)
312
           ; let itr' = itr { sr_new_work = accum_work `unionWorkList` sr_new_work itr }
313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343
           ; run_pipeline stages itr' }
\end{code}

Example 1:
  Inert:   {c ~ d, F a ~ t, b ~ Int, a ~ ty} (all given)
  Reagent: a ~ [b] (given)

React with (c~d)     ==> IR (ContinueWith (a~[b]))  True    []
React with (F a ~ t) ==> IR (ContinueWith (a~[b]))  False   [F [b] ~ t]
React with (b ~ Int) ==> IR (ContinueWith (a~[Int]) True    []

Example 2:
  Inert:  {c ~w d, F a ~g t, b ~w Int, a ~w ty}
  Reagent: a ~w [b]

React with (c ~w d)   ==> IR (ContinueWith (a~[b]))  True    []
React with (F a ~g t) ==> IR (ContinueWith (a~[b]))  True    []    (can't rewrite given with wanted!)
etc.

Example 3:
  Inert:  {a ~ Int, F Int ~ b} (given)
  Reagent: F a ~ b (wanted)

React with (a ~ Int)   ==> IR (ContinueWith (F Int ~ b)) True []
React with (F Int ~ b) ==> IR Stop True []    -- after substituting we re-canonicalize and get nothing

\begin{code}
-- Main interaction solver: we fully solve the worklist 'in one go', 
-- returning an extended inert set.
--
-- See Note [Touchables and givens].
344 345 346 347 348 349
solveInteractGiven :: InertSet -> GivenLoc -> [EvVar] -> TcS InertSet
solveInteractGiven inert gloc evs
  = do { (_, inert_ret) <- solveInteract inert $ listToBag $
                           map mk_given evs
       ; return inert_ret }
  where
dimitris's avatar
dimitris committed
350
    flav = Given gloc GivenOrig
351 352 353 354 355 356 357 358 359 360 361 362
    mk_given ev = mkEvVarX ev flav

solveInteractWanted :: InertSet -> [WantedEvVar] -> TcS InertSet
solveInteractWanted inert wvs
  = do { (_,inert_ret) <- solveInteract inert $ listToBag $
                          map wantedToFlavored wvs
       ; return inert_ret }

solveInteract :: InertSet -> Bag FlavoredEvVar -> TcS (Bool, InertSet)
-- Post: (True,  inert_set) means we managed to discharge all constraints
--                          without actually doing any interactions!
--       (False, inert_set) means some interactions occurred
363 364
solveInteract inert ws 
  = do { dyn_flags <- getDynFlags
365 366 367 368 369 370 371
       ; sctx <- getTcSContext

       ; traceTcS "solveInteract, before clever canonicalization:" $
         vcat [ text "ws = " <+>  ppr (mapBag (\(EvVarX ev ct)
                                                   -> (ct,evVarPred ev)) ws)
              , text "inert = " <+> ppr inert ]

372 373 374 375
       ; can_ws <- mkCanonicalFEVs ws

       ; (flag, inert_ret)
           <- foldrWorkListM (tryPreSolveAndInteract sctx dyn_flags) (True,inert) can_ws
376 377 378 379 380 381 382 383 384

       ; traceTcS "solveInteract, after clever canonicalization (and interaction):" $
         vcat [ text "No interaction happened = " <+> ppr flag
              , text "inert_ret = " <+> ppr inert_ret ]

       ; return (flag, inert_ret) }

tryPreSolveAndInteract :: SimplContext
                       -> DynFlags
385
                       -> CanonicalCt
386
                       -> (Bool, InertSet)
387 388
                       -> TcS (Bool, InertSet)
-- Returns: True if it was able to discharge this constraint AND all previous ones
389
tryPreSolveAndInteract sctx dyn_flags ct (all_previous_discharged, inert)
batterseapower's avatar
batterseapower committed
390
  = do { let inert_cts = get_inert_cts (predTypePredTree (evVarPred ev_var))
391

392 393 394 395 396
       ; this_one_discharged <- 
           if isCFrozenErr ct then 
               return False
           else
               dischargeFromCCans inert_cts ev_var fl
397 398 399

       ; if this_one_discharged
         then return (all_previous_discharged, inert)
400

401
         else do
402
       { inert_ret <- solveOneWithDepth (ctxtStkDepth dyn_flags,0,[]) ct inert
403 404 405
       ; return (False, inert_ret) } }

  where
406 407 408
    ev_var = cc_id ct
    fl = cc_flavor ct 

batterseapower's avatar
batterseapower committed
409
    get_inert_cts (ClassPred clas _)
410 411
      | simplEqsOnly sctx = emptyCCan
      | otherwise         = fst (getRelevantCts clas (inert_dicts inert))
batterseapower's avatar
batterseapower committed
412
    get_inert_cts (IPPred {})
413 414 415 416 417
      = emptyCCan -- We must not do the same thing for IParams, because (contrary
                  -- to dictionaries), work items /must/ override inert items.
                 -- See Note [Overriding implicit parameters] in TcInteract.
    get_inert_cts (EqPred {})
      = inert_eqs inert `unionBags` cCanMapToBag (inert_funeqs inert)
batterseapower's avatar
batterseapower committed
418 419 420 421
    get_inert_cts (TuplePred ts)
      = andCCans $ map get_inert_cts ts
    get_inert_cts (IrredPred {})
      = inert_irreds inert
422

423
dischargeFromCCans :: CanonicalCts -> EvVar -> CtFlavor -> TcS Bool
424 425 426 427
-- See if this (pre-canonicalised) work-item is identical to a 
-- one already in the inert set. Reasons:
--    a) Avoid creating superclass constraints for millions of incoming (Num a) constraints
--    b) Termination for improve_eqs in TcSimplify.simpl_loop
428
dischargeFromCCans cans ev fl
429 430 431 432 433 434
  = Bag.foldrBag discharge_ct (return False) cans
  where 
    the_pred = evVarPred ev

    discharge_ct :: CanonicalCt -> TcS Bool -> TcS Bool
    discharge_ct ct _rest
435
      | evVarPred (cc_id ct) `eqPred` the_pred
436
      , cc_flavor ct `canSolve` fl
batterseapower's avatar
batterseapower committed
437
      = do { when (isWanted fl) $ setEvBind ev (EvId (cc_id ct))
438 439 440 441 442
           	 -- Deriveds need no evidence
    	         -- For Givens, we already have evidence, and we don't need it twice 
           ; return True }

    discharge_ct _ct rest = rest
443 444 445 446
\end{code}

Note [Avoiding the superclass explosion] 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 
447 448 449 450 451 452 453 454 455 456 457 458
This note now is not as significant as it used to be because we no
longer add the superclasses of Wanted as Derived, except only if they
have equality superclasses or superclasses with functional
dependencies. The fear was that hundreds of identical wanteds would
give rise each to the same superclass or equality Derived's which
would lead to a blo-up in the number of interactions.

Instead, what we do with tryPreSolveAndCanon, is when we encounter a
new constraint, we very quickly see if it can be immediately
discharged by a class constraint in our inert set or the previous
canonicals. If so, we add nothing to the returned canonical
constraints.
459 460

\begin{code}
461 462
solveOne :: WorkItem -> InertSet -> TcS InertSet 
solveOne workItem inerts 
463
  = do { dyn_flags <- getDynFlags
464
       ; solveOneWithDepth (ctxtStkDepth dyn_flags,0,[]) workItem inerts
465 466 467 468
       }

-----------------
solveInteractWithDepth :: (Int, Int, [WorkItem])
469 470
                       -> WorkList -> InertSet -> TcS InertSet
solveInteractWithDepth ctxt@(max_depth,n,stack) ws inert
471 472 473 474 475 476 477 478
  | isEmptyWorkList ws
  = return inert

  | n > max_depth 
  = solverDepthErrorTcS n stack

  | otherwise 
  = do { traceTcS "solveInteractWithDepth" $ 
479
              vcat [ text "Current depth =" <+> ppr n
480 481
                   , text "Max depth =" <+> ppr max_depth
                   , text "ws =" <+> ppr ws ]
482

483 484 485

       ; foldrWorkListM (solveOneWithDepth ctxt) inert ws }
              -- use foldr to preserve the order
486 487 488 489 490

------------------
-- Fully interact the given work item with an inert set, and return a
-- new inert set which has assimilated the new information.
solveOneWithDepth :: (Int, Int, [WorkItem])
491 492
                  -> WorkItem -> InertSet -> TcS InertSet
solveOneWithDepth (max_depth, depth, stack) work inert
493 494
  = do { traceFireTcS depth (text "Solving {" <+> ppr work)
       ; (new_inert, new_work) <- runSolverPipeline depth thePipeline inert work
495 496 497
         
	 -- Recursively solve the new work generated 
         -- from workItem, with a greater depth
498
       ; res_inert <- solveInteractWithDepth (max_depth, depth+1, work:stack) new_work new_inert 
499

500 501
       ; traceFireTcS depth (text "Done }" <+> ppr work) 

502 503 504
       ; return res_inert }

thePipeline :: [(String,SimplifierStage)]
505 506 507 508
thePipeline = [ ("interact with inert eqs", interactWithInertEqsStage)
              , ("interact with inerts",    interactWithInertsStage)
              , ("spontaneous solve",       spontaneousSolveStage)
              , ("top-level reactions",     topReactionsStage) ]
509 510 511 512 513 514 515 516
\end{code}

*********************************************************************************
*                                                                               * 
                       The spontaneous-solve Stage
*                                                                               *
*********************************************************************************

517 518 519 520 521 522
Note [Efficient Orientation] 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

There are two cases where we have to be careful about 
orienting equalities to get better efficiency. 

523
Case 1: In Rewriting Equalities (function rewriteEqLHS) 
524

525 526 527 528 529 530 531 532 533 534
    When rewriting two equalities with the same LHS:
          (a)  (tv ~ xi1) 
          (b)  (tv ~ xi2) 
    We have a choice of producing work (xi1 ~ xi2) (up-to the
    canonicalization invariants) However, to prevent the inert items
    from getting kicked out of the inerts first, we prefer to
    canonicalize (xi1 ~ xi2) if (b) comes from the inert set, or (xi2
    ~ xi1) if (a) comes from the inert set.
    
    This choice is implemented using the WhichComesFromInert flag. 
535

536 537 538 539 540
Case 2: Functional Dependencies 
    Again, we should prefer, if possible, the inert variables on the RHS

Case 3: IP improvement work
    We must always rewrite so that the inert type is on the right. 
541

542 543
\begin{code}
spontaneousSolveStage :: SimplifierStage 
544
spontaneousSolveStage depth workItem inerts 
545 546
  = do { mSolve <- trySpontaneousSolve workItem

547
       ; case mSolve of 
548
           SPCantSolve -> -- No spontaneous solution for him, keep going
549 550
               return $ SR { sr_new_work   = emptyWorkList
                           , sr_inerts     = inerts
551 552
                           , sr_stop       = ContinueWith workItem }

553
           SPSolved workItem'
dimitris's avatar
dimitris committed
554
               | not (isGivenOrSolvedCt workItem) 
555 556 557 558 559
	       	 -- Original was wanted or derived but we have now made him 
                 -- given so we have to interact him with the inerts due to
                 -- its status change. This in turn may produce more work.
		 -- We do this *right now* (rather than just putting workItem'
		 -- back into the work-list) because we've solved 
560 561 562
               -> do { bumpStepCountTcS
	       	     ; traceFireTcS depth (ptext (sLit "Spontaneous (w/d)") <+> ppr workItem)
                     ; (new_inert, new_work) <- runSolverPipeline depth
563 564 565
                             [ ("recursive interact with inert eqs", interactWithInertEqsStage)
                             , ("recursive interact with inerts", interactWithInertsStage)
                             ] inerts workItem'
566 567 568
                     ; return $ SR { sr_new_work = new_work 
                                   , sr_inerts   = new_inert -- will include workItem' 
                                   , sr_stop     = Stop }
569
                     }
570 571 572
               | otherwise 
                   -> -- Original was given; he must then be inert all right, and
                      -- workList' are all givens from flattening
573 574 575 576 577
                      do { bumpStepCountTcS
	       	         ; traceFireTcS depth (ptext (sLit "Spontaneous (g)") <+> ppr workItem)
                         ; return $ SR { sr_new_work = emptyWorkList
                                       , sr_inerts   = inerts `updInertSet` workItem' 
                                       , sr_stop     = Stop } }
578 579 580 581
           SPError -> -- Return with no new work
               return $ SR { sr_new_work = emptyWorkList
                           , sr_inerts   = inerts
                           , sr_stop     = Stop }
582
       }
583

584 585 586 587 588 589
data SPSolveResult = SPCantSolve | SPSolved WorkItem | SPError
-- SPCantSolve means that we can't do the unification because e.g. the variable is untouchable
-- SPSolved workItem' gives us a new *given* to go on 
-- SPError means that it's completely impossible to solve this equality, eg due to a kind error


590
-- @trySpontaneousSolve wi@ solves equalities where one side is a
591
-- touchable unification variable.
592
--     	    See Note [Touchables and givens] 
593
trySpontaneousSolve :: WorkItem -> TcS SPSolveResult
batterseapower's avatar
batterseapower committed
594
trySpontaneousSolve workItem@(CTyEqCan { cc_id = eqv, cc_flavor = gw, cc_tyvar = tv1, cc_rhs = xi })
dimitris's avatar
dimitris committed
595
  | isGivenOrSolved gw
596
  = return SPCantSolve
597 598 599 600
  | Just tv2 <- tcGetTyVar_maybe xi
  = do { tch1 <- isTouchableMetaTyVar tv1
       ; tch2 <- isTouchableMetaTyVar tv2
       ; case (tch1, tch2) of
batterseapower's avatar
batterseapower committed
601 602 603
           (True,  True)  -> trySpontaneousEqTwoWay eqv gw tv1 tv2
           (True,  False) -> trySpontaneousEqOneWay eqv gw tv1 xi
           (False, True)  -> trySpontaneousEqOneWay eqv gw tv2 (mkTyVarTy tv1)
604
	   _ -> return SPCantSolve }
605 606
  | otherwise
  = do { tch1 <- isTouchableMetaTyVar tv1
batterseapower's avatar
batterseapower committed
607
       ; if tch1 then trySpontaneousEqOneWay eqv gw tv1 xi
608 609
                 else do { traceTcS "Untouchable LHS, can't spontaneously solve workitem:" 
                                    (ppr workItem) 
610
                         ; return SPCantSolve }
611
       }
612 613 614 615

  -- No need for 
  --      trySpontaneousSolve (CFunEqCan ...) = ...
  -- See Note [No touchables as FunEq RHS] in TcSMonad
616
trySpontaneousSolve _ = return SPCantSolve
617 618

----------------
batterseapower's avatar
batterseapower committed
619
trySpontaneousEqOneWay :: EqVar -> CtFlavor -> TcTyVar -> Xi -> TcS SPSolveResult
620
-- tv is a MetaTyVar, not untouchable
batterseapower's avatar
batterseapower committed
621
trySpontaneousEqOneWay eqv gw tv xi	
622
  | not (isSigTyVar tv) || isTyVarTy xi 
623 624
  = do { let kxi = typeKind xi -- NB: 'xi' is fully rewritten according to the inerts 
                               -- so we have its more specific kind in our hands
dreixel's avatar
dreixel committed
625 626
       ; is_sub_kind <- kxi `isSubKindTcS` tyVarKind tv
       ; if is_sub_kind then
batterseapower's avatar
batterseapower committed
627
             solveWithIdentity eqv gw tv xi
628 629 630
         else return SPCantSolve
{-
         else if tyVarKind tv `isSubKind` kxi then
631 632 633 634 635 636
             return SPCantSolve -- kinds are compatible but we can't solveWithIdentity this way
                                -- This case covers the  a_touchable :: * ~ b_untouchable :: ?? 
                                -- which has to be deferred or floated out for someone else to solve 
                                -- it in a scope where 'b' is no longer untouchable.
         else do { addErrorTcS KindError gw (mkTyVarTy tv) xi -- See Note [Kind errors]
                 ; return SPError }
637
-}
638
       }
639
  | otherwise -- Still can't solve, sig tyvar and non-variable rhs
640
  = return SPCantSolve
641 642

----------------
batterseapower's avatar
batterseapower committed
643
trySpontaneousEqTwoWay :: EqVar -> CtFlavor -> TcTyVar -> TcTyVar -> TcS SPSolveResult
644
-- Both tyvars are *touchable* MetaTyvars so there is only a chance for kind error here
batterseapower's avatar
batterseapower committed
645
trySpontaneousEqTwoWay eqv gw tv1 tv2
dreixel's avatar
dreixel committed
646 647 648 649 650 651 652
  = do { k1_sub_k2 <- k1 `isSubKindTcS` k2
       ; if k1_sub_k2 && nicer_to_update_tv2
         then solveWithIdentity eqv gw tv2 (mkTyVarTy tv1)
         else do
       { k2_sub_k1 <- k2 `isSubKindTcS` k1
       ; MASSERT( k2_sub_k1 )  -- they were unified in TcCanonical
       ; solveWithIdentity eqv gw tv1 (mkTyVarTy tv2) } }
653 654 655 656
  where
    k1 = tyVarKind tv1
    k2 = tyVarKind tv2
    nicer_to_update_tv2 = isSigTyVar tv1 || isSystemName (Var.varName tv2)
dreixel's avatar
dreixel committed
657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
{-
-- Previous code below (before kind polymorphism and unification):
  -- | k1 `isSubKind` k2
  -- , nicer_to_update_tv2 = solveWithIdentity eqv gw tv2 (mkTyVarTy tv1)
  -- | k2 `isSubKind` k1 
  -- = solveWithIdentity eqv gw tv1 (mkTyVarTy tv2)
  -- | otherwise -- None is a subkind of the other, but they are both touchable! 
  -- = return SPCantSolve
  --   -- do { addErrorTcS KindError gw (mkTyVarTy tv1) (mkTyVarTy tv2)
  --   --   ; return SPError }
  -- where
  --   k1 = tyVarKind tv1
  --   k2 = tyVarKind tv2
  --   nicer_to_update_tv2 = isSigTyVar tv1 || isSystemName (Var.varName tv2)
-}
672 673
\end{code}

674 675 676 677 678 679
Note [Kind errors] 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Consider the wanted problem: 
      alpha ~ (# Int, Int #) 
where alpha :: ?? and (# Int, Int #) :: (#). We can't spontaneously solve this constraint, 
but we should rather reject the program that give rise to it. If 'trySpontaneousEqTwoWay' 
680
simply returns @CantSolve@ then that wanted constraint is going to propagate all the way and 
681
get quantified over in inference mode. That's bad because we do know at this point that the 
682
constraint is insoluble. Instead, we call 'recKindErrorTcS' here, which will fail later on.
683 684

The same applies in canonicalization code in case of kind errors in the givens. 
685

686
However, when we canonicalize givens we only check for compatibility (@compatKind@). 
687
If there were a kind error in the givens, this means some form of inconsistency or dead code.
688

689 690 691 692 693
You may think that when we spontaneously solve wanteds we may have to look through the 
bindings to determine the right kind of the RHS type. E.g one may be worried that xi is 
@alpha@ where alpha :: ? and a previous spontaneous solving has set (alpha := f) with (f :: *).
But we orient our constraints so that spontaneously solved ones can rewrite all other constraint
so this situation can't happen. 
694

695 696
Note [Spontaneous solving and kind compatibility] 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
697 698 699
Note that our canonical constraints insist that *all* equalities (tv ~
xi) or (F xis ~ rhs) require the LHS and the RHS to have *compatible*
the same kinds.  ("compatible" means one is a subKind of the other.)
700

701 702 703 704 705 706 707 708 709 710 711 712 713 714
  - It can't be *equal* kinds, because
     b) wanted constraints don't necessarily have identical kinds
               eg   alpha::? ~ Int
     b) a solved wanted constraint becomes a given

  - SPJ thinks that *given* constraints (tv ~ tau) always have that
    tau has a sub-kind of tv; and when solving wanted constraints
    in trySpontaneousEqTwoWay we re-orient to achieve this.

  - Note that the kind invariant is maintained by rewriting.
    Eg wanted1 rewrites wanted2; if both were compatible kinds before,
       wanted2 will be afterwards.  Similarly givens.

Caveat:
715 716 717 718 719 720 721 722 723
  - Givens from higher-rank, such as: 
          type family T b :: * -> * -> * 
          type instance T Bool = (->) 

          f :: forall a. ((T a ~ (->)) => ...) -> a -> ... 
          flop = f (...) True 
     Whereas we would be able to apply the type instance, we would not be able to 
     use the given (T Bool ~ (->)) in the body of 'flop' 

724 725 726 727 728 729 730

Note [Avoid double unifications] 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
The spontaneous solver has to return a given which mentions the unified unification
variable *on the left* of the equality. Here is what happens if not: 
  Original wanted:  (a ~ alpha),  (alpha ~ Int) 
We spontaneously solve the first wanted, without changing the order! 
731
      given : a ~ alpha      [having unified alpha := a] 
732 733 734
Now the second wanted comes along, but he cannot rewrite the given, so we simply continue.
At the end we spontaneously solve that guy, *reunifying*  [alpha := Int] 

735
We avoid this problem by orienting the resulting given so that the unification
736 737
variable is on the left.  [Note that alternatively we could attempt to
enforce this at canonicalization]
738

739 740 741
See also Note [No touchables as FunEq RHS] in TcSMonad; avoiding
double unifications is the main reason we disallow touchable
unification variables as RHS of type family equations: F xis ~ alpha.
742 743 744

\begin{code}
----------------
745

batterseapower's avatar
batterseapower committed
746
solveWithIdentity :: EqVar -> CtFlavor -> TcTyVar -> Xi -> TcS SPSolveResult
747 748
-- Solve with the identity coercion 
-- Precondition: kind(xi) is a sub-kind of kind(tv)
simonpj@microsoft.com's avatar
simonpj@microsoft.com committed
749 750 751
-- Precondition: CtFlavor is Wanted or Derived
-- See [New Wanted Superclass Work] to see why solveWithIdentity 
--     must work for Derived as well as Wanted
752
-- Returns: workItem where 
753
--        workItem = the new Given constraint
batterseapower's avatar
batterseapower committed
754
solveWithIdentity eqv wd tv xi 
755 756
  = do { traceTcS "Sneaky unification:" $ 
                       vcat [text "Coercion variable:  " <+> ppr wd, 
757 758 759
                             text "Coercion:           " <+> pprEq (mkTyVarTy tv) xi,
                             text "Left  Kind is     : " <+> ppr (typeKind (mkTyVarTy tv)),
                             text "Right Kind is     : " <+> ppr (typeKind xi)
760
                  ]
761

762
       ; setWantedTyBind tv xi
763
       ; let refl_xi = mkReflCo xi
batterseapower's avatar
batterseapower committed
764
       ; eqv_given <- newGivenEqVar (mkTyVarTy tv) xi refl_xi
765

batterseapower's avatar
batterseapower committed
766
       ; when (isWanted wd) (setEqBind eqv refl_xi)
767
           -- We don't want to do this for Derived, that's why we use 'when (isWanted wd)'
batterseapower's avatar
batterseapower committed
768
       ; return $ SPSolved (CTyEqCan { cc_id = eqv_given
dimitris's avatar
dimitris committed
769
                                     , cc_flavor = mkSolvedFlavor wd UnkSkol
770
                                     , cc_tyvar  = tv, cc_rhs = xi }) }
771 772 773 774 775 776 777 778 779
\end{code}


*********************************************************************************
*                                                                               * 
                       The interact-with-inert Stage
*                                                                               *
*********************************************************************************

780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804
Note [The Solver Invariant]
~~~~~~~~~~~~~~~~~~~~~~~~~~~
We always add Givens first.  So you might think that the solver has
the invariant

   If the work-item is Given, 
   then the inert item must Given

But this isn't quite true.  Suppose we have, 
    c1: [W] beta ~ [alpha], c2 : [W] blah, c3 :[W] alpha ~ Int
After processing the first two, we get
     c1: [G] beta ~ [alpha], c2 : [W] blah
Now, c3 does not interact with the the given c1, so when we spontaneously
solve c3, we must re-react it with the inert set.  So we can attempt a 
reaction between inert c2 [W] and work-item c3 [G].

It *is* true that [Solver Invariant]
   If the work-item is Given, 
   AND there is a reaction
   then the inert item must Given
or, equivalently,
   If the work-item is Given, 
   and the inert item is Wanted/Derived
   then there is no reaction

805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
\begin{code}
-- Interaction result of  WorkItem <~> AtomicInert
data InteractResult
   = IR { ir_stop         :: StopOrContinue
            -- Stop
            --   => Reagent (work item) consumed.
            -- ContinueWith new_reagent
            --   => Reagent transformed but keep gathering interactions. 
            --      The transformed item remains inert with respect 
            --      to any previously encountered inerts.

        , ir_inert_action :: InertAction
            -- Whether the inert item should remain in the InertSet.

        , ir_new_work     :: WorkList
            -- new work items to add to the WorkList
821 822

        , ir_fire :: Maybe String    -- Tells whether a rule fired, and if so what
823 824 825
        }

-- What to do with the inert reactant.
826
data InertAction = KeepInert | DropInert 
827

828 829 830 831
mkIRContinue :: String -> WorkItem -> InertAction -> WorkList -> TcS InteractResult
mkIRContinue rule wi keep newWork 
  = return $ IR { ir_stop = ContinueWith wi, ir_inert_action = keep
                , ir_new_work = newWork, ir_fire = Just rule }
832

833 834
mkIRStopK :: String -> WorkList -> TcS InteractResult
mkIRStopK rule newWork
835 836
  = return $ IR { ir_stop = Stop, ir_inert_action = KeepInert
                , ir_new_work = newWork, ir_fire = Just rule }
837

838 839 840 841 842
mkIRStopD :: String -> WorkList -> TcS InteractResult
mkIRStopD rule newWork
  = return $ IR { ir_stop = Stop, ir_inert_action = DropInert
                , ir_new_work = newWork, ir_fire = Just rule }

843
noInteraction :: Monad m => WorkItem -> m InteractResult
844 845 846
noInteraction wi
  = return $ IR { ir_stop = ContinueWith wi, ir_inert_action = KeepInert
                , ir_new_work = emptyWorkList, ir_fire = Nothing }
847

dimitris@microsoft.com's avatar
dimitris@microsoft.com committed
848
data WhichComesFromInert = LeftComesFromInert | RightComesFromInert 
849
     -- See Note [Efficient Orientation] 
850

851

852
---------------------------------------------------
853 854 855
-- Interact a single WorkItem with the equalities of an inert set as
-- far as possible, i.e. until we get a Stop result from an individual
-- reaction (i.e. when the WorkItem is consumed), or until we've
856 857 858
-- interact the WorkItem with the entire equalities of the InertSet

interactWithInertEqsStage :: SimplifierStage 
859
interactWithInertEqsStage depth workItem inert
860
  = Bag.foldrBagM (interactNext depth) initITR (inert_eqs inert)
861
                        -- use foldr to preserve the order          
862 863 864 865
  where
    initITR = SR { sr_inerts   = inert { inert_eqs = emptyCCan }
                 , sr_new_work = emptyWorkList
                 , sr_stop     = ContinueWith workItem }
866

867 868 869 870
---------------------------------------------------
-- Interact a single WorkItem with *non-equality* constraints in the inert set. 
-- Precondition: equality interactions must have already happened, hence we have 
-- to pick up some information from the incoming inert, before folding over the 
871 872
-- "Other" constraints it contains!

873
interactWithInertsStage :: SimplifierStage
874
interactWithInertsStage depth workItem inert
875 876 877 878
  = let (relevant, inert_residual) = getISRelevant workItem inert 
        initITR = SR { sr_inerts   = inert_residual
                     , sr_new_work = emptyWorkList
                     , sr_stop     = ContinueWith workItem } 
879 880
    in Bag.foldrBagM (interactNext depth) initITR relevant 
                        -- use foldr to preserve the order
881
  where 
882
    getISRelevant :: CanonicalCt -> InertSet -> (CanonicalCts, InertSet) 
883 884 885 886 887 888
    getISRelevant (CFrozenErr {}) is = (emptyCCan, is)
                  -- Nothing s relevant; we have alread interacted
                  -- it with the equalities in the inert set

    getISRelevant (CDictCan { cc_class = cls } ) is
      = let (relevant, residual_map) = getRelevantCts cls (inert_dicts is)
889 890 891 892 893 894 895
        in (relevant, is { inert_dicts = residual_map }) 
    getISRelevant (CFunEqCan { cc_fun = tc } ) is 
      = let (relevant, residual_map) = getRelevantCts tc (inert_funeqs is) 
        in (relevant, is { inert_funeqs = residual_map })
    getISRelevant (CIPCan { cc_ip_nm = nm }) is 
      = let (relevant, residual_map) = getRelevantCts nm (inert_ips is)
        in (relevant, is { inert_ips = residual_map }) 
896 897
    getISRelevant (CIrredEvCan {}) is
      = (inert_irreds is, is { inert_irreds = emptyCCan })
898 899 900 901 902
    -- An equality, finally, may kick everything except equalities out 
    -- because we have already interacted the equalities in interactWithInertEqsStage
    getISRelevant _eq_ct is  -- Equality, everything is relevant for this one 
                             -- TODO: if we were caching variables, we'd know that only 
                             --       some are relevant. Experiment with this for now. 
batterseapower's avatar
batterseapower committed
903 904 905 906
      = let cts = cCanMapToBag (inert_ips is) `unionBags`
                    cCanMapToBag (inert_dicts is) `unionBags`
                    cCanMapToBag (inert_funeqs is) `unionBags`
                    inert_irreds is
907 908
        in (cts, is { inert_dicts  = emptyCCanMap
                    , inert_ips    = emptyCCanMap
batterseapower's avatar
batterseapower committed
909 910
                    , inert_funeqs = emptyCCanMap
                    , inert_irreds = emptyBag })
911

912 913
interactNext :: SubGoalDepth -> AtomicInert -> StageResult -> TcS StageResult 
interactNext depth inert it
914 915 916 917 918 919 920 921 922 923 924
  | ContinueWith work_item <- sr_stop it
  = do { let inerts = sr_inerts it 

       ; IR { ir_new_work = new_work, ir_inert_action = inert_action
            , ir_fire = fire_info, ir_stop = stop } 
            <- interactWithInert inert work_item

       ; let mk_msg rule 
      	       = text rule <+> keep_doc
      	         <+> vcat [ ptext (sLit "Inert =") <+> ppr inert
      	                  , ptext (sLit "Work =")  <+> ppr work_item
925
      	                  , ppUnless (isEmptyWorkList new_work) $
926 927 928 929 930 931 932 933 934 935 936 937 938
                            ptext (sLit "New =") <+> ppr new_work ]
             keep_doc = case inert_action of
                 	  KeepInert -> ptext (sLit "[keep]")
                 	  DropInert -> ptext (sLit "[drop]")
       ; case fire_info of
           Just rule -> do { bumpStepCountTcS
                           ; traceFireTcS depth (mk_msg rule) }
           Nothing  -> return ()

       -- New inerts depend on whether we KeepInert or not 
       ; let inerts_new = case inert_action of
                            KeepInert -> inerts `updInertSet` inert
                            DropInert -> inerts
939 940

       ; return $ SR { sr_inerts   = inerts_new
941
                     , sr_new_work = sr_new_work it `unionWorkList` new_work
942
                     , sr_stop     = stop } }
943 944
  | otherwise 
  = return $ it { sr_inerts = (sr_inerts it) `updInertSet` inert }
945 946

-- Do a single interaction of two constraints.
947
interactWithInert :: AtomicInert -> WorkItem -> TcS InteractResult
948 949 950
interactWithInert inert workItem 
  = do { ctxt <- getTcSContext
       ; let is_allowed  = allowedInteraction (simplEqsOnly ctxt) inert workItem 
951

952 953
       ; if is_allowed then 
              doInteractWithInert inert workItem 
954
          else 
955 956
              noInteraction workItem 
       }
957 958 959

allowedInteraction :: Bool -> AtomicInert -> WorkItem -> Bool 
-- Allowed interactions 
batterseapower's avatar
batterseapower committed
960 961 962
allowedInteraction eqs_only (CDictCan {})    (CDictCan {})    = not eqs_only
allowedInteraction eqs_only (CIPCan {})      (CIPCan {})      = not eqs_only
allowedInteraction eqs_only (CIrredEvCan {}) (CIrredEvCan {}) = not eqs_only
963 964 965
allowedInteraction _ _ _ = True 

--------------------------------------------
966
doInteractWithInert :: CanonicalCt -> CanonicalCt -> TcS InteractResult
967 968
-- Identical class constraints.

969
doInteractWithInert
970 971
  inertItem@(CDictCan { cc_id = d1, cc_flavor = fl1, cc_class = cls1, cc_tyargs = tys1 }) 
   workItem@(CDictCan { cc_id = d2, cc_flavor = fl2, cc_class = cls2, cc_tyargs = tys2 })
972

973
  | cls1 == cls2  
batterseapower's avatar
batterseapower committed
974 975
  = do { let pty1 = mkClassPred cls1 tys1
             pty2 = mkClassPred cls2 tys2
976
             inert_pred_loc     = (pty1, pprFlavorArising fl1)
977
             work_item_pred_loc = (pty2, pprFlavorArising fl2)
978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019

       ; any_fundeps 
           <- if isGivenOrSolved fl1 && isGivenOrSolved fl2 then return Nothing
              -- NB: We don't create fds for given (and even solved), have not seen a useful
              -- situation for these and even if we did we'd have to be very careful to only
              -- create Derived's and not Wanteds. 

              else let fd_eqns = improveFromAnother inert_pred_loc work_item_pred_loc
                       wloc    = get_workitem_wloc fl2 
                   in rewriteWithFunDeps fd_eqns tys2 wloc
                      -- See Note [Efficient Orientation], [When improvement happens]

       ; case any_fundeps of
           -- No Functional Dependencies 
           Nothing             
               | eqTypes tys1 tys2 -> solveOneFromTheOther "Cls/Cls" (EvId d1,fl1) workItem
               | otherwise         -> noInteraction workItem

           -- Actual Functional Dependencies
           Just (rewritten_tys2,cos2,fd_work) 
               | not (eqTypes tys1 rewritten_tys2) 
               -- Standard thing: create derived fds and keep on going. Importantly we don't
               -- throw workitem back in the worklist because this can cause loops. See #5236.
               -> do { fd_cans <- mkCanonicalFDAsDerived fd_work
                     ; mkIRContinue "Cls/Cls fundep (not solved)" workItem KeepInert fd_cans }

               -- This WHOLE otherwise branch is an optimization where the fd made the things match
               | otherwise  
               , let dict_co = mkTyConAppCo (classTyCon cls1) cos2
               -> case fl2 of
                    Given {} 
                        -> pprPanic "Unexpected given!" (ppr inertItem $$ ppr workItem)
                           -- The only way to have created a fundep is if the inert was
                           -- wanted or derived, in which case the workitem can't be given!
                    Derived {}
                        -- The types were made to exactly match so we don't need 
                        -- the workitem any longer.
                        -> do { fd_cans <- mkCanonicalFDAsDerived fd_work
                               -- No rewriting really, so let's create deriveds fds
                              ; mkIRStopK "Cls/Cls fundep (solved)" fd_cans }
		    Wanted  {} 
		        | isDerived fl1 
batterseapower's avatar
batterseapower committed
1020
                            -> do { setEvBind d2 (EvCast d1 dict_co)
1021
			          ; let inert_w = inertItem { cc_flavor = fl2 }
1022 1023 1024
			   -- A bit naughty: we take the inert Derived, 
			   -- turn it into a Wanted, use it to solve the work-item
			   -- and put it back into the work-list
1025 1026 1027 1028 1029 1030 1031 1032 1033
			   -- Maybe rather than starting again, we could keep going 
                           -- with the rewritten workitem, having dropped the inert, but its
                           -- safe to restart.
                          
                           -- Also: we have rewriting so lets create wanted fds
                                  ; fd_cans <- mkCanonicalFDAsWanted fd_work
                                  ; mkIRStopD "Cls/Cls fundep (solved)" $ 
                                    workListFromNonEq inert_w `unionWorkList` fd_cans }
		        | otherwise
batterseapower's avatar
batterseapower committed
1034
                        -> do { setEvBind d2 (EvCast d1 dict_co)
1035 1036 1037 1038 1039 1040 1041 1042
                          -- Rewriting is happening, so we have to create wanted fds
                              ; fd_cans <- mkCanonicalFDAsWanted fd_work
                              ; mkIRStopK "Cls/Cls fundep (solved)" fd_cans }
       }
  where get_workitem_wloc (Wanted wl)  = wl 
        get_workitem_wloc (Derived wl) = wl 
        get_workitem_wloc (Given {})   = panic "Unexpected given!"

1043 1044 1045

-- Class constraint and given equality: use the equality to rewrite
-- the class constraint. 
batterseapower's avatar
batterseapower committed
1046
doInteractWithInert (CTyEqCan { cc_id = eqv, cc_flavor = ifl, cc_tyvar = tv, cc_rhs = xi }) 
1047 1048 1049
                    (CDictCan { cc_id = dv, cc_flavor = wfl, cc_class = cl, cc_tyargs = xis }) 
  | ifl `canRewrite` wfl 
  , tv `elemVarSet` tyVarsOfTypes xis
batterseapower's avatar
batterseapower committed
1050
  = do { rewritten_dict <- rewriteDict (eqv,tv,xi) (dv,wfl,cl,xis)
1051 1052
            -- Continue with rewritten Dictionary because we can only be in the 
            -- interactWithEqsStage, so the dictionary is inert. 
1053
       ; mkIRContinue "Eq/Cls" rewritten_dict KeepInert emptyWorkList }
1054
    
1055
doInteractWithInert (CDictCan { cc_id = dv, cc_flavor = ifl, cc_class = cl, cc_tyargs = xis }) 
batterseapower's avatar
batterseapower committed
1056
           workItem@(CTyEqCan { cc_id = eqv, cc_flavor = wfl, cc_tyvar = tv, cc_rhs = xi })
1057 1058
  | wfl `canRewrite` ifl
  , tv `elemVarSet` tyVarsOfTypes xis
batterseapower's avatar
batterseapower committed
1059
  = do { rewritten_dict <- rewriteDict (eqv,tv,xi) (dv,ifl,cl,xis)
1060
       ; mkIRContinue "Cls/Eq" workItem DropInert (workListFromNonEq rewritten_dict) }
1061

1062 1063
-- Irreducible evidence and given equality: use the equality to rewrite
-- the irreducible evidence.
batterseapower's avatar
batterseapower committed
1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077
doInteractWithInert (CTyEqCan { cc_id = eqv, cc_flavor = ifl, cc_tyvar = tv, cc_rhs = xi }) 
                    (CIrredEvCan { cc_id = id, cc_flavor = wfl, cc_ty = ty })
  | ifl `canRewrite` wfl
  , tv `elemVarSet` tyVarsOfType ty 
  = do { rewritten_irred <- rewriteIrred (eqv,tv,xi) (id,wfl,ty) 
       ; mkIRStopK "Eq/Irred" rewritten_irred } 

doInteractWithInert (CIrredEvCan { cc_id = id, cc_flavor = ifl, cc_ty = ty }) 
           workItem@(CTyEqCan { cc_id = eqv, cc_flavor = wfl, cc_tyvar = tv, cc_rhs = xi })
  | wfl `canRewrite` ifl
  , tv `elemVarSet` tyVarsOfType ty
  = do { rewritten_irred <- rewriteIrred (eqv,tv,xi) (id,ifl,ty) 
       ; mkIRContinue "Irred/Eq" workItem DropInert rewritten_irred }

1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
-- Two pieces of irreducible evidence: if their types are *exactly identical* we can
-- rewrite them. We can never improve using this: if we want ty1 :: Constraint and have
-- ty2 :: Constraint it clearly does not mean that (ty1 ~ ty2)
doInteractWithInert (CIrredEvCan { cc_id = id1, cc_flavor = ifl, cc_ty = ty1 })
           workItem@(CIrredEvCan { cc_ty = ty2 })
  | ty1 `eqType` ty2
  = solveOneFromTheOther "Irred/Irred" (EvId id1,ifl) workItem

-- Implicit param and given equality: use the equality to rewrite
-- the implicit param.
batterseapower's avatar
batterseapower committed
1088
doInteractWithInert (CTyEqCan { cc_id = eqv, cc_flavor = ifl, cc_tyvar = tv, cc_rhs = xi }) 
1089 1090 1091
                    (CIPCan { cc_id = ipid, cc_flavor = wfl, cc_ip_nm = nm, cc_ip_ty = ty }) 
  | ifl `canRewrite` wfl
  , tv `elemVarSet` tyVarsOfType ty 
batterseapower's avatar
batterseapower committed
1092
  = do { rewritten_ip <- rewriteIP (eqv,tv,xi) (ipid,wfl,nm,ty) 
1093
       ; mkIRContinue "Eq/IP" rewritten_ip KeepInert emptyWorkList } 
1094

1095
doInteractWithInert (CIPCan { cc_id = ipid, cc_flavor = ifl, cc_ip_nm = nm, cc_ip_ty = ty }) 
batterseapower's avatar
batterseapower committed
1096
           workItem@(CTyEqCan { cc_id = eqv, cc_flavor = wfl, cc_tyvar = tv, cc_rhs = xi })
1097 1098
  | wfl `canRewrite` ifl
  , tv `elemVarSet` tyVarsOfType ty
batterseapower's avatar
batterseapower committed
1099
  = do { rewritten_ip <- rewriteIP (eqv,tv,xi) (ipid,ifl,nm,ty) 
1100
       ; mkIRContinue "IP/Eq" workItem DropInert (workListFromNonEq rewritten_ip) }
1101 1102 1103 1104 1105 1106

-- Two implicit parameter constraints.  If the names are the same,
-- but their types are not, we generate a wanted type equality 
-- that equates the type (this is "improvement").  
-- However, we don't actually need the coercion evidence,
-- so we just generate a fresh coercion variable that isn't used anywhere.
1107
doInteractWithInert (CIPCan { cc_id = id1, cc_flavor = ifl, cc_ip_nm = nm1, cc_ip_ty = ty1 }) 
1108
           workItem@(CIPCan { cc_flavor = wfl, cc_ip_nm = nm2, cc_ip_ty = ty2 })
dimitris's avatar
dimitris committed
1109
  | nm1 == nm2 && isGivenOrSolved wfl && isGivenOrSolved ifl
1110 1111 1112
  = 	-- See Note [Overriding implicit parameters]
        -- Dump the inert item, override totally with the new one
	-- Do not require type equality
1113 1114 1115
	-- For example, given let ?x::Int = 3 in let ?x::Bool = True in ...
	--              we must *override* the outer one with the inner one
    mkIRContinue "IP/IP override" workItem DropInert emptyWorkList
1116

Simon Peyton Jones's avatar