package org.apidesign.benchmark.matrixmul;

//import java.io.PrintStream;

//import java.util.Random;

public class Matrix {

    private final int rank;

    private float data[][];

    public Matrix(int r) {

        rank = r;

        data = new float[r][r];

    }

    public void setElement(int i, int j, float value) {

        data[i][j] = value;

    }

    public float getElement(int i, int j) {

        return data[i][j];

    }

    public void generateData() {

        //final Random rand = new Random();

        //final int x = 10;

        for (int i = 0; i < rank; i++) {

            for (int j = 0; j < rank; j++) {

                data[i][j] = i + j;

            }

        }

    }

    public Matrix multiply(Matrix m) {

        if (rank != m.rank) {

            throw new IllegalArgumentException("Rank doesn't match");

        }

        final float res[][] = new float[rank][rank];

        for (int i = 0; i < rank; i++) {

            for (int j = 0; j < rank; j++) {

                float ij = 0;

                for (int q = 0; q < rank; q++) {

                    ij += data[i][q] * m.data[q][j];

                }

                res[i][j] = ij;

            }

        }

        data = res;

        return this;

    }

    /*

    public void printOn(PrintStream s) {

        for (int i = 0; i < rank; i++) {

            for (int j = 0; j < rank; j++) {

                s.printf("%f ", data[i][j]);

            }

            s.println();

        }

    }

    */

}