Application: gvSIG desktop

package org.gvsig.app.extension;

import java.awt.BorderLayout;

import java.awt.FlowLayout;

import java.awt.event.ActionEvent;

import javax.swing.AbstractAction;

import javax.swing.JButton;

import javax.swing.JPanel;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.expressionevaluator.swing.Element;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingLocator;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.expressionevaluator.swing.JExpressionBuilder;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureSelection;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.swing.DataSwingManager;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class SelectByAttributes {

    private static final Logger LOGGER = LoggerFactory.getLogger(SelectByAttributes.class);

    public SelectByAttributes() {

    }

    public void showWindow(String title, final FeatureStore store, WindowManager.MODE mode) {

        I18nManager i18n = ToolsLocator.getI18nManager();

        WindowManager windowManager = ToolsSwingLocator.getWindowManager();

        ExpressionEvaluatorSwingManager swingManager = ExpressionEvaluatorSwingLocator.getManager();

        final JPanel panel = new JPanel();

        panel.setLayout(new BorderLayout());

        final JExpressionBuilder builder = swingManager.createJExpressionBuilder();

        builder.addSymbolTable(DataManager.FEATURE_SYMBOL_TABLE);

        Element element = swingManager.createElement("View");

        if (element != null) {

            builder.getElements().add(element);

        }

        element = swingManager.createElement(

                DataSwingManager.FEATURE_STORE_EXPRESSION_ELEMENT,

                builder,

                store

        );

        if (element != null) {

            builder.getElements().add(element);

        }

        panel.add(builder.asJComponent(), BorderLayout.CENTER);

        JPanel buttons = new JPanel();

        buttons.setLayout(new FlowLayout(FlowLayout.RIGHT, 4, 4));

        buttons.add(new JButton(new AbstractAction(i18n.getTranslation("_Set_selection")) {

            @Override

            public void actionPerformed(ActionEvent e) {

                setSelection(store, builder.getExpression());

            }

        }));

        buttons.add(new JButton(new AbstractAction(i18n.getTranslation("_Add_to_selection")) {

            @Override

            public void actionPerformed(ActionEvent e) {

                addToSelection(store, builder.getExpression());

            }

        }));

        buttons.add(new JButton(new AbstractAction(i18n.getTranslation("_Filter_selection")) {

            @Override

            public void actionPerformed(ActionEvent e) {

                filterSelection(store, builder.getExpression());

            }

        }));

        buttons.add(new JButton(new AbstractAction(i18n.getTranslation("_Close")) {

            @Override

            public void actionPerformed(ActionEvent e) {

                panel.setVisible(false);

            }

        }));

        panel.add(buttons, BorderLayout.SOUTH);

        windowManager.showWindow(

                panel,

                title,

                mode

        );

    }

    public void setSelection(FeatureStore store, Expression expression) {

        try {

            DataManager manager = DALLocator.getDataManager();

            FeatureSelection currentSelection = store.getFeatureSelection();

            FeatureQuery query = store.createFeatureQuery();

            query.setFilter(manager.createExpresion(expression));

            FeatureSet selection = store.getFeatureSet(query);

            currentSelection.deselectAll();

            currentSelection.select(selection);

        } catch (Exception ex) {

            LOGGER.warn("Can't build selecction from filter expression.", ex);

        }

    }

    public void addToSelection(FeatureStore store, Expression expression) {

        try {

            DataManager manager = DALLocator.getDataManager();

            FeatureSelection currentSelection = store.getFeatureSelection();

            FeatureQuery query = store.createFeatureQuery();

            query.setFilter(manager.createExpresion(expression));

            FeatureSet selection = store.getFeatureSet(query);

            currentSelection.select(selection);

        } catch (Exception ex) {

            LOGGER.warn("Can't build selecction from filter expression.", ex);

        }

    }

    public void filterSelection(FeatureStore store, Expression expression) {

        try {

            DataManager manager = DALLocator.getDataManager();

            FeatureSelection currentSelection = store.getFeatureSelection();

            FeatureQuery query = store.createFeatureQuery();

            query.setFilter(manager.createExpresion(expression));

            FeatureSet set = store.getFeatureSet(query);

            FeatureSelection newSelection = store.createFeatureSelection();

            for (Feature feature : set) {

                if( currentSelection.isSelected(feature) ) {

                    newSelection.select(feature);

                }

            }

            store.setSelection(newSelection);

        } catch (Exception ex) {

            LOGGER.warn("Can't build selecction from filter expression.", ex);

        }

    }

}