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.importer.sinfo.importitem;
mschaefer@8971: 
mschaefer@9010: import java.util.Date;
mschaefer@8971: import java.util.List;
mschaefer@8971: 
mschaefer@8971: import org.apache.log4j.Logger;
mschaefer@8971: import org.dive4elements.river.importer.ImporterSession;
mschaefer@8971: import org.dive4elements.river.importer.common.StoreMode;
mschaefer@8971: import org.dive4elements.river.model.River;
mschaefer@8971: import org.dive4elements.river.model.sinfo.Tkh;
mschaefer@9032: import org.dive4elements.river.model.sinfo.Tkh.Group;
mschaefer@8971: import org.hibernate.Query;
mschaefer@8971: import org.hibernate.Session;
mschaefer@8971: 
mschaefer@8971: /**
mschaefer@8971:  * Imported Transportkörperhöhe data series group of a river
mschaefer@8971:  *
mschaefer@8971:  * @author Matthias Schäfer
mschaefer@8971:  *
mschaefer@8971:  */
mschaefer@8971: public class TkhSeriesImport {
mschaefer@8971: 
mschaefer@8971:     /***** FIELDS *****/
mschaefer@8971: 
mschaefer@8971:     private static Logger log = Logger.getLogger(TkhSeriesImport.class);
mschaefer@8971: 
mschaefer@8971:     /**
mschaefer@8971:      * Name of the imported file without type extension
mschaefer@8971:      */
mschaefer@8971:     private final String filename;
mschaefer@8971: 
mschaefer@9032:     private Group group;
mschaefer@9032: 
mschaefer@8971:     private String kmrange_info;
mschaefer@8971: 
mschaefer@9038:     private String notes;
mschaefer@8971: 
mschaefer@9010:     private Date day;
mschaefer@8971: 
mschaefer@8971:     private String sounding_info;
mschaefer@8971: 
mschaefer@8971:     private String evaluation_by;
mschaefer@8971: 
mschaefer@8971:     private Tkh peer;
mschaefer@8971: 
mschaefer@8971:     private StoreMode storeMode;
mschaefer@8971: 
mschaefer@8971: 
mschaefer@8971:     /***** CONSTRUCTORS *****/
mschaefer@8971: 
mschaefer@8971:     public TkhSeriesImport(final String filename) {
mschaefer@8971:         this.filename = filename;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971: 
mschaefer@8971:     /***** METHODS *****/
mschaefer@8971: 
mschaefer@8971:     public String getFilename() {
mschaefer@8971:         return this.filename;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@9032:     public Group getGroup() {
mschaefer@9032:         return this.group;
mschaefer@9032:     }
mschaefer@9032: 
mschaefer@9032:     public void setGroup(final Group group) {
mschaefer@9032:         this.group = group;
mschaefer@9032:     }
mschaefer@9032: 
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:     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@9010:     public Date getDay() {
mschaefer@9010:         return this.day;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@9010:     public void setDay(final Date day) {
mschaefer@9010:         this.day = day;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public String getSounding_info() {
mschaefer@8971:         return this.sounding_info;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setSounding_info(final String sounding_info) {
mschaefer@8971:         this.sounding_info = sounding_info;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public String getEvaluation_by() {
mschaefer@8971:         return this.evaluation_by;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setEvaluation_by(final String evaluation_by) {
mschaefer@8971:         this.evaluation_by = evaluation_by;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public StoreMode getStoreMode() {
mschaefer@8971:         return this.storeMode;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     /**
mschaefer@8971:      * Gets the model object of the data series group, inserting it into the database if not already existing
mschaefer@8971:      */
mschaefer@8971:     public Tkh getPeer(final River river) {
mschaefer@8971:         if (this.peer != null) {
mschaefer@8971:             this.storeMode = StoreMode.NONE;
mschaefer@8971:             return this.peer;
mschaefer@8971:         }
mschaefer@8971:         final Session session = ImporterSession.getInstance().getDatabaseSession();
mschaefer@8971:         final List<Tkh> rows = querySeriesItem(session, river);
mschaefer@8971:         if (rows.isEmpty()) {
mschaefer@8971:             log.info("Create new database instance");
mschaefer@8971:             this.peer = createSeriesItem(river);
mschaefer@8971:             session.save(this.peer);
mschaefer@8971:             this.storeMode = StoreMode.INSERT;
mschaefer@8971:         }
mschaefer@8971:         else {
mschaefer@8971:             this.peer = rows.get(0);
mschaefer@8971:             this.storeMode = StoreMode.UPDATE;
mschaefer@8971:         }
mschaefer@8971:         return this.peer;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@9655:     public List<Tkh> querySeriesItem(final Session session, final River river) {
mschaefer@8971:         final Query query = session.createQuery("FROM Tkh WHERE river=:river AND lower(filename)=:filename");
mschaefer@8971:         query.setParameter("river", river);
mschaefer@8971:         query.setParameter("filename", this.filename.toLowerCase());
mschaefer@8971:         return query.list();
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     private Tkh createSeriesItem(final River river) {
mschaefer@9038:         return new Tkh(river, this.filename, this.kmrange_info, this.notes, this.day, this.sounding_info, this.evaluation_by, this.group);
mschaefer@8971:     }
mschaefer@8971: }