This commit is contained in:
ImperiumTakp 2023-05-22 17:39:47 -05:00 committed by GitHub
commit f85ea02d51
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -44,22 +44,17 @@ object EigenVectorsForSparseSymmetric {
* @return the constructed matrix * @return the constructed matrix
*/ */
def getMatrix(nonzeros: Iterable[(Int, Int, Double)], nRows: Int, nCols: Int): Matrix = { def getMatrix(nonzeros: Iterable[(Int, Int, Double)], nRows: Int, nCols: Int): Matrix = {
val matrix = new LinkedSparseMatrix(nRows, nCols) val baseCase = (new LinkedSparseMatrix(nRows, nCols), 0, 0, 0)
var numEntries = 0 val partialNonzeros = nonzeros.foldRight(baseCase)
var maxRow = 0 val (matrix, numEntries, maxRow, maxCol) = partialNonzeros{
var maxCol = 0 case ((i, j, v), (matrix, numEntries, maxRow, maxCol) ) => (i, j) match {
case (_, _) if (i > maxRow) && (j > maxCol) => (matrix.set(i, j, v), numEntries += 1, i, j)
case (_, _) if (i > maxRow) => (matrix.set(i, j, v), numEntries += 1, i, maxCol)
case (_, _) if (j > maxCol) => (matrix.set(i, j, v), numEntries += 1, maxRow, j)
case (_, _) => (matrix.set(i, j, v), numEntries += 1, maxRow, maxCol)
}
}
nonzeros.foreach {
case (i, j, v) =>
if (i > maxRow) {
maxRow = i
}
if (j > maxCol) {
maxCol = j
}
numEntries += 1
matrix.set(i, j, v)
}
log.info( log.info(
"Finished building matrix with %d entries and maxRow %d and maxCol %d" "Finished building matrix with %d entries and maxRow %d and maxCol %d"
.format(numEntries, maxRow, maxCol)) .format(numEntries, maxRow, maxCol))