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

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

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

revision 52 by mojays, Fri Apr 17 13:57:14 2009 UTC revision 112 by mojays, Wed May 13 09:18:53 2009 UTC
# Line 10  Line 10 
10   **/   **/
11  package skrueger.geotools;  package skrueger.geotools;
12    
13    
14  import java.util.Iterator;  import java.util.Iterator;
15  import java.util.Map;  import java.util.Map;
16  import java.util.TreeMap;  import java.util.TreeMap;
17  import java.util.Vector;  import java.util.Vector;
18    
19    import org.apache.log4j.Logger;
20  import org.geotools.data.DefaultQuery;  import org.geotools.data.DefaultQuery;
21  import org.geotools.data.FeatureSource;  import org.geotools.data.FeatureSource;
22  import org.geotools.data.Query;  import org.geotools.data.Query;
# Line 22  import org.geotools.data.memory.MemoryDa Line 24  import org.geotools.data.memory.MemoryDa
24  import org.geotools.feature.FeatureCollection;  import org.geotools.feature.FeatureCollection;
25  import org.opengis.filter.Filter;  import org.opengis.filter.Filter;
26    
27    import com.vividsolutions.jts.geom.Envelope;
28    
29  import schmitzm.geotools.gui.FeatureCollectionTableModel;  import schmitzm.geotools.gui.FeatureCollectionTableModel;
30  import skrueger.AttributeMetaData;  import skrueger.AttributeMetaData;
31    import skrueger.i8n.I8NUtil;
32    import skrueger.i8n.Translation;
33    
34  /**  /**
35   * This class extends the   * This class extends the the {@link FeatureCollectionTableModel} with the
36     * functionalities of the {@link AttributeMetaData} of
37     * {@linkplain StyledMapInterface styled objects}.
38     * <ul>
39     *   <li>column names are translated according to {@link AttributeMetaData#getTitle()}</li>
40     *   <li>columns are hidden according to {@link AttributeMetaData#isVisible()()}</li>
41     * </ul>
42   * @author <a href="mailto:[email protected]">Martin Schmitz</a> (University of Bonn/Germany)   * @author <a href="mailto:[email protected]">Martin Schmitz</a> (University of Bonn/Germany)
  *  
43   */   */
44  public class StyledFeatureCollectionTableModel extends FeatureCollectionTableModel {  public class StyledFeatureCollectionTableModel extends FeatureCollectionTableModel {
45        final static private Logger LOGGER = Logger.getLogger(StyledFeatureCollectionTableModel.class);
46    protected Map<Integer,AttributeMetaData> visibleAMD = null;    /** Holds the data source as styled map. */
47        protected StyledMapInterface<?> map = null;
48      /** Contains only the visible elements of the {@link AttributeMetaData}-Map */
49      protected Map<Integer, AttributeMetaData> visibleAMD = null;
50      /** Holds the data source for the table as {@code FeatureSource}. */
51      protected FeatureSource featureSource = null;
52      /** Contains the complete {@link AttributeMetaData}-Map of the styled layer. */
53      protected Map<Integer, AttributeMetaData> origAMD = null;
54      /** Holds the current filter on the table */
55      protected Filter filter = null;
56      /** Holds the Bounds for all features. Only set once during the constructor **/
57      protected Envelope bounds;
58    
59      /**
60       * Creates a new table model for a styled map.
61       * @param map the styled map
62       */
63    public StyledFeatureCollectionTableModel(StyledFeatureCollectionInterface map) {    public StyledFeatureCollectionTableModel(StyledFeatureCollectionInterface map) {
64        this(map,Filter.INCLUDE);
65      }
66    
67      /**
68       * Creates a new table model for a styled map.
69       * @param map the styled map
70       * @param filter filter applied to the table
71       */
72      public StyledFeatureCollectionTableModel(StyledFeatureCollectionInterface map, Filter filter) {
73      super();      super();
74      setFeatureCollection(map);      setFeatureCollection(map, filter);
75    }    }
76      
77      /**
78       * Creates a new table model for a styled map.
79       * @param map the styled map
80       */
81    public StyledFeatureCollectionTableModel(StyledFeatureSourceInterface map) {    public StyledFeatureCollectionTableModel(StyledFeatureSourceInterface map) {
82        this(map,Filter.INCLUDE);
83      }
84    
85      /**
86       * Creates a new table model for a styled map.
87       * @param map the styled map
88       * @param filter filter applied to the table
89       */
90      public StyledFeatureCollectionTableModel(StyledFeatureSourceInterface map, Filter filter) {
91      super();      super();
92      setFeatureCollection(map);      setFeatureCollection(map, filter);
93    }    }
94      
95    protected void setFeatureSource(FeatureSource fs, Map<Integer,AttributeMetaData> amd) throws Exception {    /**
96       * Sets a new data source for the table.
97       * @param fs     the feature source
98       * @param amd    {@link AttributeMetaData}-Map to define the visible attributes
99       *               and translation
100       */
101      protected void setFeatureSource(FeatureSource fs, Map<Integer, AttributeMetaData> amd, Filter filter) throws Exception {
102        if ( filter == null )
103          filter = Filter.INCLUDE;
104        
105        this.featureSource = fs;
106        this.filter        = filter;
107        this.origAMD       = amd;
108        this.visibleAMD    = null;
109        
110      FeatureCollection fc = null;      FeatureCollection fc = null;
111      this.visibleAMD      = new TreeMap<Integer, AttributeMetaData>();      if (fs != null) {
112      if ( fs != null ) {          
113        Query query = new DefaultQuery();         bounds = fs.getBounds();
114        if ( amd != null ) {          
115          Query query = new DefaultQuery(fs.getSchema().getTypeName(), filter);
116          if (amd != null) {
117          // determine the names of the visible Attributes          // determine the names of the visible Attributes
118          this.visibleAMD = StyledMapUtil.getVisibleAttributeMetaData(amd, true);          this.visibleAMD = StyledMapUtil.getVisibleAttributeMetaData(amd, true);
119          Vector<String> visibleAttrNames = new Vector<String>();          Vector<String> visibleAttrNames = new Vector<String>();
120          for ( int attrIdx : visibleAMD.keySet() )          // Add the column with the geometry (usually "the_geom")
121            visibleAttrNames.add( fs.getSchema().getAttributeType(attrIdx).getLocalName() );          visibleAttrNames.add(fs.getSchema().getDefaultGeometry().getLocalName());
122          // create a query for the visible attributes            for (int attrIdx : visibleAMD.keySet())
123          query = new DefaultQuery(            visibleAttrNames.add(fs.getSchema().getAttributeType(attrIdx).getLocalName());
124                         fs.getSchema().getTypeName(),  
125                         Filter.INCLUDE,          // create a query for the visible attributes
126                         visibleAttrNames.toArray(new String[0])          String[] properties = visibleAttrNames.toArray(new String[0]);
127          );          
128            LOGGER.debug("Query contains the following attributes: " + visibleAttrNames);
129            
130            query = new DefaultQuery(fs.getSchema().getTypeName(), filter, properties);
131        }        }
132        fc = fs.getFeatures(query);        fc = fs.getFeatures(query);
133    
134    // FAILS:!!!, even with  query = new DefaultQuery(fs.getSchema().getTypeName(), filter);
135                            // java.lang.UnsupportedOperationException: Unknown feature
136                            // attribute: PQM_MOD
137                            // at
138                            // schmitzm.geotools.feature.FeatureOperationTree.evaluate(FeatureOperationTree.java:93)
139                            // bounds = fc.getBounds();
140                            // SK, 17.4.2009
141                            //      
142                            // System.out.println("Filter = "+filter);
143                            // System.out.println("Size of FC = "+fc.size());
144                            // System.out.println("anz att= "+fc.getNumberOfAttributes());
145      }      }
146      setFeatureCollection( fc );      setFeatureCollection(fc);
147    }    }
148      
149    public void setFeatureCollection(StyledFeatureCollectionInterface map) {    /**
150       * Converts the {@code StyledFeatureCollection} to a {@code FeatureSource}
151       * and sets this as the new data source for the table.
152       * @param fs     the feature source
153       * @param amd    {@link AttributeMetaData}-Map to define the visible attributes
154       *               and translation
155       */
156      public void setFeatureCollection(StyledFeatureCollectionInterface map, Filter filter) {
157        this.map = map;
158      try {      try {
159        if ( map == null )        if (map == null)
160          setFeatureSource(null,null);          setFeatureSource(null, null, null);
161        else {        else {
162          FeatureCollection fc     = map.getGeoObject();          FeatureCollection fc = map.getGeoObject();
163          String            fcName = fc.getFeatureType().getTypeName();          String fcName = fc.getSchema().getTypeName();
164          FeatureSource     fs     = new MemoryDataStore(fc).getFeatureSource(fcName);          FeatureSource fs = new MemoryDataStore(fc).getFeatureSource(fcName);
165          setFeatureSource(fs, map.getAttributeMetaDataMap());          setFeatureSource(fs, map.getAttributeMetaDataMap(), filter);
166        }        }
167      } catch (Exception err) {      } catch (Exception err) {
168        throw new RuntimeException(err);        throw new RuntimeException(err);
169      }      }
170    }    }
171      
172    public void setFeatureCollection(StyledFeatureSourceInterface map) {    /**
173       * Sets the {@code StyledFeatureCollection} as new data source for the table.
174       * @param fs     the feature source
175       * @param amd    {@link AttributeMetaData}-Map to define the visible attributes
176       *               and translation
177       */
178      public void setFeatureCollection(StyledFeatureSourceInterface map, Filter filter) {
179        this.map = map;
180      try {      try {
181        if ( map == null )        if (map == null)
182          setFeatureSource(null,null);          setFeatureSource(null, null, null);
183        else        else
184          setFeatureSource(map.getGeoObject(), map.getAttributeMetaDataMap());          setFeatureSource(map.getGeoObject(), map.getAttributeMetaDataMap(), filter);
185      } catch (Exception err) {      } catch (Exception err) {
186        throw new RuntimeException(err);        throw new RuntimeException(err);
187      }      }
188    }    }
189        
190      /**
191       * Resets the filter for the table.
192       * @param filter a filter
193       */
194      public void setFilter(Filter filter) {
195        try{
196          setFeatureSource(this.featureSource, this.origAMD, filter);
197        } catch (Exception err) {
198            LOGGER.error("Setting the filter of the table model", err);
199          throw new RuntimeException(err);
200        }
201      }
202      
203      /**
204       * @return <code>Filter.INCLUDE</code> or the {@link Filter} applied to the Features
205       */
206      public Filter getFilter() {
207              return this.filter;
208      }
209    
210      /**
211       * After calling {@code super.reorganize(.)} this method replaced the
212       * column descriptions with the titles of the {@code AttributeMetaData}.
213       * @param fireTableStructureChanged indicates whether a table event is
214       *        initiated after reorganize
215       */
216    @Override    @Override
217    public void reorganize() {    protected void reorganize(boolean fireTableStructureChanged) {
218      super.reorganize();      super.reorganize(false);
219      // translate the column names      // translate the column names
220      Iterator<Integer> keys = visibleAMD.keySet().iterator();      if (visibleAMD != null) {
221      for (int i=0; keys.hasNext(); i++)        Iterator<Integer> keys = visibleAMD.keySet().iterator();
222        colNames[i] = visibleAMD.get( keys.next() ).getTitle().toString();        for (int i = 0; i < colNames.length && keys.hasNext(); i++) {
223            Translation title = visibleAMD.get(keys.next()).getTitle();
224            if (!I8NUtil.isEmpty(title)) {
225    //          System.out.println("set colname " + i + " to " + title.toString());
226              colNames[i] = title.toString();
227            }
228          }
229        }
230        if ( fireTableStructureChanged )
231          fireTableStructureChanged();
232    }    }
233    
234      /**
235             * @returns Cached bounds for the whole dataset (without applying the
236             *          filter) or <code>null</code>
237             */
238            public Envelope getBounds() {
239                    return bounds;
240            }
241  }  }
Legend:
Removed from v.52  
changed lines
  Added in v.112
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26