jdplus.sa.base.csv.CsvOutput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jdplus-sa-base-csv Show documentation
${project.parent.artifactId} - ${project.artifactId}
There is a newer version: 3.3.0
/*
 * Copyright 2013 National Bank of Belgium
 *
 * Licensed under the EUPL, Version 1.1 or – as soon they will be approved 
 * by the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 *
 * http://ec.europa.eu/idabc/eupl
 *
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and 
 * limitations under the Licence.
 */
package jdplus.sa.base.csv;

import jdplus.toolkit.base.api.information.formatters.StringFormatter;
import jdplus.toolkit.base.api.processing.Output;
import jdplus.sa.base.api.SaDocument;
import jdplus.toolkit.base.api.timeseries.TsData;
import jdplus.toolkit.base.api.util.Paths;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Optional;

/**
 *
 * @author Kristof Bayens
 */
public class CsvOutput implements Output {

    CsvOutputConfiguration config_;
    DefaultCollectionSummary summary_;
    private File folder_;

    public CsvOutput(CsvOutputConfiguration config) {
        config_ = (CsvOutputConfiguration) config.clone();
    }

    @Override
    public void process(SaDocument document) {
        List series = config_.getSeries();
        summary_.add(series.toArray(new String[series.size()]), document);
    }

    @Override
    public void start(Object context) {
        summary_ = new DefaultCollectionSummary();
        folder_ = Paths.folderFromContext(config_.getFolder(), context);
    }

    @Override
    public void end(Object context) throws Exception {

        for (String item : summary_.getItems()) {
            List series = summary_.getSeries(item);
            Optional any = series.stream().filter(s -> s != null && !s.isEmpty()).findAny();
            if (any.isPresent()) {
                String nfile = config_.getFilePrefix();
                String c = item.replaceAll("[?*.]", "_");
                nfile += "_" + StringFormatter.cleanup(c);
                nfile = Paths.changeExtension(nfile, "csv");
                write(new File(Paths.folder(folder_), nfile), summary_.getNames(), series);
            }
        }
        summary_ = null;
    }

    @Override
    public String getName() {
        return "csv";
    }

    @Override
    public boolean isAvailable() {
        return true;
    }

    private void write(File file, List names, List s) throws Exception {
        try ( FileOutputStream matrix = new FileOutputStream(file)) {
            try ( OutputStreamWriter writer = new OutputStreamWriter(matrix, StandardCharsets.ISO_8859_1)) {
                TsCollectionCsvFormatter fmt = new TsCollectionCsvFormatter();
                fmt.setFullName(config_.isFullName());
                fmt.setPresentation(config_.getPresentation());
                fmt.write(s, names, writer);
            }
        }
    }
}