org.ejml.data.DSubmatrixD1 Maven / Gradle / Ivy

/*
 * Copyright (c) 2009-2017, Peter Abeles. All Rights Reserved.
 *
 * This file is part of Efficient Java Matrix Library (EJML).
 *
 * Licensed 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.ejml.data;

import org.ejml.ops.MatrixIO;

/**
 * 
 * Describes a rectangular submatrix inside of a {@link DMatrixD1}.
 * 
 *
 * 
 * Rows are row0 ≤ i < row1 and Columns are col0 ≤ j < col1
 * 
 * 
 * @author Peter Abeles
 */
public class DSubmatrixD1 {
    public DMatrixD1 original;

    // bounding rows and columns
    public int row0,col0;
    public int row1,col1;

    public DSubmatrixD1() {
    }

    public DSubmatrixD1(DMatrixD1 original) {
        set(original);
    }

    public DSubmatrixD1(DMatrixD1 original,
                           int row0, int row1, int col0, int col1) {
        set(original,row0,row1,col0,col1);
    }

    public void set(DMatrixD1 original,
                    int row0, int row1, int col0, int col1) {
        this.original = original;
        this.row0 = row0;
        this.col0 = col0;
        this.row1 = row1;
        this.col1 = col1;
    }

    public void set(DMatrixD1 original) {
        this.original = original;
        row1 = original.numRows;
        col1 = original.numCols;
    }

    public int getRows() {
        return row1 - row0;
    }

    public int getCols() {
        return col1 - col0;
    }

    public double get(int row, int col ) {
        return original.get(row+row0,col+col0);
    }

    public void set(int row, int col, double value) {
        original.set(row+row0,col+col0,value);
    }

    public DMatrixRMaj extract() {
        DMatrixRMaj ret = new DMatrixRMaj(row1-row0,col1-col0);

        for( int i = 0; i < ret.numRows; i++ ) {
            for( int j = 0; j < ret.numCols; j++ ) {
                ret.set(i,j,get(i,j));
            }
        }

        return ret;
    }

    public void print() {
        MatrixIO.print(System.out,original,"%6.3f",row0,row1,col0,col1);
    }
}