summaryrefslogtreecommitdiff
path: root/test/Main.hs
blob: 37e75b3e579e3b527a2948ebcc9f9b7d8fb7082c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE ImportQualifiedPost #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE StandaloneDeriving #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE TypeFamilies #-}
module Main where

import Data.Dependent.EnumMap.Strict qualified as DE
import Data.Dependent.EnumMap.Strict.Internal (coe1)
import Data.Dependent.Sum
import Data.Functor.Const

data Tag = A | B | C
  deriving (Show)

data STag tag where
  SA :: STag A
  SB :: STag B
  SC :: STag C
deriving instance Show (STag tag)

instance DE.Enum1 STag where
  type Enum1Info STag = Const ()
  fromEnum1 = \case
                SA -> (0, Const ())
                SB -> (1, Const ())
                SC -> (2, Const ())
  toEnum1 n _ = case n of
                  0 -> coe1 SA
                  1 -> coe1 SB
                  2 -> coe1 SC
                  _ -> error "invalid tag"

data Value tag where
  VA :: Int -> Value A
  VB :: Bool -> Value B
  VC :: String -> Value c
deriving instance Show (Value tag)


main :: IO ()
main = do
  print $ DE.fromList @STag @Value []
  print $ DE.fromList [SB :=> VB False, SA :=> VA 3]