package shared;

import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;

/**
 * A class for writing data sets
 * @author Andrew Guillory gtg008g@mail.gatech.edu
 * @version 1.0
 */
public class DataSetWriter {
    /**
     * The dat set
     */
    private DataSet set;
    
    /**
     * The file name
     */
    private String filename;

    /**
     * Make a new data set writer
     * @param set the data set to writer
     */
    public DataSetWriter(DataSet set, String filename) {
        this.set = set;
        this.filename = filename;
    }
    
    /**
     * Write the file out
     * @throws IOException when something goes bad
     */
    public void write() throws IOException {
        PrintWriter pw = new PrintWriter(new FileWriter(filename));
        for (int i = 0; i < set.size(); i++) {
            Instance data = set.get(i);
            while (data != null) {
                for (int j = 0; j < data.size(); j++) {
                    pw.print(data.getContinuous(j));
                    if (j + 1 < data.size() || data.getLabel() != null) {
                        pw.print(", ");
                    }
                }
                data = data.getLabel();
            }
            pw.println();
        }
        pw.close();
    }
}