mschaefer@9252: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
mschaefer@9252:  * Software engineering by Intevation GmbH
mschaefer@9252:  *
mschaefer@9252:  * This file is Free Software under the GNU AGPL (>=v3)
mschaefer@9252:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
mschaefer@9252:  * documentation coming with Dive4Elements River for details.
mschaefer@9252:  */
mschaefer@9252: 
mschaefer@9252: package org.dive4elements.river.artifacts.sinfo.flood_duration;
mschaefer@9252: 
mschaefer@9252: import java.util.ArrayList;
mschaefer@9252: import java.util.List;
mschaefer@9252: 
mschaefer@9252: import org.apache.log4j.Logger;
mschaefer@9252: import org.dive4elements.artifactdatabase.state.DefaultFacet;
mschaefer@9252: import org.dive4elements.artifactdatabase.state.Facet;
mschaefer@9252: import org.dive4elements.artifacts.Artifact;
mschaefer@9252: import org.dive4elements.artifacts.CallContext;
mschaefer@9252: import org.dive4elements.river.artifacts.D4EArtifact;
mschaefer@9252: import org.dive4elements.river.artifacts.model.CalculationResult;
mschaefer@9252: import org.dive4elements.river.artifacts.model.WQDay;
mschaefer@9252: import org.dive4elements.river.artifacts.states.DefaultState.ComputeType;
mschaefer@9252: 
mschaefer@9252: 
mschaefer@9252: /**
mschaefer@9252:  * Data of a duration curve.
mschaefer@9252:  */
mschaefer@9252: public class FloodDurationCurveFacet extends DefaultFacet {
mschaefer@9252: 
mschaefer@9252:     private static Logger log = Logger.getLogger(FloodDurationCurveFacet.class);
mschaefer@9252: 
mschaefer@9252:     /** Blackboard data provider key for durationcurve (wqday) data. */
mschaefer@9252:     public static String BB_DURATIONCURVE = "durationcurve";
mschaefer@9252: 
mschaefer@9252:     /** Blackboard data provider key for km of durationcurve. */
mschaefer@9252:     public static String BB_DURATIONCURVE_KM = "durationcurve.km";
mschaefer@9252: 
mschaefer@9252:     public FloodDurationCurveFacet() {
mschaefer@9252:     }
mschaefer@9252: 
mschaefer@9252:     public FloodDurationCurveFacet(final String name, final String description) {
mschaefer@9252:         super(0, name, description);
mschaefer@9252:     }
mschaefer@9252: 
mschaefer@9252: 
mschaefer@9252:     /**
mschaefer@9252:      * Expose state computation from SINFOArtifact.
mschaefer@9252:      */
mschaefer@9252:     @Override
mschaefer@9252:     public Object getData(final Artifact artifact, final CallContext context) {
mschaefer@9252: 
mschaefer@9252:         log.debug("Get data for flood duration curve data");
mschaefer@9252: 
mschaefer@9252:         final D4EArtifact flys = (D4EArtifact) artifact;
mschaefer@9252: 
mschaefer@9252:         final CalculationResult res = (CalculationResult) flys.compute(context, ComputeType.ADVANCE, false);
mschaefer@9252: 
mschaefer@9252:         final FloodDurationCalculationResults data = (FloodDurationCalculationResults) res.getData();
mschaefer@9252: 
mschaefer@9252:         return data.getDurationCurve();
mschaefer@9252:         // return getTestData();
mschaefer@9252:     }
mschaefer@9252: 
mschaefer@9252:     private WQDay getTestData() {
mschaefer@9252:         final int[] days = new int[366];
mschaefer@9252:         final double[] ws = new double[366];
mschaefer@9252:         final double[] qs = new double[366];
mschaefer@9252:         for (int i = 0; i <= 365; i++) {
mschaefer@9252:             days[i] = i;
mschaefer@9252:             final double x = (i - 182.5) / 182.5;
mschaefer@9252:             ws[i] = 102.0 - (Math.pow(x, 5) + x);
mschaefer@9252:             qs[i] = 1600.0 - 800 * (Math.pow(x, 9) + x);
mschaefer@9252:         }
mschaefer@9252:         return new WQDay(days, ws, qs);
mschaefer@9252:     }
mschaefer@9252: 
mschaefer@9252: 
mschaefer@9252:     @Override
mschaefer@9252:     public List getStaticDataProviderKeys(final Artifact art) {
mschaefer@9252:         final List list = new ArrayList();
mschaefer@9252:         list.add(BB_DURATIONCURVE);
mschaefer@9252:         list.add(BB_DURATIONCURVE_KM);
mschaefer@9252:         return list;
mschaefer@9252:     }
mschaefer@9252: 
mschaefer@9252: 
mschaefer@9252:     /**
mschaefer@9252:      * Can provide whatever getData returns and additionally the location.
mschaefer@9252:      * @param key      will respond on BB_DURATIONCURVE +KM
mschaefer@9252:      * @param param    ignored
mschaefer@9252:      * @param context  ignored
mschaefer@9252:      * @return whatever getData delivers or location.
mschaefer@9252:      */
mschaefer@9252:     @Override
mschaefer@9252:     public Object provideBlackboardData(final Artifact artifact,
mschaefer@9252:             final Object key,
mschaefer@9252:             final Object param,
mschaefer@9252:             final CallContext context
mschaefer@9252:             ) {
mschaefer@9252:         if (key.equals(BB_DURATIONCURVE)) {
mschaefer@9252:             return getData(artifact, context);
mschaefer@9252:         }
mschaefer@9252:         else if (key.equals(BB_DURATIONCURVE_KM)) {
mschaefer@9252:             return ((D4EArtifact)artifact).getDataAsString("ld_locations");
mschaefer@9252:         }
mschaefer@9252:         else {
mschaefer@9252:             return null;
mschaefer@9252:         }
mschaefer@9252:     }
mschaefer@9252: 
mschaefer@9252: 
mschaefer@9252:     /** Create a deep copy. */
mschaefer@9252:     @Override
mschaefer@9252:     public Facet deepCopy() {
mschaefer@9252:         final FloodDurationCurveFacet copy = new FloodDurationCurveFacet();
mschaefer@9252:         copy.set(this);
mschaefer@9252:         return copy;
mschaefer@9252:     }
mschaefer@9252: }
mschaefer@9252: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :