25 |
* |
* |
26 |
* Contributors: |
* Contributors: |
27 |
* Martin O. J. Schmitz - initial API and implementation |
* Martin O. J. Schmitz - initial API and implementation |
28 |
* Stefan A. Krüger - additional utility classes |
* Stefan A. Tzeggai - additional utility classes |
29 |
******************************************************************************/ |
******************************************************************************/ |
30 |
package skrueger.geotools; |
package skrueger.geotools; |
31 |
|
|
32 |
|
import java.util.HashMap; |
33 |
|
import java.util.List; |
34 |
import java.util.Vector; |
import java.util.Vector; |
35 |
|
|
36 |
import org.apache.log4j.Logger; |
import org.apache.log4j.Logger; |
43 |
import org.opengis.feature.type.AttributeDescriptor; |
import org.opengis.feature.type.AttributeDescriptor; |
44 |
import org.opengis.filter.Filter; |
import org.opengis.filter.Filter; |
45 |
|
|
46 |
|
import schmitzm.geotools.feature.FeatureUtil; |
47 |
import schmitzm.geotools.gui.FeatureCollectionTableModel; |
import schmitzm.geotools.gui.FeatureCollectionTableModel; |
48 |
import skrueger.AttributeMetadata; |
import skrueger.AttributeMetadataImpl; |
49 |
|
import skrueger.AttributeMetadataInterface; |
50 |
|
|
51 |
import com.vividsolutions.jts.geom.Envelope; |
import com.vividsolutions.jts.geom.Envelope; |
52 |
|
|
53 |
/** |
/** |
54 |
* This class extends the the {@link FeatureCollectionTableModel} with the |
* This class extends the the {@link FeatureCollectionTableModel} with the |
55 |
* functionalities of the {@link AttributeMetadata}. |
* functionalities of the {@link AttributeMetadataImpl}. |
56 |
* <ul> |
* <ul> |
57 |
* <li>column names are translated according to |
* <li>column names are translated according to |
58 |
* {@link AttributeMetadata#getTitle()}</li> |
* {@link AttributeMetadataImpl#getTitle()}</li> |
59 |
* <li>columns are hidden according to {@link AttributeMetaData#isVisible()()}</li> |
* <li>columns are hidden according to {@link AttributeMetaData#isVisible()()}</li> |
60 |
* <li>Any filter defined in the {@link StyledFeaturesInterface} will be applied.</li> |
* <li>Any filter defined in the {@link StyledFeaturesInterface} will be |
61 |
|
* applied.</li> |
62 |
* </ul> |
* </ul> |
63 |
* |
* |
64 |
* @author Stefan A. Krüger |
* @author Stefan A. Tzeggai |
65 |
*/ |
*/ |
66 |
public class StyledFeatureCollectionTableModel extends |
public class StyledFeatureCollectionTableModel extends |
67 |
FeatureCollectionTableModel { |
FeatureCollectionTableModel { |
68 |
final static private Logger LOGGER = Logger |
final static private Logger LOGGER = Logger |
69 |
.getLogger(StyledFeatureCollectionTableModel.class); |
.getLogger(StyledFeatureCollectionTableModel.class); |
70 |
/** Contains the complete {@link AttributeMetadata}-Map of the styled layer. */ |
/** Contains the complete {@link AttributeMetadataImpl}-Map of the styled layer. */ |
71 |
protected AttributeMetadataMap origAMD = null; |
protected AttributeMetadataMap<? extends AttributeMetadataInterface> amdMap = null; |
72 |
/** Holds the current filter on the table */ |
/** Holds the current filter on the table */ |
73 |
protected Filter filter = null; |
protected Filter filter = null; |
74 |
/** Holds the Bounds for all features. Only set once during the constructor **/ |
/** Holds the Bounds for all features. Only set once during the constructor **/ |
75 |
protected Envelope bounds; |
protected Envelope bounds; |
76 |
|
/** |
77 |
|
* Tooltips für die Spaltennamen. Wird nur beim Aufruf von |
78 |
|
* {@link #reorganize} befuellt. |
79 |
|
*/ |
80 |
|
protected String[] colTooltips = null; |
81 |
|
|
82 |
|
/** A cache for the #sortedValuesVisibleOnly() **/ |
83 |
|
protected List<? extends AttributeMetadataInterface> amdMapVisibleOnly = null; |
84 |
|
|
|
|
|
85 |
/** |
/** |
86 |
* Creates a new table model for a styled layer. |
* Creates a new table model for a styled layer. |
87 |
* |
* |
90 |
* @param filter |
* @param filter |
91 |
* filter applied to the table |
* filter applied to the table |
92 |
*/ |
*/ |
93 |
public StyledFeatureCollectionTableModel(StyledFeaturesInterface<?> styledFeatures) { |
public StyledFeatureCollectionTableModel( |
94 |
|
StyledFeaturesInterface<?> styledFeatures) { |
95 |
setStyledFeatures(styledFeatures); |
setStyledFeatures(styledFeatures); |
96 |
} |
} |
97 |
|
|
98 |
/** |
/** |
99 |
|
* This overwritten method filters the values for NODATA-values defined in |
100 |
|
* the {@link AttributeMetadataImpl} |
101 |
|
*/ |
102 |
|
@Override |
103 |
|
public Object getValueAt(int row, int col) { |
104 |
|
Object rawValue = super.getValueAt(row, col); |
105 |
|
return amdMap.sortedValuesVisibleOnly().get(col).fiterNodata(rawValue); |
106 |
|
} |
107 |
|
|
108 |
|
/** |
109 |
* Sets a new data source for the table. |
* Sets a new data source for the table. |
110 |
* |
* |
111 |
* @param fs |
* @param fs |
112 |
* the feature source |
* the feature source |
113 |
* @param amd |
* @param amdm |
114 |
* {@link AttributeMetadata}-Map to define the visible attributes |
* {@link AttributeMetadataImpl}-Map to define the visible attributes |
115 |
* and translation |
* and translation |
116 |
*/ |
*/ |
117 |
protected void setFeatureSource(FeatureSource<SimpleFeatureType, SimpleFeature> fs, |
protected void setFeatureSource( |
118 |
AttributeMetadataMap amd, Filter filter) |
FeatureSource<SimpleFeatureType, SimpleFeature> fs, |
119 |
throws Exception { |
AttributeMetadataMap<? extends AttributeMetadataInterface> amdm, Filter filter) throws Exception { |
120 |
|
|
121 |
if (filter == null) |
if (filter == null) |
122 |
filter = Filter.INCLUDE; |
filter = Filter.INCLUDE; |
123 |
|
|
124 |
// this.featureSource = fs; |
// this.featureSource = fs; |
125 |
this.filter = filter; |
this.filter = filter; |
126 |
this.origAMD = amd; |
this.amdMap = amdm; |
127 |
|
this.amdMapVisibleOnly = amdMap.sortedValuesVisibleOnly(); |
128 |
|
|
129 |
FeatureCollection<SimpleFeatureType, SimpleFeature> fc = null; |
FeatureCollection<SimpleFeatureType, SimpleFeature> fc = null; |
130 |
if (fs != null) { |
if (fs != null) { |
133 |
|
|
134 |
final SimpleFeatureType schema = fs.getSchema(); |
final SimpleFeatureType schema = fs.getSchema(); |
135 |
Query query = new DefaultQuery(schema.getTypeName(), filter); |
Query query = new DefaultQuery(schema.getTypeName(), filter); |
136 |
if (amd != null) { |
if (amdm != null) { |
137 |
Vector<String> visibleAttrNames = new Vector<String>(); |
Vector<String> visibleAttrNames = new Vector<String>(); |
|
// Add the column with the geometry (usually "the_geom") |
|
|
|
|
|
for (AttributeDescriptor aDesc : schema.getAttributeDescriptors()) { |
|
|
|
|
|
// Always add the geometry |
|
|
if (schema.getGeometryDescriptor() |
|
|
.getName().equals(aDesc.getName())) { |
|
|
visibleAttrNames.add(schema.getGeometryDescriptor() |
|
|
.getLocalName()); |
|
|
continue; |
|
|
} |
|
|
|
|
|
if (amd.get(aDesc.getName()).isVisible()) |
|
|
visibleAttrNames.add(aDesc.getName().getLocalPart()); |
|
|
} |
|
|
|
|
|
// create a query for the visible attributes |
|
|
String[] properties = visibleAttrNames.toArray(new String[] {}); |
|
|
|
|
|
LOGGER.debug("Query contains the following attributes: " |
|
|
+ visibleAttrNames); |
|
138 |
|
|
139 |
|
// Add the column with the geometry (usually "the_geom") always |
140 |
/** |
visibleAttrNames.add(schema.getGeometryDescriptor() |
141 |
* I got NPEs when properties contained only [the_geom] ?!??!!?? |
.getLocalName()); |
142 |
*/ |
|
143 |
if (properties.length > 1) { |
// Add other visible attributes as ordered by weights |
144 |
query = new DefaultQuery(schema.getTypeName(), filter, |
for (AttributeMetadataInterface a : amdMapVisibleOnly) { |
145 |
properties); |
visibleAttrNames.add(a.getLocalName()); |
|
} else { |
|
|
query = new DefaultQuery(schema.getTypeName(), filter); |
|
146 |
} |
} |
147 |
|
|
148 |
|
// Tested with 2.6.x trunk from 2009-11-26 and it now works. So |
149 |
|
// we only request the properties we need! |
150 |
|
// /** |
151 |
|
// * I got NPEs when properties contained only [the_geom] |
152 |
|
// ?!??!!?? |
153 |
|
// */ |
154 |
|
// if (properties.length > 1) { |
155 |
|
query = new DefaultQuery(schema.getTypeName(), filter, |
156 |
|
visibleAttrNames.toArray(new String[] {})); |
157 |
|
// } else { |
158 |
|
// query = new DefaultQuery(schema.getTypeName(), filter); |
159 |
|
// } |
160 |
} |
} |
161 |
fc = fs.getFeatures(query); |
fc = fs.getFeatures(query); |
162 |
} |
} |
170 |
* @param fs |
* @param fs |
171 |
* the feature source |
* the feature source |
172 |
* @param amd |
* @param amd |
173 |
* {@link AttributeMetadata}-Map to define the visible attributes |
* {@link AttributeMetadataImpl}-Map to define the visible attributes |
174 |
* and translation |
* and translation |
175 |
*/ |
*/ |
176 |
public void setStyledFeatures(StyledFeaturesInterface<?> styledFeatures) { |
public void setStyledFeatures(StyledFeaturesInterface<?> styledFeatures) { |
178 |
if (styledFeatures == null) |
if (styledFeatures == null) |
179 |
setFeatureSource(null, null, null); |
setFeatureSource(null, null, null); |
180 |
else { |
else { |
181 |
setFeatureSource(styledFeatures.getFeatureSource(), styledFeatures.getAttributeMetaDataMap(), styledFeatures.getFilter()); |
setFeatureSource(styledFeatures.getFeatureSource(), |
182 |
|
styledFeatures.getAttributeMetaDataMap(), |
183 |
|
styledFeatures.getFilter()); |
184 |
} |
} |
185 |
} catch (Exception err) { |
} catch (Exception err) { |
186 |
throw new RuntimeException(err); |
throw new RuntimeException(err); |
196 |
*/ |
*/ |
197 |
@Override |
@Override |
198 |
protected void reorganize(boolean fireTableStructureChanged) { |
protected void reorganize(boolean fireTableStructureChanged) { |
199 |
|
|
200 |
super.reorganize(false); |
featureArray = FeatureUtil.featuresToArray(featureTable); |
201 |
|
if (featureArray == null || featureArray.length == 0) { |
202 |
// translate the column names |
colNames = new String[0]; |
203 |
if (origAMD != null) { |
colTooltips = new String[0]; // Only set and used in |
204 |
|
// StyledFeatureCollectionTableModel |
205 |
|
colClass = new Class[0]; |
206 |
|
} else { |
207 |
|
// Struktur der Tabelle vom AttributeMetaDtaaMap übernehmen |
208 |
|
SimpleFeatureType schema = featureArray[0].getFeatureType(); |
209 |
|
// Pruefen, welche Attribute angezeigt werden |
210 |
|
attrTypes.clear(); |
211 |
|
for (AttributeMetadataInterface amd : amdMapVisibleOnly) { |
212 |
|
AttributeDescriptor type = schema.getDescriptor(amd.getName()); |
213 |
|
if (attrFilter == null || attrFilter.accept(type)) |
214 |
|
attrTypes.add(type); |
215 |
|
} |
216 |
|
// Namen und Attribut-Indizes der angezeigten Spalten ermitteln |
217 |
|
colNames = new String[attrTypes.size()]; |
218 |
|
colTooltips = new String[attrTypes.size()]; // Only set and used in |
219 |
|
// StyledFeatureCollectionTableModel |
220 |
|
colClass = new Class[attrTypes.size()]; |
221 |
|
attrIdxForCol = new int[attrTypes.size()]; |
222 |
for (int i = 0; i < colNames.length; i++) { |
for (int i = 0; i < colNames.length; i++) { |
223 |
colNames[i] = origAMD.get(colNames[i]).getTitle().toString(); |
AttributeDescriptor descriptor = schema |
224 |
|
.getDescriptor(amdMapVisibleOnly.get(i).getName()); |
225 |
|
|
226 |
|
// Not so nice in 26: find the index of an attribute... |
227 |
|
int idx = schema.getAttributeDescriptors().indexOf(descriptor); |
228 |
|
attrIdxForCol[i] = idx; |
229 |
|
|
230 |
|
String attName = schema.getAttributeDescriptors().get(idx) |
231 |
|
.getLocalName(); |
232 |
|
colNames[i] = amdMap.get(attName).getTitle().toString(); |
233 |
|
AttributeMetadataInterface amd = amdMap.get(attName); |
234 |
|
colTooltips[i] = "<html>" + amd.getDesc().toString() + "<br>" |
235 |
|
+ amd.getName() + "</html>"; |
236 |
|
colClass[i] = schema.getAttributeDescriptors().get(idx) |
237 |
|
.getType().getBinding(); |
238 |
} |
} |
|
|
|
239 |
} |
} |
240 |
|
|
241 |
|
// store feature indexes in HashMap to optimize findFeature(.) |
242 |
|
featureIdx = new HashMap<String, Integer>(); |
243 |
|
for (int i = 0; i < featureArray.length; i++) |
244 |
|
if (featureArray[i] != null) |
245 |
|
featureIdx.put(featureArray[i].getID(), i); |
246 |
|
// |
247 |
|
// // translate the column names |
248 |
|
// if (amdMap != null) { |
249 |
|
// for (int i = 0; i < colNames.length; i++) { |
250 |
|
// colTooltips[i] = amdMap.get(colNames[i]).getDesc().toString() |
251 |
|
// + "<br>" + colNames[i]; |
252 |
|
// colNames[i] = amdMap.get(colNames[i]).getTitle().toString(); |
253 |
|
// |
254 |
|
// } |
255 |
|
// } |
256 |
if (fireTableStructureChanged) |
if (fireTableStructureChanged) |
257 |
fireTableStructureChanged(); |
fireTableStructureChanged(); |
258 |
|
|
259 |
} |
} |
260 |
|
|
261 |
/** |
/** |