org.qsardb.conversion.csv.CsvExporter Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2013 University of Tartu
 */
package org.qsardb.conversion.csv;

import java.io.*;
import java.util.*;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter;
import org.qsardb.conversion.table.*;

public class CsvExporter extends TableExporter {
	private final CSVPrinter output;

	public CsvExporter(OutputStream os) {
		output = new CSVPrinter(new OutputStreamWriter(os), CSVFormat.EXCEL);
	}

	@Override
	public void write() throws Exception {
		ArrayList header = new ArrayList();
		for (Iterator it=columns(); it.hasNext();) {
			header.add(it.next().getId());
		}
		output.printRecord(header);

		for (Iterator it=rows(); it.hasNext();) {
			ArrayList cellValues = new ArrayList();
			for (Cell v: it.next().getValues().values()) {
				cellValues.add(v.getText());
			}
			output.printRecord(cellValues);
		}
	}

	@Override
	public void close() throws IOException {
		output.close();
	}

}