编译时两个相同大小的HList的总和?

时间:2017-02-08 13:59:19

标签: scala shapeless

鉴于以下尝试检索两个HList的递归和,如下所示:

(请原谅其名下的Product,请。)

package net

import shapeless._
import shapeless.nat._
import shapeless.ops.nat.{Sum, Prod, Mod}

trait SumZippedProduct[L, M] {
  type S
}
object SumZippedProduct {

  type Aux[L, M, O] = SumZippedProduct[L, M] {
    type S = O
  }

  def apply[L <: HList, M <: HList](implicit ev: SumZippedProduct[L, M]) = ev

  // LH - L's head
  // L  - HList
  // MH - M's head
  // M  - HList
  // RS - Recursive Sum (L + H)
  // CS - Current Sum (LH + RH)
  // E  - RS + CS
  implicit def sumZippedInductiveEq5[LH <: Nat, L <: HList, MH <: Nat, M <: HList, RS <: Nat, CS <: Nat, E <: Nat](
    implicit ev: SumZippedProduct.Aux[L, M, RS],
             curr: Sum.Aux[LH, MH, CS],
             total: Sum.Aux[CS, RS, E]
  ): SumZippedProduct[LH :: L, MH :: M] = new SumZippedProduct[LH :: L, MH :: M] {
    type S = E
  }

  implicit val hnils: SumZippedProduct[HNil, HNil] = new SumZippedProduct[HNil, HNil] {
    type S = _0
  }

}

尝试时,我认为,对HNil案例有效,但对于1元素HList则无效:

scala> import net.SumZippedProduct
import net.SumZippedProduct

scala> import shapeless._, nat._
import shapeless._
import nat._

// expecting 0 (0 + 0)
scala> SumZippedProduct[HNil, HNil]
res0: net.SumZippedProduct[shapeless.HNil,shapeless.HNil] = net.SumZippedProduct$$anon$2@794b4359

// expecting _4 (1 + 3)
scala> SumZippedProduct[_1 :: HNil, _3 :: HNil]
<console>:19: error: could not find implicit value for parameter ev: net.SumZippedProduct[shapeless.::[shapeless.nat._1,shapeless.HNil],shapeless.::[shapeless.nat._3,shapeless.HNil]]
       SumZippedProduct[_1 :: HNil, _3 :: HNil]
                       ^

为什么在传递_1 :: HNil_3 :: HNil时没有编译?

另外,如何在_0中获取res0

scala> res0.S
<console>:20: error: value S is not a member of net.SumZippedProduct[shapeless.HNil,shapeless.HNil]
       res0.S
            ^

注意 - 如果这种实现已经存在于无形状中,我很感激,但我还是要问这个问题。

1 个答案:

答案 0 :(得分:2)

您必须在隐含的返回类型中使用Aux。否则S的具体类型将丢失。 对于召唤方法apply,您还必须使用更精确的返回类型,原因相同。由于SumZippedProduct延伸AnyRef,您可以使用ev.type;它不会比那更准确。

scala> :paste
// Entering paste mode (ctrl-D to finish)

import shapeless._
import shapeless.nat._
import shapeless.ops.nat.{Sum, Prod, Mod, ToInt}

trait SumZippedProduct[L, M] {
  type S <: Nat
  final def S(implicit toInt: ToInt[S]): Int = toInt()
}
object SumZippedProduct {

  type Aux[L, M, O] = SumZippedProduct[L, M] {
    type S = O
  }

  def apply[L <: HList, M <: HList](implicit ev: SumZippedProduct[L, M]): ev.type = ev

  implicit def sumZippedInductiveEq5[LH <: Nat, L <: HList, MH <: Nat, M <: HList, RS <: Nat, CS <: Nat, E <: Nat](
    implicit ev: SumZippedProduct.Aux[L, M, RS],
             curr: Sum.Aux[LH, MH, CS],
             total: Sum.Aux[CS, RS, E]
  ): SumZippedProduct.Aux[LH :: L, MH :: M, E] = new SumZippedProduct[LH :: L, MH :: M] {
    type S = E
  }

  implicit val hnils: SumZippedProduct.Aux[HNil, HNil, _0] = new SumZippedProduct[HNil, HNil] {
    type S = _0
  }

}

// Exiting paste mode, now interpreting.

import shapeless._
import shapeless.nat._
import shapeless.ops.nat.{Sum, Prod, Mod}
defined trait SumZippedProduct
defined object SumZippedProduct

scala> SumZippedProduct[HNil, HNil]
res1: SumZippedProduct.<refinement>.type = SumZippedProduct$$anon$2@673bac03

scala> val a: res1.S = _0
a: res1.S = shapeless._0@4f450e01

scala> SumZippedProduct[_1 :: HNil, _3 :: HNil]
res2: SumZippedProduct.Aux[shapeless.::[shapeless.nat._1,shapeless.HNil],shapeless.::[shapeless.nat._3,shapeless.HNil],this.Out] = SumZippedProduct$$anon$1@2a53bcfa

scala> val a: res2.S = _4
a: res2.S = Succ()

scala> val a: res2.S = _5
<console>:26: error: type mismatch;
 found   : shapeless.nat._5
    (which expands to)  shapeless.Succ[shapeless.Succ[shapeless.Succ[shapeless.Succ[shapeless.Succ[shapeless._0]]]]]
 required: res2.S
    (which expands to)  shapeless.Succ[shapeless.Succ[shapeless.Succ[shapeless.Succ[shapeless._0]]]]
       val a: res2.S = _5
                       ^

scala> res2.S
res3: Int = 4

我还添加了一种生成相应整数值的方法。我实际上并不知道从Nat类型中召唤Nat值的现有方法(或者我只是盲目......)。但我认为在大多数情况下,Nat仅在类型级别有用,而且您更倾向于使用值级别的实际Int

如果你真的想要价值等级Nat,那么你自己也难以实现。但是如果你看看它们在无形状中的实现,它们实际上只是空盒子,所以你用它们做的不多。只有他们的类型才有用。