Mercurial > dive4elements > river
comparison artifacts/src/main/java/org/dive4elements/river/artifacts/model/river/MainWstValuesCalculator.java @ 9499:853f2dafc16e
VegetationZones in CrossSectionsDiagram
| author | gernotbelger |
|---|---|
| date | Thu, 27 Sep 2018 18:06:26 +0200 |
| parents | |
| children | 8b7bf26b8782 |
comparison
equal
deleted
inserted
replaced
| 9496:d8e753d0fdb9 | 9499:853f2dafc16e |
|---|---|
| 1 /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde | |
| 2 * Software engineering by | |
| 3 * Björnsen Beratende Ingenieure GmbH | |
| 4 * Dr. Schumacher Ingenieurbüro für Wasser und Umwelt | |
| 5 * | |
| 6 * This file is Free Software under the GNU AGPL (>=v3) | |
| 7 * and comes with ABSOLUTELY NO WARRANTY! Check out the | |
| 8 * documentation coming with Dive4Elements River for details. | |
| 9 */ | |
| 10 package org.dive4elements.river.artifacts.model.river; | |
| 11 | |
| 12 import java.util.HashMap; | |
| 13 import java.util.Map; | |
| 14 | |
| 15 import org.apache.commons.lang.math.DoubleRange; | |
| 16 import org.dive4elements.artifacts.CallContext; | |
| 17 import org.dive4elements.river.artifacts.model.WstValueTable; | |
| 18 import org.dive4elements.river.artifacts.model.WstValueTable.QPosition; | |
| 19 import org.dive4elements.river.artifacts.model.WstValueTableFactory; | |
| 20 import org.dive4elements.river.model.Gauge; | |
| 21 import org.dive4elements.river.model.MainValue; | |
| 22 import org.dive4elements.river.model.MainValueType.MainValueTypeKey; | |
| 23 import org.dive4elements.river.model.River; | |
| 24 | |
| 25 /** | |
| 26 * @author Domenico Nardi Tironi | |
| 27 */ | |
| 28 public final class MainWstValuesCalculator { | |
| 29 | |
| 30 private final WstValueTable wst; | |
| 31 | |
| 32 private final Map<String, MainValueQPosition> positions; | |
| 33 | |
| 34 private static class MainValueQPosition { | |
| 35 | |
| 36 private final Map<Gauge, QPosition> gaugePositions = new HashMap<>(); | |
| 37 private QPosition refGaugePositions = null; | |
| 38 } | |
| 39 | |
| 40 public static MainWstValuesCalculator forRiver(final CallContext context, final River river, final DoubleRange calcRange, final String... mainValueNames) { | |
| 41 | |
| 42 final RiverInfoProvider info = RiverInfoProvider.forRange(context, river, calcRange); | |
| 43 | |
| 44 return forRiverInfo(info, mainValueNames); | |
| 45 } | |
| 46 | |
| 47 public static MainWstValuesCalculator forRiverInfo(final RiverInfoProvider info, final String... mainValueNames) { | |
| 48 | |
| 49 final WstValueTable wst = WstValueTableFactory.getTable(info.getRiver()); | |
| 50 | |
| 51 final Map<String, MainValueQPosition> positions = calculatePositions(info, wst, mainValueNames); | |
| 52 | |
| 53 return new MainWstValuesCalculator(wst, positions); | |
| 54 } | |
| 55 | |
| 56 private static Map<String, MainValueQPosition> calculatePositions(final RiverInfoProvider info, final WstValueTable wst, final String[] mainValueNames) { | |
| 57 | |
| 58 boolean isFirstGauge = true; | |
| 59 | |
| 60 final Map<String, MainValueQPosition> positions = new HashMap<>(); | |
| 61 | |
| 62 for (final String mainValue : mainValueNames) | |
| 63 positions.put(mainValue.toUpperCase(), new MainValueQPosition()); | |
| 64 | |
| 65 for (final Gauge gauge : info.getGauges()) { | |
| 66 | |
| 67 for (final MainValueQPosition position : positions.values()) | |
| 68 position.gaugePositions.put(gauge, null); | |
| 69 | |
| 70 final double gaugeKm = gauge.getStation().doubleValue(); | |
| 71 for (final MainValue mv : MainValue.getValuesOfGaugeAndType(gauge, MainValueTypeKey.Q)) { | |
| 72 | |
| 73 final MainValueQPosition position = positions.get(mv.getMainValue().getName().toUpperCase()); | |
| 74 if (position != null) { | |
| 75 final QPosition qPosition = wst.getQPosition(gaugeKm, mv.getValue().doubleValue()); | |
| 76 position.gaugePositions.put(gauge, qPosition); | |
| 77 | |
| 78 if (isFirstGauge) | |
| 79 position.refGaugePositions = qPosition; | |
| 80 } | |
| 81 } | |
| 82 | |
| 83 isFirstGauge = false; | |
| 84 } | |
| 85 | |
| 86 return positions; | |
| 87 } | |
| 88 | |
| 89 private MainWstValuesCalculator(final WstValueTable wst, final Map<String, MainValueQPosition> positions) { | |
| 90 this.wst = wst; | |
| 91 this.positions = positions; | |
| 92 } | |
| 93 | |
| 94 public boolean hasPosition(final String mainValueName) { | |
| 95 | |
| 96 final MainValueQPosition position = this.positions.get(mainValueName); | |
| 97 if (position == null) | |
| 98 throw new IllegalArgumentException(); | |
| 99 | |
| 100 return position.refGaugePositions != null; | |
| 101 } | |
| 102 | |
| 103 /** | |
| 104 * Interpolates the W for a station with a fixed (virtual) wst column position | |
| 105 */ | |
| 106 public double interpolateW(final double station, final String mainValueName) { | |
| 107 | |
| 108 final MainValueQPosition mainValuePosition = this.positions.get(mainValueName.toUpperCase()); | |
| 109 | |
| 110 if (mainValuePosition.refGaugePositions == null) | |
| 111 return Double.NaN; | |
| 112 | |
| 113 return this.wst.interpolateW(station, mainValuePosition.refGaugePositions); | |
| 114 } | |
| 115 } |