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

Legend:
Removed from v.93  
changed lines
  Added in v.1203

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26