net.sourceforge.cilib.problem.dataset.MatrixDataSetBuilder Maven / Gradle / Ivy

Go to download
/**           __  __
 *    _____ _/ /_/ /_    Computational Intelligence Library (CIlib)
 *   / ___/ / / / __ \   (c) CIRG @ UP
 *  / /__/ / / / /_/ /   http://cilib.net
 *  \___/_/_/_/_.___/
 */
package net.sourceforge.cilib.problem.dataset;

import java.io.EOFException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
import java.util.ArrayList;
import java.util.List;
import net.sourceforge.cilib.type.types.container.Matrix;

/**
 * TODO: This needs to implement the reading of a matrix as needed by MappingProblem.
 */
public class MatrixDataSetBuilder extends DataSetBuilder {
    private static final long serialVersionUID = 1141280214032774956L;

    private Matrix matrix;
    private int numvectors = 0;
    private int m = -1;
    private int d = -1;

    public MatrixDataSetBuilder() {
    }

    public MatrixDataSetBuilder(MatrixDataSetBuilder rhs) {
        throw new UnsupportedOperationException("'copy constructor' not implemented for '" + this.getClass().getName() + "'");
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public MatrixDataSetBuilder getClone() {
        return new MatrixDataSetBuilder(this);
    }

    @Override
    public void initialise() {
        Matrix.Builder matrixBuilder = null;

        try {
            InputStream is = this.getDataSet(0).getInputStream();

            StreamTokenizer tok = new StreamTokenizer(new InputStreamReader(is));

            if (tok.nextToken() != StreamTokenizer.TT_NUMBER)
                throw new IllegalStateException("Expected an integer number as the first token in the dataset");

            numvectors = (int) tok.nval;

            if (numvectors <= 0)
                throw new IllegalStateException("Must have a positive number of vectors in input file");

            if (tok.nextToken() != StreamTokenizer.TT_NUMBER)
                throw new IllegalStateException("Expected an integer number as the second token in the dataset");

            m = (int) tok.nval;

            if (m <= 0)
                throw new IllegalStateException("Need to have a positive number as the input dimensions");

            matrixBuilder = Matrix.builder().dimensions(numvectors, m);

            if (tok.nextToken() != StreamTokenizer.TT_NUMBER)
                throw new IllegalStateException("Expected an integer number as the third token in the dataset");

            d = (int) tok.nval;

            if (d <= 0)
                throw new IllegalStateException("Need to have a positive number as the input dimensions");

            if (!(d <= m))
                throw new IllegalStateException("Output dimension must be less than input dimension");

            for (int i = 0; i < numvectors; i++) {
                List rowVector = new ArrayList();
                for (int j = 0; j < m; j++) {
                    int tok_ret = tok.nextToken();
                    while (tok_ret != StreamTokenizer.TT_NUMBER) {
                        switch (tok_ret) {
                            case StreamTokenizer.TT_EOF:
                                throw new EOFException();
                            case StreamTokenizer.TT_WORD:
                                throw new IllegalStateException("Only numerical input expected (line " + tok.lineno() + ")");
                            default:
                                continue;
                        }
                    }

                    rowVector.add(tok.nval);
                }
                matrixBuilder.addRow(rowVector);
            }
        }
        catch (Exception e) {
            throw new RuntimeException(e);
        }

        matrix = matrixBuilder.build();
    }

    /**
     * Get the constructed {@link Matrix}.
     * @return The current {@code Matrix} instance.
     */
    public Matrix getMatrix() {
        return this.matrix;
    }

    /**
     * @return The value of d.
     */
    public int getD() {
        return d;
    }

    /**
     * @return The current value of m.
     */
    public int getM() {
        return m;
    }
}