All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • net.sourceforge.cilib.io.CSVFileWriter Maven / Gradle / Ivy

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

import java.io.IOException;
import java.util.List;
import net.sourceforge.cilib.io.exception.CIlibIOException;

/**
 * Implements a Data Writer that writes a datatable to a local CSV file.
 */
public class CSVFileWriter extends FileWriter {

    private String delimiter;
    private boolean writeColumnNames;

    /**
     * Default constructor. Sets the delimiter used to separate fields to the
     * default (comma).
     */
    public CSVFileWriter() {
        delimiter = ",";
    }

    /**
     * Writes the datatable, writing each row as new line, with data fields separated
     * by a {@link #delimiter delimiter}.
     * @param dataTable the datatable to write.
     * @throws net.sourceforge.cilib.io.exception.CIlibIOException {@inheritDoc }
     */
    @Override
    public void write(DataTable dataTable) throws CIlibIOException {
        try {
            int size;
            if (writeColumnNames) {
                List names = dataTable.getColumnNames();
                size = names.size();
                for (int i = 0; i < size - 1; i++) {
                    String name = names.get(i);
                    IOUtils.writeStrings(outputBuffer, "\"", name, "\"", delimiter);
                }
                String name = names.get(size - 1);
                IOUtils.writeLine(outputBuffer, "\"" + name + "\"");
                outputBuffer.flip();
                outputChannel.write(outputBuffer);
                outputBuffer.clear();
            }
            int rows = dataTable.getNumRows();
            for (int i = 0; i < rows; i++) {
                List row = (List) dataTable.getRow(i);
                size = row.size();
                for (int j = 0; j < size - 1; j++) {
                    Object object = row.get(j);
                    IOUtils.writeStrings(outputBuffer, object.toString(), delimiter);
                }
                Object object = row.get(size - 1);
                IOUtils.writeLine(outputBuffer, object.toString());
            }
            outputBuffer.flip();
            outputChannel.write(outputBuffer);
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

    /**
     * Gets the delimiter used to separate data fields.
     * @return the data field delimiter.
     */
    public String getDelimiter() {
        return delimiter;
    }

    /**
     * Sets the delimiter used to separate data fields.
     * @param delimiter the data field delimiter.
     */
    public void setDelimiter(String delimiter) {
        this.delimiter = delimiter;
    }

    /**
     * Gets writeColumnNames flag: whether column names will be written along
     * with data.
     * @return writeColumnNames flag.
     */
    public boolean isWriteColumnNames() {
        return writeColumnNames;
    }

    /**
     * Sets writeColumnNames flag: whether column names will be written along
     * with data.
     * @param writeColumnNames whether to write column names.
     */
    public void setWriteColumnNames(boolean writeColumnNames) {
        this.writeColumnNames = writeColumnNames;
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy