mschaefer@8971: /* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
mschaefer@8971:  * Software engineering by
mschaefer@8971:  *  Björnsen Beratende Ingenieure GmbH
mschaefer@8971:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
mschaefer@8971:  *
mschaefer@8971:  * This file is Free Software under the GNU AGPL (>=v3)
mschaefer@8971:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
mschaefer@8971:  * documentation coming with Dive4Elements River for details.
mschaefer@8971:  */
mschaefer@8971: 
mschaefer@8971: package org.dive4elements.river.model.sinfo;
mschaefer@8971: 
mschaefer@8971: import java.io.Serializable;
mschaefer@8971: import java.util.ArrayList;
mschaefer@8971: import java.util.List;
mschaefer@8971: 
mschaefer@8971: import javax.persistence.Column;
mschaefer@8971: import javax.persistence.Entity;
mschaefer@8971: import javax.persistence.GeneratedValue;
mschaefer@8971: import javax.persistence.GenerationType;
mschaefer@8971: import javax.persistence.Id;
mschaefer@8971: import javax.persistence.JoinColumn;
mschaefer@8971: import javax.persistence.OneToMany;
mschaefer@8971: import javax.persistence.OneToOne;
mschaefer@8971: import javax.persistence.SequenceGenerator;
mschaefer@8971: import javax.persistence.Table;
mschaefer@8971: 
mschaefer@9117: import org.dive4elements.river.backend.SessionHolder;
mschaefer@8971: import org.dive4elements.river.model.River;
mschaefer@9117: import org.hibernate.Query;
mschaefer@9117: import org.hibernate.Session;
mschaefer@8971: 
mschaefer@8971: /**
mschaefer@8971:  * Hibernate binding for the DB table collision
mschaefer@8971:  *
mschaefer@8971:  * @author Matthias Schäfer
mschaefer@8971:  *
mschaefer@8971:  */
mschaefer@8971: 
mschaefer@8971: @Entity
mschaefer@8971: @Table(name = "collision")
mschaefer@8971: public class Collision implements Serializable {
mschaefer@8971: 
mschaefer@8971:     /***** FIELDS *****/
mschaefer@8971: 
mschaefer@8971:     private static final long serialVersionUID = -6950745885816251322L;
mschaefer@8971: 
mschaefer@8971:     private Integer id;
mschaefer@8971: 
mschaefer@8971:     private River river;
mschaefer@8971: 
mschaefer@8971:     private String filename;
mschaefer@8971: 
mschaefer@8971:     private String kmrange_info;
mschaefer@8971: 
mschaefer@9038:     private String notes;
mschaefer@8971: 
mschaefer@8971:     private Integer year;
mschaefer@8971: 
mschaefer@8971:     private List<CollisionValue> values;
mschaefer@8971: 
mschaefer@8971: 
mschaefer@8971:     /***** CONSTRUCTORS *****/
mschaefer@8971: 
mschaefer@8971:     public Collision() {
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971: 
mschaefer@9038:     public Collision(final River river, final String filename, final String kmrange_info, final String notes, final Integer year) {
mschaefer@8971:         this.river = river;
mschaefer@8971:         this.filename = filename;
mschaefer@8971:         this.kmrange_info = kmrange_info;
mschaefer@9038:         this.notes = notes;
mschaefer@8971:         this.year = year;
mschaefer@8971:         this.values = new ArrayList<>();
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     /***** METHODS *****/
mschaefer@8971: 
mschaefer@8971:     @Id
mschaefer@8971:     @SequenceGenerator(name = "SEQUENCE_COLLISION_ID_SEQ", sequenceName = "COLLISION_ID_SEQ", allocationSize = 1)
mschaefer@8971:     @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "SEQUENCE_COLLISION_ID_SEQ")
mschaefer@8971:     @Column(name = "id")
mschaefer@8971:     public Integer getId() {
mschaefer@8971:         return this.id;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setId(final Integer id) {
mschaefer@8971:         this.id = id;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @OneToOne
mschaefer@8971:     @JoinColumn(name = "river_id")
mschaefer@8971:     public River getRiver() {
mschaefer@8971:         return this.river;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setRiver(final River river) {
mschaefer@8971:         this.river = river;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "filename")
mschaefer@8971:     public String getFilename() {
mschaefer@8971:         return this.filename;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setFilename(final String filename) {
mschaefer@8971:         this.filename = filename;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "kmrange_info")
mschaefer@8971:     public String getKmrange_info() {
mschaefer@8971:         return this.kmrange_info;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setKmrange_info(final String kmrange_info) {
mschaefer@8971:         this.kmrange_info = kmrange_info;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@9038:     @Column(name = "notes")
mschaefer@9038:     public String getNotes() {
mschaefer@9038:         return this.notes;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@9038:     public void setNotes(final String notes) {
mschaefer@9038:         this.notes = notes;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "year")
mschaefer@8971:     public Integer getYear() {
mschaefer@8971:         return this.year;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setYear(final Integer year) {
mschaefer@8971:         this.year = year;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @OneToMany
mschaefer@8971:     @JoinColumn(name = "collision_id")
mschaefer@8971:     public List<CollisionValue> getValues() {
mschaefer@8971:         return this.values;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setValues(final List<CollisionValue> values) {
mschaefer@8971:         this.values = values;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void addValue(final CollisionValue value) {
mschaefer@8971:         this.values.add(value);
mschaefer@8971:     }
mschaefer@9117: 
mschaefer@9117:     /**
mschaefer@9117:      * Get data series by id
mschaefer@9117:      */
mschaefer@9117:     public static Collision getSeries(final int id) {
mschaefer@9117:         final Session session = SessionHolder.HOLDER.get();
mschaefer@9117:         final Query query = session.createQuery("FROM Collision WHERE (id=:id)");
mschaefer@9117:         query.setParameter("id", id);
mschaefer@9117:         return (Collision) query.list().get(0);
mschaefer@9117:     }
mschaefer@9168: 
mschaefer@9168:     /**
mschaefer@9168:      * Selects from the database the years of the collisions of a river
mschaefer@9168:      */
mschaefer@9168:     public static List<Integer> getYearsOfRiver(final River river) {
mschaefer@9168:         final Session session = SessionHolder.HOLDER.get();
mschaefer@9168:         final Query query = session.createQuery("SELECT c.year, COUNT(cv) AS count"
mschaefer@9168:                 + " FROM Collision c INNER JOIN c.values cv"
mschaefer@9168:                 + " WHERE c.river=:river"
mschaefer@9168:                 + " GROUP BY c.year"
mschaefer@9168:                 + " ORDER BY c.year");
mschaefer@9168:         query.setParameter("river", river);
mschaefer@9168:         final List<Integer> years = new ArrayList<>();
mschaefer@9168:         final List<Object[]> rows = query.list();
mschaefer@9168:         if (rows != null) {
mschaefer@9168:             for (int i = 0; i <= rows.size() - 1; i++) {
mschaefer@9168:                 years.add(Integer.valueOf((int) rows.get(i)[0]));
mschaefer@9168:             }
mschaefer@9168:         }
mschaefer@9168:         return years;
mschaefer@9168:     }
mschaefer@8971: }