Application: gvSIG desktop

/* gvSIG. Sistema de Informaci�n Geogr�fica de la Generalitat Valenciana

 *

 * Copyright (C) 2005 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib��ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

 *

 * $Id$

 * $Log$

 * Revision 1.16  2007-03-06 16:34:02  caballero

 * Exceptions

 *

 * Revision 1.15  2007/01/04 07:24:32  caballero

 * isModified

 *

 * Revision 1.14  2006/12/27 11:08:56  fdiaz

 * Arreglado el metodo hasVectorLayers para que no se pare en los grupos de capas cuando estos no tienen capas vectoriales.

 *

 * Revision 1.13  2006/09/25 16:54:12  caballero

 * invert selection

 *

 * Revision 1.12  2006/09/20 13:17:40  jaume

 * *** empty log message ***

 *

 * Revision 1.11  2006/09/20 10:56:56  sbayarri

 * Refactorizaci�n clases View y ProjectView

 *

 * Revision 1.10  2006/09/15 10:40:03  caballero

 * extensibilidad de documentos

 *

 * Revision 1.9  2006/09/11 16:14:15  fjp

 * Invertir seleccion. Falta subirlo al branch de la 1.0

 *

 * Revision 1.8  2006/09/11 06:53:13  fjp

 * Antes del parto

 *

 * Revision 1.7  2006/09/04 16:26:54  fjp

 * Seleccionar por pol�gono

 *

 * Revision 1.6  2006/08/29 07:56:27  cesar

 * Rename the *View* family of classes to *Window* (ie: SingletonView to SingletonWindow, ViewInfo to WindowInfo, etc)

 *

 * Revision 1.5  2006/08/29 07:21:08  cesar

 * Rename com.iver.cit.gvsig.fmap.Fmap class to com.iver.cit.gvsig.fmap.MapContext

 *

 * Revision 1.4  2006/08/29 07:13:53  cesar

 * Rename class com.iver.andami.ui.mdiManager.View to com.iver.andami.ui.mdiManager.IWindow

 *

 * Revision 1.3  2006/08/18 08:40:04  jmvivo

 * Actualizado para que el isEnabled tenga en cuenta que las capas esten 'avialable'

 *

 * Revision 1.2  2006/07/27 06:20:28  jaume

 * *** empty log message ***

 *

 * Revision 1.1  2006/07/26 09:53:05  jaume

 * separadas las herramientas de selecci�n del viewControls al ViewSelectionControls

 *

 *

 */

package com.iver.cit.gvsig;

import org.apache.log4j.Logger;

import org.gvsig.data.ReadException;

import org.gvsig.data.vectorial.FeatureCollection;

import org.gvsig.data.vectorial.FeatureStore;

import org.gvsig.fmap.mapcontext.MapContext;

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

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

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.plugins.Extension;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.cit.gvsig.gui.selectionByTheme.DefaultSelectionByThemeModel;

import com.iver.cit.gvsig.gui.selectionByTheme.MySelectionByThemeListener;

import com.iver.cit.gvsig.gui.selectionByTheme.SelectionByTheme;

import com.iver.cit.gvsig.project.documents.ProjectDocument;

import com.iver.cit.gvsig.project.documents.view.IProjectView;

import com.iver.cit.gvsig.project.documents.view.gui.View;

/**

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

 * vectorial layers only.

 *

 * @author jaume dominguez faus - jaume.dominguez@iver.es

 *

 */

public class ViewSelectionControls extends Extension {

        private static Logger logger = Logger.getLogger(ViewControls.class

                        .getName());

        public void initialize() {

                // TODO Auto-generated method stub

                registerIcons();

        }

        private void registerIcons(){

                PluginServices.getIconTheme().registerDefault(

                                "view-select-geometry",

                                this.getClass().getClassLoader().getResource("images/Select.png")

                        );

                PluginServices.getIconTheme().registerDefault(

                                "view-select-by-rectangle",

                                this.getClass().getClassLoader().getResource("images/SelEspacial2b.png")

                        );

                PluginServices.getIconTheme().registerDefault(

                                "view-select-by-polygon",

                                this.getClass().getClassLoader().getResource("images/SelectByPolygon.png")

                        );

                PluginServices.getIconTheme().registerDefault(

                                "view-select-invert",

                                this.getClass().getClassLoader().getResource("images/invertSelection.png")

                        );

        }

        public void execute(String actionCommand) {

                View vista = (View) PluginServices.getMDIManager().getActiveWindow();

                IProjectView model = vista.getModel();

                MapContext mapa = model.getMapContext();

                MapControl mapCtrl = vista.getMapControl();

                logger.debug("Comand : " + actionCommand);

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

                        mapCtrl.setTool("rectSelection");

                        ((ProjectDocument)vista.getModel()).setModified(true);

                } else if (actionCommand.equals("SELPOINT")) {

                        mapCtrl.setTool("pointSelection");

                        ((ProjectDocument)vista.getModel()).setModified(true);

                } else if (actionCommand.equals("SELPOL")) {

                        mapCtrl.setTool("polSelection");

                        ((ProjectDocument)vista.getModel()).setModified(true);

                } else if (actionCommand.equals("SELECTIONBYSHAPE")) {

                        SelectionByTheme dlg = new SelectionByTheme();

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

                        // int count = 0;

                        dlg.setModel(new DefaultSelectionByThemeModel());

                        dlg.addSelectionListener(new MySelectionByThemeListener());

                        PluginServices.getMDIManager().addWindow(dlg);

                        ((ProjectDocument)vista.getModel()).setModified(true);

                } else if (actionCommand.equals("INVERT_SELECTION")) {

                        for (int i = 0; i < mapa.getLayers().getActives().length; i++) {

                                FLayer lyr = mapa.getLayers().getActives()[i];

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

                                        FLyrVect lyrVect = (FLyrVect) lyr;

                                        FeatureStore fs;

                                        try {

                                                fs = lyrVect.getFeatureStore();

                                                FeatureCollection fCollectionAll=(FeatureCollection)fs.getDataCollection();

                                                FeatureCollection fCollection = (FeatureCollection)fs.getSelection();

                                                fCollectionAll.invalidate(fCollection);

                                                fs.setSelection(fCollectionAll);

                                        } catch (ReadException e) {

                                                e.printStackTrace();

                                                NotificationManager.addError(e);

                                        }

                                }

                        }

                        ((ProjectDocument)vista.getModel()).setModified(true);

                }

        }

        public boolean isEnabled() {

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

                if (f instanceof View) {

                        View vista = (View) f;

                        IProjectView model = vista.getModel();

                        MapContext mapa = model.getMapContext();

                        for (int i = 0; i < mapa.getLayers().getActives().length; i++) {

                                FLayer lyr = mapa.getLayers().getActives()[i];

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

                                        return true;

                                }

                        }

                        return false;

                }

                return true;

        }

        public boolean isVisible() {

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

                if (f != null && f instanceof View) {

                        View vista = (View) f;

                        IProjectView model = vista.getModel();

                        MapContext mapa = model.getMapContext();

                        return hasVectorLayers(mapa.getLayers());

                }

                return false;

        }

        private boolean hasVectorLayers(FLayers layers) {

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

                        FLayer lyr =layers.getLayer(i);

                        if (lyr instanceof FLayers){

                                if (hasVectorLayers((FLayers) lyr)){

                                        return true;

                                }

                        } else if (lyr instanceof FLyrVect) {

                                return true;

                        }

                }

                return false;

        }

}