package de.intevation.flys.importer;

import java.util.ArrayList;
import java.util.List;

import org.apache.log4j.Logger;

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

import de.intevation.flys.model.GrainFraction;
import de.intevation.flys.model.River;
import de.intevation.flys.model.SedimentYield;
import de.intevation.flys.model.TimeInterval;
import de.intevation.flys.model.Unit;


public class ImportSedimentYield {

    private static Logger log = Logger.getLogger(ImportSedimentYield.class);

    private ImportGrainFraction grainFraction;

    private ImportUnit unit;

    private ImportTimeInterval timeInterval;

    private List<ImportSedimentYieldValue> values;

    private SedimentYield peer;


    public ImportSedimentYield() {
        this.values = new ArrayList<ImportSedimentYieldValue>();
    }


    public void setTimeInterval(ImportTimeInterval timeInterval) {
        this.timeInterval = timeInterval;
    }


    public void setUnit(ImportUnit unit) {
        this.unit = unit;
    }


    public void setGrainFraction(ImportGrainFraction grainFraction) {
        this.grainFraction = grainFraction;
    }


    public void addValue(ImportSedimentYieldValue value) {
        this.values.add(value);
    }


    public void storeDependencies(River river) {
        log.debug("store dependencies");

        if (grainFraction != null) {
            grainFraction.storeDependencies();
        }

        SedimentYield peer = getPeer(river);

        int i = 0;

        for (ImportSedimentYieldValue value: values) {
            value.storeDependencies(peer);
            i++;
        }

        log.info("stored " + i + " sediment yield values.");
    }


    public SedimentYield getPeer(River river) {
        log.debug("get peer");

        GrainFraction gf = grainFraction != null
            ? grainFraction.getPeer()
            : null;

        Unit u = unit != null ? unit.getPeer() : null;

        TimeInterval ti = timeInterval != null ? timeInterval.getPeer() : null;

        if (ti == null || u == null) {
            log.warn("Skip invalid SedimentYield: time interval or unit null!");
            return null;
        }

        if (peer == null) {
            Session session = ImporterSession.getInstance().getDatabaseSession();
            Query query = session.createQuery(
                "from SedimentYield where " +
                "   river=:river and " +
                "   grainFraction=:grainFraction and " +
                "   unit=:unit and " +
                "   timeInterval=:timeInterval"
            );

            query.setParameter("river", river);
            query.setParameter("grainFraction", gf);
            query.setParameter("unit", u);
            query.setParameter("timeInterval", ti);

            List<SedimentYield> yields = query.list();
            if (yields.isEmpty()) {
                log.debug("create new SedimentYield");

                peer = new SedimentYield(river, u, ti, gf);
                session.save(peer);
            }
            else {
                peer = yields.get(0);
            }
        }

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