[project @ 2003-10-09 11:58:39 by simonpj]
[ghc-hetmet.git] / ghc / compiler / typecheck / TcSimplify.lhs
1 %
2 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
3 %
4 \section[TcSimplify]{TcSimplify}
5
6
7
8 \begin{code}
9 module TcSimplify (
10         tcSimplifyInfer, tcSimplifyInferCheck,
11         tcSimplifyCheck, tcSimplifyRestricted,
12         tcSimplifyToDicts, tcSimplifyIPs, 
13         tcSimplifyTop, tcSimplifyInteractive,
14         tcSimplifyBracket,
15
16         tcSimplifyDeriv, tcSimplifyDefault,
17         bindInstsOfLocalFuns
18     ) where
19
20 #include "HsVersions.h"
21
22 import {-# SOURCE #-} TcUnify( unifyTauTy )
23 import TcEnv            -- temp
24 import HsSyn            ( MonoBinds(..), HsExpr(..), andMonoBinds, andMonoBindList )
25 import TcHsSyn          ( TcExpr, TcId,
26                           TcMonoBinds, TcDictBinds
27                         )
28
29 import TcRnMonad
30 import Inst             ( lookupInst, LookupInstResult(..),
31                           tyVarsOfInst, fdPredsOfInsts, fdPredsOfInst, newDicts,
32                           isDict, isClassDict, isLinearInst, linearInstType,
33                           isStdClassTyVarDict, isMethodFor, isMethod,
34                           instToId, tyVarsOfInsts,  cloneDict,
35                           ipNamesOfInsts, ipNamesOfInst, dictPred,
36                           instBindingRequired,
37                           newDictsFromOld, tcInstClassOp,
38                           getDictClassTys, isTyVarDict,
39                           instLoc, zonkInst, tidyInsts, tidyMoreInsts,
40                           Inst, pprInsts, pprInstsInFull, tcGetInstEnvs,
41                           isIPDict, isInheritableInst, pprDFuns
42                         )
43 import TcEnv            ( tcGetGlobalTyVars, tcLookupId, findGlobals )
44 import InstEnv          ( lookupInstEnv, classInstEnv )
45 import TcMType          ( zonkTcTyVarsAndFV, tcInstTyVars, checkAmbiguity )
46 import TcType           ( TcTyVar, TcTyVarSet, ThetaType, TyVarDetails(VanillaTv),
47                           mkClassPred, isOverloadedTy, mkTyConApp,
48                           mkTyVarTy, tcGetTyVar, isTyVarClassPred, mkTyVarTys,
49                           tyVarsOfPred )
50 import Id               ( idType, mkUserLocal )
51 import Var              ( TyVar )
52 import Name             ( getOccName, getSrcLoc )
53 import NameSet          ( NameSet, mkNameSet, elemNameSet )
54 import Class            ( classBigSig, classKey )
55 import FunDeps          ( oclose, grow, improve, pprEquationDoc )
56 import PrelInfo         ( isNumericClass ) 
57 import PrelNames        ( splitName, fstName, sndName, integerTyConName,
58                           showClassKey, eqClassKey, ordClassKey )
59 import Subst            ( mkTopTyVarSubst, substTheta, substTy )
60 import TysWiredIn       ( pairTyCon, doubleTy )
61 import ErrUtils         ( Message )
62 import VarSet
63 import VarEnv           ( TidyEnv )
64 import FiniteMap
65 import Outputable
66 import ListSetOps       ( equivClasses )
67 import Util             ( zipEqual, isSingleton )
68 import List             ( partition )
69 import CmdLineOpts
70 \end{code}
71
72
73 %************************************************************************
74 %*                                                                      *
75 \subsection{NOTES}
76 %*                                                                      *
77 %************************************************************************
78
79         --------------------------------------
80                 Notes on quantification
81         --------------------------------------
82
83 Suppose we are about to do a generalisation step.
84 We have in our hand
85
86         G       the environment
87         T       the type of the RHS
88         C       the constraints from that RHS
89
90 The game is to figure out
91
92         Q       the set of type variables over which to quantify
93         Ct      the constraints we will *not* quantify over
94         Cq      the constraints we will quantify over
95
96 So we're going to infer the type
97
98         forall Q. Cq => T
99
100 and float the constraints Ct further outwards.
101
102 Here are the things that *must* be true:
103
104  (A)    Q intersect fv(G) = EMPTY                       limits how big Q can be
105  (B)    Q superset fv(Cq union T) \ oclose(fv(G),C)     limits how small Q can be
106
107 (A) says we can't quantify over a variable that's free in the
108 environment.  (B) says we must quantify over all the truly free
109 variables in T, else we won't get a sufficiently general type.  We do
110 not *need* to quantify over any variable that is fixed by the free
111 vars of the environment G.
112
113         BETWEEN THESE TWO BOUNDS, ANY Q WILL DO!
114
115 Example:        class H x y | x->y where ...
116
117         fv(G) = {a}     C = {H a b, H c d}
118                         T = c -> b
119
120         (A)  Q intersect {a} is empty
121         (B)  Q superset {a,b,c,d} \ oclose({a}, C) = {a,b,c,d} \ {a,b} = {c,d}
122
123         So Q can be {c,d}, {b,c,d}
124
125 Other things being equal, however, we'd like to quantify over as few
126 variables as possible: smaller types, fewer type applications, more
127 constraints can get into Ct instead of Cq.
128
129
130 -----------------------------------------
131 We will make use of
132
133   fv(T)         the free type vars of T
134
135   oclose(vs,C)  The result of extending the set of tyvars vs
136                 using the functional dependencies from C
137
138   grow(vs,C)    The result of extend the set of tyvars vs
139                 using all conceivable links from C.
140
141                 E.g. vs = {a}, C = {H [a] b, K (b,Int) c, Eq e}
142                 Then grow(vs,C) = {a,b,c}
143
144                 Note that grow(vs,C) `superset` grow(vs,simplify(C))
145                 That is, simplfication can only shrink the result of grow.
146
147 Notice that
148    oclose is conservative one way:      v `elem` oclose(vs,C) => v is definitely fixed by vs
149    grow is conservative the other way:  if v might be fixed by vs => v `elem` grow(vs,C)
150
151
152 -----------------------------------------
153
154 Choosing Q
155 ~~~~~~~~~~
156 Here's a good way to choose Q:
157
158         Q = grow( fv(T), C ) \ oclose( fv(G), C )
159
160 That is, quantify over all variable that that MIGHT be fixed by the
161 call site (which influences T), but which aren't DEFINITELY fixed by
162 G.  This choice definitely quantifies over enough type variables,
163 albeit perhaps too many.
164
165 Why grow( fv(T), C ) rather than fv(T)?  Consider
166
167         class H x y | x->y where ...
168
169         T = c->c
170         C = (H c d)
171
172   If we used fv(T) = {c} we'd get the type
173
174         forall c. H c d => c -> b
175
176   And then if the fn was called at several different c's, each of
177   which fixed d differently, we'd get a unification error, because
178   d isn't quantified.  Solution: quantify d.  So we must quantify
179   everything that might be influenced by c.
180
181 Why not oclose( fv(T), C )?  Because we might not be able to see
182 all the functional dependencies yet:
183
184         class H x y | x->y where ...
185         instance H x y => Eq (T x y) where ...
186
187         T = c->c
188         C = (Eq (T c d))
189
190   Now oclose(fv(T),C) = {c}, because the functional dependency isn't
191   apparent yet, and that's wrong.  We must really quantify over d too.
192
193
194 There really isn't any point in quantifying over any more than
195 grow( fv(T), C ), because the call sites can't possibly influence
196 any other type variables.
197
198
199
200         --------------------------------------
201                 Notes on ambiguity
202         --------------------------------------
203
204 It's very hard to be certain when a type is ambiguous.  Consider
205
206         class K x
207         class H x y | x -> y
208         instance H x y => K (x,y)
209
210 Is this type ambiguous?
211         forall a b. (K (a,b), Eq b) => a -> a
212
213 Looks like it!  But if we simplify (K (a,b)) we get (H a b) and
214 now we see that a fixes b.  So we can't tell about ambiguity for sure
215 without doing a full simplification.  And even that isn't possible if
216 the context has some free vars that may get unified.  Urgle!
217
218 Here's another example: is this ambiguous?
219         forall a b. Eq (T b) => a -> a
220 Not if there's an insance decl (with no context)
221         instance Eq (T b) where ...
222
223 You may say of this example that we should use the instance decl right
224 away, but you can't always do that:
225
226         class J a b where ...
227         instance J Int b where ...
228
229         f :: forall a b. J a b => a -> a
230
231 (Notice: no functional dependency in J's class decl.)
232 Here f's type is perfectly fine, provided f is only called at Int.
233 It's premature to complain when meeting f's signature, or even
234 when inferring a type for f.
235
236
237
238 However, we don't *need* to report ambiguity right away.  It'll always
239 show up at the call site.... and eventually at main, which needs special
240 treatment.  Nevertheless, reporting ambiguity promptly is an excellent thing.
241
242 So here's the plan.  We WARN about probable ambiguity if
243
244         fv(Cq) is not a subset of  oclose(fv(T) union fv(G), C)
245
246 (all tested before quantification).
247 That is, all the type variables in Cq must be fixed by the the variables
248 in the environment, or by the variables in the type.
249
250 Notice that we union before calling oclose.  Here's an example:
251
252         class J a b c | a b -> c
253         fv(G) = {a}
254
255 Is this ambiguous?
256         forall b c. (J a b c) => b -> b
257
258 Only if we union {a} from G with {b} from T before using oclose,
259 do we see that c is fixed.
260
261 It's a bit vague exactly which C we should use for this oclose call.  If we
262 don't fix enough variables we might complain when we shouldn't (see
263 the above nasty example).  Nothing will be perfect.  That's why we can
264 only issue a warning.
265
266
267 Can we ever be *certain* about ambiguity?  Yes: if there's a constraint
268
269         c in C such that fv(c) intersect (fv(G) union fv(T)) = EMPTY
270
271 then c is a "bubble"; there's no way it can ever improve, and it's
272 certainly ambiguous.  UNLESS it is a constant (sigh).  And what about
273 the nasty example?
274
275         class K x
276         class H x y | x -> y
277         instance H x y => K (x,y)
278
279 Is this type ambiguous?
280         forall a b. (K (a,b), Eq b) => a -> a
281
282 Urk.  The (Eq b) looks "definitely ambiguous" but it isn't.  What we are after
283 is a "bubble" that's a set of constraints
284
285         Cq = Ca union Cq'  st  fv(Ca) intersect (fv(Cq') union fv(T) union fv(G)) = EMPTY
286
287 Hence another idea.  To decide Q start with fv(T) and grow it
288 by transitive closure in Cq (no functional dependencies involved).
289 Now partition Cq using Q, leaving the definitely-ambiguous and probably-ok.
290 The definitely-ambiguous can then float out, and get smashed at top level
291 (which squashes out the constants, like Eq (T a) above)
292
293
294         --------------------------------------
295                 Notes on principal types
296         --------------------------------------
297
298     class C a where
299       op :: a -> a
300
301     f x = let g y = op (y::Int) in True
302
303 Here the principal type of f is (forall a. a->a)
304 but we'll produce the non-principal type
305     f :: forall a. C Int => a -> a
306
307
308         --------------------------------------
309                 Notes on implicit parameters
310         --------------------------------------
311
312 Question 1: can we "inherit" implicit parameters
313 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
314 Consider this:
315
316         f x = (x::Int) + ?y
317
318 where f is *not* a top-level binding.
319 From the RHS of f we'll get the constraint (?y::Int).
320 There are two types we might infer for f:
321
322         f :: Int -> Int
323
324 (so we get ?y from the context of f's definition), or
325
326         f :: (?y::Int) => Int -> Int
327
328 At first you might think the first was better, becuase then
329 ?y behaves like a free variable of the definition, rather than
330 having to be passed at each call site.  But of course, the WHOLE
331 IDEA is that ?y should be passed at each call site (that's what
332 dynamic binding means) so we'd better infer the second.
333
334 BOTTOM LINE: when *inferring types* you *must* quantify 
335 over implicit parameters. See the predicate isFreeWhenInferring.
336
337
338 Question 2: type signatures
339 ~~~~~~~~~~~~~~~~~~~~~~~~~~~
340 BUT WATCH OUT: When you supply a type signature, we can't force you
341 to quantify over implicit parameters.  For example:
342
343         (?x + 1) :: Int
344
345 This is perfectly reasonable.  We do not want to insist on
346
347         (?x + 1) :: (?x::Int => Int)
348
349 That would be silly.  Here, the definition site *is* the occurrence site,
350 so the above strictures don't apply.  Hence the difference between
351 tcSimplifyCheck (which *does* allow implicit paramters to be inherited)
352 and tcSimplifyCheckBind (which does not).
353
354 What about when you supply a type signature for a binding?
355 Is it legal to give the following explicit, user type 
356 signature to f, thus:
357
358         f :: Int -> Int
359         f x = (x::Int) + ?y
360
361 At first sight this seems reasonable, but it has the nasty property
362 that adding a type signature changes the dynamic semantics.
363 Consider this:
364
365         (let f x = (x::Int) + ?y
366          in (f 3, f 3 with ?y=5))  with ?y = 6
367
368                 returns (3+6, 3+5)
369 vs
370         (let f :: Int -> Int
371              f x = x + ?y
372          in (f 3, f 3 with ?y=5))  with ?y = 6
373
374                 returns (3+6, 3+6)
375
376 Indeed, simply inlining f (at the Haskell source level) would change the
377 dynamic semantics.
378
379 Nevertheless, as Launchbury says (email Oct 01) we can't really give the
380 semantics for a Haskell program without knowing its typing, so if you 
381 change the typing you may change the semantics.
382
383 To make things consistent in all cases where we are *checking* against
384 a supplied signature (as opposed to inferring a type), we adopt the
385 rule: 
386
387         a signature does not need to quantify over implicit params.
388
389 [This represents a (rather marginal) change of policy since GHC 5.02,
390 which *required* an explicit signature to quantify over all implicit
391 params for the reasons mentioned above.]
392
393 But that raises a new question.  Consider 
394
395         Given (signature)       ?x::Int
396         Wanted (inferred)       ?x::Int, ?y::Bool
397
398 Clearly we want to discharge the ?x and float the ?y out.  But
399 what is the criterion that distinguishes them?  Clearly it isn't
400 what free type variables they have.  The Right Thing seems to be
401 to float a constraint that
402         neither mentions any of the quantified type variables
403         nor any of the quantified implicit parameters
404
405 See the predicate isFreeWhenChecking.
406
407
408 Question 3: monomorphism
409 ~~~~~~~~~~~~~~~~~~~~~~~~
410 There's a nasty corner case when the monomorphism restriction bites:
411
412         z = (x::Int) + ?y
413
414 The argument above suggests that we *must* generalise
415 over the ?y parameter, to get
416         z :: (?y::Int) => Int,
417 but the monomorphism restriction says that we *must not*, giving
418         z :: Int.
419 Why does the momomorphism restriction say this?  Because if you have
420
421         let z = x + ?y in z+z
422
423 you might not expect the addition to be done twice --- but it will if
424 we follow the argument of Question 2 and generalise over ?y.
425
426
427
428 Possible choices
429 ~~~~~~~~~~~~~~~~
430 (A) Always generalise over implicit parameters
431     Bindings that fall under the monomorphism restriction can't
432         be generalised
433
434     Consequences:
435         * Inlining remains valid
436         * No unexpected loss of sharing
437         * But simple bindings like
438                 z = ?y + 1
439           will be rejected, unless you add an explicit type signature
440           (to avoid the monomorphism restriction)
441                 z :: (?y::Int) => Int
442                 z = ?y + 1
443           This seems unacceptable
444
445 (B) Monomorphism restriction "wins"
446     Bindings that fall under the monomorphism restriction can't
447         be generalised
448     Always generalise over implicit parameters *except* for bindings
449         that fall under the monomorphism restriction
450
451     Consequences
452         * Inlining isn't valid in general
453         * No unexpected loss of sharing
454         * Simple bindings like
455                 z = ?y + 1
456           accepted (get value of ?y from binding site)
457
458 (C) Always generalise over implicit parameters
459     Bindings that fall under the monomorphism restriction can't
460         be generalised, EXCEPT for implicit parameters
461     Consequences
462         * Inlining remains valid
463         * Unexpected loss of sharing (from the extra generalisation)
464         * Simple bindings like
465                 z = ?y + 1
466           accepted (get value of ?y from occurrence sites)
467
468
469 Discussion
470 ~~~~~~~~~~
471 None of these choices seems very satisfactory.  But at least we should
472 decide which we want to do.
473
474 It's really not clear what is the Right Thing To Do.  If you see
475
476         z = (x::Int) + ?y
477
478 would you expect the value of ?y to be got from the *occurrence sites*
479 of 'z', or from the valuue of ?y at the *definition* of 'z'?  In the
480 case of function definitions, the answer is clearly the former, but
481 less so in the case of non-fucntion definitions.   On the other hand,
482 if we say that we get the value of ?y from the definition site of 'z',
483 then inlining 'z' might change the semantics of the program.
484
485 Choice (C) really says "the monomorphism restriction doesn't apply
486 to implicit parameters".  Which is fine, but remember that every
487 innocent binding 'x = ...' that mentions an implicit parameter in
488 the RHS becomes a *function* of that parameter, called at each
489 use of 'x'.  Now, the chances are that there are no intervening 'with'
490 clauses that bind ?y, so a decent compiler should common up all
491 those function calls.  So I think I strongly favour (C).  Indeed,
492 one could make a similar argument for abolishing the monomorphism
493 restriction altogether.
494
495 BOTTOM LINE: we choose (B) at present.  See tcSimplifyRestricted
496
497
498
499 %************************************************************************
500 %*                                                                      *
501 \subsection{tcSimplifyInfer}
502 %*                                                                      *
503 %************************************************************************
504
505 tcSimplify is called when we *inferring* a type.  Here's the overall game plan:
506
507     1. Compute Q = grow( fvs(T), C )
508
509     2. Partition C based on Q into Ct and Cq.  Notice that ambiguous
510        predicates will end up in Ct; we deal with them at the top level
511
512     3. Try improvement, using functional dependencies
513
514     4. If Step 3 did any unification, repeat from step 1
515        (Unification can change the result of 'grow'.)
516
517 Note: we don't reduce dictionaries in step 2.  For example, if we have
518 Eq (a,b), we don't simplify to (Eq a, Eq b).  So Q won't be different
519 after step 2.  However note that we may therefore quantify over more
520 type variables than we absolutely have to.
521
522 For the guts, we need a loop, that alternates context reduction and
523 improvement with unification.  E.g. Suppose we have
524
525         class C x y | x->y where ...
526
527 and tcSimplify is called with:
528         (C Int a, C Int b)
529 Then improvement unifies a with b, giving
530         (C Int a, C Int a)
531
532 If we need to unify anything, we rattle round the whole thing all over
533 again.
534
535
536 \begin{code}
537 tcSimplifyInfer
538         :: SDoc
539         -> TcTyVarSet           -- fv(T); type vars
540         -> [Inst]               -- Wanted
541         -> TcM ([TcTyVar],      -- Tyvars to quantify (zonked)
542                 TcDictBinds,    -- Bindings
543                 [TcId])         -- Dict Ids that must be bound here (zonked)
544         -- Any free (escaping) Insts are tossed into the environment
545 \end{code}
546
547
548 \begin{code}
549 tcSimplifyInfer doc tau_tvs wanted_lie
550   = inferLoop doc (varSetElems tau_tvs)
551               wanted_lie                `thenM` \ (qtvs, frees, binds, irreds) ->
552
553     extendLIEs frees                                                    `thenM_`
554     returnM (qtvs, binds, map instToId irreds)
555
556 inferLoop doc tau_tvs wanteds
557   =     -- Step 1
558     zonkTcTyVarsAndFV tau_tvs           `thenM` \ tau_tvs' ->
559     mappM zonkInst wanteds              `thenM` \ wanteds' ->
560     tcGetGlobalTyVars                   `thenM` \ gbl_tvs ->
561     let
562         preds = fdPredsOfInsts wanteds'
563         qtvs  = grow preds tau_tvs' `minusVarSet` oclose preds gbl_tvs
564
565         try_me inst
566           | isFreeWhenInferring qtvs inst = Free
567           | isClassDict inst              = DontReduceUnlessConstant    -- Dicts
568           | otherwise                     = ReduceMe                    -- Lits and Methods
569     in
570     traceTc (text "infloop" <+> vcat [ppr tau_tvs', ppr wanteds', ppr preds, ppr (grow preds tau_tvs'), ppr qtvs])      `thenM_`
571                 -- Step 2
572     reduceContext doc try_me [] wanteds'    `thenM` \ (no_improvement, frees, binds, irreds) ->
573
574                 -- Step 3
575     if no_improvement then
576         returnM (varSetElems qtvs, frees, binds, irreds)
577     else
578         -- If improvement did some unification, we go round again.  There
579         -- are two subtleties:
580         --   a) We start again with irreds, not wanteds
581         --      Using an instance decl might have introduced a fresh type variable
582         --      which might have been unified, so we'd get an infinite loop
583         --      if we started again with wanteds!  See example [LOOP]
584         --
585         --   b) It's also essential to re-process frees, because unification
586         --      might mean that a type variable that looked free isn't now.
587         --
588         -- Hence the (irreds ++ frees)
589
590         -- However, NOTICE that when we are done, we might have some bindings, but
591         -- the final qtvs might be empty.  See [NO TYVARS] below.
592                                 
593         inferLoop doc tau_tvs (irreds ++ frees) `thenM` \ (qtvs1, frees1, binds1, irreds1) ->
594         returnM (qtvs1, frees1, binds `AndMonoBinds` binds1, irreds1)
595 \end{code}
596
597 Example [LOOP]
598
599         class If b t e r | b t e -> r
600         instance If T t e t
601         instance If F t e e
602         class Lte a b c | a b -> c where lte :: a -> b -> c
603         instance Lte Z b T
604         instance (Lte a b l,If l b a c) => Max a b c
605
606 Wanted: Max Z (S x) y
607
608 Then we'll reduce using the Max instance to:
609         (Lte Z (S x) l, If l (S x) Z y)
610 and improve by binding l->T, after which we can do some reduction
611 on both the Lte and If constraints.  What we *can't* do is start again
612 with (Max Z (S x) y)!
613
614 [NO TYVARS]
615
616         class Y a b | a -> b where
617             y :: a -> X b
618         
619         instance Y [[a]] a where
620             y ((x:_):_) = X x
621         
622         k :: X a -> X a -> X a
623
624         g :: Num a => [X a] -> [X a]
625         g xs = h xs
626             where
627             h ys = ys ++ map (k (y [[0]])) xs
628
629 The excitement comes when simplifying the bindings for h.  Initially
630 try to simplify {y @ [[t1]] t2, 0 @ t1}, with initial qtvs = {t2}.
631 From this we get t1:=:t2, but also various bindings.  We can't forget
632 the bindings (because of [LOOP]), but in fact t1 is what g is
633 polymorphic in.  
634
635 The net effect of [NO TYVARS] 
636
637 \begin{code}
638 isFreeWhenInferring :: TyVarSet -> Inst -> Bool
639 isFreeWhenInferring qtvs inst
640   =  isFreeWrtTyVars qtvs inst          -- Constrains no quantified vars
641   && isInheritableInst inst             -- And no implicit parameter involved
642                                         -- (see "Notes on implicit parameters")
643
644 isFreeWhenChecking :: TyVarSet  -- Quantified tyvars
645                    -> NameSet   -- Quantified implicit parameters
646                    -> Inst -> Bool
647 isFreeWhenChecking qtvs ips inst
648   =  isFreeWrtTyVars qtvs inst
649   && isFreeWrtIPs    ips inst
650
651 isFreeWrtTyVars qtvs inst = not (tyVarsOfInst inst `intersectsVarSet` qtvs)
652 isFreeWrtIPs     ips inst = not (any (`elemNameSet` ips) (ipNamesOfInst inst))
653 \end{code}
654
655
656 %************************************************************************
657 %*                                                                      *
658 \subsection{tcSimplifyCheck}
659 %*                                                                      *
660 %************************************************************************
661
662 @tcSimplifyCheck@ is used when we know exactly the set of variables
663 we are going to quantify over.  For example, a class or instance declaration.
664
665 \begin{code}
666 tcSimplifyCheck
667          :: SDoc
668          -> [TcTyVar]           -- Quantify over these
669          -> [Inst]              -- Given
670          -> [Inst]              -- Wanted
671          -> TcM TcDictBinds     -- Bindings
672
673 -- tcSimplifyCheck is used when checking expression type signatures,
674 -- class decls, instance decls etc.
675 --
676 -- NB: tcSimplifyCheck does not consult the
677 --      global type variables in the environment; so you don't
678 --      need to worry about setting them before calling tcSimplifyCheck
679 tcSimplifyCheck doc qtvs givens wanted_lie
680   = tcSimplCheck doc get_qtvs
681                  givens wanted_lie      `thenM` \ (qtvs', binds) ->
682     returnM binds
683   where
684     get_qtvs = zonkTcTyVarsAndFV qtvs
685
686
687 -- tcSimplifyInferCheck is used when we know the constraints we are to simplify
688 -- against, but we don't know the type variables over which we are going to quantify.
689 -- This happens when we have a type signature for a mutually recursive group
690 tcSimplifyInferCheck
691          :: SDoc
692          -> TcTyVarSet          -- fv(T)
693          -> [Inst]              -- Given
694          -> [Inst]              -- Wanted
695          -> TcM ([TcTyVar],     -- Variables over which to quantify
696                  TcDictBinds)   -- Bindings
697
698 tcSimplifyInferCheck doc tau_tvs givens wanted_lie
699   = tcSimplCheck doc get_qtvs givens wanted_lie
700   where
701         -- Figure out which type variables to quantify over
702         -- You might think it should just be the signature tyvars,
703         -- but in bizarre cases you can get extra ones
704         --      f :: forall a. Num a => a -> a
705         --      f x = fst (g (x, head [])) + 1
706         --      g a b = (b,a)
707         -- Here we infer g :: forall a b. a -> b -> (b,a)
708         -- We don't want g to be monomorphic in b just because
709         -- f isn't quantified over b.
710     all_tvs = varSetElems (tau_tvs `unionVarSet` tyVarsOfInsts givens)
711
712     get_qtvs = zonkTcTyVarsAndFV all_tvs        `thenM` \ all_tvs' ->
713                tcGetGlobalTyVars                `thenM` \ gbl_tvs ->
714                let
715                   qtvs = all_tvs' `minusVarSet` gbl_tvs
716                         -- We could close gbl_tvs, but its not necessary for
717                         -- soundness, and it'll only affect which tyvars, not which
718                         -- dictionaries, we quantify over
719                in
720                returnM qtvs
721 \end{code}
722
723 Here is the workhorse function for all three wrappers.
724
725 \begin{code}
726 tcSimplCheck doc get_qtvs givens wanted_lie
727   = check_loop givens wanted_lie        `thenM` \ (qtvs, frees, binds, irreds) ->
728
729         -- Complain about any irreducible ones
730     mappM zonkInst given_dicts_and_ips                          `thenM` \ givens' ->
731     groupErrs (addNoInstanceErrs (Just doc) givens') irreds     `thenM_`
732
733         -- Done
734     extendLIEs frees            `thenM_`
735     returnM (qtvs, binds)
736
737   where
738     given_dicts_and_ips = filter (not . isMethod) givens
739         -- For error reporting, filter out methods, which are 
740         -- only added to the given set as an optimisation
741
742     ip_set = mkNameSet (ipNamesOfInsts givens)
743
744     check_loop givens wanteds
745       =         -- Step 1
746         mappM zonkInst givens   `thenM` \ givens' ->
747         mappM zonkInst wanteds  `thenM` \ wanteds' ->
748         get_qtvs                `thenM` \ qtvs' ->
749
750                     -- Step 2
751         let
752             -- When checking against a given signature we always reduce
753             -- until we find a match against something given, or can't reduce
754             try_me inst | isFreeWhenChecking qtvs' ip_set inst = Free
755                         | otherwise                            = ReduceMe
756         in
757         reduceContext doc try_me givens' wanteds'       `thenM` \ (no_improvement, frees, binds, irreds) ->
758
759                     -- Step 3
760         if no_improvement then
761             returnM (varSetElems qtvs', frees, binds, irreds)
762         else
763             check_loop givens' (irreds ++ frees)        `thenM` \ (qtvs', frees1, binds1, irreds1) ->
764             returnM (qtvs', frees1, binds `AndMonoBinds` binds1, irreds1)
765 \end{code}
766
767
768 %************************************************************************
769 %*                                                                      *
770 \subsection{tcSimplifyRestricted}
771 %*                                                                      *
772 %************************************************************************
773
774 \begin{code}
775 tcSimplifyRestricted    -- Used for restricted binding groups
776                         -- i.e. ones subject to the monomorphism restriction
777         :: SDoc
778         -> TcTyVarSet           -- Free in the type of the RHSs
779         -> [Inst]               -- Free in the RHSs
780         -> TcM ([TcTyVar],      -- Tyvars to quantify (zonked)
781                 TcDictBinds)    -- Bindings
782
783 tcSimplifyRestricted doc tau_tvs wanteds
784   =     -- First squash out all methods, to find the constrained tyvars
785         -- We can't just take the free vars of wanted_lie because that'll
786         -- have methods that may incidentally mention entirely unconstrained variables
787         --      e.g. a call to  f :: Eq a => a -> b -> b
788         -- Here, b is unconstrained.  A good example would be
789         --      foo = f (3::Int)
790         -- We want to infer the polymorphic type
791         --      foo :: forall b. b -> b
792
793         -- 'reduceMe': Reduce as far as we can.  Don't stop at
794         -- dicts; the idea is to get rid of as many type
795         -- variables as possible, and we don't want to stop
796         -- at (say) Monad (ST s), because that reduces
797         -- immediately, with no constraint on s.
798     simpleReduceLoop doc reduceMe wanteds       `thenM` \ (foo_frees, foo_binds, constrained_dicts) ->
799
800         -- Next, figure out the tyvars we will quantify over
801     zonkTcTyVarsAndFV (varSetElems tau_tvs)     `thenM` \ tau_tvs' ->
802     tcGetGlobalTyVars                           `thenM` \ gbl_tvs ->
803     let
804         constrained_tvs = tyVarsOfInsts constrained_dicts
805         qtvs = (tau_tvs' `minusVarSet` oclose (fdPredsOfInsts constrained_dicts) gbl_tvs)
806                          `minusVarSet` constrained_tvs
807     in
808     traceTc (text "tcSimplifyRestricted" <+> vcat [
809                 pprInsts wanteds, pprInsts foo_frees, pprInsts constrained_dicts,
810                 ppr foo_binds,
811                 ppr constrained_tvs, ppr tau_tvs', ppr qtvs ])  `thenM_`
812
813         -- The first step may have squashed more methods than
814         -- necessary, so try again, this time knowing the exact
815         -- set of type variables to quantify over.
816         --
817         -- We quantify only over constraints that are captured by qtvs;
818         -- these will just be a subset of non-dicts.  This in contrast
819         -- to normal inference (using isFreeWhenInferring) in which we quantify over
820         -- all *non-inheritable* constraints too.  This implements choice
821         -- (B) under "implicit parameter and monomorphism" above.
822         --
823         -- Remember that we may need to do *some* simplification, to
824         -- (for example) squash {Monad (ST s)} into {}.  It's not enough
825         -- just to float all constraints
826     restrict_loop doc qtvs wanteds
827         -- We still need a loop because improvement can take place
828         -- E.g. if we have (C (T a)) and the instance decl
829         --      instance D Int b => C (T a) where ...
830         -- and there's a functional dependency for D.   Then we may improve
831         -- the tyep variable 'b'.
832
833 restrict_loop doc qtvs wanteds
834   = mappM zonkInst wanteds                      `thenM` \ wanteds' ->
835     zonkTcTyVarsAndFV (varSetElems qtvs)        `thenM` \ qtvs' ->
836     let
837         try_me inst | isFreeWrtTyVars qtvs' inst = Free
838                     | otherwise                  = ReduceMe
839     in
840     reduceContext doc try_me [] wanteds'        `thenM` \ (no_improvement, frees, binds, irreds) ->
841     if no_improvement then
842         ASSERT( null irreds )
843         extendLIEs frees                        `thenM_`
844         returnM (varSetElems qtvs', binds)
845     else
846         restrict_loop doc qtvs' (irreds ++ frees)       `thenM` \ (qtvs1, binds1) ->
847         returnM (qtvs1, binds `AndMonoBinds` binds1)
848 \end{code}
849
850
851 %************************************************************************
852 %*                                                                      *
853 \subsection{tcSimplifyToDicts}
854 %*                                                                      *
855 %************************************************************************
856
857 On the LHS of transformation rules we only simplify methods and constants,
858 getting dictionaries.  We want to keep all of them unsimplified, to serve
859 as the available stuff for the RHS of the rule.
860
861 The same thing is used for specialise pragmas. Consider
862
863         f :: Num a => a -> a
864         {-# SPECIALISE f :: Int -> Int #-}
865         f = ...
866
867 The type checker generates a binding like:
868
869         f_spec = (f :: Int -> Int)
870
871 and we want to end up with
872
873         f_spec = _inline_me_ (f Int dNumInt)
874
875 But that means that we must simplify the Method for f to (f Int dNumInt)!
876 So tcSimplifyToDicts squeezes out all Methods.
877
878 IMPORTANT NOTE:  we *don't* want to do superclass commoning up.  Consider
879
880         fromIntegral :: (Integral a, Num b) => a -> b
881         {-# RULES "foo"  fromIntegral = id :: Int -> Int #-}
882
883 Here, a=b=Int, and Num Int is a superclass of Integral Int. But we *dont*
884 want to get
885
886         forall dIntegralInt.
887         fromIntegral Int Int dIntegralInt (scsel dIntegralInt) = id Int
888
889 because the scsel will mess up matching.  Instead we want
890
891         forall dIntegralInt, dNumInt.
892         fromIntegral Int Int dIntegralInt dNumInt = id Int
893
894 Hence "DontReduce NoSCs"
895
896 \begin{code}
897 tcSimplifyToDicts :: [Inst] -> TcM (TcDictBinds)
898 tcSimplifyToDicts wanteds
899   = simpleReduceLoop doc try_me wanteds         `thenM` \ (frees, binds, irreds) ->
900         -- Since try_me doesn't look at types, we don't need to
901         -- do any zonking, so it's safe to call reduceContext directly
902     ASSERT( null frees )
903     extendLIEs irreds           `thenM_`
904     returnM binds
905
906   where
907     doc = text "tcSimplifyToDicts"
908
909         -- Reduce methods and lits only; stop as soon as we get a dictionary
910     try_me inst | isDict inst = DontReduce NoSCs
911                 | otherwise   = ReduceMe
912 \end{code}
913
914
915
916 tcSimplifyBracket is used when simplifying the constraints arising from
917 a Template Haskell bracket [| ... |].  We want to check that there aren't
918 any constraints that can't be satisfied (e.g. Show Foo, where Foo has no
919 Show instance), but we aren't otherwise interested in the results.
920 Nor do we care about ambiguous dictionaries etc.  We will type check
921 this bracket again at its usage site.
922
923 \begin{code}
924 tcSimplifyBracket :: [Inst] -> TcM ()
925 tcSimplifyBracket wanteds
926   = simpleReduceLoop doc reduceMe wanteds       `thenM_`
927     returnM ()
928   where
929     doc = text "tcSimplifyBracket"
930 \end{code}
931
932
933 %************************************************************************
934 %*                                                                      *
935 \subsection{Filtering at a dynamic binding}
936 %*                                                                      *
937 %************************************************************************
938
939 When we have
940         let ?x = R in B
941
942 we must discharge all the ?x constraints from B.  We also do an improvement
943 step; if we have ?x::t1 and ?x::t2 we must unify t1, t2.
944
945 Actually, the constraints from B might improve the types in ?x. For example
946
947         f :: (?x::Int) => Char -> Char
948         let ?x = 3 in f 'c'
949
950 then the constraint (?x::Int) arising from the call to f will
951 force the binding for ?x to be of type Int.
952
953 \begin{code}
954 tcSimplifyIPs :: [Inst]         -- The implicit parameters bound here
955               -> [Inst]         -- Wanted
956               -> TcM TcDictBinds
957 tcSimplifyIPs given_ips wanteds
958   = simpl_loop given_ips wanteds        `thenM` \ (frees, binds) ->
959     extendLIEs frees                    `thenM_`
960     returnM binds
961   where
962     doc      = text "tcSimplifyIPs" <+> ppr given_ips
963     ip_set   = mkNameSet (ipNamesOfInsts given_ips)
964
965         -- Simplify any methods that mention the implicit parameter
966     try_me inst | isFreeWrtIPs ip_set inst = Free
967                 | otherwise                = ReduceMe
968
969     simpl_loop givens wanteds
970       = mappM zonkInst givens           `thenM` \ givens' ->
971         mappM zonkInst wanteds          `thenM` \ wanteds' ->
972
973         reduceContext doc try_me givens' wanteds'    `thenM` \ (no_improvement, frees, binds, irreds) ->
974
975         if no_improvement then
976             ASSERT( null irreds )
977             returnM (frees, binds)
978         else
979             simpl_loop givens' (irreds ++ frees)        `thenM` \ (frees1, binds1) ->
980             returnM (frees1, binds `AndMonoBinds` binds1)
981 \end{code}
982
983
984 %************************************************************************
985 %*                                                                      *
986 \subsection[binds-for-local-funs]{@bindInstsOfLocalFuns@}
987 %*                                                                      *
988 %************************************************************************
989
990 When doing a binding group, we may have @Insts@ of local functions.
991 For example, we might have...
992 \begin{verbatim}
993 let f x = x + 1     -- orig local function (overloaded)
994     f.1 = f Int     -- two instances of f
995     f.2 = f Float
996  in
997     (f.1 5, f.2 6.7)
998 \end{verbatim}
999 The point is: we must drop the bindings for @f.1@ and @f.2@ here,
1000 where @f@ is in scope; those @Insts@ must certainly not be passed
1001 upwards towards the top-level.  If the @Insts@ were binding-ified up
1002 there, they would have unresolvable references to @f@.
1003
1004 We pass in an @init_lie@ of @Insts@ and a list of locally-bound @Ids@.
1005 For each method @Inst@ in the @init_lie@ that mentions one of the
1006 @Ids@, we create a binding.  We return the remaining @Insts@ (in an
1007 @LIE@), as well as the @HsBinds@ generated.
1008
1009 \begin{code}
1010 bindInstsOfLocalFuns :: [Inst] -> [TcId] -> TcM TcMonoBinds
1011
1012 bindInstsOfLocalFuns wanteds local_ids
1013   | null overloaded_ids
1014         -- Common case
1015   = extendLIEs wanteds          `thenM_`
1016     returnM EmptyMonoBinds
1017
1018   | otherwise
1019   = simpleReduceLoop doc try_me wanteds         `thenM` \ (frees, binds, irreds) ->
1020     ASSERT( null irreds )
1021     extendLIEs frees            `thenM_`
1022     returnM binds
1023   where
1024     doc              = text "bindInsts" <+> ppr local_ids
1025     overloaded_ids   = filter is_overloaded local_ids
1026     is_overloaded id = isOverloadedTy (idType id)
1027
1028     overloaded_set = mkVarSet overloaded_ids    -- There can occasionally be a lot of them
1029                                                 -- so it's worth building a set, so that
1030                                                 -- lookup (in isMethodFor) is faster
1031
1032     try_me inst | isMethodFor overloaded_set inst = ReduceMe
1033                 | otherwise                       = Free
1034 \end{code}
1035
1036
1037 %************************************************************************
1038 %*                                                                      *
1039 \subsection{Data types for the reduction mechanism}
1040 %*                                                                      *
1041 %************************************************************************
1042
1043 The main control over context reduction is here
1044
1045 \begin{code}
1046 data WhatToDo
1047  = ReduceMe             -- Try to reduce this
1048                         -- If there's no instance, behave exactly like
1049                         -- DontReduce: add the inst to
1050                         -- the irreductible ones, but don't
1051                         -- produce an error message of any kind.
1052                         -- It might be quite legitimate such as (Eq a)!
1053
1054  | DontReduce WantSCs           -- Return as irreducible
1055
1056  | DontReduceUnlessConstant     -- Return as irreducible unless it can
1057                                 -- be reduced to a constant in one step
1058
1059  | Free                   -- Return as free
1060
1061 reduceMe :: Inst -> WhatToDo
1062 reduceMe inst = ReduceMe
1063
1064 data WantSCs = NoSCs | AddSCs   -- Tells whether we should add the superclasses
1065                                 -- of a predicate when adding it to the avails
1066 \end{code}
1067
1068
1069
1070 \begin{code}
1071 type Avails = FiniteMap Inst Avail
1072
1073 data Avail
1074   = IsFree              -- Used for free Insts
1075   | Irred               -- Used for irreducible dictionaries,
1076                         -- which are going to be lambda bound
1077
1078   | Given TcId          -- Used for dictionaries for which we have a binding
1079                         -- e.g. those "given" in a signature
1080           Bool          -- True <=> actually consumed (splittable IPs only)
1081
1082   | NoRhs               -- Used for Insts like (CCallable f)
1083                         -- where no witness is required.
1084                         -- ToDo: remove?
1085
1086   | Rhs                 -- Used when there is a RHS
1087         TcExpr          -- The RHS
1088         [Inst]          -- Insts free in the RHS; we need these too
1089
1090   | Linear              -- Splittable Insts only.
1091         Int             -- The Int is always 2 or more; indicates how
1092                         -- many copies are required
1093         Inst            -- The splitter
1094         Avail           -- Where the "master copy" is
1095
1096   | LinRhss             -- Splittable Insts only; this is used only internally
1097                         --      by extractResults, where a Linear 
1098                         --      is turned into an LinRhss
1099         [TcExpr]        -- A supply of suitable RHSs
1100
1101 pprAvails avails = vcat [sep [ppr inst, nest 2 (equals <+> pprAvail avail)]
1102                         | (inst,avail) <- fmToList avails ]
1103
1104 instance Outputable Avail where
1105     ppr = pprAvail
1106
1107 pprAvail NoRhs          = text "<no rhs>"
1108 pprAvail IsFree         = text "Free"
1109 pprAvail Irred          = text "Irred"
1110 pprAvail (Given x b)    = text "Given" <+> ppr x <+> 
1111                           if b then text "(used)" else empty
1112 pprAvail (Rhs rhs bs)   = text "Rhs" <+> ppr rhs <+> braces (ppr bs)
1113 pprAvail (Linear n i a) = text "Linear" <+> ppr n <+> braces (ppr i) <+> ppr a
1114 pprAvail (LinRhss rhss) = text "LinRhss" <+> ppr rhss
1115 \end{code}
1116
1117 Extracting the bindings from a bunch of Avails.
1118 The bindings do *not* come back sorted in dependency order.
1119 We assume that they'll be wrapped in a big Rec, so that the
1120 dependency analyser can sort them out later
1121
1122 The loop startes
1123 \begin{code}
1124 extractResults :: Avails
1125                -> [Inst]                -- Wanted
1126                -> TcM (TcDictBinds,     -- Bindings
1127                           [Inst],       -- Irreducible ones
1128                           [Inst])       -- Free ones
1129
1130 extractResults avails wanteds
1131   = go avails EmptyMonoBinds [] [] wanteds
1132   where
1133     go avails binds irreds frees [] 
1134       = returnM (binds, irreds, frees)
1135
1136     go avails binds irreds frees (w:ws)
1137       = case lookupFM avails w of
1138           Nothing    -> pprTrace "Urk: extractResults" (ppr w) $
1139                         go avails binds irreds frees ws
1140
1141           Just NoRhs  -> go avails               binds irreds     frees     ws
1142           Just IsFree -> go (add_free avails w)  binds irreds     (w:frees) ws
1143           Just Irred  -> go (add_given avails w) binds (w:irreds) frees     ws
1144
1145           Just (Given id _) -> go avails new_binds irreds frees ws
1146                             where
1147                                new_binds | id == instToId w = binds
1148                                          | otherwise        = addBind binds w (HsVar id)
1149                 -- The sought Id can be one of the givens, via a superclass chain
1150                 -- and then we definitely don't want to generate an x=x binding!
1151
1152           Just (Rhs rhs ws') -> go (add_given avails w) new_binds irreds frees (ws' ++ ws)
1153                              where
1154                                 new_binds = addBind binds w rhs
1155
1156           Just (Linear n split_inst avail)      -- Transform Linear --> LinRhss
1157             -> get_root irreds frees avail w            `thenM` \ (irreds', frees', root_id) ->
1158                split n (instToId split_inst) root_id w  `thenM` \ (binds', rhss) ->
1159                go (addToFM avails w (LinRhss rhss))
1160                   (binds `AndMonoBinds` binds')
1161                   irreds' frees' (split_inst : w : ws)
1162
1163           Just (LinRhss (rhs:rhss))             -- Consume one of the Rhss
1164                 -> go new_avails new_binds irreds frees ws
1165                 where           
1166                    new_binds  = addBind binds w rhs
1167                    new_avails = addToFM avails w (LinRhss rhss)
1168
1169     get_root irreds frees (Given id _) w = returnM (irreds, frees, id)
1170     get_root irreds frees Irred        w = cloneDict w  `thenM` \ w' ->
1171                                            returnM (w':irreds, frees, instToId w')
1172     get_root irreds frees IsFree       w = cloneDict w  `thenM` \ w' ->
1173                                            returnM (irreds, w':frees, instToId w')
1174
1175     add_given avails w 
1176         | instBindingRequired w = addToFM avails w (Given (instToId w) True)
1177         | otherwise             = addToFM avails w NoRhs
1178         -- NB: make sure that CCallable/CReturnable use NoRhs rather
1179         --      than Given, else we end up with bogus bindings.
1180
1181     add_free avails w | isMethod w = avails
1182                       | otherwise  = add_given avails w
1183         -- NB: Hack alert!  
1184         -- Do *not* replace Free by Given if it's a method.
1185         -- The following situation shows why this is bad:
1186         --      truncate :: forall a. RealFrac a => forall b. Integral b => a -> b
1187         -- From an application (truncate f i) we get
1188         --      t1 = truncate at f
1189         --      t2 = t1 at i
1190         -- If we have also have a second occurrence of truncate, we get
1191         --      t3 = truncate at f
1192         --      t4 = t3 at i
1193         -- When simplifying with i,f free, we might still notice that
1194         --   t1=t3; but alas, the binding for t2 (which mentions t1)
1195         --   will continue to float out!
1196         -- (split n i a) returns: n rhss
1197         --                        auxiliary bindings
1198         --                        1 or 0 insts to add to irreds
1199
1200
1201 split :: Int -> TcId -> TcId -> Inst 
1202       -> TcM (TcDictBinds, [TcExpr])
1203 -- (split n split_id root_id wanted) returns
1204 --      * a list of 'n' expressions, all of which witness 'avail'
1205 --      * a bunch of auxiliary bindings to support these expressions
1206 --      * one or zero insts needed to witness the whole lot
1207 --        (maybe be zero if the initial Inst is a Given)
1208 --
1209 -- NB: 'wanted' is just a template
1210
1211 split n split_id root_id wanted
1212   = go n
1213   where
1214     ty      = linearInstType wanted
1215     pair_ty = mkTyConApp pairTyCon [ty,ty]
1216     id      = instToId wanted
1217     occ     = getOccName id
1218     loc     = getSrcLoc id
1219
1220     go 1 = returnM (EmptyMonoBinds, [HsVar root_id])
1221
1222     go n = go ((n+1) `div` 2)           `thenM` \ (binds1, rhss) ->
1223            expand n rhss                `thenM` \ (binds2, rhss') ->
1224            returnM (binds1 `AndMonoBinds` binds2, rhss')
1225
1226         -- (expand n rhss) 
1227         -- Given ((n+1)/2) rhss, make n rhss, using auxiliary bindings
1228         --  e.g.  expand 3 [rhs1, rhs2]
1229         --        = ( { x = split rhs1 },
1230         --            [fst x, snd x, rhs2] )
1231     expand n rhss
1232         | n `rem` 2 == 0 = go rhss      -- n is even
1233         | otherwise      = go (tail rhss)       `thenM` \ (binds', rhss') ->
1234                            returnM (binds', head rhss : rhss')
1235         where
1236           go rhss = mapAndUnzipM do_one rhss    `thenM` \ (binds', rhss') ->
1237                     returnM (andMonoBindList binds', concat rhss')
1238
1239           do_one rhs = newUnique                        `thenM` \ uniq -> 
1240                        tcLookupId fstName               `thenM` \ fst_id ->
1241                        tcLookupId sndName               `thenM` \ snd_id ->
1242                        let 
1243                           x = mkUserLocal occ uniq pair_ty loc
1244                        in
1245                        returnM (VarMonoBind x (mk_app split_id rhs),
1246                                     [mk_fs_app fst_id ty x, mk_fs_app snd_id ty x])
1247
1248 mk_fs_app id ty var = HsVar id `TyApp` [ty,ty] `HsApp` HsVar var
1249
1250 mk_app id rhs = HsApp (HsVar id) rhs
1251
1252 addBind binds inst rhs = binds `AndMonoBinds` VarMonoBind (instToId inst) rhs
1253 \end{code}
1254
1255
1256 %************************************************************************
1257 %*                                                                      *
1258 \subsection[reduce]{@reduce@}
1259 %*                                                                      *
1260 %************************************************************************
1261
1262 When the "what to do" predicate doesn't depend on the quantified type variables,
1263 matters are easier.  We don't need to do any zonking, unless the improvement step
1264 does something, in which case we zonk before iterating.
1265
1266 The "given" set is always empty.
1267
1268 \begin{code}
1269 simpleReduceLoop :: SDoc
1270                  -> (Inst -> WhatToDo)          -- What to do, *not* based on the quantified type variables
1271                  -> [Inst]                      -- Wanted
1272                  -> TcM ([Inst],                -- Free
1273                          TcDictBinds,
1274                          [Inst])                -- Irreducible
1275
1276 simpleReduceLoop doc try_me wanteds
1277   = mappM zonkInst wanteds                      `thenM` \ wanteds' ->
1278     reduceContext doc try_me [] wanteds'        `thenM` \ (no_improvement, frees, binds, irreds) ->
1279     if no_improvement then
1280         returnM (frees, binds, irreds)
1281     else
1282         simpleReduceLoop doc try_me (irreds ++ frees)   `thenM` \ (frees1, binds1, irreds1) ->
1283         returnM (frees1, binds `AndMonoBinds` binds1, irreds1)
1284 \end{code}
1285
1286
1287
1288 \begin{code}
1289 reduceContext :: SDoc
1290               -> (Inst -> WhatToDo)
1291               -> [Inst]                 -- Given
1292               -> [Inst]                 -- Wanted
1293               -> TcM (Bool,             -- True <=> improve step did no unification
1294                          [Inst],        -- Free
1295                          TcDictBinds,   -- Dictionary bindings
1296                          [Inst])        -- Irreducible
1297
1298 reduceContext doc try_me givens wanteds
1299   =
1300     traceTc (text "reduceContext" <+> (vcat [
1301              text "----------------------",
1302              doc,
1303              text "given" <+> ppr givens,
1304              text "wanted" <+> ppr wanteds,
1305              text "----------------------"
1306              ]))                                        `thenM_`
1307
1308         -- Build the Avail mapping from "givens"
1309     foldlM addGiven emptyFM givens                      `thenM` \ init_state ->
1310
1311         -- Do the real work
1312     reduceList (0,[]) try_me wanteds init_state         `thenM` \ avails ->
1313
1314         -- Do improvement, using everything in avails
1315         -- In particular, avails includes all superclasses of everything
1316     tcImprove avails                                    `thenM` \ no_improvement ->
1317
1318     extractResults avails wanteds                       `thenM` \ (binds, irreds, frees) ->
1319
1320     traceTc (text "reduceContext end" <+> (vcat [
1321              text "----------------------",
1322              doc,
1323              text "given" <+> ppr givens,
1324              text "wanted" <+> ppr wanteds,
1325              text "----",
1326              text "avails" <+> pprAvails avails,
1327              text "frees" <+> ppr frees,
1328              text "no_improvement =" <+> ppr no_improvement,
1329              text "----------------------"
1330              ]))                                        `thenM_`
1331
1332     returnM (no_improvement, frees, binds, irreds)
1333
1334 tcImprove :: Avails -> TcM Bool         -- False <=> no change
1335 -- Perform improvement using all the predicates in Avails
1336 tcImprove avails
1337  =  tcGetInstEnvs                       `thenM` \ (home_ie, pkg_ie) ->
1338     let
1339         preds = [ (pred, pp_loc)
1340                 | inst <- keysFM avails,
1341                   let pp_loc = pprInstLoc (instLoc inst),
1342                   pred <- fdPredsOfInst inst
1343                 ]
1344                 -- Avails has all the superclasses etc (good)
1345                 -- It also has all the intermediates of the deduction (good)
1346                 -- It does not have duplicates (good)
1347                 -- NB that (?x::t1) and (?x::t2) will be held separately in avails
1348                 --    so that improve will see them separate
1349         eqns = improve get_insts preds
1350         get_insts clas = classInstEnv home_ie clas ++ classInstEnv pkg_ie clas
1351      in
1352      if null eqns then
1353         returnM True
1354      else
1355         traceTc (ptext SLIT("Improve:") <+> vcat (map pprEquationDoc eqns))     `thenM_`
1356         mappM_ unify eqns       `thenM_`
1357         returnM False
1358   where
1359     unify ((qtvs, t1, t2), doc)
1360          = addErrCtxt doc                               $
1361            tcInstTyVars VanillaTv (varSetElems qtvs)    `thenM` \ (_, _, tenv) ->
1362            unifyTauTy (substTy tenv t1) (substTy tenv t2)
1363 \end{code}
1364
1365 The main context-reduction function is @reduce@.  Here's its game plan.
1366
1367 \begin{code}
1368 reduceList :: (Int,[Inst])              -- Stack (for err msgs)
1369                                         -- along with its depth
1370            -> (Inst -> WhatToDo)
1371            -> [Inst]
1372            -> Avails
1373            -> TcM Avails
1374 \end{code}
1375
1376 @reduce@ is passed
1377      try_me:    given an inst, this function returns
1378                   Reduce       reduce this
1379                   DontReduce   return this in "irreds"
1380                   Free         return this in "frees"
1381
1382      wanteds:   The list of insts to reduce
1383      state:     An accumulating parameter of type Avails
1384                 that contains the state of the algorithm
1385
1386   It returns a Avails.
1387
1388 The (n,stack) pair is just used for error reporting.
1389 n is always the depth of the stack.
1390 The stack is the stack of Insts being reduced: to produce X
1391 I had to produce Y, to produce Y I had to produce Z, and so on.
1392
1393 \begin{code}
1394 reduceList (n,stack) try_me wanteds state
1395   | n > opt_MaxContextReductionDepth
1396   = failWithTc (reduceDepthErr n stack)
1397
1398   | otherwise
1399   =
1400 #ifdef DEBUG
1401    (if n > 8 then
1402         pprTrace "Jeepers! ReduceContext:" (reduceDepthMsg n stack)
1403     else (\x->x))
1404 #endif
1405     go wanteds state
1406   where
1407     go []     state = returnM state
1408     go (w:ws) state = reduce (n+1, w:stack) try_me w state      `thenM` \ state' ->
1409                       go ws state'
1410
1411     -- Base case: we're done!
1412 reduce stack try_me wanted state
1413     -- It's the same as an existing inst, or a superclass thereof
1414   | Just avail <- isAvailable state wanted
1415   = if isLinearInst wanted then
1416         addLinearAvailable state avail wanted   `thenM` \ (state', wanteds') ->
1417         reduceList stack try_me wanteds' state'
1418     else
1419         returnM state           -- No op for non-linear things
1420
1421   | otherwise
1422   = case try_me wanted of {
1423
1424       DontReduce want_scs -> addIrred want_scs state wanted
1425
1426     ; DontReduceUnlessConstant ->    -- It's irreducible (or at least should not be reduced)
1427                                      -- First, see if the inst can be reduced to a constant in one step
1428         try_simple (addIrred AddSCs)    -- Assume want superclasses
1429
1430     ; Free ->   -- It's free so just chuck it upstairs
1431                 -- First, see if the inst can be reduced to a constant in one step
1432         try_simple addFree
1433
1434     ; ReduceMe ->               -- It should be reduced
1435         lookupInst wanted             `thenM` \ lookup_result ->
1436         case lookup_result of
1437             GenInst wanteds' rhs -> addWanted state wanted rhs wanteds'         `thenM` \ state' ->
1438                                     reduceList stack try_me wanteds' state'
1439                 -- Experiment with doing addWanted *before* the reduceList, 
1440                 -- which has the effect of adding the thing we are trying
1441                 -- to prove to the database before trying to prove the things it
1442                 -- needs.  See note [RECURSIVE DICTIONARIES]
1443
1444             SimpleInst rhs       -> addWanted state wanted rhs []
1445
1446             NoInstance ->    -- No such instance!
1447                              -- Add it and its superclasses
1448                              addIrred AddSCs state wanted
1449
1450     }
1451   where
1452     try_simple do_this_otherwise
1453       = lookupInst wanted         `thenM` \ lookup_result ->
1454         case lookup_result of
1455             SimpleInst rhs -> addWanted state wanted rhs []
1456             other          -> do_this_otherwise state wanted
1457 \end{code}
1458
1459
1460 \begin{code}
1461 -------------------------
1462 isAvailable :: Avails -> Inst -> Maybe Avail
1463 isAvailable avails wanted = lookupFM avails wanted
1464         -- NB 1: the Ord instance of Inst compares by the class/type info
1465         -- *not* by unique.  So
1466         --      d1::C Int ==  d2::C Int
1467
1468 addLinearAvailable :: Avails -> Avail -> Inst -> TcM (Avails, [Inst])
1469 addLinearAvailable avails avail wanted
1470         -- avails currently maps [wanted -> avail]
1471         -- Extend avails to reflect a neeed for an extra copy of avail
1472
1473   | Just avail' <- split_avail avail
1474   = returnM (addToFM avails wanted avail', [])
1475
1476   | otherwise
1477   = tcLookupId splitName                        `thenM` \ split_id ->
1478     tcInstClassOp (instLoc wanted) split_id 
1479                   [linearInstType wanted]       `thenM` \ split_inst ->
1480     returnM (addToFM avails wanted (Linear 2 split_inst avail), [split_inst])
1481
1482   where
1483     split_avail :: Avail -> Maybe Avail
1484         -- (Just av) if there's a modified version of avail that
1485         --           we can use to replace avail in avails
1486         -- Nothing   if there isn't, so we need to create a Linear
1487     split_avail (Linear n i a)              = Just (Linear (n+1) i a)
1488     split_avail (Given id used) | not used  = Just (Given id True)
1489                                 | otherwise = Nothing
1490     split_avail Irred                       = Nothing
1491     split_avail IsFree                      = Nothing
1492     split_avail other = pprPanic "addLinearAvailable" (ppr avail $$ ppr wanted $$ ppr avails)
1493                   
1494 -------------------------
1495 addFree :: Avails -> Inst -> TcM Avails
1496         -- When an Inst is tossed upstairs as 'free' we nevertheless add it
1497         -- to avails, so that any other equal Insts will be commoned up right
1498         -- here rather than also being tossed upstairs.  This is really just
1499         -- an optimisation, and perhaps it is more trouble that it is worth,
1500         -- as the following comments show!
1501         --
1502         -- NB: do *not* add superclasses.  If we have
1503         --      df::Floating a
1504         --      dn::Num a
1505         -- but a is not bound here, then we *don't* want to derive
1506         -- dn from df here lest we lose sharing.
1507         --
1508 addFree avails free = returnM (addToFM avails free IsFree)
1509
1510 addWanted :: Avails -> Inst -> TcExpr -> [Inst] -> TcM Avails
1511 addWanted avails wanted rhs_expr wanteds
1512   = ASSERT2( not (wanted `elemFM` avails), ppr wanted $$ ppr avails )
1513     addAvailAndSCs avails wanted avail
1514   where
1515     avail | instBindingRequired wanted = Rhs rhs_expr wanteds
1516           | otherwise                  = ASSERT( null wanteds ) NoRhs
1517
1518 addGiven :: Avails -> Inst -> TcM Avails
1519 addGiven state given = addAvailAndSCs state given (Given (instToId given) False)
1520         -- No ASSERT( not (given `elemFM` avails) ) because in an instance
1521         -- decl for Ord t we can add both Ord t and Eq t as 'givens', 
1522         -- so the assert isn't true
1523
1524 addIrred :: WantSCs -> Avails -> Inst -> TcM Avails
1525 addIrred NoSCs  avails irred = returnM (addToFM avails irred Irred)
1526 addIrred AddSCs avails irred = ASSERT2( not (irred `elemFM` avails), ppr irred $$ ppr avails )
1527                                addAvailAndSCs avails irred Irred
1528
1529 addAvailAndSCs :: Avails -> Inst -> Avail -> TcM Avails
1530 addAvailAndSCs avails inst avail
1531   | not (isClassDict inst) = returnM avails1
1532   | otherwise              = addSCs is_loop avails1 inst 
1533   where
1534     avails1 = addToFM avails inst avail
1535     is_loop inst = inst `elem` deps     -- Note: this compares by *type*, not by Unique
1536     deps         = findAllDeps avails avail
1537
1538 findAllDeps :: Avails -> Avail -> [Inst]
1539 -- Find all the Insts that this one depends on
1540 -- See Note [SUPERCLASS-LOOP]
1541 findAllDeps avails (Rhs _ kids) = kids ++ concat (map (find_all_deps_help avails) kids)
1542 findAllDeps avails other        = []
1543
1544 find_all_deps_help :: Avails -> Inst -> [Inst]
1545 find_all_deps_help avails inst
1546   = case lookupFM avails inst of
1547         Just avail -> findAllDeps avails avail
1548         Nothing    -> []
1549
1550 addSCs :: (Inst -> Bool) -> Avails -> Inst -> TcM Avails
1551         -- Add all the superclasses of the Inst to Avails
1552         -- The first param says "dont do this because the original thing
1553         --      depends on this one, so you'd build a loop"
1554         -- Invariant: the Inst is already in Avails.
1555
1556 addSCs is_loop avails dict
1557   = newDictsFromOld dict sc_theta'      `thenM` \ sc_dicts ->
1558     foldlM add_sc avails (zipEqual "add_scs" sc_dicts sc_sels)
1559   where
1560     (clas, tys) = getDictClassTys dict
1561     (tyvars, sc_theta, sc_sels, _) = classBigSig clas
1562     sc_theta' = substTheta (mkTopTyVarSubst tyvars tys) sc_theta
1563
1564     add_sc avails (sc_dict, sc_sel)     -- Add it, and its superclasses
1565       = case lookupFM avails sc_dict of
1566           Just (Given _ _) -> returnM avails    -- Given is cheaper than
1567                                                         --   a superclass selection
1568           Just other | is_loop sc_dict -> returnM avails        -- See Note [SUPERCLASS-LOOP]
1569                      | otherwise       -> returnM avails'       -- SCs already added
1570
1571           Nothing -> addSCs is_loop avails' sc_dict
1572       where
1573         sc_sel_rhs = DictApp (TyApp (HsVar sc_sel) tys) [instToId dict]
1574         avail      = Rhs sc_sel_rhs [dict]
1575         avails'    = addToFM avails sc_dict avail
1576 \end{code}
1577
1578 Note [SUPERCLASS-LOOP]: Checking for loops
1579 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
1580 We have to be careful here.  If we are *given* d1:Ord a,
1581 and want to deduce (d2:C [a]) where
1582
1583         class Ord a => C a where
1584         instance Ord a => C [a] where ...
1585
1586 Then we'll use the instance decl to deduce C [a] and then add the
1587 superclasses of C [a] to avails.  But we must not overwrite the binding
1588 for d1:Ord a (which is given) with a superclass selection or we'll just
1589 build a loop! 
1590
1591 Here's another example 
1592         class Eq b => Foo a b
1593         instance Eq a => Foo [a] a
1594 If we are reducing
1595         (Foo [t] t)
1596
1597 we'll first deduce that it holds (via the instance decl).  We must not
1598 then overwrite the Eq t constraint with a superclass selection!
1599
1600 At first I had a gross hack, whereby I simply did not add superclass constraints
1601 in addWanted, though I did for addGiven and addIrred.  This was sub-optimal,
1602 becuase it lost legitimate superclass sharing, and it still didn't do the job:
1603 I found a very obscure program (now tcrun021) in which improvement meant the
1604 simplifier got two bites a the cherry... so something seemed to be an Irred
1605 first time, but reducible next time.
1606
1607 Now we implement the Right Solution, which is to check for loops directly 
1608 when adding superclasses.  It's a bit like the occurs check in unification.
1609
1610
1611 Note [RECURSIVE DICTIONARIES]
1612 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
1613 Consider 
1614     data D r = ZeroD | SuccD (r (D r));
1615     
1616     instance (Eq (r (D r))) => Eq (D r) where
1617         ZeroD     == ZeroD     = True
1618         (SuccD a) == (SuccD b) = a == b
1619         _         == _         = False;
1620     
1621     equalDC :: D [] -> D [] -> Bool;
1622     equalDC = (==);
1623
1624 We need to prove (Eq (D [])).  Here's how we go:
1625
1626         d1 : Eq (D [])
1627
1628 by instance decl, holds if
1629         d2 : Eq [D []]
1630         where   d1 = dfEqD d2
1631
1632 by instance decl of Eq, holds if
1633         d3 : D []
1634         where   d2 = dfEqList d2
1635                 d1 = dfEqD d2
1636
1637 But now we can "tie the knot" to give
1638
1639         d3 = d1
1640         d2 = dfEqList d2
1641         d1 = dfEqD d2
1642
1643 and it'll even run!  The trick is to put the thing we are trying to prove
1644 (in this case Eq (D []) into the database before trying to prove its
1645 contributing clauses.
1646         
1647
1648 %************************************************************************
1649 %*                                                                      *
1650 \section{tcSimplifyTop: defaulting}
1651 %*                                                                      *
1652 %************************************************************************
1653
1654
1655 @tcSimplifyTop@ is called once per module to simplify all the constant
1656 and ambiguous Insts.
1657
1658 We need to be careful of one case.  Suppose we have
1659
1660         instance Num a => Num (Foo a b) where ...
1661
1662 and @tcSimplifyTop@ is given a constraint (Num (Foo x y)).  Then it'll simplify
1663 to (Num x), and default x to Int.  But what about y??
1664
1665 It's OK: the final zonking stage should zap y to (), which is fine.
1666
1667
1668 \begin{code}
1669 tcSimplifyTop, tcSimplifyInteractive :: [Inst] -> TcM TcDictBinds
1670 tcSimplifyTop         wanteds = tc_simplify_top False {- Not interactive loop -} wanteds
1671 tcSimplifyInteractive wanteds = tc_simplify_top True  {- Interactive loop -}     wanteds
1672
1673
1674 -- The TcLclEnv should be valid here, solely to improve
1675 -- error message generation for the monomorphism restriction
1676 tc_simplify_top is_interactive wanteds
1677   = getLclEnv                                                   `thenM` \ lcl_env ->
1678     traceTc (text "tcSimplifyTop" <+> ppr (lclEnvElts lcl_env)) `thenM_`
1679     simpleReduceLoop (text "tcSimplTop") reduceMe wanteds       `thenM` \ (frees, binds, irreds) ->
1680     ASSERT( null frees )
1681
1682     let
1683                 -- All the non-std ones are definite errors
1684         (stds, non_stds) = partition isStdClassTyVarDict irreds
1685
1686                 -- Group by type variable
1687         std_groups = equivClasses cmp_by_tyvar stds
1688
1689                 -- Pick the ones which its worth trying to disambiguate
1690                 -- namely, the onese whose type variable isn't bound
1691                 -- up with one of the non-standard classes
1692         (std_oks, std_bads)     = partition worth_a_try std_groups
1693         worth_a_try group@(d:_) = not (non_std_tyvars `intersectsVarSet` tyVarsOfInst d)
1694         non_std_tyvars          = unionVarSets (map tyVarsOfInst non_stds)
1695
1696                 -- Collect together all the bad guys
1697         bad_guys               = non_stds ++ concat std_bads
1698         (bad_ips, non_ips)     = partition isIPDict bad_guys
1699         (no_insts, ambigs)     = partition no_inst non_ips
1700         no_inst d              = not (isTyVarDict d) 
1701         -- Previously, there was a more elaborate no_inst definition:
1702         --      no_inst d = not (isTyVarDict d) || tyVarsOfInst d `subVarSet` fixed_tvs
1703         --      fixed_tvs = oclose (fdPredsOfInsts tidy_dicts) emptyVarSet
1704         -- But that seems over-elaborate to me; it only bites for class decls with
1705         -- fundeps like this:           class C a b | -> b where ...
1706     in
1707
1708         -- Report definite errors
1709     groupErrs (addNoInstanceErrs Nothing []) no_insts   `thenM_`
1710     addTopIPErrs bad_ips                                `thenM_`
1711
1712         -- Deal with ambiguity errors, but only if
1713         -- if there has not been an error so far; errors often
1714         -- give rise to spurious ambiguous Insts
1715     ifErrsM (returnM []) (
1716         
1717         -- Complain about the ones that don't fall under
1718         -- the Haskell rules for disambiguation
1719         -- This group includes both non-existent instances
1720         --      e.g. Num (IO a) and Eq (Int -> Int)
1721         -- and ambiguous dictionaries
1722         --      e.g. Num a
1723         addTopAmbigErrs ambigs          `thenM_`
1724
1725         -- Disambiguate the ones that look feasible
1726         mappM (disambigGroup is_interactive) std_oks
1727     )                                   `thenM` \ binds_ambig ->
1728
1729     returnM (binds `andMonoBinds` andMonoBindList binds_ambig)
1730
1731 ----------------------------------
1732 d1 `cmp_by_tyvar` d2 = get_tv d1 `compare` get_tv d2
1733
1734 get_tv d   = case getDictClassTys d of
1735                    (clas, [ty]) -> tcGetTyVar "tcSimplify" ty
1736 get_clas d = case getDictClassTys d of
1737                    (clas, [ty]) -> clas
1738 \end{code}
1739
1740 If a dictionary constrains a type variable which is
1741         * not mentioned in the environment
1742         * and not mentioned in the type of the expression
1743 then it is ambiguous. No further information will arise to instantiate
1744 the type variable; nor will it be generalised and turned into an extra
1745 parameter to a function.
1746
1747 It is an error for this to occur, except that Haskell provided for
1748 certain rules to be applied in the special case of numeric types.
1749 Specifically, if
1750         * at least one of its classes is a numeric class, and
1751         * all of its classes are numeric or standard
1752 then the type variable can be defaulted to the first type in the
1753 default-type list which is an instance of all the offending classes.
1754
1755 So here is the function which does the work.  It takes the ambiguous
1756 dictionaries and either resolves them (producing bindings) or
1757 complains.  It works by splitting the dictionary list by type
1758 variable, and using @disambigOne@ to do the real business.
1759
1760 @disambigOne@ assumes that its arguments dictionaries constrain all
1761 the same type variable.
1762
1763 ADR Comment 20/6/94: I've changed the @CReturnable@ case to default to
1764 @()@ instead of @Int@.  I reckon this is the Right Thing to do since
1765 the most common use of defaulting is code like:
1766 \begin{verbatim}
1767         _ccall_ foo     `seqPrimIO` bar
1768 \end{verbatim}
1769 Since we're not using the result of @foo@, the result if (presumably)
1770 @void@.
1771
1772 \begin{code}
1773 disambigGroup :: Bool   -- True <=> simplifying at top-level interactive loop
1774               -> [Inst] -- All standard classes of form (C a)
1775               -> TcM TcDictBinds
1776
1777 disambigGroup is_interactive dicts
1778   |   any std_default_class classes     -- Guaranteed all standard classes
1779   =     -- THE DICTS OBEY THE DEFAULTABLE CONSTRAINT
1780         -- SO, TRY DEFAULT TYPES IN ORDER
1781
1782         -- Failure here is caused by there being no type in the
1783         -- default list which can satisfy all the ambiguous classes.
1784         -- For example, if Real a is reqd, but the only type in the
1785         -- default list is Int.
1786     get_default_tys                     `thenM` \ default_tys ->
1787     let
1788       try_default []    -- No defaults work, so fail
1789         = failM
1790
1791       try_default (default_ty : default_tys)
1792         = tryTcLIE_ (try_default default_tys) $ -- If default_ty fails, we try
1793                                                 -- default_tys instead
1794           tcSimplifyDefault theta               `thenM` \ _ ->
1795           returnM default_ty
1796         where
1797           theta = [mkClassPred clas [default_ty] | clas <- classes]
1798     in
1799         -- See if any default works
1800     tryM (try_default default_tys)      `thenM` \ mb_ty ->
1801     case mb_ty of
1802         Left  _                 -> bomb_out
1803         Right chosen_default_ty -> choose_default chosen_default_ty
1804
1805   | otherwise                           -- No defaults
1806   = bomb_out
1807
1808   where
1809     tyvar   = get_tv (head dicts)       -- Should be non-empty
1810     classes = map get_clas dicts
1811
1812     std_default_class cls
1813       =  isNumericClass cls
1814       || (is_interactive && 
1815           classKey cls `elem` [showClassKey, eqClassKey, ordClassKey])
1816                 -- In interactive mode, we default Show a to Show ()
1817                 -- to avoid graututious errors on "show []"
1818
1819     choose_default default_ty   -- Commit to tyvar = default_ty
1820       = -- Bind the type variable 
1821         unifyTauTy default_ty (mkTyVarTy tyvar) `thenM_`
1822         -- and reduce the context, for real this time
1823         simpleReduceLoop (text "disambig" <+> ppr dicts)
1824                      reduceMe dicts                     `thenM` \ (frees, binds, ambigs) ->
1825         WARN( not (null frees && null ambigs), ppr frees $$ ppr ambigs )
1826         warnDefault dicts default_ty                    `thenM_`
1827         returnM binds
1828
1829     bomb_out = addTopAmbigErrs dicts    `thenM_`
1830                returnM EmptyMonoBinds
1831
1832 get_default_tys
1833   = do  { mb_defaults <- getDefaultTys
1834         ; case mb_defaults of
1835                 Just tys -> return tys
1836                 Nothing  ->     -- No use-supplied default;
1837                                 -- use [Integer, Double]
1838                             do { integer_ty <- tcMetaTy integerTyConName
1839                                ; return [integer_ty, doubleTy] } }
1840 \end{code}
1841
1842 [Aside - why the defaulting mechanism is turned off when
1843  dealing with arguments and results to ccalls.
1844
1845 When typechecking _ccall_s, TcExpr ensures that the external
1846 function is only passed arguments (and in the other direction,
1847 results) of a restricted set of 'native' types. This is
1848 implemented via the help of the pseudo-type classes,
1849 @CReturnable@ (CR) and @CCallable@ (CC.)
1850
1851 The interaction between the defaulting mechanism for numeric
1852 values and CC & CR can be a bit puzzling to the user at times.
1853 For example,
1854
1855     x <- _ccall_ f
1856     if (x /= 0) then
1857        _ccall_ g x
1858      else
1859        return ()
1860
1861 What type has 'x' got here? That depends on the default list
1862 in operation, if it is equal to Haskell 98's default-default
1863 of (Integer, Double), 'x' has type Double, since Integer
1864 is not an instance of CR. If the default list is equal to
1865 Haskell 1.4's default-default of (Int, Double), 'x' has type
1866 Int.
1867
1868 To try to minimise the potential for surprises here, the
1869 defaulting mechanism is turned off in the presence of
1870 CCallable and CReturnable.
1871
1872 End of aside]
1873
1874
1875 %************************************************************************
1876 %*                                                                      *
1877 \subsection[simple]{@Simple@ versions}
1878 %*                                                                      *
1879 %************************************************************************
1880
1881 Much simpler versions when there are no bindings to make!
1882
1883 @tcSimplifyThetas@ simplifies class-type constraints formed by
1884 @deriving@ declarations and when specialising instances.  We are
1885 only interested in the simplified bunch of class/type constraints.
1886
1887 It simplifies to constraints of the form (C a b c) where
1888 a,b,c are type variables.  This is required for the context of
1889 instance declarations.
1890
1891 \begin{code}
1892 tcSimplifyDeriv :: [TyVar]      
1893                 -> ThetaType            -- Wanted
1894                 -> TcM ThetaType        -- Needed
1895
1896 tcSimplifyDeriv tyvars theta
1897   = tcInstTyVars VanillaTv tyvars                       `thenM` \ (tvs, _, tenv) ->
1898         -- The main loop may do unification, and that may crash if 
1899         -- it doesn't see a TcTyVar, so we have to instantiate. Sigh
1900         -- ToDo: what if two of them do get unified?
1901     newDicts DataDeclOrigin (substTheta tenv theta)     `thenM` \ wanteds ->
1902     simpleReduceLoop doc reduceMe wanteds               `thenM` \ (frees, _, irreds) ->
1903     ASSERT( null frees )                        -- reduceMe never returns Free
1904
1905     doptM Opt_AllowUndecidableInstances         `thenM` \ undecidable_ok ->
1906     let
1907         tv_set      = mkVarSet tvs
1908         simpl_theta = map dictPred irreds       -- reduceMe squashes all non-dicts
1909
1910         check_pred pred
1911           | isEmptyVarSet pred_tyvars   -- Things like (Eq T) should be rejected
1912           = addErrTc (noInstErr pred)
1913
1914           | not undecidable_ok && not (isTyVarClassPred pred)
1915           -- Check that the returned dictionaries are all of form (C a b)
1916           --    (where a, b are type variables).  
1917           -- We allow this if we had -fallow-undecidable-instances,
1918           -- but note that risks non-termination in the 'deriving' context-inference
1919           -- fixpoint loop.   It is useful for situations like
1920           --    data Min h a = E | M a (h a)
1921           -- which gives the instance decl
1922           --    instance (Eq a, Eq (h a)) => Eq (Min h a)
1923           = addErrTc (noInstErr pred)
1924   
1925           | not (pred_tyvars `subVarSet` tv_set) 
1926           -- Check for a bizarre corner case, when the derived instance decl should
1927           -- have form  instance C a b => D (T a) where ...
1928           -- Note that 'b' isn't a parameter of T.  This gives rise to all sorts
1929           -- of problems; in particular, it's hard to compare solutions for
1930           -- equality when finding the fixpoint.  So I just rule it out for now.
1931           = addErrTc (badDerivedPred pred)
1932   
1933           | otherwise
1934           = returnM ()
1935           where
1936             pred_tyvars = tyVarsOfPred pred
1937
1938         rev_env = mkTopTyVarSubst tvs (mkTyVarTys tyvars)
1939                 -- This reverse-mapping is a Royal Pain, 
1940                 -- but the result should mention TyVars not TcTyVars
1941     in
1942    
1943     mappM check_pred simpl_theta                `thenM_`
1944     checkAmbiguity tvs simpl_theta tv_set       `thenM_`
1945     returnM (substTheta rev_env simpl_theta)
1946   where
1947     doc    = ptext SLIT("deriving classes for a data type")
1948 \end{code}
1949
1950 @tcSimplifyDefault@ just checks class-type constraints, essentially;
1951 used with \tr{default} declarations.  We are only interested in
1952 whether it worked or not.
1953
1954 \begin{code}
1955 tcSimplifyDefault :: ThetaType  -- Wanted; has no type variables in it
1956                   -> TcM ()
1957
1958 tcSimplifyDefault theta
1959   = newDicts DataDeclOrigin theta               `thenM` \ wanteds ->
1960     simpleReduceLoop doc reduceMe wanteds       `thenM` \ (frees, _, irreds) ->
1961     ASSERT( null frees )        -- try_me never returns Free
1962     mappM (addErrTc . noInstErr) irreds         `thenM_`
1963     if null irreds then
1964         returnM ()
1965     else
1966         failM
1967   where
1968     doc = ptext SLIT("default declaration")
1969 \end{code}
1970
1971
1972 %************************************************************************
1973 %*                                                                      *
1974 \section{Errors and contexts}
1975 %*                                                                      *
1976 %************************************************************************
1977
1978 ToDo: for these error messages, should we note the location as coming
1979 from the insts, or just whatever seems to be around in the monad just
1980 now?
1981
1982 \begin{code}
1983 groupErrs :: ([Inst] -> TcM ()) -- Deal with one group
1984           -> [Inst]             -- The offending Insts
1985           -> TcM ()
1986 -- Group together insts with the same origin
1987 -- We want to report them together in error messages
1988
1989 groupErrs report_err [] 
1990   = returnM ()
1991 groupErrs report_err (inst:insts) 
1992   = do_one (inst:friends)               `thenM_`
1993     groupErrs report_err others
1994
1995   where
1996         -- (It may seem a bit crude to compare the error messages,
1997         --  but it makes sure that we combine just what the user sees,
1998         --  and it avoids need equality on InstLocs.)
1999    (friends, others) = partition is_friend insts
2000    loc_msg           = showSDoc (pprInstLoc (instLoc inst))
2001    is_friend friend  = showSDoc (pprInstLoc (instLoc friend)) == loc_msg
2002    do_one insts = addInstCtxt (instLoc (head insts)) (report_err insts)
2003                 -- Add location and context information derived from the Insts
2004
2005 -- Add the "arising from..." part to a message about bunch of dicts
2006 addInstLoc :: [Inst] -> Message -> Message
2007 addInstLoc insts msg = msg $$ nest 2 (pprInstLoc (instLoc (head insts)))
2008
2009 plural [x] = empty
2010 plural xs  = char 's'
2011
2012 addTopIPErrs dicts
2013   = groupErrs report tidy_dicts
2014   where
2015     (tidy_env, tidy_dicts) = tidyInsts dicts
2016     report dicts = addErrTcM (tidy_env, mk_msg dicts)
2017     mk_msg dicts = addInstLoc dicts (ptext SLIT("Unbound implicit parameter") <> 
2018                                      plural tidy_dicts <+> pprInsts tidy_dicts)
2019
2020 addNoInstanceErrs :: Maybe SDoc -- Nothing => top level
2021                                 -- Just d => d describes the construct
2022                   -> [Inst]     -- What is given by the context or type sig
2023                   -> [Inst]     -- What is wanted
2024                   -> TcM ()     
2025 addNoInstanceErrs mb_what givens [] 
2026   = returnM ()
2027 addNoInstanceErrs mb_what givens dicts
2028   =     -- Some of the dicts are here because there is no instances
2029         -- and some because there are too many instances (overlap)
2030         -- The first thing we do is separate them
2031     getDOpts            `thenM` \ dflags ->
2032     tcGetInstEnvs       `thenM` \ inst_envs ->
2033     let
2034         (tidy_env1, tidy_givens) = tidyInsts givens
2035         (tidy_env2, tidy_dicts)  = tidyMoreInsts tidy_env1 dicts
2036
2037         -- Run through the dicts, generating a message for each
2038         -- overlapping one, but simply accumulating all the 
2039         -- no-instance ones so they can be reported as a group
2040         (overlap_doc, no_inst_dicts) = foldl check_overlap (empty, []) tidy_dicts
2041         check_overlap (overlap_doc, no_inst_dicts) dict 
2042           | not (isClassDict dict) = (overlap_doc, dict : no_inst_dicts)
2043           | otherwise
2044           = case lookupInstEnv dflags inst_envs clas tys of
2045                 ([], _)  -> (overlap_doc, dict : no_inst_dicts)         -- No matches
2046                 inst_res -> (mk_overlap_msg dict inst_res $$ overlap_doc, no_inst_dicts)
2047           where
2048             (clas,tys) = getDictClassTys dict
2049     in
2050     mk_probable_fix tidy_env2 mb_what no_inst_dicts     `thenM` \ (tidy_env3, probable_fix) ->
2051     let
2052         no_inst_doc | null no_inst_dicts = empty
2053                     | otherwise = vcat [addInstLoc no_inst_dicts heading, probable_fix]
2054         heading | null givens = ptext SLIT("No instance") <> plural no_inst_dicts <+> 
2055                                 ptext SLIT("for") <+> pprInsts no_inst_dicts
2056                 | otherwise   = sep [ptext SLIT("Could not deduce") <+> pprInsts no_inst_dicts,
2057                                      nest 2 $ ptext SLIT("from the context") <+> pprInsts tidy_givens]
2058     in
2059     addErrTcM (tidy_env3, no_inst_doc $$ overlap_doc)
2060  
2061   where
2062     mk_overlap_msg dict (matches, unifiers)
2063       = vcat [  addInstLoc [dict] ((ptext SLIT("Overlapping instances for") <+> ppr dict)),
2064                 sep [ptext SLIT("Matching instances") <> colon,
2065                      nest 2 (pprDFuns (dfuns ++ unifiers))],
2066                 if null unifiers 
2067                 then empty
2068                 else parens (ptext SLIT("The choice depends on the instantiation of") <+>
2069                              quotes (pprWithCommas ppr (varSetElems (tyVarsOfInst dict))))]
2070       where
2071         dfuns = [df | (_, (_,_,df)) <- matches]
2072
2073     mk_probable_fix tidy_env Nothing dicts      -- Top level
2074       = mkMonomorphismMsg tidy_env dicts
2075     mk_probable_fix tidy_env (Just what) dicts  -- Nested (type signatures, instance decls)
2076       = returnM (tidy_env, sep [ptext SLIT("Probable fix:"), nest 2 fix1, nest 2 fix2])
2077       where
2078         fix1 = sep [ptext SLIT("Add") <+> pprInsts dicts,
2079                     ptext SLIT("to the") <+> what]
2080
2081         fix2 | null instance_dicts = empty
2082              | otherwise           = ptext SLIT("Or add an instance declaration for")
2083                                      <+> pprInsts instance_dicts
2084         instance_dicts = [d | d <- dicts, isClassDict d, not (isTyVarDict d)]
2085                 -- Insts for which it is worth suggesting an adding an instance declaration
2086                 -- Exclude implicit parameters, and tyvar dicts
2087
2088
2089 addTopAmbigErrs dicts
2090 -- Divide into groups that share a common set of ambiguous tyvars
2091   = mapM report (equivClasses cmp [(d, tvs_of d) | d <- tidy_dicts])
2092   where
2093     (tidy_env, tidy_dicts) = tidyInsts dicts
2094
2095     tvs_of :: Inst -> [TcTyVar]
2096     tvs_of d = varSetElems (tyVarsOfInst d)
2097     cmp (_,tvs1) (_,tvs2) = tvs1 `compare` tvs2
2098     
2099     report :: [(Inst,[TcTyVar])] -> TcM ()
2100     report pairs@((_,tvs) : _)  -- The pairs share a common set of ambiguous tyvars
2101         = mkMonomorphismMsg tidy_env dicts      `thenM` \ (tidy_env, mono_msg) ->
2102           addErrTcM (tidy_env, msg $$ mono_msg)
2103         where
2104           dicts = map fst pairs
2105           msg = sep [text "Ambiguous type variable" <> plural tvs <+> 
2106                              pprQuotedList tvs <+> in_msg,
2107                      nest 2 (pprInstsInFull dicts)]
2108           in_msg | isSingleton dicts = text "in the top-level constraint:"
2109                  | otherwise         = text "in these top-level constraints:"
2110
2111
2112 mkMonomorphismMsg :: TidyEnv -> [Inst] -> TcM (TidyEnv, Message)
2113 -- There's an error with these Insts; if they have free type variables
2114 -- it's probably caused by the monomorphism restriction. 
2115 -- Try to identify the offending variable
2116 -- ASSUMPTION: the Insts are fully zonked
2117 mkMonomorphismMsg tidy_env insts
2118   | isEmptyVarSet inst_tvs
2119   = returnM (tidy_env, empty)
2120   | otherwise
2121   = findGlobals inst_tvs tidy_env       `thenM` \ (tidy_env, docs) ->
2122     returnM (tidy_env, mk_msg docs)
2123
2124   where
2125     inst_tvs = tyVarsOfInsts insts
2126
2127     mk_msg []   = empty         -- This happens in things like
2128                                 --      f x = show (read "foo")
2129                                 -- whre monomorphism doesn't play any role
2130     mk_msg docs = vcat [ptext SLIT("Possible cause: the monomorphism restriction applied to the following:"),
2131                         nest 2 (vcat docs),
2132                         ptext SLIT("Probable fix: give these definition(s) an explicit type signature")]
2133     
2134 warnDefault dicts default_ty
2135   = doptM Opt_WarnTypeDefaults  `thenM` \ warn_flag ->
2136     addInstCtxt (instLoc (head dicts)) (warnTc warn_flag warn_msg)
2137   where
2138         -- Tidy them first
2139     (_, tidy_dicts) = tidyInsts dicts
2140     warn_msg  = vcat [ptext SLIT("Defaulting the following constraint(s) to type") <+>
2141                                 quotes (ppr default_ty),
2142                       pprInstsInFull tidy_dicts]
2143
2144 -- Used for the ...Thetas variants; all top level
2145 noInstErr pred = ptext SLIT("No instance for") <+> quotes (ppr pred)
2146
2147 badDerivedPred pred
2148   = vcat [ptext SLIT("Can't derive instances where the instance context mentions"),
2149           ptext SLIT("type variables that are not data type parameters"),
2150           nest 2 (ptext SLIT("Offending constraint:") <+> ppr pred)]
2151
2152 reduceDepthErr n stack
2153   = vcat [ptext SLIT("Context reduction stack overflow; size =") <+> int n,
2154           ptext SLIT("Use -fcontext-stack20 to increase stack size to (e.g.) 20"),
2155           nest 4 (pprInstsInFull stack)]
2156
2157 reduceDepthMsg n stack = nest 4 (pprInstsInFull stack)
2158 \end{code}