使用Template Haskell,如何将相同类型拼接到多个位置?

时间:2011-12-07 05:18:50

标签: haskell template-haskell

我正在为vector-space类型定义来自OpenGL的类的实例,为了节省我的打字肌肉,我想使用Template Haskell为我编写一堆实例。

我开始时通过定义函数来导出AdditiveGroup的实例:

{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE TypeFamilies #-}
module Data.VectorSpace.OpenGL.TH where

import Control.Applicative
import Control.Monad
import Data.AdditiveGroup
import Data.VectorSpace

import Language.Haskell.TH

deriveScalarAdditive ts = concat <$> forM (map conT ts) (\t -> [d| 
    instance AdditiveGroup $t where zeroV = 0; (^+^) = (+); negateV = negate
  |])

这很好,但请注意,我只是将$t一次拼接到牛津括号中。现在,派生VectorSpace个实例的函数:

deriveScalarVectorSpace ts = concat <$> forM (map conT ts) (\t -> [d|    
    instance VectorSpace $t where type Scalar $t = $t; (*^) = (*)
  |])

但是,这个barfs:

Type indexes must match class instance head
Found `t_tt' but expected `t_ts'
In the associated type instance for `Scalar'
In the instance declaration for `VectorSpace $t'
In the Template Haskell quotation
  [d| instance VectorSpace $t where
          type instance Scalar $t = $t
          { *^ = (*) } |]

错误中t_tst_tt之间的区别告诉我,每当我拼接$t时,TH都会创建一个新的唯一名称,当然这个定义只能起作用如果那些类型是相同的。

有没有办法通过牛津括号获得我想要的行为,还是我必须回到良好的旧词法范围和Language.Haskell.TH组合器?我知道CPP可能会更容易,但我想借此机会学习一些TH。

1 个答案:

答案 0 :(得分:5)

我认为你必须使用Language.Haskell.TH组合器。请参阅以下门票:

这样做非常简单。我会从这开始(稍微格式化)

*Foo Language.Haskell.TH> runQ (deriveScalarAdditive [''Int] ) >>= print

[InstanceD [] (AppT (ConT Data.AdditiveGroup.AdditiveGroup) (ConT GHC.Types.Int))
  [ValD (VarP zeroV_12) (NormalB (LitE (IntegerL 0))) [],
   ValD (VarP ^+^_13) (NormalB (VarE GHC.Num.+)) [],
   ValD (VarP negateV_14) (NormalB (VarE GHC.Num.negate)) []]
]

从这里开始,看看如何使用组合器构造实例非常简单。另请注意,您可以将引号与引用的表达式[| some code |] :: ExpQ混合使用,这通常对创建函数体很有用。