gernotbelger@9063: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
gernotbelger@9063:  * Software engineering by Intevation GmbH
gernotbelger@9063:  *
gernotbelger@9063:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@9063:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@9063:  * documentation coming with Dive4Elements River for details.
gernotbelger@9063:  */
gernotbelger@9063: 
gernotbelger@9063: package org.dive4elements.river.artifacts.sinfo.collision;
gernotbelger@9063: 
gernotbelger@9076: import java.util.HashMap;
gernotbelger@9076: import java.util.List;
gernotbelger@9076: import java.util.Map;
gernotbelger@9076: 
gernotbelger@9063: import org.apache.log4j.Logger;
gernotbelger@9076: import org.dive4elements.artifacts.Artifact;
gernotbelger@9076: import org.dive4elements.artifacts.CallContext;
gernotbelger@9076: import org.dive4elements.artifacts.common.utils.XMLUtils.ElementCreator;
gernotbelger@9076: import org.dive4elements.river.artifacts.model.CollisionHibernateFactory;
gernotbelger@9076: import org.dive4elements.river.artifacts.sinfo.SINFOArtifact;
gernotbelger@9076: import org.dive4elements.river.artifacts.states.AddTableDataHelper;
gernotbelger@9063: import org.dive4elements.river.artifacts.states.DefaultState;
gernotbelger@9076: import org.dive4elements.river.model.sinfo.Collision;
gernotbelger@9076: import org.w3c.dom.Element;
gernotbelger@9063: 
gernotbelger@9073: public class LoadMultipleEpochSelectState extends DefaultState {
gernotbelger@9063:     /** The log used in this class. */
gernotbelger@9073:     private static Logger log = Logger.getLogger(LoadMultipleEpochSelectState.class);
gernotbelger@9063: 
gernotbelger@9063:     /**
gernotbelger@9063:      * The default constructor that initializes an empty State object.
gernotbelger@9063:      */
gernotbelger@9073:     public LoadMultipleEpochSelectState() {
gernotbelger@9063:     }
gernotbelger@9063: 
gernotbelger@9063:     @Override
gernotbelger@9063:     protected String getUIProvider() {
gernotbelger@9063:         return "sinfo.collision.load_epoch_select";
gernotbelger@9063:     }
gernotbelger@9076: 
gernotbelger@9076:     @Override
gernotbelger@9076:     protected void appendItems(final Artifact artifact, final ElementCreator creator, final String name, final CallContext context, final Element select) {
gernotbelger@9076: 
gernotbelger@9076:         try {
gernotbelger@9076:             if ("epochs".equals(name)) {
gernotbelger@9076: 
gernotbelger@9076:                 final AddTableDataHelper helper = new AddTableDataHelper(creator, select, "year", context.getMeta());
gernotbelger@9076: 
gernotbelger@9076:                 helper.addColumn(0, "pinfrom", "40", "common.client.ui.from", "ICON", "CENTER", "from");
gernotbelger@9076:                 helper.addColumn(1, "pinto", "40", "common.client.ui.to", "ICON", "CENTER", "to");
gernotbelger@9076:                 helper.addColumn(2, "year", "60", "year", "INTEGER", "LEFT", null);
gernotbelger@9076: 
gernotbelger@9076:                 final CollisionAccess access = new CollisionAccess((SINFOArtifact) artifact); // Der River wurde im vorigen State bereits gesetzt
gernotbelger@9076: 
gernotbelger@9076:                 final List<Collision> collisions = CollisionHibernateFactory.getCollisionsByRiver(access.getRiver());
gernotbelger@9076:                 for (final Collision coll : collisions) { // TODO: Filter on Range (?)
gernotbelger@9076:                     final Integer year = coll.getYear();
gernotbelger@9076:                     final Map<String, String> row = new HashMap<>();
gernotbelger@9076:                     row.put("year", year.toString()); // Nullpointer?
gernotbelger@9076:                     try {
gernotbelger@9076:                         helper.addRow(row);
gernotbelger@9076:                     }
gernotbelger@9076:                     catch (final Exception e) {
gernotbelger@9076:                         e.printStackTrace();
gernotbelger@9076:                     }
gernotbelger@9076:                 }
gernotbelger@9076:                 helper.submitMapToXml();
gernotbelger@9076:             }
gernotbelger@9076:         }
gernotbelger@9076:         catch (final IllegalArgumentException iae) {
gernotbelger@9076:             iae.printStackTrace();
gernotbelger@9076:         }
gernotbelger@9076:     }
gernotbelger@9063: }