14

Scalaでshapelessを使用していますが、2つのHListを受け取り、要素のすべてのペアのHListを返す関数allPairsを作成したいと思います。例えば:

import shapeless._
val list1 = 1 :: "one" :: HNil
val list2 = 2 :: "two" :: HNil
// Has value (1, 2) :: (1, "two") :: ("one", 2) :: ("one", "two") :: HNil
val list3 = allPairs(list1, list2)

これを行う方法はありますか?

また、インライン化されたコードブロックではなく、関数を探していることを強調したいと思います。

4

1 に答える 1

17

これらのメソッドはより高いランクの関数を必要とするため、ここforで -comprehension または関数リテラルmapとの組み合わせを使用することはできませんflatMap(他の回答が示唆するように) 。静的に型付けされたリストが 2 つしかない場合、これは簡単です。HList

import shapeless._

val xs = 1 :: 'b :: 'c' :: HNil
val ys = 4.0 :: "e" :: HNil

object eachFirst extends Poly1 {
  implicit def default[A] = at[A] { a =>
    object second extends Poly1 { implicit def default[B] = at[B](a -> _) }
    ys map second
  }
}

val cartesianProductXsYs = xs flatMap eachFirst

これにより、次のようになります (適切に入力されています)。

(1,4.0) :: (1,e) :: ('b,4.0) :: ('b,e) :: (c,4.0) :: (c,e) :: HNil

HList引数でこれを行うメソッドを書くのは、よりトリッキーです。これを行う方法の簡単な例を次に示します (もう少し一般的な機構を使用します)。

2 つの通常のリストのデカルト積を見つけることは、2 つの引数を取り、それらをタプルとしてリストの applicative functor に返す関数を「持ち上げる」と考えることができることに注意することから始めます。たとえば、Haskell で次のように記述できます。

import Control.Applicative (liftA2)

cartesianProd :: [a] -> [b] -> [(a, b)]
cartesianProd = liftA2 (,)

ここに対応する多相バイナリ関数を書くことができ(,)ます:

import shapeless._

object tuple extends Poly2 {
  implicit def whatever[A, B] = at[A, B] { case (a, b) => (a, b) }
}

完全を期すために、例のリストを再度定義します。

val xs = 1 :: 'b :: 'c' :: HNil
val ys = 4.0 :: "e" :: HNil

liftA2次に、次のように記述できるという名前のメソッドに取り組みます。

liftA2(tuple)(xs, ys)

そして正しい結果を得る。liftA2アプリカティブ ファンクタ インスタンスが実際にはなく、一般的ではないため、この名前は少し誤解flatMapを招きます。mapHList

ここで、 を取得しPoly2、それを何かに部分的に適用し、結果の単項関数を にマップできる型クラスが必要HListです。

trait ApplyMapper[HF, A, X <: HList, Out <: HList] {
  def apply(a: A, x: X): Out
}

object ApplyMapper {
  implicit def hnil[HF, A] = new ApplyMapper[HF, A, HNil, HNil] {
    def apply(a: A, x: HNil) = HNil
  }
  implicit def hlist[HF, A, XH, XT <: HList, OutH, OutT <: HList](implicit
    pb: Poly.Pullback2Aux[HF, A, XH, OutH],
    am: ApplyMapper[HF, A, XT, OutT]
  ) = new ApplyMapper[HF, A, XH :: XT, OutH :: OutT] {
    def apply(a: A, x: XH :: XT) = pb(a, x.head) :: am(a, x.tail)
  }
}

そして今、リフティングを支援する型クラス:

trait LiftA2[HF, X <: HList, Y <: HList, Out <: HList] {
  def apply(x: X, y: Y): Out
}

object LiftA2 {
  implicit def hnil[HF, Y <: HList] = new LiftA2[HF, HNil, Y, HNil] {
    def apply(x: HNil, y: Y) = HNil
  }

  implicit def hlist[
    HF, XH, XT <: HList, Y <: HList,
    Out1 <: HList, Out2 <: HList, Out <: HList
  ](implicit
    am: ApplyMapper[HF, XH, Y, Out1],
    lift: LiftA2[HF, XT, Y, Out2],
    prepend : PrependAux[Out1, Out2, Out]
  ) = new LiftA2[HF, XH :: XT, Y, Out] {
    def apply(x: XH :: XT, y: Y) = prepend(am(x.head, y), lift(x.tail, y))
  }
}

最後に、メソッド自体:

def liftA2[HF, X <: HList, Y <: HList, Out <: HList](hf: HF)(x: X, y: Y)(implicit
  lift: LiftA2[HF, X, Y, Out]
) = lift(x, y)

以上で、liftA2(tuple)(xs, ys)動作するようになりました。

scala> type Result =
     |   (Int, Double) :: (Int, String) ::
     |   (Symbol, Double) :: (Symbol, String) ::
     |   (Char, Double) :: (Char, String) :: HNil
defined type alias Result

scala> val res: Result = liftA2(tuple)(xs, ys)
res: Result = (1,4.0) :: (1,e) :: ('b,4.0) :: ('b,e) :: (c,4.0) :: (c,e) :: HNil

私たちが望んでいたように。

于 2013-01-22T10:55:31.717 に答える