4

写算法HList，我需要一个zipWithIndex函数。它现在不在 shapeless 库中，所以我决定实现它。

很明显，它可以实现为

hlist.zip(indexes)

哪里indexes是HList索引（0..n），可能可以通过这种方式获得：

val indexes = Nat._0 until hlist.length

这里的问题是Nat没有until方法。我还没有找到任何Witness可用于HList索引的HList.map.

我可以用什么方法来获取从until开始HList的Nats ？Nat._0hlist.length

4

1 回答 1

6

将这样的内容添加到 Shapeless 可能是有意义的：

import shapeless._, ops.hlist.Prepend

trait Range[A <: Nat, B <: Nat] extends DepFn0 { type Out <: HList }

object Range {
  type Aux[A <: Nat, B <: Nat, Out0 <: HList] = Range[A, B] { type Out = Out0 }

  implicit def emptyRange[A <: Nat]: Aux[A, A, HNil] = new Range[A, A] {
    type Out = HNil
    def apply(): Out = HNil
  }

  implicit def slightlyBiggerRange[A <: Nat, B <: Nat, OutAB <: HList](implicit
    rangeAB: Aux[A, B, OutAB],
    appender: Prepend[OutAB, B :: HNil],
    witnessB: Witness.Aux[B]
  ): Aux[A, Succ[B], appender.Out] = new Range[A, Succ[B]] {
    type Out = appender.Out
    def apply(): Out = appender(rangeAB(), witnessB.value :: HNil)
  }
}

def range[A <: Nat, B <: Nat](implicit r: Range[A, B]): r.Out = r()

现在你可以写zipWithIndex得很干净了：

import ops.hlist.{ Length, Zip }

def zipWithIndex[L <: HList, S <: Nat, R <: HList, Out <: HList](l: L)(implicit
  len: Length.Aux[L, S],
  range: Range.Aux[nat._0, S, R],
  zipper: Zip.Aux[L :: R :: HNil, Out]
): Out = l.zip(range())

接着：

import nat._

type Expected = (Int, _0) :: (Symbol, _1) :: (String, _2) :: HNil

val xs: Expected = zipWithIndex(1 :: 'a :: "foo" :: HNil)

您还可以使用 fold 或ZippedWithIndex[L <: HList]type 类，这两者都可能更简洁一些，但不太清楚地由独立有用的部分组成，例如Range.

于 2014-07-17T22:46:52.423 回答