Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/common/PredefinedTkhProcessor.java @ 9016:6e5ff436febe
Added datacage select and chart display for TKH series loaded from database
author | mschaefer |
---|---|
date | Mon, 23 Apr 2018 15:18:48 +0200 |
parents | |
children | 384eee4b4135 |
line wrap: on
line source
/** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde * Software engineering by * Björnsen Beratende Ingenieure GmbH * Dr. Schumacher Ingenieurbüro für Wasser und Umwelt * * 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.artifacts.sinfo.common; import java.util.HashSet; import java.util.Map; import java.util.Set; import org.apache.log4j.Logger; import org.dive4elements.artifactdatabase.state.ArtifactAndFacet; import org.dive4elements.artifacts.CallContext; import org.dive4elements.river.exports.DiagramGenerator; import org.dive4elements.river.exports.StyledSeriesBuilder; import org.dive4elements.river.jfree.StyledAreaSeriesCollection; import org.dive4elements.river.jfree.StyledXYSeries; import org.dive4elements.river.themes.ThemeDocument; /** * Processor to generate a data series for tkh data loaded from the database * * @author Matthias Schäfer * */ public class PredefinedTkhProcessor extends AbstractSInfoProcessor { private final static Logger log = Logger.getLogger(PredefinedTkhProcessor.class); public static final String FACET_PREDEFINED_TKH = "sinfo_facet_predefined_tkh"; private static final String I18N_AXIS_LABEL = "sinfo.chart.tkh.section.yaxis.label"; private static final String I18N_FACET_TKH_DESCRIPTION = "sinfo.facet.tkh.description"; private static final String SINFO_CHART_TKH_YAXIS_LABEL = "sinfo.chart.tkh.yaxis.label"; private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>(); static { HANDLED_FACET_TYPES.add(FACET_PREDEFINED_TKH); } public PredefinedTkhProcessor() { super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES); } protected String generateSeries1(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) { final CallContext context = generator.getCallContext(); final Map<String, String> metaData = bundle.getFacet().getMetaData(bundle.getArtifact(), context); final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme); series.putMetaData(metaData, bundle.getArtifact(), context); final Object raw = bundle.getData(context); if (raw == null) { return null; } if (!(raw instanceof double[][])) { log.error("Unkown datatype: " + raw.getClass().getName()); return null; } final double[][] data = (double[][]) raw; StyledSeriesBuilder.addPoints(series, data, true); // REMARK: we add " " because the description is misused as id, which must be unique. final StyledXYSeries seriesDown = new StyledXYSeries(bundle.getFacetDescription() + " ", theme); // final double[][] pointsDown = data.getTkhDownPoints(); final double[][] datadown = (double[][]) raw; for (int i = 0; i <= datadown[0].length - 1; i++) datadown[1][i] = 0; StyledSeriesBuilder.addPoints(seriesDown, datadown, true); final StyledAreaSeriesCollection area = new StyledAreaSeriesCollection(theme); area.setMode(StyledAreaSeriesCollection.FILL_MODE.BETWEEN); area.addSeries(series); area.addSeries(seriesDown); generator.addAreaSeries(area, getAxisName(), visible); // generator.addAxisSeries(series, getAxisName(), visible); return null; } @Override protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) { final CallContext context = generator.getCallContext(); final String facetName = bundle.getFacetName(); final AbstractTkhCalculationResult data = (AbstractTkhCalculationResult) bundle.getData(context); if (data == null) { // Check has been here before so we keep it for security reasons // this should never happen though. throw new IllegalStateException("Data is null for facet: " + facetName); } final StyledXYSeries seriesUp = new StyledXYSeries(bundle.getFacetDescription(), theme); final double[][] pointsUp = data.getTkhUpPoints(); StyledSeriesBuilder.addPoints(seriesUp, pointsUp, true); // REMARK: we add " " because the description is misused as id, which must be unique. final StyledXYSeries seriesDown = new StyledXYSeries(bundle.getFacetDescription() + " ", theme); final double[][] pointsDown = data.getTkhDownPoints(); StyledSeriesBuilder.addPoints(seriesDown, pointsDown, true); final StyledAreaSeriesCollection area = new StyledAreaSeriesCollection(theme); area.setMode(StyledAreaSeriesCollection.FILL_MODE.BETWEEN); area.addSeries(seriesUp); area.addSeries(seriesDown); generator.addAreaSeries(area, getAxisName(), visible); return null; } }