如何将自定义类型转换为 Haskell 中的整数？

Question

我正在尝试在 haskell 中使用我自己的数据类型作为质数，但我目前运行遇到了一些问题。

newtype Prime = Prime Integer deriving (Eq, Ord, Typeable, Show)

只要我对素数进行任何数字运算（例如下面的“phi”函数），我就想将结果作为整数处理，但我不知道该怎么做。

phi :: Prime -> Prime -> Integer
phi p q = (p-1)*(q-1)

phi 应该 return 一个整数，因为它不再是素数了。我得到的只是预期的错误消息：

    • Couldn't match expected type ‘Integer’ with actual type ‘Prime’
    • In the expression: (p - 1) * (q - 1)
      In an equation for ‘genPhi’: genPhi p q = (p - 1) * (q - 1)

那么如何将我的自定义类型转换为整数？我对 Haskell.

没有太多经验

Answer 1

您可以从 Prime 数据构造函数中解包 Integer：

genPhi :: Prime -> Prime -> Integer
genPhi <b>(Prime</b> p<b>) (Prime</b> q<b>)</b> = (p-1) * (q-1)

Answer 2

如果您对现有类型有一个简单的新类型包装器，一个好技巧是使用 DerivingVia 扩展：

{-# LANGUAGE DerivingVia #-}

newtype Prime = Prime { unPrime :: Integer }
   deriving Num via Integer

phi :: Prime -> Prime -> Integer
phi p q = unPrime $ (p-1)*(q-1)

鉴于此，你可以说：

*Main> phi (Prime 3) (Prime 5)
8

此外，所有其他数值运算将自动对您的 Prime 类型起作用，只需使用它们的 Integer 等价物即可。

详情见deriving via。

NB 如评论中所述，这并不意味着 GHC 将以任何方式确保这些操作的结果为 Prime；它只是让您解除基础操作。（但是如果没有派生机制，你可能会犯同样的错误。）如果你的程序维护 Prime 构造函数总是意味着参数是质数的不变性，那么不要使用这个技巧。这通常不是问题，因为不变量没有明确定义或易于执行，构造函数只是作为提醒。但最好明确这一点，如果你非常依赖它，就不要使用推导技巧。

如何将自定义类型转换为 Haskell 中的整数？

How to convert a custom type to an Integer in Haskell?

haskell

functional-programming

algebraic-data-types

custom-data-type

derivingvia