Haskell - 使用递归方案的泛型多态代数数据类型的函子实例

Haskell - Functor instance for generic polymorphic Algebraic Data Types using recursion-schemes

问题:

最近我在这里问了以下问题,询问如何为任意多态 ADT(代数数据类型)(如列表、树等)创建通用映射函数和 Functor 的通用实例:

现在,我正在尝试重新表述上述内容以与 recursion-schemes 兼容。即,我不想定义基本仿函数,然后将类型定义为其不动点,而是想一方面定义类型,另一方面定义基本仿函数,并使用 Base 系列类型将它们关联起来。

所以不要这样做:

data ListF a b = NilF | ConsF a b
newtype Fix f = Fix { unFix :: f (Fix f) }
type List a = Fix (ListF a)

我想这样做:

data ListF a b = NilF | ConsF a b
data List a = Nil | Cons a (List a)
type instance Base (List a) = ListF a

这样我就可以利用 recursion-schemes 库的强大功能,同时仍然能够为任何这些多态类型定义通用 fmap。不仅如此,能够使用 "normal" 类型而不是固定点的类型同义词是一种更愉快的体验。

尝试:

最初我想一方面有一个 Bifunctor 实例,然后以某种方式强制或使其等于相应的 Base 系列实例。目前我只能考虑使用Data.Type.Equality中的a :~: b。这是我到目前为止所得到的:

{-# LANGUAGE TypeOperators, Rank2Types #-}
import Data.Bifunctor
import Data.Functor.Foldable
import Data.Type.Equality

gmap :: (Bifunctor p, Foldable (f a), Unfoldable (f b)) => 
        (forall x. p x :~: Base (f x)) -> (a -> b) -> f a -> f b
gmap refl f = cata alg
    where
        alg = embed . 
              castWith (apply refl Refl) . 
              bimap f id . 
              castWith (apply (sym refl) Refl)

我的问题在于尝试定义 Functor 的实例。我不知道如何在定义实例时指定那些特定的类型约束。 我正在考虑以某种方式创建一个类型类 Equals,然后做这样的事情:

instance (Bifunctor p, Foldable (f a), Unfoldable (f b), Equals (p a) (Base (f a))) 
    => Functor f where

但我不知道这是否可能,也不知道我是否以正确的方式接近它(例如,我不确定我对 gmap 的定义是否正确)。


作为参考,这是原始 SO 问题中泛型 gmap 的定义:

gmap :: (Bifunctor f) => (a -> b) -> Fix (f a) -> Fix (f b)
gmap f = unwrapFixBifunctor . cata alg . wrapFixBifunctor
  where
    alg = Fix . bimap f id

    unwrapFixBifunctor :: (Bifunctor f) => Fix (WrappedBifunctor f a) -> Fix (f a)
    unwrapFixBifunctor = Fix . unwrapBifunctor . fmap unwrapFixBifunctor . unFix

    wrapFixBifunctor :: (Bifunctor f) => Fix (f a) -> Fix (WrappedBifunctor f a)
    wrapFixBifunctor = Fix . fmap wrapFixBifunctor . WrapBifunctor . unFix

更新:

有人指出 gmap 的以下定义会更通用,不需要任何奇怪的类型级相等性应用:

gmap :: (Foldable t, Unfoldable d, Bifunctor p, Base d ~ p b, Base t ~ p a)
        => (a -> b) -> t -> d
gmap f = cata ( embed . bimap f id )

但是,我仍然找不到创建具有类似类型约束的 Functor 实例的方法

,只要您同意 UndecidableInstances

,我就能拼凑出一个可用的版本

我们的想法是从 gmap 的上下文中删除对 ab 的所有引用,方法是要求 forall x. Foldable (f x) 等等,使用 constraints包:

{-# LANGUAGE TypeFamilies, ScopedTypeVariables, TypeOperators, ConstraintKinds #-}
{-# LANGUAGE MultiParamTypeClasses, FlexibleInstances, FlexibleContexts #-}
import Data.Bifunctor
import Data.Functor.Foldable
import Data.Constraint
import Data.Constraint.Forall

-- 
class (p x ~ Base (f x)) => Based p f x
instance (p x ~ Base (f x)) => Based p f x

gmap :: forall p f a b. ( Bifunctor p 
                        , ForallF Foldable f
                        , ForallF Unfoldable f
                        , Forall (Based p f))
     => (a -> b) -> f a -> f b
gmap f = case (instF :: ForallF Foldable f :- Foldable (f a)) of
  Sub Dict -> case (instF :: ForallF Unfoldable f :- Unfoldable (f b)) of
    Sub Dict -> case (inst :: Forall (Based p f) :- Based p f a) of
      Sub Dict -> case (inst :: Forall (Based p f) :- Based p f b) of
        Sub Dict -> cata (embed . bimap f id)

去掉ab,我们可以把gmap变成fmap:

{-# LANGUAGE UndecidableInstances #-}
instance (Bifunctor p, ForallF Foldable f, ForallF Unfoldable f, Forall (Based p f)) => Functor f where
    fmap = gmap

编辑添加:如@gonzaw 所述,上述实例的问题在于它会匹配任何正确类型的类型:如果你有

data ListT a = NilT
             | ConsT a (ListT a)

data ListF a b = NilF
               | ConsF a b

type instance Base (ListT a) = ListF a

instance Bifunctor ListF where ...
instance Functor (ListF a) where ...
instance Foldable (ListT a) where ...
instance Unfoldable (ListT a) where ...

然后你得到的比你讨价还价的多,通用 Functor 实例和 ListF a(!) 重叠。

您可以再添加一层新类型包装器来解决这个问题:如果相反,您有

newtype F f x = F{ unF ::  (f x) }

instance (Bifunctor p, ForallF Foldable f, ForallF Unfoldable f, Forall (Based p f)) => Functor (F f) where
    fmap f = F . gmap f . unF

type ListT' = F ListT

然后最后进行以下类型检查:

*Main> unF . fmap (+1) . F $ ConsT 1 $ ConsT 2 NilT
ConsT 2 (ConsT 3 NilT)

这层额外的 newtype 包装是否可以接受是您必须决定的事情。