Scala:具有多种来源和异构类型的交叉(笛卡尔)产品

时间:2013-04-25 16:01:00

标签: scala reflection type-erasure cartesian-product cross-product

我正在尝试构建不同(但每个同类)类型的遍历的多个交叉产品。期望的返回类型是元组的遍历,其类型与输入遍历中的类型匹配。例如:

List(1, 2, 3) cross Seq("a", "b") cross Set(0.5, 7.3)

这应该为Traversable[(Int, String, Double)]提供来自三个来源的所有可能组合。只合并两个来源的情况很好answered here。给出的想法是:

implicit class Crossable[X](xs: Traversable[X]) {
  def cross[A](ys: Traversable[A]) = for { x <- xs; y <- ys } yield (x, y)
}

那里的评论简要提到了更多来源的问题,但我希望找到一种不依赖于无形或scalaz的解决方案(另一方面,我不介意有一些样板扩展到Tuple22)。我想做的事情如下:

implicit class Crossable[X](xs: Traversable[X]) {
  def cross[A](ys: Traversable[A]) = for { x <- xs; y <- ys } yield (x, y)
  def cross[A,B](ys: Traversable[(A,B)]) = // ... extend all Tuple2's in ys with x in xs to Tuple3's
  def cross[A,B,C](ys: Traversable[(A,B,C)]) = // ...
  // ...
}

由于类型擦除,这显然不起作用(不幸的是,可能需要在上面的示例中使用括号,因为cross将是正确的关联)。

我的问题是:是否有可能利用Scala 2.10的反射功能来解决问题?通常,将AX匹配到各种元组类型(及其类型参数,这似乎具有挑战性)并将它们合并到更大的元组应该提供满足关联法则的解决方案,对吗?

1 个答案:

答案 0 :(得分:5)

我接受了它并提出了这个问题:

trait Crosser[A,B,C] {
  def cross( as: Traversable[A], bs: Traversable[B] ): Traversable[C]
}

trait LowPriorityCrosserImplicits {
  private type T[X] = Traversable[X]

  implicit def crosser2[A,B] = new Crosser[A,B,(A,B)] {
    def cross( as: T[A], bs: T[B] ): T[(A,B)] = for { a <- as; b <- bs } yield (a, b)
  }
}

object Crosser extends LowPriorityCrosserImplicits {
  private type T[X] = Traversable[X]

  implicit def crosser3[A,B,C] = new Crosser[(A,B),C,(A,B,C)] {
    def cross( abs: T[(A,B)], cs: T[C] ): T[(A,B,C)] = for { (a,b) <- abs; c <- cs } yield (a, b, c)
  }

  implicit def crosser4[A,B,C,D] = new Crosser[(A,B,C),D,(A,B,C,D)] {
    def cross( abcs: T[(A,B,C)], ds: T[D] ): T[(A,B,C,D)] = for { (a,b,c) <- abcs; d <- ds } yield (a, b, c, d)
  }

  // and so on ...
}

implicit class Crossable[A](xs: Traversable[A]) {
  def cross[B,C](ys: Traversable[B])(implicit crosser: Crosser[A,B,C]): Traversable[C] = crosser.cross( xs, ys )
}

主要思想是将工作推迟到类型类(Crosser)并简单地通过专门针对相应arity减去1的元组Traversable来实现所有不同的元素。 在REPL中进行一些测试:

scala> List(1, 2, 3) cross Seq("a", "b") cross Set(0.5, 7.3)
res10: Traversable[(Int, String, Double)] = List((1,a,0.5), (1,a,7.3), (1,b,0.5), (1,b,7.3), (2,a,0.5), (2,a,7.3), (2,b,0.5), (2,b,7.3), (3,a,0.5), (3,a,7.3), (3,b,0.5), (3,b,7.3))