Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.actions;

import java.awt.event.ActionEvent;

import javax.swing.AbstractAction;

import javax.swing.Action;

import javax.swing.ImageIcon;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.fmap.dal.DataStore;

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.AbstractDALActionFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.swing.DALActionFactory.DALActionContext;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class SelectionFilterAction 

        extends AbstractAction 

    {

    public static class SelectionFilterActionFactory  extends AbstractDALActionFactory {

        public static final String ACTION_NAME = "SelectionFilter";

        public SelectionFilterActionFactory() {

            super(ACTION_NAME);

        }

        @Override

        public Action createAction(DALActionContext context) {

            return new SelectionFilterAction(context);

        }

    }

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

    private final DALActionContext context;

    public SelectionFilterAction(DALActionContext context) {

        this.context = context;

        ImageIcon icon = ToolsSwingLocator.getIconThemeManager().getCurrent().get("search-action-select-filter");

        if( icon == null ) {

            LOGGER.warn("Can't locate icon 'search-action-select-filter'.");

        }

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.putValue(

                Action.ACTION_COMMAND_KEY, 

                SelectionFilterActionFactory.ACTION_NAME

        );

        this.putValue(

                Action.SHORT_DESCRIPTION, 

                i18n.getTranslation("_Filter_selection")

        );

        this.putValue(

                Action.SMALL_ICON, 

                icon

        );

    }

    @Override

    public void actionPerformed(ActionEvent e) {

        try {

            DataStore store = this.context.getStore();

            if( !(store instanceof FeatureStore) ) {

                return ;

            }

            FeatureStore featureStore = (FeatureStore) store;

            FeatureSelection currentSelection = featureStore.getFeatureSelection();

            Expression filter = this.context.getFilterForSelecteds();

            if( filter == null ) {

              FeatureQuery query = this.context.getQuery();

              if( query != null ) {

                filter = query.getExpressionFilter();

              }

            }

            FeatureQuery query = featureStore.createFeatureQuery();

            query.setFilter(filter);

            FeatureSet set = featureStore.getFeatureSet(query);

            FeatureSelection newSelection = featureStore.createFeatureSelection();

            for (Feature feature : set) {

                if( currentSelection.isSelected(feature) ) {

                    newSelection.select(feature);

                }

            }

            store.setSelection(newSelection);

        } catch (Exception ex) {

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

        }

    }

}