package Lecture21;

public class Matrix {
    private double[][] data;
    // constructor para Matrix
    public Matrix(int m, int n) {
        data = new double[m][n];
    }
    
//    public Matrix() {
//        data = new double[1][1];
//    }
    
    // Método para definir la identidad de la matriz
    public void setIdentity() {
        int i,j;
        int nrows = data.length;
        int ncols = data[0].length;
        for(i=0; i<nrows; i++)
            for(j=0; j<ncols; j++)
                if(i == j)
                    data[i][j]= 1.0;
                else
                    data[i][j]= 0.0;
    }
    // El método devuelve el número de filas de la matriz
    public int getNumRows() {
        return data.length;
    }
    // El método devuelve el número de columnas de la matriz
    public int getNumCols() {
        return data[0].length;
    }
    // método para obtener elemento[i,] de la matriz
    public double getElement(int i, int j) {
        return data[i][j];
    }
    
    public void setElement(int i, int j, double val) {
        data[i][j] = val;
    }
    
    public Matrix addMatrices(Matrix b) {
        Matrix result = null;
        int nrows = data.length;
        int ncols = data[0].length;
        if (nrows == b.data.length && ncols == b.data[0].length) {
            result = new Matrix(nrows, ncols);
            for(int i=0; i<nrows; i++)
                for(int j=0; j<ncols; j++)
                    result.data[i][j]= this.data[i][j] + b.data[i][j];
        }
        return result;
    }
    
    public Matrix multMatrices(Matrix b) {
        Matrix result = null;
        int nrows = data.length;
        int p = data[0].length;
//        int i,j,k;
        if(p == b.data.length) {
            result = new Matrix(nrows, b.data[0].length);
            for(int i=0; i<nrows; i++)
                for(int j=0; j<result.data[0].length; j++) {
                    double t = 0.0;
                    for(int k=0; k<p; k++) {
                        t += data[i][k] * b.data[k][j];
                    }
                    result.data[i][j]= t;
                }
        }
        return result;
    }
    
    public void print() {
            for(int i=0; i< data.length; i++) {
                for(int j=0; j< data[0].length; j++)
                    System.out.print(data[i][j] + " ");
                System.out.println();
            }
        System.out.println();
    }
}