Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/tkhstate/TkhState.java @ 8940:82998242ba84
Preparing for additional outputs of SINFO-Tkh
author | gernotbelger |
---|---|
date | Tue, 06 Mar 2018 18:51:18 +0100 |
parents | 9c02733a1b3c |
children | a9950a3a71e5 |
line wrap: on
line source
/* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde * Software engineering by Intevation GmbH * * This file is Free Software under the GNU AGPL (>=v3) * and comes with ABSOLUTELY NO WARRANTY! Check out the * documentation coming with Dive4Elements River for details. */ package org.dive4elements.river.artifacts.sinfo.tkhstate; import java.util.List; import org.dive4elements.artifactdatabase.state.Facet; import org.dive4elements.artifactdatabase.state.FacetActivity; import org.dive4elements.artifacts.Artifact; import org.dive4elements.artifacts.CallContext; import org.dive4elements.river.artifacts.ChartArtifact; import org.dive4elements.river.artifacts.D4EArtifact; import org.dive4elements.river.artifacts.model.Calculation; import org.dive4elements.river.artifacts.model.CalculationResult; import org.dive4elements.river.artifacts.model.DataFacet; import org.dive4elements.river.artifacts.model.EmptyFacet; import org.dive4elements.river.artifacts.model.FacetTypes; import org.dive4elements.river.artifacts.model.ReportFacet; import org.dive4elements.river.artifacts.sinfo.SINFOArtifact; import org.dive4elements.river.artifacts.sinfo.common.FlowDepthProcessor; import org.dive4elements.river.artifacts.sinfo.common.TkhProcessor; import org.dive4elements.river.artifacts.states.DefaultState; /** State in which a waterlevel has been calculated. */ public class TkhState extends DefaultState { /// ** The log that is used in this state. */ // private static Logger log = Logger.getLogger(FlowDepthState.class); private static final long serialVersionUID = 1L; static { // Active/deactivate facets. // BEWARE: we can only define one activity for "sinfo", so this is not the right place.... FacetActivity.Registry.getInstance().register("sinfo", new FacetActivity() { @Override public Boolean isInitialActive(final Artifact artifact, final Facet facet, final String output) { /* only */ if ("sinfo_tkk".equals(output)) { final String name = facet.getName(); if (FlowDepthProcessor.FACET_FLOW_DEPTH_FILTERED.equals(name)) return Boolean.FALSE; } return null; } }); } /** * From this state can only be continued trivially. */ @Override protected String getUIProvider() { return "continue"; } @Override public Object computeFeed(final D4EArtifact artifact, final String hash, final CallContext context, final List<Facet> facets, final Object old) { // FIXME: why is this necessary? if (artifact instanceof ChartArtifact) { facets.add(new EmptyFacet()); return null; } return compute((SINFOArtifact) artifact, context, hash, facets, old); } @Override public Object computeAdvance(final D4EArtifact artifact, final String hash, final CallContext context, final List<Facet> facets, final Object old) { if (artifact instanceof ChartArtifact) { facets.add(new EmptyFacet()); return null; } return compute((SINFOArtifact) artifact, context, hash, facets, old); } /** * Compute result or returned object from cache, create facets. * * @param old * Object that was cached. */ private Object compute(final SINFOArtifact sinfo, final CallContext context, final String hash, final List<Facet> facets, final Object old) { final CalculationResult res = doCompute(sinfo, context, old); if (facets == null) return res; final TkhCalculationResults results = (TkhCalculationResults) res.getData(); /* add themes for chart, for each result */ final List<TkhCalculationResult> resultList = results.getResults(); for (int index = 0; index < resultList.size(); index++) { final TkhCalculationResult result = resultList.get(index); facets.add(TkhProcessor.createTkhFacet(context, hash, this.id, result, index)); // FIXME: should only be optionally visible facets.add(FlowDepthProcessor.createFlowDepthTkhFacet(context, hash, this.id, result, index)); } if (!resultList.isEmpty()) { facets.add(new DataFacet(FacetTypes.CSV, "CSV data", ComputeType.ADVANCE, hash, this.id)); facets.add(new DataFacet(FacetTypes.PDF, "PDF data", ComputeType.ADVANCE, hash, this.id)); } final Calculation report = res.getReport(); if (report.hasProblems()) facets.add(new ReportFacet(ComputeType.ADVANCE, hash, this.id)); return res; } private CalculationResult doCompute(final SINFOArtifact sinfo, final CallContext context, final Object old) { if (old instanceof CalculationResult) return (CalculationResult) old; return new TkhCalculation(context).calculate(sinfo); } }