/[schmitzm]/trunk/src/skrueger/geotools/StyledLayerUtil.java
ViewVC logotype

Contents of /trunk/src/skrueger/geotools/StyledLayerUtil.java

Parent Directory Parent Directory | Revision Log Revision Log


Revision 666 - (show annotations)
Wed Feb 3 18:05:50 2010 UTC (15 years ago) by alfonx
Original Path: branches/2.0-RC2/src/skrueger/geotools/StyledLayerUtil.java
File size: 44300 byte(s)
After importing FeatureChartStyles, their attributes names are checked against the schema
1 /*******************************************************************************
2 * Copyright (c) 2009 Martin O. J. Schmitz.
3 *
4 * This file is part of the SCHMITZM library - a collection of utility
5 * classes based on Java 1.6, focusing (not only) on Java Swing
6 * and the Geotools library.
7 *
8 * The SCHMITZM project is hosted at:
9 * http://wald.intevation.org/projects/schmitzm/
10 *
11 * This program is free software; you can redistribute it and/or
12 * modify it under the terms of the GNU Lesser General Public License
13 * as published by the Free Software Foundation; either version 3
14 * of the License, or (at your option) any later version.
15 *
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
20 *
21 * You should have received a copy of the GNU Lesser General Public License (license.txt)
22 * along with this program; if not, write to the Free Software
23 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
24 * or try this link: http://www.gnu.org/licenses/lgpl.html
25 *
26 * Contributors:
27 * Martin O. J. Schmitz - initial API and implementation
28 * Stefan A. Krüger - additional utility classes
29 ******************************************************************************/
30 package skrueger.geotools;
31
32 import java.awt.Color;
33 import java.awt.Dimension;
34 import java.awt.Graphics2D;
35 import java.awt.Rectangle;
36 import java.awt.geom.AffineTransform;
37 import java.awt.image.BufferedImage;
38 import java.awt.image.ColorModel;
39 import java.awt.image.ComponentColorModel;
40 import java.awt.image.DataBuffer;
41 import java.io.File;
42 import java.io.FileNotFoundException;
43 import java.io.FileWriter;
44 import java.net.URL;
45 import java.text.DecimalFormat;
46 import java.util.ArrayList;
47 import java.util.List;
48 import java.util.Map;
49
50 import javax.swing.BorderFactory;
51 import javax.swing.ImageIcon;
52 import javax.swing.JComponent;
53 import javax.swing.JLabel;
54
55 import net.miginfocom.swing.MigLayout;
56
57 import org.apache.log4j.Logger;
58 import org.geotools.coverage.grid.GeneralGridEnvelope;
59 import org.geotools.coverage.grid.GridCoverage2D;
60 import org.geotools.coverage.grid.GridGeometry2D;
61 import org.geotools.coverage.grid.io.AbstractGridCoverage2DReader;
62 import org.geotools.coverage.grid.io.AbstractGridFormat;
63 import org.geotools.feature.FeatureCollection;
64 import org.geotools.feature.NameImpl;
65 import org.geotools.geometry.jts.ReferencedEnvelope;
66 import org.geotools.map.DefaultMapLayer;
67 import org.geotools.map.MapLayer;
68 import org.geotools.parameter.Parameter;
69 import org.geotools.renderer.lite.gridcoverage2d.GridCoverageRenderer;
70 import org.geotools.styling.ColorMap;
71 import org.geotools.styling.ColorMapEntry;
72 import org.geotools.styling.FeatureTypeStyle;
73 import org.geotools.styling.RasterSymbolizer;
74 import org.geotools.styling.Rule;
75 import org.geotools.styling.Style;
76 import org.jdom.Document;
77 import org.jdom.Element;
78 import org.jdom.input.SAXBuilder;
79 import org.jdom.output.XMLOutputter;
80 import org.opengis.feature.simple.SimpleFeature;
81 import org.opengis.feature.simple.SimpleFeatureType;
82 import org.opengis.feature.type.AttributeDescriptor;
83 import org.opengis.feature.type.GeometryDescriptor;
84 import org.opengis.feature.type.Name;
85 import org.opengis.parameter.GeneralParameterValue;
86
87 import schmitzm.geotools.JTSUtil;
88 import schmitzm.geotools.feature.FeatureUtil;
89 import schmitzm.geotools.styling.StylingUtil;
90 import schmitzm.io.IOUtil;
91 import schmitzm.lang.LangUtil;
92 import schmitzm.swing.JPanel;
93 import schmitzm.swing.SwingUtil;
94 import skrueger.AttributeMetadata;
95 import skrueger.RasterLegendData;
96 import skrueger.i8n.Translation;
97
98 /**
99 * This class provides static helper methods for dealing with
100 * {@link StyledLayerInterface} stuff.
101 *
102 * @author <a href="mailto:[email protected]">Martin Schmitz</a>
103 * (University of Bonn/Germany)
104 * @version 1.0
105 */
106 public class StyledLayerUtil {
107 private static final Logger LOGGER = Logger.getLogger(StyledLayerUtil.class
108 .getName());
109 private static final SAXBuilder SAX_BUILDER = new SAXBuilder();
110 private static final XMLOutputter XML_OUTPUTTER = new XMLOutputter();
111
112 /** URL for Atlas XML schema */
113 public static final String AMLURI = "http://www.wikisquare.de/AtlasML";
114 /** Name of the XML Element for the attribute meta data map */
115 public static final String ELEM_NAME_AMD = "attributeMetaData";
116 /** Name of the XML Element for the raster legend data */
117 public static final String ELEM_NAME_RLD = "rasterLegendData";
118 /** Name of the XML Element for an attribute meta data map entry */
119 public static final String ELEM_NAME_ATTRIBUTE = "dataAttribute";
120 /** Name of the XML Element for an raster legend data entry */
121 public static final String ELEM_NAME_RASTERLEGEND = "rasterLegendItem";
122 /** Name of the XML Element for a translation */
123 public static final String ELEM_NAME_TRANSLATION = "translation";
124
125 /**
126 * Creates a Geotools {@link MapLayer} from an object. If the object is a
127 * {@link StyledLayerInterface} then its sytle is used. In case of direct
128 * Geotools objects ({@link GridCoverage2D},
129 * {@link AbstractGridCoverage2DReader}, {@link FeatureCollection}) a
130 * default style is generated.
131 *
132 * @param object
133 * an Object
134 * @exception Exception
135 * if {@code null} is given as object or an error occurs
136 * during layer creation
137 */
138 public static MapLayer createMapLayer(final Object object) throws Exception {
139 return createMapLayer(object, null);
140 }
141
142 /**
143 * Creates a Geotools {@link MapLayer} from an object. If the object is a
144 * {@link StyledLayerInterface} then its sytle is used. In case of direct
145 * Geotools objects ({@link GridCoverage2D},
146 * {@link AbstractGridCoverage2DReader}, {@link FeatureCollection}) a
147 * default style is generated.
148 *
149 * @param object
150 * an Object
151 * @param forcedStyle
152 * (SLD-)Style to force for the object
153 * @exception Exception
154 * if {@code null} is given as object or an error occurs
155 * during layer creation
156 */
157 public static MapLayer createMapLayer(Object object, final Style forcedStyle)
158 throws Exception {
159 MapLayer layer = null;
160 Style style = null;
161 if (object instanceof StyledLayerInterface) {
162 style = ((StyledLayerInterface<?>) object).getStyle();
163 object = ((StyledLayerInterface<?>) object).getGeoObject();
164 }
165 if (forcedStyle != null)
166 style = forcedStyle;
167 if (style == null)
168 style = StylingUtil.createDefaultStyle(object);
169
170 if (object instanceof GridCoverage2D)
171 layer = new DefaultMapLayer((GridCoverage2D) object, style);
172 if (object instanceof AbstractGridCoverage2DReader)
173 layer = new DefaultMapLayer((AbstractGridCoverage2DReader) object,
174 style);
175 if (object instanceof FeatureCollection)
176 layer = new DefaultMapLayer((FeatureCollection) object, style);
177
178 if (layer == null)
179 throw new Exception("Can not create MapLayer from "
180 + (object == null ? "null" : object.getClass()));
181
182 return layer;
183 }
184
185 /**
186 * Creates an default instance of {@link StyledLayerInterface} for a
187 * Geotools object ({@link GridCoverage2D}, {@link FeatureCollection}) with
188 * a default style.
189 *
190 * @param object
191 * an Object
192 * @param title
193 * title for the object
194 * @exception UnsupportedOperationException
195 * if {@code null} is given as object or an error occurs
196 * during creation
197 */
198 public static StyledLayerInterface<?> createStyledLayer(
199 final Object object, final String title) {
200 return createStyledLayer(object, title, null);
201 }
202
203 /**
204 * Creates an default instance of {@link StyledLayerInterface} for a
205 * Geotools object ({@link GridCoverage2D}, {@link FeatureCollection}) with
206 * a given style.
207 *
208 * @param object
209 * an Object
210 * @param title
211 * title for the object
212 * @param style
213 * style and meta data for the object
214 * @exception UnsupportedOperationException
215 * if {@code null} is given as object or an error occurs
216 * during creation
217 */
218 public static StyledLayerInterface<?> createStyledLayer(
219 final Object object, final String title,
220 final StyledLayerStyle style) {
221 StyledLayerInterface<?> styledLayer = null;
222
223 final String id = (title != null) ? title : "defaultID";
224
225 if (object instanceof GridCoverage2D)
226 styledLayer = new StyledGridCoverage((GridCoverage2D) object, id,
227 title, style);
228 else if (object instanceof AbstractGridCoverage2DReader)
229 styledLayer = new StyledGridCoverageReader(
230 (AbstractGridCoverage2DReader) object, id, title, style);
231 else if (object instanceof FeatureCollection)
232 styledLayer = new StyledFeatureCollection(
233 (FeatureCollection) object, id, title, style);
234
235 if (styledLayer == null)
236 throw new UnsupportedOperationException(
237 "Can not create StyledLayerInterface object from "
238 + (object == null ? "null" : object.getClass()));
239
240 return styledLayer;
241 }
242
243 /**
244 * Return only the visible or invisible entries of an AttributeMetaData-Map.
245 *
246 * @param amdMap
247 * AttributeMetaData-Map
248 * @param visible
249 * indicated whether the visible or invisible entries are
250 * returned
251 *
252 * TODO replace with
253 * {@link AttributeMetadataMap#sortedValuesVisibleOnly()}
254 */
255 public static AttributeMetadataMap getVisibleAttributeMetaData(
256 final AttributeMetadataMap amdMap, final boolean visible) {
257
258 final AttributeMetadataMap filteredMap = new AttributeMetadataMap(
259 amdMap.getLanguages());
260 for (final AttributeMetadata amd : amdMap.values())
261 if (amd.isVisible() == visible)
262 filteredMap.put(amd.getName(), amd);
263
264 return filteredMap;
265 }
266
267 /**
268 * Parses a {@link AttributeMetadata} object from an JDOM-{@link Element}.
269 * This method works like {@link
270 * AMLImport#parseDataAttribute(org.w3c.dom.Node}, but for JDOM.
271 *
272 * TODO 20.11.2009, SK: There are some new attribute weight, functiona,
273 * functionX and nodata in AttributeMetaData that should be parsed/exported
274 * too. but this method is only used by ISDSS, which is not supporting that
275 * stuff anyways.
276 *
277 * @param element
278 * {@link Element} to parse
279 */
280 public static AttributeMetadata parseAttributeMetaData(final Element element) {
281 final String namespace = String.valueOf(element
282 .getAttributeValue("namespace"));
283 final String localname = String.valueOf(element
284 .getAttributeValue("localname"));
285 final Name aName = new NameImpl(namespace, localname);
286 final Boolean visible = Boolean.valueOf(element
287 .getAttributeValue("visible"));
288 final String unit = element.getAttributeValue("unit");
289
290 Translation name = new Translation();
291 Translation desc = new Translation();
292 for (final Element childElement : (List<Element>) element.getChildren()) {
293 if (childElement.getName() == null)
294 continue;
295
296 if (childElement.getName().equals("name"))
297 name = parseTranslation(childElement);
298 else if (childElement.getName().equals("desc"))
299 desc = parseTranslation(childElement);
300 }
301 return new AttributeMetadata(aName, visible, name, desc, unit);
302 }
303
304 /**
305 * Parses a {@link AttributeMetadata} map from an JDOM-{@link Element} with
306 * {@code <attribute>}-childs.
307 *
308 * @param element
309 * {@link Element} to parse
310 *
311 * TODO Since GP 1.3 the {@link AttributeMetadata} class has more
312 * attributes which are not used by Xulu/ISDSS. GP
313 * exports/imports the AMD via AMLExporter and AMLImporter
314 * classes. (SK, 3.2.2010) *
315 */
316 public static AttributeMetadataMap parseAttributeMetaDataMap(
317 final Element element) {
318 final AttributeMetadataMap metaData = new AttributeMetadataMap();
319 final List<Element> attributesElements = element
320 .getChildren(ELEM_NAME_ATTRIBUTE);
321 for (final Element attibuteElement : attributesElements) {
322 final AttributeMetadata attrMetaData = parseAttributeMetaData(attibuteElement);
323 metaData.put(attrMetaData.getName(), attrMetaData);
324 }
325 return metaData;
326 }
327
328 /**
329 * Loads a {@link AttributeMetadata} object from an URL.
330 *
331 * @param documentUrl
332 * {@link URL} to parse
333 * @see #parseAttributeMetaData(Element)
334 */
335 public static AttributeMetadataMap loadAttributeMetaDataMap(
336 final URL documentUrl) throws Exception {
337 final Document document = SAX_BUILDER.build(documentUrl);
338 return parseAttributeMetaDataMap(document.getRootElement());
339 }
340
341 /**
342 * Creates an JDOM {@link Element} for the given {@link AttributeMetadata}
343 * object.
344 *
345 * @param amd
346 * meta data for one attribute
347 *
348 * TODO Since GP 1.3 the {@link AttributeMetadata} class has more
349 * attributes which are not used by Xulu/ISDSS. GP
350 * exports/imports the AMD via AMLExporter and AMLImporter
351 * classes. (SK, 3.2.2010)
352 */
353 public static Element createAttributeMetaDataElement(
354 final AttributeMetadata amd) {
355 final Element element = new Element(ELEM_NAME_ATTRIBUTE, AMLURI);
356 element.setAttribute("namespace", String.valueOf(amd.getName()
357 .getNamespaceURI()));
358 element.setAttribute("localname", String.valueOf(amd.getLocalName()));
359 element.setAttribute("visible", String.valueOf(amd.isVisible()));
360 element.setAttribute("unit", amd.getUnit());
361 // Creating a aml:name tag...
362 element.addContent(createTranslationElement("name", amd.getTitle()));
363 // Creating a aml:desc tag...
364 element.addContent(createTranslationElement("desc", amd.getDesc()));
365 return element;
366 }
367
368 /**
369 * Creates an JDOM {@link Element} for the given {@link AttributeMetadata}
370 * map.
371 *
372 * @param amdMap
373 * map of attribute meta data
374 */
375 public static Element createAttributeMetaDataMapElement(
376 final AttributeMetadataMap amdMap) {
377 final Element element = new Element(ELEM_NAME_AMD, AMLURI);
378 for (final AttributeMetadata amd : amdMap.values())
379 element.addContent(createAttributeMetaDataElement(amd));
380 return element;
381 }
382
383 /**
384 * Saves a {@link AttributeMetadata AttributeMetaData-Map} to an URL.
385 *
386 * @param amdMap
387 * map of {@link AttributeMetadata}
388 * @param documentUrl
389 * {@link URL} to store the XML
390 */
391 public static void saveAttributeMetaDataMap(
392 final AttributeMetadataMap amdMap, final URL documentUrl)
393 throws Exception {
394 // Create XML-Document
395 final FileWriter out = new FileWriter(new File(documentUrl.toURI()));
396 XML_OUTPUTTER.output(createAttributeMetaDataMapElement(amdMap), out);
397 out.flush();
398 out.close();
399 }
400
401 /**
402 * Parses a {@link RasterLegendData} object from an JDOM-{@link Element}.
403 * This method works like {@link
404 * AMLImport#parseRasterLegendData(org.w3c.dom.Node}, but for JDOM.
405 *
406 * @param element
407 * {@link Element} to parse
408 */
409 public static RasterLegendData parseRasterLegendData(final Element element) {
410
411 final boolean paintGaps = Boolean.valueOf(element
412 .getAttributeValue("paintGaps"));
413
414 final RasterLegendData rld = new RasterLegendData(paintGaps);
415
416 for (final Element childElement : (List<Element>) element.getChildren()) {
417 final String name = childElement.getName();
418 // Cancel if it's an attribute
419 if (childElement.getChildren().size() == 0)
420 continue;
421
422 if (name.equals(ELEM_NAME_RASTERLEGEND)) {
423 final String valueAttr = childElement
424 .getAttributeValue("value");
425 if (valueAttr == null)
426 throw new UnsupportedOperationException(
427 "Attribute 'value' missing for definition of <"
428 + ELEM_NAME_RASTERLEGEND + ">");
429 final double value = Double.valueOf(valueAttr);
430
431 // first and only item should be the label
432 final Element labelElement = childElement.getChild("label");
433 // id label element is missing, the translation is searched
434 // directly
435 // as childs of the rasterLegendItem element
436 final Translation label = parseTranslation(labelElement != null ? labelElement
437 : childElement);
438 rld.put(value, label);
439 }
440 }
441
442 return rld;
443 }
444
445 /**
446 * Loads a {@link RasterLegendData} object from an URL.
447 *
448 * @param documentUrl
449 * {@link URL} to parse
450 * @see #parseAttributeMetaData(Element)
451 */
452 public static RasterLegendData loadRasterLegendData(final URL documentUrl)
453 throws Exception {
454 final Document document = SAX_BUILDER.build(documentUrl);
455 return parseRasterLegendData(document.getRootElement());
456 }
457
458 /**
459 * Creates an JDOM {@link Element} for the given {@link RasterLegendData}
460 * map.
461 *
462 * @param rld
463 * raster legend data
464 */
465 public static Element createRasterLegendDataElement(
466 final RasterLegendData rld) {
467 final Element element = new Element(ELEM_NAME_RLD, AMLURI);
468 element.setAttribute("paintGaps", rld.isPaintGaps().toString());
469 for (final Double key : rld.getSortedKeys()) {
470 final Element item = new Element(ELEM_NAME_RASTERLEGEND, AMLURI);
471 item.setAttribute("value", key.toString());
472 item.addContent(createTranslationElement("label", rld.get(key)));
473 element.addContent(item);
474 }
475 return element;
476 }
477
478 /**
479 * Creates {@link RasterLegendData} from a {@link ColorMap}.
480 *
481 * @param colorMap
482 * a color map
483 * @param paintGaps
484 * indicated whether gaps are painted between the legend items
485 * @param digits
486 * number of digits the grid value classes (and legend) are
487 * rounded to (null means no round; >= 0 means digits after
488 * comma; < 0 means digits before comma)
489 */
490 public static RasterLegendData generateRasterLegendData(
491 final ColorMap colorMap, final boolean paintGaps,
492 final Integer digits) {
493 final DecimalFormat decFormat = digits != null ? new DecimalFormat(
494 SwingUtil.getNumberFormatPattern(digits)) : null;
495 final RasterLegendData rld = new RasterLegendData(paintGaps);
496 for (final ColorMapEntry cme : colorMap.getColorMapEntries()) {
497 final double value = StylingUtil.getQuantityFromColorMapEntry(cme);
498 String label = cme.getLabel();
499 // if no label is set (e.g. quantitative style),
500 // use the value as label
501 if (label == null || label.equals(""))
502 if (digits == null)
503 label = String.valueOf(value);
504 else
505 label = decFormat.format(LangUtil.round(value, digits));
506 rld.put(value, new Translation(" " + label));
507 }
508 return rld;
509 }
510
511 /**
512 * Creates {@link RasterLegendData} from the {@link ColorMap} of a style.
513 *
514 * @param style
515 * a raster style (must contain a {@link RasterSymbolizer})
516 * @param paintGaps
517 * indicated whether gaps are painted between the legend items
518 * @param digits
519 * number of digits the grid value classes (and legend) are
520 * rounded to (null means no round; >= 0 means digits after
521 * comma; < 0 means digits before comma)
522 */
523 public static RasterLegendData generateRasterLegendData(final Style style,
524 final boolean paintGaps, final Integer digits) {
525 final ColorMap colorMap = StylingUtil.getColorMapFromStyle(style);
526 if (colorMap == null)
527 throw new IllegalArgumentException(
528 "Color map can not be determined from style!");
529 return generateRasterLegendData(colorMap, paintGaps, digits);
530 }
531
532 /**
533 * Saves a {@link RasterLegendData} to an URL.
534 *
535 * @param rld
536 * raster legend data
537 * @param documentUrl
538 * {@link URL} to store the XML
539 */
540 public static void saveRasterLegendData(final RasterLegendData rld,
541 final URL documentUrl) throws Exception {
542 // Create XML-Document
543 final FileWriter out = new FileWriter(new File(documentUrl.toURI()));
544 XML_OUTPUTTER.output(createRasterLegendDataElement(rld), out);
545 out.flush();
546 out.close();
547 }
548
549 /**
550 * Parses a {@link Translation} object from an JDOM-{@link Element}. This
551 * method works like {@link AMLImport#parseTranslation(org.w3c.dom.Node},
552 * but for JDOM.
553 *
554 * @param element
555 * {@link Element} to parse
556 */
557 public final static Translation parseTranslation(final Element element) {
558 final Translation trans = new Translation();
559
560 if (element == null)
561 return trans;
562
563 for (final Element translationElement : (List<Element>) element
564 .getChildren()) {
565 final String name = translationElement.getName();
566 if (name == null)
567 continue;
568
569 // lang attribute
570 String lang = translationElement.getAttributeValue("lang");
571 // set the default, if no language code is set
572 if (lang == null)
573 lang = Translation.DEFAULT_KEY;
574
575 final String translationText = translationElement.getValue();
576 if (translationText == null)
577 trans.put(lang, "");
578 else
579 trans.put(lang, translationText);
580 }
581
582 // if no <translation> is given, the value of the node should
583 // be used as a default translation
584 if (trans.size() == 0)
585 trans.put(Translation.DEFAULT_KEY, element.getValue());
586 // trans = new Translation(
587 // ((List<Element>)element.getChildren()).get(0).getValue() );
588
589 return trans;
590 }
591
592 /**
593 * Creates an JDOM {@link Element} for the given {@link Translation}.
594 *
595 * @param tagname
596 * Name of the Element
597 * @param translation
598 * Translation to store in the Element
599 */
600 public final static Element createTranslationElement(final String tagname,
601 final Translation translation) {
602 final Element element = new Element(tagname, AMLURI);
603 if (translation == null)
604 throw new UnsupportedOperationException(
605 "Translation element can not be created from null!");
606
607 // If only a default translation is set, the <translation
608 // lang="..">..</tranlation>
609 // part is not used
610 if (translation.keySet().size() == 1
611 && translation.get(Translation.DEFAULT_KEY) != null) {
612 element.addContent(translation.get(Translation.DEFAULT_KEY));
613 return element;
614 }
615
616 // add a <translation lang="..">..</tranlation> part to the element for
617 // all languages
618 for (final String lang : translation.keySet()) {
619 final Element translationElement = new Element(
620 ELEM_NAME_TRANSLATION, AMLURI);
621 translationElement.setAttribute("lang", lang);
622 String translationString = translation.get(lang);
623 if (translationString == null)
624 translationString = "";
625 translationElement.addContent(translationString);
626 element.addContent(translationElement);
627 }
628
629 return element;
630 }
631
632 /**
633 * Sets a style to {@link StyledLayerInterface}.
634 *
635 * @param styledObject
636 * a styled object
637 * @param style
638 * a Style
639 */
640 public static void setStyledLayerStyle(
641 final StyledLayerInterface styledObject,
642 final StyledLayerStyle<?> style) {
643 // set SLD style
644 styledObject.setStyle(style.getGeoObjectStyle());
645 // set meta data
646 if (styledObject instanceof StyledGridCoverageInterface
647 && (style.getMetaData() instanceof RasterLegendData || style
648 .getMetaData() == null)) {
649 final RasterLegendData sourceRld = (RasterLegendData) style
650 .getMetaData();
651 final RasterLegendData destRld = ((StyledGridCoverageInterface) styledObject)
652 .getLegendMetaData();
653 if (destRld != null && sourceRld != null) {
654 destRld.setPaintGaps(sourceRld.isPaintGaps());
655 destRld.clear();
656 destRld.putAll(sourceRld);
657 }
658 return;
659 }
660 if (styledObject instanceof StyledFeatureCollectionInterface
661 && (style.getMetaData() instanceof Map || style.getMetaData() == null)) {
662 final AttributeMetadataMap sourceAmd = (AttributeMetadataMap) style
663 .getMetaData();
664 final AttributeMetadataMap destAmd = ((StyledFeatureCollectionInterface) styledObject)
665 .getAttributeMetaDataMap();
666 if (destAmd != null && sourceAmd != null) {
667 destAmd.clear();
668 destAmd.putAll(sourceAmd);
669 }
670 return;
671 }
672
673 throw new UnsupportedOperationException(
674 "Style is not compatible to object: "
675 + (style.getMetaData() == null ? null : style
676 .getMetaData().getClass().getSimpleName())
677 + " <-> "
678 + (styledObject == null ? null : styledObject
679 .getClass().getSimpleName()));
680 }
681
682 /**
683 * Returns the style a {@link StyledLayerInterface} as a
684 * {@link StyledLayerStyle}.
685 *
686 * @param styledObject
687 * a styled object
688 * @return {@code StyledLayerStyle<RasterLegendData>} for
689 * {@link StyledGridCoverageInterface} or {@code
690 * StyledLayerStyle<Map<Integer,AttributeMetaData>>} for
691 * {@link StyledFeatureCollectionInterface}
692 */
693 public static StyledLayerStyle<?> getStyledLayerStyle(
694 final StyledLayerInterface styledObject) {
695 if (styledObject instanceof StyledGridCoverageInterface)
696 return getStyledLayerStyle((StyledGridCoverageInterface) styledObject);
697 if (styledObject instanceof StyledFeatureCollectionInterface)
698 return getStyledLayerStyle((StyledFeatureCollectionInterface) styledObject);
699 throw new UnsupportedOperationException(
700 "Unknown type of StyledLayerInterface: "
701 + (styledObject == null ? null : styledObject
702 .getClass().getSimpleName()));
703 }
704
705 /**
706 * Returns the style and raster meta data of a
707 * {@link StyledGridCoverageInterface} as a {@link StyledLayerStyle}.
708 *
709 * @param styledGC
710 * a styled grid coverage
711 */
712 public static StyledLayerStyle<RasterLegendData> getStyledLayerStyle(
713 final StyledGridCoverageInterface styledGC) {
714 return new StyledLayerStyle<RasterLegendData>(styledGC.getStyle(),
715 styledGC.getLegendMetaData());
716 }
717
718 /**
719 * Returns the style and attribute meta data of a
720 * {@link StyledFeatureCollectionInterface} as a {@link StyledLayerStyle}.
721 *
722 * @param styledFC
723 * a styled feature collection
724 */
725 public static StyledLayerStyle<AttributeMetadataMap> getStyledLayerStyle(
726 final StyledFeatureCollectionInterface styledFC) {
727 return new StyledLayerStyle<AttributeMetadataMap>(styledFC.getStyle(),
728 styledFC.getAttributeMetaDataMap());
729 }
730
731 /**
732 * Loads a {@linkplain Style SLD-Style} and {@linkplain RasterLegendData
733 * Raster-LegendData} for a given geo-object (raster) source. The SLD file
734 * must be present. A missing raster legend-data file is tolerated.
735 *
736 * @param geoObjectURL
737 * URL of the (already read) raster object
738 * @param sldExt
739 * file extention for the SLD file
740 * @param rldExt
741 * file extention for the raster legend-data file
742 * @return {@code null} in case of any error
743 */
744 public static StyledLayerStyle<RasterLegendData> loadStyledRasterStyle(
745 final URL geoObjectURL, final String sldExt, final String rldExt) {
746 RasterLegendData metaData = null;
747 Style sldStyle = null;
748 try {
749 final Style[] styles = StylingUtil.loadSLD(IOUtil.changeUrlExt(
750 geoObjectURL, sldExt));
751 // SLD must be present
752 if (styles == null || styles.length == 0)
753 return null;
754 sldStyle = styles[0];
755 } catch (final Exception err) {
756 // SLD must be present
757 LangUtil.logDebugError(LOGGER, err);
758 return null;
759 }
760
761 try {
762 metaData = StyledLayerUtil.loadRasterLegendData(IOUtil
763 .changeUrlExt(geoObjectURL, rldExt));
764 } catch (final FileNotFoundException err) {
765 // ignore missing raster legend data
766 } catch (final Exception err) {
767 // any other error during legend data creation leads to error
768 LangUtil.logDebugError(LOGGER, err);
769 return null;
770 }
771 return new StyledLayerStyle<RasterLegendData>(sldStyle, metaData);
772 }
773
774 /**
775 * Loads a {@linkplain Style SLD-Style} from a {@code .sld} file and
776 * {@linkplain RasterLegendData Raster-LegendData} from a {@code .rld} file
777 * for a given geo-object (raster) source. The SLD file must be present. A
778 * missing raster legend-data file is tolerated.
779 *
780 * @param geoObjectURL
781 * URL of the (already read) raster object
782 * @param sldExt
783 * file extention for the SLD file
784 * @param rldExt
785 * file extention for the raster legend-data file
786 * @return {@code null} in case of any error
787 */
788 public static StyledLayerStyle<RasterLegendData> loadStyledRasterStyle(
789 final URL geoObjectURL) {
790 return loadStyledRasterStyle(geoObjectURL, "sld", "rld");
791 }
792
793 /**
794 * Loads a {@linkplain Style SLD-Style} and a {@linkplain AttributeMetadata
795 * AttributeMetaData-Map} for a given geo-object (feature) source. The SLD
796 * file must be present. A missing attribute meta-data file is tolerated.
797 *
798 * @param geoObjectURL
799 * URL of the (already read) feature object
800 * @param sldExt
801 * file extention for the SLD file
802 * @param rldExt
803 * file extention for the raster legend-data file
804 * @return {@code null} in case of any error
805 */
806 public static StyledLayerStyle<AttributeMetadataMap> loadStyledFeatureStyle(
807 final URL geoObjectURL, final String sldExt, final String rldExt) {
808 AttributeMetadataMap metaData = null;
809 Style sldStyle = null;
810 try {
811 final Style[] styles = StylingUtil.loadSLD(IOUtil.changeUrlExt(
812 geoObjectURL, sldExt));
813 // SLD must be present
814 if (styles == null || styles.length == 0)
815 return null;
816 sldStyle = styles[0];
817 } catch (final Exception err) {
818 // SLD must be present
819 LangUtil.logDebugError(LOGGER, err);
820 return null;
821 }
822
823 try {
824 metaData = StyledLayerUtil.loadAttributeMetaDataMap(IOUtil
825 .changeUrlExt(geoObjectURL, rldExt));
826 } catch (final FileNotFoundException err) {
827 // ignore missing attribute meta data
828 } catch (final Exception err) {
829 // any other error during meta data creation leads to error
830 LangUtil.logDebugError(LOGGER, err);
831 return null;
832 }
833
834 return new StyledLayerStyle<AttributeMetadataMap>(sldStyle, metaData);
835 }
836
837 /**
838 * Loads a {@linkplain Style SLD-Style} from a {@code .sld} file and
839 * {@linkplain AttributeMetadata AttributeMetaData-Map} from a {@code .amd}
840 * file for a given geo-object (feature) source. The SLD file must be
841 * present. A missing attribute meta-data file is tolerated.
842 *
843 * @param geoObjectURL
844 * URL of the (already read) feature object
845 * @param sldExt
846 * file extention for the SLD file
847 * @param rldExt
848 * file extention for the raster legend-data file
849 * @return {@code null} in case of any error
850 */
851 public static StyledLayerStyle<AttributeMetadataMap> loadStyledFeatureStyle(
852 final URL geoObjectURL) {
853 return loadStyledFeatureStyle(geoObjectURL, "sld", "amd");
854 }
855
856 /**
857 * Stores a {@linkplain Style SLD-Style} and {@linkplain RasterLegendData
858 * Raster-LegendData} for a given geo-object (raster) source.
859 *
860 * @param style
861 * style to save
862 * @param geoObjectURL
863 * URL of the raster object
864 * @param sldExt
865 * file extention for the SLD file
866 * @param mdExt
867 * file extention for the meta-data file
868 */
869 public static <T> void saveStyledLayerStyle(
870 final StyledLayerStyle<T> style, final URL geoObjectURL,
871 final String sldExt, final String mdExt) throws Exception {
872 // Store the SLD
873 final Style sldStyle = style.getGeoObjectStyle();
874 if (sldStyle != null) {
875 StylingUtil.saveStyleToSLD(sldStyle, IOUtil.changeFileExt(new File(
876 geoObjectURL.toURI()), sldExt));
877 }
878
879 // Store the meta data
880 final T metaData = style.getMetaData();
881 if (metaData != null) {
882 if (metaData instanceof RasterLegendData) {
883 saveRasterLegendData((RasterLegendData) metaData, IOUtil
884 .changeUrlExt(geoObjectURL, mdExt));
885 // } else if ( metaData instanceof
886 // Map<Integer,AttributeMetaData> ) { // LEIDER NICHT
887 // KOMPILIERBAR!!
888 } else if (metaData instanceof Map) {
889 saveAttributeMetaDataMap((AttributeMetadataMap) metaData,
890 IOUtil.changeUrlExt(geoObjectURL, mdExt));
891 } else
892 throw new UnsupportedOperationException(
893 "Export for meta data not yet supported: "
894 + metaData.getClass().getSimpleName());
895 }
896 }
897
898 /**
899 * Stores the {@linkplain Style SLD-Style} to a {@code .sld} file and the
900 * meta data ({@link RasterLegendData} or {@link AttributeMetadata}) to a
901 * {@code .rld} or {@code .amd} file. for a given geo-object source.
902 *
903 * @param style
904 * style to save
905 * @param geoObjectURL
906 * URL of the (already read) raster object
907 */
908 public static void saveStyledLayerStyle(final StyledLayerStyle<?> style,
909 final URL geoObjectURL) throws Exception {
910 if (style.getMetaData() instanceof RasterLegendData)
911 saveStyledLayerStyle(style, geoObjectURL, "sld", "rld");
912 else
913 saveStyledLayerStyle(style, geoObjectURL, "sld", "amd");
914 }
915
916 /**
917 * Creates a {@link JPanel} that shows a legend for a list of
918 * {@link FeatureTypeStyle}s and a targeted featureType
919 *
920 * @param featureType
921 * If this a legend for Point, Polygon or Line?
922 * @param list
923 * The Styles to presented in this legend
924 *
925 * @author <a href="mailto:[email protected]">Stefan Alfons
926 * Kr&uuml;ger</a>
927 */
928 public static JPanel createLegendPanel(Style style,
929 final SimpleFeatureType featureType, final int iconWidth,
930 final int iconHeight) {
931
932 final List<FeatureTypeStyle> list = style.featureTypeStyles();
933
934 final JPanel panel = new JPanel(new MigLayout("wrap 2", "[]:3:[]"));
935
936 if (style == null) {
937 // No Style => no legend
938 return panel;
939 }
940
941 for (final FeatureTypeStyle ftStyle : list) {
942
943 // One child-node for every rule
944 final List<Rule> rules = ftStyle.rules();
945 for (final Rule rule : rules) {
946
947 /**
948 * Let's not create a hbox for Rules that only contain
949 * TextSymbolizers
950 */
951 if (StylingUtil.getTextSymbolizers(rule.getSymbolizers())
952 .size() == rule.getSymbolizers().length)
953 continue;
954
955 final BufferedImage imageForRule = LegendIconFeatureRenderer
956 .getInstance().createImageForRule(rule, featureType,
957 new Dimension(iconWidth, iconHeight));
958
959 final ImageIcon legendIcon = new ImageIcon(imageForRule);
960
961 final JLabel iconLabel = new JLabel(legendIcon);
962 panel.add(iconLabel, "sgx1");
963 // hbox.setAlignmentX(0f);
964 // hbox.add(iconLabel);
965 // hbox.add(Box.createHorizontalStrut(3));
966
967 final Translation labelT = new Translation();
968 labelT.fromOneLine(rule.getDescription().getTitle());
969 final JLabel classTitleLabel = new JLabel(labelT.toString());
970
971 panel.add(classTitleLabel, "sgx2");
972 classTitleLabel.setLabelFor(iconLabel);
973 }
974 }
975
976 return panel;
977 }
978
979 /**
980 * Creates a {@link JComponent} that contains a legend for a given
981 * rasterLayer and a given {@link Style}.
982 *
983 * @param style
984 * if <code>null</code>, the default {@link Style} is extracetd
985 * from the {@link StyledRasterInterface}
986 */
987 public static JPanel createLegendPanel(
988 final StyledRasterInterface<?> styledRaster, Style style,
989 final int iconWidth, final int iconHeight) {
990
991 // If no style is given, we use the default style for this layer
992 if (style == null)
993 style = styledRaster.getStyle();
994
995 /**
996 * Determine whether a Style is responsible for the coloring
997 */
998 ColorModel colorModel = null;
999 if (!isStyleable(styledRaster)
1000 || (isStyleable(styledRaster) && style == null)) {
1001 colorModel = getColorModel(styledRaster);
1002 }
1003
1004 final RasterLegendData rasterLegendData = styledRaster
1005 .getLegendMetaData();
1006 final List<Double> legendRasterValues = rasterLegendData
1007 .getSortedKeys();
1008 final Map<Double, GridCoverage2D> sampleRasters = rasterLegendData
1009 .createSampleRasters();
1010
1011 final JPanel panel = new JPanel(new MigLayout("wrap 2, gapy 0"));
1012
1013 for (final Double rValue : legendRasterValues) {
1014
1015 // final Dimension ICON_SIZE = new Dimension(iconWidth,
1016 // new JLabel().getFontMetrics(new JLabel().getFont())
1017 // .getHeight() > 5 ? new JLabel().getFontMetrics(
1018 // new JLabel().getFont()).getHeight() : iconHeight);
1019
1020 // ****************************************************************************
1021 // Create the actual icon
1022 // ****************************************************************************
1023 final BufferedImage buffImage = new BufferedImage(iconWidth,
1024 iconHeight, BufferedImage.TYPE_INT_ARGB);
1025
1026 final Graphics2D graphics = buffImage.createGraphics();
1027
1028 if (colorModel != null) {
1029 // The colors come from the ColorModel!
1030
1031 try {
1032 Object inData = null;
1033 switch (colorModel.getTransferType()) {
1034 case DataBuffer.TYPE_BYTE:
1035 inData = new byte[] { rValue.byteValue() };
1036 break;
1037 case DataBuffer.TYPE_USHORT:
1038 inData = new short[] { rValue.shortValue() };
1039 break;
1040 case DataBuffer.TYPE_INT:
1041 inData = new int[] { rValue.intValue() };
1042 break;
1043 case DataBuffer.TYPE_SHORT:
1044 inData = new short[] { rValue.shortValue() };
1045 break;
1046 case DataBuffer.TYPE_FLOAT:
1047 inData = new float[] { rValue.floatValue() };
1048 break;
1049 case DataBuffer.TYPE_DOUBLE:
1050 inData = new double[] { rValue.doubleValue() };
1051 break;
1052 default:
1053 inData = rValue.intValue();
1054 }
1055 final Color color = new Color(colorModel.getRGB(inData));
1056 graphics.setBackground(color);
1057 graphics.setColor(color);
1058 graphics.fillRect(0, 0, iconWidth, iconHeight);
1059 } catch (final Exception e) {
1060 LOGGER.info(
1061 "Dann nehmen wir halt den GridCoverageRenderer", e);
1062 colorModel = null;
1063 }
1064 } else {
1065 // The colors come from the Style
1066
1067 /**
1068 * The coverage contains only one value of value rValue
1069 */
1070 final GridCoverage2D sampleCov = sampleRasters.get(rValue);
1071 GridCoverageRenderer renderer;
1072 try {
1073 renderer = new GridCoverageRenderer(sampleCov
1074 .getCoordinateReferenceSystem(), JTSUtil
1075 .createEnvelope(sampleCov.getEnvelope()),
1076 new Rectangle(iconWidth, iconHeight),
1077 (AffineTransform) null);
1078 } catch (final Exception e1) {
1079 throw new RuntimeException(
1080 "Creating a GridCoverageRenderer failed:", e1);
1081 }
1082
1083 /**
1084 * Iterate over all FeatureTypeStyles.
1085 */
1086 final List<RasterSymbolizer> rSymbols = StylingUtil
1087 .getRasterSymbolizers(style);
1088
1089 for (final RasterSymbolizer symbolizer : rSymbols) {
1090 try {
1091 renderer.paint(graphics, sampleCov, symbolizer);
1092 } catch (final Exception ee) {
1093 LOGGER.error("Unable to paint " + symbolizer
1094 + " into the legend image", ee);
1095 }
1096 }
1097 }
1098
1099 final JLabel iconLabel = new JLabel(new ImageIcon(buffImage));
1100 panel.add(iconLabel, "sgx1");
1101
1102 final Translation labelT = rasterLegendData.get(rValue);
1103 final JLabel classTitleLabel = new JLabel(labelT.toString());
1104 panel.add(classTitleLabel, "sgx2"
1105 + (rasterLegendData.isPaintGaps() ? ", gapy 0:0:0 5:5:5"
1106 : ""));
1107 classTitleLabel.setLabelFor(iconLabel);
1108
1109 if (rasterLegendData.isPaintGaps()) {
1110 iconLabel
1111 .setBorder(BorderFactory.createLineBorder(Color.black));
1112 }
1113
1114 }
1115
1116 return panel;
1117 }
1118
1119 /**
1120 * Extracts the {@link ColorModel} of any {@link StyledRasterInterface}. May
1121 * return <code>null</code> if the geoobject can not be accessed.
1122 */
1123 @SuppressWarnings("unchecked")
1124 public static ColorModel getColorModel(
1125 final StyledRasterInterface<?> styledGrid) {
1126 ColorModel colorModel = null;
1127 try {
1128 final Object geoObject = styledGrid.getGeoObject();
1129 if (geoObject instanceof GridCoverage2D) {
1130 final GridCoverage2D cov = (GridCoverage2D) geoObject;
1131 colorModel = cov.getRenderedImage().getColorModel();
1132 } else if (styledGrid instanceof StyledRasterPyramidInterface) {
1133
1134 final Parameter readGG = new Parameter(
1135 AbstractGridFormat.READ_GRIDGEOMETRY2D);
1136
1137 final ReferencedEnvelope mapExtend = new org.geotools.geometry.jts.ReferencedEnvelope(
1138 styledGrid.getEnvelope(), styledGrid.getCrs());
1139
1140 readGG.setValue(new GridGeometry2D(new GeneralGridEnvelope(
1141 new Rectangle(0, 0, 1, 1)), mapExtend));
1142
1143 final FeatureCollection<SimpleFeatureType, SimpleFeature> rFc = (FeatureCollection<SimpleFeatureType, SimpleFeature>) geoObject;
1144
1145 final AbstractGridCoverage2DReader aReader = (AbstractGridCoverage2DReader) FeatureUtil
1146 .getWrappedGeoObject(rFc);
1147 final GridCoverage2D cov = (GridCoverage2D) aReader
1148 .read(new GeneralParameterValue[] { readGG });
1149 colorModel = cov.getRenderedImage().getColorModel();
1150 }
1151 } catch (final Exception e) {
1152 LOGGER.error("Error reading the colormodel from " + styledGrid, e);
1153 return null;
1154 }
1155 return colorModel;
1156 }
1157
1158 /**
1159 * @return <code>true</code> if a {@link RasterSymbolizer} can be applied
1160 * and will have an effect. Some rasters (e.g. GeoTIFF) can come
1161 * with their own {@link ColorModel} and will ignore any
1162 * {@link RasterSymbolizer} = SLD.
1163 */
1164 public static boolean isStyleable(
1165 final StyledRasterInterface<?> styledRaster) {
1166 final ColorModel colorModel = getColorModel(styledRaster);
1167
1168 // LOGGER.info("The colormodel of " + styledRaster.getTitle() + " is "
1169 // + colorModel != null ? colorModel.getClass().getSimpleName() :
1170 // "NULL");
1171
1172 if (colorModel == null)
1173 return true;
1174 if (colorModel instanceof ComponentColorModel)
1175 return true;
1176 return false;
1177 }
1178
1179 /**
1180 * Set the given Style as the Style of the {@link MapLayer}, unless the
1181 * styles are the same (not comparing selection stuff). If the
1182 * {@link MapLayer}s {@link Style} is changed, the selection FTS is kept.<br/>
1183 * Remember {@link MapLayer#setStyle(Style)} triggers an event leading to a
1184 * repaint, so only use it when needed.
1185 *
1186 * @return <code>true</code> if the {@link MapLayer}'s {@link Style} has
1187 * been changed.
1188 */
1189 public static boolean updateMapLayerStyleIfChangedAndKeepSelection(
1190 MapLayer mapLayer, Style style2) {
1191
1192 Style mapLayerStyleCleaned = StylingUtil
1193 .removeSelectionFeatureTypeStyle(mapLayer.getStyle());
1194
1195 Style newStyleCleaned = StylingUtil
1196 .removeSelectionFeatureTypeStyle(style2);
1197
1198 if (StylingUtil.isStyleDifferent(mapLayerStyleCleaned, newStyleCleaned)) {
1199
1200 // They are different when compared without SELECTION FTS!
1201
1202 // Now let's copy any SELECTION FTS to the now style
1203 FeatureTypeStyle selectionFeatureTypeStyle = StylingUtil
1204 .getSelectionFeatureTypeStyle(mapLayer.getStyle());
1205 if (selectionFeatureTypeStyle != null) {
1206 newStyleCleaned.featureTypeStyles().add(
1207 selectionFeatureTypeStyle);
1208 // newStyleCleaned is not so clean anymore... We just alled a
1209 // selcetion FTS
1210 }
1211
1212 mapLayer.setStyle(newStyleCleaned);
1213
1214 return true;
1215
1216 } else {
1217 return false;
1218 }
1219 }
1220
1221 /**
1222 * After loading an atlas, the AttribteMetaData contains whatever is written
1223 * in the XML. But the DBF may have changed! This method checks an
1224 * {@link AttributeMetadataMap} against a schema and also corrects
1225 * upperCase/lowerCase problems. It will also remove any geometry column
1226 * attribute metadata.
1227 */
1228 /**
1229 * After loading an atlas, the AttribteMetaData contains whatever is written
1230 * in the XML. But the DBF may have changed!
1231 */
1232 public static void checkAttribMetaData(AttributeMetadataMap attributeMetaDataMap,
1233 SimpleFeatureType schema) {
1234
1235 ArrayList<Name> willRemove = new ArrayList<Name>();
1236
1237 // 1. Check.. all attributes in the atm should be in the schema as well.
1238 // maybe correct some upperCase/loweCase stuff
1239
1240 for (AttributeMetadata atm : attributeMetaDataMap.values()) {
1241
1242 AttributeDescriptor foundDescr = schema
1243 .getDescriptor(atm.getName());
1244 if (foundDescr == null) {
1245 Name bestMatch = FeatureUtil.findBestMatchingAttribute(schema,
1246 atm.getLocalName());
1247 if (bestMatch == null)
1248 willRemove.add(atm.getName());
1249 else
1250 atm.setName(bestMatch);
1251 } else if (foundDescr instanceof GeometryDescriptor) {
1252 // We don't want GeometryColumns in here
1253 willRemove.add(atm.getName());
1254 }
1255 }
1256
1257 // Remove the ones that were not findable in the schema
1258 for (Name removeName : willRemove) {
1259 if (attributeMetaDataMap.remove(removeName) == null){
1260 LOGGER.warn("removing the AMData didn't work");
1261 }
1262 }
1263
1264 // 2. check... all attributes from the schema must have an ATM
1265 for (AttributeDescriptor ad : schema.getAttributeDescriptors()) {
1266 if (ad instanceof GeometryDescriptor)
1267 continue;
1268 if (!attributeMetaDataMap.containsKey(ad.getName())) {
1269 attributeMetaDataMap.put(ad.getName(), new AttributeMetadata(
1270 ad, schema.getAttributeDescriptors().indexOf(ad), attributeMetaDataMap
1271 .getLanguages()));
1272 }
1273 }
1274 }
1275
1276 }

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26