package Lecture21;

public class DiagonalMatrix {   // No amplía la matriz
    private double[] data;  // Sustituye la representación de datos
    
    // // constructor especial que sólo requiere 1 argumento
    public DiagonalMatrix(int nr) {
        data = new double[nr];
    }
    
//    public DiagonalMatrix() {
//        data= new double[1];
//    }
    
    public int getNumRows() {
        return data.length;
    }
    
    public int getNumCols() {
        return data.length;
    }
    
    public double getElement(int i, int j) {
        if(i == j)
            return data[i];
        else
            return 0.0;
    }
    
    public void setElement(int i, int j, double val) {
        if(i == j)
            data[i] = val;
    }
    
    public Matrix multMatrices(Matrix b)    {
        Matrix result = null;
        int nrows = data.length;
        int p = data.length;
//        int i,j,k;
        if (p == b.getNumRows()) {
            result = new Matrix(nrows, b.getNumCols());
            for(int i=0; i<nrows; i++)
                for(int j=0; j<result.getNumCols(); j++)
                    result.setElement(i,j, b.getElement(i,j)*this.data[i] );
        }
        return result;
    }
    
    public void print() {
        for(int i=0; i< data.length; i++) {
            for(int j=0; j< data.length; j++)
                if (i==j)
                    System.out.print(data[i] + " ");
                else
                    System.out.print("0.0 ");
            System.out.println();
        }
        System.out.println();
    }
    
    // Las versiones especiales de addMatrix() y otros métodos podrían ir aquí
}