| gvSIG desktop 1

package org.gvsig.selectiontools.app.extension;

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 * 

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *  

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, 

 * MA  02110-1301, USA.

 * 

 */

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.gui.IView;

import org.gvsig.fmap.dal.exception.DataException;

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

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontrol.MapControl;

/**

 * <p>

 * Extension to add support for selecting all the features of a vector layer.

 * </p>

 * 

 * @author Pablo Piqueras Bartolom? (pablo.piqueras@iver.es)

 */

public class SelectAllExtension extends Extension {

    /*

     * @see com.iver.andami.plugins.IExtension#initialize()

     */

    public void initialize() {

            IconThemeHelper.registerIcon("action", "selection-select-all", this);

    }

    /*

     * (non-Javadoc)

     * 

     * @see com.iver.andami.plugins.IExtension#execute(java.lang.String)

     */

    public void execute(String actionCommand) {

        if (actionCommand.equals("SELALL")) {

            IWindow view = PluginServices.getMDIManager().getActiveWindow();

            if (view instanceof IView) {

                MapControl mc = ((IView) view).getMapControl();

                FLayer[] activeLayers =

                    mc.getMapContext().getLayers().getActives();

                FLayer layer;

                for (int i = 0; i < activeLayers.length; i++) {

                    layer = activeLayers[i];

                    if ((layer.isAvailable()) && (layer instanceof FLyrVect)) {

                        FLyrVect lyrVect = (FLyrVect) layer;

                        try {

                            FeatureStore fs = lyrVect.getFeatureStore();

                            fs.getFeatureSelection().selectAll();

                        } catch (DataException e) {

                            NotificationManager.showMessageError("Data exception",

                                e);

                        }

                    }

                }

            }

        }

    }

    /*

     * @see com.iver.andami.plugins.IExtension#isVisible()

     */

    public boolean isVisible() {

        IWindow f = PluginServices.getMDIManager().getActiveWindow();

        if (f == null) {

            return false;

        }

        if (f instanceof IView) {

            IView vista = (IView) f;

            ViewDocument viewDocument = vista.getViewDocument(); // .getModel();

            MapContext mapa = viewDocument.getMapContext();

            return mapa.getLayers().getLayersCount() > 0;

        }

        return false;

    }

    /*

     * @see com.iver.andami.plugins.IExtension#isEnabled()

     */

    public boolean isEnabled() {

        IWindow f = PluginServices.getMDIManager().getActiveWindow();

        if (f == null) {

            return false;

        }

        if (f instanceof IView) {

            IView vista = (IView) f;

            ViewDocument viewDocument = vista.getViewDocument();

            MapContext mapa = viewDocument.getMapContext();

            FLayer layers[] = mapa.getLayers().getActives();

            FLayer layer;

            for (int i = 0; i < layers.length; i++) {

                layer = layers[i];

                if ((layer instanceof FLyrVect) && (layer.isAvailable())

                    && (layer.isActive()))

                    return true;

            }

        }

        return false;

    }

}