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@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.FlowDepth;
mschaefer@8971: import org.hibernate.Query;
mschaefer@8971: import org.hibernate.Session;
mschaefer@8971: 
mschaefer@8971: /**
mschaefer@8971:  * Imported flow depth data series group of a river
mschaefer@8971:  *
mschaefer@8971:  * @author Matthias Schäfer
mschaefer@8971:  *
mschaefer@8971:  */
mschaefer@8971: public class FlowDepthSeriesImport {
mschaefer@8971: 
mschaefer@8971:     /***** FIELDS *****/
mschaefer@8971: 
mschaefer@8971:     private static Logger log = Logger.getLogger(FlowDepthSeriesImport.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@8971:     private String kmrange_info;
mschaefer@8971: 
mschaefer@9038:     private String notes;
mschaefer@8971: 
mschaefer@8971:     private int year;
mschaefer@8971: 
mschaefer@8971:     private String sounding_info;
mschaefer@8971: 
mschaefer@8971:     private String evaluation_by;
mschaefer@8971: 
mschaefer@8971:     private FlowDepth peer;
mschaefer@8971: 
mschaefer@8971:     private StoreMode storeMode;
mschaefer@8971: 
mschaefer@8971: 
mschaefer@8971:     /***** CONSTRUCTORS *****/
mschaefer@8971: 
mschaefer@8971:     public FlowDepthSeriesImport(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@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@8971:     public int getYear() {
mschaefer@8971:         return this.year;
mschaefer@8971:     }
mschaefer@8971: 
mschaefer@8971:     public void setYear(final int year) {
mschaefer@8971:         this.year = year;
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 FlowDepth 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<FlowDepth> 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@9656:     public List<FlowDepth> querySeriesItem(final Session session, final River river) {
mschaefer@8971:         final Query query = session.createQuery("FROM FlowDepth 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 FlowDepth createSeriesItem(final River river) {
mschaefer@9038:         return new FlowDepth(river, this.filename, this.kmrange_info, this.notes, this.year, this.sounding_info, this.evaluation_by);
mschaefer@8971:     }
mschaefer@8971: }