Language/Symantic/Lib/Data/Peano.hs

   1 {-# LANGUAGE GADTs #-}
   2 {-# LANGUAGE Rank2Types #-}
   3 {-# LANGUAGE TypeOperators #-}
   4 {-# OPTIONS_GHC -fno-warn-missing-methods #-}
   5 -- | Type variable.
   6 module Language.Symantic.Lib.Data.Peano where
   7
   8 import Data.Type.Equality ((:~:)(Refl))
   9
  10 -- * Types 'Zero' and 'Succ'
  11 -- | Type-level peano numbers of kind '*'.
  12 data Zero
  13 -- | Undefined instance required to build errors.
  14 instance Eq  Zero
  15 -- | Undefined instance required to build errors.
  16 instance Ord Zero
  17 data Succ p
  18
  19 -- * Type 'SPeano'
  20 -- | Singleton for peano numbers.
  21 data SPeano p where
  22  SZero :: SPeano Zero
  23  SSucc :: SPeano p -> SPeano (Succ p)
  24
  25 integral_from_peano :: Integral i => SPeano p -> i
  26 integral_from_peano SZero = 0
  27 integral_from_peano (SSucc x) = 1 + integral_from_peano x
  28
  29 peano_from_integral :: Integral i => i -> (forall p. SPeano p -> ret) -> ret
  30 peano_from_integral 0 k = k SZero
  31 peano_from_integral i k | i > 0 =
  32         peano_from_integral (i - 1) $ \p -> k (SSucc p)
  33 peano_from_integral _ _ = error "peano_from_integral"
  34
  35 peano_eq :: forall x y. SPeano x -> SPeano y -> Maybe (x :~: y)
  36 peano_eq SZero SZero = Just Refl
  37 peano_eq (SSucc x) (SSucc y)
  38  | Just Refl <- x `peano_eq` y
  39  = Just Refl
  40 peano_eq _ _ = Nothing