package uk.ac.leeds.ccg.andyt.projects.genesis.io;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import org.jfree.chart.JFreeChart;

public class IO {

    public static BufferedImage[] _OutputJFreeCharts(
            JFreeChart[] _JFreeCharts,
            int width,
            int height,
            String directory,
            String outputImageFileNamePrefix,
            String type)
            throws IOException {
        File directoryFile = new File(directory);
        if (!directoryFile.exists()) {
            directoryFile.mkdirs();
        }
        BufferedImage[] result = new BufferedImage[_JFreeCharts.length];
        for (int i = 0; i < _JFreeCharts.length; i++) {
            //System.out.println("i " + i);
            result[i] = _JFreeCharts[i].createBufferedImage(width, height);
            javax.imageio.ImageIO.write(
                    result[i], type,
                    new File(directory, outputImageFileNamePrefix + i + "." + type));
        }
        return result;
    }

    public static BufferedImage _OutputJFreeChart(
            JFreeChart _JFreeChart,
            int width,
            int height,
            String directory,
            String outputImageFileNamePrefix,
            String type)
            throws IOException {
        File directoryFile = new File(directory);
        if (!directoryFile.exists()) {
            directoryFile.mkdirs();
        }
        BufferedImage result = _JFreeChart.createBufferedImage(width, height);
        javax.imageio.ImageIO.write(
                result, type,
                new File(directory, outputImageFileNamePrefix + "." + type));
        return result;
    }
}