gernotbelger@8996: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9150:  * Software engineering by
gernotbelger@9150:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@8996:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@8996:  *
gernotbelger@8996:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@8996:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@8996:  * documentation coming with Dive4Elements River for details.
gernotbelger@8996:  */
gernotbelger@8996: package org.dive4elements.river.artifacts.uinfo.salix;
gernotbelger@8996: 
gernotbelger@8996: import java.util.List;
gernotbelger@8996: 
gernotbelger@8996: import org.dive4elements.artifactdatabase.state.Facet;
gernotbelger@8996: import org.dive4elements.artifacts.CallContext;
gernotbelger@8996: import org.dive4elements.river.artifacts.ChartArtifact;
gernotbelger@8996: import org.dive4elements.river.artifacts.D4EArtifact;
gernotbelger@8996: import org.dive4elements.river.artifacts.model.Calculation;
gernotbelger@8996: import org.dive4elements.river.artifacts.model.CalculationResult;
gernotbelger@8996: import org.dive4elements.river.artifacts.model.DataFacet;
gernotbelger@8996: import org.dive4elements.river.artifacts.model.EmptyFacet;
gernotbelger@8996: import org.dive4elements.river.artifacts.model.FacetTypes;
gernotbelger@8996: import org.dive4elements.river.artifacts.model.ReportFacet;
mschaefer@9316: import org.dive4elements.river.artifacts.resources.Resources;
gernotbelger@8996: import org.dive4elements.river.artifacts.states.DefaultState;
gernotbelger@8996: import org.dive4elements.river.artifacts.uinfo.UINFOArtifact;
gernotbelger@8996: 
gernotbelger@8996: /**
gernotbelger@8996:  * @author Domenico Nardi Tironi
gernotbelger@8996:  */
gernotbelger@8996: public class SalixLineState extends DefaultState {
gernotbelger@8996: 
gernotbelger@8996:     private static final long serialVersionUID = 1L;
gernotbelger@8996: 
gernotbelger@8996:     /**
gernotbelger@8996:      * From this state can only be continued trivially.
gernotbelger@8996:      */
gernotbelger@8996:     @Override
gernotbelger@8996:     protected String getUIProvider() {
gernotbelger@8996:         return "continue";
gernotbelger@8996:     }
gernotbelger@8996: 
gernotbelger@8996:     @Override
gernotbelger@8996:     public Object computeFeed(final D4EArtifact artifact, final String hash, final CallContext context, final List<Facet> facets, final Object old) {
gernotbelger@8996:         // FIXME: why is this necessary?
gernotbelger@8996:         if (artifact instanceof ChartArtifact) {
gernotbelger@8996:             facets.add(new EmptyFacet());
gernotbelger@8996:             return null;
gernotbelger@8996:         }
gernotbelger@8996: 
gernotbelger@8996:         return compute((UINFOArtifact) artifact, context, hash, facets, old);
gernotbelger@8996:     }
gernotbelger@8996: 
gernotbelger@8996:     @Override
gernotbelger@8996:     public Object computeAdvance(final D4EArtifact artifact, final String hash, final CallContext context, final List<Facet> facets, final Object old) {
gernotbelger@8996:         if (artifact instanceof ChartArtifact) {
gernotbelger@8996:             facets.add(new EmptyFacet());
gernotbelger@8996:             return null;
gernotbelger@8996:         }
gernotbelger@8996:         return compute((UINFOArtifact) artifact, context, hash, facets, old);
gernotbelger@8996:     }
gernotbelger@8996: 
gernotbelger@8996:     /**
gernotbelger@8996:      * Compute result or returned object from cache, create facets.
gernotbelger@8996:      *
gernotbelger@8996:      * @param old
gernotbelger@8996:      *            Object that was cached.
gernotbelger@8996:      */
mschaefer@9316:     private Object compute(final UINFOArtifact uinfo, final CallContext context, final String hash, final List<Facet> facets, final Object old) {
gernotbelger@8996: 
mschaefer@9316:         final CalculationResult res = doCompute(uinfo, context, old);
gernotbelger@8996: 
gernotbelger@8996:         if (facets == null)
gernotbelger@8996:             return res;
gernotbelger@8996: 
gernotbelger@8996:         final SalixLineCalculationResults results = (SalixLineCalculationResults) res.getData();
gernotbelger@8996: 
gernotbelger@9243:         final List<SalixLineCalculationNoScenarioResult> resultList = results.getResults();
gernotbelger@8996: 
gernotbelger@8996:         if (!resultList.isEmpty()) {
mschaefer@9316:             facets.add(SalixLineProcessor.createSalixLineFacet(context, hash, this.id, resultList.get(0), 0, 0));
mschaefer@9316:             facets.add(SalixLineProcessor.createSalixMnwMwFacet(context, hash, this.id, resultList.get(0), 0, 0));
mschaefer@9316:             if (resultList.get(0) instanceof SalixLineCalculationRegionalResult) {
mschaefer@9316:                 final SalixLineCalculationRegionalResult result = (SalixLineCalculationRegionalResult) resultList.get(0);
mschaefer@9316:                 for (int i = 0; i <= result.getScenarioCount() - 1; i++) {
mschaefer@9316:                     final String sublabel = Resources.getMsg(context.getMeta(), "uinfo_salix_scenario_deltaw", "uinfo_salix_scenario_deltaw",
mschaefer@9316:                             result.getScenarioLabel(i));
mschaefer@9316:                     facets.add(SalixLineProcessor.createSalixScenarioFacet(context, hash, this.id, result, i, 0, sublabel));
mschaefer@9316:                 }
mschaefer@9316:             }
mschaefer@9316:             else if (resultList.get(0) instanceof SalixLineCalculationSupraRegionalResult) {
mschaefer@9316:                 final String sublabel = Resources.getMsg(context.getMeta(), "uinfo_salix_scenario_supraregional");
mschaefer@9316:                 facets.add(SalixLineProcessor.createSalixScenarioFacet(context, hash, this.id, resultList.get(0), 0, 0, sublabel));
mschaefer@9316:             }
mschaefer@9316: 
mschaefer@9316:             else if (resultList.get(0) instanceof SalixLineCalculationHistoricalResult) {
mschaefer@9316:                 final String sublabel = Resources.getMsg(context.getMeta(), "uinfo_salix_scenario_historical");
mschaefer@9316:                 facets.add(SalixLineProcessor.createSalixScenarioFacet(context, hash, this.id, resultList.get(0), 0, 0, sublabel));
mschaefer@9316:             }
mschaefer@9295: 
gernotbelger@8996:             final Facet csv = new DataFacet(FacetTypes.CSV, "CSV data", ComputeType.ADVANCE, hash, this.id);
gernotbelger@8996:             final Facet pdf = new DataFacet(FacetTypes.PDF, "PDF data", ComputeType.ADVANCE, hash, this.id);
gernotbelger@8996: 
gernotbelger@8996:             facets.add(csv);
gernotbelger@8996:             facets.add(pdf);
gernotbelger@8996:         }
gernotbelger@8996: 
gernotbelger@8996:         final Calculation report = res.getReport();
gernotbelger@8996: 
gernotbelger@8996:         if (report.hasProblems())
gernotbelger@8996:             facets.add(new ReportFacet(ComputeType.ADVANCE, hash, this.id));
gernotbelger@8996: 
gernotbelger@8996:         return res;
gernotbelger@8996:     }
gernotbelger@8996: 
mschaefer@9295:     private CalculationResult doCompute(final UINFOArtifact uinfo, final CallContext context, final Object old) {
gernotbelger@8996:         if (old instanceof CalculationResult)
gernotbelger@8996:             return (CalculationResult) old;
gernotbelger@8996: 
mschaefer@9295:         return new SalixLineCalculation(context).calculate(uinfo);
gernotbelger@8996:     }
gernotbelger@8996: }