tim@335: package de.intevation.gnv.state.profile.verticalcrosssection; tim@335: ingo@429: import com.vividsolutions.jts.geom.Coordinate; tim@335: sascha@439: import de.intevation.artifacts.CallContext; sascha@439: sascha@521: import de.intevation.gnv.artifacts.cache.CacheFactory; sascha@521: sascha@445: import de.intevation.gnv.artifacts.context.GNVArtifactContext; sascha@445: ingo@492: import de.intevation.gnv.artifacts.ressource.RessourceFactory; sascha@432: tim@335: import de.intevation.gnv.chart.Chart; tim@335: import de.intevation.gnv.chart.ChartLabels; sascha@446: import de.intevation.gnv.chart.VerticalCrossSectionChart; sascha@432: tim@335: import de.intevation.gnv.geobackend.base.Result; ingo@429: import de.intevation.gnv.geobackend.base.ResultDescriptor; sascha@432: ingo@429: import de.intevation.gnv.geobackend.base.query.QueryExecutor; ingo@429: import de.intevation.gnv.geobackend.base.query.QueryExecutorFactory; sascha@521: ingo@429: import de.intevation.gnv.geobackend.base.query.exception.QueryException; sascha@431: sascha@521: import de.intevation.gnv.geobackend.sde.datasources.RasterObject; sascha@521: sascha@521: import de.intevation.gnv.jfreechart.PolygonDataset; sascha@521: import de.intevation.gnv.jfreechart.PolygonSeries; sascha@521: ingo@429: import de.intevation.gnv.math.AttributedXYColumns; ingo@429: import de.intevation.gnv.math.HeightValue; sascha@521: import de.intevation.gnv.math.IJKey; sascha@521: import de.intevation.gnv.math.Interpolation3D; sascha@521: import de.intevation.gnv.math.LinearMetrics; tim@455: import de.intevation.gnv.math.QueriedXYDepth; ingo@429: import de.intevation.gnv.math.XYColumn; sascha@521: sascha@521: import de.intevation.gnv.raster.Filter; sascha@521: import de.intevation.gnv.raster.IsoAttributeGenerator; sascha@521: import de.intevation.gnv.raster.IsoPolygonSeriesProducer; sascha@521: import de.intevation.gnv.raster.Palette; sascha@521: import de.intevation.gnv.raster.PaletteManager; sascha@521: import de.intevation.gnv.raster.PolygonDatasetProducer; sascha@521: import de.intevation.gnv.raster.Raster; sascha@521: import de.intevation.gnv.raster.Vectorizer; sascha@431: sascha@440: import de.intevation.gnv.state.InputData; sascha@440: tim@335: import de.intevation.gnv.state.describedata.KeyValueDescibeData; sascha@521: tim@335: import de.intevation.gnv.state.exception.StateException; sascha@521: tim@335: import de.intevation.gnv.state.timeseries.TimeSeriesOutputState; sascha@432: tim@335: import de.intevation.gnv.statistics.Statistics; sascha@454: import de.intevation.gnv.statistics.VerticalCrossSectionStatistics; sascha@432: sascha@445: import de.intevation.gnv.utils.DistanceCalculator; sascha@521: import de.intevation.gnv.utils.StringUtils; ingo@429: import de.intevation.gnv.utils.WKTUtils; tim@335: ingo@527: import java.awt.Color; sascha@521: import java.awt.Dimension; sascha@521: import java.awt.Paint; sascha@445: sascha@521: import java.io.IOException; sascha@521: import java.io.OutputStream; sascha@521: import java.io.UnsupportedEncodingException; sascha@445: sascha@521: import java.util.ArrayList; sascha@521: import java.util.Arrays; sascha@521: import java.util.Collection; sascha@521: import java.util.HashMap; sascha@521: import java.util.Iterator; sascha@521: import java.util.List; sascha@521: import java.util.Locale; sascha@521: import java.util.Map; sascha@521: sascha@521: import net.sf.ehcache.Element; sascha@521: sascha@521: import org.apache.log4j.Logger; sascha@521: sascha@521: import org.jfree.chart.ChartTheme; sascha@463: tim@335: /** sascha@445: * @author Tim Englich (tim.englich@intevation.de) ingo@492: * @author Ingo Weinzierl (ingo.weinzierl@intevation.de) sascha@445: * @author Sascha L. Teichmann (sascha.teichmann@intevation.de) tim@335: */ tim@335: public class VerticalCrossSectionOutputState extends TimeSeriesOutputState { tim@335: ingo@492: public static final String CHART_TYPE = "verticalcrosssection"; ingo@492: sascha@521: public static final Integer GROUND_FILL_INDEX = Integer.valueOf(-2); sascha@521: sascha@528: public static final boolean USE_INDEX_BUFFER = sascha@528: Boolean.getBoolean("gnv.vertical.cross.section.index.buffer"); sascha@528: 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: ingo@492: private String rangeLabel; ingo@492: 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"; ingo@492: this.rangeLabel = "chart.verticalcrosssection.title.yaxis"; ingo@492: } ingo@492: ingo@533: @Override ingo@533: public void initialize(String uuid, CallContext callContext) ingo@533: throws StateException { ingo@533: super.initialize(uuid, callContext); ingo@533: ingo@533: getChartResult(uuid, callContext); ingo@533: } ingo@492: ingo@492: @Override ingo@492: protected ChartLabels createChartLabels(Locale locale, String uuid) { ingo@492: RessourceFactory factory = RessourceFactory.getInstance(); ingo@492: String parameterName = getSelectedInputDataName( ingo@492: uuid, parameterValuesName); ingo@492: ingo@492: if (parameterName == null) ingo@492: parameterName = "parameterid"; ingo@492: ingo@492: return new ChartLabels( ingo@492: createChartTitle(locale, uuid), ingo@492: createChartSubtitle(locale, uuid), ingo@492: factory.getRessource(locale, domainLable, domainLable), ingo@492: factory.getRessource(locale, rangeLabel, rangeLabel), ingo@492: parameterName ingo@492: ); ingo@492: } ingo@492: ingo@492: ingo@492: @Override ingo@492: protected String createChartSubtitle(Locale locale, String uuid) { ingo@492: String date = getSelectedInputDataName(uuid, dateValueName); ingo@492: ingo@492: if (date == null) ingo@492: date = "dateid"; ingo@492: ingo@492: RessourceFactory factory = RessourceFactory.getInstance(); ingo@492: String chartType = factory.getRessource( ingo@492: locale, CHART_TYPE, CHART_TYPE); ingo@492: ingo@492: return chartType + ": " + date; tim@335: } tim@335: ingo@429: ingo@429: @Override sascha@439: protected Object getChartResult(String uuid, CallContext callContext) { ingo@429: log.debug("VerticalCrossSectionOutputState.getChartResult"); ingo@429: Collection 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: sascha@440: InputData meshLine = inputData.get("mesh_linestring"); sascha@440: InputData meshId = inputData.get("meshid"); ingo@429: sascha@440: if (meshLine == null) { sascha@440: log.error("mesh_linestring is not defined"); sascha@440: throw new IllegalStateException("missing mesh_linestring"); ingo@429: } ingo@429: sascha@440: if (meshId == null) { sascha@440: log.error("meshid is not defined"); sascha@440: throw new IllegalStateException("missing meshid"); sascha@440: } sascha@440: sascha@440: Coordinate [] coords = WKTUtils.toCoordinates( sascha@440: meshLine.getValue()); sascha@440: sascha@440: if (coords == null) { sascha@440: throw new IllegalStateException("cannot read coordinates"); sascha@440: } sascha@440: sascha@440: try { sascha@528: String additionWhere = USE_INDEX_BUFFER sascha@528: ? WKTUtils.worldCoordinatesToIndex( sascha@528: coords, sascha@528: result, sascha@528: meshId.getValue(), sascha@528: ijkQueryID) sascha@528: : WKTUtils.TRUE_EXPRESSION; sascha@440: sascha@440: String[] addedFilterValues = StringUtils.append( sascha@440: generateFilterValuesFromInputData(), sascha@440: additionWhere); sascha@440: sascha@440: QueryExecutor exec = QueryExecutorFactory sascha@440: .getInstance() sascha@440: .getQueryExecutor(); sascha@440: sascha@440: result = exec.executeQuery(queryID, addedFilterValues); sascha@440: } sascha@440: catch (QueryException qe) { sascha@440: log.error(qe, qe); sascha@440: } sascha@440: sascha@440: Object obj = process( sascha@440: Arrays.asList(coords), sascha@440: preProcess(result), sascha@440: callContext); sascha@440: ingo@429: CacheFactory.getInstance().getCache().put(new Element(key, obj)); ingo@429: ingo@429: return obj; ingo@429: } ingo@429: ingo@492: ingo@492: protected String getSelectedInputDataName(String uuid, String id) { ingo@492: Collection values = getCollection(id, uuid); ingo@492: ingo@492: if (values != null) { ingo@492: Iterator it = values.iterator(); ingo@492: ingo@492: while (it.hasNext()) { ingo@492: KeyValueDescibeData data = (KeyValueDescibeData) it.next(); ingo@492: ingo@492: if (data.isSelected()) { ingo@492: return data.getValue(); ingo@492: } ingo@492: } ingo@492: } ingo@492: return null; ingo@492: } ingo@492: sascha@463: private static int getGroundInterpolation(CallContext callContext) { sascha@463: GNVArtifactContext context = sascha@463: (GNVArtifactContext)callContext.globalContext(); sascha@463: sascha@463: String interpolation = (String)context.get( sascha@463: GNVArtifactContext.VERTICAL_CROSS_SECTION_GROUND_INTERPOLATION_KEY); sascha@463: sascha@463: return RasterObject.getInterpolationType(interpolation); sascha@463: } sascha@463: sascha@445: private static Dimension getRasterSize(CallContext callContext) { sascha@445: GNVArtifactContext context = sascha@445: (GNVArtifactContext)callContext.globalContext(); sascha@445: Dimension size = (Dimension)context.get( sascha@445: GNVArtifactContext.VERTICAL_CROSS_SECTION_SAMPLES_KEY); sascha@445: return size != null sascha@445: ? size sascha@445: : GNVArtifactContext.DEFAULT_VERTICAL_CROSS_SECTION_SAMPLES; sascha@445: } sascha@445: sascha@446: private static List getFilterFactories( sascha@446: CallContext callContext sascha@446: ) { sascha@445: GNVArtifactContext context = sascha@445: (GNVArtifactContext)callContext.globalContext(); sascha@445: List factories = (List)context.get( sascha@445: GNVArtifactContext.VERTICAL_CROSS_SECTION_FILTER_FACTORIES_KEY); sascha@445: return factories != null sascha@445: ? factories sascha@445: : new ArrayList(); sascha@445: } sascha@445: sascha@445: private static Map getPalettes( sascha@445: CallContext callContext sascha@445: ) { sascha@445: GNVArtifactContext context = sascha@445: (GNVArtifactContext)callContext.globalContext(); sascha@445: Map palettes = sascha@445: (Map)context.get( sascha@445: GNVArtifactContext.PALETTES_KEY); sascha@445: return palettes != null sascha@445: ? palettes sascha@445: : new HashMap(); sascha@445: } sascha@445: sascha@521: private static Paint getGroundFill(CallContext callContext) { sascha@521: GNVArtifactContext context = sascha@521: (GNVArtifactContext)callContext.globalContext(); sascha@521: Paint fill = (Paint)context.get( sascha@521: GNVArtifactContext.VERTICAL_CROSS_SECTION_GROUND_FILL_KEY); sascha@521: return fill != null sascha@521: ? fill sascha@521: : GNVArtifactContext.DEFAULT_VERTICAL_CROSS_SECTION_GROUND_FILL; sascha@521: } sascha@521: sascha@445: public static final double EPSILON = 1e-5d; ingo@429: sascha@440: protected Object process( sascha@440: List path, sascha@440: AttributedXYColumns columns, sascha@440: CallContext callContext sascha@440: ) { sascha@445: Integer parameterId = sascha@445: (Integer)columns.getAttribute("GROUP1"); // XXX: hardcoded ingo@429: sascha@445: if (parameterId == null) { sascha@445: log.error("missing parameter id"); sascha@445: return null; sascha@445: } sascha@445: sascha@445: Map paletteManagers = sascha@445: getPalettes(callContext); sascha@445: sascha@445: PaletteManager paletteManager = paletteManagers.get(parameterId); sascha@445: sascha@445: if (paletteManager == null) { sascha@445: log.error("no palette found for parameter id " + parameterId); sascha@445: return null; sascha@445: } sascha@445: sascha@445: boolean debug = log.isDebugEnabled(); sascha@445: sascha@445: if (debug) { sascha@445: log.debug("using palette '" + paletteManager.getName() + "'"); sascha@445: } sascha@445: sascha@445: Dimension rasterSize = getRasterSize(callContext); sascha@445: List filterFactories = getFilterFactories(callContext); sascha@445: Interpolation3D interpolation = new Interpolation3D(rasterSize); sascha@445: sascha@446: double distance = DistanceCalculator.calculateDistance(path); sascha@445: sascha@445: if (distance < EPSILON) { sascha@445: log.warn("distance too short for interpolation"); sascha@445: return null; sascha@445: } sascha@445: sascha@445: boolean success = interpolation.interpolate( sascha@445: path, sascha@445: columns.getXYColumns(), sascha@445: 0d, sascha@445: distance, sascha@445: LinearMetrics.INSTANCE, sascha@463: new QueriedXYDepth( sascha@463: getGroundInterpolation(callContext))); sascha@445: sascha@445: if (!success) { sascha@445: log.warn("interpolation failed"); sascha@445: return null; sascha@445: } sascha@445: sascha@445: // Do the post processing sascha@445: Raster raster = new Raster( sascha@445: interpolation.getRaster(), sascha@445: rasterSize.width); sascha@445: sascha@445: for (Filter.Factory factory: filterFactories) { sascha@445: raster = factory.create().filter(raster); sascha@445: } sascha@445: sascha@445: if (debug) { sascha@445: log.debug("to indexed raster"); sascha@445: } sascha@445: sascha@445: // scan for regions with base palette sascha@448: Palette basePalette = paletteManager.getBase(); sascha@445: sascha@448: int [] intRaster = raster.toIndexed(basePalette); sascha@445: sascha@445: // produce JFreeChart compatible polygons sascha@445: sascha@445: if (debug) { sascha@445: log.debug("vectorize indexed raster"); sascha@445: } sascha@445: sascha@447: double maxDepth = interpolation.getMaxDepth(); sascha@447: sascha@445: PolygonDatasetProducer pdsp = new PolygonDatasetProducer( sascha@445: 0, 0, sascha@447: distance, maxDepth); sascha@445: sascha@447: int numRegions = new Vectorizer(intRaster, rasterSize.width) sascha@447: .process(pdsp); sascha@445: sascha@448: PolygonDataset pds = pdsp.getPolygonDataset(); sascha@446: sascha@521: // Count number of colors before generating seabed sascha@521: // because its used to determine the number of iso lines. sascha@448: int numColors = pds.getSeriesCount(); sascha@445: sascha@445: if (debug) { sascha@445: log.debug("number of regions: " + numRegions); sascha@448: log.debug("number of colors: " + numColors); sascha@445: } sascha@445: sascha@521: // generate seabed polygon sascha@521: sascha@521: PolygonSeries seabed = OutputHelper.createSeabedPolygon( sascha@521: interpolation, sascha@521: GROUND_FILL_INDEX); sascha@521: sascha@521: if (seabed != null) { sascha@521: pds.addSeries(seabed); sascha@521: } sascha@521: sascha@447: // generate iso lines sascha@447: sascha@448: int numIso; sascha@447: sascha@448: if (numColors < 5) { numIso = 5; } sascha@448: else if (numColors < 10) { numIso = 2; } sascha@448: else { numIso = 0; } sascha@448: sascha@448: Palette isoPalette; sascha@448: sascha@448: if (numIso == 0) { // same palette sascha@448: isoPalette = basePalette; sascha@448: /* intRaster = intRaster; */ sascha@448: } sascha@448: else { sascha@448: isoPalette = paletteManager.getLevel(numIso); sascha@448: intRaster = raster.toIndexed(isoPalette); sascha@448: } sascha@447: sascha@447: IsoPolygonSeriesProducer ipsp = new IsoPolygonSeriesProducer( sascha@447: 0, 0, sascha@447: distance, maxDepth); sascha@447: sascha@447: numRegions = new Vectorizer(false, intRaster, rasterSize.width) sascha@447: .process(ipsp); sascha@447: sascha@448: IsoAttributeGenerator iag = new IsoAttributeGenerator(isoPalette); sascha@448: Collection ps = ipsp.getSeries(iag); sascha@448: ipsp.clear(); sascha@447: sascha@447: if (debug) { sascha@447: log.debug("num of iso regions: " + numRegions); sascha@447: log.debug("num of iso series: " + ps.size()); sascha@447: } sascha@447: sascha@447: pds.addAllSeries(ps); sascha@447: sascha@446: columns.setInterpolation(interpolation); sascha@446: columns.setPolygonDataset(pds); sascha@446: sascha@446: return columns; ingo@429: } ingo@429: ingo@429: ingo@429: protected AttributedXYColumns preProcess(Collection results) { sascha@432: ingo@429: AttributedXYColumns attColumns = new AttributedXYColumns(); sascha@432: 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: sascha@439: if (!StringUtils.contains(ATTRIBUTE_LIST, colName)) { ingo@429: attColumns.setAttribute( ingo@429: colName, ingo@429: result.getObject(colName)); ingo@429: } ingo@429: } ingo@429: } ingo@429: sascha@432: double v = result.getDouble(vIdx); sascha@432: double z = result.getDouble(zIdx); sascha@432: int i = result.getInteger(iIdx); sascha@432: int j = result.getInteger(jIdx); sascha@432: int k = result.getInteger(kIdx); ingo@429: sascha@432: IJKey key = new IJKey(i, j); sascha@431: sascha@432: XYColumn col = (XYColumn)map.get(key); ingo@429: sascha@432: if (col == null) { sascha@432: Coordinate coord = WKTUtils.toCoordinate(result.getString(sIdx)); sascha@432: if (coord == null) coord = new Coordinate(); sascha@445: col = new XYColumn(coord.x, coord.y, i, j); sascha@432: map.put(key, col); ingo@429: } sascha@432: sascha@432: col.add(new HeightValue(z, v, k)); ingo@429: } ingo@429: sascha@432: ArrayList cols = new ArrayList(map.values()); ingo@429: attColumns.setXYColumns(cols); ingo@429: ingo@429: return attColumns; 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, sascha@439: boolean shapesVisible, sascha@439: CallContext callContext 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."); sascha@439: chart = (Chart) getChartFromCache(uuid, callContext); 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: sascha@462: if (!(result instanceof AttributedXYColumns)) { sascha@462: log.error("result of wrong type"); sascha@462: return null; sascha@462: } sascha@462: sascha@521: AttributedXYColumns columns = (AttributedXYColumns)result; sascha@446: sascha@446: Integer parameterId = sascha@446: (Integer)columns.getAttribute("GROUP1"); // XXX: hardcoded sascha@446: sascha@446: if (parameterId == null) { sascha@446: log.error("missing parameter id"); sascha@446: return null; sascha@446: } sascha@446: sascha@446: Map paletteManagers = sascha@446: getPalettes(callContext); sascha@446: sascha@446: PaletteManager paletteManager = paletteManagers.get(parameterId); sascha@446: sascha@446: if (paletteManager == null) { sascha@446: log.error("no palette found for parameter id " + parameterId); sascha@446: return null; sascha@446: } sascha@446: sascha@521: HashMap special = new HashMap(); sascha@521: special.put(GROUND_FILL_INDEX, getGroundFill(callContext)); sascha@521: sascha@446: chart = new VerticalCrossSectionChart( sascha@446: columns, sascha@446: paletteManager.getBase(), sascha@521: special, ingo@492: locale, ingo@492: chartLables); sascha@446: tim@335: chart.generateChart(); ingo@527: ((VerticalCrossSectionChart)chart).setBackgroundPaint(Color.WHITE); 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: return chart; 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() { sascha@454: return new VerticalCrossSectionStatistics(); 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 sascha@432: protected void createCSV( sascha@432: OutputStream outputStream, sascha@432: Collection chartResult sascha@432: ) sascha@432: throws UnsupportedEncodingException, IOException, StateException sascha@432: { sascha@521: // TODO: Implement a substitution which makes sense. tim@335: } tim@335: } sascha@521: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :