ingo@2311: package de.intevation.flys.artifacts.model;
ingo@2311: 
ingo@2311: import gnu.trove.TDoubleArrayList;
ingo@2311: 
christian@4183: import java.util.ArrayList;
christian@4183: import java.util.Collections;
christian@4183: import java.util.List;
christian@4183: 
ingo@2311: 
ingo@2311: /**
ingo@2311:  * A subclass of WQTimerange that stores besides W, Q and Timerange values
ingo@2311:  * another double value.
ingo@2311:  *
ingo@2311:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@2311:  */
ingo@2311: public class HistoricalWQTimerange extends WQTimerange {
ingo@2311: 
christian@4183:     public static class HistoricalTimerangeItem extends TimerangeItem {
christian@4183:         public double diff;
christian@4183: 
christian@4183:         public HistoricalTimerangeItem (Timerange timerange, double q, double w, double diff) {
christian@4183:             super(timerange, q, w);
christian@4183:             this.diff = diff;
christian@4183:         }
christian@4183: 
christian@4183:         public double[] get(double[] wq) {
christian@4183:             if (wq.length >= 3) {
christian@4183:                 wq[0] = w;
christian@4183:                 wq[1] = q;
christian@4183:                 wq[2] = diff;
christian@4183:             }
christian@4183:             else if (wq.length >= 2) {
christian@4183:                 return super.get(wq);
christian@4183:             }
christian@4183: 
christian@4183:             return wq;
christian@4183:         }
christian@4183:     }
christian@4183: 
ingo@2311:     protected TDoubleArrayList diffs;
ingo@2311: 
ingo@2311: 
ingo@2311:     public HistoricalWQTimerange(String name) {
ingo@2311:         super(name);
ingo@2311: 
ingo@2311:         diffs = new TDoubleArrayList();
ingo@2311:     }
ingo@2311: 
ingo@2311: 
ingo@2311:     public void add(double w, double q, double diff, Timerange t) {
ingo@2311:         ws.add(w);
ingo@2311:         qs.add(q);
ingo@2311:         ts.add(t);
ingo@2311:         diffs.add(diff);
ingo@2311:     }
ingo@2311: 
ingo@2311: 
ingo@2311:     /**
ingo@2311:      * This method requires a 3dim double array for <i>res</i>!
ingo@2311:      */
ingo@2311:     @Override
ingo@2311:     public double[] get(int idx, double[] res) {
ingo@2311:         res[0] = ws.getQuick(idx);
ingo@2311:         res[1] = qs.getQuick(idx);
ingo@2311:         res[2] = diffs.getQuick(idx);
ingo@2311: 
ingo@2311:         return res;
ingo@2311:     }
ingo@2311: 
ingo@2311: 
ingo@2311:     public double[] getDiffs() {
ingo@2311:         return diffs.toNativeArray();
ingo@2311:     }
christian@4183: 
christian@4183:     @Override
christian@4183:     public List<TimerangeItem> sort() {
christian@4183:         ArrayList<TimerangeItem> items = new ArrayList<TimerangeItem>(ts.size());
christian@4183:         for (int i = 0, n = size(); i < n; i++) {
christian@4183:             items.add(new HistoricalTimerangeItem(getTimerange(i), getQ(i), getW(i), diffs.get(i)));
christian@4183:         }
christian@4183: 
christian@4183:         Collections.sort(items);
christian@4183:         return items;
christian@4183:     }
ingo@2311: }
ingo@2311: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :