Application: gvSIG desktop

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

import java.awt.event.ActionEvent;

import java.util.Objects;

import javax.swing.AbstractAction;

import javax.swing.Action;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.mutable.MutableBoolean;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.fmap.dal.DataStore;

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

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.fmap.dal.swing.DALActionFactory.DALActionContext;

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.i18n.I18nManager;

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

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class SelectionSetAction 

        extends AbstractAction 

    {

    public static class SelectionSetActionFactory extends AbstractDALActionFactory {

        public static final String ACTION_NAME = "SelectionSet";

        public SelectionSetActionFactory() {

            super(ACTION_NAME);

        }

        @Override

        public Action createAction(DALActionContext context) {

            return new SelectionSetAction(context);

        }

        public static void selfRegister() {

            DataSwingManager dalSwingManager = DALSwingLocator.getSwingManager();

            dalSwingManager.registerStoreAction(new SelectionSetActionFactory());

        }        

    }

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

    private final DALActionContext context;

    @SuppressWarnings("OverridableMethodCallInConstructor")

    public SelectionSetAction(DALActionContext context) {

        this.context = context;

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.putValue(

                Action.ACTION_COMMAND_KEY, 

                SelectionSetActionFactory.ACTION_NAME

        );

        this.putValue(

                Action.SHORT_DESCRIPTION, 

                i18n.getTranslation("_New_selection")

        );

        this.putValue(

                Action.SMALL_ICON, 

                ToolsSwingLocator.getIconThemeManager().getCurrent().get("storeaction-select")

        );

    }

    @Override

    public void actionPerformed(ActionEvent e) {

        FeatureSet selection = null;

        try {

            DataStore store = this.context.getStore();

            if (!(store instanceof FeatureStore)) {

                return;

            }

            FeatureStore featureStore = (FeatureStore) store;

            FeatureSelection currentSelection = this.context.getSelecteds();

            if (currentSelection != null) {

                featureStore.setSelection(currentSelection);

                return;

            }

            FeatureQuery query = null;

            FeatureQuery contextQuery = this.context.getQuery();

            if (contextQuery != null) {

                Expression filter = contextQuery.getExpressionFilter();

                if (filter != null) {

                    query = featureStore.createFeatureQuery();

                    query.setFilter(filter);

                    for (EditableFeatureAttributeDescriptor extraColumn : contextQuery.getExtraColumns()) {

                        if(expressionUseIdentifier(filter, extraColumn.getName())){

                            query.getExtraColumns().getColumns().add(extraColumn);

                        }

                    }

//                    query.retrievesAllAttributes();

                }

            } 

            currentSelection = featureStore.getFeatureSelection();

            if( query == null ) {

                currentSelection.selectAll();

                return;

            }

            selection = featureStore.getFeatureSet(query);

            currentSelection.deselectAll();

            currentSelection.select(selection);

        } catch (Exception ex) {

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

        } finally {

            DisposeUtils.disposeQuietly(selection);

        }

    }

    @SuppressWarnings("Convert2Lambda")

    private boolean expressionUseIdentifier(Expression filter, String name) {

        MutableBoolean res = new MutableBoolean(false);

        try {

            filter.getCode().accept(new Visitor() {

                @Override

                public void visit(Object obj) throws VisitCanceledException, BaseException {

                    if (((Code) obj).code() == Code.IDENTIFIER) {

                        if (StringUtils.equalsIgnoreCase(name, ((Code.Identifier) obj).name())) {

                            res.setTrue();

                            throw new VisitCanceledException();

                        }

                    } else if (((Code) obj).code() == Code.CALLABLE) {

                        Code.Callable callable = (Code.Callable) obj;

                        if (StringUtils.equalsIgnoreCase("GETATTR", callable.name()) &&

                            callable.parameters().size() == 2 &&

                            callable.parameters().get(1).code() == Code.CONSTANT) {

                            Code.Constant p1 = (Code.Constant) callable.parameters().get(1);

                            if (StringUtils.equalsIgnoreCase(name, Objects.toString(p1.value(), null))) {

                                res.setTrue();

                                throw new VisitCanceledException();

                            }

                        }

                    }

                }

            });

        } catch (VisitCanceledException e) {

        } catch (Exception e) {

            throw new RuntimeException("Can't check if '"+name+"' is in '"+filter.getPhrase()+"'", e);

        }

        return res.booleanValue();

    }

}