teichmann@5844: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
teichmann@5844:  * Software engineering by Intevation GmbH
teichmann@5844:  *
teichmann@5992:  * This file is Free Software under the GNU AGPL (>=v3)
teichmann@5844:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
teichmann@5992:  * documentation coming with Dive4Elements River for details.
teichmann@5844:  */
teichmann@5844: 
teichmann@5829: package org.dive4elements.river.importer;
sascha@196: 
teichmann@5829: import org.dive4elements.river.model.Wst;
teichmann@5829: import org.dive4elements.river.model.River;
sascha@201: 
ingo@478: import org.apache.log4j.Logger;
ingo@478: 
sascha@201: import org.hibernate.Session;
sascha@201: import org.hibernate.Query;
sascha@201: 
ingo@478: import java.util.ArrayList;
sascha@201: import java.util.List;
sascha@196: 
felix@6321: /** Not (yet) db-mapped WST object. */
sascha@196: public class ImportWst
sascha@196: {
sascha@499:     private static Logger log = Logger.getLogger(ImportWst.class);
ingo@478: 
teichmann@6336:     public interface ImportWstColumnFactory {
teichmann@6336:         ImportWstColumn create(ImportWst iw, int position);
teichmann@6336:     }
teichmann@6336: 
teichmann@6336:     public static final ImportWstColumnFactory COLUMN_FACTORY =
teichmann@6336:         new ImportWstColumnFactory() {
teichmann@6336:             @Override
teichmann@6336:             public ImportWstColumn create(ImportWst importWst, int position) {
teichmann@6336:                 return new ImportWstColumn(importWst, null, null, position);
teichmann@6336:             }
teichmann@6336:         };
teichmann@6336: 
sascha@201:     protected String description;
sascha@201: 
sascha@467:     protected Integer kind;
sascha@467: 
sascha@474:     protected List<ImportWstColumn> columns;
sascha@201: 
ingo@2346:     protected ImportUnit unit;
ingo@2346: 
teichmann@6336:     protected ImportWstColumnFactory columnFactory;
teichmann@6336: 
felix@5000:     /** Wst as in db. */
sascha@196:     protected Wst peer;
sascha@196: 
sascha@196:     public ImportWst() {
teichmann@6336:         this(COLUMN_FACTORY);
teichmann@6336:     }
teichmann@6336: 
teichmann@6336:     public ImportWst(ImportWstColumnFactory columnFactory) {
teichmann@6336:         this.columnFactory = columnFactory;
sascha@467:         kind = 0;
sascha@474:         columns = new ArrayList<ImportWstColumn>();
sascha@196:     }
sascha@196: 
sascha@201:     public ImportWst(String description) {
teichmann@6336:         this(description, COLUMN_FACTORY);
teichmann@6336:     }
teichmann@6336: 
teichmann@6336:     public ImportWst(String description, ImportWstColumnFactory columnFactory) {
teichmann@6336:         this(columnFactory);
sascha@201:         this.description = description;
sascha@201:     }
sascha@201: 
sascha@201:     public String getDescription() {
sascha@201:         return description;
sascha@201:     }
sascha@201: 
sascha@482:     public Integer getKind() {
sascha@482:         return kind;
sascha@482:     }
sascha@482: 
sascha@482:     public void setKind(Integer kind) {
sascha@482:         this.kind = kind;
sascha@482:     }
sascha@482: 
sascha@482: 
sascha@201:     public void setDescription(String description) {
sascha@201:         this.description = description;
sascha@201:     }
sascha@201: 
felix@5263:     /** Create columns that can be accessed with getColumn. */
sascha@474:     public void setNumberColumns(int numColumns) {
sascha@474:         for (int i = 0; i < numColumns; ++i) {
teichmann@6336:             columns.add(columnFactory.create(this, i));
sascha@201:         }
sascha@474:     }
sascha@474: 
sascha@474:     public int getNumberColumns() {
sascha@474:         return columns.size();
sascha@474:     }
sascha@474: 
sascha@474:     public ImportWstColumn getColumn(int index) {
sascha@474:         return columns.get(index);
sascha@201:     }
sascha@201: 
felix@5263:     public List<ImportWstColumn> getColumns() {
felix@5263:         return columns;
felix@5263:     }
felix@5263: 
felix@5263:     /** Adds a column. Assumes that columns wst is this instance. */
felix@5263:     public void addColumn(ImportWstColumn column) {
felix@5263:         columns.add(column);
felix@5263:     }
felix@5263: 
ingo@2347:     public ImportUnit getUnit() {
ingo@2347:         return unit;
ingo@2347:     }
ingo@2347: 
ingo@2346:     public void setUnit(ImportUnit unit) {
ingo@2346:         this.unit = unit;
ingo@2346:     }
ingo@2346: 
sascha@201:     public void storeDependencies(River river) {
sascha@499: 
sascha@499:         log.info("store '" + description + "'");
sascha@201:         Wst wst = getPeer(river);
sascha@201: 
sascha@474:         for (ImportWstColumn column: columns) {
sascha@201:             column.storeDependencies(river);
sascha@201:         }
ingo@2346: 
sascha@503:         Session session = ImporterSession.getInstance().getDatabaseSession();
sascha@503:         session.flush();
sascha@201:     }
sascha@201: 
teichmann@5542:     public void fixRangesOrder() {
teichmann@5542:         for (ImportWstColumn column: columns) {
teichmann@5542:             column.fixRangesOrder();
teichmann@5542:         }
teichmann@5542:     }
teichmann@5542: 
felix@5000:     /** Get corresponding mapped wst (from database). */
sascha@201:     public Wst getPeer(River river) {
sascha@196:         if (peer == null) {
sascha@497:             Session session = ImporterSession.getInstance().getDatabaseSession();
sascha@201:             Query query = session.createQuery(
sascha@201:                 "from Wst where " +
sascha@467:                 "river=:river and description=:description and kind=:kind");
sascha@201:             query.setParameter("river",       river);
sascha@201:             query.setParameter("description", description);
sascha@467:             query.setParameter("kind",        kind);
sascha@201:             List<Wst> wsts = query.list();
sascha@201:             if (wsts.isEmpty()) {
ingo@2347:                 peer = new Wst(river, description, kind);
sascha@201:                 session.save(peer);
sascha@201:             }
sascha@201:             else {
sascha@201:                 peer = wsts.get(0);
sascha@201:             }
sascha@201: 
sascha@196:         }
sascha@196:         return peer;
sascha@196:     }
sascha@196: }
sascha@196: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :