mschaefer@9295: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
mschaefer@9295:  * Software engineering by
mschaefer@9295:  *  Björnsen Beratende Ingenieure GmbH
mschaefer@9295:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
mschaefer@9295:  *
mschaefer@9295:  * This file is Free Software under the GNU AGPL (>=v3)
mschaefer@9295:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
mschaefer@9295:  * documentation coming with Dive4Elements River for details.
mschaefer@9295:  */
mschaefer@9295: 
mschaefer@9295: package org.dive4elements.river.artifacts.uinfo.salix;
mschaefer@9295: 
mschaefer@9295: import java.util.HashSet;
mschaefer@9295: import java.util.Set;
mschaefer@9295: 
mschaefer@9295: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
mschaefer@9295: import org.dive4elements.artifactdatabase.state.Facet;
mschaefer@9295: import org.dive4elements.artifacts.CallContext;
mschaefer@9295: import org.dive4elements.river.artifacts.D4EArtifact;
mschaefer@9295: import org.dive4elements.river.artifacts.common.AbstractCalculationResult;
mschaefer@9295: import org.dive4elements.river.artifacts.uinfo.commons.UInfoResultType;
mschaefer@9295: import org.dive4elements.river.exports.DiagramGenerator;
mschaefer@9295: import org.dive4elements.river.exports.StyledSeriesBuilder;
mschaefer@9295: import org.dive4elements.river.exports.process.DefaultProcessor;
mschaefer@9295: import org.dive4elements.river.exports.process.WOutProcessor;
mschaefer@9295: import org.dive4elements.river.jfree.StyledXYSeries;
mschaefer@9295: import org.dive4elements.river.themes.ThemeDocument;
mschaefer@9295: import org.dive4elements.river.utils.RiverUtils;
mschaefer@9295: import org.jfree.data.xy.XYSeries;
mschaefer@9295: 
mschaefer@9295: /**
mschaefer@9295:  * Processor to generate the facets and data series of salix line
mschaefer@9295:  *
mschaefer@9295:  * @author Matthias Schäfer
mschaefer@9295:  *
mschaefer@9295:  */
mschaefer@9295: public final class SalixLineProcessor extends DefaultProcessor {
mschaefer@9295: 
mschaefer@9295:     private static final String FACET_SALIX_LINE = "uinfo_facet_salix";
mschaefer@9295: 
mschaefer@9295:     private static final String FACET_SALIX_MNWMW = "uinfo_facet_mnwmw";
mschaefer@9295: 
mschaefer@9295:     private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>();
mschaefer@9295: 
mschaefer@9295:     static {
mschaefer@9295:         HANDLED_FACET_TYPES.add(FACET_SALIX_LINE);
mschaefer@9295:         HANDLED_FACET_TYPES.add(FACET_SALIX_MNWMW);
mschaefer@9295:     }
mschaefer@9295: 
mschaefer@9295:     public static Facet createSalixLineFacet(final CallContext context, final String hash, final String id, final AbstractCalculationResult result,
mschaefer@9295:             final int facetIndex, final int resultIndex, final String description) {
mschaefer@9295: 
mschaefer@9295:         return new SalixLineResultFacet(FACET_SALIX_LINE, description);
mschaefer@9295:     }
mschaefer@9295: 
mschaefer@9295:     public static Facet createSalixMnwMwFacet(final CallContext context, final String hash, final String id, final AbstractCalculationResult result,
mschaefer@9295:             final int facetIndex, final int resultIndex, final String description) {
mschaefer@9295: 
mschaefer@9295:         return new SalixMnwMwResultFacet(FACET_SALIX_MNWMW, description);
mschaefer@9295:     }
mschaefer@9295: 
mschaefer@9295:     @Override
mschaefer@9295:     public final String getAxisLabel(final DiagramGenerator generator) {
mschaefer@9295: 
mschaefer@9295:         final D4EArtifact flys = (D4EArtifact) generator.getMaster();
mschaefer@9295:         final String unit = RiverUtils.getRiver(flys).getWstUnit().getName();
mschaefer@9295:         return generator.msg(WOutProcessor.I18N_AXIS_LABEL, WOutProcessor.I18N_AXIS_LABEL_DEFAULT, unit);
mschaefer@9295:     }
mschaefer@9295: 
mschaefer@9295:     @Override
mschaefer@9295:     public final boolean canHandle(final String facettype) {
mschaefer@9295:         return HANDLED_FACET_TYPES.contains(facettype);
mschaefer@9295:     }
mschaefer@9295: 
mschaefer@9295:     @Override
mschaefer@9295:     public void doOut(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
mschaefer@9295: 
mschaefer@9295:         // log.debug("Processing facet: " + bundle.getFacetName());
mschaefer@9295:         final CallContext context = generator.getContext();
mschaefer@9295:         final Object data = bundle.getData(context);
mschaefer@9295: 
mschaefer@9295:         final XYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
mschaefer@9295: 
mschaefer@9295:         if (bundle.getFacetName().equals(FACET_SALIX_LINE)) {
mschaefer@9295:             StyledSeriesBuilder.addPoints(series, ((SalixLineCalculationNoScenarioResult) data).getStationPoints(UInfoResultType.salixline), true);
mschaefer@9295:             generator.addAxisSeries(series, this.axisName, visible);
mschaefer@9295:             return;
mschaefer@9295:         }
mschaefer@9295: 
mschaefer@9295:         if (bundle.getFacetName().equals(FACET_SALIX_MNWMW)) {
mschaefer@9295:             StyledSeriesBuilder.addPoints(series, ((SalixLineCalculationNoScenarioResult) data).getStationPoints(UInfoResultType.salix_delta_mw), true);
mschaefer@9295:             generator.addAxisSeries(series, this.axisName, visible);
mschaefer@9295:             return;
mschaefer@9295:         }
mschaefer@9295:     }
mschaefer@9295: }