dive4elements/river: artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/tkhstate/TkhCalculation.java annotate

annotate artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/tkhstate/TkhCalculation.java @ 8948:a4f1ac81f26d

Work on SINFO-FlowDepthMinMax. Also rework of result row stuff, in order to reduce abstraction, using result type concept

author	gernotbelger
date	Wed, 14 Mar 2018 14:10:32 +0100
parents	5d5d482da3e9
children	183f42641ab6

rev	line source
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	1 /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	2 * Software engineering by
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	3 * Björnsen Beratende Ingenieure GmbH
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	4 * Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	5 *
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	6 * This file is Free Software under the GNU AGPL (>=v3)
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	7 * and comes with ABSOLUTELY NO WARRANTY! Check out the
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	8 * documentation coming with Dive4Elements River for details.
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	9 */
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	10 package org.dive4elements.river.artifacts.sinfo.tkhstate;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	11
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	12 import java.util.ArrayList;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	13 import java.util.Collection;
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	14 import java.util.HashMap;
8916 5d5d0051723f Working on outputmodes of tkh calculation gernotbelger parents: 8915 diff changeset	15 import java.util.List;
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	16 import java.util.Map;
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	17 import java.util.Map.Entry;
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	18 import java.util.Set;
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	19
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	20 import org.apache.commons.lang.math.DoubleRange;
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	21 import org.apache.commons.lang.math.NumberRange;
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	22 import org.dive4elements.artifacts.CallContext;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	23 import org.dive4elements.river.artifacts.WINFOArtifact;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	24 import org.dive4elements.river.artifacts.model.Calculation;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	25 import org.dive4elements.river.artifacts.model.Calculation.Problem;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	26 import org.dive4elements.river.artifacts.model.CalculationResult;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	27 import org.dive4elements.river.artifacts.model.WQKms;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	28 import org.dive4elements.river.artifacts.resources.Resources;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	29 import org.dive4elements.river.artifacts.sinfo.SINFOArtifact;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	30 import org.dive4elements.river.artifacts.sinfo.common.RiverInfoProvider;
8948 a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	31 import org.dive4elements.river.artifacts.sinfo.common.SInfoResultRow;
a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	32 import org.dive4elements.river.artifacts.sinfo.common.SInfoResultType;
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	33 import org.dive4elements.river.artifacts.sinfo.tkhcalculation.DischargeValuesFinder;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	34 import org.dive4elements.river.artifacts.sinfo.tkhcalculation.TkhCalculator;
8946 5d5d482da3e9 Implementing SINFO - FlowDepthMinMax calculation gernotbelger parents: 8942 diff changeset	35 import org.dive4elements.river.artifacts.sinfo.tkhcalculation.WaterlevelValuesFinder;
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	36 import org.dive4elements.river.artifacts.sinfo.util.BedHeightInfo;
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	37 import org.dive4elements.river.artifacts.sinfo.util.CalculationUtils;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	38 import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	39 import org.dive4elements.river.artifacts.sinfo.util.WstInfo;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	40 import org.dive4elements.river.artifacts.states.WaterlevelData;
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	41 import org.dive4elements.river.exports.WaterlevelDescriptionBuilder;
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	42 import org.dive4elements.river.model.BedHeight;
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	43 import org.dive4elements.river.model.River;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	44
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	45 /**
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	46 * @author Gernot Belger
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	47 */
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	48 final class TkhCalculation {
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	49
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	50 private final CallContext context;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	51
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	52 public TkhCalculation(final CallContext context) {
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	53 this.context = context;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	54 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	55
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	56 public CalculationResult calculate(final SINFOArtifact sinfo) {
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	57
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	58 /* access input data */
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	59 final TkhAccess access = new TkhAccess(sinfo);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	60 final River river = access.getRiver();
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	61 final RiverInfo riverInfo = new RiverInfo(river);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	62 final DoubleRange calcRange = access.getRange();
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	63
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	64 final Calculation problems = new Calculation();
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	65
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	66 /* find relevant bed-heights */
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	67 final List<BedHeight> defaultBedHeights = new DefaultBedHeights(river).getBedHeights(problems);
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	68 final Collection<BedHeightsFinder> bedHeights = BedHeightsFinder.createTkhBedHeights(calcRange, defaultBedHeights);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	69
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	70 /* misuse winfo-artifact to calculate waterlevels in the same way */
9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	71 final WINFOArtifact winfo = new WinfoArtifactWrapper(sinfo);
9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	72
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	73 /* calculate waterlevels */
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	74 final WQKms[] kms = calculateWaterlevels(winfo, problems);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	75
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	76 final RiverInfoProvider infoProvider = RiverInfoProvider.forRange(this.context, river, calcRange);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	77
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	78 final String user = CalculationUtils.findArtifactUser(this.context, sinfo);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	79
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	80 final String calcModeLabel = Resources.getMsg(this.context.getMeta(), sinfo.getCalculationMode().name());
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	81
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	82 final WaterlevelDescriptionBuilder descBuilder = new WaterlevelDescriptionBuilder(winfo, this.context);
9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	83 final String descriptionHeader = descBuilder.getColumnHeader();
9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	84
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	85 /* for each waterlevel, do a tkh calculation */
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	86 final TkhCalculationResults results = new TkhCalculationResults(calcModeLabel, user, riverInfo, calcRange, descriptionHeader);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	87
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	88 for (final WQKms wqKms : kms) {
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	89
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	90 final TkhCalculationResult result = calculateResult(calcRange, infoProvider, wqKms, bedHeights, descBuilder, problems);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	91 if (result != null)
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	92 // FIXME: must be sorted by station!
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	93 results.addResult(result);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	94 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	95
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	96 return new CalculationResult(results, problems);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	97 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	98
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	99 private WQKms[] calculateWaterlevels(final WINFOArtifact winfo, final Calculation problems) {
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	100
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	101 final CalculationResult waterlevelData = winfo.getWaterlevelData(this.context);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	102
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	103 /* copy all problems */
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	104 final Calculation winfoProblems = waterlevelData.getReport();
8916 5d5d0051723f Working on outputmodes of tkh calculation gernotbelger parents: 8915 diff changeset	105 final List<Problem> problems2 = winfoProblems.getProblems();
5d5d0051723f Working on outputmodes of tkh calculation gernotbelger parents: 8915 diff changeset	106 if (problems2 != null) {
5d5d0051723f Working on outputmodes of tkh calculation gernotbelger parents: 8915 diff changeset	107 for (final Problem problem : problems2) {
5d5d0051723f Working on outputmodes of tkh calculation gernotbelger parents: 8915 diff changeset	108 problems.addProblem(problem);
5d5d0051723f Working on outputmodes of tkh calculation gernotbelger parents: 8915 diff changeset	109 }
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	110 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	111
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	112 return (WQKms[]) waterlevelData.getData();
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	113 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	114
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	115 private TkhCalculationResult calculateResult(final DoubleRange calcRange, final RiverInfoProvider riverInfo, final WQKms wkms,
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	116 final Collection<BedHeightsFinder> bedHeights, final WaterlevelDescriptionBuilder descBuilder, final Calculation problems) {
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	117
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	118 // FIXME: wo kommt das her? via winfo kein jahr vorhanden, oder doch? aber soll in metadaten ausgegeben werden...
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	119 final int wspYear = -1;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	120 // FIXME: richtig? vgl. WInfo?
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	121 final boolean showAllGauges = false;
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	122 final WaterlevelData waterlevel = new WaterlevelData(wkms, wspYear, showAllGauges);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	123
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	124 final RiverInfoProvider riverInfoProvider = riverInfo.forWaterlevel(waterlevel);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	125
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	126 // FIXME: check with winfo how the name is generated
9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	127 final String wstLabel = waterlevel.getName();
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	128
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	129 final WstInfo wstInfo = new WstInfo(wstLabel, wspYear, riverInfoProvider.getReferenceGauge());
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	130
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	131 /* build tkh calculators per bedheight */
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	132 final Map<NumberRange, TkhCalculator> calculatorsByRanges = buildCalculators(calcRange, wkms, bedHeights, problems, riverInfoProvider, wstLabel);
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	133 if (calculatorsByRanges.isEmpty()) {
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	134 /* there should already be some problems, so just abort */
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	135 return null;
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	136 }
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	137
8948 a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	138 final Collection<SInfoResultRow> rows = new ArrayList<>();
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	139
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	140 /* using wst-kms as basis, because we know that they are generated wst's with a fixed km-step */
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	141 // FIXME: das führt dazu, das aktuell die Sohlhöhen beliebig linear interpolierrt werden. ist das immer richtig? z.b.
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	142 // bei großen abständen?
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	143
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	144 final int size = wkms.size();
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	145 for (int i = 0; i < size; i++) {
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	146
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	147 final double station = wkms.getKm(i);
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	148
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	149 /* find the right calculator (i.e. bedheigh) depending on station, there should only be one maximal */
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	150 final TkhCalculator tkhCalculator = findCalculator(calculatorsByRanges, station);
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	151 if (tkhCalculator == null)
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	152 continue;
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	153
8948 a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	154 final SInfoResultRow row = SInfoResultRow.create();
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	155
8948 a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	156 row.putValue(SInfoResultType.waterlevelLabel, descBuilder.getDesc(wkms));
a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	157 row.putValue(SInfoResultType.gaugeLabel, riverInfoProvider.findGauge(station));
a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	158 row.putValue(SInfoResultType.location, riverInfoProvider.getLocation(station));
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	159
8948 a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	160 tkhCalculator.calculateTkh(station, row);
a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	161
a4f1ac81f26d Work on SINFO-FlowDepthMinMax. gernotbelger parents: 8946 diff changeset	162 rows.add(row);
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	163 }
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	164
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	165 return new TkhCalculationResult(wstLabel, wstInfo, true, rows);
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	166 }
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	167
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	168 private TkhCalculator findCalculator(final Map<NumberRange, TkhCalculator> calculators, final double station) {
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	169
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	170 // REMAKR: linear search at this point, put we expect the number of bed heights to be very small (1-2 items)
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	171 final Set<Entry<NumberRange, TkhCalculator>> x = calculators.entrySet();
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	172 for (final Entry<NumberRange, TkhCalculator> entry : x) {
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	173 final NumberRange range = entry.getKey();
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	174 // FIXME: check if we need comparison with a tolerance
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	175 if (range.containsDouble(station))
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	176 return entry.getValue();
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	177 }
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	178
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	179 return null;
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	180 }
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	181
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	182 private Map<NumberRange, TkhCalculator> buildCalculators(final DoubleRange calcRange, final WQKms wkms, final Collection<BedHeightsFinder> bedHeights,
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	183 final Calculation problems, final RiverInfoProvider riverInfoProvider, final String wstLabel) {
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	184 final Map<NumberRange, TkhCalculator> calculatorByRanges = new HashMap<>();
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	185 for (final BedHeightsFinder bedHeightsProvider : bedHeights) {
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	186
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	187 final BedHeightInfo info = bedHeightsProvider.getInfo();
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	188
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	189 final NumberRange range = new NumberRange(info.getFrom(), info.getTo());
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	190
8946 5d5d482da3e9 Implementing SINFO - FlowDepthMinMax calculation gernotbelger parents: 8942 diff changeset	191 final WaterlevelValuesFinder waterlevelProvider = WaterlevelValuesFinder.fromKms(wkms);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	192 final DischargeValuesFinder dischargeProvider = DischargeValuesFinder.fromKms(wkms);
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	193
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	194 /* initialize tkh calculator */
8938 9c02733a1b3c Work on Sinfo-tkh - exports; using same logic for wst-description as winfo gernotbelger parents: 8916 diff changeset	195 final TkhCalculator tkhCalculator = TkhCalculator.buildTkhCalculator(true, this.context, problems, wstLabel, riverInfoProvider.getRiver(),
8946 5d5d482da3e9 Implementing SINFO - FlowDepthMinMax calculation gernotbelger parents: 8942 diff changeset	196 calcRange, waterlevelProvider, dischargeProvider, bedHeightsProvider);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	197
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	198 if (tkhCalculator != null) {
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	199 /* just ignore null ones, problems have already been updated by buildTkhCalculator() */
11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	200 calculatorByRanges.put(range, tkhCalculator);
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	201 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	202 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	203
8942 11bf13cf0463 Minor changes to tkh calculation. Loading default bed heights form config file. gernotbelger parents: 8938 diff changeset	204 return calculatorByRanges;
8915 d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	205 }
d9dbf0b74bc2 Refaktoring of flow depth calculation, extracting tkh part. First implementation of tkh calculation. gernotbelger parents: diff changeset	206 }

Mercurial > dive4elements > river

annotate artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/tkhstate/TkhCalculation.java @ 8948:a4f1ac81f26d