package de.intevation.flys.artifacts.model;

import java.util.List;
import java.util.Map;
import java.util.HashMap;
import java.util.Arrays;

import java.io.Serializable;

import org.hibernate.SessionFactory;
import org.hibernate.Session;
import org.hibernate.Query;

import org.apache.log4j.Logger;

import de.intevation.flys.backend.SessionFactoryProvider;

import de.intevation.flys.model.Gauge;
import de.intevation.flys.model.DischargeTable;
import de.intevation.flys.model.DischargeTableValue;

public class DischargeTables
implements   Serializable
{
    private static Logger log = Logger.getLogger(DischargeTables.class);

    protected List<String> gaugeNames;

    protected String riverName;

    protected Map<String, double [][]> values;

    public DischargeTables() {
    }

    public DischargeTables(String riverName, String [] gaugeNames) {
        this(riverName, Arrays.asList(gaugeNames));
    }

    public DischargeTables(String riverName, List<String> gaugeNames) {
        this.riverName  = riverName;
        this.gaugeNames = gaugeNames;
    }

    public Map<String, double [][]> getValues(double scale) {
        if (values == null) {
            values = loadValues(scale);
        }
        return values;
    }

    protected Map<String, double [][]> loadValues(double scale) {
        Map<String, double [][]> values = new HashMap<String, double [][]>();

        SessionFactory sf = SessionFactoryProvider.getSessionFactory();
        Session session = sf.openSession();

        try {
            Query gaugeQuery = session.createQuery(
                "from Gauge where name=:gauge and river.name=:river");
            gaugeQuery.setParameter("river", riverName);

            for (String gaugeName: gaugeNames) {
                gaugeQuery.setParameter("gauge", gaugeName);
                List<Gauge> gauges = gaugeQuery.list();
                if (gauges.isEmpty()) {
                    log.warn(
                        "no gauge '"+gaugeName+"' at river '"+riverName+"'");
                    continue;
                }
                Gauge gauge = gauges.get(0);

                List<DischargeTable> tables = gauge.getDischargeTables();

                if (tables.isEmpty()) {
                    log.warn(
                        "no discharge table for gauge '" + gaugeName + "'");
                    continue;
                }

                // TODO: Filter by time interval
                DischargeTable table = tables.get(0);

                List<DischargeTableValue> dtvs =
                    table.getDischargeTableValues();

                double [][] vs = new double[2][dtvs.size()];

                int idx = 0;
                for (DischargeTableValue dtv: dtvs) {
                    vs[0][idx] = dtv.getQ().doubleValue() * scale;
                    vs[1][idx] = dtv.getW().doubleValue() * scale;
                    ++idx;
                }

                values.put(gaugeName, vs);
            }

            return values;
        }
        finally {
            session.close();
        }
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :