Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * 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 3 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.

 *

 * For any additional information, do not hesitate to contact us at info AT

 * gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.app.extension;

import java.util.List;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

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

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

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Extension that handles the selection tools, selection tools have sense on

 * vectorial layers only.

 *

 */

public class ViewSelectionControls extends Extension {

    private static final Logger logger = LoggerFactory

            .getLogger(ViewSelectionControls.class);

    @Override

    public void initialize() {

        registerIcons();

    }

    private void registerIcons() {

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

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

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

    }

    @Override

    public void execute(String actionCommand) {

        ApplicationManager application = ApplicationLocator.getManager();

        IView view = (IView) application.getActiveComponent(ViewDocument.class);

        if (view == null) {

            return;

        }

        ViewDocument document = view.getViewDocument();

        MapControl mapCtrl = view.getMapControl();

        if (actionCommand.equalsIgnoreCase("selection-select-by-rectangle-view")) {

            mapCtrl.setTool("rectSelection");

            document.setModified(true);

        } else if (actionCommand.equalsIgnoreCase("selection-select-by-polygon")) {

            mapCtrl.setTool("polSelection");

            document.setModified(true);

        }

    }

    @Override

    public boolean isEnabled() {

        ApplicationManager application = ApplicationLocator.getManager();

        IView view = (IView) application.getActiveComponent(ViewDocument.class);

        if (view == null) {

            return false;

        }

        ViewDocument document = view.getViewDocument();

        if( document == null ) {

            return false;

        }

        List<FLayer> layers = document.getMapContext().getLayers().getLayers(

                FLayers.LAYER_ACTIVE

                        .and(FLayers.LAYER_AVALAIBLE)

                        .and(FLayers.LAYER_SELECTION_AVAILABLE)

        );

        if (layers.size()>=1) {

            return true;

        }

        return false;

    }

    @Override

    public boolean isVisible() {

        ApplicationManager application = ApplicationLocator.getManager();

        IView view = (IView) application.getActiveComponent(ViewDocument.class);

        if (view == null) {

            return false;

        }

        ViewDocument document = view.getViewDocument();

        return document.getMapContext().hasVectorLayers();

    }

}