package de.intevation.flys.exports;

import java.awt.Color;

import org.apache.log4j.Logger;

import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;

import org.w3c.dom.Document;

import de.intevation.artifacts.Artifact;

import de.intevation.flys.artifacts.WINFOArtifact;
import de.intevation.flys.artifacts.model.WQCKms;
import de.intevation.flys.artifacts.model.WQKms;


/**
 * An OutGenerator that generates discharge longitudinal section curves.
 *
 * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
 */
public class DischargeLongitudinalSectionGenerator
extends      LongitudinalSectionGenerator
{
    private static Logger logger =
        Logger.getLogger(DischargeLongitudinalSectionGenerator.class);


    /** The facet for Ws.*/
    public static final String FACET_W = "discharge_longitudinal_section.w";

    /** The facet for Qs.*/
    public static final String FACET_Q = "discharge_longitudinal_section.q";


    /** The storage for the corrected W series to be drawn in this chart.*/
    protected XYSeriesCollection cw;


    public DischargeLongitudinalSectionGenerator() {
        super();

        this.cw = new XYSeriesCollection();
    }


    public void addDatasets(JFreeChart chart) {
        super.addDatasets(chart);

        XYPlot plot = (XYPlot) chart.getPlot();

        plot.setDataset(2, cw);
    }


    protected void adjustPlot(XYPlot plot) {
        super.adjustPlot(plot);

        // TODO REMOVE THIS CODE, IF WE HAVE INTRODUCED THEMES!
        XYLineAndShapeRenderer rw = (XYLineAndShapeRenderer)
            plot.getRendererForDataset(w);

        XYLineAndShapeRenderer rcw = null;
        try {
            rcw = (XYLineAndShapeRenderer) rw.clone();
        }
        catch (Exception e) {
            logger.error(e, e);
        }

        int cwNum = cw.getSeriesCount();

        for (int i = 0; i < cwNum; i++) {
            rcw.setSeriesPaint(i, Color.RED);
        }

        plot.setRenderer(2, rcw);
    }


    protected void adjustAxes(XYPlot plot) {
        super.adjustAxes(plot);

        plot.mapDatasetToRangeAxis(2, 1);
    }



    @Override
    public void doOut(Artifact artifact, String facet, Document attr) {
        logger.debug("DischargeLongitudinalSectionGenerator.doOut");

        if (facet != null && facet.equals(FACET_W)) {
            doWOut(getDischargeLongitudinalSectionData(artifact));
        }
        else if (facet != null && facet.equals(FACET_Q)) {
            doQOut(getDischargeLongitudinalSectionData(artifact));
        }
        else {
            logger.warn("Unknown facet name: " + facet);
            return;
        }
    }


    protected WQKms[] getDischargeLongitudinalSectionData(Artifact artifact) {
        WINFOArtifact winfoArtifact = (WINFOArtifact) artifact;
        return winfoArtifact.getDischargeLongitudinalSectionData();
    }


    /**
     * This method adds new Series for the wqkms objects. If there are instances
     * of WQCKms in this array, there is a further curve that displays the
     * corrected W values.
     *
     * @param wqkms An array of WQKms objects to be displayed.
     */
    protected void doWOut(WQKms[] wqkms) {
        logger.debug("LongitudinalSectionGenerator.doWOut");

        int idx = 0;
        for (WQKms tmp: wqkms) {
            if (tmp instanceof WQCKms) {
                doCorrectedWOut((WQCKms) tmp);
            }

            doWOut(tmp);
        }
    }


    /**
     * Adds a new series for the W curve.
     *
     * @param wqkms The object that contains the W values.
     */
    protected void doWOut(WQKms wqkms) {
        logger.debug("DischargeLongitudinalSectionGenerator.doWOut");

        XYSeries series = new XYSeries(getSeriesName(wqkms, "W"));

        double[] target = new double[4];
        int      size   = wqkms.size();

        for (int i = 0; i < size; i++) {
            target = wqkms.get(i, target);

            series.add(target[2], target[0]);
        }

        w.addSeries(series);
    }


    /**
     * Adds a new series for the corrected W curve.
     *
     * @param wqckms The object that contains the corrected W values.
     */
    protected void doCorrectedWOut(WQCKms wqckms) {
        logger.debug("DischargeLongitudinalSectionGenerator.doCorrectedWOut");

        XYSeries series = new XYSeries(getSeriesNameForCorrected(wqckms, "W"));

        double[] target = new double[4];
        int      size   = wqckms.size();

        for (int i = 0; i < size; i++) {
            target = wqckms.get(i, target);

            series.add(target[2], target[3]);
        }

        if (series.getItemCount() > 0) {
            cw.addSeries(series);
        }
    }


    protected String getSeriesNameForCorrected(WQKms wqkms, String mode) {
        String name = wqkms.getName();

        name = name.replace(
            "Benutzerdefiniert",
            "Benutzerdefiniert [korrigiert]");

        String prefix = name != null && name.indexOf(mode) >= 0 ? null : mode;

        return prefix != null && prefix.length() > 0
            ? prefix + "(" + name +")"
            : name;
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :