diff options
author | Iulian Dragos <jaguarul@gmail.com> | 2010-02-22 13:11:49 +0000 |
---|---|---|
committer | Iulian Dragos <jaguarul@gmail.com> | 2010-02-22 13:11:49 +0000 |
commit | 57d38b321e73f78e03469ad5dab23e84999a817d (patch) | |
tree | eacda0bb39b039f625567edbee4cf1c6d40eeb50 /test/files/run/spec-matrix.scala | |
parent | 6e061d6f2597a3e4bdc7fcf192abbdb419917b70 (diff) | |
download | scala-57d38b321e73f78e03469ad5dab23e84999a817d.tar.gz scala-57d38b321e73f78e03469ad5dab23e84999a817d.tar.bz2 scala-57d38b321e73f78e03469ad5dab23e84999a817d.zip |
Specialized types are not substituted inside ty...
Specialized types are not substituted inside type arguments. Closes
#3085, no review.
Diffstat (limited to 'test/files/run/spec-matrix.scala')
-rw-r--r-- | test/files/run/spec-matrix.scala | 70 |
1 files changed, 70 insertions, 0 deletions
diff --git a/test/files/run/spec-matrix.scala b/test/files/run/spec-matrix.scala new file mode 100644 index 0000000000..212a800672 --- /dev/null +++ b/test/files/run/spec-matrix.scala @@ -0,0 +1,70 @@ +/** Test matrix multiplication with specialization. + */ + +class Matrix[@specialized A: ClassManifest](val rows: Int, val cols: Int) { + private val arr: Array[Array[A]] = new Array[Array[A]](rows, cols) + + def apply(i: Int, j: Int): A = { + if (i < 0 || i >= rows || j < 0 || j >= cols) + throw new NoSuchElementException("Indexes out of bounds: " + (i, j)) + + arr(i)(j) + } + + def update(i: Int, j: Int, e: A) { + arr(i)(j) = e + } + + def rowsIterator: Iterator[Array[A]] = new Iterator[Array[A]] { + var idx = 0; + def hasNext = idx < rows + def next = { + idx += 1 + arr(idx - 1) + } + } +} + +object Test { + def main(args: Array[String]) { + val m = randomMatrix(200, 100) + val n = randomMatrix(100, 200) + + mult(m, n) + println("*") + } + + def randomMatrix(n: Int, m: Int) = { + val r = new util.Random(10) + val x = new Matrix[Int](n, m) + for (i <- 0 until n; j <- 0 until m) + x(i, j) = r.nextInt + x + } + + + def multManifest[@specialized("Int") T](m: Matrix[T], n: Matrix[T])(implicit cm: ClassManifest[T], num: Numeric[T]) { + val p = new Matrix[T](m.rows, n.cols) + import num._ + + for (i <- 0 until m.rows) + for (j <- 0 until n.cols) { + var sum = num.zero + for (k <- 0 until n.rows) + sum += m(i, k) * n(k, j) + p(i, j) = sum + } + } + + def mult(m: Matrix[Int], n: Matrix[Int]) { + val p = new Matrix[Int](m.rows, n.cols) + + for (i <- 0 until m.rows) + for (j <- 0 until n.cols) { + var sum = 0 + for (k <- 0 until n.rows) + sum += m(i, k) * n(k, j) + p(i, j) = sum + } + } +} |