/* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
 * Software engineering by
 *  Björnsen Beratende Ingenieure GmbH
 *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
 *
 * This file is Free Software under the GNU AGPL (>=v3)
 * and comes with ABSOLUTELY NO WARRANTY! Check out the
 * documentation coming with Dive4Elements River for details.
 */

package org.dive4elements.river.model.sinfo;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;
import javax.persistence.JoinColumn;
import javax.persistence.OneToMany;
import javax.persistence.OneToOne;
import javax.persistence.SequenceGenerator;
import javax.persistence.Table;

import org.dive4elements.river.backend.SessionHolder;
import org.dive4elements.river.model.River;
import org.hibernate.Query;
import org.hibernate.Session;

/**
 * Hibernate binding for the DB table collision
 *
 * @author Matthias Schäfer
 *
 */

@Entity
@Table(name = "collision")
public class Collision implements Serializable {

    /***** FIELDS *****/

    private static final long serialVersionUID = -6950745885816251322L;

    private Integer id;

    private River river;

    private String filename;

    private String kmrange_info;

    private String notes;

    private Integer year;

    private List<CollisionValue> values;


    /***** CONSTRUCTORS *****/

    public Collision() {
    }


    public Collision(final River river, final String filename, final String kmrange_info, final String notes, final Integer year) {
        this.river = river;
        this.filename = filename;
        this.kmrange_info = kmrange_info;
        this.notes = notes;
        this.year = year;
        this.values = new ArrayList<>();
    }

    /***** METHODS *****/

    @Id
    @SequenceGenerator(name = "SEQUENCE_COLLISION_ID_SEQ", sequenceName = "COLLISION_ID_SEQ", allocationSize = 1)
    @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "SEQUENCE_COLLISION_ID_SEQ")
    @Column(name = "id")
    public Integer getId() {
        return this.id;
    }

    public void setId(final Integer id) {
        this.id = id;
    }

    @OneToOne
    @JoinColumn(name = "river_id")
    public River getRiver() {
        return this.river;
    }

    public void setRiver(final River river) {
        this.river = river;
    }

    @Column(name = "filename")
    public String getFilename() {
        return this.filename;
    }

    public void setFilename(final String filename) {
        this.filename = filename;
    }

    @Column(name = "kmrange_info")
    public String getKmrange_info() {
        return this.kmrange_info;
    }

    public void setKmrange_info(final String kmrange_info) {
        this.kmrange_info = kmrange_info;
    }

    @Column(name = "notes")
    public String getNotes() {
        return this.notes;
    }

    public void setNotes(final String notes) {
        this.notes = notes;
    }

    @Column(name = "year")
    public Integer getYear() {
        return this.year;
    }

    public void setYear(final Integer year) {
        this.year = year;
    }

    @OneToMany
    @JoinColumn(name = "collision_id")
    public List<CollisionValue> getValues() {
        return this.values;
    }

    public void setValues(final List<CollisionValue> values) {
        this.values = values;
    }

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

    /**
     * Get data series by id
     */
    public static Collision getSeries(final int id) {
        final Session session = SessionHolder.HOLDER.get();
        final Query query = session.createQuery("FROM Collision WHERE (id=:id)");
        query.setParameter("id", id);
        return (Collision) query.list().get(0);
    }

    /**
     * Selects from the database the years of the collisions of a river
     */
    public static List<Integer> getYearsOfRiver(final River river) {
        final Session session = SessionHolder.HOLDER.get();
        final Query query = session.createQuery("SELECT c.year, COUNT(cv) AS count"
                + " FROM Collision c INNER JOIN c.values cv"
                + " WHERE c.river=:river"
                + " GROUP BY c.year"
                + " ORDER BY c.year");
        query.setParameter("river", river);
        final List<Integer> years = new ArrayList<>();
        final List<Object[]> rows = query.list();
        if (rows != null) {
            for (int i = 0; i <= rows.size() - 1; i++) {
                years.add(Integer.valueOf((int) rows.get(i)[0]));
            }
        }
        return years;
    }
}