gernotbelger@8996: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9024:  * Software engineering by
gernotbelger@9024:  *  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.vegetationzones;
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;
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 VegetationZonesState extends DefaultState {
gernotbelger@8996: 
gernotbelger@8996:     /// ** The log that is used in this state. */
gernotbelger@8996:     // private static Logger log = Logger.getLogger(FlowDepthState.class);
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:      */
gernotbelger@8996:     private Object compute(final UINFOArtifact sinfo, final CallContext context, final String hash, final List<Facet> facets, final Object old) {
gernotbelger@8996: 
gernotbelger@8996:         final CalculationResult res = doCompute(sinfo, context, old);
gernotbelger@8996: 
gernotbelger@8996:         if (facets == null)
gernotbelger@8996:             return res;
gernotbelger@8996: 
gernotbelger@8996:         final VegetationZonesCalculationResults results = (VegetationZonesCalculationResults) res.getData();
gernotbelger@8996: 
gernotbelger@9024:         final List<VegetationZonesCalculationResult> resultList = results.getResults();
gernotbelger@8996: 
gernotbelger@8996:         if (!resultList.isEmpty()) {
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: 
gernotbelger@8996:     private CalculationResult doCompute(final UINFOArtifact sinfo, final CallContext context, final Object old) {
gernotbelger@8996:         if (old instanceof CalculationResult)
gernotbelger@8996:             return (CalculationResult) old;
gernotbelger@8996: 
gernotbelger@9024:         return new VegetationZonesCalculation(context).calculate(sinfo); // Test2
gernotbelger@8996:     }
gernotbelger@8996: }