简单的GHC.Generics示例

时间:2013-07-15 13:57:01

标签: generics haskell

我正在尝试按照wiki article创建如何使用GHC.Generics的最小工作示例。这就是我所拥有的:

{-# LANGUAGE DefaultSignatures, DeriveGeneric, TypeOperators, FlexibleContexts #-}

import GHC.Generics

data Bit = O | I deriving Show

class Serialize a where
  put :: a -> [Bit]

  default put :: (Generic a, GSerialize (Rep a)) => a -> [Bit]
  put a = gput (from a)

class GSerialize f where
  gput :: f a -> [Bit]

instance GSerialize U1 where
  gput U1 = []

instance (GSerialize a, GSerialize b) => GSerialize (a :*: b) where
  gput (a :*: b) = gput a ++ gput b

instance (GSerialize a, GSerialize b) => GSerialize (a :+: b) where
  gput (L1 x) = O : gput x
  gput (R1 x) = I : gput x

instance (GSerialize a) => GSerialize (M1 i c a) where
  gput (M1 x) = gput x

instance (Serialize a) => GSerialize (K1 i a) where
  gput (K1 x) = put x


--
-- Try it out...
--

data UserTree a = Node a (UserTree a) (UserTree a) | Leaf
  deriving Generic

instance (Serialize a) => Serialize (UserTree a)

instance Serialize Int


main = do
  print . put $ (Leaf :: UserTree Int)
  print . put $ (Node 7 Leaf Leaf :: UserTree Int)
  print . put $ (3 :: Int)

然而,当我尝试运行它时,程序挂起:

λ> main
[I]
[O     -- the program hangs here

我做错了什么?

1 个答案:

答案 0 :(得分:5)

您需要Int的正确实例。这是一种内置类型,你不能指望魔法。为Int提供一个空实例将导致循环(这可能是一个糟糕的设计决策,但目前的情况就是如此)。

这是一个有效的方法(但效率不高):

import Data.Bits

boolToBit :: Bool -> Bit
boolToBit False = O
boolToBit True  = I

instance Serialize Int where
  put x = map (boolToBit . testBit x) [0 .. bitSize x - 1]

如果您真的想要一个最小的示例,请不要使用Int,而是使用Tree ()Tree Bool