@@ -1,32 +1,27 @@
1−{-# LANGUAGE GADTs, DataKinds, KindSignatures, TypeOperators, TypeFamilies,
2− MultiParamTypeClasses, FlexibleInstances, PolyKinds,
3− FlexibleContexts, UndecidableInstances, ConstraintKinds,
4− ScopedTypeVariables, TypeInType #-}
1+{-# LANGUAGE GADTs, DataKinds, KindSignatures, TypeOperators, TypeFamilies,
2+ MultiParamTypeClasses, FlexibleInstances, PolyKinds, FlexibleContexts,
3+ UndecidableInstances, ConstraintKinds, OverlappingInstances, ScopedTypeVariables #-}
54
6−module Data.Type.Set (Set(..), Union, Unionable, union, quicksort, append,
7− Sort, Sortable, (:++), Split(..), Cmp, Filter, Flag(..),
5+module Data.Type.Set (Set(..), Union, Unionable, union, quicksort, append,
6+ Sort, Sortable, Append(..), Split(..), Cmp,
87 Nub, Nubable(..), AsSet, asSet, IsSet, Subset(..),
9− Delete(..), Proxy(..), remove, Remove, (:\),
10− Member(..), NonMember, MemberP(..)) where
8+ (:->)(..), Var(..)) where
119
1210import GHC.TypeLits
1311import Data.Type.Bool
1412import Data.Type.Equality
13+import Data.Proxy
1514
16−data Proxy (p :: k) = Proxy
17−
18−-- Value-level 'Set' representation, essentially a list
19−data Set (n :: [k]) where
20− {--| Construct an empty set -}
15+{-| Core Set definition, in terms of lists -}
16+data Set (n :: [*]) where
2117 Empty :: Set '[]
22− {--| Extend a set with an element -}
2318 Ext :: e -> Set s -> Set (e ': s)
2419
2520instance Show (Set '[]) where
2621 show Empty = "{}"
2722
2823instance (Show e, Show' (Set s)) => Show (Set (e ': s)) where
29− show (Ext e s) = "{" ++ show e ++ (show' s) ++ "}"
24+ show (Ext e s) = "{" ++ show e ++ (show' s) ++ "}"
3025
3126class Show' t where
3227 show' :: t -> String
@@ -35,10 +30,6 @@ instance Show' (Set '[]) where
3530instance (Show' (Set s), Show e) => Show' (Set (e ': s)) where
3631 show' (Ext e s) = ", " ++ show e ++ (show' s)
3732
38−instance (Eq e, Eq (Set s)) => Eq (Set (e ': s)) where
39− (Ext e m) == (Ext e' m') = e == e' && m == m'
40−
41−
4233{-| At the type level, normalise the list form to the set form -}
4334type AsSet s = Nub (Sort s)
4435
@@ -50,68 +41,54 @@ asSet x = nub (quicksort x)
5041type IsSet s = (s ~ Nub (Sort s))
5142
5243{-| Useful properties to be able to refer to someties -}
53−type SetProperties f = (Union f '[] ~ f, Split f '[] f,
54− Union '[] f ~ f, Split '[] f f,
44+type SetProperties f = (Union f '[] ~ f, Split f '[] f,
45+ Union '[] f ~ f, Split '[] f f,
5546 Union f f ~ f, Split f f f,
5647 Unionable f '[], Unionable '[] f)
5748
5849{-- Union --}
5950
6051{-| Union of sets -}
61−type Union s t = Nub (Sort (s :++ t))
52+type Union s t = Nub (Sort (Append s t))
6253
6354union :: (Unionable s t) => Set s -> Set t -> Set (Union s t)
6455union s t = nub (quicksort (append s t))
6556
66−type Unionable s t = (Sortable (s :++ t), Nubable (Sort (s :++ t)))
57+type Unionable s t = (Sortable (Append s t), Nubable (Sort (Append s t)))
6758
6859{-| List append (essentially set disjoint union) -}
69−type family (:++) (x :: [k]) (y :: [k]) :: [k] where
70− '[] :++ xs = xs
71− (x ': xs) :++ ys = x ': (xs :++ ys)
60+type family Append (s :: [k]) (t :: [k]) :: [k] where
61+ Append '[] t = t
62+ Append (x ': xs) ys = x ': (Append xs ys)
7263
73−append :: Set s -> Set t -> Set (s :++ t)
64+append :: Set s -> Set t -> Set (Append s t)
7465append Empty x = x
7566append (Ext e xs) ys = Ext e (append xs ys)
7667
77−{-| Delete elements from a set -}
78−type family (m :: [k]) :\ (x :: k) :: [k] where
79− '[] :\ x = '[]
80− (x ': xs) :\ x = xs :\ x
81− (y ': xs) :\ x = y ': (xs :\ x)
82−
83−class Remove s t where
84− remove :: Set s -> Proxy t -> Set (s :\ t)
85−
86−instance Remove '[] t where
87− remove Empty Proxy = Empty
88−
89−instance {-# OVERLAPS #-} Remove xs x => Remove (x ': xs) x where
90− remove (Ext _ xs) x@Proxy = remove xs x
91−
92−instance {-# OVERLAPPABLE #-} (((y : xs) :\ x) ~ (y : (xs :\ x)), Remove xs x)
93− => Remove (y ': xs) x where
94− remove (Ext y xs) (x@Proxy) = Ext y (remove xs x)
68+{-| Useful alias for append -}
69+type (s :: [k]) :++ (t :: [k]) = Append s t
9570
9671{-| Splitting a union a set, given the sets we want to split it into -}
9772class Split s t st where
9873 -- where st ~ Union s t
99− split :: Set st -> (Set s, Set t)
74+ split :: Set st -> (Set s, Set t)
10075
10176instance Split '[] '[] '[] where
10277 split Empty = (Empty, Empty)
10378
104−instance {-# OVERLAPPABLE #-} Split s t st => Split (x ': s) (x ': t) (x ': st) where
79+instance Split s t st => Split (x ': s) (x ': t) (x ': st) where
10580 split (Ext x st) = let (s, t) = split st
10681 in (Ext x s, Ext x t)
10782
108−instance {-# OVERLAPS #-} Split s t st => Split (x ': s) t (x ': st) where
83+instance Split s t st => Split (x ': s) t (x ': st) where
10984 split (Ext x st) = let (s, t) = split st
110− in (Ext x s, t)
85+ in (Ext x s, t)
11186
112−instance {-# OVERLAPS #-} (Split s t st) => Split s (x ': t) (x ': st) where
87+instance (Split s t st) => Split s (x ': t) (x ': st) where
11388 split (Ext x st) = let (s, t) = split st
114− in (s, Ext x t)
89+ in (s, Ext x t)
90+
91+
11592
11693{-| Remove duplicates from a sorted list -}
11794type family Nub t where
@@ -121,7 +98,7 @@ type family Nub t where
12198 Nub (e ': f ': s) = e ': Nub (f ': s)
12299
123100{-| Value-level counterpart to the type-level 'Nub'
124− Note: the value-level case for equal types is not define here,
101+ Note: the value-level case for equal types is not define here,
125102 but should be given per-application, e.g., custom 'merging' behaviour may be required -}
126103
127104class Nubable t where
@@ -136,7 +113,7 @@ instance Nubable '[e] where
136113instance Nubable (e ': s) => Nubable (e ': e ': s) where
137114 nub (Ext _ (Ext e s)) = nub (Ext e s)
138115
139−instance {-# OVERLAPS #-} (Nub (e ': f ': s) ~ (e ': Nub (f ': s)),
116+instance (Nub (e ': f ': s) ~ (e ': Nub (f ': s)),
140117 Nubable (f ': s)) => Nubable (e ': f ': s) where
141118 nub (Ext e (Ext f s)) = Ext e (nub (Ext f s))
142119
@@ -145,13 +122,13 @@ instance {-# OVERLAPS #-} (Nub (e ': f ': s) ~ (e ': Nub (f ': s)),
145122class Subset s t where
146123 subset :: Set t -> Set s
147124
148−instance Subset '[] '[] where
125+instance Subset '[] '[] where
149126 subset xs = Empty
150127
151−instance {-# OVERLAPPABLE #-} Subset s t => Subset s (x ': t) where
152− subset (Ext _ xs) = subset xs
128+instance Subset '[] (x ': t) where
129+ subset xs = Empty
153130
154−instance {-# OVERLAPS #-} Subset s t => Subset (x ': s) (x ': t) where
131+instance Subset s t => Subset (x ': s) (x ': t) where
155132 subset (Ext x xs) = Ext x (subset xs)
156133
157134
@@ -164,17 +141,8 @@ data Flag = FMin | FMax
164141
165142type family Filter (f :: Flag) (p :: k) (xs :: [k]) :: [k] where
166143 Filter f p '[] = '[]
167− Filter FMin p (x ': xs) = If (Cmp x p == LT) (x ': (Filter FMin p xs)) (Filter FMin p xs)
168− Filter FMax p (x ': xs) = If (Cmp x p == GT || Cmp x p == EQ) (x ': (Filter FMax p xs)) (Filter FMax p xs)
169−
170−type family DeleteFromList (e :: elem) (list :: [elem]) where
171− DeleteFromList elem '[] = '[]
172− DeleteFromList elem (x ': xs) = If (Cmp elem x == EQ)
173− xs
174− (x ': DeleteFromList elem xs)
175−
176−type family Delete elem set where
177− Delete elem (Set xs) = Set (DeleteFromList elem xs)
144+ Filter FMin p (x ': xs) = If (Cmp x p == LT) (x ': (Filter FMin p xs)) (Filter FMin p xs)
145+ Filter FMax p (x ': xs) = If (Cmp x p == GT || Cmp x p == EQ) (x ': (Filter FMax p xs)) (Filter FMax p xs)
178146
179147{-| Value-level quick sort that respects the type-level ordering -}
180148class Sortable xs where
@@ -198,11 +166,11 @@ instance FilterV f p '[] where
198166
199167instance (Conder ((Cmp x p) == LT), FilterV FMin p xs) => FilterV FMin p (x ': xs) where
200168 filterV f@Proxy p (Ext x xs) = cond (Proxy::(Proxy ((Cmp x p) == LT)))
201− (Ext x (filterV f p xs)) (filterV f p xs)
169+ (Ext x (filterV f p xs)) (filterV f p xs)
202170
203171instance (Conder (((Cmp x p) == GT) || ((Cmp x p) == EQ)), FilterV FMax p xs) => FilterV FMax p (x ': xs) where
204172 filterV f@Proxy p (Ext x xs) = cond (Proxy::(Proxy (((Cmp x p) == GT) || ((Cmp x p) == EQ))))
205− (Ext x (filterV f p xs)) (filterV f p xs)
173+ (Ext x (filterV f p xs)) (filterV f p xs)
206174
207175class Conder g where
208176 cond :: Proxy g -> Set s -> Set t -> Set (If g s t)
@@ -217,20 +185,30 @@ instance Conder False where
217185
218186type family Cmp (a :: k) (b :: k) :: Ordering
219187
220−{-| Membership of an element in a set, with an acommanying
221− value-level function that returns a bool -}
222−class Member a s where
223− member :: Proxy a -> Set s -> Bool
224−
225−instance {-# OVERLAPS #-} Member a (a ': s) where
226− member _ (Ext x _) = True
227−
228−instance {-# OVERLAPPABLE #-} Member a s => Member a (b ': s) where
229− member a (Ext _ xs) = member a xs
230−
231−type family MemberP a s :: Bool where
232− MemberP a '[] = False
233− MemberP a (a ': s) = True
234− MemberP a (b ': s) = MemberP a s
235−
236−type NonMember a s = MemberP a s ~ False
188+{-| Pair a symbol (represetning a variable) with a type -}
189+infixl 2 :->
190+data (k :: Symbol) :-> (v :: *) = (Var k) :-> v
191+
192+data Var (k :: Symbol) where Var :: Var k
193+ {-| Some special defaults for some common names -}
194+ X :: Var "x"
195+ Y :: Var "y"
196+ Z :: Var "z"
197+
198+
199+instance (Show (Var k), Show v) => Show (k :-> v) where
200+ show (k :-> v) = "(" ++ show k ++ " :-> " ++ show v ++ ")"
201+instance Show (Var "x") where
202+ show X = "x"
203+ show Var = "Var"
204+instance Show (Var "y") where
205+ show Y = "y"
206+ show Var = "Var"
207+instance Show (Var "z") where
208+ show Z = "z"
209+ show Var = "Var"
210+instance Show (Var v) where
211+ show _ = "Var"
212+
213+{-| Symbol comparison -}
214+type instance Cmp (v :-> a) (u :-> b) = CmpSymbol v u