Reorganise TcSimplify (again); FIX Trac #1919
[ghc-hetmet.git] / compiler / typecheck / TcSimplify.lhs
1 %
2 % (c) The University of Glasgow 2006
3 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
4 %
5
6 TcSimplify
7
8 \begin{code}
9 {-# OPTIONS -w #-}
10 -- The above warning supression flag is a temporary kludge.
11 -- While working on this module you are encouraged to remove it and fix
12 -- any warnings in the module. See
13 --     http://hackage.haskell.org/trac/ghc/wiki/Commentary/CodingStyle#Warnings
14 -- for details
15
16 module TcSimplify (
17         tcSimplifyInfer, tcSimplifyInferCheck,
18         tcSimplifyCheck, tcSimplifyRestricted,
19         tcSimplifyRuleLhs, tcSimplifyIPs, 
20         tcSimplifySuperClasses,
21         tcSimplifyTop, tcSimplifyInteractive,
22         tcSimplifyBracket, tcSimplifyCheckPat,
23
24         tcSimplifyDeriv, tcSimplifyDefault,
25         bindInstsOfLocalFuns, 
26
27         misMatchMsg
28     ) where
29
30 #include "HsVersions.h"
31
32 import {-# SOURCE #-} TcUnify( unifyType )
33 import HsSyn
34
35 import TcRnMonad
36 import Inst
37 import TcEnv
38 import InstEnv
39 import TcGadt
40 import TcType
41 import TcMType
42 import TcIface
43 import TcTyFuns
44 import TypeRep
45 import Var
46 import Name
47 import NameSet
48 import Class
49 import FunDeps
50 import PrelInfo
51 import PrelNames
52 import Type
53 import TysWiredIn
54 import ErrUtils
55 import BasicTypes
56 import VarSet
57 import VarEnv
58 import Module
59 import FiniteMap
60 import Bag
61 import Outputable
62 import Maybes
63 import ListSetOps
64 import Util
65 import UniqSet
66 import SrcLoc
67 import DynFlags
68
69 import Data.List
70 \end{code}
71
72
73 %************************************************************************
74 %*                                                                      *
75 \subsection{NOTES}
76 %*                                                                      *
77 %************************************************************************
78
79         --------------------------------------
80         Notes on functional dependencies (a bug)
81         --------------------------------------
82
83 Consider this:
84
85         class C a b | a -> b
86         class D a b | a -> b
87
88         instance D a b => C a b -- Undecidable 
89                                 -- (Not sure if it's crucial to this eg)
90         f :: C a b => a -> Bool
91         f _ = True
92         
93         g :: C a b => a -> Bool
94         g = f
95
96 Here f typechecks, but g does not!!  Reason: before doing improvement,
97 we reduce the (C a b1) constraint from the call of f to (D a b1).
98
99 Here is a more complicated example:
100
101 | > class Foo a b | a->b
102 | >
103 | > class Bar a b | a->b
104 | >
105 | > data Obj = Obj
106 | >
107 | > instance Bar Obj Obj
108 | >
109 | > instance (Bar a b) => Foo a b
110 | >
111 | > foo:: (Foo a b) => a -> String
112 | > foo _ = "works"
113 | >
114 | > runFoo:: (forall a b. (Foo a b) => a -> w) -> w
115 | > runFoo f = f Obj
116
117 | *Test> runFoo foo
118
119 | <interactive>:1:
120 |     Could not deduce (Bar a b) from the context (Foo a b)
121 |       arising from use of `foo' at <interactive>:1
122 |     Probable fix:
123 |         Add (Bar a b) to the expected type of an expression
124 |     In the first argument of `runFoo', namely `foo'
125 |     In the definition of `it': it = runFoo foo
126
127 | Why all of the sudden does GHC need the constraint Bar a b? The
128 | function foo didn't ask for that... 
129
130 The trouble is that to type (runFoo foo), GHC has to solve the problem:
131
132         Given constraint        Foo a b
133         Solve constraint        Foo a b'
134
135 Notice that b and b' aren't the same.  To solve this, just do
136 improvement and then they are the same.  But GHC currently does
137         simplify constraints
138         apply improvement
139         and loop
140
141 That is usually fine, but it isn't here, because it sees that Foo a b is
142 not the same as Foo a b', and so instead applies the instance decl for
143 instance Bar a b => Foo a b.  And that's where the Bar constraint comes
144 from.
145
146 The Right Thing is to improve whenever the constraint set changes at
147 all.  Not hard in principle, but it'll take a bit of fiddling to do.  
148
149 Note [Choosing which variables to quantify]
150 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
151 Suppose we are about to do a generalisation step.  We have in our hand
152
153         G       the environment
154         T       the type of the RHS
155         C       the constraints from that RHS
156
157 The game is to figure out
158
159         Q       the set of type variables over which to quantify
160         Ct      the constraints we will *not* quantify over
161         Cq      the constraints we will quantify over
162
163 So we're going to infer the type
164
165         forall Q. Cq => T
166
167 and float the constraints Ct further outwards.
168
169 Here are the things that *must* be true:
170
171  (A)    Q intersect fv(G) = EMPTY                       limits how big Q can be
172  (B)    Q superset fv(Cq union T) \ oclose(fv(G),C)     limits how small Q can be
173
174  (A) says we can't quantify over a variable that's free in the environment. 
175  (B) says we must quantify over all the truly free variables in T, else 
176      we won't get a sufficiently general type.  
177
178 We do not *need* to quantify over any variable that is fixed by the
179 free vars of the environment G.
180
181         BETWEEN THESE TWO BOUNDS, ANY Q WILL DO!
182
183 Example:        class H x y | x->y where ...
184
185         fv(G) = {a}     C = {H a b, H c d}
186                         T = c -> b
187
188         (A)  Q intersect {a} is empty
189         (B)  Q superset {a,b,c,d} \ oclose({a}, C) = {a,b,c,d} \ {a,b} = {c,d}
190
191         So Q can be {c,d}, {b,c,d}
192
193 In particular, it's perfectly OK to quantify over more type variables
194 than strictly necessary; there is no need to quantify over 'b', since
195 it is determined by 'a' which is free in the envt, but it's perfectly
196 OK to do so.  However we must not quantify over 'a' itself.
197
198 Other things being equal, however, we'd like to quantify over as few
199 variables as possible: smaller types, fewer type applications, more
200 constraints can get into Ct instead of Cq.  Here's a good way to
201 choose Q:
202
203         Q = grow( fv(T), C ) \ oclose( fv(G), C )
204
205 That is, quantify over all variable that that MIGHT be fixed by the
206 call site (which influences T), but which aren't DEFINITELY fixed by
207 G.  This choice definitely quantifies over enough type variables,
208 albeit perhaps too many.
209
210 Why grow( fv(T), C ) rather than fv(T)?  Consider
211
212         class H x y | x->y where ...
213
214         T = c->c
215         C = (H c d)
216
217   If we used fv(T) = {c} we'd get the type
218
219         forall c. H c d => c -> b
220
221   And then if the fn was called at several different c's, each of
222   which fixed d differently, we'd get a unification error, because
223   d isn't quantified.  Solution: quantify d.  So we must quantify
224   everything that might be influenced by c.
225
226 Why not oclose( fv(T), C )?  Because we might not be able to see
227 all the functional dependencies yet:
228
229         class H x y | x->y where ...
230         instance H x y => Eq (T x y) where ...
231
232         T = c->c
233         C = (Eq (T c d))
234
235 Now oclose(fv(T),C) = {c}, because the functional dependency isn't
236 apparent yet, and that's wrong.  We must really quantify over d too.
237
238 There really isn't any point in quantifying over any more than
239 grow( fv(T), C ), because the call sites can't possibly influence
240 any other type variables.
241
242
243
244 -------------------------------------
245         Note [Ambiguity]
246 -------------------------------------
247
248 It's very hard to be certain when a type is ambiguous.  Consider
249
250         class K x
251         class H x y | x -> y
252         instance H x y => K (x,y)
253
254 Is this type ambiguous?
255         forall a b. (K (a,b), Eq b) => a -> a
256
257 Looks like it!  But if we simplify (K (a,b)) we get (H a b) and
258 now we see that a fixes b.  So we can't tell about ambiguity for sure
259 without doing a full simplification.  And even that isn't possible if
260 the context has some free vars that may get unified.  Urgle!
261
262 Here's another example: is this ambiguous?
263         forall a b. Eq (T b) => a -> a
264 Not if there's an insance decl (with no context)
265         instance Eq (T b) where ...
266
267 You may say of this example that we should use the instance decl right
268 away, but you can't always do that:
269
270         class J a b where ...
271         instance J Int b where ...
272
273         f :: forall a b. J a b => a -> a
274
275 (Notice: no functional dependency in J's class decl.)
276 Here f's type is perfectly fine, provided f is only called at Int.
277 It's premature to complain when meeting f's signature, or even
278 when inferring a type for f.
279
280
281
282 However, we don't *need* to report ambiguity right away.  It'll always
283 show up at the call site.... and eventually at main, which needs special
284 treatment.  Nevertheless, reporting ambiguity promptly is an excellent thing.
285
286 So here's the plan.  We WARN about probable ambiguity if
287
288         fv(Cq) is not a subset of  oclose(fv(T) union fv(G), C)
289
290 (all tested before quantification).
291 That is, all the type variables in Cq must be fixed by the the variables
292 in the environment, or by the variables in the type.
293
294 Notice that we union before calling oclose.  Here's an example:
295
296         class J a b c | a b -> c
297         fv(G) = {a}
298
299 Is this ambiguous?
300         forall b c. (J a b c) => b -> b
301
302 Only if we union {a} from G with {b} from T before using oclose,
303 do we see that c is fixed.
304
305 It's a bit vague exactly which C we should use for this oclose call.  If we
306 don't fix enough variables we might complain when we shouldn't (see
307 the above nasty example).  Nothing will be perfect.  That's why we can
308 only issue a warning.
309
310
311 Can we ever be *certain* about ambiguity?  Yes: if there's a constraint
312
313         c in C such that fv(c) intersect (fv(G) union fv(T)) = EMPTY
314
315 then c is a "bubble"; there's no way it can ever improve, and it's
316 certainly ambiguous.  UNLESS it is a constant (sigh).  And what about
317 the nasty example?
318
319         class K x
320         class H x y | x -> y
321         instance H x y => K (x,y)
322
323 Is this type ambiguous?
324         forall a b. (K (a,b), Eq b) => a -> a
325
326 Urk.  The (Eq b) looks "definitely ambiguous" but it isn't.  What we are after
327 is a "bubble" that's a set of constraints
328
329         Cq = Ca union Cq'  st  fv(Ca) intersect (fv(Cq') union fv(T) union fv(G)) = EMPTY
330
331 Hence another idea.  To decide Q start with fv(T) and grow it
332 by transitive closure in Cq (no functional dependencies involved).
333 Now partition Cq using Q, leaving the definitely-ambiguous and probably-ok.
334 The definitely-ambiguous can then float out, and get smashed at top level
335 (which squashes out the constants, like Eq (T a) above)
336
337
338         --------------------------------------
339                 Notes on principal types
340         --------------------------------------
341
342     class C a where
343       op :: a -> a
344
345     f x = let g y = op (y::Int) in True
346
347 Here the principal type of f is (forall a. a->a)
348 but we'll produce the non-principal type
349     f :: forall a. C Int => a -> a
350
351
352         --------------------------------------
353         The need for forall's in constraints
354         --------------------------------------
355
356 [Exchange on Haskell Cafe 5/6 Dec 2000]
357
358   class C t where op :: t -> Bool
359   instance C [t] where op x = True
360
361   p y = (let f :: c -> Bool; f x = op (y >> return x) in f, y ++ [])
362   q y = (y ++ [], let f :: c -> Bool; f x = op (y >> return x) in f)
363
364 The definitions of p and q differ only in the order of the components in
365 the pair on their right-hand sides.  And yet:
366
367   ghc and "Typing Haskell in Haskell" reject p, but accept q;
368   Hugs rejects q, but accepts p;
369   hbc rejects both p and q;
370   nhc98 ... (Malcolm, can you fill in the blank for us!).
371
372 The type signature for f forces context reduction to take place, and
373 the results of this depend on whether or not the type of y is known,
374 which in turn depends on which component of the pair the type checker
375 analyzes first.  
376
377 Solution: if y::m a, float out the constraints
378         Monad m, forall c. C (m c)
379 When m is later unified with [], we can solve both constraints.
380
381
382         --------------------------------------
383                 Notes on implicit parameters
384         --------------------------------------
385
386 Note [Inheriting implicit parameters]
387 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
388 Consider this:
389
390         f x = (x::Int) + ?y
391
392 where f is *not* a top-level binding.
393 From the RHS of f we'll get the constraint (?y::Int).
394 There are two types we might infer for f:
395
396         f :: Int -> Int
397
398 (so we get ?y from the context of f's definition), or
399
400         f :: (?y::Int) => Int -> Int
401
402 At first you might think the first was better, becuase then
403 ?y behaves like a free variable of the definition, rather than
404 having to be passed at each call site.  But of course, the WHOLE
405 IDEA is that ?y should be passed at each call site (that's what
406 dynamic binding means) so we'd better infer the second.
407
408 BOTTOM LINE: when *inferring types* you *must* quantify 
409 over implicit parameters. See the predicate isFreeWhenInferring.
410
411
412 Note [Implicit parameters and ambiguity] 
413 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
414 Only a *class* predicate can give rise to ambiguity
415 An *implicit parameter* cannot.  For example:
416         foo :: (?x :: [a]) => Int
417         foo = length ?x
418 is fine.  The call site will suppply a particular 'x'
419
420 Furthermore, the type variables fixed by an implicit parameter
421 propagate to the others.  E.g.
422         foo :: (Show a, ?x::[a]) => Int
423         foo = show (?x++?x)
424 The type of foo looks ambiguous.  But it isn't, because at a call site
425 we might have
426         let ?x = 5::Int in foo
427 and all is well.  In effect, implicit parameters are, well, parameters,
428 so we can take their type variables into account as part of the
429 "tau-tvs" stuff.  This is done in the function 'FunDeps.grow'.
430
431
432 Question 2: type signatures
433 ~~~~~~~~~~~~~~~~~~~~~~~~~~~
434 BUT WATCH OUT: When you supply a type signature, we can't force you
435 to quantify over implicit parameters.  For example:
436
437         (?x + 1) :: Int
438
439 This is perfectly reasonable.  We do not want to insist on
440
441         (?x + 1) :: (?x::Int => Int)
442
443 That would be silly.  Here, the definition site *is* the occurrence site,
444 so the above strictures don't apply.  Hence the difference between
445 tcSimplifyCheck (which *does* allow implicit paramters to be inherited)
446 and tcSimplifyCheckBind (which does not).
447
448 What about when you supply a type signature for a binding?
449 Is it legal to give the following explicit, user type 
450 signature to f, thus:
451
452         f :: Int -> Int
453         f x = (x::Int) + ?y
454
455 At first sight this seems reasonable, but it has the nasty property
456 that adding a type signature changes the dynamic semantics.
457 Consider this:
458
459         (let f x = (x::Int) + ?y
460          in (f 3, f 3 with ?y=5))  with ?y = 6
461
462                 returns (3+6, 3+5)
463 vs
464         (let f :: Int -> Int
465              f x = x + ?y
466          in (f 3, f 3 with ?y=5))  with ?y = 6
467
468                 returns (3+6, 3+6)
469
470 Indeed, simply inlining f (at the Haskell source level) would change the
471 dynamic semantics.
472
473 Nevertheless, as Launchbury says (email Oct 01) we can't really give the
474 semantics for a Haskell program without knowing its typing, so if you 
475 change the typing you may change the semantics.
476
477 To make things consistent in all cases where we are *checking* against
478 a supplied signature (as opposed to inferring a type), we adopt the
479 rule: 
480
481         a signature does not need to quantify over implicit params.
482
483 [This represents a (rather marginal) change of policy since GHC 5.02,
484 which *required* an explicit signature to quantify over all implicit
485 params for the reasons mentioned above.]
486
487 But that raises a new question.  Consider 
488
489         Given (signature)       ?x::Int
490         Wanted (inferred)       ?x::Int, ?y::Bool
491
492 Clearly we want to discharge the ?x and float the ?y out.  But
493 what is the criterion that distinguishes them?  Clearly it isn't
494 what free type variables they have.  The Right Thing seems to be
495 to float a constraint that
496         neither mentions any of the quantified type variables
497         nor any of the quantified implicit parameters
498
499 See the predicate isFreeWhenChecking.
500
501
502 Question 3: monomorphism
503 ~~~~~~~~~~~~~~~~~~~~~~~~
504 There's a nasty corner case when the monomorphism restriction bites:
505
506         z = (x::Int) + ?y
507
508 The argument above suggests that we *must* generalise
509 over the ?y parameter, to get
510         z :: (?y::Int) => Int,
511 but the monomorphism restriction says that we *must not*, giving
512         z :: Int.
513 Why does the momomorphism restriction say this?  Because if you have
514
515         let z = x + ?y in z+z
516
517 you might not expect the addition to be done twice --- but it will if
518 we follow the argument of Question 2 and generalise over ?y.
519
520
521 Question 4: top level
522 ~~~~~~~~~~~~~~~~~~~~~
523 At the top level, monomorhism makes no sense at all.
524
525     module Main where
526         main = let ?x = 5 in print foo
527
528         foo = woggle 3
529
530         woggle :: (?x :: Int) => Int -> Int
531         woggle y = ?x + y
532
533 We definitely don't want (foo :: Int) with a top-level implicit parameter
534 (?x::Int) becuase there is no way to bind it.  
535
536
537 Possible choices
538 ~~~~~~~~~~~~~~~~
539 (A) Always generalise over implicit parameters
540     Bindings that fall under the monomorphism restriction can't
541         be generalised
542
543     Consequences:
544         * Inlining remains valid
545         * No unexpected loss of sharing
546         * But simple bindings like
547                 z = ?y + 1
548           will be rejected, unless you add an explicit type signature
549           (to avoid the monomorphism restriction)
550                 z :: (?y::Int) => Int
551                 z = ?y + 1
552           This seems unacceptable
553
554 (B) Monomorphism restriction "wins"
555     Bindings that fall under the monomorphism restriction can't
556         be generalised
557     Always generalise over implicit parameters *except* for bindings
558         that fall under the monomorphism restriction
559
560     Consequences
561         * Inlining isn't valid in general
562         * No unexpected loss of sharing
563         * Simple bindings like
564                 z = ?y + 1
565           accepted (get value of ?y from binding site)
566
567 (C) Always generalise over implicit parameters
568     Bindings that fall under the monomorphism restriction can't
569         be generalised, EXCEPT for implicit parameters
570     Consequences
571         * Inlining remains valid
572         * Unexpected loss of sharing (from the extra generalisation)
573         * Simple bindings like
574                 z = ?y + 1
575           accepted (get value of ?y from occurrence sites)
576
577
578 Discussion
579 ~~~~~~~~~~
580 None of these choices seems very satisfactory.  But at least we should
581 decide which we want to do.
582
583 It's really not clear what is the Right Thing To Do.  If you see
584
585         z = (x::Int) + ?y
586
587 would you expect the value of ?y to be got from the *occurrence sites*
588 of 'z', or from the valuue of ?y at the *definition* of 'z'?  In the
589 case of function definitions, the answer is clearly the former, but
590 less so in the case of non-fucntion definitions.   On the other hand,
591 if we say that we get the value of ?y from the definition site of 'z',
592 then inlining 'z' might change the semantics of the program.
593
594 Choice (C) really says "the monomorphism restriction doesn't apply
595 to implicit parameters".  Which is fine, but remember that every
596 innocent binding 'x = ...' that mentions an implicit parameter in
597 the RHS becomes a *function* of that parameter, called at each
598 use of 'x'.  Now, the chances are that there are no intervening 'with'
599 clauses that bind ?y, so a decent compiler should common up all
600 those function calls.  So I think I strongly favour (C).  Indeed,
601 one could make a similar argument for abolishing the monomorphism
602 restriction altogether.
603
604 BOTTOM LINE: we choose (B) at present.  See tcSimplifyRestricted
605
606
607
608 %************************************************************************
609 %*                                                                      *
610 \subsection{tcSimplifyInfer}
611 %*                                                                      *
612 %************************************************************************
613
614 tcSimplify is called when we *inferring* a type.  Here's the overall game plan:
615
616     1. Compute Q = grow( fvs(T), C )
617
618     2. Partition C based on Q into Ct and Cq.  Notice that ambiguous
619        predicates will end up in Ct; we deal with them at the top level
620
621     3. Try improvement, using functional dependencies
622
623     4. If Step 3 did any unification, repeat from step 1
624        (Unification can change the result of 'grow'.)
625
626 Note: we don't reduce dictionaries in step 2.  For example, if we have
627 Eq (a,b), we don't simplify to (Eq a, Eq b).  So Q won't be different
628 after step 2.  However note that we may therefore quantify over more
629 type variables than we absolutely have to.
630
631 For the guts, we need a loop, that alternates context reduction and
632 improvement with unification.  E.g. Suppose we have
633
634         class C x y | x->y where ...
635
636 and tcSimplify is called with:
637         (C Int a, C Int b)
638 Then improvement unifies a with b, giving
639         (C Int a, C Int a)
640
641 If we need to unify anything, we rattle round the whole thing all over
642 again.
643
644
645 \begin{code}
646 tcSimplifyInfer
647         :: SDoc
648         -> TcTyVarSet           -- fv(T); type vars
649         -> [Inst]               -- Wanted
650         -> TcM ([TcTyVar],      -- Tyvars to quantify (zonked and quantified)
651                 [Inst],         -- Dict Ids that must be bound here (zonked)
652                 TcDictBinds)    -- Bindings
653         -- Any free (escaping) Insts are tossed into the environment
654 \end{code}
655
656
657 \begin{code}
658 tcSimplifyInfer doc tau_tvs wanted
659   = do  { tau_tvs1 <- zonkTcTyVarsAndFV (varSetElems tau_tvs)
660         ; wanted'  <- mappM zonkInst wanted     -- Zonk before deciding quantified tyvars
661         ; gbl_tvs  <- tcGetGlobalTyVars
662         ; let preds1   = fdPredsOfInsts wanted'
663               gbl_tvs1 = oclose preds1 gbl_tvs
664               qtvs     = grow preds1 tau_tvs1 `minusVarSet` gbl_tvs1
665                         -- See Note [Choosing which variables to quantify]
666
667                 -- To maximise sharing, remove from consideration any 
668                 -- constraints that don't mention qtvs at all
669         ; let (free, bound) = partition (isFreeWhenInferring qtvs) wanted'
670         ; extendLIEs free
671
672                 -- To make types simple, reduce as much as possible
673         ; traceTc (text "infer" <+> (ppr preds1 $$ ppr (grow preds1 tau_tvs1) $$ ppr gbl_tvs $$ 
674                    ppr gbl_tvs1 $$ ppr free $$ ppr bound))
675         ; (irreds1, binds1) <- tryHardCheckLoop doc bound
676
677                 -- Note [Inference and implication constraints]
678         ; let want_dict d = tyVarsOfInst d `intersectsVarSet` qtvs
679         ; (irreds2, binds2) <- approximateImplications doc want_dict irreds1
680
681                 -- Now work out all over again which type variables to quantify,
682                 -- exactly in the same way as before, but starting from irreds2.  Why?
683                 -- a) By now improvment may have taken place, and we must *not*
684                 --    quantify over any variable free in the environment
685                 --    tc137 (function h inside g) is an example
686                 --
687                 -- b) Do not quantify over constraints that *now* do not
688                 --    mention quantified type variables, because they are
689                 --    simply ambiguous (or might be bound further out).  Example:
690                 --      f :: Eq b => a -> (a, b)
691                 --      g x = fst (f x)
692                 --    From the RHS of g we get the MethodInst f77 :: alpha -> (alpha, beta)
693                 --    We decide to quantify over 'alpha' alone, but free1 does not include f77
694                 --    because f77 mentions 'alpha'.  Then reducing leaves only the (ambiguous)
695                 --    constraint (Eq beta), which we dump back into the free set
696                 --    See test tcfail181
697                 --
698                 -- c) irreds may contain type variables not previously mentioned,
699                 --    e.g.  instance D a x => Foo [a] 
700                 --          wanteds = Foo [a]
701                 --       Then after simplifying we'll get (D a x), and x is fresh
702                 --       We must quantify over x else it'll be totally unbound
703         ; tau_tvs2 <- zonkTcTyVarsAndFV (varSetElems tau_tvs1)
704         ; gbl_tvs2 <- zonkTcTyVarsAndFV (varSetElems gbl_tvs1)
705                 -- Note that we start from gbl_tvs1
706                 -- We use tcGetGlobalTyVars, then oclose wrt preds2, because
707                 -- we've already put some of the original preds1 into frees
708                 -- E.g.         wanteds = C a b   (where a->b)
709                 --              gbl_tvs = {a}
710                 --              tau_tvs = {b}
711                 -- Then b is fixed by gbl_tvs, so (C a b) will be in free, and
712                 -- irreds2 will be empty.  But we don't want to generalise over b!
713         ; let preds2 = fdPredsOfInsts irreds2   -- irreds2 is zonked
714               qtvs   = grow preds2 tau_tvs2 `minusVarSet` oclose preds2 gbl_tvs2
715         ; let (free, irreds3) = partition (isFreeWhenInferring qtvs) irreds2
716         ; extendLIEs free
717
718                 -- Turn the quantified meta-type variables into real type variables
719         ; qtvs2 <- zonkQuantifiedTyVars (varSetElems qtvs)
720
721                 -- We can't abstract over any remaining unsolved 
722                 -- implications so instead just float them outwards. Ugh.
723         ; let (q_dicts0, implics) = partition isAbstractableInst irreds3
724         ; loc <- getInstLoc (ImplicOrigin doc)
725         ; implic_bind <- bindIrreds loc qtvs2 q_dicts0 implics
726
727                 -- Prepare equality instances for quantification
728         ; let (q_eqs0,q_dicts) = partition isEqInst q_dicts0
729         ; q_eqs <- mappM finalizeEqInst q_eqs0
730
731         ; return (qtvs2, q_eqs ++ q_dicts, binds1 `unionBags` binds2 `unionBags` implic_bind) }
732         -- NB: when we are done, we might have some bindings, but
733         -- the final qtvs might be empty.  See Note [NO TYVARS] below.
734
735 approximateImplications :: SDoc -> (Inst -> Bool) -> [Inst] -> TcM ([Inst], TcDictBinds)
736 -- Note [Inference and implication constraints]
737 -- Given a bunch of Dict and ImplicInsts, try to approximate the implications by
738 --      - fetching any dicts inside them that are free
739 --      - using those dicts as cruder constraints, to solve the implications
740 --      - returning the extra ones too
741
742 approximateImplications doc want_dict irreds
743   | null extra_dicts 
744   = return (irreds, emptyBag)
745   | otherwise
746   = do  { extra_dicts' <- mapM cloneDict extra_dicts
747         ; tryHardCheckLoop doc (extra_dicts' ++ irreds) }
748                 -- By adding extra_dicts', we make them 
749                 -- available to solve the implication constraints
750   where 
751     extra_dicts = get_dicts (filter isImplicInst irreds)
752
753     get_dicts :: [Inst] -> [Inst]       -- Returns only Dicts
754         -- Find the wanted constraints in implication constraints that satisfy
755         -- want_dict, and are not bound by forall's in the constraint itself
756     get_dicts ds = concatMap get_dict ds
757
758     get_dict d@(Dict {}) | want_dict d = [d]
759                          | otherwise   = []
760     get_dict (ImplicInst {tci_tyvars = tvs, tci_wanted = wanteds})
761         = [ d | let tv_set = mkVarSet tvs
762               , d <- get_dicts wanteds 
763               , not (tyVarsOfInst d `intersectsVarSet` tv_set)]
764     get_dict i@(EqInst {}) | want_dict i = [i]
765                            | otherwise   = [] 
766     get_dict other = pprPanic "approximateImplications" (ppr other)
767 \end{code}
768
769 Note [Inference and implication constraints]
770 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 
771 Suppose we have a wanted implication constraint (perhaps arising from
772 a nested pattern match) like
773         C a => D [a]
774 and we are now trying to quantify over 'a' when inferring the type for
775 a function.  In principle it's possible that there might be an instance
776         instance (C a, E a) => D [a]
777 so the context (E a) would suffice.  The Right Thing is to abstract over
778 the implication constraint, but we don't do that (a) because it'll be
779 surprising to programmers and (b) because we don't have the machinery to deal
780 with 'given' implications.
781
782 So our best approximation is to make (D [a]) part of the inferred
783 context, so we can use that to discharge the implication. Hence
784 the strange function get_dicts in approximateImplications.
785
786 The common cases are more clear-cut, when we have things like
787         forall a. C a => C b
788 Here, abstracting over (C b) is not an approximation at all -- but see
789 Note [Freeness and implications].
790  
791 See Trac #1430 and test tc228.
792
793
794 \begin{code}
795 -----------------------------------------------------------
796 -- tcSimplifyInferCheck is used when we know the constraints we are to simplify
797 -- against, but we don't know the type variables over which we are going to quantify.
798 -- This happens when we have a type signature for a mutually recursive group
799 tcSimplifyInferCheck
800          :: InstLoc
801          -> TcTyVarSet          -- fv(T)
802          -> [Inst]              -- Given
803          -> [Inst]              -- Wanted
804          -> TcM ([TyVar],       -- Fully zonked, and quantified
805                  TcDictBinds)   -- Bindings
806
807 tcSimplifyInferCheck loc tau_tvs givens wanteds
808   = do  { traceTc (text "tcSimplifyInferCheck <-" <+> ppr wanteds)
809         ; (irreds, binds) <- gentleCheckLoop loc givens wanteds
810
811         -- Figure out which type variables to quantify over
812         -- You might think it should just be the signature tyvars,
813         -- but in bizarre cases you can get extra ones
814         --      f :: forall a. Num a => a -> a
815         --      f x = fst (g (x, head [])) + 1
816         --      g a b = (b,a)
817         -- Here we infer g :: forall a b. a -> b -> (b,a)
818         -- We don't want g to be monomorphic in b just because
819         -- f isn't quantified over b.
820         ; let all_tvs = varSetElems (tau_tvs `unionVarSet` tyVarsOfInsts givens)
821         ; all_tvs <- zonkTcTyVarsAndFV all_tvs
822         ; gbl_tvs <- tcGetGlobalTyVars
823         ; let qtvs = varSetElems (all_tvs `minusVarSet` gbl_tvs)
824                 -- We could close gbl_tvs, but its not necessary for
825                 -- soundness, and it'll only affect which tyvars, not which
826                 -- dictionaries, we quantify over
827
828         ; qtvs' <- zonkQuantifiedTyVars qtvs
829
830                 -- Now we are back to normal (c.f. tcSimplCheck)
831         ; implic_bind <- bindIrreds loc qtvs' givens irreds
832
833         ; traceTc (text "tcSimplifyInferCheck ->" <+> ppr (implic_bind))
834         ; return (qtvs', binds `unionBags` implic_bind) }
835 \end{code}
836
837 Note [Squashing methods]
838 ~~~~~~~~~~~~~~~~~~~~~~~~~
839 Be careful if you want to float methods more:
840         truncate :: forall a. RealFrac a => forall b. Integral b => a -> b
841 From an application (truncate f i) we get
842         t1 = truncate at f
843         t2 = t1 at i
844 If we have also have a second occurrence of truncate, we get
845         t3 = truncate at f
846         t4 = t3 at i
847 When simplifying with i,f free, we might still notice that
848 t1=t3; but alas, the binding for t2 (which mentions t1)
849 may continue to float out!
850
851
852 Note [NO TYVARS]
853 ~~~~~~~~~~~~~~~~~
854         class Y a b | a -> b where
855             y :: a -> X b
856         
857         instance Y [[a]] a where
858             y ((x:_):_) = X x
859         
860         k :: X a -> X a -> X a
861
862         g :: Num a => [X a] -> [X a]
863         g xs = h xs
864             where
865             h ys = ys ++ map (k (y [[0]])) xs
866
867 The excitement comes when simplifying the bindings for h.  Initially
868 try to simplify {y @ [[t1]] t2, 0 @ t1}, with initial qtvs = {t2}.
869 From this we get t1:=:t2, but also various bindings.  We can't forget
870 the bindings (because of [LOOP]), but in fact t1 is what g is
871 polymorphic in.  
872
873 The net effect of [NO TYVARS] 
874
875 \begin{code}
876 isFreeWhenInferring :: TyVarSet -> Inst -> Bool
877 isFreeWhenInferring qtvs inst
878   =  isFreeWrtTyVars qtvs inst  -- Constrains no quantified vars
879   && isInheritableInst inst     -- and no implicit parameter involved
880                                 --   see Note [Inheriting implicit parameters]
881
882 {-      No longer used (with implication constraints)
883 isFreeWhenChecking :: TyVarSet  -- Quantified tyvars
884                    -> NameSet   -- Quantified implicit parameters
885                    -> Inst -> Bool
886 isFreeWhenChecking qtvs ips inst
887   =  isFreeWrtTyVars qtvs inst
888   && isFreeWrtIPs    ips inst
889 -}
890
891 isFreeWrtTyVars qtvs inst = tyVarsOfInst inst `disjointVarSet` qtvs
892 isFreeWrtIPs     ips inst = not (any (`elemNameSet` ips) (ipNamesOfInst inst))
893 \end{code}
894
895
896 %************************************************************************
897 %*                                                                      *
898 \subsection{tcSimplifyCheck}
899 %*                                                                      *
900 %************************************************************************
901
902 @tcSimplifyCheck@ is used when we know exactly the set of variables
903 we are going to quantify over.  For example, a class or instance declaration.
904
905 \begin{code}
906 -----------------------------------------------------------
907 -- tcSimplifyCheck is used when checking expression type signatures,
908 -- class decls, instance decls etc.
909 tcSimplifyCheck :: InstLoc
910                 -> [TcTyVar]            -- Quantify over these
911                 -> [Inst]               -- Given
912                 -> [Inst]               -- Wanted
913                 -> TcM TcDictBinds      -- Bindings
914 tcSimplifyCheck loc qtvs givens wanteds 
915   = ASSERT( all isTcTyVar qtvs && all isSkolemTyVar qtvs )
916     do  { traceTc (text "tcSimplifyCheck")
917         ; (irreds, binds) <- gentleCheckLoop loc givens wanteds
918         ; implic_bind <- bindIrreds loc qtvs givens irreds
919         ; return (binds `unionBags` implic_bind) }
920
921 -----------------------------------------------------------
922 -- tcSimplifyCheckPat is used for existential pattern match
923 tcSimplifyCheckPat :: InstLoc
924                    -> [CoVar] -> Refinement
925                    -> [TcTyVar]         -- Quantify over these
926                    -> [Inst]            -- Given
927                    -> [Inst]            -- Wanted
928                    -> TcM TcDictBinds   -- Bindings
929 tcSimplifyCheckPat loc co_vars reft qtvs givens wanteds
930   = ASSERT( all isTcTyVar qtvs && all isSkolemTyVar qtvs )
931     do  { traceTc (text "tcSimplifyCheckPat")
932         ; (irreds, binds) <- gentleCheckLoop loc givens wanteds
933         ; implic_bind <- bindIrredsR loc qtvs co_vars reft 
934                                     givens irreds
935         ; return (binds `unionBags` implic_bind) }
936
937 -----------------------------------------------------------
938 bindIrreds :: InstLoc -> [TcTyVar]
939            -> [Inst] -> [Inst]
940            -> TcM TcDictBinds
941 bindIrreds loc qtvs givens irreds 
942   = bindIrredsR loc qtvs [] emptyRefinement givens irreds
943
944 bindIrredsR :: InstLoc -> [TcTyVar] -> [CoVar]
945             -> Refinement -> [Inst] -> [Inst]
946             -> TcM TcDictBinds  
947 -- Make a binding that binds 'irreds', by generating an implication
948 -- constraint for them, *and* throwing the constraint into the LIE
949 bindIrredsR loc qtvs co_vars reft givens irreds
950   | null irreds
951   = return emptyBag
952   | otherwise
953   = do  { let givens' = filter isAbstractableInst givens
954                 -- The givens can (redundantly) include methods
955                 -- We want to retain both EqInsts and Dicts
956                 -- There should be no implicadtion constraints
957                 -- See Note [Pruning the givens in an implication constraint]
958
959            -- If there are no 'givens' *and* the refinement is empty
960            -- (the refinement is like more givens), then it's safe to 
961            -- partition the 'wanteds' by their qtvs, thereby trimming irreds
962            -- See Note [Freeness and implications]
963         ; irreds' <- if null givens' && isEmptyRefinement reft
964                      then do
965                         { let qtv_set = mkVarSet qtvs
966                               (frees, real_irreds) = partition (isFreeWrtTyVars qtv_set) irreds
967                         ; extendLIEs frees
968                         ; return real_irreds }
969                      else return irreds
970         
971         ; let all_tvs = qtvs ++ co_vars -- Abstract over all these
972         ; (implics, bind) <- makeImplicationBind loc all_tvs reft givens' irreds'
973                         -- This call does the real work
974                         -- If irreds' is empty, it does something sensible
975         ; extendLIEs implics
976         ; return bind } 
977
978
979 makeImplicationBind :: InstLoc -> [TcTyVar] -> Refinement
980                     -> [Inst] -> [Inst]
981                     -> TcM ([Inst], TcDictBinds)
982 -- Make a binding that binds 'irreds', by generating an implication
983 -- constraint for them, *and* throwing the constraint into the LIE
984 -- The binding looks like
985 --      (ir1, .., irn) = f qtvs givens
986 -- where f is (evidence for) the new implication constraint
987 --      f :: forall qtvs. {reft} givens => (ir1, .., irn)
988 -- qtvs includes coercion variables
989 --
990 -- This binding must line up the 'rhs' in reduceImplication
991 makeImplicationBind loc all_tvs reft
992                     givens      -- Guaranteed all Dicts
993                                 -- or EqInsts
994                     irreds
995  | null irreds                  -- If there are no irreds, we are done
996  = return ([], emptyBag)
997  | otherwise                    -- Otherwise we must generate a binding
998  = do   { uniq <- newUnique 
999         ; span <- getSrcSpanM
1000         ; let (eq_givens, dict_givens) = partition isEqInst givens
1001               eq_tyvar_cos = mkTyVarTys (varSetElems $ tyVarsOfTypes $ map eqInstType eq_givens)
1002                 -- Urgh! See line 2187 or thereabouts.  I believe that all these
1003                 -- 'givens' must be a simple CoVar.  This MUST be cleaned up.
1004
1005         ; let name = mkInternalName uniq (mkVarOcc "ic") span
1006               implic_inst = ImplicInst { tci_name = name, tci_reft = reft,
1007                                          tci_tyvars = all_tvs, 
1008                                          tci_given = (eq_givens ++ dict_givens),
1009                                          tci_wanted = irreds, tci_loc = loc }
1010         ; let   -- only create binder for dict_irreds
1011               (eq_irreds, dict_irreds) = partition isEqInst irreds
1012               n_dict_irreds = length dict_irreds
1013               dict_irred_ids = map instToId dict_irreds
1014               tup_ty = mkTupleTy Boxed n_dict_irreds (map idType dict_irred_ids)
1015               pat = TuplePat (map nlVarPat dict_irred_ids) Boxed tup_ty
1016               rhs = L span (mkHsWrap co (HsVar (instToId implic_inst)))
1017               co  = mkWpApps (map instToId dict_givens)
1018                     <.> mkWpTyApps eq_tyvar_cos
1019                     <.> mkWpTyApps (mkTyVarTys all_tvs)
1020               bind | [dict_irred_id] <- dict_irred_ids  = VarBind dict_irred_id rhs
1021                    | otherwise        = PatBind { pat_lhs = L span pat, 
1022                                                   pat_rhs = unguardedGRHSs rhs, 
1023                                                   pat_rhs_ty = tup_ty,
1024                                                   bind_fvs = placeHolderNames }
1025         ; traceTc $ text "makeImplicationBind" <+> ppr implic_inst
1026         ; return ([implic_inst], unitBag (L span bind)) 
1027         }
1028
1029 -----------------------------------------------------------
1030 tryHardCheckLoop :: SDoc
1031              -> [Inst]                  -- Wanted
1032              -> TcM ([Inst], TcDictBinds)
1033
1034 tryHardCheckLoop doc wanteds
1035   = do { (irreds,binds) <- checkLoop (mkRedEnv doc try_me []) wanteds
1036        ; return (irreds,binds)
1037        }
1038   where
1039     try_me inst = ReduceMe AddSCs
1040         -- Here's the try-hard bit
1041
1042 -----------------------------------------------------------
1043 gentleCheckLoop :: InstLoc
1044                -> [Inst]                -- Given
1045                -> [Inst]                -- Wanted
1046                -> TcM ([Inst], TcDictBinds)
1047
1048 gentleCheckLoop inst_loc givens wanteds
1049   = do { (irreds,binds) <- checkLoop env wanteds
1050        ; return (irreds,binds)
1051        }
1052   where
1053     env = mkRedEnv (pprInstLoc inst_loc) try_me givens
1054
1055     try_me inst | isMethodOrLit inst = ReduceMe AddSCs
1056                 | otherwise          = Stop
1057         -- When checking against a given signature 
1058         -- we MUST be very gentle: Note [Check gently]
1059
1060 gentleInferLoop :: SDoc -> [Inst]
1061                 -> TcM ([Inst], TcDictBinds)
1062 gentleInferLoop doc wanteds
1063   = do  { (irreds, binds) <- checkLoop env wanteds
1064         ; return (irreds, binds) }
1065   where
1066     env = mkRedEnv doc try_me []
1067     try_me inst | isMethodOrLit inst = ReduceMe AddSCs
1068                 | otherwise          = Stop
1069 \end{code}
1070
1071 Note [Check gently]
1072 ~~~~~~~~~~~~~~~~~~~~
1073 We have to very careful about not simplifying too vigorously
1074 Example:  
1075   data T a where
1076     MkT :: a -> T [a]
1077
1078   f :: Show b => T b -> b
1079   f (MkT x) = show [x]
1080
1081 Inside the pattern match, which binds (a:*, x:a), we know that
1082         b ~ [a]
1083 Hence we have a dictionary for Show [a] available; and indeed we 
1084 need it.  We are going to build an implication contraint
1085         forall a. (b~[a]) => Show [a]
1086 Later, we will solve this constraint using the knowledge (Show b)
1087         
1088 But we MUST NOT reduce (Show [a]) to (Show a), else the whole
1089 thing becomes insoluble.  So we simplify gently (get rid of literals
1090 and methods only, plus common up equal things), deferring the real
1091 work until top level, when we solve the implication constraint
1092 with tryHardCheckLooop.
1093
1094
1095 \begin{code}
1096 -----------------------------------------------------------
1097 checkLoop :: RedEnv
1098           -> [Inst]                     -- Wanted
1099           -> TcM ([Inst], TcDictBinds) 
1100 -- Precondition: givens are completely rigid
1101 -- Postcondition: returned Insts are zonked
1102
1103 checkLoop env wanteds
1104   = go env wanteds
1105   where go env wanteds
1106           = do {  -- We do need to zonk the givens; cf Note [Zonking RedEnv]
1107                 ; env'     <- zonkRedEnv env
1108                 ; wanteds' <- zonkInsts  wanteds
1109         
1110                 ; (improved, binds, irreds) <- reduceContext env' wanteds'
1111
1112                 ; if not improved then
1113                      return (irreds, binds)
1114                   else do
1115         
1116                 -- If improvement did some unification, we go round again.
1117                 -- We start again with irreds, not wanteds
1118                 -- Using an instance decl might have introduced a fresh type variable
1119                 -- which might have been unified, so we'd get an infinite loop
1120                 -- if we started again with wanteds!  See Note [LOOP]
1121                 { (irreds1, binds1) <- go env' irreds
1122                 ; return (irreds1, binds `unionBags` binds1) } }
1123 \end{code}
1124
1125 Note [Zonking RedEnv]
1126 ~~~~~~~~~~~~~~~~~~~~~
1127 It might appear as if the givens in RedEnv are always rigid, but that is not
1128 necessarily the case for programs involving higher-rank types that have class
1129 contexts constraining the higher-rank variables.  An example from tc237 in the
1130 testsuite is
1131
1132   class Modular s a | s -> a
1133
1134   wim ::  forall a w. Integral a 
1135                         => a -> (forall s. Modular s a => M s w) -> w
1136   wim i k = error "urk"
1137
1138   test5  ::  (Modular s a, Integral a) => M s a
1139   test5  =   error "urk"
1140
1141   test4   =   wim 4 test4'
1142
1143 Notice how the variable 'a' of (Modular s a) in the rank-2 type of wim is
1144 quantified further outside.  When type checking test4, we have to check
1145 whether the signature of test5 is an instance of 
1146
1147   (forall s. Modular s a => M s w)
1148
1149 Consequently, we will get (Modular s t_a), where t_a is a TauTv into the
1150 givens. 
1151
1152 Given the FD of Modular in this example, class improvement will instantiate
1153 t_a to 'a', where 'a' is the skolem from test5's signatures (due to the
1154 Modular s a predicate in that signature).  If we don't zonk (Modular s t_a) in
1155 the givens, we will get into a loop as improveOne uses the unification engine
1156 TcGadt.tcUnifyTys, which doesn't know about mutable type variables.
1157
1158
1159 Note [LOOP]
1160 ~~~~~~~~~~~
1161         class If b t e r | b t e -> r
1162         instance If T t e t
1163         instance If F t e e
1164         class Lte a b c | a b -> c where lte :: a -> b -> c
1165         instance Lte Z b T
1166         instance (Lte a b l,If l b a c) => Max a b c
1167
1168 Wanted: Max Z (S x) y
1169
1170 Then we'll reduce using the Max instance to:
1171         (Lte Z (S x) l, If l (S x) Z y)
1172 and improve by binding l->T, after which we can do some reduction
1173 on both the Lte and If constraints.  What we *can't* do is start again
1174 with (Max Z (S x) y)!
1175
1176
1177
1178 %************************************************************************
1179 %*                                                                      *
1180                 tcSimplifySuperClasses
1181 %*                                                                      *
1182 %************************************************************************
1183
1184 Note [SUPERCLASS-LOOP 1]
1185 ~~~~~~~~~~~~~~~~~~~~~~~~
1186 We have to be very, very careful when generating superclasses, lest we
1187 accidentally build a loop. Here's an example:
1188
1189   class S a
1190
1191   class S a => C a where { opc :: a -> a }
1192   class S b => D b where { opd :: b -> b }
1193   
1194   instance C Int where
1195      opc = opd
1196   
1197   instance D Int where
1198      opd = opc
1199
1200 From (instance C Int) we get the constraint set {ds1:S Int, dd:D Int}
1201 Simplifying, we may well get:
1202         $dfCInt = :C ds1 (opd dd)
1203         dd  = $dfDInt
1204         ds1 = $p1 dd
1205 Notice that we spot that we can extract ds1 from dd.  
1206
1207 Alas!  Alack! We can do the same for (instance D Int):
1208
1209         $dfDInt = :D ds2 (opc dc)
1210         dc  = $dfCInt
1211         ds2 = $p1 dc
1212
1213 And now we've defined the superclass in terms of itself.
1214
1215 Solution: never generate a superclass selectors at all when
1216 satisfying the superclass context of an instance declaration.
1217
1218 Two more nasty cases are in
1219         tcrun021
1220         tcrun033
1221
1222 \begin{code}
1223 tcSimplifySuperClasses 
1224         :: InstLoc 
1225         -> [Inst]       -- Given 
1226         -> [Inst]       -- Wanted
1227         -> TcM TcDictBinds
1228 tcSimplifySuperClasses loc givens sc_wanteds
1229   = do  { traceTc (text "tcSimplifySuperClasses")
1230         ; (irreds,binds1) <- checkLoop env sc_wanteds
1231         ; let (tidy_env, tidy_irreds) = tidyInsts irreds
1232         ; reportNoInstances tidy_env (Just (loc, givens)) tidy_irreds
1233         ; return binds1 }
1234   where
1235     env = mkRedEnv (pprInstLoc loc) try_me givens
1236     try_me inst = ReduceMe NoSCs
1237         -- Like tryHardCheckLoop, but with NoSCs
1238 \end{code}
1239
1240
1241 %************************************************************************
1242 %*                                                                      *
1243 \subsection{tcSimplifyRestricted}
1244 %*                                                                      *
1245 %************************************************************************
1246
1247 tcSimplifyRestricted infers which type variables to quantify for a 
1248 group of restricted bindings.  This isn't trivial.
1249
1250 Eg1:    id = \x -> x
1251         We want to quantify over a to get id :: forall a. a->a
1252         
1253 Eg2:    eq = (==)
1254         We do not want to quantify over a, because there's an Eq a 
1255         constraint, so we get eq :: a->a->Bool  (notice no forall)
1256
1257 So, assume:
1258         RHS has type 'tau', whose free tyvars are tau_tvs
1259         RHS has constraints 'wanteds'
1260
1261 Plan A (simple)
1262   Quantify over (tau_tvs \ ftvs(wanteds))
1263   This is bad. The constraints may contain (Monad (ST s))
1264   where we have         instance Monad (ST s) where...
1265   so there's no need to be monomorphic in s!
1266
1267   Also the constraint might be a method constraint,
1268   whose type mentions a perfectly innocent tyvar:
1269           op :: Num a => a -> b -> a
1270   Here, b is unconstrained.  A good example would be
1271         foo = op (3::Int)
1272   We want to infer the polymorphic type
1273         foo :: forall b. b -> b
1274
1275
1276 Plan B (cunning, used for a long time up to and including GHC 6.2)
1277   Step 1: Simplify the constraints as much as possible (to deal 
1278   with Plan A's problem).  Then set
1279         qtvs = tau_tvs \ ftvs( simplify( wanteds ) )
1280
1281   Step 2: Now simplify again, treating the constraint as 'free' if 
1282   it does not mention qtvs, and trying to reduce it otherwise.
1283   The reasons for this is to maximise sharing.
1284
1285   This fails for a very subtle reason.  Suppose that in the Step 2
1286   a constraint (Foo (Succ Zero) (Succ Zero) b) gets thrown upstairs as 'free'.
1287   In the Step 1 this constraint might have been simplified, perhaps to
1288   (Foo Zero Zero b), AND THEN THAT MIGHT BE IMPROVED, to bind 'b' to 'T'.
1289   This won't happen in Step 2... but that in turn might prevent some other
1290   constraint (Baz [a] b) being simplified (e.g. via instance Baz [a] T where {..}) 
1291   and that in turn breaks the invariant that no constraints are quantified over.
1292
1293   Test typecheck/should_compile/tc177 (which failed in GHC 6.2) demonstrates
1294   the problem.
1295
1296
1297 Plan C (brutal)
1298   Step 1: Simplify the constraints as much as possible (to deal 
1299   with Plan A's problem).  Then set
1300         qtvs = tau_tvs \ ftvs( simplify( wanteds ) )
1301   Return the bindings from Step 1.
1302   
1303
1304 A note about Plan C (arising from "bug" reported by George Russel March 2004)
1305 Consider this:
1306
1307       instance (HasBinary ty IO) => HasCodedValue ty
1308
1309       foo :: HasCodedValue a => String -> IO a
1310
1311       doDecodeIO :: HasCodedValue a => () -> () -> IO a
1312       doDecodeIO codedValue view  
1313         = let { act = foo "foo" } in  act
1314
1315 You might think this should work becuase the call to foo gives rise to a constraint
1316 (HasCodedValue t), which can be satisfied by the type sig for doDecodeIO.  But the
1317 restricted binding act = ... calls tcSimplifyRestricted, and PlanC simplifies the
1318 constraint using the (rather bogus) instance declaration, and now we are stuffed.
1319
1320 I claim this is not really a bug -- but it bit Sergey as well as George.  So here's
1321 plan D
1322
1323
1324 Plan D (a variant of plan B)
1325   Step 1: Simplify the constraints as much as possible (to deal 
1326   with Plan A's problem), BUT DO NO IMPROVEMENT.  Then set
1327         qtvs = tau_tvs \ ftvs( simplify( wanteds ) )
1328
1329   Step 2: Now simplify again, treating the constraint as 'free' if 
1330   it does not mention qtvs, and trying to reduce it otherwise.
1331
1332   The point here is that it's generally OK to have too few qtvs; that is,
1333   to make the thing more monomorphic than it could be.  We don't want to
1334   do that in the common cases, but in wierd cases it's ok: the programmer
1335   can always add a signature.  
1336
1337   Too few qtvs => too many wanteds, which is what happens if you do less
1338   improvement.
1339
1340
1341 \begin{code}
1342 tcSimplifyRestricted    -- Used for restricted binding groups
1343                         -- i.e. ones subject to the monomorphism restriction
1344         :: SDoc
1345         -> TopLevelFlag
1346         -> [Name]               -- Things bound in this group
1347         -> TcTyVarSet           -- Free in the type of the RHSs
1348         -> [Inst]               -- Free in the RHSs
1349         -> TcM ([TyVar],        -- Tyvars to quantify (zonked and quantified)
1350                 TcDictBinds)    -- Bindings
1351         -- tcSimpifyRestricted returns no constraints to
1352         -- quantify over; by definition there are none.
1353         -- They are all thrown back in the LIE
1354
1355 tcSimplifyRestricted doc top_lvl bndrs tau_tvs wanteds
1356         -- Zonk everything in sight
1357   = do  { traceTc (text "tcSimplifyRestricted")
1358         ; wanteds' <- zonkInsts wanteds
1359
1360         -- 'ReduceMe': Reduce as far as we can.  Don't stop at
1361         -- dicts; the idea is to get rid of as many type
1362         -- variables as possible, and we don't want to stop
1363         -- at (say) Monad (ST s), because that reduces
1364         -- immediately, with no constraint on s.
1365         --
1366         -- BUT do no improvement!  See Plan D above
1367         -- HOWEVER, some unification may take place, if we instantiate
1368         --          a method Inst with an equality constraint
1369         ; let env = mkNoImproveRedEnv doc (\i -> ReduceMe AddSCs)
1370         ; (_imp, _binds, constrained_dicts) <- reduceContext env wanteds'
1371
1372         -- Next, figure out the tyvars we will quantify over
1373         ; tau_tvs' <- zonkTcTyVarsAndFV (varSetElems tau_tvs)
1374         ; gbl_tvs' <- tcGetGlobalTyVars
1375         ; constrained_dicts' <- zonkInsts constrained_dicts
1376
1377         ; let qtvs1 = tau_tvs' `minusVarSet` oclose (fdPredsOfInsts constrained_dicts) gbl_tvs'
1378                                 -- As in tcSimplifyInfer
1379
1380                 -- Do not quantify over constrained type variables:
1381                 -- this is the monomorphism restriction
1382               constrained_tvs' = tyVarsOfInsts constrained_dicts'
1383               qtvs = qtvs1 `minusVarSet` constrained_tvs'
1384               pp_bndrs = pprWithCommas (quotes . ppr) bndrs
1385
1386         -- Warn in the mono
1387         ; warn_mono <- doptM Opt_WarnMonomorphism
1388         ; warnTc (warn_mono && (constrained_tvs' `intersectsVarSet` qtvs1))
1389                  (vcat[ ptext SLIT("the Monomorphism Restriction applies to the binding")
1390                                 <> plural bndrs <+> ptext SLIT("for") <+> pp_bndrs,
1391                         ptext SLIT("Consider giving a type signature for") <+> pp_bndrs])
1392
1393         ; traceTc (text "tcSimplifyRestricted" <+> vcat [
1394                 pprInsts wanteds, pprInsts constrained_dicts',
1395                 ppr _binds,
1396                 ppr constrained_tvs', ppr tau_tvs', ppr qtvs ])
1397
1398         -- The first step may have squashed more methods than
1399         -- necessary, so try again, this time more gently, knowing the exact
1400         -- set of type variables to quantify over.
1401         --
1402         -- We quantify only over constraints that are captured by qtvs;
1403         -- these will just be a subset of non-dicts.  This in contrast
1404         -- to normal inference (using isFreeWhenInferring) in which we quantify over
1405         -- all *non-inheritable* constraints too.  This implements choice
1406         -- (B) under "implicit parameter and monomorphism" above.
1407         --
1408         -- Remember that we may need to do *some* simplification, to
1409         -- (for example) squash {Monad (ST s)} into {}.  It's not enough
1410         -- just to float all constraints
1411         --
1412         -- At top level, we *do* squash methods becuase we want to 
1413         -- expose implicit parameters to the test that follows
1414         ; let is_nested_group = isNotTopLevel top_lvl
1415               try_me inst | isFreeWrtTyVars qtvs inst,
1416                            (is_nested_group || isDict inst) = Stop
1417                           | otherwise            = ReduceMe AddSCs
1418               env = mkNoImproveRedEnv doc try_me
1419         ; (_imp, binds, irreds) <- reduceContext env wanteds'
1420
1421         -- See "Notes on implicit parameters, Question 4: top level"
1422         ; ASSERT( all (isFreeWrtTyVars qtvs) irreds )   -- None should be captured
1423           if is_nested_group then
1424                 extendLIEs irreds
1425           else do { let (bad_ips, non_ips) = partition isIPDict irreds
1426                   ; addTopIPErrs bndrs bad_ips
1427                   ; extendLIEs non_ips }
1428
1429         ; qtvs' <- zonkQuantifiedTyVars (varSetElems qtvs)
1430         ; return (qtvs', binds) }
1431 \end{code}
1432
1433
1434 %************************************************************************
1435 %*                                                                      *
1436                 tcSimplifyRuleLhs
1437 %*                                                                      *
1438 %************************************************************************
1439
1440 On the LHS of transformation rules we only simplify methods and constants,
1441 getting dictionaries.  We want to keep all of them unsimplified, to serve
1442 as the available stuff for the RHS of the rule.
1443
1444 Example.  Consider the following left-hand side of a rule
1445         
1446         f (x == y) (y > z) = ...
1447
1448 If we typecheck this expression we get constraints
1449
1450         d1 :: Ord a, d2 :: Eq a
1451
1452 We do NOT want to "simplify" to the LHS
1453
1454         forall x::a, y::a, z::a, d1::Ord a.
1455           f ((==) (eqFromOrd d1) x y) ((>) d1 y z) = ...
1456
1457 Instead we want 
1458
1459         forall x::a, y::a, z::a, d1::Ord a, d2::Eq a.
1460           f ((==) d2 x y) ((>) d1 y z) = ...
1461
1462 Here is another example:
1463
1464         fromIntegral :: (Integral a, Num b) => a -> b
1465         {-# RULES "foo"  fromIntegral = id :: Int -> Int #-}
1466
1467 In the rule, a=b=Int, and Num Int is a superclass of Integral Int. But
1468 we *dont* want to get
1469
1470         forall dIntegralInt.
1471            fromIntegral Int Int dIntegralInt (scsel dIntegralInt) = id Int
1472
1473 because the scsel will mess up RULE matching.  Instead we want
1474
1475         forall dIntegralInt, dNumInt.
1476           fromIntegral Int Int dIntegralInt dNumInt = id Int
1477
1478 Even if we have 
1479
1480         g (x == y) (y == z) = ..
1481
1482 where the two dictionaries are *identical*, we do NOT WANT
1483
1484         forall x::a, y::a, z::a, d1::Eq a
1485           f ((==) d1 x y) ((>) d1 y z) = ...
1486
1487 because that will only match if the dict args are (visibly) equal.
1488 Instead we want to quantify over the dictionaries separately.
1489
1490 In short, tcSimplifyRuleLhs must *only* squash LitInst and MethInts, leaving
1491 all dicts unchanged, with absolutely no sharing.  It's simpler to do this
1492 from scratch, rather than further parameterise simpleReduceLoop etc
1493
1494 \begin{code}
1495 tcSimplifyRuleLhs :: [Inst] -> TcM ([Inst], TcDictBinds)
1496 tcSimplifyRuleLhs wanteds
1497   = go [] emptyBag wanteds
1498   where
1499     go dicts binds []
1500         = return (dicts, binds)
1501     go dicts binds (w:ws)
1502         | isDict w
1503         = go (w:dicts) binds ws
1504         | otherwise
1505         = do { w' <- zonkInst w  -- So that (3::Int) does not generate a call
1506                                  -- to fromInteger; this looks fragile to me
1507              ; lookup_result <- lookupSimpleInst w'
1508              ; case lookup_result of
1509                  GenInst ws' rhs -> 
1510                    go dicts (addInstToDictBind binds w rhs) (ws' ++ ws)
1511                  NoInstance      -> pprPanic "tcSimplifyRuleLhs" (ppr w)
1512           }
1513 \end{code}
1514
1515 tcSimplifyBracket is used when simplifying the constraints arising from
1516 a Template Haskell bracket [| ... |].  We want to check that there aren't
1517 any constraints that can't be satisfied (e.g. Show Foo, where Foo has no
1518 Show instance), but we aren't otherwise interested in the results.
1519 Nor do we care about ambiguous dictionaries etc.  We will type check
1520 this bracket again at its usage site.
1521
1522 \begin{code}
1523 tcSimplifyBracket :: [Inst] -> TcM ()
1524 tcSimplifyBracket wanteds
1525   = do  { tryHardCheckLoop doc wanteds
1526         ; return () }
1527   where
1528     doc = text "tcSimplifyBracket"
1529 \end{code}
1530
1531
1532 %************************************************************************
1533 %*                                                                      *
1534 \subsection{Filtering at a dynamic binding}
1535 %*                                                                      *
1536 %************************************************************************
1537
1538 When we have
1539         let ?x = R in B
1540
1541 we must discharge all the ?x constraints from B.  We also do an improvement
1542 step; if we have ?x::t1 and ?x::t2 we must unify t1, t2.
1543
1544 Actually, the constraints from B might improve the types in ?x. For example
1545
1546         f :: (?x::Int) => Char -> Char
1547         let ?x = 3 in f 'c'
1548
1549 then the constraint (?x::Int) arising from the call to f will
1550 force the binding for ?x to be of type Int.
1551
1552 \begin{code}
1553 tcSimplifyIPs :: [Inst]         -- The implicit parameters bound here
1554               -> [Inst]         -- Wanted
1555               -> TcM TcDictBinds
1556         -- We need a loop so that we do improvement, and then
1557         -- (next time round) generate a binding to connect the two
1558         --      let ?x = e in ?x
1559         -- Here the two ?x's have different types, and improvement 
1560         -- makes them the same.
1561
1562 tcSimplifyIPs given_ips wanteds
1563   = do  { wanteds'   <- zonkInsts wanteds
1564         ; given_ips' <- zonkInsts given_ips
1565                 -- Unusually for checking, we *must* zonk the given_ips
1566
1567         ; let env = mkRedEnv doc try_me given_ips'
1568         ; (improved, binds, irreds) <- reduceContext env wanteds'
1569
1570         ; if not improved then 
1571                 ASSERT( all is_free irreds )
1572                 do { extendLIEs irreds
1573                    ; return binds }
1574           else
1575                 tcSimplifyIPs given_ips wanteds }
1576   where
1577     doc    = text "tcSimplifyIPs" <+> ppr given_ips
1578     ip_set = mkNameSet (ipNamesOfInsts given_ips)
1579     is_free inst = isFreeWrtIPs ip_set inst
1580
1581         -- Simplify any methods that mention the implicit parameter
1582     try_me inst | is_free inst = Stop
1583                 | otherwise    = ReduceMe NoSCs
1584 \end{code}
1585
1586
1587 %************************************************************************
1588 %*                                                                      *
1589 \subsection[binds-for-local-funs]{@bindInstsOfLocalFuns@}
1590 %*                                                                      *
1591 %************************************************************************
1592
1593 When doing a binding group, we may have @Insts@ of local functions.
1594 For example, we might have...
1595 \begin{verbatim}
1596 let f x = x + 1     -- orig local function (overloaded)
1597     f.1 = f Int     -- two instances of f
1598     f.2 = f Float
1599  in
1600     (f.1 5, f.2 6.7)
1601 \end{verbatim}
1602 The point is: we must drop the bindings for @f.1@ and @f.2@ here,
1603 where @f@ is in scope; those @Insts@ must certainly not be passed
1604 upwards towards the top-level.  If the @Insts@ were binding-ified up
1605 there, they would have unresolvable references to @f@.
1606
1607 We pass in an @init_lie@ of @Insts@ and a list of locally-bound @Ids@.
1608 For each method @Inst@ in the @init_lie@ that mentions one of the
1609 @Ids@, we create a binding.  We return the remaining @Insts@ (in an
1610 @LIE@), as well as the @HsBinds@ generated.
1611
1612 \begin{code}
1613 bindInstsOfLocalFuns :: [Inst] -> [TcId] -> TcM TcDictBinds
1614 -- Simlifies only MethodInsts, and generate only bindings of form 
1615 --      fm = f tys dicts
1616 -- We're careful not to even generate bindings of the form
1617 --      d1 = d2
1618 -- You'd think that'd be fine, but it interacts with what is
1619 -- arguably a bug in Match.tidyEqnInfo (see notes there)
1620
1621 bindInstsOfLocalFuns wanteds local_ids
1622   | null overloaded_ids
1623         -- Common case
1624   = extendLIEs wanteds          `thenM_`
1625     returnM emptyLHsBinds
1626
1627   | otherwise
1628   = do  { (irreds, binds) <- gentleInferLoop doc for_me
1629         ; extendLIEs not_for_me 
1630         ; extendLIEs irreds
1631         ; return binds }
1632   where
1633     doc              = text "bindInsts" <+> ppr local_ids
1634     overloaded_ids   = filter is_overloaded local_ids
1635     is_overloaded id = isOverloadedTy (idType id)
1636     (for_me, not_for_me) = partition (isMethodFor overloaded_set) wanteds
1637
1638     overloaded_set = mkVarSet overloaded_ids    -- There can occasionally be a lot of them
1639                                                 -- so it's worth building a set, so that
1640                                                 -- lookup (in isMethodFor) is faster
1641 \end{code}
1642
1643
1644 %************************************************************************
1645 %*                                                                      *
1646 \subsection{Data types for the reduction mechanism}
1647 %*                                                                      *
1648 %************************************************************************
1649
1650 The main control over context reduction is here
1651
1652 \begin{code}
1653 data RedEnv 
1654   = RedEnv { red_doc    :: SDoc                 -- The context
1655            , red_try_me :: Inst -> WhatToDo
1656            , red_improve :: Bool                -- True <=> do improvement
1657            , red_givens :: [Inst]               -- All guaranteed rigid
1658                                                 -- Always dicts
1659                                                 -- but see Note [Rigidity]
1660            , red_reft :: Refinement             -- The refinement to apply to the 'givens'
1661                                                 -- You should think of it as 'given equalities'
1662            , red_stack  :: (Int, [Inst])        -- Recursion stack (for err msg)
1663                                                 -- See Note [RedStack]
1664   }
1665
1666 -- Note [Rigidity]
1667 -- The red_givens are rigid so far as cmpInst is concerned.
1668 -- There is one case where they are not totally rigid, namely in tcSimplifyIPs
1669 --      let ?x = e in ...
1670 -- Here, the given is (?x::a), where 'a' is not necy a rigid type
1671 -- But that doesn't affect the comparison, which is based only on mame.
1672
1673 -- Note [RedStack]
1674 -- The red_stack pair (n,insts) pair is just used for error reporting.
1675 -- 'n' is always the depth of the stack.
1676 -- The 'insts' is the stack of Insts being reduced: to produce X
1677 -- I had to produce Y, to produce Y I had to produce Z, and so on.
1678
1679
1680 mkRedEnv :: SDoc -> (Inst -> WhatToDo) -> [Inst] -> RedEnv
1681 mkRedEnv doc try_me givens
1682   = RedEnv { red_doc = doc, red_try_me = try_me,
1683              red_givens = givens, 
1684              red_reft = emptyRefinement,
1685              red_stack = (0,[]),
1686              red_improve = True }       
1687
1688 mkNoImproveRedEnv :: SDoc -> (Inst -> WhatToDo) -> RedEnv
1689 -- Do not do improvement; no givens
1690 mkNoImproveRedEnv doc try_me
1691   = RedEnv { red_doc = doc, red_try_me = try_me,
1692              red_givens = [], red_reft = emptyRefinement,
1693              red_stack = (0,[]),
1694              red_improve = True }       
1695
1696 data WhatToDo
1697  = ReduceMe WantSCs     -- Try to reduce this
1698                         -- If there's no instance, add the inst to the 
1699                         -- irreductible ones, but don't produce an error 
1700                         -- message of any kind.
1701                         -- It might be quite legitimate such as (Eq a)!
1702
1703  | Stop         -- Return as irreducible unless it can
1704                         -- be reduced to a constant in one step
1705                         -- Do not add superclasses; see 
1706
1707 data WantSCs = NoSCs | AddSCs   -- Tells whether we should add the superclasses
1708                                 -- of a predicate when adding it to the avails
1709         -- The reason for this flag is entirely the super-class loop problem
1710         -- Note [SUPER-CLASS LOOP 1]
1711
1712 zonkRedEnv :: RedEnv -> TcM RedEnv
1713 zonkRedEnv env 
1714   = do { givens' <- mappM zonkInst (red_givens env)
1715        ; return $ env {red_givens = givens'}
1716        }
1717 \end{code}
1718
1719
1720 %************************************************************************
1721 %*                                                                      *
1722 \subsection[reduce]{@reduce@}
1723 %*                                                                      *
1724 %************************************************************************
1725
1726 Note [Ancestor Equalities]
1727 ~~~~~~~~~~~~~~~~~~~~~~~~~~
1728 During context reduction, we add to the wanted equalities also those
1729 equalities that (transitively) occur in superclass contexts of wanted
1730 class constraints.  Consider the following code
1731
1732   class a ~ Int => C a
1733   instance C Int
1734
1735 If (C a) is wanted, we want to add (a ~ Int), which will be discharged by
1736 substituting Int for a.  Hence, we ultimately want (C Int), which we
1737 discharge with the explicit instance.
1738
1739 \begin{code}
1740 reduceContext :: RedEnv
1741               -> [Inst]                 -- Wanted
1742               -> TcM (ImprovementDone,
1743                       TcDictBinds,      -- Dictionary bindings
1744                       [Inst])           -- Irreducible
1745
1746 reduceContext env wanteds
1747   = do  { traceTc (text "reduceContext" <+> (vcat [
1748              text "----------------------",
1749              red_doc env,
1750              text "given" <+> ppr (red_givens env),
1751              text "wanted" <+> ppr wanteds,
1752              text "----------------------"
1753              ]))
1754
1755
1756         ; let givens                       = red_givens env
1757               (given_eqs0, given_dicts0)   = partition isEqInst givens
1758               (wanted_eqs0, wanted_non_eqs) = partition isEqInst wanteds
1759               (wanted_implics0, wanted_dicts0) = partition isImplicInst wanted_non_eqs
1760
1761           -- We want to add as wanted equalities those that (transitively) 
1762           -- occur in superclass contexts of wanted class constraints.
1763           -- See Note [Ancestor Equalities]
1764         ; ancestor_eqs <- ancestorEqualities wanted_dicts0
1765         ; let wanted_eqs = wanted_eqs0 ++ ancestor_eqs
1766         ; traceTc $ text "reduceContext: ancestor eqs" <+> ppr ancestor_eqs
1767
1768           -- 1. Normalise the *given* *equality* constraints
1769         ; (given_eqs, eliminate_skolems) <- normaliseGivenEqs given_eqs0
1770
1771           -- 2. Normalise the *given* *dictionary* constraints
1772           --    wrt. the toplevel and given equations
1773         ; (given_dicts, given_binds) <- normaliseGivenDicts given_eqs
1774                                                             given_dicts0
1775
1776           -- 5. Build the Avail mapping from "given_dicts"
1777           --    Add dicts refined by the current type refinement
1778         ; (init_state, extra_givens) <- getLIE $ do 
1779                 { init_state <- foldlM addGiven emptyAvails given_dicts
1780                 ; let reft = red_reft env
1781                 ; if isEmptyRefinement reft then return init_state
1782                   else foldlM (addRefinedGiven reft)
1783                                     init_state given_dicts }
1784
1785         -- *** ToDo: what to do with the "extra_givens"?  For the
1786         -- moment I'm simply discarding them, which is probably wrong
1787
1788           -- 7. Normalise the *wanted* *dictionary* constraints
1789           --    wrt. the toplevel and given equations
1790           -- NB: normalisation includes zonking as part of what it does
1791           --     so it's important to do it after any unifications
1792           --     that happened as a result of the addGivens
1793         ; (wanted_dicts,normalise_binds1) <- normaliseWantedDicts given_eqs wanted_dicts0
1794
1795           -- 6. Solve the *wanted* *dictionary* constraints (not implications)
1796           --    This may expose some further equational constraints...
1797         ; (avails, extra_eqs) <- getLIE (reduceList env wanted_dicts init_state)
1798         ; (dict_binds, bound_dicts, dict_irreds) <- extractResults avails wanted_dicts
1799         ; traceTc $ text "reduceContext extractresults" <+> vcat
1800                       [ppr avails,ppr wanted_dicts,ppr dict_binds]
1801
1802         -- *** ToDo: what to do with the "extra_eqs"?  For the
1803         -- moment I'm simply discarding them, which is probably wrong
1804
1805           -- Solve the wanted *implications*.  In doing so, we can provide
1806           -- as "given"   all the dicts that were originally given, 
1807           --              *or* for which we now have bindings, 
1808           --              *or* which are now irreds
1809         ; let implic_env = env { red_givens = givens ++ bound_dicts ++ dict_irreds }
1810         ; (implic_binds_s, implic_irreds_s) <- mapAndUnzipM (reduceImplication implic_env) wanted_implics0
1811         ; let implic_binds  = unionManyBags implic_binds_s
1812               implic_irreds = concat implic_irreds_s
1813
1814           -- 3. Solve the *wanted* *equation* constraints
1815         ; eq_irreds0 <- solveWantedEqs given_eqs wanted_eqs
1816
1817           -- 4. Normalise the *wanted* equality constraints with respect to
1818           --    each other 
1819         ; eq_irreds <- normaliseWantedEqs eq_irreds0
1820
1821           -- 8. Substitute the wanted *equations* in the wanted *dictionaries*
1822         ; let irreds = dict_irreds ++ implic_irreds
1823         ; (norm_irreds, normalise_binds2) <- substEqInDictInsts eq_irreds irreds
1824                 
1825           -- 9. eliminate the artificial skolem constants introduced in 1.
1826         ; eliminate_skolems     
1827
1828           -- Figure out whether we should go round again
1829           -- My current plan is to see if any of the mutable tyvars in
1830           -- givens or irreds has been filled in by improvement.  
1831           -- If so, there is merit in going around again, because
1832           -- we may make further progress
1833           -- 
1834           -- ToDo: is it only mutable stuff?  We may have exposed new
1835           --       equality constraints and should probably go round again
1836           --       then as well.  But currently we are dropping them on the
1837           --       floor anyway.
1838
1839         ; let all_irreds = norm_irreds ++ eq_irreds
1840         ; improved <- anyM isFilledMetaTyVar $ varSetElems $
1841                       tyVarsOfInsts (givens ++ all_irreds)
1842
1843         -- The old plan (fragile)
1844         -- improveed   = availsImproved avails 
1845         --               || (not $ isEmptyBag normalise_binds1)
1846         --               || (not $ isEmptyBag normalise_binds2)
1847         --               || (any isEqInst irreds)
1848
1849         ; traceTc (text "reduceContext end" <+> (vcat [
1850              text "----------------------",
1851              red_doc env,
1852              text "given" <+> ppr givens,
1853              text "given_eqs" <+> ppr given_eqs,
1854              text "wanted" <+> ppr wanteds,
1855              text "wanted_dicts" <+> ppr wanted_dicts,
1856              text "----",
1857              text "avails" <+> pprAvails avails,
1858              text "improved =" <+> ppr improved,
1859              text "(all) irreds = " <+> ppr all_irreds,
1860              text "dict-binds = " <+> ppr dict_binds,
1861              text "implic-binds = " <+> ppr implic_binds,
1862              text "----------------------"
1863              ]))
1864
1865         ; return (improved, 
1866                   given_binds `unionBags` normalise_binds1 
1867                               `unionBags` normalise_binds2 
1868                               `unionBags` dict_binds 
1869                               `unionBags` implic_binds, 
1870                   all_irreds) 
1871         }
1872
1873 tcImproveOne :: Avails -> Inst -> TcM ImprovementDone
1874 tcImproveOne avails inst
1875   | not (isDict inst) = return False
1876   | otherwise
1877   = do  { inst_envs <- tcGetInstEnvs
1878         ; let eqns = improveOne (classInstances inst_envs)
1879                                 (dictPred inst, pprInstArising inst)
1880                                 [ (dictPred p, pprInstArising p)
1881                                 | p <- availsInsts avails, isDict p ]
1882                 -- Avails has all the superclasses etc (good)
1883                 -- It also has all the intermediates of the deduction (good)
1884                 -- It does not have duplicates (good)
1885                 -- NB that (?x::t1) and (?x::t2) will be held separately in avails
1886                 --    so that improve will see them separate
1887         ; traceTc (text "improveOne" <+> ppr inst)
1888         ; unifyEqns eqns }
1889
1890 unifyEqns :: [(Equation,(PredType,SDoc),(PredType,SDoc))] 
1891           -> TcM ImprovementDone
1892 unifyEqns [] = return False
1893 unifyEqns eqns
1894   = do  { traceTc (ptext SLIT("Improve:") <+> vcat (map pprEquationDoc eqns))
1895         ; mappM_ unify eqns
1896         ; return True }
1897   where
1898     unify ((qtvs, pairs), what1, what2)
1899          = addErrCtxtM (mkEqnMsg what1 what2)   $
1900            tcInstTyVars (varSetElems qtvs)      `thenM` \ (_, _, tenv) ->
1901            mapM_ (unif_pr tenv) pairs
1902     unif_pr tenv (ty1,ty2) =  unifyType (substTy tenv ty1) (substTy tenv ty2)
1903
1904 pprEquationDoc (eqn, (p1,w1), (p2,w2)) = vcat [pprEquation eqn, nest 2 (ppr p1), nest 2 (ppr p2)]
1905
1906 mkEqnMsg (pred1,from1) (pred2,from2) tidy_env
1907   = do  { pred1' <- zonkTcPredType pred1; pred2' <- zonkTcPredType pred2
1908         ; let { pred1'' = tidyPred tidy_env pred1'; pred2'' = tidyPred tidy_env pred2' }
1909         ; let msg = vcat [ptext SLIT("When using functional dependencies to combine"),
1910                           nest 2 (sep [ppr pred1'' <> comma, nest 2 from1]), 
1911                           nest 2 (sep [ppr pred2'' <> comma, nest 2 from2])]
1912         ; return (tidy_env, msg) }
1913 \end{code}
1914
1915 The main context-reduction function is @reduce@.  Here's its game plan.
1916
1917 \begin{code}
1918 reduceList :: RedEnv -> [Inst] -> Avails -> TcM Avails
1919 reduceList env@(RedEnv {red_stack = (n,stk)}) wanteds state
1920   = do  { traceTc (text "reduceList " <+> (ppr wanteds $$ ppr state))
1921         ; dopts <- getDOpts
1922 #ifdef DEBUG
1923         ; if n > 8 then
1924                 dumpTcRn (hang (ptext SLIT("Interesting! Context reduction stack depth") <+> int n) 
1925                              2 (ifPprDebug (nest 2 (pprStack stk))))
1926           else return ()
1927 #endif
1928         ; if n >= ctxtStkDepth dopts then
1929             failWithTc (reduceDepthErr n stk)
1930           else
1931             go wanteds state }
1932   where
1933     go []     state = return state
1934     go (w:ws) state = do { state' <- reduce (env {red_stack = (n+1, w:stk)}) w state
1935                          ; go ws state' }
1936
1937     -- Base case: we're done!
1938 reduce env wanted avails
1939     -- It's the same as an existing inst, or a superclass thereof
1940   | Just avail <- findAvail avails wanted
1941   = do { traceTc (text "reduce: found " <+> ppr wanted)
1942        ; returnM avails 
1943        }
1944
1945   | otherwise
1946   = do  { traceTc (text "reduce" <+> ppr wanted $$ ppr avails)
1947         ; case red_try_me env wanted of {
1948             Stop -> try_simple (addIrred NoSCs);
1949                         -- See Note [No superclasses for Stop]
1950
1951             ReduceMe want_scs -> do     -- It should be reduced
1952                 { (avails, lookup_result) <- reduceInst env avails wanted
1953                 ; case lookup_result of
1954                     NoInstance -> addIrred want_scs avails wanted
1955                              -- Add it and its superclasses
1956                              
1957                     GenInst [] rhs -> addWanted want_scs avails wanted rhs []
1958
1959                     GenInst wanteds' rhs
1960                           -> do { avails1 <- addIrred NoSCs avails wanted
1961                                 ; avails2 <- reduceList env wanteds' avails1
1962                                 ; addWanted want_scs avails2 wanted rhs wanteds' } }
1963                 -- Temporarily do addIrred *before* the reduceList, 
1964                 -- which has the effect of adding the thing we are trying
1965                 -- to prove to the database before trying to prove the things it
1966                 -- needs.  See note [RECURSIVE DICTIONARIES]
1967                 -- NB: we must not do an addWanted before, because that adds the
1968                 --     superclasses too, and that can lead to a spurious loop; see
1969                 --     the examples in [SUPERCLASS-LOOP]
1970                 -- So we do an addIrred before, and then overwrite it afterwards with addWanted
1971     } }
1972   where
1973         -- First, see if the inst can be reduced to a constant in one step
1974         -- Works well for literals (1::Int) and constant dictionaries (d::Num Int)
1975         -- Don't bother for implication constraints, which take real work
1976     try_simple do_this_otherwise
1977       = do { res <- lookupSimpleInst wanted
1978            ; case res of
1979                 GenInst [] rhs -> addWanted AddSCs avails wanted rhs []
1980                 other          -> do_this_otherwise avails wanted }
1981 \end{code}
1982
1983
1984 Note [SUPERCLASS-LOOP 2]
1985 ~~~~~~~~~~~~~~~~~~~~~~~~
1986 But the above isn't enough.  Suppose we are *given* d1:Ord a,
1987 and want to deduce (d2:C [a]) where
1988
1989         class Ord a => C a where
1990         instance Ord [a] => C [a] where ...
1991
1992 Then we'll use the instance decl to deduce C [a] from Ord [a], and then add the
1993 superclasses of C [a] to avails.  But we must not overwrite the binding
1994 for Ord [a] (which is obtained from Ord a) with a superclass selection or we'll just
1995 build a loop! 
1996
1997 Here's another variant, immortalised in tcrun020
1998         class Monad m => C1 m
1999         class C1 m => C2 m x
2000         instance C2 Maybe Bool
2001 For the instance decl we need to build (C1 Maybe), and it's no good if
2002 we run around and add (C2 Maybe Bool) and its superclasses to the avails 
2003 before we search for C1 Maybe.
2004
2005 Here's another example 
2006         class Eq b => Foo a b
2007         instance Eq a => Foo [a] a
2008 If we are reducing
2009         (Foo [t] t)
2010
2011 we'll first deduce that it holds (via the instance decl).  We must not
2012 then overwrite the Eq t constraint with a superclass selection!
2013
2014 At first I had a gross hack, whereby I simply did not add superclass constraints
2015 in addWanted, though I did for addGiven and addIrred.  This was sub-optimal,
2016 becuase it lost legitimate superclass sharing, and it still didn't do the job:
2017 I found a very obscure program (now tcrun021) in which improvement meant the
2018 simplifier got two bites a the cherry... so something seemed to be an Stop
2019 first time, but reducible next time.
2020
2021 Now we implement the Right Solution, which is to check for loops directly 
2022 when adding superclasses.  It's a bit like the occurs check in unification.
2023
2024
2025 Note [RECURSIVE DICTIONARIES]
2026 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2027 Consider 
2028     data D r = ZeroD | SuccD (r (D r));
2029     
2030     instance (Eq (r (D r))) => Eq (D r) where
2031         ZeroD     == ZeroD     = True
2032         (SuccD a) == (SuccD b) = a == b
2033         _         == _         = False;
2034     
2035     equalDC :: D [] -> D [] -> Bool;
2036     equalDC = (==);
2037
2038 We need to prove (Eq (D [])).  Here's how we go:
2039
2040         d1 : Eq (D [])
2041
2042 by instance decl, holds if
2043         d2 : Eq [D []]
2044         where   d1 = dfEqD d2
2045
2046 by instance decl of Eq, holds if
2047         d3 : D []
2048         where   d2 = dfEqList d3
2049                 d1 = dfEqD d2
2050
2051 But now we can "tie the knot" to give
2052
2053         d3 = d1
2054         d2 = dfEqList d3
2055         d1 = dfEqD d2
2056
2057 and it'll even run!  The trick is to put the thing we are trying to prove
2058 (in this case Eq (D []) into the database before trying to prove its
2059 contributing clauses.
2060         
2061
2062 %************************************************************************
2063 %*                                                                      *
2064                 Reducing a single constraint
2065 %*                                                                      *
2066 %************************************************************************
2067
2068 \begin{code}
2069 ---------------------------------------------
2070 reduceInst :: RedEnv -> Avails -> Inst -> TcM (Avails, LookupInstResult)
2071 reduceInst env avails other_inst
2072   = do  { result <- lookupSimpleInst other_inst
2073         ; return (avails, result) }
2074 \end{code}
2075
2076 Note [Equational Constraints in Implication Constraints]
2077 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2078
2079 An implication constraint is of the form 
2080         Given => Wanted 
2081 where Given and Wanted may contain both equational and dictionary
2082 constraints. The delay and reduction of these two kinds of constraints
2083 is distinct:
2084
2085 -) In the generated code, wanted Dictionary constraints are wrapped up in an
2086    implication constraint that is created at the code site where the wanted
2087    dictionaries can be reduced via a let-binding. This let-bound implication
2088    constraint is deconstructed at the use-site of the wanted dictionaries.
2089
2090 -) While the reduction of equational constraints is also delayed, the delay
2091    is not manifest in the generated code. The required evidence is generated
2092    in the code directly at the use-site. There is no let-binding and deconstruction
2093    necessary. The main disadvantage is that we cannot exploit sharing as the
2094    same evidence may be generated at multiple use-sites. However, this disadvantage
2095    is limited because it only concerns coercions which are erased.
2096
2097 The different treatment is motivated by the different in representation. Dictionary
2098 constraints require manifest runtime dictionaries, while equations require coercions
2099 which are types.
2100
2101 \begin{code}
2102 ---------------------------------------------
2103 reduceImplication :: RedEnv
2104                   -> Inst
2105                   -> TcM (TcDictBinds, [Inst])
2106 \end{code}
2107
2108 Suppose we are simplifying the constraint
2109         forall bs. extras => wanted
2110 in the context of an overall simplification problem with givens 'givens',
2111 and refinment 'reft'.
2112
2113 Note that
2114   * The refinement is often empty
2115
2116   * The 'extra givens' need not mention any of the quantified type variables
2117         e.g.    forall {}. Eq a => Eq [a]
2118                 forall {}. C Int => D (Tree Int)
2119
2120     This happens when you have something like
2121         data T a where
2122           T1 :: Eq a => a -> T a
2123
2124         f :: T a -> Int
2125         f x = ...(case x of { T1 v -> v==v })...
2126
2127 \begin{code}
2128         -- ToDo: should we instantiate tvs?  I think it's not necessary
2129         --
2130         -- Note on coercion variables:
2131         --
2132         --      The extra given coercion variables are bound at two different sites:
2133         --      -) in the creation context of the implication constraint        
2134         --              the solved equational constraints use these binders
2135         --
2136         --      -) at the solving site of the implication constraint
2137         --              the solved dictionaries use these binders               
2138         --              these binders are generated by reduceImplication
2139         --
2140 reduceImplication env
2141         orig_implic@(ImplicInst { tci_name = name, tci_loc = inst_loc,
2142                                   tci_tyvars = tvs, tci_reft = reft,
2143                                   tci_given = extra_givens, tci_wanted = wanteds })
2144   = do  {       -- Add refined givens, and the extra givens
2145                 -- Todo fix this 
2146 --        (refined_red_givens,refined_avails)
2147 --              <- if isEmptyRefinement reft then return (red_givens env,orig_avails)
2148 --                 else foldlM (addRefinedGiven reft) ([],orig_avails) (red_givens env)
2149 --      Commented out SLPJ Sept 07; see comment with extractLocalResults below
2150           let refined_red_givens = []
2151
2152                 -- Solve the sub-problem
2153         ; let try_me inst = ReduceMe AddSCs     -- Note [Freeness and implications]
2154               env' = env { red_givens = extra_givens ++ red_givens env
2155                          , red_reft = reft
2156                          , red_doc = sep [ptext SLIT("reduceImplication for") <+> ppr name,
2157                                           nest 2 (parens $ ptext SLIT("within") <+> red_doc env)]
2158                          , red_try_me = try_me }
2159
2160         ; traceTc (text "reduceImplication" <+> vcat
2161                         [ ppr (red_givens env), ppr extra_givens, 
2162                           ppr reft, ppr wanteds])
2163         ; (irreds, binds) <- checkLoop env' wanteds
2164         ; let   (extra_eq_givens, extra_dict_givens) = partition isEqInst extra_givens
2165                         -- SLPJ Sept 07: I think this is bogus; currently
2166                         -- there are no Eqinsts in extra_givens
2167                 dict_ids = map instToId extra_dict_givens 
2168
2169                 -- Note [Reducing implication constraints]
2170                 -- Tom -- update note, put somewhere!
2171
2172         ; traceTc (text "reduceImplication result" <+> vcat
2173                         [ppr irreds, ppr binds])
2174
2175         ; -- extract superclass binds
2176           --  (sc_binds,_) <- extractResults avails []
2177 --      ; traceTc (text "reduceImplication sc_binds" <+> vcat
2178 --                      [ppr sc_binds, ppr avails])
2179 --  
2180
2181         -- SLPJ Sept 07: what if improvement happened inside the checkLoop?
2182         -- Then we must iterate the outer loop too!
2183
2184         ; traceTc (text "reduceImplication condition" <+> ppr ((isEmptyLHsBinds binds) || (null irreds)))
2185
2186 --      Progress is no longer measered by the number of bindings
2187         ; if (isEmptyLHsBinds binds) && (not $ null irreds) then        -- No progress
2188                 -- If there are any irreds, we back off and do nothing
2189                 return (emptyBag, [orig_implic])
2190           else do
2191         { (simpler_implic_insts, bind) <- makeImplicationBind inst_loc tvs reft extra_givens irreds
2192                         -- This binding is useless if the recursive simplification
2193                         -- made no progress; but currently we don't try to optimise that
2194                         -- case.  After all, we only try hard to reduce at top level, or
2195                         -- when inferring types.
2196
2197         ; let   dict_wanteds = filter (not . isEqInst) wanteds
2198                 -- TOMDO: given equational constraints bug!
2199                 --  we need a different evidence for given
2200                 --  equations depending on whether we solve
2201                 --  dictionary constraints or equational constraints
2202
2203                 eq_tyvars = varSetElems $ tyVarsOfTypes $ map eqInstType extra_eq_givens
2204                         -- SLPJ Sept07: this looks Utterly Wrong to me, but I think
2205                         --              that current extra_givens has no EqInsts, so
2206                         --              it makes no difference
2207                 co  = wrap_inline       -- Note [Always inline implication constraints]
2208                       <.> mkWpTyLams tvs
2209                       <.> mkWpTyLams eq_tyvars
2210                       <.> mkWpLams dict_ids
2211                       <.> WpLet (binds `unionBags` bind)
2212                 wrap_inline | null dict_ids = idHsWrapper
2213                             | otherwise     = WpInline
2214                 rhs = mkHsWrap co payload
2215                 loc = instLocSpan inst_loc
2216                 payload | [dict_wanted] <- dict_wanteds = HsVar (instToId dict_wanted)
2217                         | otherwise = ExplicitTuple (map (L loc . HsVar . instToId) dict_wanteds) Boxed
2218
2219         
2220         ; traceTc (vcat [text "reduceImplication" <+> ppr name,
2221                          ppr simpler_implic_insts,
2222                          text "->" <+> ppr rhs])
2223         ; return (unitBag (L loc (VarBind (instToId orig_implic) (L loc rhs))),
2224                   simpler_implic_insts)
2225         } 
2226     }
2227 \end{code}
2228
2229 Note [Always inline implication constraints]
2230 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2231 Suppose an implication constraint floats out of an INLINE function.
2232 Then although the implication has a single call site, it won't be 
2233 inlined.  And that is bad because it means that even if there is really
2234 *no* overloading (type signatures specify the exact types) there will
2235 still be dictionary passing in the resulting code.  To avert this,
2236 we mark the implication constraints themselves as INLINE, at least when
2237 there is no loss of sharing as a result.
2238
2239 Note [Freeness and implications]
2240 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2241 It's hard to say when an implication constraint can be floated out.  Consider
2242         forall {} Eq a => Foo [a]
2243 The (Foo [a]) doesn't mention any of the quantified variables, but it
2244 still might be partially satisfied by the (Eq a). 
2245
2246 There is a useful special case when it *is* easy to partition the 
2247 constraints, namely when there are no 'givens'.  Consider
2248         forall {a}. () => Bar b
2249 There are no 'givens', and so there is no reason to capture (Bar b).
2250 We can let it float out.  But if there is even one constraint we
2251 must be much more careful:
2252         forall {a}. C a b => Bar (m b)
2253 because (C a b) might have a superclass (D b), from which we might 
2254 deduce (Bar [b]) when m later gets instantiated to [].  Ha!
2255
2256 Here is an even more exotic example
2257         class C a => D a b
2258 Now consider the constraint
2259         forall b. D Int b => C Int
2260 We can satisfy the (C Int) from the superclass of D, so we don't want
2261 to float the (C Int) out, even though it mentions no type variable in
2262 the constraints!
2263
2264 Note [Pruning the givens in an implication constraint]
2265 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2266 Suppose we are about to form the implication constraint
2267         forall tvs.  Eq a => Ord b
2268 The (Eq a) cannot contribute to the (Ord b), because it has no access to
2269 the type variable 'b'.  So we could filter out the (Eq a) from the givens.
2270
2271 Doing so would be a bit tidier, but all the implication constraints get
2272 simplified away by the optimiser, so it's no great win.   So I don't take
2273 advantage of that at the moment.
2274
2275 If you do, BE CAREFUL of wobbly type variables.
2276
2277
2278 %************************************************************************
2279 %*                                                                      *
2280                 Avails and AvailHow: the pool of evidence
2281 %*                                                                      *
2282 %************************************************************************
2283
2284
2285 \begin{code}
2286 data Avails = Avails !ImprovementDone !AvailEnv
2287
2288 type ImprovementDone = Bool     -- True <=> some unification has happened
2289                                 -- so some Irreds might now be reducible
2290                                 -- keys that are now 
2291
2292 type AvailEnv = FiniteMap Inst AvailHow
2293 data AvailHow
2294   = IsIrred             -- Used for irreducible dictionaries,
2295                         -- which are going to be lambda bound
2296
2297   | Given Inst          -- Used for dictionaries for which we have a binding
2298                         -- e.g. those "given" in a signature
2299
2300   | Rhs                 -- Used when there is a RHS
2301         (LHsExpr TcId)  -- The RHS
2302         [Inst]          -- Insts free in the RHS; we need these too
2303
2304 instance Outputable Avails where
2305   ppr = pprAvails
2306
2307 pprAvails (Avails imp avails)
2308   = vcat [ ptext SLIT("Avails") <> (if imp then ptext SLIT("[improved]") else empty)
2309          , nest 2 $ braces $ 
2310            vcat [ sep [ppr inst, nest 2 (equals <+> ppr avail)]
2311                 | (inst,avail) <- fmToList avails ]]
2312
2313 instance Outputable AvailHow where
2314     ppr = pprAvail
2315
2316 -------------------------
2317 pprAvail :: AvailHow -> SDoc
2318 pprAvail IsIrred        = text "Irred"
2319 pprAvail (Given x)      = text "Given" <+> ppr x
2320 pprAvail (Rhs rhs bs)   = sep [text "Rhs" <+> ppr bs,
2321                                nest 2 (ppr rhs)]
2322
2323 -------------------------
2324 extendAvailEnv :: AvailEnv -> Inst -> AvailHow -> AvailEnv
2325 extendAvailEnv env inst avail = addToFM env inst avail
2326
2327 findAvailEnv :: AvailEnv -> Inst -> Maybe AvailHow
2328 findAvailEnv env wanted = lookupFM env wanted
2329         -- NB 1: the Ord instance of Inst compares by the class/type info
2330         --  *not* by unique.  So
2331         --      d1::C Int ==  d2::C Int
2332
2333 emptyAvails :: Avails
2334 emptyAvails = Avails False emptyFM
2335
2336 findAvail :: Avails -> Inst -> Maybe AvailHow
2337 findAvail (Avails _ avails) wanted = findAvailEnv avails wanted
2338
2339 elemAvails :: Inst -> Avails -> Bool
2340 elemAvails wanted (Avails _ avails) = wanted `elemFM` avails
2341
2342 extendAvails :: Avails -> Inst -> AvailHow -> TcM Avails
2343 -- Does improvement
2344 extendAvails avails@(Avails imp env) inst avail 
2345   = do  { imp1 <- tcImproveOne avails inst      -- Do any improvement
2346         ; return (Avails (imp || imp1) (extendAvailEnv env inst avail)) }
2347
2348 availsInsts :: Avails -> [Inst]
2349 availsInsts (Avails _ avails) = keysFM avails
2350
2351 availsImproved (Avails imp _) = imp
2352
2353 updateImprovement :: Avails -> Avails -> Avails
2354 -- (updateImprovement a1 a2) sets a1's improvement flag from a2
2355 updateImprovement (Avails _ avails1) (Avails imp2 _) = Avails imp2 avails1
2356 \end{code}
2357
2358 Extracting the bindings from a bunch of Avails.
2359 The bindings do *not* come back sorted in dependency order.
2360 We assume that they'll be wrapped in a big Rec, so that the
2361 dependency analyser can sort them out later
2362
2363 \begin{code}
2364 type DoneEnv = FiniteMap Inst [Id]
2365 -- Tracks which things we have evidence for
2366
2367 extractResults :: Avails
2368                -> [Inst]                -- Wanted
2369                -> TcM (TcDictBinds,     -- Bindings
2370                        [Inst],          -- The insts bound by the bindings
2371                        [Inst])          -- Irreducible ones
2372                         -- Note [Reducing implication constraints]
2373
2374 extractResults (Avails _ avails) wanteds
2375   = go emptyBag [] [] emptyFM wanteds
2376   where
2377     go  :: TcDictBinds  -- Bindings for dicts
2378         -> [Inst]       -- Bound by the bindings
2379         -> [Inst]       -- Irreds
2380         -> DoneEnv      -- Has an entry for each inst in the above three sets
2381         -> [Inst]       -- Wanted
2382         -> TcM (TcDictBinds, [Inst], [Inst])
2383     go binds bound_dicts irreds done [] 
2384       = return (binds, bound_dicts, irreds)
2385
2386     go binds bound_dicts irreds done (w:ws)
2387       | Just done_ids@(done_id : rest_done_ids) <- lookupFM done w
2388       = if w_id `elem` done_ids then
2389            go binds bound_dicts irreds done ws
2390         else
2391            go (add_bind (nlHsVar done_id)) bound_dicts irreds
2392               (addToFM done w (done_id : w_id : rest_done_ids)) ws
2393
2394       | otherwise       -- Not yet done
2395       = case findAvailEnv avails w of
2396           Nothing -> pprTrace "Urk: extractResults" (ppr w) $
2397                      go binds bound_dicts irreds done ws
2398
2399           Just IsIrred -> go binds bound_dicts (w:irreds) done' ws
2400
2401           Just (Rhs rhs ws') -> go (add_bind rhs) (w:bound_dicts) irreds done' (ws' ++ ws)
2402
2403           Just (Given g) -> go binds' bound_dicts irreds (addToFM done w [g_id]) ws 
2404                 where
2405                   g_id = instToId g
2406                   binds' | w_id == g_id = binds
2407                          | otherwise    = add_bind (nlHsVar g_id)
2408       where
2409         w_id  = instToId w      
2410         done' = addToFM done w [w_id]
2411         add_bind rhs = addInstToDictBind binds w rhs
2412 \end{code}
2413
2414
2415 Note [No superclasses for Stop]
2416 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2417 When we decide not to reduce an Inst -- the 'WhatToDo' --- we still
2418 add it to avails, so that any other equal Insts will be commoned up
2419 right here.  However, we do *not* add superclasses.  If we have
2420         df::Floating a
2421         dn::Num a
2422 but a is not bound here, then we *don't* want to derive dn from df
2423 here lest we lose sharing.
2424
2425 \begin{code}
2426 addWanted :: WantSCs -> Avails -> Inst -> LHsExpr TcId -> [Inst] -> TcM Avails
2427 addWanted want_scs avails wanted rhs_expr wanteds
2428   = addAvailAndSCs want_scs avails wanted avail
2429   where
2430     avail = Rhs rhs_expr wanteds
2431
2432 addGiven :: Avails -> Inst -> TcM Avails
2433 addGiven avails given = addAvailAndSCs AddSCs avails given (Given given)
2434         -- Always add superclasses for 'givens'
2435         --
2436         -- No ASSERT( not (given `elemAvails` avails) ) because in an instance
2437         -- decl for Ord t we can add both Ord t and Eq t as 'givens', 
2438         -- so the assert isn't true
2439
2440 addRefinedGiven :: Refinement -> Avails -> Inst -> TcM Avails
2441 addRefinedGiven reft avails given
2442   | isDict given        -- We sometimes have 'given' methods, but they
2443                         -- are always optional, so we can drop them
2444   , let pred = dictPred given
2445   , isRefineablePred pred       -- See Note [ImplicInst rigidity]
2446   , Just (co, pred) <- refinePred reft pred
2447   = do  { new_given <- newDictBndr (instLoc given) pred
2448         ; let rhs = L (instSpan given) $
2449                     HsWrap (WpCo co) (HsVar (instToId given))
2450         ; addAvailAndSCs AddSCs avails new_given (Rhs rhs [given]) }
2451             -- ToDo: the superclasses of the original given all exist in Avails 
2452             -- so we could really just cast them, but it's more awkward to do,
2453             -- and hopefully the optimiser will spot the duplicated work
2454   | otherwise
2455   = return avails
2456 \end{code}
2457
2458 Note [ImplicInst rigidity]
2459 ~~~~~~~~~~~~~~~~~~~~~~~~~~
2460 Consider
2461         C :: forall ab. (Eq a, Ord b) => b -> T a
2462         
2463         ...(case x of C v -> <body>)...
2464
2465 From the case (where x::T ty) we'll get an implication constraint
2466         forall b. (Eq ty, Ord b) => <body-constraints>
2467 Now suppose <body-constraints> itself has an implication constraint 
2468 of form
2469         forall c. <reft> => <payload>
2470 Then, we can certainly apply the refinement <reft> to the Ord b, becuase it is
2471 existential, but we probably should not apply it to the (Eq ty) because it may
2472 be wobbly. Hence the isRigidInst
2473
2474 @Insts@ are ordered by their class/type info, rather than by their
2475 unique.  This allows the context-reduction mechanism to use standard finite
2476 maps to do their stuff.  It's horrible that this code is here, rather
2477 than with the Avails handling stuff in TcSimplify
2478
2479 \begin{code}
2480 addIrred :: WantSCs -> Avails -> Inst -> TcM Avails
2481 addIrred want_scs avails irred = ASSERT2( not (irred `elemAvails` avails), ppr irred $$ ppr avails )
2482                                  addAvailAndSCs want_scs avails irred IsIrred
2483
2484 addAvailAndSCs :: WantSCs -> Avails -> Inst -> AvailHow -> TcM Avails
2485 addAvailAndSCs want_scs avails inst avail
2486   | not (isClassDict inst) = extendAvails avails inst avail
2487   | NoSCs <- want_scs      = extendAvails avails inst avail
2488   | otherwise              = do { traceTc (text "addAvailAndSCs" <+> vcat [ppr inst, ppr deps])
2489                                 ; avails' <- extendAvails avails inst avail
2490                                 ; addSCs is_loop avails' inst }
2491   where
2492     is_loop pred = any (`tcEqType` mkPredTy pred) dep_tys
2493                         -- Note: this compares by *type*, not by Unique
2494     deps         = findAllDeps (unitVarSet (instToVar inst)) avail
2495     dep_tys      = map idType (varSetElems deps)
2496
2497     findAllDeps :: IdSet -> AvailHow -> IdSet
2498     -- Find all the Insts that this one depends on
2499     -- See Note [SUPERCLASS-LOOP 2]
2500     -- Watch out, though.  Since the avails may contain loops 
2501     -- (see Note [RECURSIVE DICTIONARIES]), so we need to track the ones we've seen so far
2502     findAllDeps so_far (Rhs _ kids) = foldl find_all so_far kids
2503     findAllDeps so_far other        = so_far
2504
2505     find_all :: IdSet -> Inst -> IdSet
2506     find_all so_far kid
2507       | isEqInst kid                       = so_far
2508       | kid_id `elemVarSet` so_far         = so_far
2509       | Just avail <- findAvail avails kid = findAllDeps so_far' avail
2510       | otherwise                          = so_far'
2511       where
2512         so_far' = extendVarSet so_far kid_id    -- Add the new kid to so_far
2513         kid_id = instToId kid
2514
2515 addSCs :: (TcPredType -> Bool) -> Avails -> Inst -> TcM Avails
2516         -- Add all the superclasses of the Inst to Avails
2517         -- The first param says "don't do this because the original thing
2518         --      depends on this one, so you'd build a loop"
2519         -- Invariant: the Inst is already in Avails.
2520
2521 addSCs is_loop avails dict
2522   = ASSERT( isDict dict )
2523     do  { sc_dicts <- newDictBndrs (instLoc dict) sc_theta'
2524         ; foldlM add_sc avails (zipEqual "add_scs" sc_dicts sc_sels) }
2525   where
2526     (clas, tys) = getDictClassTys dict
2527     (tyvars, sc_theta, sc_sels, _) = classBigSig clas
2528     sc_theta' = filter (not . isEqPred) $
2529                   substTheta (zipTopTvSubst tyvars tys) sc_theta
2530
2531     add_sc avails (sc_dict, sc_sel)
2532       | is_loop (dictPred sc_dict) = return avails      -- See Note [SUPERCLASS-LOOP 2]
2533       | is_given sc_dict           = return avails
2534       | otherwise                  = do { avails' <- extendAvails avails sc_dict (Rhs sc_sel_rhs [dict])
2535                                         ; addSCs is_loop avails' sc_dict }
2536       where
2537         sc_sel_rhs = L (instSpan dict) (HsWrap co_fn (HsVar sc_sel))
2538         co_fn      = WpApp (instToVar dict) <.> mkWpTyApps tys
2539
2540     is_given :: Inst -> Bool
2541     is_given sc_dict = case findAvail avails sc_dict of
2542                           Just (Given _) -> True        -- Given is cheaper than superclass selection
2543                           other          -> False       
2544
2545 -- From the a set of insts obtain all equalities that (transitively) occur in
2546 -- superclass contexts of class constraints (aka the ancestor equalities). 
2547 --
2548 ancestorEqualities :: [Inst] -> TcM [Inst]
2549 ancestorEqualities
2550   =   mapM mkWantedEqInst               -- turn only equality predicates..
2551     . filter isEqPred                   -- ..into wanted equality insts
2552     . bagToList 
2553     . addAEsToBag emptyBag              -- collect the superclass constraints..
2554     . map dictPred                      -- ..of all predicates in a bag
2555     . filter isClassDict
2556   where
2557     addAEsToBag :: Bag PredType -> [PredType] -> Bag PredType
2558     addAEsToBag bag []           = bag
2559     addAEsToBag bag (pred:preds)
2560       | pred `elemBag` bag = addAEsToBag bag         preds
2561       | isEqPred pred      = addAEsToBag bagWithPred preds
2562       | isClassPred pred   = addAEsToBag bagWithPred predsWithSCs
2563       | otherwise          = addAEsToBag bag         preds
2564       where
2565         bagWithPred  = bag `snocBag` pred
2566         predsWithSCs = preds ++ substTheta (zipTopTvSubst tyvars tys) sc_theta
2567         --
2568         (tyvars, sc_theta, _, _) = classBigSig clas
2569         (clas, tys)              = getClassPredTys pred 
2570 \end{code}
2571
2572
2573 %************************************************************************
2574 %*                                                                      *
2575 \section{tcSimplifyTop: defaulting}
2576 %*                                                                      *
2577 %************************************************************************
2578
2579
2580 @tcSimplifyTop@ is called once per module to simplify all the constant
2581 and ambiguous Insts.
2582
2583 We need to be careful of one case.  Suppose we have
2584
2585         instance Num a => Num (Foo a b) where ...
2586
2587 and @tcSimplifyTop@ is given a constraint (Num (Foo x y)).  Then it'll simplify
2588 to (Num x), and default x to Int.  But what about y??
2589
2590 It's OK: the final zonking stage should zap y to (), which is fine.
2591
2592
2593 \begin{code}
2594 tcSimplifyTop, tcSimplifyInteractive :: [Inst] -> TcM TcDictBinds
2595 tcSimplifyTop wanteds
2596   = tc_simplify_top doc False wanteds
2597   where 
2598     doc = text "tcSimplifyTop"
2599
2600 tcSimplifyInteractive wanteds
2601   = tc_simplify_top doc True wanteds
2602   where 
2603     doc = text "tcSimplifyInteractive"
2604
2605 -- The TcLclEnv should be valid here, solely to improve
2606 -- error message generation for the monomorphism restriction
2607 tc_simplify_top doc interactive wanteds
2608   = do  { dflags <- getDOpts
2609         ; wanteds <- zonkInsts wanteds
2610         ; mapM_ zonkTopTyVar (varSetElems (tyVarsOfInsts wanteds))
2611
2612         ; traceTc (text "tc_simplify_top 0: " <+> ppr wanteds)
2613         ; (irreds1, binds1) <- tryHardCheckLoop doc1 wanteds
2614 --      ; (irreds1, binds1) <- gentleInferLoop doc1 wanteds
2615         ; traceTc (text "tc_simplify_top 1: " <+> ppr irreds1)
2616         ; (irreds2, binds2) <- approximateImplications doc2 (\d -> True) irreds1
2617         ; traceTc (text "tc_simplify_top 2: " <+> ppr irreds2)
2618
2619                 -- Use the defaulting rules to do extra unification
2620                 -- NB: irreds2 are already zonked
2621         ; (irreds3, binds3) <- disambiguate doc3 interactive dflags irreds2
2622
2623                 -- Deal with implicit parameters
2624         ; let (bad_ips, non_ips) = partition isIPDict irreds3
2625               (ambigs, others)   = partition isTyVarDict non_ips
2626
2627         ; topIPErrs bad_ips     -- Can arise from   f :: Int -> Int
2628                                 --                  f x = x + ?y
2629         ; addNoInstanceErrs others
2630         ; addTopAmbigErrs ambigs        
2631
2632         ; return (binds1 `unionBags` binds2 `unionBags` binds3) }
2633   where
2634     doc1 = doc <+> ptext SLIT("(first round)")
2635     doc2 = doc <+> ptext SLIT("(approximate)")
2636     doc3 = doc <+> ptext SLIT("(disambiguate)")
2637 \end{code}
2638
2639 If a dictionary constrains a type variable which is
2640         * not mentioned in the environment
2641         * and not mentioned in the type of the expression
2642 then it is ambiguous. No further information will arise to instantiate
2643 the type variable; nor will it be generalised and turned into an extra
2644 parameter to a function.
2645
2646 It is an error for this to occur, except that Haskell provided for
2647 certain rules to be applied in the special case of numeric types.
2648 Specifically, if
2649         * at least one of its classes is a numeric class, and
2650         * all of its classes are numeric or standard
2651 then the type variable can be defaulted to the first type in the
2652 default-type list which is an instance of all the offending classes.
2653
2654 So here is the function which does the work.  It takes the ambiguous
2655 dictionaries and either resolves them (producing bindings) or
2656 complains.  It works by splitting the dictionary list by type
2657 variable, and using @disambigOne@ to do the real business.
2658
2659 @disambigOne@ assumes that its arguments dictionaries constrain all
2660 the same type variable.
2661
2662 ADR Comment 20/6/94: I've changed the @CReturnable@ case to default to
2663 @()@ instead of @Int@.  I reckon this is the Right Thing to do since
2664 the most common use of defaulting is code like:
2665 \begin{verbatim}
2666         _ccall_ foo     `seqPrimIO` bar
2667 \end{verbatim}
2668 Since we're not using the result of @foo@, the result if (presumably)
2669 @void@.
2670
2671 \begin{code}
2672 disambiguate :: SDoc -> Bool -> DynFlags -> [Inst] -> TcM ([Inst], TcDictBinds)
2673         -- Just does unification to fix the default types
2674         -- The Insts are assumed to be pre-zonked
2675 disambiguate doc interactive dflags insts
2676   | null insts
2677   = return (insts, emptyBag)
2678
2679   | null defaultable_groups
2680   = do  { traceTc (text "disambigutate, no defaultable groups" <+> vcat [ppr unaries, ppr insts, ppr bad_tvs, ppr defaultable_groups])
2681         ; return (insts, emptyBag) }
2682
2683   | otherwise
2684   = do  {       -- Figure out what default types to use
2685           default_tys <- getDefaultTys extended_defaulting ovl_strings
2686
2687         ; traceTc (text "disambiguate1" <+> vcat [ppr insts, ppr unaries, ppr bad_tvs, ppr defaultable_groups])
2688         ; mapM_ (disambigGroup default_tys) defaultable_groups
2689
2690         -- disambigGroup does unification, hence try again
2691         ; tryHardCheckLoop doc insts }
2692
2693   where
2694    extended_defaulting = interactive || dopt Opt_ExtendedDefaultRules dflags
2695    ovl_strings = dopt Opt_OverloadedStrings dflags
2696
2697    unaries :: [(Inst, Class, TcTyVar)]  -- (C tv) constraints
2698    bad_tvs :: TcTyVarSet  -- Tyvars mentioned by *other* constraints
2699    (unaries, bad_tvs_s) = partitionWith find_unary insts 
2700    bad_tvs              = unionVarSets bad_tvs_s
2701
2702         -- Finds unary type-class constraints
2703    find_unary d@(Dict {tci_pred = ClassP cls [ty]})
2704         | Just tv <- tcGetTyVar_maybe ty = Left (d,cls,tv)
2705    find_unary inst                       = Right (tyVarsOfInst inst)
2706
2707                 -- Group by type variable
2708    defaultable_groups :: [[(Inst,Class,TcTyVar)]]
2709    defaultable_groups = filter defaultable_group (equivClasses cmp_tv unaries)
2710    cmp_tv (_,_,tv1) (_,_,tv2) = tv1 `compare` tv2
2711
2712    defaultable_group :: [(Inst,Class,TcTyVar)] -> Bool
2713    defaultable_group ds@((_,_,tv):_)
2714         =  isTyConableTyVar tv  -- Note [Avoiding spurious errors]
2715         && not (tv `elemVarSet` bad_tvs)
2716         && defaultable_classes [c | (_,c,_) <- ds]
2717    defaultable_group [] = panic "defaultable_group"
2718
2719    defaultable_classes clss 
2720         | extended_defaulting = any isInteractiveClass clss
2721         | otherwise           = all is_std_class clss && (any is_num_class clss)
2722
2723         -- In interactive mode, or with -fextended-default-rules,
2724         -- we default Show a to Show () to avoid graututious errors on "show []"
2725    isInteractiveClass cls 
2726         = is_num_class cls || (classKey cls `elem` [showClassKey, eqClassKey, ordClassKey])
2727
2728    is_num_class cls = isNumericClass cls || (ovl_strings && (cls `hasKey` isStringClassKey))
2729         -- is_num_class adds IsString to the standard numeric classes, 
2730         -- when -foverloaded-strings is enabled
2731
2732    is_std_class cls = isStandardClass cls || (ovl_strings && (cls `hasKey` isStringClassKey))
2733         -- Similarly is_std_class
2734
2735 -----------------------
2736 disambigGroup :: [Type]                 -- The default types
2737               -> [(Inst,Class,TcTyVar)] -- All standard classes of form (C a)
2738               -> TcM () -- Just does unification, to fix the default types
2739
2740 disambigGroup default_tys dicts
2741   = try_default default_tys
2742   where
2743     (_,_,tyvar) = ASSERT(not (null dicts)) head dicts   -- Should be non-empty
2744     classes = [c | (_,c,_) <- dicts]
2745
2746     try_default [] = return ()
2747     try_default (default_ty : default_tys)
2748       = tryTcLIE_ (try_default default_tys) $
2749         do { tcSimplifyDefault [mkClassPred clas [default_ty] | clas <- classes]
2750                 -- This may fail; then the tryTcLIE_ kicks in
2751                 -- Failure here is caused by there being no type in the
2752                 -- default list which can satisfy all the ambiguous classes.
2753                 -- For example, if Real a is reqd, but the only type in the
2754                 -- default list is Int.
2755
2756                 -- After this we can't fail
2757            ; warnDefault dicts default_ty
2758            ; unifyType default_ty (mkTyVarTy tyvar) 
2759            ; return () -- TOMDO: do something with the coercion
2760            }
2761
2762
2763 -----------------------
2764 getDefaultTys :: Bool -> Bool -> TcM [Type]
2765 getDefaultTys extended_deflts ovl_strings
2766   = do  { mb_defaults <- getDeclaredDefaultTys
2767         ; case mb_defaults of {
2768            Just tys -> return tys ;     -- User-supplied defaults
2769            Nothing  -> do
2770
2771         -- No use-supplied default
2772         -- Use [Integer, Double], plus modifications
2773         { integer_ty <- tcMetaTy integerTyConName
2774         ; checkWiredInTyCon doubleTyCon
2775         ; string_ty <- tcMetaTy stringTyConName
2776         ; return (opt_deflt extended_deflts unitTy
2777                         -- Note [Default unitTy]
2778                         ++
2779                   [integer_ty,doubleTy]
2780                         ++
2781                   opt_deflt ovl_strings string_ty) } } }
2782   where
2783     opt_deflt True  ty = [ty]
2784     opt_deflt False ty = []
2785 \end{code}
2786
2787 Note [Default unitTy]
2788 ~~~~~~~~~~~~~~~~~~~~~
2789 In interative mode (or with -fextended-default-rules) we add () as the first type we
2790 try when defaulting.  This has very little real impact, except in the following case.
2791 Consider: 
2792         Text.Printf.printf "hello"
2793 This has type (forall a. IO a); it prints "hello", and returns 'undefined'.  We don't
2794 want the GHCi repl loop to try to print that 'undefined'.  The neatest thing is to
2795 default the 'a' to (), rather than to Integer (which is what would otherwise happen;
2796 and then GHCi doesn't attempt to print the ().  So in interactive mode, we add
2797 () to the list of defaulting types.  See Trac #1200.
2798
2799 Note [Avoiding spurious errors]
2800 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2801 When doing the unification for defaulting, we check for skolem
2802 type variables, and simply don't default them.  For example:
2803    f = (*)      -- Monomorphic
2804    g :: Num a => a -> a
2805    g x = f x x
2806 Here, we get a complaint when checking the type signature for g,
2807 that g isn't polymorphic enough; but then we get another one when
2808 dealing with the (Num a) context arising from f's definition;
2809 we try to unify a with Int (to default it), but find that it's
2810 already been unified with the rigid variable from g's type sig
2811
2812
2813 %************************************************************************
2814 %*                                                                      *
2815 \subsection[simple]{@Simple@ versions}
2816 %*                                                                      *
2817 %************************************************************************
2818
2819 Much simpler versions when there are no bindings to make!
2820
2821 @tcSimplifyThetas@ simplifies class-type constraints formed by
2822 @deriving@ declarations and when specialising instances.  We are
2823 only interested in the simplified bunch of class/type constraints.
2824
2825 It simplifies to constraints of the form (C a b c) where
2826 a,b,c are type variables.  This is required for the context of
2827 instance declarations.
2828
2829 \begin{code}
2830 tcSimplifyDeriv :: InstOrigin
2831                 -> [TyVar]      
2832                 -> ThetaType            -- Wanted
2833                 -> TcM ThetaType        -- Needed
2834 -- Given  instance (wanted) => C inst_ty 
2835 -- Simplify 'wanted' as much as possible
2836
2837 tcSimplifyDeriv orig tyvars theta
2838   = do  { (tvs, _, tenv) <- tcInstTyVars tyvars
2839         -- The main loop may do unification, and that may crash if 
2840         -- it doesn't see a TcTyVar, so we have to instantiate. Sigh
2841         -- ToDo: what if two of them do get unified?
2842         ; wanteds <- newDictBndrsO orig (substTheta tenv theta)
2843         ; (irreds, _) <- tryHardCheckLoop doc wanteds
2844
2845         ; let (tv_dicts, others) = partition ok irreds
2846         ; addNoInstanceErrs others
2847         -- See Note [Exotic derived instance contexts] in TcMType
2848
2849         ; let rev_env = zipTopTvSubst tvs (mkTyVarTys tyvars)
2850               simpl_theta = substTheta rev_env (map dictPred tv_dicts)
2851                 -- This reverse-mapping is a pain, but the result
2852                 -- should mention the original TyVars not TcTyVars
2853
2854         ; return simpl_theta }
2855   where
2856     doc = ptext SLIT("deriving classes for a data type")
2857
2858     ok dict | isDict dict = validDerivPred (dictPred dict)
2859             | otherwise   = False
2860 \end{code}
2861
2862
2863 @tcSimplifyDefault@ just checks class-type constraints, essentially;
2864 used with \tr{default} declarations.  We are only interested in
2865 whether it worked or not.
2866
2867 \begin{code}
2868 tcSimplifyDefault :: ThetaType  -- Wanted; has no type variables in it
2869                   -> TcM ()
2870
2871 tcSimplifyDefault theta
2872   = newDictBndrsO DefaultOrigin theta   `thenM` \ wanteds ->
2873     tryHardCheckLoop doc wanteds        `thenM` \ (irreds, _) ->
2874     addNoInstanceErrs  irreds           `thenM_`
2875     if null irreds then
2876         returnM ()
2877     else
2878         traceTc (ptext SLIT("tcSimplifyDefault failing")) >> failM
2879   where
2880     doc = ptext SLIT("default declaration")
2881 \end{code}
2882
2883
2884 %************************************************************************
2885 %*                                                                      *
2886 \section{Errors and contexts}
2887 %*                                                                      *
2888 %************************************************************************
2889
2890 ToDo: for these error messages, should we note the location as coming
2891 from the insts, or just whatever seems to be around in the monad just
2892 now?
2893
2894 \begin{code}
2895 groupErrs :: ([Inst] -> TcM ()) -- Deal with one group
2896           -> [Inst]             -- The offending Insts
2897           -> TcM ()
2898 -- Group together insts with the same origin
2899 -- We want to report them together in error messages
2900
2901 groupErrs report_err [] 
2902   = return ()
2903 groupErrs report_err (inst:insts) 
2904   = do  { do_one (inst:friends)
2905         ; groupErrs report_err others }
2906   where
2907         -- (It may seem a bit crude to compare the error messages,
2908         --  but it makes sure that we combine just what the user sees,
2909         --  and it avoids need equality on InstLocs.)
2910    (friends, others) = partition is_friend insts
2911    loc_msg           = showSDoc (pprInstLoc (instLoc inst))
2912    is_friend friend  = showSDoc (pprInstLoc (instLoc friend)) == loc_msg
2913    do_one insts = addInstCtxt (instLoc (head insts)) (report_err insts)
2914                 -- Add location and context information derived from the Insts
2915
2916 -- Add the "arising from..." part to a message about bunch of dicts
2917 addInstLoc :: [Inst] -> Message -> Message
2918 addInstLoc insts msg = msg $$ nest 2 (pprInstArising (head insts))
2919
2920 addTopIPErrs :: [Name] -> [Inst] -> TcM ()
2921 addTopIPErrs bndrs [] 
2922   = return ()
2923 addTopIPErrs bndrs ips
2924   = do  { dflags <- getDOpts
2925         ; addErrTcM (tidy_env, mk_msg dflags tidy_ips) }
2926   where
2927     (tidy_env, tidy_ips) = tidyInsts ips
2928     mk_msg dflags ips 
2929         = vcat [sep [ptext SLIT("Implicit parameters escape from"),
2930                 nest 2 (ptext SLIT("the monomorphic top-level binding") 
2931                                             <> plural bndrs <+> ptext SLIT("of")
2932                                             <+> pprBinders bndrs <> colon)],
2933                 nest 2 (vcat (map ppr_ip ips)),
2934                 monomorphism_fix dflags]
2935     ppr_ip ip = pprPred (dictPred ip) <+> pprInstArising ip
2936
2937 topIPErrs :: [Inst] -> TcM ()
2938 topIPErrs dicts
2939   = groupErrs report tidy_dicts
2940   where
2941     (tidy_env, tidy_dicts) = tidyInsts dicts
2942     report dicts = addErrTcM (tidy_env, mk_msg dicts)
2943     mk_msg dicts = addInstLoc dicts (ptext SLIT("Unbound implicit parameter") <> 
2944                                      plural tidy_dicts <+> pprDictsTheta tidy_dicts)
2945
2946 addNoInstanceErrs :: [Inst]     -- Wanted (can include implications)
2947                   -> TcM ()     
2948 addNoInstanceErrs insts
2949   = do  { let (tidy_env, tidy_insts) = tidyInsts insts
2950         ; reportNoInstances tidy_env Nothing tidy_insts }
2951
2952 reportNoInstances 
2953         :: TidyEnv
2954         -> Maybe (InstLoc, [Inst])      -- Context
2955                         -- Nothing => top level
2956                         -- Just (d,g) => d describes the construct
2957                         --               with givens g
2958         -> [Inst]       -- What is wanted (can include implications)
2959         -> TcM ()       
2960
2961 reportNoInstances tidy_env mb_what insts 
2962   = groupErrs (report_no_instances tidy_env mb_what) insts
2963
2964 report_no_instances tidy_env mb_what insts
2965   = do { inst_envs <- tcGetInstEnvs
2966        ; let (implics, insts1)  = partition isImplicInst insts
2967              (insts2, overlaps) = partitionWith (check_overlap inst_envs) insts1
2968              (eqInsts, insts3)  = partition isEqInst insts2
2969        ; traceTc (text "reportNoInstances" <+> vcat 
2970                        [ppr insts, ppr implics, ppr insts1, ppr insts2])
2971        ; mapM_ complain_implic implics
2972        ; mapM_ (\doc -> addErrTcM (tidy_env, doc)) overlaps
2973        ; groupErrs complain_no_inst insts3 
2974        ; mapM_ (addErrTcM . mk_eq_err) eqInsts
2975        }
2976   where
2977     complain_no_inst insts = addErrTcM (tidy_env, mk_no_inst_err insts)
2978
2979     complain_implic inst        -- Recurse!
2980       = reportNoInstances tidy_env 
2981                           (Just (tci_loc inst, tci_given inst)) 
2982                           (tci_wanted inst)
2983
2984     check_overlap :: (InstEnv,InstEnv) -> Inst -> Either Inst SDoc
2985         -- Right msg  => overlap message
2986         -- Left  inst => no instance
2987     check_overlap inst_envs wanted
2988         | not (isClassDict wanted) = Left wanted
2989         | otherwise
2990         = case lookupInstEnv inst_envs clas tys of
2991                 -- The case of exactly one match and no unifiers means a
2992                 -- successful lookup.  That can't happen here, because dicts
2993                 -- only end up here if they didn't match in Inst.lookupInst
2994 #ifdef DEBUG
2995                 ([m],[]) -> pprPanic "reportNoInstance" (ppr wanted)
2996 #endif
2997                 ([], _)  -> Left wanted         -- No match
2998                 res      -> Right (mk_overlap_msg wanted res)
2999           where
3000             (clas,tys) = getDictClassTys wanted
3001
3002     mk_overlap_msg dict (matches, unifiers)
3003       = ASSERT( not (null matches) )
3004         vcat [  addInstLoc [dict] ((ptext SLIT("Overlapping instances for") 
3005                                         <+> pprPred (dictPred dict))),
3006                 sep [ptext SLIT("Matching instances") <> colon,
3007                      nest 2 (vcat [pprInstances ispecs, pprInstances unifiers])],
3008                 if not (isSingleton matches)
3009                 then    -- Two or more matches
3010                      empty
3011                 else    -- One match, plus some unifiers
3012                 ASSERT( not (null unifiers) )
3013                 parens (vcat [ptext SLIT("The choice depends on the instantiation of") <+>
3014                                  quotes (pprWithCommas ppr (varSetElems (tyVarsOfInst dict))),
3015                               ptext SLIT("To pick the first instance above, use -fallow-incoherent-instances"),
3016                               ptext SLIT("when compiling the other instance declarations")])]
3017       where
3018         ispecs = [ispec | (ispec, _) <- matches]
3019
3020     mk_eq_err :: Inst -> (TidyEnv, SDoc)
3021     mk_eq_err inst = misMatchMsg tidy_env (eqInstTys inst)
3022
3023     mk_no_inst_err insts
3024       | null insts = empty
3025
3026       | Just (loc, givens) <- mb_what,   -- Nested (type signatures, instance decls)
3027         not (isEmptyVarSet (tyVarsOfInsts insts))
3028       = vcat [ addInstLoc insts $
3029                sep [ ptext SLIT("Could not deduce") <+> pprDictsTheta insts
3030                    , nest 2 $ ptext SLIT("from the context") <+> pprDictsTheta givens]
3031              , show_fixes (fix1 loc : fixes2) ]
3032
3033       | otherwise       -- Top level 
3034       = vcat [ addInstLoc insts $
3035                ptext SLIT("No instance") <> plural insts
3036                     <+> ptext SLIT("for") <+> pprDictsTheta insts
3037              , show_fixes fixes2 ]
3038
3039       where
3040         fix1 loc = sep [ ptext SLIT("add") <+> pprDictsTheta insts
3041                                  <+> ptext SLIT("to the context of"),
3042                          nest 2 (ppr (instLocOrigin loc)) ]
3043                          -- I'm not sure it helps to add the location
3044                          -- nest 2 (ptext SLIT("at") <+> ppr (instLocSpan loc)) ]
3045
3046         fixes2 | null instance_dicts = []
3047                | otherwise           = [sep [ptext SLIT("add an instance declaration for"),
3048                                         pprDictsTheta instance_dicts]]
3049         instance_dicts = [d | d <- insts, isClassDict d, not (isTyVarDict d)]
3050                 -- Insts for which it is worth suggesting an adding an instance declaration
3051                 -- Exclude implicit parameters, and tyvar dicts
3052
3053         show_fixes :: [SDoc] -> SDoc
3054         show_fixes []     = empty
3055         show_fixes (f:fs) = sep [ptext SLIT("Possible fix:"), 
3056                                  nest 2 (vcat (f : map (ptext SLIT("or") <+>) fs))]
3057
3058 addTopAmbigErrs dicts
3059 -- Divide into groups that share a common set of ambiguous tyvars
3060   = ifErrsM (return ()) $       -- Only report ambiguity if no other errors happened
3061                                 -- See Note [Avoiding spurious errors]
3062     mapM_ report (equivClasses cmp [(d, tvs_of d) | d <- tidy_dicts])
3063   where
3064     (tidy_env, tidy_dicts) = tidyInsts dicts
3065
3066     tvs_of :: Inst -> [TcTyVar]
3067     tvs_of d = varSetElems (tyVarsOfInst d)
3068     cmp (_,tvs1) (_,tvs2) = tvs1 `compare` tvs2
3069     
3070     report :: [(Inst,[TcTyVar])] -> TcM ()
3071     report pairs@((inst,tvs) : _)       -- The pairs share a common set of ambiguous tyvars
3072         = mkMonomorphismMsg tidy_env tvs        `thenM` \ (tidy_env, mono_msg) ->
3073           setSrcSpan (instSpan inst) $
3074                 -- the location of the first one will do for the err message
3075           addErrTcM (tidy_env, msg $$ mono_msg)
3076         where
3077           dicts = map fst pairs
3078           msg = sep [text "Ambiguous type variable" <> plural tvs <+> 
3079                           pprQuotedList tvs <+> in_msg,
3080                      nest 2 (pprDictsInFull dicts)]
3081           in_msg = text "in the constraint" <> plural dicts <> colon
3082     report [] = panic "addTopAmbigErrs"
3083
3084
3085 mkMonomorphismMsg :: TidyEnv -> [TcTyVar] -> TcM (TidyEnv, Message)
3086 -- There's an error with these Insts; if they have free type variables
3087 -- it's probably caused by the monomorphism restriction. 
3088 -- Try to identify the offending variable
3089 -- ASSUMPTION: the Insts are fully zonked
3090 mkMonomorphismMsg tidy_env inst_tvs
3091   = do  { dflags <- getDOpts
3092         ; (tidy_env, docs) <- findGlobals (mkVarSet inst_tvs) tidy_env
3093         ; return (tidy_env, mk_msg dflags docs) }
3094   where
3095     mk_msg _ _ | any isRuntimeUnk inst_tvs
3096         =  vcat [ptext SLIT("Cannot resolve unknown runtime types:") <+>
3097                    (pprWithCommas ppr inst_tvs),
3098                 ptext SLIT("Use :print or :force to determine these types")]
3099     mk_msg _ []   = ptext SLIT("Probable fix: add a type signature that fixes these type variable(s)")
3100                         -- This happens in things like
3101                         --      f x = show (read "foo")
3102                         -- where monomorphism doesn't play any role
3103     mk_msg dflags docs 
3104         = vcat [ptext SLIT("Possible cause: the monomorphism restriction applied to the following:"),
3105                 nest 2 (vcat docs),
3106                 monomorphism_fix dflags]
3107
3108 monomorphism_fix :: DynFlags -> SDoc
3109 monomorphism_fix dflags
3110   = ptext SLIT("Probable fix:") <+> vcat
3111         [ptext SLIT("give these definition(s) an explicit type signature"),
3112          if dopt Opt_MonomorphismRestriction dflags
3113            then ptext SLIT("or use -fno-monomorphism-restriction")
3114            else empty]  -- Only suggest adding "-fno-monomorphism-restriction"
3115                         -- if it is not already set!
3116     
3117 warnDefault ups default_ty
3118   = doptM Opt_WarnTypeDefaults  `thenM` \ warn_flag ->
3119     addInstCtxt (instLoc (head (dicts))) (warnTc warn_flag warn_msg)
3120   where
3121     dicts = [d | (d,_,_) <- ups]
3122
3123         -- Tidy them first
3124     (_, tidy_dicts) = tidyInsts dicts
3125     warn_msg  = vcat [ptext SLIT("Defaulting the following constraint(s) to type") <+>
3126                                 quotes (ppr default_ty),
3127                       pprDictsInFull tidy_dicts]
3128
3129 reduceDepthErr n stack
3130   = vcat [ptext SLIT("Context reduction stack overflow; size =") <+> int n,
3131           ptext SLIT("Use -fcontext-stack=N to increase stack size to N"),
3132           nest 4 (pprStack stack)]
3133
3134 pprStack stack = vcat (map pprInstInFull stack)
3135 \end{code}