package de.intevation.flys.exports;

import java.awt.Color;

import org.apache.log4j.Logger;

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

import org.w3c.dom.Document;

import de.intevation.artifacts.Artifact;

import de.intevation.artifactdatabase.state.Facet;

import de.intevation.flys.artifacts.FLYSArtifact;
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);



    public DischargeLongitudinalSectionGenerator() {
        super();
    }


    @Override
    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);
    }


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

        if (facet == null) {
            return;
        }

        String name = facet.getName();

        if (name == null) {
            return;
        }

        FLYSArtifact flys = (FLYSArtifact) artifact;
        Facet        f    = flys.getNativeFacet(facet);

        if (name.equals(DISCHARGE_LONGITUDINAL_W)) {
            doWOut((WQKms) f.getData(artifact, context));
        }
        else if (name.equals(DISCHARGE_LONGITUDINAL_Q)) {
            doQOut((WQKms) f.getData(artifact, context));
        }
        else if (name.equals(DISCHARGE_LONGITUDINAL_C)) {
            doCorrectedWOut((WQCKms) f.getData(artifact, context));
        }
        else {
            logger.warn("Unknown facet name: " + name);
        }
    }


    /**
     * 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");

        int size = wqckms.size();

        if (size > 0) {
            XYSeries series = new XYSeries(getSeriesNameForCorrected(wqckms, "W"));
            for (int i = 0; i < size; i++) {
                series.add(wqckms.getKms(i), wqckms.getC(i));
            }

            addFirstAxisSeries(series);
        }

        if (wqckms.guessWaterIncreasing()) {
            setInverted(true);
        }
    }


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

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

        String prefix = 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 :