felix@7503: /* Copyright (C) 2013 by Bundesanstalt für Gewässerkunde
felix@7503:  * Software engineering by Intevation GmbH
felix@7503:  *
felix@7503:  * This file is Free Software under the GNU AGPL (>=v3)
felix@7503:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
felix@7503:  * documentation coming with Dive4Elements River for details.
felix@7503:  */
felix@7503: 
felix@7503: package org.dive4elements.river.exports.process;
felix@7503: 
felix@7503: import java.util.Set;
felix@7503: 
felix@7503: import org.apache.log4j.Logger;
felix@7503: import org.jfree.data.xy.XYSeries;
felix@7503: import org.dive4elements.river.artifacts.D4EArtifact;
felix@7503: 
felix@7503: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
felix@7503: import org.dive4elements.artifacts.CallContext;
felix@7503: import org.dive4elements.river.artifacts.access.SedimentLoadAccess;
felix@7503: import org.dive4elements.river.artifacts.model.FacetTypes;
felix@7503: import org.dive4elements.river.exports.DiagramGenerator;
felix@7503: import org.dive4elements.river.exports.StyledSeriesBuilder;
felix@7503: import org.dive4elements.river.jfree.StyledXYSeries;
felix@7503: import org.dive4elements.river.themes.ThemeDocument;
felix@7503: 
felix@7503: import org.dive4elements.river.artifacts.model.minfo.SedimentLoad;
felix@7503: import org.dive4elements.river.artifacts.model.minfo.SedimentLoadFraction;
felix@7503: 
felix@7503: public class SedimentLoadM3AProcessor extends DefaultProcessor {
felix@7503: 
felix@7503:     private final static Logger logger =
felix@7503:             Logger.getLogger(SedimentLoadM3AProcessor.class);
felix@7503: 
felix@7503:     public static final String I18N_YAXIS_LABEL =
felix@7503:         "chart.sedimentload.ls.yaxis.label.m3pera";
felix@7503:     public static final String I18N_YAXIS_LABEL_DEFAULT = "[m\u00b3/a]";
felix@7503: 
felix@7503:     @Override
felix@7503:     public void doOut(
felix@7503:             DiagramGenerator generator,
felix@7503:             ArtifactAndFacet bundle,
felix@7503:             ThemeDocument    theme,
felix@7503:             boolean          visible) {
felix@7503:         logger.debug("doOut " + bundle.getFacetName());
felix@7503:         CallContext context = generator.getCallContext();
felix@7503:         XYSeries series = new StyledXYSeries(bundle.getFacetDescription(),
felix@7503:                 theme);
felix@7503:         Object data = bundle.getData(context);
felix@7503:         String facetName = bundle.getFacetName();
felix@7503:         double [][] points;
felix@7503: 
felix@7503:         if (FacetTypes.IS.SEDIMENT_LOAD_M3A(facetName)) {
felix@7503:             points = (double[][]) data;
felix@7503:         } else if (FacetTypes.IS.SEDIMENT_LOAD_UNKNOWN(facetName)) {
felix@7503:             SedimentLoad load = (SedimentLoad) data;
felix@7503:             Set<Double> kms = load.getKms();
felix@7503:             points = new double[2][kms.size()];
felix@7503:             int counter = 0;
felix@7503:             for (Double km: kms) {
felix@7503:                 SedimentLoadFraction fraction = load.getFraction(km);
felix@7503:                 points[0][counter] = km;
felix@7503:                 points[1][counter] = fraction.getUnknown();
felix@7503:                 counter++;
felix@7503:             }
felix@7503:         } else {
felix@7503:             logger.error("Unknown facet name: " + facetName);
felix@7503:             return;
felix@7503:         }
felix@7503: 
felix@7503:         StyledSeriesBuilder.addPoints(series, points, true);
felix@7503: 
felix@7503:         generator.addAxisSeries(series, axisName, visible);
felix@7503:     }
felix@7503: 
felix@7503:     @Override
felix@7503:     public boolean canHandle(String facettype) {
felix@7503:         return FacetTypes.IS.SEDIMENT_LOAD_M3A(facettype) ||
felix@7503:             FacetTypes.SEDIMENT_LOAD_M3A_UNKNOWN.equals(facettype);
felix@7503:     }
felix@7503: 
felix@7503:     @Override
felix@7503:     public String getAxisLabel(DiagramGenerator generator) {
felix@7503:         return generator.msg(I18N_YAXIS_LABEL, I18N_YAXIS_LABEL_DEFAULT);
felix@7503:     }
felix@7503: }