package com.imc.workshop.core.limits;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

public class Matrix {
    private double[][] theMatrix;

    public Matrix(int aNumberOfRows, int aNumberOfColumns) {
        assert aNumberOfRows >= 0 && aNumberOfColumns >= 0;
        theMatrix = new double[aNumberOfRows][aNumberOfColumns];
    }

    public Matrix(double[][] aMatrix) {
        theMatrix = aMatrix;
    }

    public double get(int aRowIndex, int aColumnIndex) {
        return theMatrix[aRowIndex][aColumnIndex];
    }

    public void set(int aRowIndex, int aColumnIndex, double aValue) {
        theMatrix[aRowIndex][aColumnIndex] = aValue;
    }

    public int getNumberOfRows() {
        return theMatrix.length;
    }

    @Override
    public String toString() {
        return theMatrix.toString();
    }

    @Override
    public boolean equals(Object aOtherMatrix) {
        if (this == aOtherMatrix) {
            return true;
        }
        if (aOtherMatrix == null || getClass() != aOtherMatrix.getClass()) {
            return false;
        }
        final Matrix myMatrix = (Matrix) aOtherMatrix;
        return theMatrix.equals(myMatrix.theMatrix);
    }

    @Override
    public int hashCode() {
        return Objects.hash(theMatrix);
    }

    public int getNumberOfColumns() {
        return theMatrix.length == 0
                ? 0
                : theMatrix[0].length;
    }
}