/*
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;
import cern.colt.list.DoubleArrayList;
import cern.colt.list.IntArrayList;
import cern.colt.matrix.impl.AbstractMatrix2D;
import cern.colt.matrix.impl.DenseDoubleMatrix1D;
import cern.colt.matrix.impl.DenseDoubleMatrix2D;
/**
Abstract base class for 2-d matrices holding double elements.
First see the package summary and javadoc tree view to get the broad picture.
A matrix has a number of rows and columns, which are assigned upon instance construction - The matrix's size is then rows()*columns().
Elements are accessed via [row,column] coordinates.
Legal coordinates range from [0,0] to [rows()-1,columns()-1].
Any attempt to access an element at a coordinate column<0 || column>=columns() || row<0 || row>=rows() will throw an IndexOutOfBoundsException.
Note that this implementation is not synchronized.
@author wolfgang.hoschek@cern.ch
@version 1.0, 09/24/99
*/
public abstract class DoubleMatrix2D extends AbstractMatrix2D {
/**
* Makes this class non instantiable, but still let's others inherit from it.
*/
protected DoubleMatrix2D() {}
/**
Applies a function to each cell and aggregates the results.
Returns a value v such that v==a(size()) where a(i) == aggr( a(i-1), f(get(row,column)) ) and terminators are a(1) == f(get(0,0)), a(0)==Double.NaN.
Example:
cern.jet.math.Functions F = cern.jet.math.Functions.functions;
2 x 2 matrix
0 1
2 3
// Sum( x[row,col]*x[row,col] )
matrix.aggregate(F.plus,F.square);
--> 14
For further examples, see the package doc.
@param aggr an aggregation function taking as first argument the current aggregation and as second argument the transformed current cell value.
@param f a function transforming the current cell value.
@return the aggregated measure.
@see cern.jet.math.Functions
*/
public double aggregate(cern.colt.function.DoubleDoubleFunction aggr, cern.colt.function.DoubleFunction f) {
if (size()==0) return Double.NaN;
double a = f.apply(getQuick(rows-1,columns-1));
int d = 1; // last cell already done
for (int row=rows; --row >= 0; ) {
for (int column=columns-d; --column >= 0; ) {
a = aggr.apply(a, f.apply(getQuick(row,column)));
}
d = 0;
}
return a;
}
/**
Applies a function to each corresponding cell of two matrices and aggregates the results.
Returns a value v such that v==a(size()) where a(i) == aggr( a(i-1), f(get(row,column),other.get(row,column)) ) and terminators are a(1) == f(get(0,0),other.get(0,0)), a(0)==Double.NaN.
Example:
cern.jet.math.Functions F = cern.jet.math.Functions.functions;
x == 2 x 2 matrix
0 1
2 3
y == 2 x 2 matrix
0 1
2 3
// Sum( x[row,col] * y[row,col] )
x.aggregate(y, F.plus, F.mult);
--> 14
// Sum( (x[row,col] + y[row,col])^2 )
x.aggregate(y, F.plus, F.chain(F.square,F.plus));
--> 56
For further examples, see the package doc.
@param aggr an aggregation function taking as first argument the current aggregation and as second argument the transformed current cell values.
@param f a function transforming the current cell values.
@return the aggregated measure.
@throws IllegalArgumentException if columns() != other.columns() || rows() != other.rows()
@see cern.jet.math.Functions
*/
public double aggregate(DoubleMatrix2D other, cern.colt.function.DoubleDoubleFunction aggr, cern.colt.function.DoubleDoubleFunction f) {
checkShape(other);
if (size()==0) return Double.NaN;
double a = f.apply(getQuick(rows-1,columns-1),other.getQuick(rows-1,columns-1));
int d = 1; // last cell already done
for (int row=rows; --row >= 0; ) {
for (int column=columns-d; --column >= 0; ) {
a = aggr.apply(a, f.apply(getQuick(row,column), other.getQuick(row,column)));
}
d = 0;
}
return a;
}
/**
* Sets all cells to the state specified by values.
* values is required to have the form values[row][column]
* and have exactly the same number of rows and columns as the receiver.
*
* 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 cells.
* @return this (for convenience only).
* @throws IllegalArgumentException if values.length != rows() || for any 0 <= row < rows(): values[row].length != columns().
*/
public DoubleMatrix2D assign(double[][] values) {
if (values.length != rows) throw new IllegalArgumentException("Must have same number of rows: rows="+values.length+"rows()="+rows());
for (int row=rows; --row >= 0;) {
double[] currentRow = values[row];
if (currentRow.length != columns) throw new IllegalArgumentException("Must have same number of columns in every row: columns="+currentRow.length+"columns()="+columns());
for (int column=columns; --column >= 0;) {
setQuick(row,column,currentRow[column]);
}
}
return this;
}
/**
* Sets all cells to the state specified by value.
* @param value the value to be filled into the cells.
* @return this (for convenience only).
*/
public DoubleMatrix2D assign(double value) {
int r = rows;
int c = columns;
//for (int row=rows; --row >= 0;) {
// for (int column=columns; --column >= 0;) {
for (int row=0; row < r; row++) {
for (int column=0; column < c; column++) {
setQuick(row,column,value);
}
}
return this;
}
/**
Assigns the result of a function to each cell; x[row,col] = function(x[row,col]).
Example:
matrix = 2 x 2 matrix
0.5 1.5
2.5 3.5
// change each cell to its sine
matrix.assign(cern.jet.math.Functions.sin);
-->
2 x 2 matrix
0.479426 0.997495
0.598472 -0.350783
For further examples, see the package doc.
@param function a function object taking as argument the current cell's value.
@return this (for convenience only).
@see cern.jet.math.Functions
*/
public DoubleMatrix2D assign(cern.colt.function.DoubleFunction function) {
for (int row=rows; --row >= 0; ) {
for (int column=columns; --column >= 0; ) {
setQuick(row,column, function.apply(getQuick(row,column)));
}
}
return this;
}
/**
* Replaces all cell values of the receiver with the values of another matrix.
* Both matrices must have the same number of rows and columns.
* If both matrices share the same cells (as is the case if they are views derived from the same matrix) and intersect in an ambiguous way, then replaces as if using an intermediate auxiliary deep copy of other.
*
* @param other the source matrix to copy from (may be identical to the receiver).
* @return this (for convenience only).
* @throws IllegalArgumentException if columns() != other.columns() || rows() != other.rows()
*/
public DoubleMatrix2D assign(DoubleMatrix2D other) {
if (other==this) return this;
checkShape(other);
if (haveSharedCells(other)) other = other.copy();
//for (int row=0; row= 0;) {
for (int column=columns; --column >= 0;) {
setQuick(row,column,other.getQuick(row,column));
}
}
return this;
}
/**
Assigns the result of a function to each cell; x[row,col] = function(x[row,col],y[row,col]).
Example:
// assign x[row,col] = x[row,col]y[row,col]
m1 = 2 x 2 matrix
0 1
2 3
m2 = 2 x 2 matrix
0 2
4 6
m1.assign(m2, cern.jet.math.Functions.pow);
-->
m1 == 2 x 2 matrix
1 1
16 729
For further examples, see the package doc.
@param y the secondary matrix to operate on.
@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).
@throws IllegalArgumentException if columns() != other.columns() || rows() != other.rows()
@see cern.jet.math.Functions
*/
public DoubleMatrix2D assign(DoubleMatrix2D y, cern.colt.function.DoubleDoubleFunction function) {
checkShape(y);
for (int row=rows; --row >= 0; ) {
for (int column=columns; --column >= 0; ) {
setQuick(row,column, function.apply(getQuick(row,column), y.getQuick(row,column)));
}
}
return this;
}
/**
* Returns the number of cells having non-zero values; ignores tolerance.
*/
public int cardinality() {
int cardinality = 0;
for (int row=rows; --row >= 0;) {
for (int column=columns; --column >= 0;) {
if (getQuick(row,column) != 0) cardinality++;
}
}
return cardinality;
}
/**
* Constructs and returns a deep copy of the receiver.
*
* Note that the returned matrix is an independent deep copy.
* The returned matrix is not backed by this matrix, so changes in the returned matrix are not reflected in this matrix, and vice-versa.
*
* @return a deep copy of the receiver.
*/
public DoubleMatrix2D copy() {
return like().assign(this);
}
/**
* Returns whether all cells are equal to the given value.
*
* @param value the value to test against.
* @return true if all cells are equal to the given value, false otherwise.
*/
public boolean equals(double value) {
return cern.colt.matrix.linalg.Property.DEFAULT.equals(this,value);
}
/**
* Compares this object against the specified object.
* The result is true if and only if the argument is
* not null and is at least a DoubleMatrix2D object
* that has the same number of columns and rows as the receiver and
* has exactly the same values at the same coordinates.
* @param obj the object to compare with.
* @return true if the objects are the same;
* false otherwise.
*/
public boolean equals(Object obj) {
if (this == obj) return true;
if (obj == null) return false;
if (!(obj instanceof DoubleMatrix2D)) return false;
return cern.colt.matrix.linalg.Property.DEFAULT.equals(this,(DoubleMatrix2D) obj);
}
/**
* Assigns the result of a function to each non-zero cell; x[row,col] = function(x[row,col]).
* Use this method for fast special-purpose iteration.
* If you want to modify another matrix instead of this (i.e. work in read-only mode), simply return the input value unchanged.
*
* Parameters to function are as follows: first==row, second==column, third==nonZeroValue.
*
* @param function a function object taking as argument the current non-zero cell's row, column and value.
* @return this (for convenience only).
*/
public DoubleMatrix2D forEachNonZero(final cern.colt.function.IntIntDoubleFunction function) {
for (int row=rows; --row >= 0;) {
for (int column=columns; --column >= 0;) {
double value = getQuick(row,column);
if (value!=0) {
double r = function.apply(row,column,value);
if (r!=value) setQuick(row,column,r);
}
}
}
return this;
}
/**
* Returns the matrix cell value at coordinate [row,column].
*
* @param row the index of the row-coordinate.
* @param column the index of the column-coordinate.
* @return the value of the specified cell.
* @throws IndexOutOfBoundsException if column<0 || column>=columns() || row<0 || row>=rows()
*/
public double get(int row, int column) {
if (column<0 || column>=columns || row<0 || row>=rows) throw new IndexOutOfBoundsException("row:"+row+", column:"+column);
return getQuick(row,column);
}
/**
* Returns the content of this matrix if it is a wrapper; or this otherwise.
* Override this method in wrappers.
*/
protected DoubleMatrix2D getContent() {
return this;
}
/**
Fills the coordinates and values of cells having non-zero values into the specified lists.
Fills into the lists, starting at index 0.
After this call returns the specified lists all have a new size, the number of non-zero values.
In general, fill order is unspecified.
This implementation fills like for (row = 0..rows-1) for (column = 0..columns-1) do ... .
However, subclasses are free to us any other order, even an order that may change over time as cell values are changed.
(Of course, result lists indexes are guaranteed to correspond to the same cell).
Example:
2 x 3 matrix:
0, 0, 8
0, 7, 0
-->
rowList = (0,1)
columnList = (2,1)
valueList = (8,7)
In other words, get(0,2)==8, get(1,1)==7.
@param rowList the list to be filled with row indexes, can have any size.
@param columnList the list to be filled with column indexes, can have any size.
@param valueList the list to be filled with values, can have any size.
*/
public void getNonZeros(IntArrayList rowList, IntArrayList columnList, DoubleArrayList valueList) {
rowList.clear();
columnList.clear();
valueList.clear();
int r = rows;
int c = columns;
for (int row=0; row < r; row++) {
for (int column=0; column < c; column++) {
double value = getQuick(row,column);
if (value != 0) {
rowList.add(row);
columnList.add(column);
valueList.add(value);
}
}
}
}
/**
* Returns the matrix cell value at coordinate [row,column].
*
* Provided with invalid parameters this method may return invalid objects without throwing any exception.
* You should only use this method when you are absolutely sure that the coordinate is within bounds.
* Precondition (unchecked): 0 <= column < columns() && 0 <= row < rows().
*
* @param row the index of the row-coordinate.
* @param column the index of the column-coordinate.
* @return the value at the specified coordinate.
*/
public abstract double getQuick(int row, int column);
/**
* Returns true if both matrices share at least one identical cell.
*/
protected boolean haveSharedCells(DoubleMatrix2D other) {
if (other==null) return false;
if (this==other) return true;
return getContent().haveSharedCellsRaw(other.getContent());
}
/**
* Returns true if both matrices share at least one identical cell.
*/
protected boolean haveSharedCellsRaw(DoubleMatrix2D other) {
return false;
}
/**
* Construct and returns a new empty matrix of the same dynamic type as the receiver, having the same number of rows and columns.
* For example, if the receiver is an instance of type DenseDoubleMatrix2D the new matrix must also be of type DenseDoubleMatrix2D,
* if the receiver is an instance of type SparseDoubleMatrix2D the new matrix must also be of type SparseDoubleMatrix2D, etc.
* In general, the new matrix should have internal parametrization as similar as possible.
*
* @return a new empty matrix of the same dynamic type.
*/
public DoubleMatrix2D like() {
return like(rows,columns);
}
/**
* Construct and returns a new empty matrix of the same dynamic type as the receiver, having the specified number of rows and columns.
* For example, if the receiver is an instance of type DenseDoubleMatrix2D the new matrix must also be of type DenseDoubleMatrix2D,
* if the receiver is an instance of type SparseDoubleMatrix2D the new matrix must also be of type SparseDoubleMatrix2D, etc.
* In general, the new matrix should have internal parametrization as similar as possible.
*
* @param rows the number of rows the matrix shall have.
* @param columns the number of columns the matrix shall have.
* @return a new empty matrix of the same dynamic type.
*/
public abstract DoubleMatrix2D like(int rows, int columns);
/**
* Construct and returns a new 1-d matrix of the corresponding dynamic type, entirelly independent of the receiver.
* For example, if the receiver is an instance of type DenseDoubleMatrix2D the new matrix must be of type DenseDoubleMatrix1D,
* if the receiver is an instance of type SparseDoubleMatrix2D the new matrix must be of type SparseDoubleMatrix1D, etc.
*
* @param size the number of cells the matrix shall have.
* @return a new matrix of the corresponding dynamic type.
*/
public abstract DoubleMatrix1D like1D(int size);
/**
* Construct and returns a new 1-d matrix of the corresponding dynamic type, sharing the same cells.
* For example, if the receiver is an instance of type DenseDoubleMatrix2D the new matrix must be of type DenseDoubleMatrix1D,
* if the receiver is an instance of type SparseDoubleMatrix2D the new matrix must be of type SparseDoubleMatrix1D, etc.
*
* @param size the number of cells the matrix shall have.
* @param zero the index of the first element.
* @param stride the number of indexes between any two elements, i.e. index(i+1)-index(i).
* @return a new matrix of the corresponding dynamic type.
*/
protected abstract DoubleMatrix1D like1D(int size, int zero, int stride);
/**
* Sets the matrix cell at coordinate [row,column] to the specified value.
*
* @param row the index of the row-coordinate.
* @param column the index of the column-coordinate.
* @param value the value to be filled into the specified cell.
* @throws IndexOutOfBoundsException if column<0 || column>=columns() || row<0 || row>=rows()
*/
public void set(int row, int column, double value) {
if (column<0 || column>=columns || row<0 || row>=rows) throw new IndexOutOfBoundsException("row:"+row+", column:"+column);
setQuick(row,column,value);
}
/**
* Sets the matrix cell at coordinate [row,column] to the specified value.
*
*
Provided with invalid parameters this method may access illegal indexes without throwing any exception.
* You should only use this method when you are absolutely sure that the coordinate is within bounds.
* Precondition (unchecked): 0 <= column < columns() && 0 <= row < rows().
*
* @param row the index of the row-coordinate.
* @param column the index of the column-coordinate.
* @param value the value to be filled into the specified cell.
*/
public abstract void setQuick(int row, int column, double value);
/**
* Constructs and returns a 2-dimensional array containing the cell values.
* The returned array values has the form values[row][column]
* and has the same number of rows and columns as the receiver.
*
* The values are copied. So subsequent changes in values are not reflected in the matrix, and vice-versa.
*
* @return an array filled with the values of the cells.
*/
public double[][] toArray() {
double[][] values = new double[rows][columns];
for (int row=rows; --row >= 0;) {
double[] currentRow = values[row];
for (int column=columns; --column >= 0;) {
currentRow[column] = getQuick(row,column);
}
}
return values;
}
/**
* Returns a string representation using default formatting.
* @see cern.colt.matrix.doublealgo.Formatter
*/
public String toString() {
return new cern.colt.matrix.doublealgo.Formatter().toString(this);
}
/**
* Constructs and returns a new view equal to the receiver.
* The view is a shallow clone. Calls clone() and casts the result.
*
* Note that the view is not a deep copy.
* The returned matrix is backed by this matrix, so changes in the returned matrix are reflected in this matrix, and vice-versa.
*
* Use {@link #copy()} to construct an independent deep copy rather than a new view.
*
* @return a new view of the receiver.
*/
protected DoubleMatrix2D view() {
return (DoubleMatrix2D) clone();
}
/**
Constructs and returns a new slice view representing the rows of the given column.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
To obtain a slice view on subranges, construct a sub-ranging view (viewPart(...)), then apply this method to the sub-range view.
Example:
2 x 3 matrix:
1, 2, 3
4, 5, 6
viewColumn(0) ==>
Matrix1D of size 2:
1, 4
@param column the column to fix.
@return a new slice view.
@throws IndexOutOfBoundsException if column < 0 || column >= columns().
@see #viewRow(int)
*/
public DoubleMatrix1D viewColumn(int column) {
checkColumn(column);
int viewSize = this.rows;
int viewZero = index(0,column);
int viewStride = this.rowStride;
return like1D(viewSize,viewZero,viewStride);
}
/**
Constructs and returns a new flip view along the column axis.
What used to be column 0 is now column columns()-1, ..., what used to be column columns()-1 is now column 0.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
Example:
2 x 3 matrix:
1, 2, 3
4, 5, 6
columnFlip ==>
2 x 3 matrix:
3, 2, 1
6, 5, 4
columnFlip ==>
2 x 3 matrix:
1, 2, 3
4, 5, 6
@return a new flip view.
@see #viewRowFlip()
*/
public DoubleMatrix2D viewColumnFlip() {
return (DoubleMatrix2D) (view().vColumnFlip());
}
/**
Constructs and returns a new dice (transposition) view; Swaps axes; example: 3 x 4 matrix --> 4 x 3 matrix.
The view has both dimensions exchanged; what used to be columns become rows, what used to be rows become columns.
In other words: view.get(row,column)==this.get(column,row).
This is a zero-copy transposition, taking O(1), i.e. constant time.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
Use idioms like result = viewDice(A).copy() to generate an independent transposed matrix.
Example:
2 x 3 matrix:
1, 2, 3
4, 5, 6
transpose ==>
3 x 2 matrix:
1, 4
2, 5
3, 6
transpose ==>
2 x 3 matrix:
1, 2, 3
4, 5, 6
@return a new dice view.
*/
public DoubleMatrix2D viewDice() {
return (DoubleMatrix2D) (view().vDice());
}
/**
Constructs and returns a new sub-range view that is a height x width sub matrix starting at [row,column].
Operations on the returned view can only be applied to the restricted range.
Any attempt to access coordinates not contained in the view will throw an IndexOutOfBoundsException.
Note that the view is really just a range restriction:
The returned matrix is backed by this matrix, so changes in the returned matrix are reflected in this matrix, and vice-versa.
The view contains the cells from [row,column] to [row+height-1,column+width-1], all inclusive.
and has view.rows() == height; view.columns() == width;.
A view's legal coordinates are again zero based, as usual.
In other words, legal coordinates of the view range from [0,0] to [view.rows()-1==height-1,view.columns()-1==width-1].
As usual, any attempt to access a cell at a coordinate column<0 || column>=view.columns() || row<0 || row>=view.rows() will throw an IndexOutOfBoundsException.
@param row The index of the row-coordinate.
@param column The index of the column-coordinate.
@param height The height of the box.
@param width The width of the box.
@throws IndexOutOfBoundsException if column<0 || width<0 || column+width>columns() || row<0 || height<0 || row+height>rows()
@return the new view.
*/
public DoubleMatrix2D viewPart(int row, int column, int height, int width) {
return (DoubleMatrix2D) (view().vPart(row,column,height,width));
}
/**
Constructs and returns a new slice view representing the columns of the given row.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
To obtain a slice view on subranges, construct a sub-ranging view (viewPart(...)), then apply this method to the sub-range view.
Example:
2 x 3 matrix:
1, 2, 3
4, 5, 6
viewRow(0) ==>
Matrix1D of size 3:
1, 2, 3
@param row the row to fix.
@return a new slice view.
@throws IndexOutOfBoundsException if row < 0 || row >= rows().
@see #viewColumn(int)
*/
public DoubleMatrix1D viewRow(int row) {
checkRow(row);
int viewSize = this.columns;
int viewZero = index(row,0);
int viewStride = this.columnStride;
return like1D(viewSize,viewZero,viewStride);
}
/**
Constructs and returns a new flip view along the row axis.
What used to be row 0 is now row rows()-1, ..., what used to be row rows()-1 is now row 0.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
Example:
2 x 3 matrix:
1, 2, 3
4, 5, 6
rowFlip ==>
2 x 3 matrix:
4, 5, 6
1, 2, 3
rowFlip ==>
2 x 3 matrix:
1, 2, 3
4, 5, 6
@return a new flip view.
@see #viewColumnFlip()
*/
public DoubleMatrix2D viewRowFlip() {
return (DoubleMatrix2D) (view().vRowFlip());
}
/**
Constructs and returns a new selection view that is a matrix holding the indicated cells.
There holds view.rows() == rowIndexes.length, view.columns() == columnIndexes.length and view.get(i,j) == this.get(rowIndexes[i],columnIndexes[j]).
Indexes can occur multiple times and can be in arbitrary order.
Example:
this = 2 x 3 matrix:
1, 2, 3
4, 5, 6
rowIndexes = (0,1)
columnIndexes = (1,0,1,0)
-->
view = 2 x 4 matrix:
2, 1, 2, 1
5, 4, 5, 4
Note that modifying the index arguments after this call has returned has no effect on the view.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
To indicate "all" rows or "all columns", simply set the respective parameter
@param rowIndexes The rows of the cells that shall be visible in the new view. To indicate that all rows shall be visible, simply set this parameter to null.
@param columnIndexes The columns of the cells that shall be visible in the new view. To indicate that all columns shall be visible, simply set this parameter to null.
@return the new view.
@throws IndexOutOfBoundsException if !(0 <= rowIndexes[i] < rows()) for any i=0..rowIndexes.length()-1.
@throws IndexOutOfBoundsException if !(0 <= columnIndexes[i] < columns()) for any i=0..columnIndexes.length()-1.
*/
public DoubleMatrix2D viewSelection(int[] rowIndexes, int[] columnIndexes) {
// check for "all"
if (rowIndexes==null) {
rowIndexes = new int[rows];
for (int i=rows; --i >= 0; ) rowIndexes[i] = i;
}
if (columnIndexes==null) {
columnIndexes = new int[columns];
for (int i=columns; --i >= 0; ) columnIndexes[i] = i;
}
checkRowIndexes(rowIndexes);
checkColumnIndexes(columnIndexes);
int[] rowOffsets = new int[rowIndexes.length];
int[] columnOffsets = new int[columnIndexes.length];
for (int i=rowIndexes.length; --i >= 0; ) {
rowOffsets[i] = _rowOffset(_rowRank(rowIndexes[i]));
}
for (int i=columnIndexes.length; --i >= 0; ) {
columnOffsets[i] = _columnOffset(_columnRank(columnIndexes[i]));
}
return viewSelectionLike(rowOffsets,columnOffsets);
}
/**
Constructs and returns a new selection view that is a matrix holding all rows matching the given condition.
Applies the condition to each row and takes only those row where condition.apply(viewRow(i)) yields true.
To match columns, use a dice view.
Example:
// extract and view all rows which have a value < threshold in the first column (representing "age")
final double threshold = 16;
matrix.viewSelection(
new DoubleMatrix1DProcedure() {
public final boolean apply(DoubleMatrix1D m) { return m.get(0) < threshold; }
}
);
// extract and view all rows with RMS < threshold
// The RMS (Root-Mean-Square) is a measure of the average "size" of the elements of a data sequence.
matrix = 0 1 2 3
final double threshold = 0.5;
matrix.viewSelection(
new DoubleMatrix1DProcedure() {
public final boolean apply(DoubleMatrix1D m) { return Math.sqrt(m.aggregate(F.plus,F.square) / m.size()) < threshold; }
}
);
For further examples, see the package doc.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
@param condition The condition to be matched.
@return the new view.
*/
public DoubleMatrix2D viewSelection(DoubleMatrix1DProcedure condition) {
IntArrayList matches = new IntArrayList();
for (int i=0; i < rows; i++) {
if (condition.apply(viewRow(i))) matches.add(i);
}
matches.trimToSize();
return viewSelection(matches.elements(), null); // take all columns
}
/**
* Construct and returns a new selection view.
*
* @param rowOffsets the offsets of the visible elements.
* @param columnOffsets the offsets of the visible elements.
* @return a new view.
*/
protected abstract DoubleMatrix2D viewSelectionLike(int[] rowOffsets, int[] columnOffsets);
/**
Sorts the matrix rows into ascending order, according to the natural ordering of the matrix values in the given column.
This sort is guaranteed to be stable.
For further information, see {@link cern.colt.matrix.doublealgo.Sorting#sort(DoubleMatrix2D,int)}.
For more advanced sorting functionality, see {@link cern.colt.matrix.doublealgo.Sorting}.
@return a new sorted vector (matrix) view.
@throws IndexOutOfBoundsException if column < 0 || column >= columns().
*/
public DoubleMatrix2D viewSorted(int column) {
return cern.colt.matrix.doublealgo.Sorting.mergeSort.sort(this,column);
}
/**
Constructs and returns a new stride view which is a sub matrix consisting of every i-th cell.
More specifically, the view has this.rows()/rowStride rows and this.columns()/columnStride columns holding cells this.get(i*rowStride,j*columnStride) for all i = 0..rows()/rowStride - 1, j = 0..columns()/columnStride - 1.
The returned view is backed by this matrix, so changes in the returned view are reflected in this matrix, and vice-versa.
@param rowStride the row step factor.
@param columnStride the column step factor.
@return a new view.
@throws IndexOutOfBoundsException if rowStride<=0 || columnStride<=0.
*/
public DoubleMatrix2D viewStrides(int rowStride, int columnStride) {
return (DoubleMatrix2D) (view().vStrides(rowStride, columnStride));
}
/**
* Applies a procedure to each cell's value.
* Iterates downwards from [rows()-1,columns()-1] to [0,0],
* as demonstrated by this snippet:
*
* for (int row=rows; --row >=0;) {
* for (int column=columns; --column >= 0;) {
* if (!procedure.apply(getQuick(row,column))) return false;
* }
* }
* return true;
*
* Note that an implementation may use more efficient techniques, but must not use any other order.
*
* @param procedure a procedure object taking as argument the current cell's value. Stops iteration if the procedure returns false, otherwise continues.
* @return false if the procedure stopped before all elements where iterated over, true otherwise.
*/
private boolean xforEach(final cern.colt.function.DoubleProcedure procedure) {
for (int row=rows; --row >= 0;) {
for (int column=columns; --column >= 0;) {
if (!procedure.apply(getQuick(row,column))) return false;
}
}
return true;
}
/**
8 neighbor stencil transformation. For efficient finite difference operations.
Applies a function to a moving 3 x 3 window.
Does nothing if rows() < 3 || columns() < 3.
B[i,j] = function.apply(
A[i-1,j-1], A[i-1,j], A[i-1,j+1],
A[i, j-1], A[i, j], A[i, j+1],
A[i+1,j-1], A[i+1,j], A[i+1,j+1]
)
x x x - - x x x - - - -
x o x - - x o x - - - -
x x x - - x x x ... - x x x
- - - - - - - - - x o x
- - - - - - - - - x x x
Make sure that cells of this and B do not overlap.
In case of overlapping views, behaviour is unspecified.
Example:
final double alpha = 0.25;
final double beta = 0.75;
// 8 neighbors
cern.colt.function.Double9Function f = new cern.colt.function.Double9Function() {
public final double apply(
double a00, double a01, double a02,
double a10, double a11, double a12,
double a20, double a21, double a22) {
return beta*a11 + alpha*(a00+a01+a02 + a10+a12 + a20+a21+a22);
}
};
A.zAssign8Neighbors(B,f);
// 4 neighbors
cern.colt.function.Double9Function g = new cern.colt.function.Double9Function() {
public final double apply(
double a00, double a01, double a02,
double a10, double a11, double a12,
double a20, double a21, double a22) {
return beta*a11 + alpha*(a01+a10+a12+a21);
}
C.zAssign8Neighbors(B,g); // fast, even though it doesn't look like it
};
@param B the matrix to hold the results.
@param function the function to be applied to the 9 cells.
@throws NullPointerException if function==null.
@throws IllegalArgumentException if rows() != B.rows() || columns() != B.columns().
*/
public void zAssign8Neighbors(DoubleMatrix2D B, cern.colt.function.Double9Function function) {
if (function==null) throw new NullPointerException("function must not be null.");
checkShape(B);
if (rows<3 || columns<3) return; // nothing to do
int r = rows-1;
int c = columns-1;
double a00, a01, a02;
double a10, a11, a12;
double a20, a21, a22;
for (int i=1; i
Ads help maintain this website.