/*
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.tobject.impl;

import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;

import cern.colt.function.tobject.ObjectFunction;
import cern.colt.function.tobject.ObjectObjectFunction;
import cern.colt.function.tobject.ObjectProcedure;
import cern.colt.list.tint.IntArrayList;
import cern.colt.list.tobject.ObjectArrayList;
import cern.colt.matrix.tobject.ObjectMatrix1D;
import cern.colt.matrix.tobject.ObjectMatrix2D;
import edu.emory.utils.ConcurrencyUtils;

/**
 * Dense 2-d matrix holding Object elements. First see the package summary and javadoc tree view to get the broad picture.
 * 

* Implementation: *

* Internally holds one single contigous one-dimensional array, addressed in * column major. Note that this implementation is not synchronized. *

* Time complexity: *

* O(1) (i.e. constant time) for the basic operations get, * getQuick, set, setQuick and size, *

* Cells are internally addressed in column-major. Applications demanding utmost * speed can exploit this fact. Setting/getting values in a loop * column-by-column is quicker than row-by-row. Thus * *

 * for (int column = 0; column < columns; column++) {
 *     for (int row = 0; row < rows; row++) {
 *         matrix.setQuick(row, column, someValue);
 *     }
 * }
 * 
 * 
* * is quicker than * *
 * for (int row = 0; row < rows; row++) {
 *     for (int column = 0; column < columns; column++) {
 *         matrix.setQuick(row, column, someValue);
 *     }
 * }
 * 
 * 
* * @author Piotr Wendykier (piotr.wendykier@gmail.com) * */ public class DenseColumnObjectMatrix2D extends ObjectMatrix2D { static final long serialVersionUID = 1020177651L; protected Object[] 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 DenseColumnObjectMatrix2D(Object[][] values) { this(values.length, values.length == 0 ? 0 : values[0].length); assign(values); } /** * Constructs a matrix with a given number of rows and columns. 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 DenseColumnObjectMatrix2D(int rows, int columns) { setUp(rows, columns, 0, 0, 1, rows); this.elements = new Object[rows * columns]; } /** * Constructs a matrix 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). * @param isView * if true then a matrix view is constructed * @throws IllegalArgumentException * if * rows<0 || columns<0 || (double)columns*rows > Integer.MAX_VALUE * or flip's are illegal. */ public DenseColumnObjectMatrix2D(int rows, int columns, Object[] elements, int rowZero, int columnZero, int rowStride, int columnStride, boolean isView) { setUp(rows, columns, rowZero, columnZero, rowStride, columnStride); this.elements = elements; this.isNoView = !isView; } public Object aggregate(final ObjectObjectFunction aggr, final ObjectFunction f) { if (size() == 0) throw new IllegalArgumentException("size() = 0"); final int zero = (int) index(0, 0); Object a = 0; int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Callable() { public Object call() throws Exception { Object a = f.apply(elements[zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride]); int d = 1; for (int c = firstColumn; --c >= lastColumn;) { int cidx = zero + c * columnStride; for (int r = rows - d; --r >= 0;) { a = aggr.apply(a, f.apply(elements[r * rowStride + cidx])); } d = 0; } return a; } }); } a = ConcurrencyUtils.waitForCompletion(futures, aggr); } else { a = f.apply(elements[zero + (rows - 1) * rowStride + (columns - 1) * columnStride]); int d = 1; for (int c = columns; --c >= 0;) { int cidx = zero + c * columnStride; for (int r = rows - d; --r >= 0;) { a = aggr.apply(a, f.apply(elements[r * rowStride + cidx])); } d = 0; } } return a; } public Object aggregate(final ObjectObjectFunction aggr, final ObjectFunction f, final ObjectProcedure cond) { if (size() == 0) throw new IllegalArgumentException("size() = 0"); final int zero = (int) index(0, 0); Object a = 0; int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Callable() { public Object call() throws Exception { Object elem = elements[zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride]; Object a = 0; if (cond.apply(elem) == true) { a = f.apply(elem); } int d = 1; for (int c = firstColumn; --c >= lastColumn;) { int cidx = zero + c * columnStride; for (int r = rows - d; --r >= 0;) { elem = elements[r * rowStride + cidx]; if (cond.apply(elem) == true) { a = aggr.apply(a, f.apply(elem)); } } d = 0; } return a; } }); } a = ConcurrencyUtils.waitForCompletion(futures, aggr); } else { Object elem = elements[zero + (rows - 1) * rowStride + (columns - 1) * columnStride]; if (cond.apply(elem) == true) { a = f.apply(elem); } int d = 1; for (int c = columns; --c >= 0;) { int cidx = zero + c * columnStride; for (int r = rows - d; --r >= 0;) { elem = elements[r * rowStride + cidx]; if (cond.apply(elem) == true) { a = aggr.apply(a, f.apply(elem)); } } d = 0; } } return a; } public Object aggregate(final ObjectObjectFunction aggr, final ObjectFunction f, final IntArrayList rowList, final IntArrayList columnList) { if (size() == 0) throw new IllegalArgumentException("size() = 0"); final int zero = (int) index(0, 0); final int size = rowList.size(); final int[] rowElements = rowList.elements(); final int[] columnElements = columnList.elements(); Object a = 0; int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = size / nthreads; for (int j = 0; j < nthreads; j++) { final int firstIdx = size - j * k; final int lastIdx = (j == (nthreads - 1)) ? 0 : firstIdx - k; futures[j] = ConcurrencyUtils.submit(new Callable() { public Object call() throws Exception { Object a = f.apply(elements[zero + rowElements[firstIdx - 1] * rowStride + columnElements[firstIdx - 1] * columnStride]); for (int i = firstIdx - 1; --i >= lastIdx;) { a = aggr.apply(a, f.apply(elements[zero + rowElements[i] * rowStride + columnElements[i] * columnStride])); } return a; } }); } a = ConcurrencyUtils.waitForCompletion(futures, aggr); } else { a = f.apply(elements[zero + rowElements[size - 1] * rowStride + columnElements[size - 1] * columnStride]); for (int i = size - 1; --i >= 0;) { a = aggr.apply(a, f .apply(elements[zero + rowElements[i] * rowStride + columnElements[i] * columnStride])); } } return a; } public Object aggregate(final ObjectMatrix2D other, final ObjectObjectFunction aggr, final ObjectObjectFunction f) { if (!(other instanceof DenseColumnObjectMatrix2D)) { return super.aggregate(other, aggr, f); } checkShape(other); if (size() == 0) throw new IllegalArgumentException("size() = 0"); final int zero = (int) index(0, 0); final int zeroOther = (int) other.index(0, 0); final int rowStrideOther = other.rowStride(); final int columnStrideOther = other.columnStride(); final Object[] otherElements = (Object[]) other.elements(); Object a = 0; int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Callable() { public Object call() throws Exception { Object a = f.apply(elements[zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride], otherElements[zeroOther + (rows - 1) * rowStrideOther + (firstColumn - 1) * columnStrideOther]); int d = 1; for (int c = firstColumn; --c >= lastColumn;) { int cidx = zero + c * columnStride; int cidxOther = zeroOther + c * columnStrideOther; for (int r = rows - d; --r >= 0;) { a = aggr.apply(a, f.apply(elements[r * rowStride + cidx], otherElements[r * rowStrideOther + cidxOther])); } d = 0; } return a; } }); } a = ConcurrencyUtils.waitForCompletion(futures, aggr); } else { int d = 1; a = f.apply(elements[zero + (rows - 1) * rowStride + (columns - 1) * columnStride], otherElements[zeroOther + (rows - 1) * rowStrideOther + (columns - 1) * columnStrideOther]); for (int c = columns; --c >= 0;) { int cidx = zero + c * columnStride; int cidxOther = zeroOther + c * columnStrideOther; for (int r = rows - d; --r >= 0;) { a = aggr.apply(a, f.apply(elements[r * rowStride + cidx], otherElements[r * rowStrideOther + cidxOther])); } d = 0; } } return a; } public ObjectMatrix2D assign(final ObjectFunction function) { final int zero = (int) index(0, 0); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { elements[i] = function.apply(elements[i]); i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { elements[i] = function.apply(elements[i]); i -= rowStride; } idx -= columnStride; } } return this; } public ObjectMatrix2D assign(final ObjectProcedure cond, final ObjectFunction function) { final int zero = (int) index(0, 0); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { Object elem; int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { elem = elements[i]; if (cond.apply(elem) == true) { elements[i] = function.apply(elem); } i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { Object elem; int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { elem = elements[i]; if (cond.apply(elem) == true) { elements[i] = function.apply(elem); } i -= rowStride; } idx -= columnStride; } } return this; } public ObjectMatrix2D assign(final ObjectProcedure cond, final Object value) { final int zero = (int) index(0, 0); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { Object elem; int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { elem = elements[i]; if (cond.apply(elem) == true) { elements[i] = value; } i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { Object elem; int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { elem = elements[i]; if (cond.apply(elem) == true) { elements[i] = value; } i -= rowStride; } idx -= columnStride; } } return this; } public ObjectMatrix2D assign(final Object value) { final int zero = (int) index(0, 0); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { elements[i] = value; i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { elements[i] = value; i -= rowStride; } idx -= columnStride; } } return this; } public ObjectMatrix2D assign(final Object[] values) { if (values.length != size()) throw new IllegalArgumentException("Must have same length: length=" + values.length + " rows()*columns()=" + rows() * columns()); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if (this.isNoView) { System.arraycopy(values, 0, this.elements, 0, values.length); } else { final int zero = (int) index(0, 0); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; int idxOther = (rows - 1) + (firstColumn - 1) * rows; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { elements[i] = values[idxOther--]; i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; int idxOther = values.length - 1; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { elements[i] = values[idxOther--]; i -= rowStride; } idx -= columnStride; } } } return this; } public ObjectMatrix2D assign(final Object[][] values) { if (values.length != rows) throw new IllegalArgumentException("Must have same number of rows: rows=" + values.length + "columns()=" + rows()); int nthreads = ConcurrencyUtils.getNumberOfThreads(); final int zero = (int) index(0, 0); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = rows / nthreads; for (int j = 0; j < nthreads; j++) { final int firstRow = rows - j * k; final int lastRow = (j == (nthreads - 1)) ? 0 : firstRow - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (firstRow - 1) * rowStride + (columns - 1) * columnStride; for (int r = firstRow; --r >= lastRow;) { Object[] currentRow = values[r]; if (currentRow.length != columns) throw new IllegalArgumentException( "Must have same number of columns in every row: column=" + currentRow.length + "columns()=" + columns()); for (int i = idx, c = columns; --c >= 0;) { elements[i] = currentRow[c]; i -= columnStride; } idx -= rowStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int r = rows; --r >= 0;) { Object[] currentRow = values[r]; if (currentRow.length != columns) throw new IllegalArgumentException("Must have same number of columns in every row: column=" + currentRow.length + "columns()=" + columns()); for (int i = idx, c = columns; --c >= 0;) { elements[i] = currentRow[c]; i -= columnStride; } idx -= rowStride; } } return this; } public ObjectMatrix2D assign(final ObjectMatrix2D source) { // overriden for performance only if (!(source instanceof DenseColumnObjectMatrix2D)) { super.assign(source); return this; } DenseColumnObjectMatrix2D other = (DenseColumnObjectMatrix2D) source; if (other == this) return this; // nothing to do checkShape(other); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if (this.isNoView && other.isNoView) { // quickest System.arraycopy(other.elements, 0, elements, 0, elements.length); return this; } if (haveSharedCells(other)) { ObjectMatrix2D c = other.copy(); if (!(c instanceof DenseColumnObjectMatrix2D)) { // should not happen super.assign(other); return this; } other = (DenseColumnObjectMatrix2D) c; } final int zeroOther = (int) other.index(0, 0); final int zero = (int) index(0, 0); final int columnStrideOther = other.columnStride; final int rowStrideOther = other.rowStride; final Object[] otherElements = other.elements; if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; int idxOther = zeroOther + (rows - 1) * rowStrideOther + (firstColumn - 1) * columnStrideOther; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, j = idxOther, r = rows; --r >= 0;) { elements[i] = otherElements[j]; i -= rowStride; j -= rowStrideOther; } idx -= columnStride; idxOther -= columnStrideOther; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; int idxOther = zeroOther + (rows - 1) * rowStrideOther + (columns - 1) * columnStrideOther; for (int c = columns; --c >= 0;) { for (int i = idx, j = idxOther, r = rows; --r >= 0;) { elements[i] = otherElements[j]; i -= rowStride; j -= rowStrideOther; } idx -= columnStride; idxOther -= columnStrideOther; } } return this; } public ObjectMatrix2D assign(final ObjectMatrix2D y, final ObjectObjectFunction function) { if (!(y instanceof DenseColumnObjectMatrix2D)) { super.assign(y, function); return this; } DenseColumnObjectMatrix2D other = (DenseColumnObjectMatrix2D) y; checkShape(y); final Object[] otherElements = other.elements; final int zeroOther = (int) other.index(0, 0); final int zero = (int) index(0, 0); final int columnStrideOther = other.columnStride; final int rowStrideOther = other.rowStride; int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; int idxOther = zeroOther + (rows - 1) * rowStrideOther + (firstColumn - 1) * columnStrideOther; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, j = idxOther, r = rows; --r >= 0;) { elements[i] = function.apply(elements[i], otherElements[j]); i -= rowStride; j -= rowStrideOther; } idx -= columnStride; idxOther -= columnStrideOther; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; int idxOther = zeroOther + (rows - 1) * rowStrideOther + (columns - 1) * columnStrideOther; for (int c = columns; --c >= 0;) { for (int i = idx, j = idxOther, r = rows; --r >= 0;) { elements[i] = function.apply(elements[i], otherElements[j]); i -= rowStride; j -= rowStrideOther; } idx -= columnStride; idxOther -= columnStrideOther; } } return this; } public ObjectMatrix2D assign(final ObjectMatrix2D y, final ObjectObjectFunction function, IntArrayList rowList, IntArrayList columnList) { checkShape(y); if (!(y instanceof DenseColumnObjectMatrix2D)) { super.assign(y, function); return this; } DenseColumnObjectMatrix2D other = (DenseColumnObjectMatrix2D) y; final int size = rowList.size(); final int[] rowElements = rowList.elements(); final int[] columnElements = columnList.elements(); final Object[] otherElements = other.elements(); final int zeroOther = (int) other.index(0, 0); final int zero = (int) index(0, 0); final int columnStrideOther = other.columnStride(); final int rowStrideOther = other.rowStride(); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = size / nthreads; for (int j = 0; j < nthreads; j++) { final int firstIdx = size - j * k; final int lastIdx = (j == (nthreads - 1)) ? 0 : firstIdx - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx; int idxOther; for (int i = firstIdx; --i >= lastIdx;) { idx = zero + rowElements[i] * rowStride + columnElements[i] * columnStride; idxOther = zeroOther + rowElements[i] * rowStrideOther + columnElements[i] * columnStrideOther; elements[idx] = function.apply(elements[idx], otherElements[idxOther]); } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx; int idxOther; for (int i = size; --i >= 0;) { idx = zero + rowElements[i] * rowStride + columnElements[i] * columnStride; idxOther = zeroOther + rowElements[i] * rowStrideOther + columnElements[i] * columnStrideOther; elements[idx] = function.apply(elements[idx], otherElements[idxOther]); } } return this; } public int cardinality() { int cardinality = 0; int nthreads = ConcurrencyUtils.getNumberOfThreads(); final int zero = (int) index(0, 0); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; Integer[] results = new Integer[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Callable() { public Integer call() throws Exception { int cardinality = 0; int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { if (elements[i] != null) cardinality++; i -= rowStride; } idx -= columnStride; } return cardinality; } }); } try { for (int j = 0; j < nthreads; j++) { results[j] = (Integer) futures[j].get(); } cardinality = results[0]; for (int j = 1; j < nthreads; j++) { cardinality += results[j]; } } catch (ExecutionException ex) { ex.printStackTrace(); } catch (InterruptedException e) { e.printStackTrace(); } } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { if (elements[i] != null) cardinality++; i -= rowStride; } idx -= columnStride; } } return cardinality; } public Object[] elements() { return elements; } public ObjectMatrix2D forEachNonZero(final cern.colt.function.tobject.IntIntObjectFunction function) { final int zero = (int) index(0, 0); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { Object value = elements[i]; if (value != null) { elements[i] = function.apply(r, c, value); } i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { Object value = elements[i]; if (value != null) { elements[i] = function.apply(r, c, value); } i -= rowStride; } idx -= columnStride; } } return this; } /** * Returns a new matrix that has the same elements as this matrix, but they * are addressed internally in row major. This method creates a new object * (not a view), so changes in the returned matrix are NOT reflected in this * matrix. * * @return this matrix with elements addressed internally in row major */ public DenseObjectMatrix2D getRowMajor() { DenseObjectMatrix2D R = new DenseObjectMatrix2D(rows, columns); final int zeroR = (int) R.index(0, 0); final int rowStrideR = R.rowStride(); final int columnStrideR = R.columnStride(); final Object[] elementsR = R.elements(); final int zero = (int) index(0, 0); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; int idxR = zeroR + (rows - 1) * rowStrideR + (firstColumn - 1) * columnStrideR; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, j = idxR, r = rows; --r >= 0;) { elementsR[j] = elements[i]; i -= rowStride; j -= rowStrideR; } idx -= columnStride; idxR -= columnStrideR; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; int idxR = zeroR + (rows - 1) * rowStrideR + (columns - 1) * columnStrideR; for (int c = columns; --c >= 0;) { for (int i = idx, j = idxR, r = rows; --r >= 0;) { elementsR[j] = elements[i]; i -= rowStride; j -= rowStrideR; } idx -= columnStride; idxR -= columnStrideR; } } return R; } public void getNonZeros(final IntArrayList rowList, final IntArrayList columnList, final ObjectArrayList valueList) { rowList.clear(); columnList.clear(); valueList.clear(); int idx = (int) index(0, 0); for (int c = 0; c < columns; c++) { for (int i = idx, r = 0; r < rows; r++) { Object value = elements[i]; if (value != null) { rowList.add(r); columnList.add(c); valueList.add(value); } i += rowStride; } idx += columnStride; } } public Object getQuick(int row, int column) { return elements[rowZero + row * rowStride + columnZero + column * columnStride]; } public long index(int row, int column) { return rowZero + row * rowStride + columnZero + column * columnStride; } public ObjectMatrix2D like(int rows, int columns) { return new DenseColumnObjectMatrix2D(rows, columns); } public ObjectMatrix1D like1D(int size) { return new DenseObjectMatrix1D(size); } public void setQuick(int row, int column, Object value) { elements[rowZero + row * rowStride + columnZero + column * columnStride] = value; } public Object[][] toArray() { final Object[][] values = new Object[rows][columns]; int nthreads = ConcurrencyUtils.getNumberOfThreads(); final int zero = (int) index(0, 0); if ((nthreads > 1) && (size() >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { values[r][c] = elements[i]; i -= rowStride; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { values[r][c] = elements[i]; i -= rowStride; } idx -= columnStride; } } return values; } public ObjectMatrix1D vectorize() { final int size = (int) size(); ObjectMatrix1D v = new DenseObjectMatrix1D(size); if (isNoView == true) { System.arraycopy(elements, 0, v.elements(), 0, size); } else { final int zero = (int) index(0, 0); final int zeroOther = (int) v.index(0); final int strideOther = v.stride(); final Object[] elementsOther = (Object[]) v.elements(); int nthreads = ConcurrencyUtils.getNumberOfThreads(); if ((nthreads > 1) && (size >= ConcurrencyUtils.getThreadsBeginN_2D())) { nthreads = Math.min(nthreads, columns); Future[] futures = new Future[nthreads]; int k = columns / nthreads; for (int j = 0; j < nthreads; j++) { final int firstColumn = columns - j * k; final int lastColumn = (j == (nthreads - 1)) ? 0 : firstColumn - k; final int firstIdxOther = size - j * k * rows; futures[j] = ConcurrencyUtils.submit(new Runnable() { public void run() { int idx = zero + (rows - 1) * rowStride + (firstColumn - 1) * columnStride; int idxOther = zeroOther + (firstIdxOther - 1) * strideOther; for (int c = firstColumn; --c >= lastColumn;) { for (int i = idx, r = rows; --r >= 0;) { elementsOther[idxOther] = elements[i]; i -= rowStride; idxOther -= strideOther; } idx -= columnStride; } } }); } ConcurrencyUtils.waitForCompletion(futures); } else { int idx = zero + (rows - 1) * rowStride + (columns - 1) * columnStride; int idxOther = zeroOther + size - 1; for (int c = columns; --c >= 0;) { for (int i = idx, r = rows; --r >= 0;) { elementsOther[idxOther] = elements[i]; i -= rowStride; idxOther--; } idx -= columnStride; } } } return v; } protected boolean haveSharedCellsRaw(ObjectMatrix2D other) { if (other instanceof SelectedDenseColumnObjectMatrix2D) { SelectedDenseColumnObjectMatrix2D otherMatrix = (SelectedDenseColumnObjectMatrix2D) other; return this.elements == otherMatrix.elements; } else if (other instanceof DenseColumnObjectMatrix2D) { DenseColumnObjectMatrix2D otherMatrix = (DenseColumnObjectMatrix2D) other; return this.elements == otherMatrix.elements; } return false; } protected ObjectMatrix1D like1D(int size, int zero, int stride) { return new DenseObjectMatrix1D(size, this.elements, zero, stride, true); } protected ObjectMatrix2D viewSelectionLike(int[] rowOffsets, int[] columnOffsets) { return new SelectedDenseColumnObjectMatrix2D(this.elements, rowOffsets, columnOffsets, 0); } }

Ads help maintain this website.