db5d597e0403aea40df63aa0ee0fdd58b33a665d
[ghc-hetmet.git] / ghc / compiler / types / Class.lhs
1 %
2 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
3 %
4 \section[Class]{The @Class@ datatype}
5
6 \begin{code}
7 module Class (
8         Class, ClassOpItem, FunDep,
9         DefMeth (..),
10
11         mkClass, classTyVars, classArity,
12         classKey, className, classSelIds, classTyCon,
13         classBigSig, classExtraBigSig, classTvsFds, classSCTheta
14     ) where
15
16 #include "HsVersions.h"
17
18 import {-# SOURCE #-} TyCon     ( TyCon )
19 import {-# SOURCE #-} TypeRep   ( PredType )
20
21 import Var              ( Id, TyVar )
22 import Name             ( NamedThing(..), Name )
23 import BasicTypes       ( Arity )
24 import Unique           ( Unique, Uniquable(..) )
25 import Outputable
26 import Util             ( notNull )
27 \end{code}
28
29 %************************************************************************
30 %*                                                                      *
31 \subsection[Class-basic]{@Class@: basic definition}
32 %*                                                                      *
33 %************************************************************************
34
35 A @Class@ corresponds to a Greek kappa in the static semantics:
36
37 \begin{code}
38 data Class
39   = Class {
40         classKey  :: Unique,                    -- Key for fast comparison
41         className :: Name,
42         
43         classTyVars  :: [TyVar],                -- The class type variables
44         classFunDeps :: [FunDep TyVar],         -- The functional dependencies
45
46         classSCTheta :: [PredType],             -- Immediate superclasses, and the
47         classSCSels  :: [Id],                   -- corresponding selector functions to
48                                                 -- extract them from a dictionary of this
49                                                 -- class
50
51         classOpStuff :: [ClassOpItem],          -- Ordered by tag
52
53         classTyCon :: TyCon             -- The data type constructor for dictionaries
54   }                                     -- of this class
55
56 type FunDep a     = ([a],[a])   --  e.g. class C a b c |  a b -> c, a c -> b  where ...
57                                 --  Here fun-deps are [([a,b],[c]), ([a,c],[b])]
58
59 type ClassOpItem = (Id, DefMeth)
60         -- Selector function; contains unfolding
61         -- Default-method info
62
63 data DefMeth = NoDefMeth                -- No default method
64              | DefMeth                  -- A polymorphic default method
65              | GenDefMeth               -- A generic default method
66              deriving Eq  
67 \end{code}
68
69 The @mkClass@ function fills in the indirect superclasses.
70
71 \begin{code}
72 mkClass :: Name -> [TyVar]
73         -> [([TyVar], [TyVar])]
74         -> [PredType] -> [Id]
75         -> [ClassOpItem]
76         -> TyCon
77         -> Class
78
79 mkClass name tyvars fds super_classes superdict_sels
80         op_stuff tycon
81   = Class {     classKey = getUnique name, 
82                 className = name,
83                 classTyVars = tyvars,
84                 classFunDeps = fds,
85                 classSCTheta = super_classes,
86                 classSCSels = superdict_sels,
87                 classOpStuff = op_stuff,
88                 classTyCon = tycon }
89 \end{code}
90
91 %************************************************************************
92 %*                                                                      *
93 \subsection[Class-selectors]{@Class@: simple selectors}
94 %*                                                                      *
95 %************************************************************************
96
97 The rest of these functions are just simple selectors.
98
99 \begin{code}
100 classArity :: Class -> Arity
101 classArity clas = length (classTyVars clas)
102         -- Could memoise this
103
104 classSelIds (Class {classSCSels = sc_sels, classOpStuff = op_stuff})
105   = sc_sels ++ [op_sel | (op_sel, _) <- op_stuff]
106
107 classTvsFds c
108   = (classTyVars c, classFunDeps c)
109
110 classBigSig (Class {classTyVars = tyvars, classSCTheta = sc_theta, 
111                     classSCSels = sc_sels, classOpStuff = op_stuff})
112   = (tyvars, sc_theta, sc_sels, op_stuff)
113 classExtraBigSig (Class {classTyVars = tyvars, classFunDeps = fundeps,
114                          classSCTheta = sc_theta, classSCSels = sc_sels,
115                          classOpStuff = op_stuff})
116   = (tyvars, fundeps, sc_theta, sc_sels, op_stuff)
117 \end{code}
118
119
120 %************************************************************************
121 %*                                                                      *
122 \subsection[Class-instances]{Instance declarations for @Class@}
123 %*                                                                      *
124 %************************************************************************
125
126 We compare @Classes@ by their keys (which include @Uniques@).
127
128 \begin{code}
129 instance Eq Class where
130     c1 == c2 = classKey c1 == classKey c2
131     c1 /= c2 = classKey c1 /= classKey c2
132
133 instance Ord Class where
134     c1 <= c2 = classKey c1 <= classKey c2
135     c1 <  c2 = classKey c1 <  classKey c2
136     c1 >= c2 = classKey c1 >= classKey c2
137     c1 >  c2 = classKey c1 >  classKey c2
138     compare c1 c2 = classKey c1 `compare` classKey c2
139 \end{code}
140
141 \begin{code}
142 instance Uniquable Class where
143     getUnique c = classKey c
144
145 instance NamedThing Class where
146     getName clas = className clas
147
148 instance Outputable Class where
149     ppr c = ppr (getName c)
150
151 instance Show Class where
152     showsPrec p c = showsPrecSDoc p (ppr c)
153
154 instance Outputable DefMeth where
155     ppr DefMeth     =  text "{- has default method -}"
156     ppr GenDefMeth  =  text "{- has generic method -}"
157     ppr NoDefMeth   =  empty   -- No default method
158 \end{code}
159
160