gernotbelger@8951: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@8951:  * Software engineering by
gernotbelger@8951:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@8951:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@8951:  *
gernotbelger@8951:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@8951:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@8951:  * documentation coming with Dive4Elements River for details.
gernotbelger@8951:  */
gernotbelger@8951: package org.dive4elements.river.artifacts.sinfo.flowdepthdev;
gernotbelger@8951: 
gernotbelger@8951: import java.util.ArrayList;
gernotbelger@8951: import java.util.Collection;
gernotbelger@8951: 
gernotbelger@8951: import org.apache.commons.lang.math.DoubleRange;
gernotbelger@8951: import org.dive4elements.artifacts.CallContext;
gernotbelger@8951: import org.dive4elements.river.artifacts.model.Calculation;
gernotbelger@8951: import org.dive4elements.river.artifacts.model.CalculationResult;
gernotbelger@8951: import org.dive4elements.river.artifacts.model.WKms;
gernotbelger@8951: import org.dive4elements.river.artifacts.resources.Resources;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.SINFOArtifact;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.common.RiverInfoProvider;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.common.SInfoResultRow;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.common.SInfoResultType;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.flowdepth.FlowDepthUtils;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.flowdepth.WstSoundingIdPair;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.tkhcalculation.WaterlevelValuesFinder;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.tkhstate.BedHeightsFinder;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.util.BedHeightInfo;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.util.CalculationUtils;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
gernotbelger@8951: import org.dive4elements.river.artifacts.sinfo.util.WstInfo;
gernotbelger@8951: import org.dive4elements.river.artifacts.states.WaterlevelData;
gernotbelger@8951: import org.dive4elements.river.artifacts.states.WaterlevelFetcher;
gernotbelger@8951: import org.dive4elements.river.model.River;
gernotbelger@8951: 
gernotbelger@8951: /**
gernotbelger@8951:  * @author Gernot Belger
gernotbelger@8951:  */
gernotbelger@8951: final class FlowDepthDevelopmentCalculation {
gernotbelger@8951: 
gernotbelger@8951:     private final CallContext context;
gernotbelger@8951: 
gernotbelger@8951:     public FlowDepthDevelopmentCalculation(final CallContext context) {
gernotbelger@8951:         this.context = context;
gernotbelger@8951:     }
gernotbelger@8951: 
gernotbelger@8951:     public CalculationResult calculate(final SINFOArtifact sinfo) {
gernotbelger@8951: 
gernotbelger@8951:         final String user = CalculationUtils.findArtifactUser(this.context, sinfo);
gernotbelger@8951: 
gernotbelger@8951:         /* access input data */
gernotbelger@8951:         final FlowDepthDevelopmentAccess access = new FlowDepthDevelopmentAccess(sinfo);
gernotbelger@8951:         final River river = access.getRiver();
gernotbelger@8951:         final RiverInfo riverInfo = new RiverInfo(river);
gernotbelger@8951: 
gernotbelger@8951:         final WstSoundingIdPair currentPair = access.getCurrentPair();
gernotbelger@8951:         final WstSoundingIdPair histPair = access.getHistoricalPair();
gernotbelger@8951: 
gernotbelger@8951:         final DoubleRange calcRange = access.getRange();
gernotbelger@8951: 
gernotbelger@8951:         /* calculate results for each diff pair */
gernotbelger@8951:         final Calculation problems = new Calculation();
gernotbelger@8951: 
gernotbelger@8951:         final RiverInfoProvider infoProvider = RiverInfoProvider.forRange(this.context, river, calcRange);
gernotbelger@8951: 
gernotbelger@8951:         final String calcModeLabel = Resources.getMsg(this.context.getMeta(), sinfo.getCalculationMode().name());
gernotbelger@8951: 
gernotbelger@8951:         final FlowDepthDevelopmentCalculationResults results = new FlowDepthDevelopmentCalculationResults(calcModeLabel, user, riverInfo, calcRange);
gernotbelger@8951: 
gernotbelger@8951:         final FlowDepthDevelopmentCalculationResult result = calculateResult(calcRange, currentPair, histPair, problems, infoProvider);
gernotbelger@8951:         if (result != null)
gernotbelger@8951:             results.addResult(result);
gernotbelger@8951: 
gernotbelger@8951:         return new CalculationResult(results, problems);
gernotbelger@8951:     }
gernotbelger@8951: 
gernotbelger@8951:     private FlowDepthDevelopmentCalculationResult calculateResult(final DoubleRange calcRange, final WstSoundingIdPair currentPair,
gernotbelger@8951:             final WstSoundingIdPair histPair, final Calculation problems, final RiverInfoProvider infoProvider) {
gernotbelger@8951: 
gernotbelger@8951:         /* access real input data from database */
gernotbelger@8951:         final WaterlevelData currentWaterlevel = loadWaterlevel(currentPair, problems);
gernotbelger@8951:         if (currentWaterlevel == null)
gernotbelger@8951:             return null;
gernotbelger@8951: 
gernotbelger@8951:         final WaterlevelData historicalWaterlevel = loadWaterlevel(histPair, problems);
gernotbelger@8951:         if (historicalWaterlevel == null)
gernotbelger@8951:             return null;
gernotbelger@8951: 
gernotbelger@8951:         final BedHeightsFinder currentSounding = loadBedHeight(currentPair, calcRange, problems);
gernotbelger@8951:         if (currentSounding == null)
gernotbelger@8951:             return null;
gernotbelger@8951: 
gernotbelger@8951:         final BedHeightsFinder historicalSounding = loadBedHeight(histPair, calcRange, problems);
gernotbelger@8951:         if (historicalSounding == null)
gernotbelger@8951:             return null;
gernotbelger@8951: 
gernotbelger@8951:         // FIXME: check current/hist wst have same discharge...
gernotbelger@8951: 
gernotbelger@8951:         final BedHeightInfo currentSoundingInfo = currentSounding.getInfo();
gernotbelger@8951:         final BedHeightInfo historicalSoundingInfo = historicalSounding.getInfo();
gernotbelger@8951: 
gernotbelger@8951:         // final String label = createLabel(waterlevel, minBedHeight, maxBedHeight);
gernotbelger@8951: 
gernotbelger@8951:         // final WKms wstKms = waterlevel.getWkms();
gernotbelger@8951:         final int currentWstYear = currentWaterlevel.getYear();
gernotbelger@8951:         final int historicalWstYear = historicalWaterlevel.getYear();
gernotbelger@8951:         final int currentSoundingYear = currentSoundingInfo.getYear();
gernotbelger@8951:         final int historicalSoundingYear = historicalSoundingInfo.getYear();
gernotbelger@8951: 
gernotbelger@8951:         // FIXME: distinguish error messages
gernotbelger@8951:         FlowDepthUtils.checkYearDifference("", currentWstYear, currentSoundingYear, problems);
gernotbelger@8951:         FlowDepthUtils.checkYearDifference("", historicalWstYear, historicalSoundingYear, problems);
gernotbelger@8951: 
gernotbelger@8951:         // checkWaterlevelDiscretisation(wstKms, calcRange, problems);
gernotbelger@8951:         // TODO: prüfen, ob sohlhöhen die calcRange abdecken/überschneiden
gernotbelger@8951: 
gernotbelger@8951:         /* re-determine the reference gauge, in the same way as the WaterlevelArtifact would do it */
gernotbelger@8951:         final RiverInfoProvider currentRiverInfoProvider = infoProvider.forWaterlevel(currentWaterlevel);
gernotbelger@8951:         final RiverInfoProvider histRiverInfoProvider = infoProvider.forWaterlevel(historicalWaterlevel);
gernotbelger@8951: 
gernotbelger@8951:         final WstInfo currentWstInfo = new WstInfo(currentWaterlevel.getName(), currentWstYear, currentRiverInfoProvider.getReferenceGauge());
gernotbelger@8951:         final WstInfo historicalWstInfo = new WstInfo(historicalWaterlevel.getName(), historicalWstYear, histRiverInfoProvider.getReferenceGauge());
gernotbelger@8951: 
gernotbelger@8951:         final WKms currentWkms = currentWaterlevel.getWkms();
gernotbelger@8951:         final WaterlevelValuesFinder currentWstProvider = WaterlevelValuesFinder.fromKms(currentWkms);
gernotbelger@8951:         // final DischargeValuesFinder currentDischargeProvider = DischargeValuesFinder.fromKms(currentWkms);
gernotbelger@8951: 
gernotbelger@8951:         final WKms historicalWkms = historicalWaterlevel.getWkms();
gernotbelger@8951:         final WaterlevelValuesFinder historicalWstProvider = WaterlevelValuesFinder.fromKms(historicalWkms);
gernotbelger@8951:         // final DischargeValuesFinder historicalDischargeProvider = DischargeValuesFinder.fromKms(historicalWkms);
gernotbelger@8951: 
gernotbelger@8951:         final int currentMeanYear = (currentWstYear + currentSoundingYear) / 2;
gernotbelger@8951:         final int historcialMeanYear = (historicalWstYear + historicalSoundingYear) / 2;
gernotbelger@8951: 
gernotbelger@8951:         // final String waterlevelLabel = waterlevel.getName();
gernotbelger@8951:         // final String soundingLabel = buildSoundingLabel(minBedHeight, maxBedHeight);
gernotbelger@8951: 
gernotbelger@8951:         final double diffYear = currentMeanYear - historcialMeanYear;
gernotbelger@8951: 
gernotbelger@8951:         /* real calculation loop */
gernotbelger@8951:         final Collection<SInfoResultRow> rows = new ArrayList<>();
gernotbelger@8951: 
gernotbelger@8951:         // FIXME: determine what is the spatial discretisation that we will use...
gernotbelger@8951:         final double[] allKms = currentWkms.allKms().toNativeArray();
gernotbelger@8951:         for (final double station : allKms) {
gernotbelger@8951:             if (calcRange.containsDouble(station)) {
gernotbelger@8951: 
gernotbelger@8951:                 final double currentWst = currentWstProvider.getWaterlevel(station);
gernotbelger@8951:                 // final double currentDischarge = currentDischargeProvider.getDischarge(station);
gernotbelger@8951:                 final double currentBedHeight = currentSounding.getMeanBedHeight(station);
gernotbelger@8951: 
gernotbelger@8951:                 final double historicalWst = historicalWstProvider.getWaterlevel(station);
gernotbelger@8951:                 // final double historicalDischarge = historicalDischargeProvider.getDischarge(station);
gernotbelger@8951:                 final double historicalBedHeight = historicalSounding.getMeanBedHeight(station);
gernotbelger@8951: 
gernotbelger@8951:                 final double diffWst = (currentWst - historicalWst) * 100;
gernotbelger@8951:                 final double diffBedHeight = (currentBedHeight - historicalBedHeight) * 100;
gernotbelger@8951: 
gernotbelger@8951:                 final double flowDepthDevelopment = diffWst - diffBedHeight;
gernotbelger@8951: 
gernotbelger@8951:                 final double flowDepthDevelopmentPerYear = flowDepthDevelopment / diffYear;
gernotbelger@8951: 
gernotbelger@8951:                 final double currentFlowDepth = currentWst - currentBedHeight;
gernotbelger@8951:                 final double historicalFlowDepth = historicalWst - historicalBedHeight;
gernotbelger@8951: 
gernotbelger@8951:                 // REMARK: access the location once only during calculation
gernotbelger@8951:                 final String location = currentRiverInfoProvider.getLocation(station);
gernotbelger@8951: 
gernotbelger@8951:                 final SInfoResultRow row = SInfoResultRow.create().//
gernotbelger@8951:                         putValue(SInfoResultType.station, station). //
gernotbelger@8951:                         putValue(SInfoResultType.flowdepthDevelopment, flowDepthDevelopment). //
gernotbelger@8951:                         putValue(SInfoResultType.flowdepthDevelopmentPerYear, flowDepthDevelopmentPerYear). //
gernotbelger@8951:                         putValue(SInfoResultType.waterlevelDifference, diffWst). //
gernotbelger@8951:                         putValue(SInfoResultType.bedHeightDifference, diffBedHeight). //
gernotbelger@8951:                         putValue(SInfoResultType.flowdepthCurrent, currentFlowDepth). //
gernotbelger@8951:                         putValue(SInfoResultType.flowdepthHistorical, historicalFlowDepth). //
gernotbelger@8951:                         putValue(SInfoResultType.location, location);
gernotbelger@8951:                 rows.add(row);
gernotbelger@8951:             }
gernotbelger@8951:         }
gernotbelger@8951: 
gernotbelger@8951:         return new FlowDepthDevelopmentCalculationResult("", currentWstInfo, historicalWstInfo, currentSoundingInfo, historicalSoundingInfo,
gernotbelger@8951:                 rows);
gernotbelger@8951:     }
gernotbelger@8951: 
gernotbelger@8951:     /* REMARK: fetch ALL wst kms, because we need to determine the original reference gauge */
gernotbelger@8951:     private WaterlevelData loadWaterlevel(final WstSoundingIdPair pair, final Calculation problems) {
gernotbelger@8951:         final String wstId = pair.getWstId();
gernotbelger@8951:         return new WaterlevelFetcher().findWaterlevel(this.context, wstId, Double.NaN, Double.NaN, problems);
gernotbelger@8951:     }
gernotbelger@8951: 
gernotbelger@8951:     private BedHeightsFinder loadBedHeight(final WstSoundingIdPair pair, final DoubleRange calcRange, final Calculation problems) {
gernotbelger@8951:         final String soundingId = pair.getSoundingId();
gernotbelger@8951:         return BedHeightsFinder.forId(this.context, soundingId, calcRange, problems);
gernotbelger@8951:     }
gernotbelger@8951: }