aheinecke@7240: /* Copyright (C) 2013 by Bundesanstalt für Gewässerkunde
aheinecke@7240:  * Software engineering by Intevation GmbH
aheinecke@7240:  *
aheinecke@7240:  * This file is Free Software under the GNU AGPL (>=v3)
aheinecke@7240:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
aheinecke@7240:  * documentation coming with Dive4Elements River for details.
aheinecke@7240:  */
aheinecke@7240: 
aheinecke@7240: package org.dive4elements.river.exports.process;
aheinecke@7240: 
aheinecke@7240: import org.apache.log4j.Logger;
aheinecke@7240: import org.jfree.data.xy.XYSeries;
aheinecke@7240: 
aheinecke@7240: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
aheinecke@7240: import org.dive4elements.artifacts.CallContext;
aheinecke@7240: import org.dive4elements.river.exports.DiagramGenerator;
aheinecke@7240: import org.dive4elements.river.jfree.StyledXYSeries;
aheinecke@7240: import org.dive4elements.river.themes.ThemeDocument;
aheinecke@7240: import org.dive4elements.river.artifacts.model.FacetTypes;
aheinecke@7240: 
aheinecke@7240: import org.dive4elements.river.jfree.JFreeUtil;
aheinecke@7240: 
aheinecke@7240: import org.dive4elements.river.artifacts.model.sq.SQ;
aheinecke@7240: import org.dive4elements.river.artifacts.model.sq.SQFunction;
aheinecke@7240: 
aheinecke@7240: public class SQRelationProcessor extends DefaultProcessor {
aheinecke@7240: 
aheinecke@7240:     public static final String I18N_AXIS_LABEL =
aheinecke@7240:         "chart.sq_relation.yaxis.label";
aheinecke@7240:     public static final String I18N_AXIS_LABEL_DEFAULT =
aheinecke@7240:         "";
aheinecke@7240: 
aheinecke@7240:     private final static Logger logger =
aheinecke@7240:             Logger.getLogger(SQRelationProcessor.class);
aheinecke@7240: 
aheinecke@7240:     @Override
aheinecke@7240:     public void doOut(
aheinecke@7240:             DiagramGenerator generator,
aheinecke@7240:             ArtifactAndFacet bundle,
aheinecke@7240:             ThemeDocument    theme,
aheinecke@7240:             boolean          visible) {
aheinecke@7240:         CallContext context = generator.getCallContext();
aheinecke@7240:         String facetName = bundle.getFacetName();
aheinecke@7240:         XYSeries series;
aheinecke@7240:         Object data = bundle.getData(context);
aheinecke@7240:         String desc = bundle.getFacetDescription();
aheinecke@7240: 
aheinecke@7240:         if (data == null) {
aheinecke@7240:             // Check has been here before so we keep it but
aheinecke@7240:             // this should never happen.
aheinecke@7240:             logger.error("Data is null for facet: " + facetName);
aheinecke@7240:             return;
aheinecke@7240:         }
aheinecke@7240: 
aheinecke@7240:         if (FacetTypes.IS.SQ_CURVE(facetName)) {
aheinecke@7240:             SQFunction func = (SQFunction) data;
aheinecke@7240: 
aheinecke@7240:             series = JFreeUtil.sampleFunction2DPositive(
aheinecke@7240:                 func.getFunction(),
aheinecke@7240:                 theme,
aheinecke@7240:                 desc,
aheinecke@7240:                 500,
aheinecke@7240:                 Math.max(func.getMinQ(), 0.01),
aheinecke@7240:                 Math.max(func.getMaxQ(), 0.02));
aheinecke@7240: 
aheinecke@7240:         } else if (FacetTypes.IS.SQ_MEASUREMENT(facetName) ||
aheinecke@7240:                FacetTypes.IS.SQ_OUTLIER(facetName)) {
aheinecke@7240: 
aheinecke@7240:             SQ[] sqs = (SQ[]) data;
aheinecke@7240:             series = new StyledXYSeries(desc, theme);
aheinecke@7240: 
aheinecke@7240:             for (SQ sq: sqs) {
aheinecke@7240:                 double q = sq.getQ();
aheinecke@7240:                 double s = sq.getS();
aheinecke@7240:                 if (s > 0d && q > 0d) {
aheinecke@7240:                     series.add(q, s, false);
aheinecke@7240:                 }
aheinecke@7240:             }
aheinecke@7240:         } else {
aheinecke@7240:             logger.error("Could not handle: " + facetName);
aheinecke@7240:             return;
aheinecke@7240:         }
aheinecke@7240: 
aheinecke@7240:         if (logger.isDebugEnabled()) {
aheinecke@7240:             logger.debug("Series '" + desc + "' has "
aheinecke@7240:                 + series.getItemCount() + " items.");
aheinecke@7240: 
aheinecke@7240:             logger.debug("   -> min x = " + series.getMinX());
aheinecke@7240:             logger.debug("   -> max x = " + series.getMaxX());
aheinecke@7240:             logger.debug("   -> min y = " + series.getMinY());
aheinecke@7240:             logger.debug("   -> max y = " + series.getMaxY());
aheinecke@7240:         }
aheinecke@7240: 
aheinecke@7240:         generator.addAxisSeries(series, axisName, visible);
aheinecke@7240:     }
aheinecke@7240: 
aheinecke@7240:     @Override
aheinecke@7240:     public boolean canHandle(String facettype) {
aheinecke@7240:         return FacetTypes.IS.SQ_CURVE(facettype) ||
aheinecke@7240:             FacetTypes.IS.SQ_MEASUREMENT(facettype) ||
aheinecke@7240:             FacetTypes.IS.SQ_OUTLIER(facettype);
aheinecke@7240:     }
aheinecke@7240: 
aheinecke@7240:     @Override
aheinecke@7240:     public String getAxisLabel(DiagramGenerator generator) {
aheinecke@7240:         return generator.msg(
aheinecke@7240:                 I18N_AXIS_LABEL,
aheinecke@7240:                 I18N_AXIS_LABEL_DEFAULT);
aheinecke@7240:     }
aheinecke@7240: }