mschaefer@9125: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
mschaefer@9125:  * Software engineering by
mschaefer@9125:  *  Björnsen Beratende Ingenieure GmbH
mschaefer@9125:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
mschaefer@9125:  *
mschaefer@9125:  * This file is Free Software under the GNU AGPL (>=v3)
mschaefer@9125:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
mschaefer@9125:  * documentation coming with Dive4Elements River for details.
mschaefer@9125:  */
mschaefer@9125: 
mschaefer@9125: package org.dive4elements.river.artifacts.sinfo.common;
mschaefer@9125: 
mschaefer@9125: import java.util.HashSet;
mschaefer@9125: import java.util.Map;
mschaefer@9125: import java.util.Set;
mschaefer@9125: 
mschaefer@9125: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
mschaefer@9125: import org.dive4elements.artifactdatabase.state.Facet;
mschaefer@9125: import org.dive4elements.artifacts.Artifact;
mschaefer@9125: import org.dive4elements.artifacts.CallContext;
mschaefer@9125: import org.dive4elements.artifacts.CallMeta;
mschaefer@9125: import org.dive4elements.river.artifacts.resources.Resources;
mschaefer@9125: import org.dive4elements.river.artifacts.sinfo.collision.CollisionFacet;
mschaefer@9125: import org.dive4elements.river.artifacts.sinfo.collision.CollisionQueryCalculationResult;
mschaefer@9125: import org.dive4elements.river.exports.DiagramGenerator;
mschaefer@9125: import org.dive4elements.river.exports.StyledSeriesBuilder;
mschaefer@9125: import org.dive4elements.river.jfree.StyledXYSeries;
mschaefer@9125: import org.dive4elements.river.themes.ThemeDocument;
mschaefer@9125: 
mschaefer@9125: /**
mschaefer@9125:  * Processor to generate a data series for bed collision count data loaded from the database
mschaefer@9125:  *
mschaefer@9125:  * @author Matthias Schäfer
mschaefer@9125:  *
mschaefer@9125:  */
mschaefer@9125: public class CollisionCountProcessor extends AbstractSInfoProcessor {
mschaefer@9125: 
mschaefer@9125:     // private final static Logger log = Logger.getLogger(CollisionCountProcessor.class);
mschaefer@9125: 
mschaefer@9125:     public static final String FACET_COLLISION_COUNT = "sinfo_facet_collision_count";
mschaefer@9125: 
mschaefer@9125:     private static final String I18N_AXIS_LABEL = "chart.longitudinal.section.yaxis.label";
mschaefer@9125: 
mschaefer@9125:     private static final String I18N_SERIES_NAME_PATTERN = "collision.count.title";
mschaefer@9125: 
mschaefer@9125:     private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>();
mschaefer@9125: 
mschaefer@9125:     static {
mschaefer@9125:         HANDLED_FACET_TYPES.add(FACET_COLLISION_COUNT);
mschaefer@9125:     }
mschaefer@9125: 
mschaefer@9125:     public CollisionCountProcessor() {
mschaefer@9125:         super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES);
mschaefer@9125:     }
mschaefer@9125: 
mschaefer@9125:     @Override
mschaefer@9125:     protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
mschaefer@9125: 
mschaefer@9125:         final CallContext context = generator.getContext();
mschaefer@9125:         final Map<String, String> metaData = bundle.getFacet().getMetaData();
mschaefer@9125: 
mschaefer@9125:         final Artifact artifact = bundle.getArtifact();
mschaefer@9125: 
mschaefer@9125:         final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
mschaefer@9125:         series.putMetaData(metaData, artifact, context);
mschaefer@9125: 
mschaefer@9125:         final String facetName = bundle.getFacetName();
mschaefer@9125:         final CollisionQueryCalculationResult data = (CollisionQueryCalculationResult) bundle.getData(context);
mschaefer@9125:         if (data == null) {
mschaefer@9125:             // Check has been here before so we keep it for security reasons
mschaefer@9125:             // this should never happen though.
mschaefer@9125:             throw new IllegalStateException("Data is null for facet: " + facetName);
mschaefer@9125:         }
mschaefer@9125: 
mschaefer@9125:         final double[][] points = data.getStationPoints(SInfoResultType.collisionCount);
mschaefer@9125: 
mschaefer@9125:         StyledSeriesBuilder.addPoints(series, points, true);
mschaefer@9125:         generator.addAxisSeries(series, getAxisName(), visible);
mschaefer@9125: 
mschaefer@9125:         return metaData.get("Y");
mschaefer@9125:     }
mschaefer@9125: 
mschaefer@9125:     public static Facet createFacet(final CallMeta callMeta, final String seriesName) {
mschaefer@9125:         return new CollisionFacet(FACET_COLLISION_COUNT,
mschaefer@9125:                 Resources.getMsg(callMeta, I18N_SERIES_NAME_PATTERN, I18N_SERIES_NAME_PATTERN, seriesName), I18N_AXIS_LABEL);
mschaefer@9125:     }
mschaefer@9125: }