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