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@9160: import java.util.ArrayList;
mschaefer@8971: import java.util.Date;
mschaefer@9117: 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.OneToOne;
mschaefer@8971: import javax.persistence.SequenceGenerator;
mschaefer@8971: import javax.persistence.Table;
mschaefer@8971: 
mschaefer@9117: import org.dive4elements.river.backend.SessionHolder;
mschaefer@9160: import org.dive4elements.river.model.River;
mschaefer@9117: import org.hibernate.Query;
mschaefer@9160: import org.hibernate.SQLQuery;
mschaefer@9117: import org.hibernate.Session;
mschaefer@9160: import org.hibernate.type.StandardBasicTypes;
mschaefer@9117: 
mschaefer@8971: 
mschaefer@8971: /**
mschaefer@8971:  * Hibernate binding for the DB table collision_values
mschaefer@8971:  *
mschaefer@8971:  * @author Matthias Schäfer
mschaefer@8971:  *
mschaefer@8971:  */
mschaefer@8971: @Entity
mschaefer@8971: @Table(name = "collision_values")
mschaefer@8971: public class CollisionValue implements Serializable {
mschaefer@8971: 
mschaefer@8971:     /***** FIELDS *****/
mschaefer@8971: 
mschaefer@8971:     private static final long serialVersionUID = -1157324854470346513L;
mschaefer@8971: 
mschaefer@8971:     private Integer id;
mschaefer@8971: 
mschaefer@8971:     private Collision collision;
mschaefer@8971: 
mschaefer@8971:     private Double station;
mschaefer@8971: 
mschaefer@8971:     private CollisionType collisionType;
mschaefer@8971: 
mschaefer@8971:     private Date eventDate;
mschaefer@8971: 
mschaefer@8971:     private String gaugeName;
mschaefer@8971: 
mschaefer@8971:     private Double gaugeW;
mschaefer@8971: 
mschaefer@8971: 
mschaefer@8971:     /***** CONSTRUCTORS *****/
mschaefer@8971: 
mschaefer@8971:     public CollisionValue() {
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public CollisionValue(final Collision collision, final Double station, final CollisionType collisionType, final Date eventDate, final String gaugeName,
mschaefer@8971:             final Double gaugeW) {
mschaefer@8971:         this.collision = collision;
mschaefer@8971:         this.station = station;
mschaefer@8971:         this.collisionType = collisionType;
mschaefer@8971:         this.eventDate = eventDate;
mschaefer@8971:         this.gaugeName = gaugeName;
mschaefer@8971:         this.gaugeW = gaugeW;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     /**
mschaefer@8971:      * Constructor with primitive parameter types
mschaefer@8971:      */
mschaefer@8971:     public CollisionValue(final Collision collision, final double km, final CollisionType collisionType, final Date eventDate, final String gaugeName,
mschaefer@8971:             final double gaugeW) {
mschaefer@8971:         this(collision, Double.valueOf(km), collisionType, eventDate, gaugeName, Double.valueOf(gaugeW));
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971: 
mschaefer@8971:     /***** METHODS *****/
mschaefer@8971: 
mschaefer@8971:     @Id
mschaefer@8971:     @SequenceGenerator(name = "SEQUENCE_COLLISION_VALUE_ID_SEQ", sequenceName = "COLLISION_VALUES_ID_SEQ", allocationSize = 1)
mschaefer@8971:     @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "SEQUENCE_COLLISION_VALUE_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 = "collision_id")
mschaefer@8971:     public Collision getCollision() {
mschaefer@8971:         return this.collision;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setCollision(final Collision collision) {
mschaefer@8971:         this.collision = collision;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "station")
mschaefer@8971:     public Double getStation() {
mschaefer@8971:         return this.station;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setStation(final Double station) {
mschaefer@8971:         this.station = station;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @OneToOne
mschaefer@8971:     @JoinColumn(name = "collision_type_id")
mschaefer@8971:     public CollisionType getCollisionType() {
mschaefer@8971:         return this.collisionType;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setCollisionType(final CollisionType collisionType) {
mschaefer@8971:         this.collisionType = collisionType;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "event_date")
mschaefer@8971:     public Date getEventDate() {
mschaefer@8971:         return this.eventDate;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setEventDate(final Date eventDate) {
mschaefer@8971:         this.eventDate = eventDate;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "gauge_name")
mschaefer@8971:     public String getGaugeName() {
mschaefer@8971:         return this.gaugeName;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setGaugeName(final String gaugeName) {
mschaefer@8971:         this.gaugeName = gaugeName;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     @Column(name = "gauge_w")
mschaefer@8971:     public Double getGaugeW() {
mschaefer@8971:         return this.gaugeW;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setGaugeW(final Double gaugeW) {
mschaefer@8971:         this.gaugeW = gaugeW;
mschaefer@8971:     }
mschaefer@9117: 
mschaefer@9117:     /**
mschaefer@9117:      * Selects the collision values of a data series in a km range from the database
mschaefer@9117:      */
mschaefer@9117:     public static List<CollisionValue> getValues(final Collision parent, final double kmLo, final double kmHi) {
mschaefer@9117:         final Session session = SessionHolder.HOLDER.get();
mschaefer@9117:         final Query query = session.createQuery("FROM CollisionValue WHERE (collision=:parent)"
mschaefer@9117:                 + " AND (station >= :kmLo - 0.0001) AND (station <= :kmHi + 0.0001)");
mschaefer@9117:         query.setParameter("parent", parent);
mschaefer@9117:         query.setParameter("kmLo", new Double(kmLo));
mschaefer@9117:         query.setParameter("kmHi", new Double(kmHi));
mschaefer@9117:         return query.list();
mschaefer@9117:     }
mschaefer@9160:     /**
mschaefer@9160:      * Selects from the database the collision values of a km range of a river and a date range
mschaefer@9160:      */
mschaefer@9160:     public static List<CollisionValue> getValues(final River river, final double kmLo, final double kmHi, final Date fromDate, final Date toDate) {
mschaefer@9160:         final Session session = SessionHolder.HOLDER.get();
mschaefer@9160:         final SQLQuery query = session.createSQLQuery("SELECT station, event_date, gauge_w, gauge_name, ct.name AS collisiontype"
mschaefer@9160:                 + " FROM collision_values v INNER JOIN collision s ON v.collision_id=s.id"
mschaefer@9160:                 + " INNER JOIN collision_type ct ON v.collision_type_id=ct.id"
mschaefer@9160:                 + " WHERE (s.river_id=:river_id)"
mschaefer@9160:                 + " AND (station BETWEEN (:kmLo - 0.0001) AND (:kmHi + 0.0001))"
mschaefer@9160:                 + " AND (event_date BETWEEN :fromDate AND :toDate)"
mschaefer@9160:                 + " ORDER BY station, event_date")
mschaefer@9160:                 .addScalar("station", StandardBasicTypes.DOUBLE)
mschaefer@9160:                 .addScalar("event_date", StandardBasicTypes.DATE)
mschaefer@9160:                 .addScalar("gauge_w", StandardBasicTypes.DOUBLE)
mschaefer@9160:                 .addScalar("gauge_name", StandardBasicTypes.STRING)
mschaefer@9160:                 .addScalar("collisiontype", StandardBasicTypes.STRING);
mschaefer@9160:         query.setInteger("river_id", river.getId());
mschaefer@9160:         query.setDouble("kmLo", new Double(kmLo));
mschaefer@9160:         query.setDouble("kmHi", new Double(kmHi));
mschaefer@9160:         query.setDate("fromDate", fromDate);
mschaefer@9160:         query.setDate("toDate", toDate);
mschaefer@9160:         return buildCollisionValueList(query.list());
mschaefer@9160:     }
mschaefer@9160: 
mschaefer@9160:     private static List<CollisionValue> buildCollisionValueList(final List<Object[]> rows) {
mschaefer@9160:         final List<CollisionValue> values = new ArrayList<>();
mschaefer@9160:         for (int i = 0; i <= rows.size() - 1; i++)
mschaefer@9160:             values.add(new CollisionValue(null, (double) rows.get(i)[0], null, (Date) rows.get(i)[1], (String) rows.get(i)[3], (double) rows.get(i)[2]));
mschaefer@9160:         return values;
mschaefer@9160:     }
mschaefer@8971: }