如何使用Data.Reify来修改数据列表?

时间:2012-07-27 20:09:23

标签: haskell

我尝试阅读论文(http://www.ittc.ku.edu/csdl/fpg/sites/default/files/Gill-09-TypeSafeReification.pdf)并设法重新启用我的符号表达式类型,但我无法弄清楚如何重新列出它们的列表。这是简化的代码:

{-# OPTIONS_GHC -Wall #-}
{-# Language TypeOperators #-}
{-# Language TypeFamilies #-}
{-# Language FlexibleInstances #-}

import Control.Applicative
import Data.Reify

-- symbolic expression type
data Expr a = EConst a
            | EBin (Expr a) (Expr a)
            deriving Show

-- corresponding node type
data GraphExpr a b = GConst a
                   | GBin b b
                   deriving Show

instance MuRef (Expr a) where
  type DeRef (Expr a) = GraphExpr a
  mapDeRef _ (EConst c)  = pure (GConst c)
  mapDeRef f (EBin u v) = GBin <$> f u <*> f v

-- this works as expected
main :: IO ()
main = reifyGraph (EBin x (EBin x y)) >>= print
  where
    x = EConst "x"
    y = EConst "y"
-- (output: "let [(1,GBin 2 3),(3,GBin 2 4),(4,GConst "y"),(2,GConst "x")] in 1")

-- but what if I want to reify a list of Exprs?
data ExprList a = ExprList [Expr a]
data GraphList a b = GraphList [GraphExpr a b]

instance MuRef (ExprList a) where
  type DeRef (ExprList a) = GraphList a
  --  mapDeRef f (ExprList xs) = ???????

2 个答案:

答案 0 :(得分:4)

我遇到了完全相同的问题,我找到了一个使用data-reify的解决方案。

要达成解决方案,您必须意识到的事情是: 1.即使EDSL没有列表,图表类型也可以包含它们 2.可以将不同类型的数据统一为相同的结果类型。

所以我们首先在结果类型中添加列表构造函数:

data GraphExpr a b = GConst a
                   | GBin b b
                   | Cons b b
                   | Nil
                   deriving Show

然后我们需要MuRef的第二个实例,它将Expr a列表与GraphExpr相结合。

instance MuRef [Expr a] where
    type DeRef [Expr a] = GraphExpr a
    mapDeRef _ [] = pure Nil
    mapDeRef f (x:xs) = Cons <$> f x <*> f xs

现在有了这个,如果我们试图重新列出一个列表表达式

reified = reifyGraph [EBin x (EBin x y), Ebin y (EBin x y)]
              where x = EConst "x"
                    y = EConst "y"

我们会得到结果

let [(1,Cons 2 6),(6,Cons 7 9),(9,Nil),(7,GBin 5 8),(8,GBin 3 5),(2,GBin 3 4),(4,GBin 3 5),(5,GConst "y"),(3,GConst "x")] in 1

要从此图中提取已知的节点ID列表,我们可以定义一个小函数来遍历Conses并将节点ID从它们中提取到列表中。

walkConses :: Graph (GraphExpr t) -> [Unique]
walkConses (Graph xs root) = go (lookup root xs)
where 
    go (Just (Cons n1 n2)) = n1 : go (lookup n2 xs)
    go (Just Nil) = []

(如果图形很大,最好在开始游走之前将它们转换为IntMap)

这看起来像是一个部分函数,​​但是因为我们知道DAG的根将始终是一个Cons节点(因为我们重新列出了一个列表),并且因为我们知道所有node-id都在xs中,所以这个函数将返回结果列表中所有node-id的列表。

因此,如果我们在结果图上运行walkConses,我们将得到结果:

[2, 7]

希望这会有所帮助,我也一直在努力解决这个问题。

答案 1 :(得分:3)

你真的不能用MuRef那样做。 GraphLists不包含GraphLists。您可以依次重新启动每个Expr并编写一个一次性组合器,将它们粉碎到您的GraphList中:

只需在遍历列表内容上使用遍历reifyGraph。

此外,后者都应该是新类型。