Question

有没有通过scala中的值来参数化类型？例如，使用它的大小来参数化矩阵，就像......

val m1 = new Matrix[2,3]()
val m2 = new Matrix[5,1]()

val m3 = m1 multiply m2

会无法编译，因为你不能将[2,3]矩阵乘以[5,1]吗？

这在实现其他类型（如元组或向量）时也很有用。有谁知道实现这个目标的方法？

Answer 1

使用Peano数字我们可以定义所有类型都是从0开始的自然数。这里它们都是Nat的子类型，但_1和_2是不同的类型，因此它们可以'用它们代替对方而没有差异。

定义自然数：

scala> sealed trait Nat
defined trait Nat

scala> sealed trait _0 extends Nat
defined trait _0

scala> sealed trait Succ[N <: Nat] extends Nat
defined trait Succ

scala> type _1 = Succ[_0]
defined type alias _1

scala> type _2 = Succ[_1]
defined type alias _2

Matrix的参数类型不变：

scala> case class Matrix[A <: Nat, B <: Nat](ignoreThis: String)
defined class Matrix

乘法函数也是不变的：

scala> def multiply[R1 <: Nat, C1 <: Nat, C2 <: Nat](m1: Matrix[R1, C1], m2: Matrix[C1, C2]) = Matrix[R1, C2](m1.ignoreThis + m2.ignoreThis)
multiply: [R1 <: Nat, C1 <: Nat, C2 <: Nat](m1: Matrix[R1,C1], m2: Matrix[C1,C2])Matrix[R1,C2]

编译器将为您进行检查，尺寸匹配：

scala> multiply(Matrix[_1, _2]("one"), Matrix[_2, _1]("two"))
res0: Matrix[_1,_1] = Matrix(onetwo)

维度不匹配，编译时错误比运行时好得多：

scala> multiply(Matrix[_1, _2]("one"), Matrix[_1, _1]("two"))
<console>:19: error: type mismatch;
 found   : Matrix[_1(in object $iw),_2]
    (which expands to)  Matrix[Succ[_0],Succ[Succ[_0]]]
 required: Matrix[_1(in object $iw),Succ[_ >: _0 with _1(in object $iw) <: Nat]]
    (which expands to)  Matrix[Succ[_0],Succ[_ >: _0 with Succ[_0] <: Nat]]
Note: _2 <: Succ[_ >: _0 with _1 <: Nat], but class Matrix is invariant in type B.
You may wish to define B as +B instead. (SLS 4.5)
       multiply(Matrix[_1, _2]("one"), Matrix[_1, _1]("two"))
                              ^
<console>:19: error: type mismatch;
 found   : Matrix[_1(in object $iw),_1(in object $iw)]
    (which expands to)  Matrix[Succ[_0],Succ[_0]]
 required: Matrix[Succ[_ >: _0 with _1(in object $iw) <: Nat],_1(in object $iw)]
    (which expands to)  Matrix[Succ[_ >: _0 with Succ[_0] <: Nat],Succ[_0]]
Note: _1 <: Succ[_ >: _0 with _1 <: Nat], but class Matrix is invariant in type A.
You may wish to define A as +A instead. (SLS 4.5)
       multiply(Matrix[_1, _2]("one"), Matrix[_1, _1]("two"))
                                                     ^

我懒得写实际的乘法实现，因此ignoreThis占位符。

Scala中的大小参数化

1 个答案: