From 878924ac03219f02c857b0c3a95bb47a4a427d55 Mon Sep 17 00:00:00 2001
From: Manuel M T Chakravarty <chak@cse.unsw.edu.au>
Date: Wed, 27 Jun 2007 05:48:34 +0000
Subject: [PATCH] Fixed deriving of associated data types - We forgot to pull
 the data declarations nested in class instances out of  
 the instances when collecting all the predicates that we
 need derive.   Thanks to Roman for spotting this.

---
 compiler/typecheck/TcDeriv.lhs    |   21 +++++++++++++++------
 compiler/typecheck/TcInstDcls.lhs |    5 ++++-
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/compiler/typecheck/TcDeriv.lhs b/compiler/typecheck/TcDeriv.lhs
index e26c97d..ba11079 100644
--- a/compiler/typecheck/TcDeriv.lhs
+++ b/compiler/typecheck/TcDeriv.lhs
@@ -205,16 +205,18 @@ And then translate it to:
 %************************************************************************
 
 \begin{code}
-tcDeriving  :: [LTyClDecl Name]	-- All type constructors
+tcDeriving  :: [LTyClDecl Name]  -- All type constructors
+            -> [LInstDecl Name]  -- All instance declarations
             -> [LDerivDecl Name] -- All stand-alone deriving declarations
 	    -> TcM ([InstInfo],		-- The generated "instance decls"
 		    HsValBinds Name)	-- Extra generated top-level bindings
 
-tcDeriving tycl_decls deriv_decls
+tcDeriving tycl_decls inst_decls deriv_decls
   = recoverM (returnM ([], emptyValBindsOut)) $
     do	{   	-- Fish the "deriving"-related information out of the TcEnv
 		-- and make the necessary "equations".
-	; (ordinary_eqns, newtype_inst_info) <- makeDerivEqns tycl_decls deriv_decls
+	; (ordinary_eqns, newtype_inst_info) 
+                <- makeDerivEqns tycl_decls inst_decls deriv_decls
 
 	; (ordinary_inst_info, deriv_binds) 
 		<- extendLocalInstEnv (map iSpec newtype_inst_info)  $
@@ -338,17 +340,24 @@ when the dict is constructed in TcInstDcls.tcInstDecl2
 
 \begin{code}
 makeDerivEqns :: [LTyClDecl Name] 
+              -> [LInstDecl Name]
 	      -> [LDerivDecl Name] 
 	      -> TcM ([DerivEqn],	-- Ordinary derivings
 		      [InstInfo])	-- Special newtype derivings
 
-makeDerivEqns tycl_decls deriv_decls
+makeDerivEqns tycl_decls inst_decls deriv_decls
   = do	{ eqns1 <- mapM deriveTyData $
-		   [ (p,d) | d@(L _ (TyData {tcdDerivs = Just preds})) <- tycl_decls
-			   , p <- preds ]
+                     extractTyDataPreds tycl_decls ++
+		     [ pd                        -- traverse assoc data families
+                     | L _ (InstDecl _ _ _ ats) <- inst_decls
+		     , pd <- extractTyDataPreds ats ]
 	; eqns2 <- mapM deriveStandalone deriv_decls
 	; return ([eqn  | (Just eqn, _)  <- eqns1 ++ eqns2],
 		  [inst | (_, Just inst) <- eqns1 ++ eqns2]) }
+  where
+    extractTyDataPreds decls = 		   
+      [(p, d) | d@(L _ (TyData {tcdDerivs = Just preds})) <- decls, p <- preds]
+
 
 ------------------------------------------------------------------
 deriveStandalone :: LDerivDecl Name -> TcM (Maybe DerivEqn, Maybe InstInfo)
diff --git a/compiler/typecheck/TcInstDcls.lhs b/compiler/typecheck/TcInstDcls.lhs
index 0dbb775..d314e1e 100644
--- a/compiler/typecheck/TcInstDcls.lhs
+++ b/compiler/typecheck/TcInstDcls.lhs
@@ -179,7 +179,10 @@ tcInstDecls1 tycl_decls inst_decls deriv_decls
 	        -- (4) Compute instances from "deriving" clauses; 
 		-- This stuff computes a context for the derived instance
 		-- decl, so it needs to know about all the instances possible
-       ; (deriv_inst_info, deriv_binds) <- tcDeriving tycl_decls deriv_decls
+                -- NB: class instance declarations can contain derivings as
+                --     part of associated data type declarations
+       ; (deriv_inst_info, deriv_binds) <- tcDeriving tycl_decls inst_decls 
+                                                      deriv_decls
        ; addInsts deriv_inst_info   $ do {
 
        ; gbl_env <- getGblEnv
-- 
1.7.10.4