cern.colt.matrix.tfloat.impl.SparseFloatMatrix2D Maven / Gradle / Ivy
Show all versions of parallelcolt Show documentation
/*
Copyright (C) 1999 CERN - European Organization for Nuclear Research.
Permission to use, copy, modify, distribute and sell this software and its documentation for any purpose
is hereby granted without fee, provided that the above copyright notice appear in all copies and
that both that copyright notice and this permission notice appear in supporting documentation.
CERN makes no representations about the suitability of this software for any purpose.
It is provided "as is" without expressed or implied warranty.
*/
package cern.colt.matrix.tfloat.impl;
import java.io.IOException;
import cern.colt.map.tfloat.AbstractLongFloatMap;
import cern.colt.map.tfloat.OpenLongFloatHashMap;
import cern.colt.matrix.io.MatrixInfo;
import cern.colt.matrix.io.MatrixSize;
import cern.colt.matrix.io.MatrixVectorReader;
import cern.colt.matrix.tfloat.FloatMatrix1D;
import cern.colt.matrix.tfloat.FloatMatrix2D;
/**
* Sparse hashed 2-d matrix holding float elements. First see the package summary and javadoc tree view to get the broad picture.
*
* Implementation:
*
* Note that this implementation is not synchronized. Uses a
* {@link cern.colt.map.tfloat.OpenLongFloatHashMap}, which is a compact and
* performant hashing technique.
*
* Memory requirements:
*
* Cells that
*
* - are never set to non-zero values do not use any memory.
*
- switch from zero to non-zero state do use memory.
*
- switch back from non-zero to zero state also do use memory. However,
* their memory is automatically reclaimed from time to time. It can also
* manually be reclaimed by calling {@link #trimToSize()}.
*
*
* worst case: memory [bytes] = (1/minLoadFactor) * nonZeros * 13.
* best case: memory [bytes] = (1/maxLoadFactor) * nonZeros * 13.
* Where nonZeros = cardinality() is the number of non-zero cells.
* Thus, a 1000 x 1000 matrix with minLoadFactor=0.25 and maxLoadFactor=0.5 and
* 1000000 non-zero cells consumes between 25 MB and 50 MB. The same 1000 x 1000
* matrix with 1000 non-zero cells consumes between 25 and 50 KB.
*
* Time complexity:
*
* This class offers expected time complexity O(1) (i.e.
* constant time) for the basic operations get, getQuick,
* set, setQuick and size assuming the hash function
* disperses the elements properly among the buckets. Otherwise, pathological
* cases, although highly improbable, can occur, degrading performance to
* O(N) in the worst case. As such this sparse class is expected to
* have no worse time complexity than its dense counterpart
* {@link DenseFloatMatrix2D}. However, constant factors are considerably
* larger.
*
* Cells are internally addressed in row-major. Performance sensitive
* applications can exploit this fact. Setting values in a loop row-by-row is
* quicker than column-by-column, because fewer hash collisions occur. Thus
*
*
* for (int row = 0; row < rows; row++) {
* for (int column = 0; column < columns; column++) {
* matrix.setQuick(row, column, someValue);
* }
* }
*
*
* is quicker than
*
*
* for (int column = 0; column < columns; column++) {
* for (int row = 0; row < rows; row++) {
* matrix.setQuick(row, column, someValue);
* }
* }
*
*
* @see cern.colt.map
* @see cern.colt.map.tfloat.OpenLongFloatHashMap
* @author [email protected]
* @version 1.0, 09/24/99
*
* @author Piotr Wendykier ([email protected])
*/
public class SparseFloatMatrix2D extends FloatMatrix2D {
/**
*
*/
private static final long serialVersionUID = 1L;
/*
* The elements of the matrix.
*/
protected AbstractLongFloatMap elements;
/**
* Constructs a matrix with a copy of the given values. values is
* required to have the form values[row][column] and have exactly
* the same number of columns in every row.
*
* The values are copied. So subsequent changes in values are not
* reflected in the matrix, and vice-versa.
*
* @param values
* The values to be filled into the new matrix.
* @throws IllegalArgumentException
* if
* for any 1 <= row < values.length: values[row].length != values[row-1].length
* .
*/
public SparseFloatMatrix2D(float[][] values) {
this(values.length, values.length == 0 ? 0 : values[0].length);
assign(values);
}
/**
* Constructs a matrix with a given number of rows and columns and default
* memory usage. All entries are initially 0.
*
* @param rows
* the number of rows the matrix shall have.
* @param columns
* the number of columns the matrix shall have.
* @throws IllegalArgumentException
* if
* rows<0 || columns<0 || (double)columns*rows > Integer.MAX_VALUE
* .
*/
public SparseFloatMatrix2D(int rows, int columns) {
this(rows, columns, rows * (columns / 1000), 0.2f, 0.5f);
}
/**
* Constructs a matrix with a given number of rows and columns using memory
* as specified. All entries are initially 0. For details related
* to memory usage see {@link cern.colt.map.tfloat.OpenLongFloatHashMap}.
*
* @param rows
* the number of rows the matrix shall have.
* @param columns
* the number of columns the matrix shall have.
* @param initialCapacity
* the initial capacity of the hash map. If not known, set
* initialCapacity=0 or small.
* @param minLoadFactor
* the minimum load factor of the hash map.
* @param maxLoadFactor
* the maximum load factor of the hash map.
* @throws IllegalArgumentException
* if
*
* initialCapacity < 0 || (minLoadFactor < 0.0 || minLoadFactor >= 1.0) || (maxLoadFactor <= 0.0 || maxLoadFactor >= 1.0) || (minLoadFactor >= maxLoadFactor)
* .
* @throws IllegalArgumentException
* if
* rows<0 || columns<0 || (double)columns*rows > Integer.MAX_VALUE
* .
*/
public SparseFloatMatrix2D(int rows, int columns, int initialCapacity, float minLoadFactor, float maxLoadFactor) {
try {
setUp(rows, columns);
} catch (IllegalArgumentException exc) { // we can hold rows*columns>Integer.MAX_VALUE cells !
if (!"matrix too large".equals(exc.getMessage()))
throw exc;
}
this.elements = new OpenLongFloatHashMap(initialCapacity, minLoadFactor, maxLoadFactor);
}
/**
* Constructs a matrix with a copy of the given indexes and a single value.
*
* @param rows
* the number of rows the matrix shall have.
* @param columns
* the number of columns the matrix shall have.
* @param rowIndexes
* row indexes
* @param columnIndexes
* column indexes
* @param value
* numerical value
*/
public SparseFloatMatrix2D(int rows, int columns, int[] rowIndexes, int[] columnIndexes, float value) {
try {
setUp(rows, columns);
} catch (IllegalArgumentException exc) { // we can hold rows*columns>Integer.MAX_VALUE cells !
if (!"matrix too large".equals(exc.getMessage()))
throw exc;
}
this.elements = new OpenLongFloatHashMap(rowIndexes.length);
insert(rowIndexes, columnIndexes, value);
}
/**
* Constructs a matrix with a copy of the given indexes and values.
*
* @param rows
* the number of rows the matrix shall have.
* @param columns
* the number of columns the matrix shall have.
* @param rowIndexes
* row indexes
* @param columnIndexes
* column indexes
* @param values
* numerical values
*/
public SparseFloatMatrix2D(int rows, int columns, int[] rowIndexes, int[] columnIndexes, float[] values) {
try {
setUp(rows, columns);
} catch (IllegalArgumentException exc) { // we can hold rows*columns>Integer.MAX_VALUE cells !
if (!"matrix too large".equals(exc.getMessage()))
throw exc;
}
this.elements = new OpenLongFloatHashMap(rowIndexes.length);
insert(rowIndexes, columnIndexes, values);
}
/**
* Constructs a matrix from MatrixVectorReader.
*
* @param reader
* matrix reader
* @throws IOException
*/
public SparseFloatMatrix2D(MatrixVectorReader reader) throws IOException {
MatrixInfo info;
if (reader.hasInfo())
info = reader.readMatrixInfo();
else
info = new MatrixInfo(true, MatrixInfo.MatrixField.Real, MatrixInfo.MatrixSymmetry.General);
if (info.isPattern())
throw new UnsupportedOperationException("Pattern matrices are not supported");
if (info.isDense())
throw new UnsupportedOperationException("Dense matrices are not supported");
if (info.isComplex())
throw new UnsupportedOperationException("Complex matrices are not supported");
MatrixSize size = reader.readMatrixSize(info);
try {
setUp(size.numRows(), size.numColumns());
} catch (IllegalArgumentException exc) { // we can hold rows*columns>Integer.MAX_VALUE cells !
if (!"matrix too large".equals(exc.getMessage()))
throw exc;
}
int numEntries = size.numEntries();
int[] columnIndexes = new int[numEntries];
int[] rowIndexes = new int[numEntries];
float[] values = new float[numEntries];
reader.readCoordinate(rowIndexes, columnIndexes, values);
if (info.isSymmetric() || info.isSkewSymmetric()) {
this.elements = new OpenLongFloatHashMap(2 * rowIndexes.length);
} else {
this.elements = new OpenLongFloatHashMap(rowIndexes.length);
}
insert(rowIndexes, columnIndexes, values);
if (info.isSymmetric()) {
for (int i = 0; i < numEntries; i++) {
if (rowIndexes[i] != columnIndexes[i]) {
set(columnIndexes[i], rowIndexes[i], values[i]);
}
}
} else if (info.isSkewSymmetric()) {
for (int i = 0; i < numEntries; i++) {
if (rowIndexes[i] != columnIndexes[i]) {
set(columnIndexes[i], rowIndexes[i], -values[i]);
}
}
}
}
/**
* Constructs a view with the given parameters.
*
* @param rows
* the number of rows the matrix shall have.
* @param columns
* the number of columns the matrix shall have.
* @param elements
* the cells.
* @param rowZero
* the position of the first element.
* @param columnZero
* the position of the first element.
* @param rowStride
* the number of elements between two rows, i.e.
* index(i+1,j)-index(i,j).
* @param columnStride
* the number of elements between two columns, i.e.
* index(i,j+1)-index(i,j).
* @throws IllegalArgumentException
* if
* rows<0 || columns<0 || (double)columns*rows > Integer.MAX_VALUE
* or flip's are illegal.
*/
protected SparseFloatMatrix2D(int rows, int columns, AbstractLongFloatMap elements, int rowZero, int columnZero,
int rowStride, int columnStride) {
try {
setUp(rows, columns, rowZero, columnZero, rowStride, columnStride);
} catch (IllegalArgumentException exc) { // we can hold rows*columns>Integer.MAX_VALUE cells !
if (!"matrix too large".equals(exc.getMessage()))
throw exc;
}
this.elements = elements;
this.isNoView = false;
}
public FloatMatrix2D assign(cern.colt.function.tfloat.FloatFunction function) {
if (this.isNoView && function instanceof cern.jet.math.tfloat.FloatMult) { // x[i] = mult*x[i]
this.elements.assign(function);
} else {
super.assign(function);
}
return this;
}
public FloatMatrix2D assign(float value) {
// overriden for performance only
if (this.isNoView && value == 0)
this.elements.clear();
else
super.assign(value);
return this;
}
public FloatMatrix2D assign(FloatMatrix2D source) {
// overriden for performance only
if (!(source instanceof SparseFloatMatrix2D)) {
return super.assign(source);
}
SparseFloatMatrix2D other = (SparseFloatMatrix2D) source;
if (other == this)
return this; // nothing to do
checkShape(other);
if (this.isNoView && other.isNoView) { // quickest
this.elements.assign(other.elements);
return this;
}
return super.assign(source);
}
public FloatMatrix2D assign(final FloatMatrix2D y, cern.colt.function.tfloat.FloatFloatFunction function) {
if (!this.isNoView)
return super.assign(y, function);
checkShape(y);
if (function instanceof cern.jet.math.tfloat.FloatPlusMultSecond) { // x[i] = x[i] + alpha*y[i]
final float alpha = ((cern.jet.math.tfloat.FloatPlusMultSecond) function).multiplicator;
if (alpha == 0)
return this; // nothing to do
y.forEachNonZero(new cern.colt.function.tfloat.IntIntFloatFunction() {
public float apply(int i, int j, float value) {
setQuick(i, j, getQuick(i, j) + alpha * value);
return value;
}
});
} else if (function == cern.jet.math.tfloat.FloatFunctions.mult) { // x[i] = x[i] * y[i]
this.elements.forEachPair(new cern.colt.function.tfloat.LongFloatProcedure() {
public boolean apply(long key, float value) {
int i = (int) (key / columns);
int j = (int) (key % columns);
float r = value * y.getQuick(i, j);
if (r != value)
elements.put(key, r);
return true;
}
});
} else if (function == cern.jet.math.tfloat.FloatFunctions.div) { // x[i] = x[i] / y[i]
this.elements.forEachPair(new cern.colt.function.tfloat.LongFloatProcedure() {
public boolean apply(long key, float value) {
int i = (int) (key / columns);
int j = (int) (key % columns);
float r = value / y.getQuick(i, j);
if (r != value)
elements.put(key, r);
return true;
}
});
} else {
super.assign(y, function);
}
return this;
}
/**
* Assigns the result of a function to each cell;
* x[row,col] = function(x[row,col],y[row,col]), where y is given
* in the coordinate form with single numerical value.
*
* @param rowIndexes
* row indexes of y
* @param columnIndexes
* column indexes of y
* @param value
* numerical value of y
* @param function
* a function object taking as first argument the current cell's
* value of this, and as second argument the current
* cell's value of y,
* @return this (for convenience only).
*/
public SparseFloatMatrix2D assign(final int[] rowIndexes, final int[] columnIndexes, final float value,
final cern.colt.function.tfloat.FloatFloatFunction function) {
int size = rowIndexes.length;
if (function == cern.jet.math.tfloat.FloatFunctions.plus) { // x[i] = x[i] + y[i]
for (int i = 0; i < size; i++) {
long row = rowIndexes[i];
long column = columnIndexes[i];
if (row >= rows || column >= columns) {
throw new IndexOutOfBoundsException("row: " + row + ", column: " + column);
}
long index = rowZero + row * rowStride + columnZero + column * columnStride;
float elem = elements.get(index);
float sum = elem + value;
if (sum != 0) {
elements.put(index, sum);
} else {
elements.removeKey(index);
}
}
} else {
for (int i = 0; i < size; i++) {
long row = rowIndexes[i];
long column = columnIndexes[i];
if (row >= rows || column >= columns) {
throw new IndexOutOfBoundsException("row: " + row + ", column: " + column);
}
long index = rowZero + row * rowStride + columnZero + column * columnStride;
float elem = elements.get(index);
float result = function.apply(elem, value);
if (result != 0) {
elements.put(index, result);
} else {
elements.removeKey(index);
}
}
}
return this;
}
/**
* Assigns the result of a function to each cell;
* x[row,col] = function(x[row,col],y[row,col]), where y is given
* in the coordinate form.
*
* @param rowIndexes
* row indexes of y
* @param columnIndexes
* column indexes of y
* @param values
* numerical values of y
* @param function
* a function object taking as first argument the current cell's
* value of this, and as second argument the current
* cell's value of y,
* @return this (for convenience only).
*/
public SparseFloatMatrix2D assign(final int[] rowIndexes, final int[] columnIndexes, final float[] values,
final cern.colt.function.tfloat.FloatFloatFunction function) {
int size = rowIndexes.length;
if (function == cern.jet.math.tfloat.FloatFunctions.plus) { // x[i] = x[i] + y[i]
for (int i = 0; i < size; i++) {
float value = values[i];
long row = rowIndexes[i];
long column = columnIndexes[i];
if (row >= rows || column >= columns) {
throw new IndexOutOfBoundsException("row: " + row + ", column: " + column);
}
long index = rowZero + row * rowStride + columnZero + column * columnStride;
float elem = elements.get(index);
value += elem;
if (value != 0) {
elements.put(index, value);
} else {
elements.removeKey(index);
}
}
} else {
for (int i = 0; i < size; i++) {
float value = values[i];
long row = rowIndexes[i];
long column = columnIndexes[i];
if (row >= rows || column >= columns) {
throw new IndexOutOfBoundsException("row: " + row + ", column: " + column);
}
long index = rowZero + row * rowStride + columnZero + column * columnStride;
float elem = elements.get(index);
value = function.apply(elem, value);
if (value != 0) {
elements.put(index, value);
} else {
elements.removeKey(index);
}
}
}
return this;
}
public int cardinality() {
if (this.isNoView)
return this.elements.size();
else
return super.cardinality();
}
/**
* Returns a new matrix that has the same elements as this matrix, but is in
* a column-compressed form. This method creates a new object (not a view),
* so changes in the returned matrix are NOT reflected in this matrix.
*
* @param sortRowIndexes
* if true, then row indexes in column compressed matrix are
* sorted
*
* @return this matrix in a column-compressed form
*/
public SparseCCFloatMatrix2D getColumnCompressed(boolean sortRowIndexes) {
int nnz = cardinality();
long[] keys = elements.keys().elements();
float[] values = elements.values().elements();
int[] rowIndexes = new int[nnz];
int[] columnIndexes = new int[nnz];
for (int k = 0; k < nnz; k++) {
long key = keys[k];
rowIndexes[k] = (int) (key / columns);
columnIndexes[k] = (int) (key % columns);
}
return new SparseCCFloatMatrix2D(rows, columns, rowIndexes, columnIndexes, values, false, false, sortRowIndexes);
}
/**
* Returns a new matrix that has the same elements as this matrix, but is in
* a column-compressed modified form. This method creates a new object (not
* a view), so changes in the returned matrix are NOT reflected in this
* matrix.
*
* @return this matrix in a column-compressed modified form
*/
public SparseCCMFloatMatrix2D getColumnCompressedModified() {
SparseCCMFloatMatrix2D A = new SparseCCMFloatMatrix2D(rows, columns);
int nnz = cardinality();
long[] keys = elements.keys().elements();
float[] values = elements.values().elements();
for (int i = 0; i < nnz; i++) {
int row = (int) (keys[i] / columns);
int column = (int) (keys[i] % columns);
A.setQuick(row, column, values[i]);
}
return A;
}
/**
* Returns a new matrix that has the same elements as this matrix, but is in
* a row-compressed form. This method creates a new object (not a view), so
* changes in the returned matrix are NOT reflected in this matrix.
*
* @param sortColumnIndexes
* if true, then column indexes in row compressed matrix are
* sorted
*
* @return this matrix in a row-compressed form
*/
public SparseRCFloatMatrix2D getRowCompressed(boolean sortColumnIndexes) {
int nnz = cardinality();
long[] keys = elements.keys().elements();
float[] values = elements.values().elements();
final int[] rowIndexes = new int[nnz];
final int[] columnIndexes = new int[nnz];
for (int k = 0; k < nnz; k++) {
long key = keys[k];
rowIndexes[k] = (int) (key / columns);
columnIndexes[k] = (int) (key % columns);
}
return new SparseRCFloatMatrix2D(rows, columns, rowIndexes, columnIndexes, values, false, false,
sortColumnIndexes);
}
/**
* Returns a new matrix that has the same elements as this matrix, but is in
* a row-compressed modified form. This method creates a new object (not a
* view), so changes in the returned matrix are NOT reflected in this
* matrix.
*
* @return this matrix in a row-compressed modified form
*/
public SparseRCMFloatMatrix2D getRowCompressedModified() {
SparseRCMFloatMatrix2D A = new SparseRCMFloatMatrix2D(rows, columns);
int nnz = cardinality();
long[] keys = elements.keys().elements();
float[] values = elements.values().elements();
for (int i = 0; i < nnz; i++) {
int row = (int) (keys[i] / columns);
int column = (int) (keys[i] % columns);
A.setQuick(row, column, values[i]);
}
return A;
}
public AbstractLongFloatMap elements() {
return elements;
}
public void ensureCapacity(int minCapacity) {
this.elements.ensureCapacity(minCapacity);
}
public FloatMatrix2D forEachNonZero(final cern.colt.function.tfloat.IntIntFloatFunction function) {
if (this.isNoView) {
this.elements.forEachPair(new cern.colt.function.tfloat.LongFloatProcedure() {
public boolean apply(long key, float value) {
int i = (int) (key / columns);
int j = (int) (key % columns);
float r = function.apply(i, j, value);
if (r != value)
elements.put(key, r);
return true;
}
});
} else {
super.forEachNonZero(function);
}
return this;
}
public synchronized float getQuick(int row, int column) {
return this.elements.get((long) rowZero + (long) row * (long) rowStride + (long) columnZero + (long) column
* (long) columnStride);
}
public long index(int row, int column) {
return (long) rowZero + (long) row * (long) rowStride + (long) columnZero + (long) column * (long) columnStride;
}
public FloatMatrix2D like(int rows, int columns) {
return new SparseFloatMatrix2D(rows, columns);
}
public FloatMatrix1D like1D(int size) {
return new SparseFloatMatrix1D(size);
}
public synchronized void setQuick(int row, int column, float value) {
long index = (long) rowZero + (long) row * (long) rowStride + (long) columnZero + (long) column
* (long) columnStride;
if (value == 0)
this.elements.removeKey(index);
else
this.elements.put(index, value);
}
public String toString() {
StringBuilder builder = new StringBuilder();
builder.append(rows).append(" x ").append(columns).append(" sparse matrix, nnz = ").append(cardinality())
.append('\n');
for (int r = 0; r < rows; r++) {
for (int c = 0; c < columns; c++) {
float elem = getQuick(r, c);
if (elem != 0) {
builder.append('(').append(r).append(',').append(c).append(')').append('\t').append(elem).append(
'\n');
}
}
}
return builder.toString();
}
public void trimToSize() {
this.elements.trimToSize();
}
public FloatMatrix1D vectorize() {
SparseFloatMatrix1D v = new SparseFloatMatrix1D((int) size());
int idx = 0;
for (int c = 0; c < columns; c++) {
for (int r = 0; r < rows; r++) {
float elem = getQuick(r, c);
v.setQuick(idx++, elem);
}
}
return v;
}
public FloatMatrix1D zMult(FloatMatrix1D y, FloatMatrix1D z, final float alpha, float beta, final boolean transposeA) {
int rowsA = rows;
int columnsA = columns;
if (transposeA) {
rowsA = columns;
columnsA = rows;
}
boolean ignore = (z == null);
if (z == null)
z = new DenseFloatMatrix1D(rowsA);
if (!(this.isNoView && y instanceof DenseFloatMatrix1D && z instanceof DenseFloatMatrix1D)) {
return super.zMult(y, z, alpha, beta, transposeA);
}
if (columnsA != y.size() || rowsA > z.size())
throw new IllegalArgumentException("Incompatible args: "
+ ((transposeA ? viewDice() : this).toStringShort()) + ", " + y.toStringShort() + ", "
+ z.toStringShort());
if (!ignore)
z.assign(cern.jet.math.tfloat.FloatFunctions.mult(beta));
DenseFloatMatrix1D zz = (DenseFloatMatrix1D) z;
final float[] elementsZ = zz.elements;
final int strideZ = zz.stride();
final int zeroZ = (int) z.index(0);
DenseFloatMatrix1D yy = (DenseFloatMatrix1D) y;
final float[] elementsY = yy.elements;
final int strideY = yy.stride();
final int zeroY = (int) y.index(0);
if (elementsY == null || elementsZ == null)
throw new InternalError();
this.elements.forEachPair(new cern.colt.function.tfloat.LongFloatProcedure() {
public boolean apply(long key, float value) {
int i = (int) (key / columns);
int j = (int) (key % columns);
if (transposeA) {
int tmp = i;
i = j;
j = tmp;
}
elementsZ[zeroZ + strideZ * i] += alpha * value * elementsY[zeroY + strideY * j];
return true;
}
});
return z;
}
public FloatMatrix2D zMult(FloatMatrix2D B, FloatMatrix2D C, final float alpha, float beta,
final boolean transposeA, boolean transposeB) {
if (!(this.isNoView)) {
return super.zMult(B, C, alpha, beta, transposeA, transposeB);
}
if (transposeB)
B = B.viewDice();
int rowsA = rows;
int columnsA = columns;
if (transposeA) {
rowsA = columns;
columnsA = rows;
}
int p = B.columns();
boolean ignore = (C == null);
if (C == null)
C = new DenseFloatMatrix2D(rowsA, p);
if (B.rows() != columnsA)
throw new IllegalArgumentException("Matrix2D inner dimensions must agree:" + toStringShort() + ", "
+ (transposeB ? B.viewDice() : B).toStringShort());
if (C.rows() != rowsA || C.columns() != p)
throw new IllegalArgumentException("Incompatibel result matrix: " + toStringShort() + ", "
+ (transposeB ? B.viewDice() : B).toStringShort() + ", " + C.toStringShort());
if (this == C || B == C)
throw new IllegalArgumentException("Matrices must not be identical");
if (!ignore)
C.assign(cern.jet.math.tfloat.FloatFunctions.mult(beta));
// cache views
final FloatMatrix1D[] Brows = new FloatMatrix1D[columnsA];
for (int i = columnsA; --i >= 0;)
Brows[i] = B.viewRow(i);
final FloatMatrix1D[] Crows = new FloatMatrix1D[rowsA];
for (int i = rowsA; --i >= 0;)
Crows[i] = C.viewRow(i);
final cern.jet.math.tfloat.FloatPlusMultSecond fun = cern.jet.math.tfloat.FloatPlusMultSecond.plusMult(0);
this.elements.forEachPair(new cern.colt.function.tfloat.LongFloatProcedure() {
public boolean apply(long key, float value) {
int i = (int) (key / columns);
int j = (int) (key % columns);
fun.multiplicator = value * alpha;
if (!transposeA)
Crows[i].assign(Brows[j], fun);
else
Crows[j].assign(Brows[i], fun);
return true;
}
});
return C;
}
private void insert(int[] rowIndexes, int[] columnIndexes, float value) {
int size = rowIndexes.length;
for (int i = 0; i < size; i++) {
long row = rowIndexes[i];
long column = columnIndexes[i];
if (row >= rows || column >= columns) {
throw new IndexOutOfBoundsException("row: " + row + ", column: " + column);
}
if (value != 0) {
long index = rowZero + row * rowStride + columnZero + column * columnStride;
float elem = elements.get(index);
if (elem == 0) {
elements.put(index, value);
} else {
float sum = elem + value;
if (sum == 0) {
elements.removeKey(index);
} else {
elements.put(index, sum);
}
}
}
}
}
private void insert(int[] rowIndexes, int[] columnIndexes, float[] values) {
int size = rowIndexes.length;
for (int i = 0; i < size; i++) {
float value = values[i];
long row = rowIndexes[i];
long column = columnIndexes[i];
if (row >= rows || column >= columns) {
throw new IndexOutOfBoundsException("row: " + row + ", column: " + column);
}
if (value != 0) {
long index = rowZero + row * rowStride + columnZero + column * columnStride;
float elem = elements.get(index);
if (elem == 0) {
elements.put(index, value);
} else {
float sum = elem + value;
if (sum == 0) {
elements.removeKey(index);
} else {
elements.put(index, sum);
}
}
}
}
}
protected boolean haveSharedCellsRaw(FloatMatrix2D other) {
if (other instanceof SelectedSparseFloatMatrix2D) {
SelectedSparseFloatMatrix2D otherMatrix = (SelectedSparseFloatMatrix2D) other;
return this.elements == otherMatrix.elements;
} else if (other instanceof SparseFloatMatrix2D) {
SparseFloatMatrix2D otherMatrix = (SparseFloatMatrix2D) other;
return this.elements == otherMatrix.elements;
}
return false;
}
protected FloatMatrix1D like1D(int size, int offset, int stride) {
return new SparseFloatMatrix1D(size, this.elements, offset, stride);
}
protected FloatMatrix2D viewSelectionLike(int[] rowOffsets, int[] columnOffsets) {
return new SelectedSparseFloatMatrix2D(this.elements, rowOffsets, columnOffsets, 0);
}
}