eu.clarussecure.dataoperations.testing.DatasetParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dataoperations.splittingmodule Show documentation

CLARUS Splitting Module

The newest version!

package eu.clarussecure.dataoperations.testing;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.LinkedList;

public class DatasetParser {

    private final File f;
    private final String fieldSeparator;

    public DatasetParser(File f, String fieldSeparator) {
        this.f = f;
        this.fieldSeparator = fieldSeparator;
    }

    public String[] parseHeaders() throws IOException {
        BufferedReader in = new BufferedReader(new FileReader(this.f));
        String line = in.readLine();
        in.close();
        return line.split(this.fieldSeparator);
    }

    public String[][] parseDataset() throws IOException {
        BufferedReader in = new BufferedReader(new FileReader(this.f));
        LinkedList lines = new LinkedList();
        String line = "";
        while ((line = in.readLine()) != null) {
            lines.add(line);
        }
        in.close();
        line = lines.pollFirst();
        int recordSize = line.split(this.fieldSeparator).length;
        String[][] content = new String[lines.size()][recordSize];
        for (int i = 0; i < lines.size(); i++) {
            content[i] = lines.get(i).split(this.fieldSeparator);
        }
        return content;
    }
}