view artifacts/src/main/java/org/dive4elements/river/exports/process/KMIndexProcessor.java @ 6601:5ecc6d4d73f2

Add official fixings to Waterlevel CSV Export (issue1384) This searches the collection for staticwqkms artifacts that contain official data and adds that data to the export. The data is filtered by the calculation range and sorted by the calculation direction.
author Andre Heinecke <aheinecke@intevation.de>
date Thu, 18 Jul 2013 13:16:33 +0200
parents 87b49f9f27a1
children 1b35b2ddfc28
line wrap: on
line source
/* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
 * Software engineering by Intevation GmbH
 *
 * This file is Free Software under the GNU AGPL (>=v3)
 * and comes with ABSOLUTELY NO WARRANTY! Check out the
 * documentation coming with Dive4Elements River for details.
 */

package org.dive4elements.river.exports.process;

import org.apache.log4j.Logger;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.w3c.dom.Document;

import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
import org.dive4elements.artifacts.CallContext;
import org.dive4elements.river.artifacts.model.FacetTypes;
import org.dive4elements.river.artifacts.model.fixings.AnalysisPeriod;
import org.dive4elements.river.artifacts.model.fixings.QWD;
import org.dive4elements.river.exports.XYChartGenerator;
import org.dive4elements.river.jfree.StyledXYSeries;
import org.dive4elements.river.utils.KMIndex;

public class KMIndexProcessor implements Processor {

    private static final Logger logger = Logger.getLogger(KMIndexProcessor.class);

    @Override
    public void doOut(XYChartGenerator generator, ArtifactAndFacet aandf,
            Document theme, boolean visible, int index) {
        String facettype = aandf.getFacetName();
        if (facettype.contains(FacetTypes.FIX_SECTOR_AVERAGE_LS)) {
            doSectorAverageOut(generator, aandf, theme, visible, index);
        }
        else if (facettype.equals(FacetTypes.FIX_REFERENCE_EVENTS_LS)) {
            doReferenceEventsOut(generator, aandf, theme, visible, index);
        }
        else if (facettype.equals(FacetTypes.FIX_ANALYSIS_EVENTS_LS)) {
            doAnalysisEventsOut(generator, aandf, theme, visible, index);
        }

    }

    @Override
    public boolean canHandle(String facettype) {
        if (facettype == null) {
            return false;
        }

        if (facettype.contains(FacetTypes.FIX_SECTOR_AVERAGE_LS)
                || facettype.equals(FacetTypes.FIX_REFERENCE_EVENTS_LS)
                || facettype.equals(FacetTypes.FIX_ANALYSIS_EVENTS_LS))
        {
            return true;
        }
        return false;
    }

    private void doSectorAverageOut(XYChartGenerator generator, ArtifactAndFacet aaf,
            Document doc, boolean visible, int idx) {
        logger.debug("doSectorAverageOut" + aaf.getFacet().getIndex());

        CallContext context = generator.getCallContext();
        int index = aaf.getFacet().getIndex();
        int sectorNdx = index & 3;

        @SuppressWarnings("unchecked")
        KMIndex<AnalysisPeriod> kms =
                (KMIndex<AnalysisPeriod>)aaf.getData(context);

        if(kms == null) {
            return;
        }

        XYSeries series = new StyledXYSeries(aaf.getFacetDescription(), doc);

        for (KMIndex.Entry<AnalysisPeriod> entry: kms) {
            double km = entry.getKm();
            AnalysisPeriod ap = entry.getValue();
            QWD qwd = ap.getQSectorAverages()[sectorNdx];
            if (qwd == null) {
                continue;
            }
            double deltaW = qwd.getDeltaW();
            series.add(km, deltaW);
        }

        generator.addAxisSeries(series, idx, visible);
    }

    private void doReferenceEventsOut(XYChartGenerator generator,
            ArtifactAndFacet aaf, Document doc, boolean visible, int idx) {
        logger.debug("doReferenceEventOut");

        CallContext context = generator.getCallContext();

        @SuppressWarnings("unchecked")
        KMIndex<QWD> kms =
                (KMIndex<QWD>)aaf.getData(context);

        if(kms == null) {
            return;
        }

        XYSeriesCollection col = new XYSeriesCollection();

        StyledXYSeries series = new StyledXYSeries(aaf.getFacetDescription(), false,
                doc);

        for (KMIndex.Entry<QWD> entry: kms) {
            double km = entry.getKm();
            QWD qwd = entry.getValue();

            series.add(km, qwd.getDeltaW());
        }
        col.addSeries(series);

        generator.addAxisDataset(col, idx, visible);
    }

    private void doAnalysisEventsOut(XYChartGenerator generator,
            ArtifactAndFacet aaf, Document doc, boolean visible, int idx) {
        logger.debug("doAnalysisEventsOut");

        CallContext context = generator.getCallContext();

        @SuppressWarnings("unchecked")
        KMIndex<QWD> kms =
                (KMIndex<QWD>)aaf.getData(context);

        if(kms == null) {
            return;
        }

        XYSeriesCollection col = new XYSeriesCollection();

        StyledXYSeries series = new StyledXYSeries(aaf.getFacetDescription(), false, doc);

        for (KMIndex.Entry<QWD> entry: kms) {
            double km = entry.getKm();
            QWD qwd = entry.getValue();

            series.add(km, qwd.getDeltaW());
        }
        col.addSeries(series);

        generator.addAxisDataset(col, idx, visible);
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

http://dive4elements.wald.intevation.org