Haskell - 匹配类型实例

时间:2013-09-09 21:10:51

标签: haskell types filtering instance data-partitioning

我已经定义了类似于以下内容的Haskell类型:

data TypeData = TypeA Int | TypeB String | TypeC Char deriving (Eq, Show)

在某些时候,我需要一种方法来过滤所有非TypeC实例的[TypeData]。我想写的函数的签名是:

-- Returns a tuple containing (TypeC elements, non-TypeC elements)
partitionTypeCs :: [TypeData] -> ([TypeData],[TypeData])

partition函数似乎适合这个:

-- Attempt:
partitionTypeCs data = partition (TypeData -> Bool) data

但是,我无法弄清楚哪种函数与类型签名TypeData -> Bool匹配。看起来我需要一个可以确定类型实例是否属于特定实例的函数。我知道我可以通过编写另一个函数(isTypeC (TypeC _) = True)来使用模式匹配,但是有更通用的方法或匹配类型实例的线条方式吗?

2 个答案:

答案 0 :(得分:5)

我会这样做:

partitionTypeCs = partition f where
  f (TypeC _) = False
  f _ = True

我必须承认,我不明白为什么你不喜欢这个。您无法与使用相等的构造函数进行比较,因为涉及未知参数。模式匹配完全可以在这里完成。

答案 1 :(得分:3)

模式匹配就是这里的答案。它需要你额外的3行(包括类型签名),所以它不是一个很长的解决方案。你可能会写一些模板haskell一般可以做到这一点,但对于这么简单的东西来说,这似乎太复杂了。

isTypeC :: TypeData -> Bool
isTypeC (TypeC _) = True
isTypeC _ = False