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



问题:

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

哈斯克尔中通用多态 ADT 的函子实例?

现在,我正在尝试重新制定上述内容以与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。不仅如此,能够使用"普通"类型而不将其作为固定点的类型同义词是一种更愉快的体验。

尝试:

最初,我想到一方面有一个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实例的方法

在@kosmikus的帮助下,我能够破解一个有效的版本,只要你对UndecidableInstances感到满意。

这个想法是从gmap的上下文中删除所有对ab的引用,方法是要求使用约束包进行编码forall x. Foldable (f x)等:

{-# LANGUAGE TypeFamilies, ScopedTypeVariables, TypeOperators, ConstraintKinds #-}
{-# LANGUAGE MultiParamTypeClasses, FlexibleInstances, FlexibleContexts #-}
import Data.Bifunctor
import Data.Functor.Foldable
import Data.Constraint
import Data.Constraint.Forall
-- https://stackoverflow.com/a/28067872/477476
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 包装器来解决这个问题:如果相反,你有

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包装对你来说是否可以接受是你必须决定的事情。

最新更新