/*
    * Licensed to the Apache Software Foundation (ASF) under one or more
    * contributor license agreements.  See the NOTICE file distributed with
    * this work for additional information regarding copyright ownership.
    * The ASF licenses this file to You under the Apache License, Version 2.0
    * (the "License"); you may not use this file except in compliance with
    * the License.  You may obtain a copy of the License at
    *
    *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package org.apache.commons.math4.legacy.linear;
  
  import java.io.Serializable;
  
  import org.apache.commons.math4.legacy.core.Field;
  import org.apache.commons.math4.legacy.core.FieldElement;
  import org.apache.commons.math4.legacy.exception.DimensionMismatchException;
  import org.apache.commons.math4.legacy.exception.MathIllegalStateException;
  import org.apache.commons.math4.legacy.exception.NoDataException;
  import org.apache.commons.math4.legacy.exception.NotStrictlyPositiveException;
  import org.apache.commons.math4.legacy.exception.NullArgumentException;
  import org.apache.commons.math4.legacy.exception.NumberIsTooSmallException;
  import org.apache.commons.math4.legacy.exception.OutOfRangeException;
  import org.apache.commons.math4.legacy.exception.util.LocalizedFormats;
  import org.apache.commons.math4.legacy.core.MathArrays;
  
  /**
   * Implementation of {@code FieldMatrix<T>} using a {@link FieldElement}[][] array to store entries.
   * <p>
   * As specified in the {@link FieldMatrix} interface, matrix element indexing
   * is 0-based -- e.g., <code>getEntry(0, 0)</code>
   * returns the element in the first row, first column of the matrix.
   * </p>
   *
   * @param <T> the type of the field elements
   */
  public class Array2DRowFieldMatrix<T extends FieldElement<T>>
      extends AbstractFieldMatrix<T>
      implements Serializable {
      /** Serializable version identifier. */
      private static final long serialVersionUID = 7260756672015356458L;
      /** Entries of the matrix. */
      private T[][] data;
  
      /**
       * Creates a matrix with no data.
       * @param field field to which the elements belong
       */
      public Array2DRowFieldMatrix(final Field<T> field) {
          super(field);
      }
  
      /**
       * Create a new {@code FieldMatrix<T>} with the supplied row and column dimensions.
       *
       * @param field Field to which the elements belong.
       * @param rowDimension Number of rows in the new matrix.
       * @param columnDimension Number of columns in the new matrix.
       * @throws NotStrictlyPositiveException if row or column dimension is not positive.
       */
      public Array2DRowFieldMatrix(final Field<T> field, final int rowDimension,
                                   final int columnDimension)
          throws NotStrictlyPositiveException {
          super(field, rowDimension, columnDimension);
          data = MathArrays.buildArray(field, rowDimension, columnDimension);
      }
  
      /**
       * Create a new {@code FieldMatrix<T>} using the input array as the underlying
       * data array.
       * <p>The input array is copied, not referenced. This constructor has
       * the same effect as calling {@link #Array2DRowFieldMatrix(FieldElement[][], boolean)}
       * with the second argument set to {@code true}.</p>
       *
       * @param d Data for the new matrix.
       * @throws DimensionMismatchException if {@code d} is not rectangular.
       * @throws NullArgumentException if {@code d} is {@code null}.
       * @throws NoDataException if there are not at least one row and one column.
       * @see #Array2DRowFieldMatrix(FieldElement[][], boolean)
       */
      public Array2DRowFieldMatrix(final T[][] d)
          throws DimensionMismatchException, NullArgumentException,
          NoDataException {
          this(extractField(d), d);
      }
  
      /**
       * Create a new {@code FieldMatrix<T>} using the input array as the underlying
       * data array.
       * <p>The input array is copied, not referenced. This constructor has
       * the same effect as calling {@link #Array2DRowFieldMatrix(FieldElement[][], boolean)}
       * with the second argument set to {@code true}.</p>
      *
      * @param field Field to which the elements belong.
      * @param d Data for the new matrix.
      * @throws DimensionMismatchException if {@code d} is not rectangular.
      * @throws NullArgumentException if {@code d} is {@code null}.
      * @throws NoDataException if there are not at least one row and one column.
      * @see #Array2DRowFieldMatrix(FieldElement[][], boolean)
      */
     public Array2DRowFieldMatrix(final Field<T> field, final T[][] d)
         throws DimensionMismatchException, NullArgumentException,
         NoDataException {
         super(field);
         copyIn(d);
     }
 
     /**
      * Create a new {@code FieldMatrix<T>} using the input array as the underlying
      * data array.
      * <p>If an array is built specially in order to be embedded in a
      * {@code FieldMatrix<T>} and not used directly, the {@code copyArray} may be
      * set to {@code false}. This will prevent the copying and improve
      * performance as no new array will be built and no data will be copied.</p>
      *
      * @param d Data for the new matrix.
      * @param copyArray Whether to copy or reference the input array.
      * @throws DimensionMismatchException if {@code d} is not rectangular.
      * @throws NoDataException if there are not at least one row and one column.
      * @throws NullArgumentException if {@code d} is {@code null}.
      * @see #Array2DRowFieldMatrix(FieldElement[][])
      */
     public Array2DRowFieldMatrix(final T[][] d, final boolean copyArray)
         throws DimensionMismatchException, NoDataException,
         NullArgumentException {
         this(extractField(d), d, copyArray);
     }
 
     /**
      * Create a new {@code FieldMatrix<T>} using the input array as the underlying
      * data array.
      * <p>If an array is built specially in order to be embedded in a
      * {@code FieldMatrix<T>} and not used directly, the {@code copyArray} may be
      * set to {@code false}. This will prevent the copying and improve
      * performance as no new array will be built and no data will be copied.</p>
      *
      * @param field Field to which the elements belong.
      * @param d Data for the new matrix.
      * @param copyArray Whether to copy or reference the input array.
      * @throws DimensionMismatchException if {@code d} is not rectangular.
      * @throws NoDataException if there are not at least one row and one column.
      * @throws NullArgumentException if {@code d} is {@code null}.
      * @see #Array2DRowFieldMatrix(FieldElement[][])
      */
     public Array2DRowFieldMatrix(final Field<T> field, final T[][] d, final boolean copyArray)
         throws DimensionMismatchException, NoDataException, NullArgumentException {
         super(field);
         if (copyArray) {
             copyIn(d);
         } else {
             NullArgumentException.check(d);
             final int nRows = d.length;
             if (nRows == 0) {
                 throw new NoDataException(LocalizedFormats.AT_LEAST_ONE_ROW);
             }
             final int nCols = d[0].length;
             if (nCols == 0) {
                 throw new NoDataException(LocalizedFormats.AT_LEAST_ONE_COLUMN);
             }
             for (int r = 1; r < nRows; r++) {
                 if (d[r].length != nCols) {
                     throw new DimensionMismatchException(nCols, d[r].length);
                 }
             }
             data = d;
         }
     }
 
     /**
      * Create a new (column) {@code FieldMatrix<T>} using {@code v} as the
      * data for the unique column of the created matrix.
      * The input array is copied.
      *
      * @param v Column vector holding data for new matrix.
      * @throws NoDataException if v is empty
      */
     public Array2DRowFieldMatrix(final T[] v) throws NoDataException {
         this(extractField(v), v);
     }
 
     /**
      * Create a new (column) {@code FieldMatrix<T>} using {@code v} as the
      * data for the unique column of the created matrix.
      * The input array is copied.
      *
      * @param field Field to which the elements belong.
      * @param v Column vector holding data for new matrix.
      */
     public Array2DRowFieldMatrix(final Field<T> field, final T[] v) {
         super(field);
         final int nRows = v.length;
         data = MathArrays.buildArray(getField(), nRows, 1);
         for (int row = 0; row < nRows; row++) {
             data[row][0] = v[row];
         }
     }
 
     /** {@inheritDoc} */
     @Override
     public FieldMatrix<T> createMatrix(final int rowDimension,
                                        final int columnDimension)
         throws NotStrictlyPositiveException {
         return new Array2DRowFieldMatrix<>(getField(), rowDimension, columnDimension);
     }
 
     /** {@inheritDoc} */
     @Override
     public FieldMatrix<T> copy() {
         return new Array2DRowFieldMatrix<>(getField(), copyOut(), false);
     }
 
     /**
      * Add {@code m} to this matrix.
      *
      * @param m Matrix to be added.
      * @return {@code this} + m.
      * @throws MatrixDimensionMismatchException if {@code m} is not the same
      * size as this matrix.
      */
     public Array2DRowFieldMatrix<T> add(final Array2DRowFieldMatrix<T> m)
         throws MatrixDimensionMismatchException {
         // safety check
         checkAdd(m);
 
         final int rowCount    = getRowDimension();
         final int columnCount = getColumnDimension();
         final T[][] outData = MathArrays.buildArray(getField(), rowCount, columnCount);
         for (int row = 0; row < rowCount; row++) {
             final T[] dataRow    = data[row];
             final T[] mRow       = m.data[row];
             final T[] outDataRow = outData[row];
             for (int col = 0; col < columnCount; col++) {
                 outDataRow[col] = dataRow[col].add(mRow[col]);
             }
         }
 
         return new Array2DRowFieldMatrix<>(getField(), outData, false);
     }
 
     /**
      * Subtract {@code m} from this matrix.
      *
      * @param m Matrix to be subtracted.
      * @return {@code this} + m.
      * @throws MatrixDimensionMismatchException if {@code m} is not the same
      * size as this matrix.
      */
     public Array2DRowFieldMatrix<T> subtract(final Array2DRowFieldMatrix<T> m)
         throws MatrixDimensionMismatchException {
         // safety check
         checkAdd(m);
 
         final int rowCount    = getRowDimension();
         final int columnCount = getColumnDimension();
         final T[][] outData = MathArrays.buildArray(getField(), rowCount, columnCount);
         for (int row = 0; row < rowCount; row++) {
             final T[] dataRow    = data[row];
             final T[] mRow       = m.data[row];
             final T[] outDataRow = outData[row];
             for (int col = 0; col < columnCount; col++) {
                 outDataRow[col] = dataRow[col].subtract(mRow[col]);
             }
         }
 
         return new Array2DRowFieldMatrix<>(getField(), outData, false);
     }
 
     /**
      * Postmultiplying this matrix by {@code m}.
      *
      * @param m Matrix to postmultiply by.
      * @return {@code this} * m.
      * @throws DimensionMismatchException if the number of columns of this
      * matrix is not equal to the number of rows of {@code m}.
      */
     public Array2DRowFieldMatrix<T> multiply(final Array2DRowFieldMatrix<T> m)
         throws DimensionMismatchException {
         // safety check
         checkMultiply(m);
 
         final int nRows = this.getRowDimension();
         final int nCols = m.getColumnDimension();
         final int nSum = this.getColumnDimension();
         final T[][] outData = MathArrays.buildArray(getField(), nRows, nCols);
         for (int row = 0; row < nRows; row++) {
             final T[] dataRow    = data[row];
             final T[] outDataRow = outData[row];
             for (int col = 0; col < nCols; col++) {
                 T sum = getField().getZero();
                 for (int i = 0; i < nSum; i++) {
                     sum = sum.add(dataRow[i].multiply(m.data[i][col]));
                 }
                 outDataRow[col] = sum;
             }
         }
 
         return new Array2DRowFieldMatrix<>(getField(), outData, false);
     }
 
     /** {@inheritDoc} */
     @Override
     public T[][] getData() {
         return copyOut();
     }
 
     /**
      * Get a reference to the underlying data array.
      * This methods returns internal data, <strong>not</strong> fresh copy of it.
      *
      * @return the 2-dimensional array of entries.
      */
     public T[][] getDataRef() {
         return data;
     }
 
     /** {@inheritDoc} */
     @Override
     public void setSubMatrix(final T[][] subMatrix, final int row,
                              final int column)
         throws OutOfRangeException, NullArgumentException, NoDataException,
         DimensionMismatchException {
         if (data == null) {
             if (row > 0) {
                 throw new MathIllegalStateException(LocalizedFormats.FIRST_ROWS_NOT_INITIALIZED_YET, row);
             }
             if (column > 0) {
                 throw new MathIllegalStateException(LocalizedFormats.FIRST_COLUMNS_NOT_INITIALIZED_YET, column);
             }
             final int nRows = subMatrix.length;
             if (nRows == 0) {
                 throw new NoDataException(LocalizedFormats.AT_LEAST_ONE_ROW);
             }
 
             final int nCols = subMatrix[0].length;
             if (nCols == 0) {
                 throw new NoDataException(LocalizedFormats.AT_LEAST_ONE_COLUMN);
             }
             data = MathArrays.buildArray(getField(), subMatrix.length, nCols);
             for (int i = 0; i < data.length; ++i) {
                 if (subMatrix[i].length != nCols) {
                     throw new DimensionMismatchException(nCols, subMatrix[i].length);
                 }
                 System.arraycopy(subMatrix[i], 0, data[i + row], column, nCols);
             }
         } else {
             super.setSubMatrix(subMatrix, row, column);
         }
     }
 
     /** {@inheritDoc} */
     @Override
     public T getEntry(final int row, final int column)
         throws OutOfRangeException {
         checkRowIndex(row);
         checkColumnIndex(column);
 
         return data[row][column];
     }
 
     /** {@inheritDoc} */
     @Override
     public void setEntry(final int row, final int column, final T value)
         throws OutOfRangeException {
         checkRowIndex(row);
         checkColumnIndex(column);
 
         data[row][column] = value;
     }
 
     /** {@inheritDoc} */
     @Override
     public void addToEntry(final int row, final int column, final T increment)
         throws OutOfRangeException {
         checkRowIndex(row);
         checkColumnIndex(column);
 
         data[row][column] = data[row][column].add(increment);
     }
 
     /** {@inheritDoc} */
     @Override
     public void multiplyEntry(final int row, final int column, final T factor)
         throws OutOfRangeException {
         checkRowIndex(row);
         checkColumnIndex(column);
 
         data[row][column] = data[row][column].multiply(factor);
     }
 
     /** {@inheritDoc} */
     @Override
     public int getRowDimension() {
         return (data == null) ? 0 : data.length;
     }
 
     /** {@inheritDoc} */
     @Override
     public int getColumnDimension() {
         return (data == null || data[0] == null) ? 0 : data[0].length;
     }
 
     /** {@inheritDoc} */
     @Override
     public T[] operate(final T[] v) throws DimensionMismatchException {
         final int nRows = this.getRowDimension();
         final int nCols = this.getColumnDimension();
         if (v.length != nCols) {
             throw new DimensionMismatchException(v.length, nCols);
         }
         final T[] out = MathArrays.buildArray(getField(), nRows);
         for (int row = 0; row < nRows; row++) {
             final T[] dataRow = data[row];
             T sum = getField().getZero();
             for (int i = 0; i < nCols; i++) {
                 sum = sum.add(dataRow[i].multiply(v[i]));
             }
             out[row] = sum;
         }
         return out;
     }
 
     /** {@inheritDoc} */
     @Override
     public T[] preMultiply(final T[] v) throws DimensionMismatchException {
         final int nRows = getRowDimension();
         final int nCols = getColumnDimension();
         if (v.length != nRows) {
             throw new DimensionMismatchException(v.length, nRows);
         }
 
         final T[] out = MathArrays.buildArray(getField(), nCols);
         for (int col = 0; col < nCols; ++col) {
             T sum = getField().getZero();
             for (int i = 0; i < nRows; ++i) {
                 sum = sum.add(data[i][col].multiply(v[i]));
             }
             out[col] = sum;
         }
 
         return out;
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInRowOrder(final FieldMatrixChangingVisitor<T> visitor) {
         final int rows    = getRowDimension();
         final int columns = getColumnDimension();
         visitor.start(rows, columns, 0, rows - 1, 0, columns - 1);
         for (int i = 0; i < rows; ++i) {
             final T[] rowI = data[i];
             for (int j = 0; j < columns; ++j) {
                 rowI[j] = visitor.visit(i, j, rowI[j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInRowOrder(final FieldMatrixPreservingVisitor<T> visitor) {
         final int rows    = getRowDimension();
         final int columns = getColumnDimension();
         visitor.start(rows, columns, 0, rows - 1, 0, columns - 1);
         for (int i = 0; i < rows; ++i) {
             final T[] rowI = data[i];
             for (int j = 0; j < columns; ++j) {
                 visitor.visit(i, j, rowI[j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInRowOrder(final FieldMatrixChangingVisitor<T> visitor,
                             final int startRow, final int endRow,
                             final int startColumn, final int endColumn)
         throws OutOfRangeException, NumberIsTooSmallException {
         checkSubMatrixIndex(startRow, endRow, startColumn, endColumn);
         visitor.start(getRowDimension(), getColumnDimension(),
                       startRow, endRow, startColumn, endColumn);
         for (int i = startRow; i <= endRow; ++i) {
             final T[] rowI = data[i];
             for (int j = startColumn; j <= endColumn; ++j) {
                 rowI[j] = visitor.visit(i, j, rowI[j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInRowOrder(final FieldMatrixPreservingVisitor<T> visitor,
                             final int startRow, final int endRow,
                             final int startColumn, final int endColumn)
         throws OutOfRangeException, NumberIsTooSmallException {
         checkSubMatrixIndex(startRow, endRow, startColumn, endColumn);
         visitor.start(getRowDimension(), getColumnDimension(),
                       startRow, endRow, startColumn, endColumn);
         for (int i = startRow; i <= endRow; ++i) {
             final T[] rowI = data[i];
             for (int j = startColumn; j <= endColumn; ++j) {
                 visitor.visit(i, j, rowI[j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInColumnOrder(final FieldMatrixChangingVisitor<T> visitor) {
         final int rows    = getRowDimension();
         final int columns = getColumnDimension();
         visitor.start(rows, columns, 0, rows - 1, 0, columns - 1);
         for (int j = 0; j < columns; ++j) {
             for (int i = 0; i < rows; ++i) {
                 final T[] rowI = data[i];
                 rowI[j] = visitor.visit(i, j, rowI[j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInColumnOrder(final FieldMatrixPreservingVisitor<T> visitor) {
         final int rows    = getRowDimension();
         final int columns = getColumnDimension();
         visitor.start(rows, columns, 0, rows - 1, 0, columns - 1);
         for (int j = 0; j < columns; ++j) {
             for (int i = 0; i < rows; ++i) {
                 visitor.visit(i, j, data[i][j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInColumnOrder(final FieldMatrixChangingVisitor<T> visitor,
                                final int startRow, final int endRow,
                                final int startColumn, final int endColumn)
         throws OutOfRangeException, NumberIsTooSmallException {
     checkSubMatrixIndex(startRow, endRow, startColumn, endColumn);
         visitor.start(getRowDimension(), getColumnDimension(),
                       startRow, endRow, startColumn, endColumn);
         for (int j = startColumn; j <= endColumn; ++j) {
             for (int i = startRow; i <= endRow; ++i) {
                 final T[] rowI = data[i];
                 rowI[j] = visitor.visit(i, j, rowI[j]);
             }
         }
         return visitor.end();
     }
 
     /** {@inheritDoc} */
     @Override
     public T walkInColumnOrder(final FieldMatrixPreservingVisitor<T> visitor,
                                final int startRow, final int endRow,
                                final int startColumn, final int endColumn)
         throws OutOfRangeException, NumberIsTooSmallException {
         checkSubMatrixIndex(startRow, endRow, startColumn, endColumn);
         visitor.start(getRowDimension(), getColumnDimension(),
                       startRow, endRow, startColumn, endColumn);
         for (int j = startColumn; j <= endColumn; ++j) {
             for (int i = startRow; i <= endRow; ++i) {
                 visitor.visit(i, j, data[i][j]);
             }
         }
         return visitor.end();
     }
 
     /**
      * Get a fresh copy of the underlying data array.
      *
      * @return a copy of the underlying data array.
      */
     private T[][] copyOut() {
         final int nRows = this.getRowDimension();
         final T[][] out = MathArrays.buildArray(getField(), nRows, getColumnDimension());
         // can't copy 2-d array in one shot, otherwise get row references
         for (int i = 0; i < nRows; i++) {
             System.arraycopy(data[i], 0, out[i], 0, data[i].length);
         }
         return out;
     }
 
     /**
      * Replace data with a fresh copy of the input array.
      *
      * @param in Data to copy.
      * @throws NoDataException if the input array is empty.
      * @throws DimensionMismatchException if the input array is not rectangular.
      * @throws NullArgumentException if the input array is {@code null}.
      */
     private void copyIn(final T[][] in)
         throws NullArgumentException, NoDataException,
         DimensionMismatchException {
         setSubMatrix(in, 0, 0);
     }
 }