Application: gvSIG desktop

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

 *

 * Copyright (C) 2004 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

 */

package com.iver.cit.gvsig.fmap.operations.strategies;

import java.util.ArrayList;

import java.util.BitSet;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;

import com.iver.cit.gvsig.fmap.layers.layerOperations.AlphanumericData;

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class SelectedShapeVisitor implements FeatureVisitor {

        private SelectableDataSource sds=null;

        private ArrayList fgs=new ArrayList();

        private BitSet bitset;

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor#visit(com.iver.cit.gvsig.fmap.core.IGeometry, int)

         */

        public void visit(IGeometry g, int index) throws VisitException {

                if (bitset.get(index)){

                        fgs.add(g);

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor#stop(com.iver.cit.gvsig.fmap.layers.FLayer)

         */

        public void stop(FLayer layer) {

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor#start(com.iver.cit.gvsig.fmap.layers.FLayer)

         */

        public boolean start(FLayer layer) {

                if (layer instanceof AlphanumericData) {

                        try {

                                sds = ((AlphanumericData) layer).getRecordset();

                        } catch (DriverException e) {

                                e.printStackTrace();

                        }

                        try {

                                bitset = ((AlphanumericData) layer).getRecordset().getSelection();

                        } catch (DriverException e) {

                                e.printStackTrace();

                        }

                        return true;

                }

                return false;

        }

        public IGeometry[] getSelectedGeometries(){

                return (IGeometry[])fgs.toArray(new IGeometry[0]);

        }

        public SelectableDataSource getSelectableDataSource(){

                return sds;

        }

        public BitSet getBitSet(){

                return bitset;

        }

        public String getProcessDescription() {

                return "Adding selected geometries to a memory structure";

        }

}