Statistics
| Revision:

svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.compat.cdc / org.gvsig.fmap.dal / org.gvsig.fmap.dal.swing / org.gvsig.fmap.dal.swing.impl / src / main / java / org / gvsig / fmap / dal / swing / impl / searchpanel / SearchFieldController.java @ 44669

History | View | Annotate | Download (21.1 KB)

1
package org.gvsig.fmap.dal.swing.impl.searchpanel;
2

    
3
import java.awt.Cursor;
4
import java.awt.Dimension;
5
import java.awt.event.ActionEvent;
6
import java.awt.event.ActionListener;
7
import java.awt.event.ItemEvent;
8
import java.awt.event.ItemListener;
9
import java.awt.event.MouseAdapter;
10
import java.awt.event.MouseEvent;
11
import java.util.ArrayList;
12
import java.util.HashMap;
13
import java.util.List;
14
import java.util.Map;
15
import java.util.Objects;
16
import javax.swing.ComboBoxModel;
17
import javax.swing.DefaultComboBoxModel;
18
import javax.swing.ImageIcon;
19
import javax.swing.JComboBox;
20
import javax.swing.JLabel;
21
import javax.swing.JScrollPane;
22
import javax.swing.JTree;
23
import javax.swing.SwingUtilities;
24
import javax.swing.tree.TreePath;
25
import org.apache.commons.lang3.ObjectUtils;
26
import org.apache.commons.lang3.StringUtils;
27
import org.gvsig.expressionevaluator.ExpressionBuilder;
28
import org.gvsig.expressionevaluator.ExpressionUtils;
29
import org.gvsig.fmap.dal.DataManager;
30
import org.gvsig.fmap.dal.complements.Search;
31
import org.gvsig.fmap.dal.exception.DataException;
32
import org.gvsig.fmap.dal.feature.Feature;
33
import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;
34
import org.gvsig.fmap.dal.feature.FeatureQuery;
35
import org.gvsig.fmap.dal.feature.FeatureSet;
36
import org.gvsig.fmap.dal.feature.FeatureStore;
37
import org.gvsig.fmap.dal.feature.FeatureType;
38
import org.gvsig.fmap.dal.swing.impl.searchpanel.AdvancedAttributeSelectionTreeModel.Node;
39
import static org.gvsig.fmap.dal.swing.impl.searchpanel.DefaultSearchPanel.getAttributeDescriptorLabel;
40
import org.gvsig.tools.ToolsLocator;
41
import org.gvsig.tools.dataTypes.Coercion;
42
import org.gvsig.tools.dataTypes.CoercionException;
43
import org.gvsig.tools.dataTypes.DataTypesManager;
44
import org.gvsig.tools.exception.BaseException;
45
import org.gvsig.tools.swing.api.DropDown;
46
import org.gvsig.tools.swing.api.ToolsSwingLocator;
47
import org.gvsig.tools.swing.api.ToolsSwingManager;
48
import org.gvsig.tools.swing.api.windowmanager.Dialog;
49
import org.gvsig.tools.swing.api.windowmanager.WindowManager;
50
import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;
51
import org.gvsig.tools.swing.icontheme.IconTheme;
52
import org.gvsig.tools.util.LabeledValue;
53
import org.gvsig.tools.util.LabeledValueImpl;
54
import org.gvsig.tools.visitor.VisitCanceledException;
55
import org.gvsig.tools.visitor.Visitor;
56
import org.slf4j.Logger;
57
import org.slf4j.LoggerFactory;
58

    
59
/**
60
 *
61
 * @author jjdelcerro
62
 */
63
@SuppressWarnings("UseSpecificCatch")
64
public class SearchFieldController {
65

    
66
    private static final Logger LOGGER = LoggerFactory.getLogger(SearchFieldController.class);
67
    
68
    private static class FeatureAttribute extends LabeledValueImpl<String> {
69

    
70
        FeatureAttributeDescriptor attrdesc;
71
        private final FeatureStore store;
72
        private final int type;
73
        private final boolean showStoreName;
74

    
75
        public FeatureAttribute(FeatureStore store, FeatureAttributeDescriptor attrdesc, int type) {
76
            this(store, attrdesc, null, type, false);
77
        }
78

    
79
        public FeatureAttribute(
80
                FeatureStore store,
81
                FeatureAttributeDescriptor attrdesc, 
82
                String value,
83
                int type,
84
                boolean showStoreName
85
            ) {
86
            super(
87
                    getAttributeDescriptorLabel(attrdesc, store.getName()),
88
                    ObjectUtils.defaultIfNull(value, attrdesc.getName())
89
            );
90
            this.store = store;
91
            this.attrdesc = attrdesc;
92
            this.type = type;
93
            this.showStoreName = showStoreName;
94
        }
95

    
96
        @Override
97
        public String getLabel() {
98
            String theLabel = getAttributeDescriptorLabel(attrdesc, showStoreName? store.getName():null);
99
            switch(this.type) {
100
                case Search.OrderedAttribute.TYPE_REGURAL:
101
                    break;
102
                case Search.OrderedAttribute.TYPE_FAVORITE:
103
                    theLabel = "<html><b>"+theLabel+"</b></html>";
104
                    break;
105
                case Search.OrderedAttribute.TYPE_RECENT:
106
                    theLabel = "<html><i><b>"+theLabel+"</b></i></html>";
107
                    break;
108
            }
109
            return theLabel;
110
        }
111
        
112
        public FeatureAttributeDescriptor getDescriptor() {
113
            return this.attrdesc;
114
        }
115

    
116
        public FeatureStore getFeatureStore() {
117
            return this.store;
118
        }
119
        
120
        public boolean isExpression() {
121
            FeatureType type = this.attrdesc.getFeatureType();
122
            if (type == null) {
123
                return false;
124
            }
125
            Object x = type.get(this.getValue());
126
            return x == null;
127
        }
128

    
129
    }
130

    
131
    private FeatureStore store;
132
    private final JLabel lblFields;
133
    private final JLabel lblExtraFields;
134
    private final JLabel lblLogicalOperators;
135
    private final JLabel lblRelationalOperators;
136
    private final JComboBox cboValue;
137
    private Object valueAssigned = null;
138

    
139
    private DropDown ddnFields;
140
    private DropDown ddnLogicalOperators;
141
    private DropDown ddnRelationalOperators;
142

    
143
    private final LabeledValue[] relationalOperators = {
144
        new LabeledValueImpl("Equals to", ExpressionBuilder.OPERATOR_EQ),
145
        new LabeledValueImpl("Like to", ExpressionBuilder.OPERATOR_ILIKE),
146
        new LabeledValueImpl("Not equals to", ExpressionBuilder.OPERATOR_NE),
147
        new LabeledValueImpl("Greater than", ExpressionBuilder.OPERATOR_GT),
148
        new LabeledValueImpl("Greater or equal to", ExpressionBuilder.OPERATOR_GE),
149
        new LabeledValueImpl("Less than", ExpressionBuilder.OPERATOR_LT),
150
        new LabeledValueImpl("Less or equal to", ExpressionBuilder.OPERATOR_LE)
151
    };
152
    private final LabeledValue[] logicalOperators = {
153
        new LabeledValueImpl("Or", ExpressionBuilder.OPERATOR_OR),
154
        new LabeledValueImpl("And", ExpressionBuilder.OPERATOR_AND)
155
    };
156

    
157
    public SearchFieldController(
158
            FeatureStore store,
159
            JLabel lblFields,
160
            JLabel lblExtraFields,
161
            JLabel lblRelationalOperators,
162
            JComboBox cboValue,
163
            JLabel lblLogicalOperators
164
    ) {
165
        this.store = store;
166
        this.lblFields = lblFields;
167
        this.lblExtraFields = lblExtraFields;
168
        this.lblRelationalOperators = lblRelationalOperators;
169
        this.cboValue = cboValue;
170
        this.lblLogicalOperators = lblLogicalOperators;
171
        this.initComponents();
172
    }
173

    
174
    public boolean isAValidRelationOperator(String name) {
175
        for (LabeledValue relationalOperator : relationalOperators) {
176
            if( StringUtils.equalsIgnoreCase(name, (CharSequence) relationalOperator.getValue())) {
177
                return true;
178
            }
179
        }
180
        return false;
181
    }
182
    
183
    private void initComponents() {
184
        try {
185
            ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();
186
            this.lblExtraFields.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
187

    
188
            this.ddnFields = toolsSwingManager.createDropDown(lblFields);
189
            this.ddnFields.setVisibleDropdownArrow(false);
190
            this.ddnRelationalOperators = toolsSwingManager.createDropDown(lblRelationalOperators);
191
            this.ddnRelationalOperators.setVisibleDropdownArrow(false);
192
            if (lblLogicalOperators != null) {
193
                this.ddnLogicalOperators = toolsSwingManager.createDropDown(lblLogicalOperators);
194
                this.ddnLogicalOperators.setVisibleDropdownArrow(false);
195
            }
196

    
197
            DefaultComboBoxModel modelRelationalOperators = new DefaultComboBoxModel();
198
            for (LabeledValue op : relationalOperators) {
199
                modelRelationalOperators.addElement(op);
200
            }
201
            this.ddnRelationalOperators.setModel(modelRelationalOperators);
202

    
203
            if (this.ddnLogicalOperators != null) {
204
                DefaultComboBoxModel modelLogicalOperators = new DefaultComboBoxModel();
205
                for (LabeledValue op : logicalOperators) {
206
                    modelLogicalOperators.addElement(op);
207
                }
208
                this.ddnLogicalOperators.setModel(modelLogicalOperators);
209
            }
210
            FeatureType featureType = store.getDefaultFeatureType();
211
            Search search = (Search) ToolsLocator.getComplementsManager().get(
212
                    Search.COMPLEMENT_MANE, featureType
213
            );
214
            List<Search.OrderedAttribute> orderedAttributes = search.getOrderedAttributes(
215
                    Search.BASIC_TYPES_FILTER,
216
                    Search.STR_INT_LONG_LABEL_ORDER,
217
                    20
218
            );
219
            List<ImageIcon>icons = new ArrayList<>();
220
//            DataTypesManager dataTypeManager = ToolsLocator.getDataTypesManager();
221
            IconTheme iconTheme = ToolsSwingLocator.getIconThemeManager().getCurrent();
222
            DefaultComboBoxModel model = new DefaultComboBoxModel();
223
            for (Search.OrderedAttribute attr : orderedAttributes) {
224
                FeatureAttributeDescriptor attrdesc = attr.getDescriptor();
225
                model.addElement(new FeatureAttribute(this.store, attrdesc, attr.getType()));
226
                String iconName = attrdesc.getDataType().getIconName();
227
                if( iconTheme.exists(iconName) ) {
228
                    icons.add(iconTheme.get(iconName));
229
                } else {
230
                    icons.add(null);
231
                }
232
            }
233
            this.ddnFields.setIcons(icons);
234
            this.ddnFields.setModel(model);
235
            this.ddnFields.addItemListener(new ItemListener() {
236
                @Override
237
                public void itemStateChanged(ItemEvent e) {
238
                    if (e.getStateChange() == ItemEvent.SELECTED) {
239
                        doUpdateValuesList();
240
                    }
241

    
242
                }
243
            });
244

    
245
            this.lblExtraFields.addMouseListener(new MouseAdapter() {
246
                @Override
247
                public void mouseClicked(MouseEvent e) {
248
                    doSelectExtraField();
249
                }
250
            });
251
            clear();
252
        } catch (Exception ex) {
253
            throw new RuntimeException(ex);
254
        }
255
    }
256

    
257
    private FeatureType getFeatureType() {
258
        try {
259
            return this.store.getDefaultFeatureType();
260
        } catch (DataException ex) {
261
            return null;
262
        }
263
    }
264

    
265
    private void doSelectExtraField() {
266
        FeatureType featureType = this.getFeatureType();
267
        AdvancedAttributeSelectionTreeModel treeModel = new AdvancedAttributeSelectionTreeModel(
268
                this.store,
269
                Search.BASIC_TYPES_FILTER
270
        );
271
        final JTree tree = new JTree();
272
        tree.setCellRenderer(new AdvancedAttributeSelectionTreeCellRenderer());
273
        tree.setModel(treeModel);
274
        try {
275
            tree.setSelectionRow(1);
276
        } catch(Throwable th) {
277
        }
278
        JScrollPane scrollpane = new JScrollPane(tree);
279
        scrollpane.setPreferredSize(new Dimension(400, 300));
280
        scrollpane.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);
281
        scrollpane.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
282

    
283
        WindowManager_v2 winManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();
284
        final Dialog dialog = winManager.createDialog(
285
                scrollpane,
286
                "Select attribute",
287
                null,
288
                WindowManager_v2.BUTTONS_OK_CANCEL
289
        );
290
        dialog.addActionListener(new ActionListener() {
291
            @Override
292
            public void actionPerformed(ActionEvent e) {
293
                if (dialog.getAction() == WindowManager_v2.BUTTONS_OK) {
294
                    TreePath path = tree.getSelectionPath();
295
                    doAddAndSelect(path.getPath());
296
                }
297
            }
298
        });
299
        dialog.show(WindowManager.MODE.DIALOG);
300

    
301
    }
302

    
303
    private void doAddAndSelect(Object[] nodes) {
304
        ExpressionBuilder builder = ExpressionUtils.createExpressionBuilder();
305
        List<String> list = new ArrayList<>();
306
        for (int i = 1; i < nodes.length; i++) {
307
            Node node = (Node) nodes[i];
308
            FeatureAttributeDescriptor attrdesc = node.getValue();
309
            list.add(attrdesc.getName());
310
        }
311
        Node node = (Node) nodes[nodes.length - 1];
312
        FeatureStore theStore = node.getFeatureStore();
313
        FeatureAttributeDescriptor attrdesc = node.getValue();
314
        String storeFullName = theStore.getFullName();
315
        DefaultComboBoxModel<FeatureAttribute> model = (DefaultComboBoxModel) this.ddnFields.getModel();
316
        for (int i = 0; i < model.getSize(); i++) {
317
            FeatureAttribute attr = model.getElementAt(i);
318
            FeatureAttributeDescriptor attrdescN = attr.getDescriptor();
319
            if (StringUtils.equalsIgnoreCase(storeFullName, attrdescN.getStore().getFullName())
320
                    && StringUtils.equalsIgnoreCase(attrdesc.getName(), attrdescN.getName())) {
321
                this.setAttribute(i);
322
                return;
323
            }
324
        }
325
        String formula;
326
        boolean showStoreName;
327
        if (StringUtils.equalsIgnoreCase(storeFullName, this.store.getFullName())) {
328
            formula = attrdesc.getName();
329
            showStoreName = false;
330
        } else {
331
            formula = builder.function(
332
                    DataManager.FUNCTION_FOREING_VALUE, 
333
                    builder.constant(StringUtils.join(list,"."))
334
            ).toString();
335
            showStoreName = true;
336
        }
337
        FeatureAttribute attribute = new FeatureAttribute(theStore, attrdesc, formula, Search.OrderedAttribute.TYPE_REGURAL, showStoreName);
338
        model.addElement(attribute);
339
        IconTheme iconTheme = ToolsSwingLocator.getIconThemeManager().getCurrent();
340
        this.ddnFields.getIcons().add(iconTheme.get(attrdesc.getDataType().getIconName()));
341
        this.setAttribute(model.getSize() - 1);
342
    }
343

    
344
    public void clear() {
345
        this.ddnRelationalOperators.setSelectedIndex(0);
346
        if (this.ddnLogicalOperators != null) {
347
            this.ddnLogicalOperators.setSelectedIndex(0);
348
        }
349
        this.cboValue.setSelectedIndex(-1);
350
    }
351

    
352
    private void doUpdateValuesList() {
353
        final FeatureAttribute attribute = (FeatureAttribute) this.ddnFields.getSelectedItem();
354
        if (attribute == null) {
355
            return;
356
        }
357

    
358
        final List<Object> values = new ArrayList<>();
359
        final int limit = 60;
360
        final long timeLimit = System.currentTimeMillis() + limit * 1000;
361
        final DefaultComboBoxModel model = new DefaultComboBoxModel();
362
        this.setEnabled(false);
363
        final FeatureStore theStore = attribute.getFeatureStore();
364
        final FeatureQuery query = theStore.createFeatureQuery();
365
        query.addAttributeName(attribute.getDescriptor().getName());
366
        query.setGroup(true);
367
        query.setLimit(1000);
368
        Thread th = new Thread(new Runnable() {
369
            @Override
370
            public void run() {
371
                try {
372
                    FeatureSet set = theStore.getFeatureSet(query);
373
                    set.accept(new Visitor() {
374
                        @Override
375
                        public void visit(Object o) throws VisitCanceledException, BaseException {
376
                            Object value = ((Feature) o).get(attribute.getDescriptor().getName());
377
                            if (!values.contains(value)) {
378
                                values.add(value);
379
                            }
380
                            if (System.currentTimeMillis() > timeLimit) {
381
                                throw new VisitCanceledException();
382
                            }
383
                            if (values.size() > 1000) {
384
                                throw new VisitCanceledException();
385
                            }
386
                        }
387
                    });
388
                } catch (VisitCanceledException ex) {
389

    
390
                } catch (Exception ex) {
391
                    LOGGER.warn("Can't update list of values of '"+attribute.getLabel()+"'.", ex);
392
                }
393
                List<LabeledValue> elements = new ArrayList<>();
394
                if (!values.isEmpty()) {
395
                    LabeledValue[] availableValues = attribute.getDescriptor().getAvailableValues();
396
                    Map<String, String> availableValuesMap = new HashMap<>();
397
                    if (availableValues != null) {
398
                        for (LabeledValue availableValue : availableValues) {
399
                            availableValuesMap.put(
400
                                    Objects.toString(availableValue.getValue()),
401
                                    availableValue.getLabel()
402
                            );
403
                        }
404
                    }
405
                    elements.add(new LabeledValueImpl("", null));
406
                    for (Object value : values) {
407
                        String key = Objects.toString(value);
408
                        String label = availableValuesMap.getOrDefault(key, key);
409
                        elements.add(new LabeledValueImpl(label, value));
410
                    }
411
                    elements.sort(null);
412
                    
413
                }
414
                for (LabeledValue element : elements) {
415
                    model.addElement(element);
416
                }
417
                SwingUtilities.invokeLater(new Runnable() {
418
                    @Override
419
                    public void run() {
420
                        cboValue.setModel(model);
421
                        if( valueAssigned!=null ) {
422
                            cboValue.setSelectedItem(valueAssigned);
423
                            valueAssigned = null;
424
                        }
425
                        setEnabled(true);
426
                    }
427
                });
428
            }
429
        });
430
        th.start();
431
    }
432

    
433
    public void setEnabled(boolean enabled) {
434
        this.ddnFields.setEnabled(enabled);
435
        if( this.ddnLogicalOperators!=null ) {
436
            this.ddnLogicalOperators.setEnabled(enabled);
437
        }
438
        this.ddnRelationalOperators.setEnabled(enabled);
439
        this.lblExtraFields.setEnabled(enabled);
440
    }
441
    
442
    public String getRelationalOperator() {
443
        LabeledValue<String> op = (LabeledValue) this.ddnRelationalOperators.getSelectedItem();
444
        if (op == null) {
445
            return null;
446
        }
447
        return op.getValue();
448
    }
449

    
450
    public int setRelationalOperator(String name) {
451
        int n = 0;
452
        for (LabeledValue relationalOperator : relationalOperators) {
453
            if( StringUtils.equalsIgnoreCase(name, (CharSequence) relationalOperator.getValue())) {
454
                break;
455
            }
456
            n++;
457
        }
458
        if( this.relationalOperators.length<=n ) {
459
            return -1;
460
        }
461
        this.ddnRelationalOperators.setSelectedIndex(n);
462
        return n;
463
    }
464
    
465
    public String getLogicalOperator() {
466
        if (this.ddnLogicalOperators == null) {
467
            return null;
468
        }
469
        LabeledValue<String> rel = (LabeledValue) this.ddnLogicalOperators.getSelectedItem();
470
        if (rel == null) {
471
            return null;
472
        }
473
        return rel.getValue();
474
    }
475

    
476
    public Object getValue() {
477
        final FeatureAttribute attribute = (FeatureAttribute) this.ddnFields.getSelectedItem();
478
        if (attribute == null) {
479
            return null;
480
        }
481
        Object v = this.cboValue.getSelectedItem();
482
        if (v == null) {
483
            return null;
484
        }
485
        if (v instanceof LabeledValue) {
486
            v = ((LabeledValue) v).getValue();
487
            if (v == null) {
488
                return null;
489
            }
490
        }
491
        if (v instanceof CharSequence) {
492
            if (StringUtils.isBlank((CharSequence) v)) {
493
                return null;
494
            }
495
        }
496
        Coercion coercion = attribute.getDescriptor().getDataType().getCoercion();
497
        try {
498
            return coercion.coerce(v);
499
        } catch (CoercionException ex) {
500
            return null;
501
        }
502
    }
503
    
504
    public void setValue(Object value) {
505
        this.cboValue.setSelectedItem(value);
506
        this.valueAssigned = value;
507
    }
508
    
509
    public boolean isAttributeAnExpression() {
510
        final FeatureAttribute attribute = (FeatureAttribute) this.ddnFields.getSelectedItem();
511
        if (attribute == null) {
512
            return false;
513
        }
514
        return attribute.isExpression();
515
    }
516

    
517
    public String getAttribute() {
518
        final FeatureAttribute attribute = (FeatureAttribute) this.ddnFields.getSelectedItem();
519
        if (attribute == null) {
520
            return null;
521
        }
522
        if( this.getValue()!=null ) {
523
            attribute.getDescriptor().recentUsed();
524
        }
525
        return attribute.getValue();
526
    }
527

    
528
    public int setAttribute(String name) {
529
        ComboBoxModel<FeatureAttribute> model = this.ddnFields.getModel();
530
        for (int i = 0; i < model.getSize(); i++) {
531
            FeatureAttribute x = model.getElementAt(i);
532
            if (StringUtils.equalsIgnoreCase(name, x.getValue())) {
533
                this.setAttribute(i);
534
                return i;
535
            }
536
        }
537
        this.setAttribute(-1);
538
        return -1;
539
    }
540

    
541
    public void setAttribute(int index) {
542
        try {
543
            this.ddnFields.setSelectedIndex(index);
544
        } catch (Exception ex) {
545
            this.ddnFields.setSelectedIndex(-1);
546
        }
547
        doUpdateValuesList();
548
    }
549

    
550
}