com.github.rcaller.io.CSVFileWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of RCaller Show documentation

http://www.mhsatman.com

There is a newer version: 4.0.2

package com.github.rcaller.io;

import com.github.rcaller.datatypes.DataFrame;

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

public class CSVFileWriter extends FileWriter{

    private CSVFileWriter(String filePath) throws IOException {
        super(filePath);
    }

    public static CSVFileWriter create(String filePath) throws IOException {
        return new CSVFileWriter(filePath);
    }

    public void writeDataFrameToFile(DataFrame dataFrame) throws IOException {
        appendArray(dataFrame.getNames());
        for (int i = 0; i < dataFrame.getNumberOfRows(); i++) {
            appendArray(dataFrame.getRow(i));
        }
    }

    private void appendArray(Object[] array) throws IOException {
        for (int i = 0; i < array.length; i++) {
            this.append(array[i].toString());

            if (i == array.length - 1) {
                this.append('\n');
            } else {
                this.append(',');
            }
        }
        this.flush();
    }


}