package de.intevation.flys.importer;

import java.util.Iterator;
import java.util.Map;
import java.util.HashMap;

import java.math.BigDecimal;

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

import de.intevation.flys.backend.SessionFactoryProvider;

import de.intevation.flys.model.WstColumnValue;
import de.intevation.flys.model.WstColumn;
import de.intevation.flys.model.DischargeTableValue;
import de.intevation.flys.model.DischargeTable;

public class ImporterSession
{
    private static final ThreadLocal<ImporterSession> SESSION =
        new ThreadLocal<ImporterSession>() {
            @Override
            protected ImporterSession initialValue() {
                return new ImporterSession();
            }
        };

    protected Session databaseSession;

    protected Map<IdValueKey, WstColumnValue> wstColumnValues;

    protected Map<IdValueKey, DischargeTableValue> dischargeTableValues;

    public static ImporterSession getInstance() {
        return SESSION.get();
    }

    public ImporterSession() {
        SessionFactory sessionFactory =
            SessionFactoryProvider.createSessionFactory();
        databaseSession = sessionFactory.openSession();
    }

    public Session getDatabaseSession() {
        return databaseSession;
    }

    public WstColumnValue getWstColumnValue(
        WstColumn  column,
        BigDecimal position,
        BigDecimal w
    ) {
        if (wstColumnValues == null) {
            loadWstColumnValues();
        }

        IdValueKey key = new IdValueKey(
            column.getId(),
            position.doubleValue(),
            w.doubleValue());

        WstColumnValue wcv = wstColumnValues.get(key);

        if (wcv != null) {
            return wcv;
        }

        wcv = new WstColumnValue(column, position, w);

        databaseSession.save(wcv);

        wstColumnValues.put(key, wcv);

        return wcv;
    }

    protected void loadWstColumnValues() {
        wstColumnValues = new HashMap<IdValueKey, WstColumnValue>();

        Query query = databaseSession.createQuery("from WstColumnValue");

        for (Iterator iter = query.iterate(); iter.hasNext();) {
            WstColumnValue wcv = (WstColumnValue)iter.next();
            wstColumnValues.put(new IdValueKey(wcv), wcv);
        }
    }

    public DischargeTableValue getDischargeTableValue(
        DischargeTable table,
        BigDecimal     q,
        BigDecimal     w
    ) {
        if (dischargeTableValues == null) {
            loadDischargeTableValues();
        }

        IdValueKey key = new IdValueKey(
            table.getId(),
            q.doubleValue(),
            w.doubleValue());

        DischargeTableValue dtv = dischargeTableValues.get(key);

        if (dtv != null) {
            return dtv;
        }

        dtv = new DischargeTableValue(table, q, w);

        databaseSession.save(dtv);

        dischargeTableValues.put(key, dtv);

        return dtv;
    }

    protected void loadDischargeTableValues() {
        dischargeTableValues = new HashMap<IdValueKey, DischargeTableValue>();

        Query query = databaseSession.createQuery("from DischargeTableValue");

        for (Iterator iter = query.iterate(); iter.hasNext();) {
            DischargeTableValue dtv = (DischargeTableValue)iter.next();
            dischargeTableValues.put(new IdValueKey(dtv), dtv);
        }
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :