view artifacts/src/main/java/org/dive4elements/river/artifacts/model/FlowVelocityFilterFacet.java @ 9425:3f49835a00c3

Extended CrossSectionFacet so it may fetch different data from within the artifact result. Also allows to have acces to the potentially already computed artifact result via its normal computation cache.
author gernotbelger
date Fri, 17 Aug 2018 15:31:02 +0200
parents 5e38e2924c07
children 0a5239a1e46e
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.model;

import org.dive4elements.artifactdatabase.state.Facet;

import org.dive4elements.artifacts.Artifact;
import org.dive4elements.artifacts.CallContext;

import org.dive4elements.river.artifacts.D4EArtifact;

import org.dive4elements.river.artifacts.access.RiverAccess;
import org.dive4elements.river.artifacts.context.RiverContext;

import org.dive4elements.river.artifacts.math.MovingAverage;
import org.dive4elements.river.artifacts.states.DefaultState.ComputeType;

import org.apache.log4j.Logger;

/**
 * Facet of a FlowVelocity curve.
 */
public class FlowVelocityFilterFacet extends DataFacet {

    private static Logger log = Logger.getLogger(
        FlowVelocityFilterFacet.class);

    public FlowVelocityFilterFacet() {
        // required for clone operation deepCopy()
    }


    public FlowVelocityFilterFacet(
        int         idx,
        String      name,
        String      description,
        ComputeType type,
        String      stateId,
        String      hash
    ) {
        super(idx, name, description, type, hash, stateId);
        this.metaData.put("X", "chart.longitudinal.section.xaxis.label");
        this.metaData.put("Y", "chart.flow_velocity.section.yaxis.label");
    }


    public Object getData(Artifact artifact, CallContext context) {
        log.debug("Get data for flow velocity at index: " + index);

        Double start = (Double)context.getContextValue("startkm");
        Double end = (Double)context.getContextValue("endkm");
        D4EArtifact flys = (D4EArtifact) artifact;

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

        FlowVelocityData[] data = (FlowVelocityData[]) res.getData();
        if(start != null && end != null) {
            RiverContext fc = (RiverContext)context.globalContext();
            ZoomScale scales = (ZoomScale)fc.get("zoomscale");
            RiverAccess access = new RiverAccess((D4EArtifact)artifact);
            String river = access.getRiverName();

            double radius = scales.getRadius(river, start, end);
            FlowVelocityData oldData = data[index];
            FlowVelocityData newData = new FlowVelocityData();
            double[][] q = oldData.getQPoints();
            double[][] totalV = MovingAverage.weighted(
                oldData.getTotalChannelPoints(), radius);
            double[][] mainV = MovingAverage.weighted(
                oldData.getMainChannelPoints(), radius);
            double[][] tau = MovingAverage.weighted(
                oldData.getTauPoints(), radius);
            for(int j = 0; j < q[0].length; j++) {
                newData.addKM(q[0][j]);
                newData.addQ(q[1][j]);
                newData.addTauMain(tau[1][j]);
                newData.addVMain(mainV[1][j]);
                newData.addVTotal(totalV[1][j]);
            }
            return newData;
        }
        return data[index];
    }


    /** Copy deeply. */
    @Override
    public Facet deepCopy() {
        FlowVelocityFilterFacet copy = new FlowVelocityFilterFacet();
        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