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.fmap.mapcontrol.tools;

import java.awt.Image;

import org.gvsig.fmap.IconThemeHelper;

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

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

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

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

import org.gvsig.fmap.dal.feature.exception.CreateGeometryException;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.Geometry.TYPES;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.primitive.GeneralPathX;

import org.gvsig.fmap.geom.primitive.Surface;

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.tools.Events.MeasureEvent;

import org.gvsig.fmap.mapcontrol.tools.Listeners.PolylineListener;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * <p>Listener that selects all features of the active and vector layers which intersect with the defined

 *  polygon area in the associated {@link MapControl MapControl} object.</p>

 *

 * <p>The selection will be produced after user finishes the creation of the polyline.</p>

 *

 * @author Vicente Caballero Navarro

 */

public class PolygonSelectionListener implements PolylineListener {

        private static final GeometryManager geomManager = GeometryLocator.getGeometryManager();

        private static final Logger logger = LoggerFactory.getLogger(PolygonSelectionListener.class);

        /**

         * The cursor used to work with this tool listener.

         *

         * @see #getCursor()

         */

        private MapControl mapCtrl;

        /**

          * <p>Creates a new <code>PolygonSelectionListener</code> object.</p>

         *

         * @param mc the <code>MapControl</code> where is drawn the polyline

         */

        public PolygonSelectionListener(MapControl mc) {

                this.mapCtrl = mc;

        }

        public Image getImageCursor() {

                return IconThemeHelper.getImage("cursor-select-by-polygon");

        }

        public boolean cancelDrawing() {

                return false;

        }

        public void points(MeasureEvent event) throws BehaviorException {

                // TODO Auto-generated method stub

        }

        public void pointFixed(MeasureEvent event) throws BehaviorException {

        }

        public void polylineFinished(MeasureEvent event) throws BehaviorException {

                try {

                        GeometryManager geomManager = GeometryLocator.getGeometryManager();

            GeneralPathX gp = event.getGP();

            Surface geom = (Surface)geomManager.create(TYPES.SURFACE, SUBTYPES.GEOM2D);

            geom.setGeneralPath(gp);           

            FLayer[] actives = mapCtrl.getMapContext().getLayers().getActives();

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

                if (actives[i] instanceof FLyrVect) {

                    FLyrVect lyrVect = (FLyrVect) actives[i];

                                        FeatureSet newSelection = null;

                                        try {

                                                newSelection = lyrVect.queryByGeometry(geom,

                                                                                lyrVect.getFeatureStore()

                                                                                                .getDefaultFeatureType());

                                                if (event.getEvent().isControlDown()) {

                                                        FeatureSelection currentSelection = (FeatureSelection)lyrVect.getFeatureStore().getSelection();

                                                        if(isAnyGeomAlreadySelected(currentSelection, newSelection)) {

                                                                currentSelection.deselect(newSelection);

                                                        } else {

                                                                currentSelection.select(newSelection);

                                                        }

                                                } else {

                                                        lyrVect.getDataStore().setSelection(newSelection);

                                                }

                                        } finally {

                                                DisposeUtils.dispose(newSelection);

                    }

                }

            }

                } catch (DataException e) {

                        throw new BehaviorException("No se pudo hacer la selecci?n", e);

                } catch (org.gvsig.fmap.geom.exception.CreateGeometryException e) {

                        throw new BehaviorException("No se pudo hacer la selecci?n", 

                                        new CreateGeometryException(TYPES.SURFACE, SUBTYPES.GEOM2D, e));

                }

        }

        private boolean isAnyGeomAlreadySelected(FeatureSelection currentSelection, FeatureSet newSelection) throws DataException {

                DisposableIterator it = newSelection.fastIterator();

                while(it.hasNext()) {

                        Object obj = it.next();

                        if(obj instanceof Feature) {

                                Feature feat = ((Feature)obj);

                                if(currentSelection.isSelected(feat))

                                        return true;

                        }

                }

                return false;

        }

}