ingo@361: package de.intevation.flys.artifacts.model;
ingo@361: 
ingo@365: import gnu.trove.TDoubleArrayList;
ingo@361: 
ingo@402: import org.apache.log4j.Logger;
ingo@402: 
ingo@361: 
ingo@361: /**
ingo@361:  * This class represents a pool of data triples that consists of 'W', 'Q' and
ingo@361:  * 'KM' data.
ingo@361:  *
ingo@361:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@361:  */
ingo@447: public class WQKms extends NamedObject {
ingo@361: 
ingo@402:     private static Logger logger = Logger.getLogger(WQKms.class);
ingo@402: 
ingo@402: 
ingo@361:     /** The array that contains the 'W' values.*/
ingo@365:     protected TDoubleArrayList w;
ingo@361: 
ingo@361:     /** The array that contains the 'Q' values.*/
ingo@365:     protected TDoubleArrayList q;
ingo@361: 
ingo@361:     /** The array that contains the 'KMs' values.*/
ingo@365:     protected TDoubleArrayList kms;
ingo@361: 
ingo@361: 
ingo@361:     public WQKms() {
ingo@447:         this("");
ingo@447:     }
ingo@447: 
ingo@447: 
ingo@447:     public WQKms(String name) {
ingo@447:         super(name);
ingo@447: 
ingo@365:         this.w   = new TDoubleArrayList();
ingo@365:         this.q   = new TDoubleArrayList();
ingo@365:         this.kms = new TDoubleArrayList();
ingo@361:     }
ingo@361: 
ingo@447: 
sascha@380:     public WQKms(int capacity) {
ingo@447:         this(capacity, "");
ingo@447:     }
ingo@447: 
ingo@447: 
ingo@447:     public WQKms(int capacity, String name) {
ingo@447:         super(name);
ingo@447: 
sascha@380:         this.w   = new TDoubleArrayList(capacity);
sascha@380:         this.q   = new TDoubleArrayList(capacity);
sascha@380:         this.kms = new TDoubleArrayList(capacity);
sascha@380:     }
sascha@380: 
sascha@675:     public static void removeNaNs(TDoubleArrayList [] arrays) {
sascha@675: 
sascha@675:         int dest = 0;
sascha@675: 
sascha@675:         int A = arrays.length;
sascha@675:         int N = arrays[0].size();
sascha@675: 
sascha@675:         OUTER: for (int i = 0; i < N; ++i) {
sascha@675:             for (int j = 0; j < A; ++j) {
sascha@675:                 TDoubleArrayList a = arrays[j];
sascha@675:                 double v = a.getQuick(i);
sascha@675:                 if (Double.isNaN(v)) {
sascha@675:                     continue OUTER;
sascha@675:                 }
sascha@675:                 a.setQuick(dest, v);
sascha@675:             }
sascha@675:             ++dest;
sascha@675:         }
sascha@675: 
sascha@675:         if (dest < N) {
sascha@675:             for (int i = 0; i < A; ++i) {
sascha@675:                 arrays[i].remove(dest, N-dest);
sascha@675:             }
sascha@675:         }
sascha@675:     }
sascha@675: 
sascha@675:     public void removeNaNs() {
sascha@675:         removeNaNs(new TDoubleArrayList [] { w, q, kms });
sascha@675:     }
sascha@675: 
ingo@361: 
ingo@402:     public WQKms(double[] kms, double[] qs, double[] ws) {
ingo@447:         this(kms, qs, ws, "");
ingo@447:     }
ingo@447: 
ingo@447: 
ingo@447:     public WQKms(double[] kms, double[] qs, double[] ws, String name) {
ingo@447:         super(name);
ingo@447: 
ingo@402:         this.w   = new TDoubleArrayList(ws);
ingo@402:         this.q   = new TDoubleArrayList(qs);
ingo@402:         this.kms = new TDoubleArrayList(kms);
ingo@402:     }
ingo@402: 
ingo@402: 
ingo@361:     /**
ingo@361:      * Adds a new row to this data pool.
ingo@361:      *
ingo@361:      * @param w a W.
ingo@361:      * @param q a Q.
ingo@361:      * @param kms a Kms.
ingo@361:      */
ingo@361:     public void add(double w, double q, double kms) {
ingo@365:         this.w.add(w);
ingo@365:         this.q.add(q);
ingo@365:         this.kms.add(kms);
ingo@361:     }
ingo@361: 
ingo@361:     /**
ingo@367:      * Returns the number of triples stored in this data pool.
ingo@361:      *
ingo@367:      * @return the number of triples stored in this data pool.
ingo@361:      */
ingo@367:     public int size() {
ingo@367:         return kms.size();
sascha@366:     }
sascha@366: 
sascha@366:     /**
sascha@366:      * This method returns a triple of W, Q and Kms in a single 3dim array.
sascha@366:      *
sascha@366:      * @param idx The position of the triple.
sascha@366:      * @param dst destination array
sascha@366:      *
sascha@366:      * @return a triple of [W, Q, Kms] in dst.
sascha@366:      */
sascha@366:     public double[] get(int idx, double [] dst) {
sascha@675:         dst[0] = w  .getQuick(idx);
sascha@675:         dst[1] = q  .getQuick(idx);
sascha@675:         dst[2] = kms.getQuick(idx);
sascha@366:         return dst;
ingo@361:     }
ingo@367: 
ingo@367: 
ingo@367:     public double getKms(int idx) {
sascha@675:         return kms.getQuick(idx);
ingo@367:     }
ingo@402: 
ingo@402: 
ingo@402:     public double[] getKms() {
ingo@402:         return kms.toNativeArray();
ingo@402:     }
ingo@402: 
ingo@402: 
ingo@402:     public double[] getWs() {
ingo@402:         return w.toNativeArray();
ingo@402:     }
ingo@402: 
ingo@402: 
ingo@402:     public double[] getQs() {
ingo@402:         return q.toNativeArray();
ingo@402:     }
ingo@402: 
ingo@402: 
ingo@402:     /**
ingo@402:      * Returns a string that consist of the first and last kilometer.
ingo@402:      *
ingo@402:      * @return a string that consist of the first and last kilometer.
ingo@402:      */
ingo@402:     public String toString() {
ingo@402:         double from = getKms(0);
ingo@402:         double to   = getKms(size()-1);
sascha@678:         return from + " - " + to;
ingo@402:     }
ingo@361: }
ingo@361: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :