/[schmitzm]/branches/2.0-RC2/src/skrueger/geotools/selection/FeatureMapLayerSelectionSynchronizer.java
ViewVC logotype

Diff of /branches/2.0-RC2/src/skrueger/geotools/selection/FeatureMapLayerSelectionSynchronizer.java

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

branches/1.0-gt2-2.6/src/skrueger/geotools/selection/FeatureMapLayerSelectionSynchronizer.java revision 315 by mojays, Wed Aug 26 11:03:27 2009 UTC branches/2.0-RC1/src/skrueger/geotools/selection/FeatureMapLayerSelectionSynchronizer.java revision 604 by alfonx, Wed Dec 9 14:15:53 2009 UTC
# Line 29  Line 29 
29   ******************************************************************************/   ******************************************************************************/
30  /**  /**
31   Copyright 2008 Stefan Alfons Krüger   Copyright 2008 Stefan Alfons Krüger
32    
33   atlas-framework - This file is part of the Atlas Framework   atlas-framework - This file is part of the Atlas Framework
34    
35   This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version.   This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version.
# Line 42  Line 42 
42   **/   **/
43  package skrueger.geotools.selection;  package skrueger.geotools.selection;
44    
45    import java.awt.RenderingHints;
46  import java.beans.PropertyChangeEvent;  import java.beans.PropertyChangeEvent;
47  import java.beans.PropertyChangeListener;  import java.beans.PropertyChangeListener;
48  import java.util.Arrays;  import java.io.File;
49  import java.util.Collection;  import java.util.HashMap;
50  import java.util.HashSet;  import java.util.HashSet;
51    import java.util.Iterator;
52  import java.util.Set;  import java.util.Set;
53  import java.util.Vector;  import java.util.Vector;
54    
# Line 54  import javax.swing.JTable; Line 56  import javax.swing.JTable;
56  import javax.swing.ListSelectionModel;  import javax.swing.ListSelectionModel;
57  import javax.swing.event.ListSelectionListener;  import javax.swing.event.ListSelectionListener;
58    
59  import org.geotools.feature.Feature;  import org.geotools.feature.FeatureCollection;
60    import org.geotools.feature.FeatureIterator;
61    import org.geotools.filter.FidFilterImpl;
62  import org.geotools.map.MapLayer;  import org.geotools.map.MapLayer;
63    import org.geotools.renderer.GTRenderer;
64    import org.geotools.renderer.label.LabelCacheImpl;
65    import org.geotools.renderer.lite.StreamingRenderer;
66  import org.geotools.styling.FeatureTypeStyle;  import org.geotools.styling.FeatureTypeStyle;
67  import org.geotools.styling.Style;  import org.geotools.styling.Style;
68    import org.geotools.styling.visitor.DuplicatingStyleVisitor;
69    import org.opengis.feature.simple.SimpleFeature;
70    import org.opengis.feature.simple.SimpleFeatureType;
71    import org.opengis.filter.Filter;
72    import org.opengis.filter.Not;
73  import org.opengis.filter.identity.FeatureId;  import org.opengis.filter.identity.FeatureId;
74    
75  import schmitzm.geotools.FilterUtil;  import schmitzm.geotools.FilterUtil;
76  import schmitzm.geotools.gui.JMapPane;  import schmitzm.geotools.GTUtil;
77    import schmitzm.geotools.gui.SelectableXMapPane;
78  import schmitzm.geotools.map.event.FeatureSelectedEvent;  import schmitzm.geotools.map.event.FeatureSelectedEvent;
 import schmitzm.geotools.map.event.JMapPaneEvent;  
79  import schmitzm.geotools.map.event.JMapPaneListener;  import schmitzm.geotools.map.event.JMapPaneListener;
80    import schmitzm.geotools.map.event.MapPaneEvent;
81  import schmitzm.geotools.styling.StylingUtil;  import schmitzm.geotools.styling.StylingUtil;
 import schmitzm.lang.LangUtil;  
82  import skrueger.geotools.MapPaneToolBar;  import skrueger.geotools.MapPaneToolBar;
83  import skrueger.geotools.StyledLayerInterface;  import skrueger.geotools.StyledFeaturesInterface;
84    
85  /**  /**
86   * This class keeps the selection of a (feature) {@link JTable} synchronized   * This class keeps the selection of a (feature) {@link JTable} synchronized
# Line 91  import skrueger.geotools.StyledLayerInte Line 103  import skrueger.geotools.StyledLayerInte
103  public class FeatureMapLayerSelectionSynchronizer extends  public class FeatureMapLayerSelectionSynchronizer extends
104                  StyledLayerSelectionModelSynchronizer<StyledFeatureLayerSelectionModel>                  StyledLayerSelectionModelSynchronizer<StyledFeatureLayerSelectionModel>
105                  implements JMapPaneListener {                  implements JMapPaneListener {
106          public static final String SELECTION_STYLING = "SELECTION";  
107            /**
108             * This constant is set as the {@link FeatureTypeStyle#getName()} attribute
109             * in the {@link FeatureTypeStyle}s that only exist to present the selected
110             * features
111             **/
112            public static final String SELECTION_STYLING_FTS_NAME = "SELECTION";
113    
114          /**          /**
115           * Holds the {@link MapLayer} to keep synchronized with the layer selection           * Holds the {@link MapLayer} to keep synchronized with the layer selection
116           * model.           * model.
117           */           */
118          protected final MapLayer mapLayer;          protected final MapLayer mapLayer;
119          protected final StyledLayerInterface<?> styledLayer;          protected final StyledFeaturesInterface<?> styledLayer;
120          protected final JMapPane mapPane;          protected final SelectableXMapPane mapPane;
121          private final MapPaneToolBar toolBar;          private final MapPaneToolBar toolBar;
122    
123          /**          /**
# Line 113  public class FeatureMapLayerSelectionSyn Line 132  public class FeatureMapLayerSelectionSyn
132           */           */
133          public FeatureMapLayerSelectionSynchronizer(          public FeatureMapLayerSelectionSynchronizer(
134                          StyledFeatureLayerSelectionModel layerSelModel,                          StyledFeatureLayerSelectionModel layerSelModel,
135                          StyledLayerInterface<?> styledLayer, MapLayer mapLayer,                          StyledFeaturesInterface<?> styledLayer, MapLayer mapLayer,
136                          JMapPane mapPane, MapPaneToolBar toolBar) {                          SelectableXMapPane mapPane, MapPaneToolBar toolBar) {
137    
138                  super(layerSelModel);                  super(layerSelModel);
139                  this.styledLayer = styledLayer;                  this.styledLayer = styledLayer;
# Line 167  public class FeatureMapLayerSelectionSyn Line 186  public class FeatureMapLayerSelectionSyn
186           * selection.           * selection.
187           *           *
188           * @param newSelection           * @param newSelection
189           *            A {@link Vector} of Feature-IDs that are selected.           *            A {@link Vector} of SimpleFeature-IDs that are selected.
190           */           */
191          private void changeLayerStyle(final Vector<String> newSelection) {          private void changeLayerStyle(final Vector<String> newSelection) {
                 Style selectionMapStyle = null;  
192                  try {                  try {
193    
194                            Style originalStyle = mapLayer.getStyle();
195    
196                          if (newSelection.isEmpty()) {                          if (newSelection.isEmpty()) {
197    
198                                  selectionMapStyle = styledLayer.getStyle();                                  // Check if the Style contains a SELECTION FTS
199                                  // LOGGER.debug("NO SELECTION .. set to original style directly");  
200                                    FeatureTypeStyle[] clone = originalStyle.featureTypeStyles()
201                                                    .toArray(new FeatureTypeStyle[] {}).clone();
202    
203                                    for (FeatureTypeStyle fts : clone) {
204                                            if (fts.getName() != null
205                                                            && fts.getName().equals(SELECTION_STYLING_FTS_NAME)) {
206                                                    originalStyle.featureTypeStyles().remove(fts);
207    
208                                                    mapLayer.setStyle(originalStyle);
209                                                    // replaceRenderer();
210                                                    // // mapPane.refresh();
211    
212                                                    return;
213                                            }
214                                    }
215    
216                          } else {                          } else {
217    
218                                  LOGGER.debug("SELECTION .. change style");                                  LOGGER.debug("SELECTION .. change style");
219    
220                                    // Saving a repaint if the selection didn't change
221                                    if (!selectionChanged(newSelection, originalStyle)) {
222                                            return;
223                                    }
224    
225                                  // We take Style from the MapLayer that is displayed at the                                  // We take Style from the MapLayer that is displayed at the
226                                  // moment. We do not use the styledLayer.getStyle, because in                                  // moment. We do not use the styledLayer.getStyle, because in
227                                  // the atlas, this always return the default style, but                                  // the atlas, this always return the default style, but
228                                  // additional styles might be selected.                                  // additional styles might be selected.
229                                  // Taking the style from the mapLayer indicated, that we have to                                  // Taking the style from the mapLayer indicated, that we have to
230                                  // remove any selection rules first.                                  // remove any selection rules first.
                                 Style originalStyle = mapLayer.getStyle();  
231    
232                                  // TODO The default style is not good here. We need                                  FeatureTypeStyle selectionFTStyle = StylingUtil
                                 // .createSelectionStyle(normalStyle, geoObj);  
                                 selectionMapStyle = StylingUtil  
233                                                  .createSelectionStyle(styledLayer.getGeoObject());                                                  .createSelectionStyle(styledLayer.getGeoObject());
                                 // selectionMapStyle = StylingUtil  
                                 // .createDefaultStyle(styledLayer.getGeoObject());  
234    
235                                  selectionMapStyle.getFeatureTypeStyles()[0]                                  selectionFTStyle.setName(SELECTION_STYLING_FTS_NAME);
                                                 .setName(SELECTION_STYLING);  
236    
237                                  /**                                  /**
238                                   *                                   *
239                                   * Add a Filter to the selectionMapStyle, so that it is only                                   * Add a Filter to the selectionMapStyle, so that it is only
240                                   * used on objects that are selected. <br/>                                   * used on objects that are selected. <br/>
241                                     * To optimize rendering speed, the filter checks whether more
242                                     * than half of the features are selected. If so, the filter is
243                                     * inverted to be shorter.
244                                   *                                   *
                                  * Note 1:<br/>  
                                  * It is NEVER allowed to GeoTools extend Filter () { .. } (and write  
                                  * tests into the evaluate block). Especially for the  
                                  * ShapeFileRenderer, we may only use a geotools Filter.<br/>  
                                  *  
                                  * Note 2:<br/>  
                                  * The FilterUtil.FILTER_FAC2.id(fids) wants a set of FeatureId-Objects!  
245                                   */                                   */
246                                    final FeatureCollection<SimpleFeatureType, SimpleFeature> featureCollectionFiltered = styledLayer
247                                                    .getFeatureCollectionFiltered();
248                                    if (newSelection.size() > featureCollectionFiltered.size() / 2) {
249                                            FeatureIterator<SimpleFeature> iterator = featureCollectionFiltered
250                                                            .features();
251                                            Set<FeatureId> antiFids = new HashSet<FeatureId>();
252                                            try {
253                                                    while (iterator.hasNext()) {
254                                                            SimpleFeature next = iterator.next();
255                                                            if (!newSelection.contains(next.getID()))
256                                                                    antiFids.add(FilterUtil.FILTER_FAC2
257                                                                                    .featureId(next.getID()));
258                                                    }
259    
260                                                    selectionFTStyle.rules().get(0).setFilter(
261                                                                    FilterUtil.FILTER_FAC2
262                                                                                    .not(FilterUtil.FILTER_FAC2
263                                                                                                    .id(antiFids)));
264    
265                                            } finally {
266                                                    featureCollectionFiltered.close(iterator);
267                                            }
268                                    } else {
269                                            Set<FeatureId> fids = new HashSet<FeatureId>();
270                                            for (String fid : newSelection) {
271                                                    fids.add(FilterUtil.FILTER_FAC2.featureId(fid));
272                                            }
273    
274                                  Set<FeatureId> fids = new HashSet<FeatureId>();                                          selectionFTStyle.rules().get(0).setFilter(
275                                  for (String fid : newSelection) {                                                          FilterUtil.FILTER_FAC2.id(fids));
                                         fids.add(FilterUtil.FILTER_FAC2.featureId(fid));  
276                                  }                                  }
277    
278                                  selectionMapStyle.getFeatureTypeStyles()[0].getRules()[0]                                  // Maybe there has already been another selection
279                                                  .setFilter(FilterUtil.FILTER_FAC2.id(fids));                                  // FeatureTypeStyle... Let's replace it...
280                                    boolean foundAndReplaced = false;
281                                  FeatureTypeStyle[] originalFeatureTypeStyles = originalStyle                                  for (FeatureTypeStyle fts : originalStyle.featureTypeStyles()) {
282                                                  .getFeatureTypeStyles();                                          if (fts.getName() != null
283                                  FeatureTypeStyle[] newFeatureTypes;                                                          && fts.getName().equals(SELECTION_STYLING_FTS_NAME)) {
284                                  if (originalFeatureTypeStyles[originalFeatureTypeStyles.length - 1]                                                  foundAndReplaced = true;
285                                                  .getName() != null                                                  fts.rules().clear();
286                                                  && originalFeatureTypeStyles[originalFeatureTypeStyles.length - 1]                                                  fts.rules().addAll(selectionFTStyle.rules());
287                                                                  .getName().equals(SELECTION_STYLING)) {                                                  break;
288                                          newFeatureTypes = Arrays.copyOf(originalFeatureTypeStyles,                                          }
289                                                          originalFeatureTypeStyles.length - 1);                                  }
290                                  } else {                                  if (!foundAndReplaced) {
291                                          newFeatureTypes = originalFeatureTypeStyles;                                          originalStyle.featureTypeStyles().add(selectionFTStyle);
292                                  }                                  }
293    
294                                  // The last FTS is the selection FTS                                  // Refresh the map
295                                  newFeatureTypes = LangUtil.extendArray(newFeatureTypes,                                  // mapPane.refresh();
                                                 selectionMapStyle.getFeatureTypeStyles());  
296    
297                                  selectionMapStyle.setFeatureTypeStyles(newFeatureTypes);                                  DuplicatingStyleVisitor dsv = new DuplicatingStyleVisitor();
298                                    dsv.visit(originalStyle);
299                                    Style newStyle = (Style) dsv.getCopy();
300    
301                                    // SK-Debug
302                                    try {
303                                            //
304                                            StylingUtil.saveStyleToSLD(newStyle, new File(
305                                                            "/home/stefan/Desktop/selection.sld"));
306                                    } catch (Exception e) {
307                                    }
308    
309                                    // DuplicatingStyleVisitor dsv = new DuplicatingStyleVisitor();
310                                    // dsv.visit(originalStyle);
311                                    // Style newStyle = originalStyle;
312    
313                                    // Style newStyle = StylingUtil.STYLE_BUILDER.createStyle();
314                                    // newStyle.featureTypeStyles().addAll(originalStyle.featureTypeStyles());
315                                    mapLayer.setStyle(newStyle);
316    
317                                    replaceRenderer();
318                          }                          }
319    
                         mapLayer.setStyle(selectionMapStyle);  
                         mapPane.refresh();  
320                  } catch (Exception e) {                  } catch (Exception e) {
321                          LOGGER.error("Error while trying to create a selection style", e);                          LOGGER.error("Error while trying to create a selection style", e);
322                  }                  }
323          }          }
324    
325          /**          /**
326             * Analyzes whether the selection has changed in comparison to the selection
327             * stored in the mapLayer.Style
328             *
329             * @param newSelection a List<String> of all newly selected FIDs
330             * @param originalStyle the original {@link Style} that has an earlier selection coded into the {@link Style}
331             * @return <code>true</code> if changed.
332             */
333            private boolean selectionChanged(Vector<String> newSelection,
334                            Style originalStyle) {
335    
336                    boolean SELECTION_STYLING_foundInMapStyle = false;
337    
338                    /**
339                     * testing, whether the selection really changed. If not, we can save
340                     * one rendering!
341                     */
342                    for (FeatureTypeStyle fts : originalStyle.featureTypeStyles()) {
343    
344                            if (fts.getName() != null
345                                            && fts.getName().equals(SELECTION_STYLING_FTS_NAME)) {
346    
347                                    SELECTION_STYLING_foundInMapStyle = true;
348    
349                                    Filter filter = fts.rules().get(0).getFilter();
350                                    if (filter instanceof Not) {
351                                            FidFilterImpl antiOrigFidsFilter = (FidFilterImpl) ((Not) filter)
352                                                            .getFilter();
353    
354                                            // Check one way
355                                            final Set<String> antiFids = antiOrigFidsFilter
356                                                            .getFidsSet();
357    
358                                            if (antiFids.isEmpty() && !newSelection.isEmpty())
359                                                    return true;
360    
361                                            for (String fid : antiFids) {
362                                                    if (newSelection.contains(fid)) {
363                                                            return true;
364                                                    }
365                                            }
366    
367                                            // Check the other way
368                                            for (String fid : newSelection) {
369                                                    if (antiFids.contains(fid)) {
370                                                            return true;
371                                                    }
372                                            }
373    
374                                            if (antiFids.size() + newSelection.size() != styledLayer
375                                                            .getFeatureCollectionFiltered().size())
376                                                    return true;
377    
378                                    } else {
379                                            FidFilterImpl origFidsFilter = (FidFilterImpl) filter;
380    
381                                            // Check one way
382                                            final Set<String> fids = origFidsFilter.getFidsSet();
383    
384                                            if (fids.isEmpty() && newSelection.isEmpty())
385                                                    return false;
386                                            if (fids.size() != newSelection.size())
387                                                    return true;
388    
389                                            for (String fid : fids) {
390                                                    if (!newSelection.contains(fid)) {
391                                                            return true;
392                                                    }
393                                            }
394    
395                                            // Check the other way
396                                            for (String fid : newSelection) {
397                                                    if (!fids.contains(fid)) {
398                                                            return true;
399                                                    }
400                                            }
401    
402                                    }
403    
404                                    break;
405                            }
406                    }
407    
408                    if (!SELECTION_STYLING_foundInMapStyle && !newSelection.isEmpty())
409                            return true;
410    
411                    return false;
412            }
413    
414            /**
415             * Replaces the local renderer
416             */
417            private void replaceRenderer() {
418            }
419    
420            /**
421           * Used to synchronize {@link FeatureSelectedEvent}s with the           * Used to synchronize {@link FeatureSelectedEvent}s with the
422           * {@link StyledFeatureLayerSelectionModel}           * {@link StyledFeatureLayerSelectionModel}
423           */           */
424          @Override          @Override
425          public void performMapPaneEvent(JMapPaneEvent e) {          public void performMapPaneEvent(MapPaneEvent e) {
426    
427                  // Ignore event if it is caused by us or the synchronizer is disabled.                  // Ignore event if it is caused by us or the synchronizer is disabled.
428                  if (!isEnabled() || selectionChangeCausedByMe)                  if (!isEnabled() || selectionChangeCausedByMe)
# Line 280  public class FeatureMapLayerSelectionSyn Line 449  public class FeatureMapLayerSelectionSyn
449                  if (e.getSourceObject() != this.mapPane)                  if (e.getSourceObject() != this.mapPane)
450                          return;                          return;
451    
                 FeatureSelectedEvent fse = (FeatureSelectedEvent) e;  
   
452                  /**                  /**
453                   * Checking, that the FeatureSelectedEvent actually contains features                   * Checking, that the FeatureSelectedEvent actually contains features
454                   * from this layer                   * from this layer
455                   */                   */
456                    FeatureSelectedEvent fse = (FeatureSelectedEvent) e;
457                  final String sourceID = fse.getSourceLayer().getTitle();                  final String sourceID = fse.getSourceLayer().getTitle();
458                  final String syncForID = mapLayer.getTitle();                  final String syncForID = mapLayer.getTitle();
459                  if (sourceID != null && syncForID != null                  if (sourceID != null && syncForID != null
# Line 294  public class FeatureMapLayerSelectionSyn Line 462  public class FeatureMapLayerSelectionSyn
462                          return;                          return;
463                  }                  }
464    
465                  LOGGER.debug("do event " + fse);                  // LOGGER.debug("do event " + fse);
466    
467                  // Avoid event circles in propertyChange(..)                  // Avoid event circles in propertyChange(..)
468                  selectionChangeCausedByMe = true;                  selectionChangeCausedByMe = true;
469    
470                  Collection<Feature> selectionResult = (Collection<Feature>) fse                  final FeatureCollection<SimpleFeatureType, SimpleFeature> selectionResult = fse
471                                  .getSelectionResult();                                  .getSelectionResult();
472                    Iterator<SimpleFeature> fi = selectionResult.iterator();
473                    try {
474    
475                  // reset the selection of the DpLayerSelectionModel                          // reset the selection of the DpLayerSelectionModel
476                  // layerSelModel.setValueIsAdjusting(true);                          // layerSelModel.setValueIsAdjusting(true);
477    
478                  if (selectedTool == MapPaneToolBar.TOOL_SELECTION_ADD) {                          if (selectedTool == MapPaneToolBar.TOOL_SELECTION_ADD) {
479                          layerSelModel.setValueIsAdjusting(true);                                  layerSelModel.setValueIsAdjusting(true);
                         for (Feature f : selectionResult) {  
                                 layerSelModel.addSelection(f.getID());  
                         }  
                         layerSelModel.setValueIsAdjusting(false);  
                 } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_SET) {  
                         layerSelModel.setValueIsAdjusting(true);  
                         layerSelModel.clearSelection();  
480    
481                          for (Feature f : selectionResult) {                                  for (int fIdx = 0; fi.hasNext(); fIdx++) {
482                                  layerSelModel.addSelection(f.getID());                                          SimpleFeature f = fi.next();
483                          }                                          layerSelModel.addSelection(f.getID());
484                                    }
485                                    layerSelModel.setValueIsAdjusting(false);
486                            } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_SET) {
487                                    layerSelModel.setValueIsAdjusting(true);
488                                    layerSelModel.clearSelection();
489    
490                                    for (int fIdx = 0; fi.hasNext(); fIdx++) {
491                                            SimpleFeature f = fi.next();
492                                            layerSelModel.addSelection(f.getID());
493                                    }
494    
495                          LOGGER.debug("Setting selection to " + selectionResult.size());                                  // LOGGER.debug("Setting selection to " + fi.());
496                          layerSelModel.setValueIsAdjusting(false);  
497                  } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_REMOVE) {                                  layerSelModel.setValueIsAdjusting(false);
498                          layerSelModel.setValueIsAdjusting(true);                          } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_REMOVE) {
499                          for (Feature f : selectionResult) {                                  layerSelModel.setValueIsAdjusting(true);
500                                  layerSelModel.removeSelection(f.getID());                                  for (int fIdx = 0; fi.hasNext(); fIdx++) {
501                                            SimpleFeature f = fi.next();
502                                            layerSelModel.removeSelection(f.getID());
503                                    }
504                                    layerSelModel.setValueIsAdjusting(false);
505                          }                          }
506                          layerSelModel.setValueIsAdjusting(false);  
507                    } finally {
508                            selectionResult.close(fi);
509                  }                  }
510    
511                  // Show selected features in the map, which is not automatically done by                  // Show selected features in the map, which is not automatically done by
Legend:
Removed from v.315  
changed lines
  Added in v.604
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26