Haskell 中的 AST 转换

Question

我有一个 AST，其根节​​点的类型为 E.Root。我需要将其转换为具有类型为I.Root 的根节点的AST。

我现在可以定义一个带有类型签名的函数eToI：

eToI :: E.Root -> AdditionalInfo -> I.Root

但是，两个 AST 共享很多节点。所以，eToF 函数有很多样板代码，从 E's 节点构建 I's 节点基本相同。

我想解决这两个问题：

1. 在类型级别，避免定义I 的节点。我有所有节点 E 的定义。我在I 中定义了哪些节点发生了变化。我可以拥有吗 编译器生成I 中的所有类型，通过某种方式映射什么 需要改变吗？

2. 在价值层面，我只想为 变化的节点（假设 E 的 A 映射到 I 的 Z ..）：

       aToZ :: E.A -> AdditionalInfo -> I.Z
       bToY :: E.B -> AdditionalInfo -> I.Y
   

   现在，编译器可以生成像 eToI 这样的函数吗？

       eToI :: E.Root -> AdditionalInfo -> I.Root
   

Haskell 的惯用方法是什么？

Answer 1

有时您可以将E.Root 和I.Root 的共同方面分解为一种或多种可重用的数据类型。它通常还有助于将 AdditionalInfo -> ... 隐藏在 monad 或应用函子中。例如，这里有一些伪代码：

module Common where
  {-# LANGUAGE DeriveTraversable #-}
  data Reusable root = ... root ...
    deriving (Functor, Foldable, Traversable)

module E where
  import Common
  data Root = Leaf (Reusable Root) | Node Root

module I where
  {-# LANGUAGE GeneralizedNewtypeDeriving #-}
  import Common

  import Data.Monoid

  data Root = Root [Reusable Root]
    deriving Monoid

module Transform where
  import Common
  import qualified E
  import qualified I

  import Control.Applicative
  import Control.Monad.Reader
  import Data.Monoid

  type AdditionalInput = ...
  type F = Reader AdditionalInput

  convertRoot :: E.Root -> F I.Root
  convertRoot (Leaf reusable) =
    traverse convertRoot reusable
  convertRoot (Node left right) =
    liftA2 mappend (convertRoot left) (convertRoot right)

现在我可以使用traverse 在Reusable E.Root 和Reusable I.Root 之间进行转换。