package de.intevation.flys.artifacts.model;

import gnu.trove.TDoubleArrayList;

import org.apache.log4j.Logger;


/**
 * This class represents a pool of data triples that consists of 'W', 'Q' and
 * 'KM' data.
 *
 * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
 */
public class WQKms extends NamedObject {

    private static Logger logger = Logger.getLogger(WQKms.class);


    /** The array that contains the 'W' values.*/
    protected TDoubleArrayList w;

    /** The array that contains the 'Q' values.*/
    protected TDoubleArrayList q;

    /** The array that contains the 'KMs' values.*/
    protected TDoubleArrayList kms;


    public WQKms() {
        this("");
    }


    public WQKms(String name) {
        super(name);

        this.w   = new TDoubleArrayList();
        this.q   = new TDoubleArrayList();
        this.kms = new TDoubleArrayList();
    }


    public WQKms(int capacity) {
        this(capacity, "");
    }


    public WQKms(int capacity, String name) {
        super(name);

        this.w   = new TDoubleArrayList(capacity);
        this.q   = new TDoubleArrayList(capacity);
        this.kms = new TDoubleArrayList(capacity);
    }


    public WQKms(double[] kms, double[] qs, double[] ws) {
        this(kms, qs, ws, "");
    }


    public WQKms(double[] kms, double[] qs, double[] ws, String name) {
        super(name);

        this.w   = new TDoubleArrayList(ws);
        this.q   = new TDoubleArrayList(qs);
        this.kms = new TDoubleArrayList(kms);
    }


    /**
     * Adds a new row to this data pool.
     *
     * @param w a W.
     * @param q a Q.
     * @param kms a Kms.
     */
    public void add(double w, double q, double kms) {
        this.w.add(w);
        this.q.add(q);
        this.kms.add(kms);
    }


    public void add(double[] w, double[] q, double[] kms) {
        this.w.add(w);
        this.q.add(q);
        this.kms.add(kms);
    }


    /**
     * Returns the number of triples stored in this data pool.
     *
     * @return the number of triples stored in this data pool.
     */
    public int size() {
        return kms.size();
    }

    /**
     * This method returns a triple of W, Q and Kms in a single 3dim array.
     *
     * @param idx The position of the triple.
     * @param dst destination array
     *
     * @return a triple of [W, Q, Kms] in dst.
     */
    public double[] get(int idx, double [] dst) {
        dst[0] = w  .get(idx);
        dst[1] = q  .get(idx);
        dst[2] = kms.get(idx);
        return dst;
    }


    public double getKms(int idx) {
        return kms.get(idx);
    }


    public double[] getKms() {
        return kms.toNativeArray();
    }


    public double[] getWs() {
        return w.toNativeArray();
    }


    public double[] getQs() {
        return q.toNativeArray();
    }


    /**
     * Returns a string that consist of the first and last kilometer.
     *
     * @return a string that consist of the first and last kilometer.
     */
    public String toString() {
        double from = getKms(0);
        double to   = getKms(size()-1);
        return Double.toString(from) + " - " + Double.toString(to);
    }


    /**
     * Merges the WQKms objects of an incoming 2dim array (table) where the
     * objects of a column belong together.
     *
     * @param toMerge The objects that need to be merged.
     *
     * @return an array of merged WQKms objects.
     */
    public static WQKms[] merge(WQKms[][] toMerge) {
        int num = toMerge[0].length;

        // TODO IS THE LENGTH CORRECT?
        WQKms[] merged = new WQKms[num];
        for (int i = 0; i < num; i++) {
            merged[i] = new WQKms();
        }

        for (int i = 0; i < toMerge.length; i++) {
            WQKms[] tmp = toMerge[i];

            for (int j = 0; j < num; j++) {
                WQKms toAdd = tmp[j];
                merged[j].add(toAdd.getWs(), toAdd.getQs(), toAdd.getKms());
            }
        }

        return merged;
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :