view flys-artifacts/src/main/java/de/intevation/flys/artifacts/model/WaterlevelFacet.java @ 4241:49cb65d5932d

Improved the historical discharge calculation. The calculation now creates new HistoricalWQKms (new subclass of WQKms). Those WQKms are used to create new facets from (new) type 'HistoricalDischargeCurveFacet'. The chart generator is improved to support those facets.
author Ingo Weinzierl <ingo.weinzierl@intevation.de>
date Wed, 24 Oct 2012 14:34:35 +0200
parents de6e2b933f33
children d65cf8e40230
line wrap: on
line source
package de.intevation.flys.artifacts.model;

import de.intevation.artifactdatabase.state.Facet;
import de.intevation.artifacts.Artifact;
import de.intevation.artifacts.CallContext;
import de.intevation.flys.artifacts.FLYSArtifact;
import de.intevation.flys.artifacts.math.Linear;
import de.intevation.flys.artifacts.states.DefaultState.ComputeType;

import org.apache.log4j.Logger;

/**
 * Facet of a Waterlevel (WQKms).
 */
public class WaterlevelFacet extends DataFacet {

    private static Logger logger = Logger.getLogger(WaterlevelFacet.class);

    public WaterlevelFacet(int index, String name, String description) {
        super(index, name, description, ComputeType.ADVANCE, null, null);
    }

    public WaterlevelFacet(
        int         index,
        String      name,
        String      description,
        ComputeType type,
        String      stateID,
        String      hash
    ) {
        super(index, name, description, type, hash, stateID);
    }

    public WaterlevelFacet() {
    }

    protected WQKms [] getWQKms(CalculationResult res) {
        return (WQKms [])res.getData();
    }

    /**
     * Get waterlevel data.
     * @return a WQKms at given index.
     */
    @Override
    public Object getData(Artifact artifact, CallContext context) {

        if (logger.isDebugEnabled()) {
            logger.debug("Get data for waterlevels at index: " + index +
                " /stateId: " + stateId);
        }

        if (artifact == null) {
            logger.error("WaterlevelFacet.getData: artifact is null");
            return null;
        }

        FLYSArtifact winfo = (FLYSArtifact) artifact;

        CalculationResult res = (CalculationResult)
            winfo.compute(context, hash, stateId, type, false);

        if (res == null) {
            logger.error("WaterlevelFacet.getData: null result");
            return null;
        }

        WQKms [] wqkms = getWQKms(res);
        Object KM = context.getContextValue("currentKm");
        if (KM != null) {
            logger.debug("interpolate at given km");
            // TODO handle exact match.

            WQKms wqkmsI = wqkms[index];
            double km = (Double)KM;

            // TODO employ DataUtils interface to TDoubleArraList
            int size = wqkmsI.size();
            boolean kmIncreasing = wqkmsI.getKm(0) < wqkmsI.getKm(size-1);
            int mod = kmIncreasing ? +1 : -1;
            int idx = 0;
            if (!kmIncreasing) {
                while (idx < size && wqkmsI.getKm(idx) < km) {
                    idx++;
                }
            }
            else {
                idx = size-1;
                while (idx > 0 && wqkmsI.getKm(idx) > km) {
                    idx--;
                }
            }

            WQKms resultWQKms = new WQKms();
            if ((idx != -1) && (idx < size) && (idx - mod != -1) && (idx - mod < size)) {
                double inW = Linear.linear(
                    km,
                    wqkmsI.getKm(idx), wqkmsI.getKm(idx - mod),
                    wqkmsI.getW(idx), wqkmsI.getW(idx - mod));
                double inQ = Linear.linear(
                    km,
                    wqkmsI.getKm(idx), wqkmsI.getKm(idx - mod),
                    wqkmsI.getQ(idx), wqkmsI.getQ(idx - mod));
                resultWQKms.add(inW, inQ, km);
            }

            return resultWQKms;
        }

        return wqkms != null ? wqkms[index] : null;
    }


    /** Copy deeply. */
    @Override
    public Facet deepCopy() {
        WaterlevelFacet copy = new WaterlevelFacet();
        copy.set(this);
        copy.type    = type;
        copy.hash    = hash;
        copy.stateId = stateId;
        return copy;
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

http://dive4elements.wald.intevation.org