/* 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 logger = 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);
    }


    public Object getData(Artifact artifact, CallContext context) {
        logger.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.getRiver();

            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 :