tim@335: /**
tim@335:  *
tim@335:  */
tim@335: package de.intevation.gnv.state.profile.verticalcrosssection;
tim@335: 
tim@335: import java.io.IOException;
tim@335: import java.io.OutputStream;
tim@335: import java.io.OutputStreamWriter;
tim@335: import java.io.UnsupportedEncodingException;
tim@335: import java.util.Collection;
ingo@429: import java.util.HashMap;
tim@335: import java.util.Iterator;
tim@335: import java.util.Locale;
ingo@429: import java.util.Map;
tim@335: 
tim@335: import org.apache.log4j.Logger;
tim@335: 
ingo@358: import org.jfree.chart.ChartTheme;
ingo@358: 
ingo@429: import org.w3c.dom.Node;
ingo@429: 
ingo@429: import net.sf.ehcache.Element;
ingo@429: 
tim@335: import au.com.bytecode.opencsv.CSVWriter;
tim@335: 
tim@335: import com.vividsolutions.jts.geom.Point;
ingo@429: import com.vividsolutions.jts.geom.Coordinate;
ingo@429: import com.vividsolutions.jts.geom.LineString;
tim@335: import com.vividsolutions.jts.io.ParseException;
tim@335: import com.vividsolutions.jts.io.WKTReader;
tim@335: 
ingo@429: import de.intevation.artifactdatabase.Config;
ingo@429: import de.intevation.gnv.artifacts.cache.CacheFactory;
tim@335: import de.intevation.gnv.chart.Chart;
tim@335: import de.intevation.gnv.chart.ChartLabels;
tim@335: import de.intevation.gnv.chart.ChartStyle;
tim@335: import de.intevation.gnv.chart.exception.TechnicalChartException;
tim@335: import de.intevation.gnv.geobackend.base.Result;
ingo@429: import de.intevation.gnv.geobackend.base.ResultDescriptor;
ingo@429: import de.intevation.gnv.geobackend.base.query.QueryExecutor;
ingo@429: import de.intevation.gnv.geobackend.base.query.QueryExecutorFactory;
ingo@429: import de.intevation.gnv.geobackend.base.query.exception.QueryException;
sascha@431: 
ingo@429: import de.intevation.gnv.math.AttributedXYColumns;
ingo@429: import de.intevation.gnv.math.HeightValue;
ingo@429: import de.intevation.gnv.math.XYColumn;
sascha@431: import de.intevation.gnv.math.IJKey;
sascha@431: 
tim@335: import de.intevation.gnv.state.describedata.KeyValueDescibeData;
tim@335: import de.intevation.gnv.state.exception.StateException;
tim@335: import de.intevation.gnv.state.timeseries.TimeSeriesOutputState;
tim@335: import de.intevation.gnv.statistics.Statistics;
tim@335: import de.intevation.gnv.statistics.VerticalProfileStatistics;
ingo@429: import de.intevation.gnv.utils.WKTUtils;
tim@335: 
tim@335: /**
tim@335:  * @author Tim Englich <tim.englich@intevation.de>
tim@335:  * 
tim@335:  */
tim@335: public class VerticalCrossSectionOutputState extends TimeSeriesOutputState {
tim@335: 
ingo@429:     public static final String[] ATTRIBUTE_LIST = {
ingo@429:         "SHAPE",
ingo@429:         "Z",
ingo@429:         "YORDINATE",
ingo@429:         "IPOSITION",
ingo@429:         "JPOSITION",
ingo@429:         "KPOSITION"
ingo@429:     };
ingo@429: 
ingo@429:     private static Logger log = Logger.getLogger(
ingo@429:         VerticalCrossSectionOutputState.class);
ingo@429: 
ingo@429:     private String ijkQueryID = "horizontalprofile_meshpoint_cross_ij";
ingo@429: 
tim@335:     /**
tim@335:      * The UID of this Class
tim@335:      */
tim@335:     private static final long serialVersionUID = 3233620652465061860L;
tim@335: 
tim@335:     /**
tim@335:      * Constructor
tim@335:      */
tim@335:     public VerticalCrossSectionOutputState() {
tim@335:         super();
ingo@343:         super.domainLable = "chart.verticalcrosssection.title.xaxis";
tim@335:     }
tim@335: 
ingo@429: 
ingo@429:     @Override
ingo@429:     protected Object getChartResult(String uuid) {
ingo@429:         log.debug("VerticalCrossSectionOutputState.getChartResult");
ingo@429:         Collection<Result> result = null;
ingo@429:         String key = uuid + super.getID();
ingo@429: 
ingo@429:         Element element = CacheFactory.getInstance().getCache().get(key);
ingo@429:         if (element != null)
ingo@429:             return element.getObjectValue();
ingo@429: 
ingo@429:         log.debug("No results in cache yet.");
ingo@429:         if (inputData.containsKey("mesh_linestring")) {
ingo@429: 
ingo@429:             try {
ingo@429:                 LineString ls = (LineString) new WKTReader().read(
ingo@429:                     inputData.get("mesh_linestring").getValue());
ingo@429: 
ingo@429:                 Coordinate[] coords = ls.getCoordinates();
ingo@429: 
ingo@429:                 String additionWhere = WKTUtils.worldCoordinatesToIndex(
ingo@429:                     result,
ingo@429:                     inputData,
ingo@429:                     ijkQueryID
ingo@429:                 );
ingo@429: 
ingo@429: 
ingo@429:                 String[] filterValues =
ingo@429:                     generateFilterValuesFromInputData();
ingo@429:                 String[] addedFilterValues =
ingo@429:                     new String[filterValues.length + 1];
ingo@429: 
ingo@429:                 System.arraycopy(
ingo@429:                     filterValues, 0,
ingo@429:                     addedFilterValues, 0,
ingo@429:                     filterValues.length
ingo@429:                 );
ingo@429:                 addedFilterValues[filterValues.length] = additionWhere;
ingo@429: 
ingo@429:                 QueryExecutor exec = QueryExecutorFactory
ingo@429:                     .getInstance()
ingo@429:                     .getQueryExecutor();
ingo@429: 
ingo@429:                 result = exec.executeQuery(queryID, addedFilterValues);
ingo@429:             }
ingo@429:             catch (ParseException pe) {
ingo@429:                 log.error(pe, pe);
ingo@429:             }
ingo@429:             catch (QueryException qe) {
ingo@429:                 log.error(qe, qe);
ingo@429:             }
ingo@429:         }
ingo@429:         else {
ingo@429:             // TODO What should happen if there is no linestring?
ingo@429:             log.warn("No linestring in inputData.");
ingo@429:         }
ingo@429: 
ingo@429:         Object obj = process(preProcess(result));
ingo@429:         CacheFactory.getInstance().getCache().put(new Element(key, obj));
ingo@429: 
ingo@429:         return obj;
ingo@429:     }
ingo@429: 
ingo@429: 
ingo@429:     protected Object process(AttributedXYColumns columns) {
ingo@429: 
ingo@429:         // TODO Implement me
ingo@429:         return null;
ingo@429:     }
ingo@429: 
ingo@429: 
ingo@429:     protected AttributedXYColumns preProcess(Collection results) {
ingo@429:         AttributedXYColumns attColumns = new AttributedXYColumns();
sascha@431:         Map        map                 = new HashMap(1013);
ingo@429:         Iterator   iter                = results.iterator();
ingo@429: 
ingo@429:         int sIdx = -1;
ingo@429:         int iIdx = -1;
ingo@429:         int jIdx = -1;
ingo@429:         int kIdx = -1;
ingo@429:         int vIdx = -1;
ingo@429:         int zIdx = -1;
ingo@429: 
ingo@429:         while (iter.hasNext()) {
ingo@429:             Result result = (Result) iter.next();
ingo@429: 
ingo@429:             if (sIdx == -1) {
ingo@429:                 ResultDescriptor rd = result.getResultDescriptor();
ingo@429:                 int columnCount     = rd.getColumnCount();
ingo@429: 
ingo@429:                 sIdx = rd.getColumnIndex("SHAPE");
ingo@429:                 iIdx = rd.getColumnIndex("IPOSITION");
ingo@429:                 jIdx = rd.getColumnIndex("JPOSITION");
ingo@429:                 kIdx = rd.getColumnIndex("KPOSITION");
ingo@429:                 vIdx = rd.getColumnIndex("YORDINATE");
ingo@429:                 zIdx = rd.getColumnIndex("Z");
ingo@429: 
ingo@429:                 for (int i = 0; i < columnCount; i++) {
ingo@429:                     String colName = rd.getColumnName(i);
ingo@429: 
ingo@429:                     if (!attributeInList(colName)) {
ingo@429:                         attColumns.setAttribute(
ingo@429:                             colName,
ingo@429:                             result.getObject(colName));
ingo@429:                     }
ingo@429:                 }
ingo@429:             }
ingo@429: 
ingo@429:             try {
ingo@429:                 Point point  = (Point) new WKTReader().read(
ingo@429:                     result.getString(sIdx));
ingo@429:                 double v     = result.getDouble(vIdx);
ingo@429:                 int i        = result.getInteger(iIdx);
ingo@429:                 int j        = result.getInteger(jIdx);
ingo@429:                 int k        = result.getInteger(kIdx);
ingo@429:                 int z        = result.getInteger(zIdx);
ingo@429: 
sascha@431:                 IJKey key = new IJKey(i, j);
ingo@429: 
sascha@431:                 XYColumn col = (XYColumn)map.get(key);
sascha@431: 
sascha@431:                 if (col == null) {
sascha@431:                     col = new XYColumn(point.getX(), point.getY(), i, j);
sascha@431:                     map.put(key, col);
ingo@429:                 }
ingo@429: 
sascha@431:                 col.add(new HeightValue(z, v, k));
ingo@429:             }
ingo@429:             catch (ParseException pe) {
ingo@429:                 log.warn("Error while parsing geometry.", pe);
ingo@429:             }
ingo@429:         }
ingo@429: 
ingo@429:         XYColumn[] cols = (XYColumn[])map.values().toArray(
ingo@429:             new XYColumn[map.size()]);
ingo@429:         attColumns.setXYColumns(cols);
ingo@429: 
ingo@429:         return attColumns;
ingo@429:     }
ingo@429: 
ingo@429: 
ingo@429:     protected boolean attributeInList(String name) {
ingo@429:         for (int i = 0; i < ATTRIBUTE_LIST.length; i++) {
ingo@429:             if (name.equals(ATTRIBUTE_LIST[i]))
ingo@429:                 return true;
ingo@429:         }
ingo@429: 
ingo@429:         return false;
ingo@429:     }
ingo@429: 
ingo@429: 
tim@335:     @Override
tim@335:     protected Chart getChart(
tim@335:         ChartLabels  chartLables,
ingo@358:         ChartTheme   theme,
tim@335:         Collection   parameters,
tim@335:         Collection   measurements,
tim@335:         Collection   dates,
ingo@429:         Object       result,
tim@335:         Locale       locale,
tim@335:         String       uuid,
tim@335:         boolean      linesVisible,
tim@335:         boolean      shapesVisible
tim@335:     ) {
tim@335:         Chart chart = null;
tim@335: 
tim@335:         if (CACHE_CHART) {
tim@335:             log.info("Try to get verticalcrosssection chart from cache.");
tim@335:             chart = (Chart) getChartFromCache(uuid);
tim@335:         }
tim@335: 
tim@335:         if (chart != null)
tim@335:             return chart;
tim@335: 
tim@335:         log.info("Chart not in cache yet.");
ingo@429: 
tim@335:         log.warn("This sort of chart is not implemented yet.");
tim@335:         /* TODO Implement a special chart for this sort of charts.
tim@335:         chart = new VerticalProfileChart(
tim@335:             chartLables,
tim@335:             chartTheme,
tim@335:             parameters,
tim@335:             measurements,
tim@335:             result,
tim@335:             dates,
tim@335:             locale
tim@335:         );
tim@335:         chart.generateChart();
tim@335: 
tim@335:         if (CACHE_CHART) {
tim@335:             log.info("Put chart into cache.");
tim@335:             purifyChart(chart, uuid);
tim@335:         }
tim@335:         */
tim@335: 
tim@335:         return chart;
tim@335:     }
tim@335: 
tim@335:     /**
tim@335:      * @see de.intevation.gnv.state.timeseries.TimeSeriesOutputState#createChart(java.io.OutputStream,
tim@335:      *      java.util.Collection, java.util.Collection, java.lang.String,
tim@335:      *      de.intevation.gnv.chart.ChartStyle,
tim@335:      *      de.intevation.gnv.chart.ChartLabels)
tim@335:      */
tim@335:     /*
tim@335:     @Override
tim@335:     protected void createChart(OutputStream outputStream,
tim@335:                                Collection<KeyValueDescibeData> parameters,
tim@335:                                Collection<KeyValueDescibeData> measurements,
tim@335:                                Collection<KeyValueDescibeData> dates,
tim@335:                                ChartStyle chartStyle, ChartLabels chartLables,
tim@335:                                String uuid) throws IOException,
tim@335:                                            TechnicalChartException {
tim@335:         new VerticalCrossSectionChartFactory().createProfileChart(chartLables,
tim@335:                 chartStyle, parameters, measurements, dates, outputStream, this
tim@335:                         .getChartResult(uuid));
tim@335:     }
tim@335:     */
tim@335: 
tim@335:     /**
tim@335:      * @see de.intevation.gnv.state.timeseries.TimeSeriesOutputState#getStatisticsGenerator()
tim@335:      */
tim@335:     @Override
tim@335:     protected Statistics getStatisticsGenerator() {
tim@335:         return new VerticalProfileStatistics();
tim@335:     }
ingo@429: 
tim@335:     /**
tim@335:      * @see de.intevation.gnv.state.timeseries.TimeSeriesOutputState#createCSV(java.io.OutputStream, java.util.Collection)
tim@335:      */
tim@335:     @Override
tim@335:     protected void createCSV(OutputStream outputStream,
tim@335:                              Collection<Result> chartResult)
tim@335:                                                             throws UnsupportedEncodingException,
tim@335:                                                             IOException,
tim@335:                                                             StateException {
tim@335:         if (chartResult != null) {
tim@335:             try {
tim@335:                 CSVWriter writer = new CSVWriter(new OutputStreamWriter(
tim@335:                         outputStream, "ISO-8859-1"), ','); 
tim@335:                 // USE THIS ENCODING BECAUSE OF
tim@335:                 // PROBLEMS WITH EXCEL AND UTF-8
tim@335:                 Iterator<Result> it = chartResult.iterator();
tim@335:                 WKTReader wktReader = new WKTReader();
tim@335:                 while (it.hasNext()) {
tim@335:                     Result result = it.next();
tim@335:                     int i = 0;
tim@335:                     String[] entries = new String[9];
tim@335:                     Point p = (Point)wktReader.read(result.getString("SHAPE"));
tim@335:                     entries[i++] = ""+p.getX();
tim@335:                     entries[i++] = ""+p.getY();
tim@335:                     entries[i++] = result.getString("Z");
tim@335:                     entries[i++] = result.getString("YORDINATE");
tim@335:                     entries[i++] = result.getString("GROUP1");
tim@335:                     entries[i++] = result.getString("GROUP2");
tim@335:                     entries[i++] = result.getString("IPOSITION");
tim@335:                     entries[i++] = result.getString("JPOSITION");
tim@335:                     entries[i++] = result.getString("KPOSITION");
tim@335:                     writer.writeNext(entries);
tim@335:                 }
tim@335:                 writer.close();
tim@335:             } catch (ParseException e) {
tim@335:                 log.error(e,e);
tim@335:                 throw new StateException(
tim@335:                 "Exception occured while parsing an Point from WKT.");
tim@335:             }
tim@335:         } else {
tim@335:             log.error("No Data given for generating an CSV-File.");
tim@335:             throw new StateException(
tim@335:                     "No Data given for generating an CSV-File.");
tim@335:         }
tim@335:     }
tim@335: 
tim@335: }