teichmann@5863: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
teichmann@5863:  * Software engineering by Intevation GmbH
teichmann@5863:  *
teichmann@5994:  * This file is Free Software under the GNU AGPL (>=v3)
teichmann@5863:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
teichmann@5994:  * documentation coming with Dive4Elements River for details.
teichmann@5863:  */
teichmann@5863: 
teichmann@5831: package org.dive4elements.river.artifacts.model;
felix@1085: 
ingo@1679: import java.util.ArrayList;
ingo@1679: import java.util.List;
ingo@1679: 
felix@2775: import org.apache.log4j.Logger;
felix@2775: 
teichmann@5831: import org.dive4elements.artifacts.Artifact;
teichmann@5831: import org.dive4elements.artifacts.CallContext;
teichmann@5831: import org.dive4elements.artifacts.DataProvider;
ingo@1679: 
teichmann@5831: import org.dive4elements.artifactdatabase.state.DefaultFacet;
ingo@1679: 
teichmann@5831: import org.dive4elements.river.artifacts.MainValuesArtifact;
teichmann@5864: import org.dive4elements.river.jfree.RiverAnnotation;
teichmann@5831: import org.dive4elements.river.jfree.StickyAxisAnnotation;
felix@6525: import org.dive4elements.river.exports.fixings.FixChartGenerator;
felix@1085: 
tom@8331: import static org.dive4elements.river.exports.injector.InjectorConstants.PNP;
felix@2161: 
felix@1085: /**
felix@1085:  * Facet to show Main W Values.
felix@1085:  */
felix@1085: public class MainValuesWFacet
felix@1085: extends      DefaultFacet
felix@1085: implements   FacetTypes {
felix@1085: 
teichmann@8202:     /** Own log. */
teichmann@8202:     private static Logger log = Logger.getLogger(MainValuesWFacet.class);
felix@2775: 
felix@1085:     /** Trivial Constructor. */
tom@8331:     public MainValuesWFacet(String name, String description) {
felix@1112:         this.description = description;
felix@1809:         this.name = name;
felix@1957:         this.index = 0;
felix@1085:     }
felix@1085: 
felix@1085: 
felix@1085:     /**
felix@2775:      * Set the hit-point in W where a line drawn from the axis would hit the
felix@2775:      * curve in WQDay (if hit).
felix@2775:      * Employ linear interpolation.
felix@2775:      */
tom@8856:     protected static void setHitPoint(
tom@8856:         WQDay wqday,
tom@8856:         StickyAxisAnnotation annotation
tom@8856:     ) {
teichmann@7494:         float w = annotation.getPos();
teichmann@7483: 
teichmann@7494:         Double day = wqday.interpolateDayByW(w);
teichmann@7494: 
teichmann@7494:         if (day != null) {
teichmann@7494:             annotation.setHitPoint(day.floatValue());
felix@2775:         }
teichmann@8202:         else if (log.isDebugEnabled()) {
teichmann@8202:             log.debug("StickyAnnotation does not hit wqday curve: " + w);
felix@2775:         }
felix@2775:     }
felix@2775: 
felix@2775: 
felix@2775:     /**
felix@4235:      * Returns the data this facet provides.
felix@1085:      *
felix@1085:      * @param artifact the owner artifact.
felix@6574:      * @param context  the CallContext (can be used to find out if in
felix@6574:      *                 navigable fixation-setting, or durationcurve).
felix@1085:      *
felix@1085:      * @return the data.
felix@1085:      */
felix@1085:     @Override
felix@1085:     public Object getData(Artifact artifact, CallContext context) {
felix@1085:         MainValuesArtifact mvArtifact = (MainValuesArtifact) artifact;
ingo@1679: 
tom@8331:         List<NamedDouble> ws = mvArtifact.getMainValuesW(
tom@8331:             context.getContextValue(PNP));
felix@2161:         List<StickyAxisAnnotation> xy = new ArrayList<StickyAxisAnnotation>();
ingo@1679: 
felix@2780:         // Find whether a duration curve is on the blackboard.
felix@2775:         WQDay wqdays = null;
felix@2775:         List<DataProvider> providers = context.
felix@2775:             getDataProvider(DurationCurveFacet.BB_DURATIONCURVE);
felix@2775:         if (providers.size() < 1) {
felix@6525:             // Do we have a current km in context?
felix@6525:             // If so, we are likely fetching data for a navigable
felix@6525:             // diagram (i.e. in fixation branch).
teichmann@7489:             Object xkm = context.getContextValue(FixChartGenerator.CURRENT_KM);
teichmann@7489:             if (xkm != null) {
teichmann@7489:                 Double ckm = (Double)xkm;
tom@8331:                 // Return linearly interpolated values. Always in m, as
tom@8331:                 // cm over datum ist represented by a second axis.
tom@8331:                 ws = mvArtifact.getMainValuesW(
tom@8331:                     new double[] {ckm},
tom@8331:                     context.getContextValue(PNP)
tom@8331:                 );
felix@6525:             }
felix@2775:         }
felix@2775:         else {
felix@2775:             wqdays = (WQDay) providers.get(0).provideData(
felix@2775:                 DurationCurveFacet.BB_DURATIONCURVE,
felix@2775:                 null,
felix@2775:                 context);
felix@2775:         }
felix@2775: 
ingo@1679:         for (NamedDouble w: ws) {
felix@6695:             if (Double.isNaN(w.getValue())) {
teichmann@8202:                 log.warn("NaN MainValue " + w.getName());
felix@6695:                 continue;
felix@6695:             }
felix@2775:             StickyAxisAnnotation annotation =
felix@2775:                 new StickyAxisAnnotation(
felix@2775:                     w.getName(),
felix@2775:                     (float) w.getValue(),
felix@2775:                     StickyAxisAnnotation.SimpleAxis.Y_AXIS);
felix@2775:             xy.add(annotation);
felix@2775:             if (wqdays != null) {
felix@2775:                 setHitPoint(wqdays, annotation);
felix@2775:             }
ingo@1679:         }
ingo@1679: 
teichmann@5864:         return new RiverAnnotation(description, xy);
felix@1085:     }
felix@1085: 
felix@1085: 
felix@1085:     /**
felix@1085:      * Create a deep copy of this Facet.
felix@1085:      * @return a deep copy.
felix@1085:      */
felix@1085:     @Override
felix@1085:     public MainValuesWFacet deepCopy() {
felix@1957:         MainValuesWFacet copy = new MainValuesWFacet(this.name,
tom@8331:             description);
felix@1085:         copy.set(this);
felix@1085:         return copy;
felix@1085:     }
felix@1085: }
felix@1809: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :