gernotbelger@9499: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9499:  * Software engineering by
gernotbelger@9499:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@9499:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@9499:  *
gernotbelger@9499:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@9499:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@9499:  * documentation coming with Dive4Elements River for details.
gernotbelger@9499:  */
gernotbelger@9499: package org.dive4elements.river.artifacts.uinfo.vegetationzones;
gernotbelger@9499: 
gernotbelger@9499: import java.awt.Color;
gernotbelger@9499: import java.util.HashSet;
gernotbelger@9499: import java.util.List;
gernotbelger@9499: import java.util.Set;
gernotbelger@9499: 
gernotbelger@9499: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
gernotbelger@9499: import org.dive4elements.artifactdatabase.state.Facet;
gernotbelger@9499: import org.dive4elements.artifacts.Artifact;
gernotbelger@9499: import org.dive4elements.artifacts.CallContext;
gernotbelger@9499: import org.dive4elements.artifacts.DataProvider;
gernotbelger@9506: import org.dive4elements.river.artifacts.D4EArtifact;
gernotbelger@9499: import org.dive4elements.river.artifacts.common.AbstractProcessor;
gernotbelger@9499: import org.dive4elements.river.artifacts.model.CrossSectionFacetUtils;
gernotbelger@9499: import org.dive4elements.river.artifacts.model.river.MainWstValuesCalculator;
gernotbelger@9499: import org.dive4elements.river.exports.CrossSectionGenerator;
gernotbelger@9499: import org.dive4elements.river.exports.DiagramGenerator;
gernotbelger@9499: import org.dive4elements.river.jfree.StripedAreaDataset;
gernotbelger@9499: import org.dive4elements.river.jfree.StripedAreaDataset.Stripe;
gernotbelger@9499: import org.dive4elements.river.model.FastCrossSectionLine;
gernotbelger@9499: import org.dive4elements.river.model.River;
gernotbelger@9499: import org.dive4elements.river.themes.ThemeDocument;
gernotbelger@9499: 
gernotbelger@9499: /**
gernotbelger@9499:  * @author Domenico Nardi Tironi
gernotbelger@9499:  *
gernotbelger@9499:  */
gernotbelger@9499: public class VegetationZonesCrossSectionProcessor extends AbstractProcessor {
gernotbelger@9499: 
gernotbelger@9499:     private static final String MAIN_VALUE_MQ = "mq";
gernotbelger@9499: 
gernotbelger@9499:     public static final String FACET_VEGETATION_ZONES_CROSS_SECTION = "uinfo_facet_vegetation_zones_cross_section";
gernotbelger@9499: 
gernotbelger@9499: 
gernotbelger@9499:     private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>();
gernotbelger@9499: 
gernotbelger@9499:     static {
gernotbelger@9499:         HANDLED_FACET_TYPES.add(FACET_VEGETATION_ZONES_CROSS_SECTION);
gernotbelger@9499:     }
gernotbelger@9499: 
gernotbelger@9506:     public static Facet createVegetationZonesCrossSectionFacet(final String description) {
gernotbelger@9506: 
gernotbelger@9506:         return new VegetationZonesCrossSectionFacet(FACET_VEGETATION_ZONES_CROSS_SECTION, description);
gernotbelger@9499:     }
gernotbelger@9499: 
gernotbelger@9499:     public static void generateSeries(final CrossSectionGenerator generator, final ArtifactAndFacet bundle, final CallContext context,
gernotbelger@9499:             final ThemeDocument theme, final boolean visible) {
gernotbelger@9499: 
gernotbelger@9499:         final DataProvider provider = CrossSectionFacetUtils.getDataProvider(context);
gernotbelger@9499:         final FastCrossSectionLine crossSection = CrossSectionFacetUtils.getCrossSection(provider, context);
gernotbelger@9499:         if (crossSection == null)
gernotbelger@9499:             return;
gernotbelger@9499:         final double currentStation = crossSection.getKm();
gernotbelger@9499: 
gernotbelger@9499:         if (bundle.getFacetName().equals(FACET_VEGETATION_ZONES_CROSS_SECTION)) {
gernotbelger@9499: 
gernotbelger@9499:             final StripedAreaDataset dataset = new StripedAreaDataset(theme);
gernotbelger@9499: 
gernotbelger@9499:             final Artifact artifact = bundle.getArtifact();
gernotbelger@9506:             final VegetationzonesAccess vAccess = new VegetationzonesAccess((D4EArtifact) artifact);
gernotbelger@9499:             final River river = vAccess.getRiver();
gernotbelger@9499:             final List<VegetationZoneServerClientXChange> zones = VegetationZoneServerClientXChange.parse(vAccess.getVegZones());
gernotbelger@9499: 
gernotbelger@9499:             for (final VegetationZoneServerClientXChange zone : zones) {
gernotbelger@9499: 
gernotbelger@9499:                 final double lower = uefdToHeight(context, river, currentStation, zone.getLowerFromTo());
gernotbelger@9499:                 final double upper = uefdToHeight(context, river, currentStation, zone.getUpperFromTo());
gernotbelger@9499: 
gernotbelger@9499:                 final Color color = Color.decode(zone.getHexColor());
gernotbelger@9501:                 final String label = String.format("%s (%dd-%dd)", zone.getZoneName(), zone.getLowerFromTo(), zone.getUpperFromTo());
gernotbelger@9501:                 dataset.addStripe(new Stripe(label, color, lower, upper));
gernotbelger@9499:             }
gernotbelger@9499: 
gernotbelger@9499:             generator.addAxisDataset(dataset, 0, visible);
gernotbelger@9499:             return;
gernotbelger@9499:         }
gernotbelger@9499: 
gernotbelger@9499:         throw new UnsupportedOperationException();
gernotbelger@9499:     }
gernotbelger@9499: 
gernotbelger@9499:     private static double uefdToHeight(final CallContext context, final River river, final double station, final int uefd) {
gernotbelger@9499: 
gernotbelger@9499:         final MainWstValuesCalculator mainWstValues = MainWstValuesCalculator.forRiver(context, river, null, MAIN_VALUE_MQ);
gernotbelger@9499: 
gernotbelger@9499:         final double mw = mainWstValues.interpolateW(station, MAIN_VALUE_MQ);
gernotbelger@9499: 
gernotbelger@9499:         // Üfd = -70,559 ∗ ln((DGM - MW) + 0,5) + 80,711
gernotbelger@9499:         final double f1 = -70.559;
mschaefer@9509:         final double f2 = 88.711;
gernotbelger@9499: 
gernotbelger@9499:         final double dgm = Math.exp((uefd - f2) / f1) + mw - 0.5;
gernotbelger@9499:         return dgm;
gernotbelger@9499:     }
gernotbelger@9499: 
gernotbelger@9499:     public VegetationZonesCrossSectionProcessor() {
gernotbelger@9499:         super(CrossSectionGenerator.I18N_YAXIS_LABEL, HANDLED_FACET_TYPES);
gernotbelger@9499: 
gernotbelger@9499:         throw new UnsupportedOperationException();
gernotbelger@9499:     }
gernotbelger@9499: 
gernotbelger@9499:     @Override
gernotbelger@9499:     protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
gernotbelger@9499:         throw new UnsupportedOperationException();
gernotbelger@9499:     }
gernotbelger@9499: 
gernotbelger@9499: }