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

Diff of /trunk/src/skrueger/geotools/StyledFS.java

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 169 by alfonx, Tue Jun 30 08:03:29 2009 UTC revision 863 by alfonx, Sun May 23 13:42:13 2010 UTC
# Line 1  Line 1 
1  package skrueger.geotools;  /*******************************************************************************
2     * Copyright (c) 2009 Martin O. J. Schmitz.
3  import java.io.File;   *
4  import java.io.FileNotFoundException;   * This file is part of the SCHMITZM library - a collection of utility
5  import java.io.IOException;   * classes based on Java 1.6, focusing (not only) on Java Swing
6  import java.net.URL;   * and the Geotools library.
7  import java.util.Date;   *
8  import java.util.HashMap;   * The SCHMITZM project is hosted at:
9  import java.util.Map;   * http://wald.intevation.org/projects/schmitzm/
10  import java.util.Random;   *
11     * This program is free software; you can redistribute it and/or
12  import javax.swing.ImageIcon;   * modify it under the terms of the GNU Lesser General Public License
13  import javax.swing.JPanel;   * as published by the Free Software Foundation; either version 3
14     * of the License, or (at your option) any later version.
15  import org.apache.log4j.Logger;   *
16  import org.geotools.data.FeatureSource;   * This program is distributed in the hope that it will be useful,
17  import org.geotools.feature.AttributeType;   * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  import org.geotools.gui.swing.ExceptionMonitor;   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  import org.geotools.styling.Style;   * GNU General Public License for more details.
20  import org.opengis.referencing.crs.CoordinateReferenceSystem;   *
21     * You should have received a copy of the GNU Lesser General Public License (license.txt)
22  import schmitzm.geotools.styling.StylingUtil;   * along with this program; if not, write to the Free Software
23  import skrueger.AttributeMetaData;   * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
24  import skrueger.i8n.Translation;   * or try this link: http://www.gnu.org/licenses/lgpl.html
25     *
26  import com.vividsolutions.jts.geom.Envelope;   * Contributors:
27     *     Martin O. J. Schmitz - initial API and implementation
28  /**   *     Stefan A. Tzeggai - additional utility classes
29   * This class enables a non Atlas context to use the Atlas LayerPanel   ******************************************************************************/
30   * {@link JPanel} as a {@link MapContextManagerInterface}  package skrueger.geotools;
31   *  
32   * @author <a href="mailto:[email protected]">Stefan Alfons Kr&uuml;ger</a>  import java.io.File;
33   */  import java.io.FileNotFoundException;
34  public class StyledFS implements StyledFeatureSourceInterface {  import java.io.IOException;
35          private static final Logger LOGGER = Logger.getLogger(StyledFS.class);  import java.net.URL;
36    import java.util.Date;
37          private final FeatureSource fs;  import java.util.Random;
38    
39          /**  import javax.swing.ImageIcon;
40           * A unique ID which identifies the Layer in the Atlas. It's more important  import javax.swing.JPanel;
41           * than it should be ;-)  
42           */  import org.apache.log4j.Logger;
43          private String id;  import org.geotools.data.FeatureSource;
44    import org.geotools.feature.FeatureCollection;
45          private Style style;  import org.geotools.feature.NameImpl;
46    import org.geotools.styling.Style;
47          private Translation title;  import org.opengis.feature.simple.SimpleFeature;
48    import org.opengis.feature.simple.SimpleFeatureType;
49          private Translation desc;  import org.opengis.feature.type.AttributeDescriptor;
50    import org.opengis.filter.Filter;
51          private File sldFile;  import org.opengis.referencing.crs.CoordinateReferenceSystem;
52    
53          private HashMap<Integer, AttributeMetaData> map;  import schmitzm.geotools.io.GeoImportUtil;
54    import schmitzm.geotools.styling.StylingUtil;
55          /**  import skrueger.AttributeMetadataImpl;
56           * This class enables a non Atlas context to use the Atlas LayerPanel  import skrueger.AttributeMetadataInterface;
57           * {@link JPanel} as a {@link MapContextManagerInterface}  import skrueger.i8n.Translation;
58           *  
59           * @param fs  import com.vividsolutions.jts.geom.Envelope;
60           *            {@link FeatureSource} that is beeing styled.  
61           *  /**
62           * @param sldFile   * This class enables a non Atlas context to use the Atlas LayerPanel
63           *            may be <code>null</code>. Otherwise the SLD {@link File} to   * {@link JPanel} as a {@link MapContextManagerInterface}
64           *            import and associate with this {@link StyledFS}   *
65           */   * @author <a href="mailto:[email protected]">Stefan Alfons Kr&uuml;ger</a>
66          public StyledFS(FeatureSource fs, File sldFile) {   *
67     *         TODO Rename to StyledShapefile
68                  this.fs = fs;   */
69                  id = StyledFS.class.getSimpleName()  public class StyledFS implements StyledFeatureSourceInterface {
70                                  + new Random(new Date().getTime()).nextInt(10000000);          private static final Logger LOGGER = Logger.getLogger(StyledFS.class);
71    
72                  this.sldFile = sldFile;          private final FeatureSource<SimpleFeatureType, SimpleFeature> fs;
73    
74                  if ((sldFile != null) && (sldFile.exists())) {          /** Caching the CRS of the layer **/
75                          try {          CoordinateReferenceSystem crs = null;
76                                  style = StylingUtil.loadSLD(sldFile)[0];  
77                          } catch (FileNotFoundException e) {          /**
78                                  LOGGER           * A unique ID which identifies the Layer in the Atlas. It's more important
79                                                  .debug("The SLD file passed was empty. Leaving the Style untouched. (We are in the constructor.. so its null");           * than it should be ;-)
80                          } catch (Exception e) {           */
81                                  LOGGER.warn("Reading SLD failed: " + sldFile, e);          private String id;
82                          }  
83                  }          private Style style;
84                  title = new Translation();  
85                  title.fromOneLine(sldFile.getName());          private Translation title;
86    
87                  desc = new Translation();          private Translation desc;
88                  desc.fromOneLine(sldFile.getAbsolutePath());  
89          }          private File sldFile;
90    
91          public void dispose() {          /** A map of simple attribute names to their meta-data **/
92          }          private AttributeMetadataMap<AttributeMetadataImpl> map;
93    
94          /**          private Filter filter = Filter.INCLUDE;
95           * Returnes human readable {@link String} of the CRS natively used by this  
96           * {@link DpLayer}          /**
97           *           * This class enables a non Atlas context to use the Atlas LayerPanel
98           * If crs == null, it will call {@link #getGeoObject()}           * {@link JPanel} as a {@link MapContextManagerInterface}
99           *           *
100           */           * @param fs
101          public String getCRSString() {           *            {@link FeatureSource} that is beeing styled.
102                  if (getCrs() == null)           *
103                          return "CRS?";           * @param sldFile
104             *            may be <code>null</code>. Otherwise the SLD {@link File} to
105                  return getCrs().getName().getCode();           *            import and associate with this {@link StyledFS}
106          }           */
107            public StyledFS(FeatureSource<SimpleFeatureType, SimpleFeature> fs,
108          public CoordinateReferenceSystem getCrs() {                          File sldFile) {
109                  return fs.getSchema().getDefaultGeometry().getCoordinateSystem();  
110          }                  this.fs = fs;
111    
112          public Translation getDesc() {                  id = StyledFS.class.getSimpleName()
113                  return desc;                                  + new Random(new Date().getTime()).nextInt(10000000);
114          }  
115                    this.sldFile = sldFile;
116          public Envelope getEnvelope() {  
117                  try {                  // datei existiert, dann lesen
118                          return fs.getBounds();                  if (sldFile != null && sldFile.exists()) {
119                  } catch (IOException e) {                          try {
120                          e.printStackTrace();                                  style = StylingUtil.loadSLD(sldFile)[0];
121                          return null;                          } catch (Exception e) {
122                  }                                  LOGGER.warn("Reading SLD failed: " + sldFile, e);
123          }                                  style = null;
124                            }
125          public FeatureSource getGeoObject() throws Exception {                  }
126                  return fs;  
127          }                  title = new Translation();
128                    desc = new Translation();
129          public String getId() {  
130                  return id;                  if (sldFile != null) {
131          }                          title.fromOneLine(sldFile.getName());
132                            desc.fromOneLine(sldFile.getAbsolutePath());
133          public ImageIcon getImageIcon() {                  }
134                  return null;  
135          }          }
136    
137          public URL getInfoURL() {          public StyledFS(FeatureSource<SimpleFeatureType, SimpleFeature> fs) {
138                  return null;                  this(fs, null);
139          }          }
140    
141          public Translation getKeywords() {          public void dispose() {
142                  return null;          }
143          }  
144            /**
145          public Style getStyle() {           * Returns human readable {@link String} of the CRS natively used by this
146                  return style;           * {@link DpLayer}
147          }           *
148             * If CRS == null, it will call {@link #getGeoObject()}
149          public Translation getTitle() {           *
150                  return title;           */
151          }          public String getCRSString() {
152                    if (getCrs() == null)
153          public boolean isDisposed() {                          return "CRS?";
154                  return false;  
155          }                  return getCrs().getName().getCode();
156            }
157          /**  
158           * If true, this layer will not be shown in the legend. Default = false          public CoordinateReferenceSystem getCrs() {
159           */                  if (crs == null) {
160          /**                          crs = fs.getSchema().getCoordinateReferenceSystem();
161           *                          if (fs.getSchema().getCoordinateReferenceSystem() == null) {
162           * Killed by SK: 6. April 09: Ein Layer soll nicht generell auf                                  LOGGER.warn("Could not determine the CRS of " + getTitle()
163           * verstecken/nicht verstecken gestellt werden können. Das sind                                                  + ". Using default " + GeoImportUtil.getDefaultCRS());
164           * Eigenschaften der Karte/MapContext, ebenso wie die Reihenfolge der Layer.                                  crs = GeoImportUtil.getDefaultCRS();
165           * Im Atlas verwaltet deshalb nun die Klasse skrueger.atlas.Map welche Layer                          }
166           * nicht in der Legende auftauchen sollen. Meines Wissens hat keiner bisher                  }
167           * die Funktion genutzt.                  return crs;
168           *          }
169           * // public boolean isHideInLegend() { // return false; // }  
170           */          public Translation getDesc() {
171                    return desc;
172          public void setDesc(Translation dec) {          }
173                  this.desc = dec;  
174          }          public Envelope getEnvelope() {
175                    try {
176          public void setImageIcon(ImageIcon icon) {                          return fs.getBounds();
177          }                  } catch (IOException e) {
178                            e.printStackTrace();
179          public void setKeywords(Translation keywords) {                          return null;
180          }                  }
181            }
182          public void setStyle(Style style) {  
183                  this.style = style;          public FeatureSource<SimpleFeatureType, SimpleFeature> getGeoObject() {
184                    return fs;
185          }          }
186    
187          public void setTitle(Translation title) {          public String getId() {
188                  this.title = title;                  return id;
189            }
190          }  
191            public ImageIcon getImageIcon() {
192          public void uncache() {                  return null;
193          }          }
194    
195          /**          public URL getInfoURL() {
196           *                  return null;
197           */          }
198          public Map<Integer, AttributeMetaData> getAttributeMetaDataMap() {  
199                  if (map == null) {          public Translation getKeywords() {
200                    return null;
201                          map = new HashMap<Integer, AttributeMetaData>();          }
202    
203                          // Leaving out the first one, it will be the_geom          public Style getStyle() {
204                          for (int i = 1; i < fs.getSchema().getAttributeCount(); i++) {                  return style;
205                                  AttributeType att = fs.getSchema().getAttributeType(i);          }
206    
207                                  AttributeMetaData attMetaData = new AttributeMetaData(i, att          public Translation getTitle() {
208                                                  .getLocalName());                  return title;
209                                  map.put(i, attMetaData);          }
210                          }  
211                  }          public boolean isDisposed() {
212                  return map;                  return false;
213          }          }
214    
215          /**          /**
216           * @return The {@link File} where the SLD was loaded from or           * If true, this layer will not be shown in the legend. Default = false
217           *         <code>null</code> if there didn't exist a {@link File}. (It could           */
218           *         be a WFS or a PostGIS          /**
219           *           *
220           * @author <a href="mailto:[email protected]">Stefan Alfons           * Killed by SK: 6. April 09: Ein Layer soll nicht generell auf
221           *         Kr&uuml;ger</a>           * verstecken/nicht verstecken gestellt werden können. Das sind
222           */           * Eigenschaften der Karte/MapContext, ebenso wie die Reihenfolge der Layer.
223          public File getSldFile() {           * Im Atlas verwaltet deshalb nun die Klasse skrueger.atlas.Map welche Layer
224                  return sldFile;           * nicht in der Legende auftauchen sollen. Meines Wissens hat keiner bisher
225          }           * die Funktion genutzt.
226             *
227          public void setSldFile(File sldFile) {           * // public boolean isHideInLegend() { // return false; // }
228                  this.sldFile = sldFile;           */
229          }  
230            public void setDesc(Translation dec) {
231  }                  this.desc = dec;
232            }
233    
234            public void setImageIcon(ImageIcon icon) {
235            }
236    
237            public void setKeywords(Translation keywords) {
238            }
239    
240            public void setStyle(Style style) {
241                    this.style = style;
242    
243            }
244    
245            public void setTitle(Translation title) {
246                    this.title = title;
247    
248            }
249    
250            public void uncache() {
251            }
252    
253            /**
254             *
255             */
256            public AttributeMetadataMap<AttributeMetadataImpl> getAttributeMetaDataMap() {
257                    if (map == null) {
258    
259                            map = new AttributeMetadataImplMap();
260    
261                            // // Leaving out the first one, it will be the_geom
262                            // for (int i = 1; i < fs.getSchema().getAttributeCount(); i++) {
263                            // AttributeDescriptor attDesc = fs.getSchema().getDescriptor(i);
264                            //
265                            // AttributeMetadataImpl attMetaData = new AttributeMetadataImpl(
266                            // new NameImpl(attDesc
267                            // .getName().getNamespaceURI(), attDesc
268                            // .getName().getLocalPart()), map.getLanguages());
269                            // map.put(attDesc.getName(), attMetaData);
270                            // }
271    
272                            // Leaving out the first one, it will be the_geom
273                            for (int i = 1; i < fs.getSchema().getAttributeCount(); i++) {
274                                    AttributeDescriptor attDesc = fs.getSchema().getDescriptor(i);
275    
276                                    // TODO AttributeMetadataAS would be nicer, which would not work
277                                    // with Translations ;-)
278                                    AttributeMetadataImpl attMetaData = new AttributeMetadataImpl(
279                                                    new NameImpl(attDesc.getName().getNamespaceURI(),
280                                                                    attDesc.getName().getLocalPart()), map
281                                                                    .getLanguages());
282                                    if (String.class.isAssignableFrom(attDesc.getType()
283                                                    .getBinding())) {
284                                            // For Strings we add the "" as NODATA values
285                                            attMetaData.addNodataValue("");
286                                    }
287                                    map.put(attDesc.getName(), attMetaData);
288                            }
289                    }
290                    return map;
291            }
292    
293            /**
294             * @return The {@link File} where the SLD was loaded from or
295             *         <code>null</code> if there didn't exist a {@link File}.
296             *
297             * @author <a href="mailto:[email protected]">Stefan Alfons
298             *         Kr&uuml;ger</a>
299             */
300            public File getSldFile() {
301                    return sldFile;
302            }
303    
304            public void setSldFile(File sldFile) {
305                    this.sldFile = sldFile;
306            }
307    
308            /**
309             * Returns the features of the {@link FeatureSource}.
310             *
311             * @see {@link StyledFeaturesInterface}
312             */
313            @Override
314            public FeatureCollection<SimpleFeatureType, SimpleFeature> getFeatureCollection() {
315                    FeatureCollection<SimpleFeatureType, SimpleFeature> features;
316                    try {
317                            features = getGeoObject().getFeatures();
318                    } catch (IOException e) {
319                            throw new RuntimeException(
320                                            "Error getting the features of the  FeatureSource");
321                    }
322                    return features;
323            }
324    
325            /**
326             * Same as {@link #getGeoObject()} method, but complies to the
327             * {@link StyledFeaturesInterface}
328             *
329             * @see {@link StyledFeaturesInterface}
330             */
331            @Override
332            public FeatureSource<SimpleFeatureType, SimpleFeature> getFeatureSource() {
333                    return getGeoObject();
334            }
335    
336            @Override
337            public FeatureCollection<SimpleFeatureType, SimpleFeature> getFeatureCollectionFiltered() {
338                    // final FeatureCollection<SimpleFeatureType, SimpleFeature> fc =
339                    // getFeatureCollection();
340                    // if (filter == Filter.EXCLUDE)
341                    // return new EmptyFeatureCollection(fc.getSchema());
342                    // if (filter == Filter.INCLUDE)
343                    // return fc;
344                    // return fc.subCollection(filter);
345    
346                    try {
347                            return getFeatureSource().getFeatures(filter);
348                    } catch (IOException e) {
349                            throw new RuntimeException(e);
350                    }
351            }
352    
353            @Override
354            public Filter getFilter() {
355                    return filter;
356            }
357    
358            @Override
359            public void setFilter(Filter filter) {
360                    this.filter = filter;
361            }
362    
363            @Override
364            public SimpleFeatureType getSchema() {
365                    return getGeoObject().getSchema();
366            }
367    
368            /**
369             * Tries to load a style from the file denoted in {@link #getSldFile()}. If
370             * the file doesn't exits, return <code>null</code>;
371             * @return <code>true</code> is style was loaded
372             */
373            public boolean loadStyle() {
374                    if (getSldFile() == null)
375                            return false;
376                    
377                    try {
378                            Style[] loadSLD = StylingUtil.loadSLD(getSldFile());
379                            setStyle(loadSLD[0]);
380                            return true;
381                    } catch (Exception e) {
382                            return false;
383                    }
384    
385            }
386    
387    }

Legend:
Removed from v.169  
changed lines
  Added in v.863

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26