Statistics
| Revision:

svn-gvsig-desktop / branches / v2_0_0_prep / applications / appgvSIG / src / com / iver / cit / gvsig / FiltroExtension.java @ 22932

History | View | Annotate | Download (11.5 KB)

1
/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana
2
 *
3
 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.
4
 *
5
 * This program is free software; you can redistribute it and/or
6
 * modify it under the terms of the GNU General Public License
7
 * as published by the Free Software Foundation; either version 2
8
 * of the License, or (at your option) any later version.
9
 *
10
 * This program is distributed in the hope that it will be useful,
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 * GNU General Public License for more details.
14
 *
15
 * You should have received a copy of the GNU General Public License
16
 * along with this program; if not, write to the Free Software
17
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.
18
 *
19
 * For more information, contact:
20
 *
21
 *  Generalitat Valenciana
22
 *   Conselleria d'Infraestructures i Transport
23
 *   Av. Blasco Ib??ez, 50
24
 *   46010 VALENCIA
25
 *   SPAIN
26
 *
27
 *      +34 963862235
28
 *   gvsig@gva.es
29
 *      www.gvsig.gva.es
30
 *
31
 *    or
32
 *
33
 *   IVER T.I. S.A
34
 *   Salamanca 50
35
 *   46005 Valencia
36
 *   Spain
37
 *
38
 *   +34 963163400
39
 *   dac@iver.es
40
 */
41
package com.iver.cit.gvsig;
42

    
43
import java.awt.Component;
44
import java.util.Iterator;
45

    
46
import javax.swing.JOptionPane;
47

    
48
import org.gvsig.fmap.data.DataException;
49
import org.gvsig.fmap.data.ReadException;
50
import org.gvsig.fmap.data.feature.Feature;
51
import org.gvsig.fmap.data.feature.FeatureCollection;
52
import org.gvsig.fmap.data.feature.FeatureStore;
53
import org.gvsig.fmap.data.feature.MemoryFeatureCollection;
54
import org.gvsig.fmap.mapcontext.layers.FLayer;
55
import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;
56

    
57
import com.iver.andami.PluginServices;
58
import com.iver.andami.messages.NotificationManager;
59
import com.iver.andami.plugins.Extension;
60
import com.iver.andami.ui.mdiManager.IWindow;
61
import com.iver.cit.gvsig.gui.filter.ExpressionListener;
62
import com.iver.cit.gvsig.gui.filter.FilterDialog;
63
import com.iver.cit.gvsig.project.documents.ProjectDocument;
64
import com.iver.cit.gvsig.project.documents.table.gui.Table;
65
import com.iver.cit.gvsig.project.documents.view.IProjectView;
66
import com.iver.utiles.exceptionHandling.ExceptionListener;
67

    
68

    
69
/**
70
 * Extensi?n que abre un di?logo para poder hacer un filtro de una capa o tabla.
71
 *
72
 * @author Vicente Caballero Navarro
73
 */
74
public class FiltroExtension extends Extension implements ExpressionListener {
75
        protected FeatureStore featureStore = null;
76
        protected Table table;
77
        private String filterTitle;
78

    
79
        /**
80
         * DOCUMENT ME!
81
         */
82
        public void initialize() {
83
                registerIcons();
84
        }
85

    
86
        private void registerIcons(){
87
                PluginServices.getIconTheme().registerDefault(
88
                                "table-filter",
89
                                this.getClass().getClassLoader().getResource("images/Filtro.png")
90
                        );
91
        }
92

    
93
        /**
94
         * DOCUMENT ME!
95
         *
96
         * @param actionCommand DOCUMENT ME!
97
         */
98
        public void execute(String actionCommand) {
99
                if ("FILTRO".equals(actionCommand)) {
100
                        try {
101
                                IWindow v = PluginServices.getMDIManager().getActiveWindow();
102

    
103
                                if (v instanceof Table) {
104
                                        table = (Table) v;
105

    
106
                                        featureStore = table.getModel().getModel();
107
                                        filterTitle = table.getModel().getName();
108
                                        table.getModel().setModified(true);
109
                                } else if (v instanceof com.iver.cit.gvsig.project.documents.view.gui.View) {
110
                                        IProjectView pv = ((com.iver.cit.gvsig.project.documents.view.gui.View) v).getModel();
111
                                        filterTitle = ((com.iver.cit.gvsig.project.documents.view.gui.View) v).getModel().getName();
112
                                        FLayer layer = pv.getMapContext()
113
                                        .getLayers().getActives()[0];
114
                                        featureStore = ((FLyrVect)layer).getFeatureStore();//pv.getProject().getDataSourceByLayer(layer);
115
                                        ((ProjectDocument)pv).setModified(true);
116
                                }
117
                        }  catch (ReadException e) {
118
                                NotificationManager.addError("Error filtrando", e);
119
                        }
120

    
121
                        doExecute();
122
                }
123
        }
124

    
125
        /**
126
         * "execute" method action.
127
         *
128
         */
129
        protected void doExecute(){
130
//                DefaultExpressionDataSource ds = new DefaultExpressionDataSource();
131
//                ds.setTable(featureStore);
132
                FilterDialog dlg = new FilterDialog(filterTitle);
133
                dlg.addExpressionListener(this);
134
                dlg.addExceptionListener(new ExceptionListener() {
135
                        public void exceptionThrown(Throwable t) {
136
                                NotificationManager.addError(t.getMessage(), t);
137
                        }
138
                });
139
                dlg.setModel(featureStore);
140
                PluginServices.getMDIManager().addWindow(dlg);
141
        }
142

    
143
        /**
144
         * DOCUMENT ME!
145
         *
146
         * @return DOCUMENT ME!
147
         */
148
        public boolean isEnabled() {
149
                IWindow v = PluginServices.getMDIManager().getActiveWindow();
150

    
151
                if (v == null) {
152
                        return false;
153
                }
154

    
155
                if (v instanceof Table) {
156
                        return true;
157
                } else {
158
                        if (v instanceof com.iver.cit.gvsig.project.documents.view.gui.View) {
159
                                com.iver.cit.gvsig.project.documents.view.gui.View view = (com.iver.cit.gvsig.project.documents.view.gui.View) v;
160
                                IProjectView pv = view.getModel();
161
                                FLayer[] seleccionadas = pv.getMapContext().getLayers()
162
                                .getActives();
163

    
164
                                if (seleccionadas.length == 1) {
165
                                        if (seleccionadas[0].isAvailable() && seleccionadas[0] instanceof FLyrVect) {
166
                                                return true;
167
                                        }
168
                                }
169
                        }
170

    
171
                        return false;
172
                }
173

    
174
        }
175

    
176
        /**
177
         * DOCUMENT ME!
178
         *
179
         * @return DOCUMENT ME!
180
         */
181
        public boolean isVisible() {
182
                IWindow v = PluginServices.getMDIManager().getActiveWindow();
183

    
184
                if (v == null) {
185
                        return false;
186
                }
187

    
188
                if (v instanceof Table) {
189
                        return true;
190
                } else {
191
                        if (v instanceof com.iver.cit.gvsig.project.documents.view.gui.View) {
192
                                com.iver.cit.gvsig.project.documents.view.gui.View view = (com.iver.cit.gvsig.project.documents.view.gui.View) v;
193
                                IProjectView pv = view.getModel();
194
                                FLayer[] seleccionadas = pv.getMapContext().getLayers()
195
                                .getActives();
196

    
197
                                if (seleccionadas.length == 1) {
198
                                        if (seleccionadas[0] instanceof FLyrVect) {
199
                                                return true;
200
                                        }
201
                                }
202
                        }
203

    
204
                        return false;
205
                }
206
        }
207

    
208
        /**
209
         * DOCUMENT ME!
210
         *
211
         * @param expression DOCUMENT ME!
212
         */
213
        public void newSet(String expression) throws DataException {
214
                // By Pablo: if no filter expression -> no element selected
215
                if (! this.filterExpressionFromWhereIsEmpty(expression)) {
216
                        try {
217
                                FeatureCollection sel = doSet(expression);
218

    
219
                                if (sel == null) {
220
                                        //throw new RuntimeException("Not a 'where' clause?");
221
                                        return;
222
                                }
223
                                featureStore.setSelection(sel);
224

    
225
//                                FBitSet selection = new FBitSet();
226
//
227
//                                for (int i = 0; i < sel.length; i++) {
228
//                                        selection.set((int) sel[i]);
229
//                                }
230
//
231
//                                featureStore.getSelection().clear();
232
//                                featureStore.setSelection(selection);
233
                        }catch(Exception e){
234
                                JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(), "Asegurate de que la consulta es correcta.");
235
                        }
236
                }
237
                else {
238
                        // By Pablo: if no expression -> no element selected
239
                        featureStore.getSelection().clear();
240
                }
241
        }
242

    
243
        /**
244
         * @throws ReadException
245
         * @see com.iver.cit.gvsig.gui.filter.ExpressionListener#newSet(java.lang.String)
246
         */
247
        private FeatureCollection doSet(String expression) throws ReadException {
248
                return (FeatureCollection)featureStore.getDataCollection(featureStore.getDefaultFeatureType(),expression,null);
249
//                try {
250
//                        DataSource ds = LayerFactory.getDataSourceFactory().executeSQL(expression,
251
//                                        DataSourceFactory.MANUAL_OPENING);
252
//
253
//                        return ds.getWhereFilter();
254
//                } catch (DriverLoadException e) {
255
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"driver_error")+"\n"+e.getMessage());
256
//                } catch (ReadDriverException e) {
257
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"driver_error")+"\n"+e.getMessage());
258
//                } catch (ParseException e) {
259
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"parse_expresion_error")+"\n"+e.getMessage());
260
//                } catch (SemanticException e) {
261
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"semantic_expresion_error")+"\n"+e.getMessage());
262
//                } catch (IOException e) {
263
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"input_output_error")+"\n"+e.getMessage());
264
//                } catch (EvaluationException e) {
265
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"parse_expresion_error")+"\n"+e.getMessage());
266
//                } catch (com.hardcode.gdbms.parser.TokenMgrError e) {
267
//                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"expresion_error")+"\n"+e.getMessage());
268
//                }
269
//                return null;
270
        }
271

    
272
        /**
273
         * DOCUMENT ME!
274
         *
275
         * @param expression
276
         *            DOCUMENT ME!
277
         * @throws DataException
278
         */
279
        public void addToSet(String expression) throws DataException {
280
                // By Pablo: if no filter expression -> don't add more elements to set
281
                if (! this.filterExpressionFromWhereIsEmpty(expression)) {
282
                        FeatureCollection sel=null;
283
                        try {
284
                                sel = doSet(expression);
285
                        } catch (ReadException e) {
286
                                // TODO Auto-generated catch block
287
                                e.printStackTrace();
288
                        }
289

    
290
                        if (sel == null) {
291
                                //throw new RuntimeException("Not a 'where' clause?");
292
                                return;
293
                        }
294
                        FeatureCollection oldSelection=(FeatureCollection)featureStore.getSelection();
295
                        oldSelection.addAll(sel);
296
                        featureStore.setSelection(oldSelection);
297

    
298
//                        FBitSet selection = new FBitSet();
299
//
300
//                        for (int i = 0; i < sel.length; i++) {
301
//                                selection.set((int) sel[i]);
302
//                        }
303
//
304
//                        FBitSet fbs = featureStore.getSelection();
305
//                        fbs.or(selection);
306
//                        featureStore.setSelection(fbs);
307
                }
308
        }
309

    
310
        /**
311
         * DOCUMENT ME!
312
         *
313
         * @param expression DOCUMENT ME!
314
         */
315
        public void fromSet(String expression) throws DataException {
316
                // By Pablo: if no filter expression -> no element selected
317
                try{
318
                        if (! this.filterExpressionFromWhereIsEmpty(expression)) {
319

    
320
                                FeatureCollection sel=null;
321
                                sel = doSet(expression);
322

    
323
                                if (sel == null) {
324
                                        throw new RuntimeException("Not a 'where' clause?");
325
                                }
326
                                FeatureCollection oldSelection=(FeatureCollection)featureStore.getSelection();
327

    
328
                                MemoryFeatureCollection mfc=new MemoryFeatureCollection(featureStore);
329
                                Iterator iterator=sel.iterator();
330
                                while (iterator.hasNext()) {
331
                                        Feature feature = (Feature) iterator.next();
332
                                        if (oldSelection.contains(feature)){
333
                                                mfc.add(feature);
334
                                        }
335
                                }
336
                                featureStore.setSelection(mfc);
337

    
338

    
339

    
340

    
341

    
342
                                        //                        FBitSet selection = new FBitSet();
343
                                        //
344
                                        //                        for (int i = 0; i < sel.length; i++) {
345
                                        //                                selection.set((int) sel[i]);
346
                                        //                        }
347
                                        //
348
                                        //                        FBitSet fbs = featureStore.getSelection();
349
                                        //                        fbs.and(selection);
350
                                        //                        featureStore.setSelection(fbs);
351
                        } else {
352
                                        // By Pablo: if no expression -> no element selected
353
                                        featureStore.setSelection(featureStore.createSelection());
354
                                }
355
                        } catch (DataException e) {
356
                                NotificationManager.addError(e);
357
                        }
358

    
359
                }
360

    
361
        /**
362
         * Returns true if the WHERE subconsultation of the filterExpression is empty ("")
363
         *
364
         * @author Pablo Piqueras Bartolom? (p_queras@hotmail.com)
365
         * @param expression An string
366
         * @return A boolean value
367
         */
368
        private boolean filterExpressionFromWhereIsEmpty(String expression) {
369
                String subExpression = expression.trim();
370
                int pos;
371

    
372
                // Remove last ';' if exists
373
                if (subExpression.charAt(subExpression.length() -1) == ';') {
374
                        subExpression = subExpression.substring(0, subExpression.length() -1).trim();
375
                }
376

    
377
                // If there is no 'where' clause
378
                if ((pos = subExpression.indexOf("where")) == -1) {
379
                        return false;
380
                }
381

    
382
                // If there is no subexpression in the WHERE clause -> true
383
                subExpression = subExpression.substring(pos + 5, subExpression.length()).trim(); // + 5 is the length of 'where'
384
                if ( subExpression.length() == 0 ) {
385
                        return true;
386
                } else {
387
                        return false;
388
                }
389
        }
390
}