Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/artifacts/model/WQDay.java @ 8098:09725b65955a
Add new and simplyfied SedimentLoadFacet
The SedimentLoadFacet is intended to work with the Measurement stations.
It uses the same mechanismn to access the Mesurement station values
as the calculation does.
SedimentLoadLS values need a different facet that will come soon.
author | Andre Heinecke <andre.heinecke@intevation.de> |
---|---|
date | Fri, 15 Aug 2014 18:27:19 +0200 |
parents | 4de4b19b6be6 |
children | 5e38e2924c07 |
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 java.util.ArrayList; import java.util.Collections; import java.util.Comparator; import org.dive4elements.river.artifacts.math.Linear; import gnu.trove.TIntArrayList; /** * This class represents a pool of data triples that consists of 'W', 'Q' and * 'Day' data. * * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> */ public class WQDay extends WQ { public static final Comparator<double []> FIRST_CMP = new Comparator<double []>() { @Override public int compare(double [] a, double [] b) { double diff = a[0] - b[0]; if (diff < 0d) return -1; if (diff > 0d) return +1; return 0; } }; public static final double EPSILON = 1e-4; protected TIntArrayList days; public WQDay() { super(""); days = new TIntArrayList(); } public WQDay(int capacity) { super(capacity); days = new TIntArrayList(capacity); } public WQDay(int [] days, double [] ws, double [] qs) { super(qs, ws, ""); this.days = new TIntArrayList(days); } public void add(int day, double w, double q) { super.add(w, q); days.add(day); } public boolean isIncreasing() { return size() == 0 || getDay(0) < getDay(size()-1); } public int getDay(int idx) { return days.getQuick(idx); } private static final Double interpolateX(ArrayList<double []> dxs, double x) { Collections.sort(dxs, FIRST_CMP); if (Math.abs(x - dxs.get(0)[1]) < EPSILON) { return dxs.get(0)[0]; } for (int i = 1, S = dxs.size(); i < S; ++i) { double [] curr = dxs.get(i); if (Math.abs(x - curr[1]) < EPSILON) { return curr[0]; } double [] prev = dxs.get(i-1); double x1 = Math.min(prev[1], curr[1]); double x2 = Math.max(prev[1], curr[1]); if (x > x1 && x < x2) { return Linear.linear( x, prev[1], curr[1], prev[0], curr[0]); } } return null; } public Double interpolateDayByW(double w) { int S = days.size(); if (S == 0) { return null; } ArrayList<double[]> dws = new ArrayList<double[]>(S); for (int i = 0; i < S; ++i) { dws.add(new double[] { getDay(i), getW(i) }); } return interpolateX(dws, w); } public Double interpolateDayByQ(double q) { int S = days.size(); if (S == 0) { return null; } ArrayList<double[]> dqs = new ArrayList<double[]>(S); for (int i = 0; i < S; ++i) { dqs.add(new double[] { getDay(i), getQ(i) }); } return interpolateX(dqs, q); } @Override public void removeNaNs() { int dest = 0; int N = ws.size(); for (int i = 0; i < N; ++i) { double wi = ws.getQuick(i); double qi = qs.getQuick(i); if (Double.isNaN(wi) || Double.isNaN(qi)) { continue; } days.setQuick(dest, days.getQuick(i)); ws.setQuick(dest, wi); qs.setQuick(dest, qi); ++dest; } if (dest < N) { days.remove(dest, N-dest); ws .remove(dest, N-dest); qs .remove(dest, N-dest); } } } // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :