/[schmitzm]/branches/2.1/src/skrueger/geotools/selection/FeatureMapLayerSelectionSynchronizer.java
ViewVC logotype

Diff of /branches/2.1/src/skrueger/geotools/selection/FeatureMapLayerSelectionSynchronizer.java

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

revision 318 by alfonx, Wed Aug 26 13:33:32 2009 UTC revision 509 by alfonx, Thu Nov 5 08:51:33 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 44  package skrueger.geotools.selection; Line 44  package skrueger.geotools.selection;
44    
45  import java.beans.PropertyChangeEvent;  import java.beans.PropertyChangeEvent;
46  import java.beans.PropertyChangeListener;  import java.beans.PropertyChangeListener;
47  import java.util.Arrays;  import java.io.File;
 import java.util.Collection;  
48  import java.util.HashSet;  import java.util.HashSet;
49    import java.util.Iterator;
50  import java.util.Set;  import java.util.Set;
51  import java.util.Vector;  import java.util.Vector;
52    
# Line 54  import javax.swing.JTable; Line 54  import javax.swing.JTable;
54  import javax.swing.ListSelectionModel;  import javax.swing.ListSelectionModel;
55  import javax.swing.event.ListSelectionListener;  import javax.swing.event.ListSelectionListener;
56    
57  import org.geotools.feature.SimpleFeature;  import org.geotools.feature.FeatureCollection;
58    import org.geotools.feature.FeatureIterator;
59    import org.geotools.filter.FidFilterImpl;
60  import org.geotools.map.MapLayer;  import org.geotools.map.MapLayer;
61  import org.geotools.styling.FeatureTypeStyle;  import org.geotools.styling.FeatureTypeStyle;
62  import org.geotools.styling.Style;  import org.geotools.styling.Style;
63    import org.geotools.styling.visitor.DuplicatingStyleVisitor;
64    import org.opengis.feature.simple.SimpleFeature;
65    import org.opengis.feature.simple.SimpleFeatureType;
66    import org.opengis.filter.Filter;
67    import org.opengis.filter.Not;
68  import org.opengis.filter.identity.FeatureId;  import org.opengis.filter.identity.FeatureId;
69    
70  import schmitzm.geotools.FilterUtil;  import schmitzm.geotools.FilterUtil;
71  import schmitzm.geotools.gui.JMapPane;  import schmitzm.geotools.gui.SelectableXMapPane;
72  import schmitzm.geotools.map.event.FeatureSelectedEvent;  import schmitzm.geotools.map.event.FeatureSelectedEvent;
73  import schmitzm.geotools.map.event.JMapPaneEvent;  import schmitzm.geotools.map.event.MapPaneEvent;
74  import schmitzm.geotools.map.event.JMapPaneListener;  import schmitzm.geotools.map.event.JMapPaneListener;
75  import schmitzm.geotools.styling.StylingUtil;  import schmitzm.geotools.styling.StylingUtil;
 import schmitzm.lang.LangUtil;  
76  import skrueger.geotools.MapPaneToolBar;  import skrueger.geotools.MapPaneToolBar;
77  import skrueger.geotools.StyledLayerInterface;  import skrueger.geotools.StyledFeaturesInterface;
78    
79  /**  /**
80   * This class keeps the selection of a (feature) {@link JTable} synchronized   * This class keeps the selection of a (feature) {@link JTable} synchronized
# Line 97  public class FeatureMapLayerSelectionSyn Line 103  public class FeatureMapLayerSelectionSyn
103           * model.           * model.
104           */           */
105          protected final MapLayer mapLayer;          protected final MapLayer mapLayer;
106          protected final StyledLayerInterface<?> styledLayer;          protected final StyledFeaturesInterface<?> styledLayer;
107          protected final JMapPane mapPane;          protected final SelectableXMapPane mapPane;
108          private final MapPaneToolBar toolBar;          private final MapPaneToolBar toolBar;
109    
110          /**          /**
# Line 113  public class FeatureMapLayerSelectionSyn Line 119  public class FeatureMapLayerSelectionSyn
119           */           */
120          public FeatureMapLayerSelectionSynchronizer(          public FeatureMapLayerSelectionSynchronizer(
121                          StyledFeatureLayerSelectionModel layerSelModel,                          StyledFeatureLayerSelectionModel layerSelModel,
122                          StyledLayerInterface<?> styledLayer, MapLayer mapLayer,                          StyledFeaturesInterface<?> styledLayer, MapLayer mapLayer,
123                          JMapPane mapPane, MapPaneToolBar toolBar) {                          SelectableXMapPane mapPane, MapPaneToolBar toolBar) {
124    
125                  super(layerSelModel);                  super(layerSelModel);
126                  this.styledLayer = styledLayer;                  this.styledLayer = styledLayer;
# Line 170  public class FeatureMapLayerSelectionSyn Line 176  public class FeatureMapLayerSelectionSyn
176           *            A {@link Vector} of SimpleFeature-IDs that are selected.           *            A {@link Vector} of SimpleFeature-IDs that are selected.
177           */           */
178          private void changeLayerStyle(final Vector<String> newSelection) {          private void changeLayerStyle(final Vector<String> newSelection) {
                 Style selectionMapStyle = null;  
179                  try {                  try {
180    
181                            Style originalStyle = mapLayer.getStyle();
182    
183                          if (newSelection.isEmpty()) {                          if (newSelection.isEmpty()) {
184    
185                                  selectionMapStyle = styledLayer.getStyle();                                  // Check if the Style contains a SELECTION FTS
186                                  // LOGGER.debug("NO SELECTION .. set to original style directly");  
187                                    FeatureTypeStyle[] clone = originalStyle.featureTypeStyles()
188                                                    .toArray(new FeatureTypeStyle[] {}).clone();
189    
190                                    for (FeatureTypeStyle fts : clone) {
191                                            if (fts.getName() != null
192                                                            && fts.getName().equals(SELECTION_STYLING)) {
193                                                    originalStyle.featureTypeStyles().remove(fts);
194    
195                                                    replaceRenderer();
196                                                    // mapPane.refresh();
197    
198                                                    return;
199                                            }
200                                    }
201    
202                          } else {                          } else {
203    
204                                  LOGGER.debug("SELECTION .. change style");                                  LOGGER.debug("SELECTION .. change style");
205    
206                                    // Saving a repaint if the selection didn't change
207                                    if (!selectionChanged(newSelection, originalStyle)) {
208                                            return;
209                                    }
210    
211                                  // We take Style from the MapLayer that is displayed at the                                  // We take Style from the MapLayer that is displayed at the
212                                  // moment. We do not use the styledLayer.getStyle, because in                                  // moment. We do not use the styledLayer.getStyle, because in
213                                  // the atlas, this always return the default style, but                                  // the atlas, this always return the default style, but
214                                  // additional styles might be selected.                                  // additional styles might be selected.
215                                  // Taking the style from the mapLayer indicated, that we have to                                  // Taking the style from the mapLayer indicated, that we have to
216                                  // remove any selection rules first.                                  // remove any selection rules first.
                                 Style originalStyle = mapLayer.getStyle();  
217    
218                                  // TODO The default style is not good here. We need                                  FeatureTypeStyle selectionFTStyle = StylingUtil
                                 // .createSelectionStyle(normalStyle, geoObj);  
                                 selectionMapStyle = StylingUtil  
219                                                  .createSelectionStyle(styledLayer.getGeoObject());                                                  .createSelectionStyle(styledLayer.getGeoObject());
                                 // selectionMapStyle = StylingUtil  
                                 // .createDefaultStyle(styledLayer.getGeoObject());  
220    
221                                  selectionMapStyle.getFeatureTypeStyles()[0]                                  selectionFTStyle.setName(SELECTION_STYLING);
                                                 .setName(SELECTION_STYLING);  
222    
223                                  /**                                  /**
224                                   *                                   *
225                                   * Add a Filter to the selectionMapStyle, so that it is only                                   * Add a Filter to the selectionMapStyle, so that it is only
226                                   * used on objects that are selected. <br/>                                   * used on objects that are selected. <br/>
227                                     * To optimize rendering speed, the filter checks whether more
228                                     * than half of the features are selected. If so, the filter is
229                                     * inverted to be shorter.
230                                   *                                   *
                                  * 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!  
231                                   */                                   */
232                                    final FeatureCollection<SimpleFeatureType, SimpleFeature> featureCollectionFiltered = styledLayer
233                                                    .getFeatureCollectionFiltered();
234                                    if (newSelection.size() > featureCollectionFiltered.size() / 2) {
235                                            FeatureIterator<SimpleFeature> iterator = featureCollectionFiltered
236                                                            .features();
237                                            Set<FeatureId> antiFids = new HashSet<FeatureId>();
238                                            try {
239                                                    while (iterator.hasNext()) {
240                                                            SimpleFeature next = iterator.next();
241                                                            if (!newSelection.contains(next.getID()))
242                                                                    antiFids.add(FilterUtil.FILTER_FAC2
243                                                                                    .featureId(next.getID()));
244                                                    }
245    
246                                                    selectionFTStyle.rules().get(0).setFilter(
247                                                                    FilterUtil.FILTER_FAC2
248                                                                                    .not(FilterUtil.FILTER_FAC2
249                                                                                                    .id(antiFids)));
250    
251                                            } finally {
252                                                    featureCollectionFiltered.close(iterator);
253                                            }
254                                    } else {
255                                            Set<FeatureId> fids = new HashSet<FeatureId>();
256                                            for (String fid : newSelection) {
257                                                    fids.add(FilterUtil.FILTER_FAC2.featureId(fid));
258                                            }
259    
260                                  Set<FeatureId> fids = new HashSet<FeatureId>();                                          selectionFTStyle.rules().get(0).setFilter(
261                                  for (String fid : newSelection) {                                                          FilterUtil.FILTER_FAC2.id(fids));
                                         fids.add(FilterUtil.FILTER_FAC2.featureId(fid));  
262                                  }                                  }
263    
264                                  selectionMapStyle.getFeatureTypeStyles()[0].getRules()[0]                                  // Maybe there has already been another selection
265                                                  .setFilter(FilterUtil.FILTER_FAC2.id(fids));                                  // FeatureTypeStyle... Let's replace it...
266                                    boolean foundAndReplaced = false;
267                                  FeatureTypeStyle[] originalFeatureTypeStyles = originalStyle                                  for (FeatureTypeStyle fts : originalStyle.featureTypeStyles()) {
268                                                  .getFeatureTypeStyles();                                          if (fts.getName() != null
269                                  FeatureTypeStyle[] newFeatureTypes;                                                          && fts.getName().equals(SELECTION_STYLING)) {
270                                  if (originalFeatureTypeStyles[originalFeatureTypeStyles.length - 1]                                                  foundAndReplaced = true;
271                                                  .getName() != null                                                  fts.rules().clear();
272                                                  && originalFeatureTypeStyles[originalFeatureTypeStyles.length - 1]                                                  fts.rules().addAll(selectionFTStyle.rules());
273                                                                  .getName().equals(SELECTION_STYLING)) {                                                  break;
274                                          newFeatureTypes = Arrays.copyOf(originalFeatureTypeStyles,                                          }
                                                         originalFeatureTypeStyles.length - 1);  
                                 } else {  
                                         newFeatureTypes = originalFeatureTypeStyles;  
275                                  }                                  }
276                                    if (!foundAndReplaced) {
277                                            originalStyle.featureTypeStyles().add(selectionFTStyle);
278                                    }
279    
280                                    // Refresh the map
281                                    // mapPane.refresh();
282    
283                                  // The last FTS is the selection FTS                                  DuplicatingStyleVisitor dsv = new DuplicatingStyleVisitor();
284                                  newFeatureTypes = LangUtil.extendArray(newFeatureTypes,                                  dsv.visit(originalStyle);
285                                                  selectionMapStyle.getFeatureTypeStyles());                                  Style newStyle = (Style) dsv.getCopy();
286    
287                                    // SK-Debug
288                                    try {
289                                            //
290                                            StylingUtil.saveStyleToSLD(newStyle, new File(
291                                                            "/home/stefan/Desktop/selection.sld"));
292                                    } catch (Exception e) {
293                                    }
294    
295                                  selectionMapStyle.setFeatureTypeStyles(newFeatureTypes);                                  // DuplicatingStyleVisitor dsv = new DuplicatingStyleVisitor();
296                                    // dsv.visit(originalStyle);
297                                    // Style newStyle = originalStyle;
298    
299                                    // Style newStyle = StylingUtil.STYLE_BUILDER.createStyle();
300                                    // newStyle.featureTypeStyles().addAll(originalStyle.featureTypeStyles());
301                                    mapLayer.setStyle(newStyle);
302    
303                                    replaceRenderer();
304                          }                          }
305    
                         mapLayer.setStyle(selectionMapStyle);  
                         mapPane.refresh();  
306                  } catch (Exception e) {                  } catch (Exception e) {
307                          LOGGER.error("Error while trying to create a selection style", e);                          LOGGER.error("Error while trying to create a selection style", e);
308                  }                  }
309          }          }
310    
311          /**          /**
312             * Analyses whether the selection has changed in comparison to the selection
313             * stored in the mapLayer.Style
314             *
315             * @param newSelection
316             * @param originalStyle
317             * @return
318             */
319            private boolean selectionChanged(Vector<String> newSelection,
320                            Style originalStyle) {
321    
322                    boolean SELECTION_STYLING_foundInMapStyle = false;
323    
324                    /**
325                     * testing, whether the selection really changed. If not, we can save
326                     * one paint!
327                     */
328                    for (FeatureTypeStyle fts : originalStyle.featureTypeStyles()) {
329    
330                            if (fts.getName() != null
331                                            && fts.getName().equals(SELECTION_STYLING)) {
332    
333                                    SELECTION_STYLING_foundInMapStyle = true;
334    
335                                    Filter filter = fts.rules().get(0).getFilter();
336                                    if (filter instanceof Not) {
337                                            FidFilterImpl antiOrigFidsFilter = (FidFilterImpl) ((Not) filter)
338                                                            .getFilter();
339    
340                                            // Check one way
341                                            final Set<String> antiFids = antiOrigFidsFilter
342                                                            .getFidsSet();
343    
344                                            if (antiFids.isEmpty() && !newSelection.isEmpty())
345                                                    return true;
346    
347                                            for (String fid : antiFids) {
348                                                    if (newSelection.contains(fid)) {
349                                                            return true;
350                                                    }
351                                            }
352    
353                                            // Check the other way
354                                            for (String fid : newSelection) {
355                                                    if (antiFids.contains(fid)) {
356                                                            return true;
357                                                    }
358                                            }
359    
360                                    } else {
361                                            FidFilterImpl origFidsFilter = (FidFilterImpl) filter;
362    
363                                            // Check one way
364                                            final Set<String> fids = origFidsFilter.getFidsSet();
365    
366                                            if (fids.isEmpty() && newSelection.isEmpty())
367                                                    return false;
368                                            if (fids.size() != newSelection.size())
369                                                    return true;
370    
371                                            for (String fid : fids) {
372                                                    if (!newSelection.contains(fid)) {
373                                                            return true;
374                                                    }
375                                            }
376    
377                                            // Check the other way
378                                            for (String fid : newSelection) {
379                                                    if (!fids.contains(fid)) {
380                                                            return true;
381                                                    }
382                                            }
383    
384                                    }
385    
386                                    break;
387                            }
388                    }
389    
390                    if (!SELECTION_STYLING_foundInMapStyle && !newSelection.isEmpty())
391                            return true;
392    
393                    return false;
394            }
395    
396            /**
397             * Replaces the local renderer
398             */
399            private void replaceRenderer() {
400    //              //
401    //              // // Has to be done before we apply the new Renderer
402    //              // mapLayer.setStyle(style);
403    //
404    //              GTRenderer oldRenderer = mapPane.getLocalRenderer();
405    //
406    //              /**
407    //               * Explicitly putting a new instance of LabelCacheDefault into the
408    //               * renderer instance, so JMapPane doesn't reuse the old one. This is
409    //               * very useful when changing the TextSymbolizer with AtlasStyler<br/>
410    //               * SK 9.7.09: It's not enought to user LabelCache.clear(). We can not
411    //               * reuse the old Renderer - better to create a new one!
412    //               */
413    //              final GTRenderer newRenderer = GTUtil.createGTRenderer();
414    //
415    //              final HashMap<Object, Object> rendererHints = defaultGTRenderingHints;
416    //              rendererHints.put(StreamingRenderer.LABEL_CACHE_KEY,
417    //                              new LabelCacheImpl());
418    //
419    //              newRenderer.setRendererHints(rendererHints);
420    //              mapPane.setLocalRenderer(newRenderer);
421    //
422    //              if (oldRenderer != null) {
423    //
424    //                      RenderingHints java2DHints = oldRenderer.getJava2DHints();
425    //                      if (java2DHints != null) {
426    //                              newRenderer.setJava2DHints(java2DHints);
427    //                      }
428    //
429    //                      oldRenderer.setContext(null);
430    //                      oldRenderer = null;
431    //              }
432    
433                    mapPane.refresh();
434    
435            }
436    
437            /**
438           * Used to synchronize {@link FeatureSelectedEvent}s with the           * Used to synchronize {@link FeatureSelectedEvent}s with the
439           * {@link StyledFeatureLayerSelectionModel}           * {@link StyledFeatureLayerSelectionModel}
440           */           */
441          @Override          @Override
442          public void performMapPaneEvent(JMapPaneEvent e) {          public void performMapPaneEvent(MapPaneEvent e) {
443    
444                  // 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.
445                  if (!isEnabled() || selectionChangeCausedByMe)                  if (!isEnabled() || selectionChangeCausedByMe)
# Line 280  public class FeatureMapLayerSelectionSyn Line 466  public class FeatureMapLayerSelectionSyn
466                  if (e.getSourceObject() != this.mapPane)                  if (e.getSourceObject() != this.mapPane)
467                          return;                          return;
468    
                 FeatureSelectedEvent fse = (FeatureSelectedEvent) e;  
   
469                  /**                  /**
470                   * Checking, that the FeatureSelectedEvent actually contains features                   * Checking, that the FeatureSelectedEvent actually contains features
471                   * from this layer                   * from this layer
472                   */                   */
473                    FeatureSelectedEvent fse = (FeatureSelectedEvent) e;
474                  final String sourceID = fse.getSourceLayer().getTitle();                  final String sourceID = fse.getSourceLayer().getTitle();
475                  final String syncForID = mapLayer.getTitle();                  final String syncForID = mapLayer.getTitle();
476                  if (sourceID != null && syncForID != null                  if (sourceID != null && syncForID != null
# Line 294  public class FeatureMapLayerSelectionSyn Line 479  public class FeatureMapLayerSelectionSyn
479                          return;                          return;
480                  }                  }
481    
482                  LOGGER.debug("do event " + fse);                  // LOGGER.debug("do event " + fse);
483    
484                  // Avoid event circles in propertyChange(..)                  // Avoid event circles in propertyChange(..)
485                  selectionChangeCausedByMe = true;                  selectionChangeCausedByMe = true;
486    
487                  Collection<SimpleFeature> selectionResult = (Collection<SimpleFeature>) fse                  final FeatureCollection<SimpleFeatureType, SimpleFeature> selectionResult = fse
488                                  .getSelectionResult();                                  .getSelectionResult();
489                    Iterator<SimpleFeature> fi = selectionResult.iterator();
490                    try {
491    
492                  // reset the selection of the DpLayerSelectionModel                          // reset the selection of the DpLayerSelectionModel
493                  // layerSelModel.setValueIsAdjusting(true);                          // layerSelModel.setValueIsAdjusting(true);
494    
495                  if (selectedTool == MapPaneToolBar.TOOL_SELECTION_ADD) {                          if (selectedTool == MapPaneToolBar.TOOL_SELECTION_ADD) {
496                          layerSelModel.setValueIsAdjusting(true);                                  layerSelModel.setValueIsAdjusting(true);
                         for (SimpleFeature f : selectionResult) {  
                                 layerSelModel.addSelection(f.getID());  
                         }  
                         layerSelModel.setValueIsAdjusting(false);  
                 } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_SET) {  
                         layerSelModel.setValueIsAdjusting(true);  
                         layerSelModel.clearSelection();  
497    
498                          for (SimpleFeature f : selectionResult) {                                  for (int fIdx = 0; fi.hasNext(); fIdx++) {
499                                  layerSelModel.addSelection(f.getID());                                          SimpleFeature f = fi.next();
500                          }                                          layerSelModel.addSelection(f.getID());
501                                    }
502                                    layerSelModel.setValueIsAdjusting(false);
503                            } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_SET) {
504                                    layerSelModel.setValueIsAdjusting(true);
505                                    layerSelModel.clearSelection();
506    
507                                    for (int fIdx = 0; fi.hasNext(); fIdx++) {
508                                            SimpleFeature f = fi.next();
509                                            layerSelModel.addSelection(f.getID());
510                                    }
511    
512                          LOGGER.debug("Setting selection to " + selectionResult.size());                                  // LOGGER.debug("Setting selection to " + fi.());
513                          layerSelModel.setValueIsAdjusting(false);  
514                  } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_REMOVE) {                                  layerSelModel.setValueIsAdjusting(false);
515                          layerSelModel.setValueIsAdjusting(true);                          } else if (selectedTool == MapPaneToolBar.TOOL_SELECTION_REMOVE) {
516                          for (SimpleFeature f : selectionResult) {                                  layerSelModel.setValueIsAdjusting(true);
517                                  layerSelModel.removeSelection(f.getID());                                  for (int fIdx = 0; fi.hasNext(); fIdx++) {
518                                            SimpleFeature f = fi.next();
519                                            layerSelModel.removeSelection(f.getID());
520                                    }
521                                    layerSelModel.setValueIsAdjusting(false);
522                          }                          }
523                          layerSelModel.setValueIsAdjusting(false);  
524                    } finally {
525                            selectionResult.close(fi);
526                  }                  }
527    
528                  // 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.318  
changed lines
  Added in v.509
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26