org.solovyev.common.math.matrix.AbstractArrayMatrix Maven / Gradle / Ivy

Go to download
package org.solovyev.common.math.matrix;

import org.solovyev.common.math.graph.Graph;

import java.io.IOException;
import java.util.Arrays;

/**
 * User: serso
 * Date: 30.04.2009
 * Time: 0:15:02
 */
public abstract class AbstractArrayMatrix extends AbstractMatrix implements ArrayMatrix  {

    protected Object[] array;

	protected AbstractArrayMatrix() {
	}

	protected AbstractArrayMatrix(int m) {
        super(m);
    }

    protected AbstractArrayMatrix(int m, int n) {
		this(m, n, null);
    }

    protected AbstractArrayMatrix(int m, int n, T defaultValue) {
        super(m, n, defaultValue);
    }

    protected AbstractArrayMatrix(Graph g) {
        super(g);
    }

    protected AbstractArrayMatrix(String fName, MatrixFileFormat fileFormat) throws IOException, IllegalArgumentException {
        super(fName, fileFormat);
    }

    public void init(int m, int n, T defaultValue) {
        this.m = m;
        this.n = n;
        this.array = new Object[m * n];
        for (int i = 0; i < this.array.length; i++) {
            if (defaultValue == null) {
                this.array[i] = this.getEmptyValue();
            } else {
                this.array[i] = defaultValue;
            }
        }
    }

    public T getChecked(int i, int j) {
		return MatrixUtils.getElement(this.getArray(), i, j, this.getNumberOfColumns());
    }

    public void setChecked(int i, int j, T value) {
        MatrixUtils.setElement(this.getArray(), value, i, j, this.getNumberOfColumns());
    }

    public void transpose() {
        //todo serso: hint: use just boolean stored in Matrix (transposed) and use it proper in each method working with matrix
        Object[] array = new Object[this.getNumberOfColumns() * this.getNumberOfRows()];
        for (int i = 0; i < this.getNumberOfColumns(); i++) {
            for (int j = 0; j < this.getNumberOfRows(); j++) {
                array[i * this.getNumberOfRows() + j] = this.get(j, i);
            }
        }
        Integer tmpInt = this.m;
        this.m = this.n;
        this.n = tmpInt;
        this.array = array;
    }

    @SuppressWarnings("unchecked")
    public Matrix clone() {
        final AbstractArrayMatrix clone = (AbstractArrayMatrix) super.clone();

        //todo serso: only for simple types
        clone.array = Arrays.copyOf(this.array, this.array.length);

        return clone;
    }

	@Override
	public T[] getArray() {
		//noinspection unchecked
		return (T[])array;
	}
}