| gvSIG desktop 1

/* 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.drivers.shp.write;

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

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

import com.iver.cit.gvsig.fmap.core.v02.FConstant;

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

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;

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

import com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor;

import com.iver.cit.gvsig.fmap.operations.strategies.VisitException;

import java.io.File;

import java.io.IOException;

import java.io.OutputStream;

import java.io.RandomAccessFile;

import java.net.MalformedURLException;

import java.net.URL;

import java.nio.channels.Channels;

import java.nio.channels.FileChannel;

import java.nio.channels.WritableByteChannel;

/**

 * Visitor de creaci?n de un shape de los elementos seleccionados.

 *

 * @author Vicente Caballero Navarro

 */

public class SHPSHXFromSelectedVisitor implements FeatureVisitor {

        private IGeometry[] fgs = null;

        private SelectableDataSource sds = null;

        private String shpPath;

        private String shxPath;

        private int temp = 0;

        private int type;

        private FBitSet bitset = null;

        private int num = 0;

        /**

         * DOCUMENT ME!

         *

         * @param f DOCUMENT ME!

         */

        public void setFile(File f) {

                shpPath = f.getAbsolutePath();

                String strFichshx = f.getAbsolutePath().replaceAll("\\.shp", ".shx");

                shxPath = strFichshx.replaceAll("\\.SHP", ".SHX");

        }

        /**

         * DOCUMENT ME!

         *

         * @param layer DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        /*public void setType(int i) {

           type = i;

           }

         */

        /**

         * Inicializa el visitor.

         *

         * @param layer Capa.

         *

         * @return True si se inicializa correctamente.

         */

        public boolean start(FLayer layer) {

                if (layer instanceof AlphanumericData && layer instanceof Selectable) {

                        try {

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

                                bitset = ((Selectable) layer).getSelection();

                                ///fgs = new IGeometry[(int) sds.getRowCount()];

                                fgs = new IGeometry[bitset.cardinality()];

                                ///} catch (DriverException e) {

                                ///        e.printStackTrace();

                        } catch (com.iver.cit.gvsig.fmap.DriverException e) {

                                e.printStackTrace();

                        }

                        return true;

                }

                return false;

        }

        /**

         * Finaliza el visitor.

         *

         * @param layer Capa.

         */

        public void stop(FLayer layer) {

                SHPFileWrite filewrite;

                try {

                        filewrite = new SHPFileWrite((FileChannel) getWriteChannel(shpPath),

                                        (FileChannel) getWriteChannel(shxPath));

                        if (fgs.length > 0) {

                                type = getTypeShape(fgs[0].getGeometryType());

                                filewrite.write(fgs, type);

                        }

                } catch (ShapefileException e1) {

                        e1.printStackTrace();

                } catch (MalformedURLException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

                System.out.println("Acabado SHP y SHX");

        }

        /**

         * @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[num] = g;

                        num++;

                }

        }

        /**

         * DOCUMENT ME!

         *

         * @param url DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         *

         * @throws IOException DOCUMENT ME!

         */

        private WritableByteChannel getWriteChannel(String path)

                throws IOException {

                WritableByteChannel channel;

                        File f = new File(path);

                        if (!f.exists())

                        {

                            System.out.println("Creando fichero " + f.getAbsolutePath());

                            if (!f.createNewFile()) {

                                System.err.print("Error al crear el fichero " + f.getAbsolutePath());

                                throw new IOException("Cannot create file " + f);

                            }

                        }

                        RandomAccessFile raf = new RandomAccessFile(f, "rw");

                        channel = raf.getChannel();

                return channel;

        }

        /**

         * DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        public IGeometry[] getGeometries() {

                return fgs;

        }

        /**

         * DOCUMENT ME!

         *

         * @param geometryType DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        private int getTypeShape(int geometryType) {

                switch (geometryType) {

                        case FShape.POINT:

                                return FConstant.SHAPE_TYPE_POINT;

                        case FShape.LINE:

                                return FConstant.SHAPE_TYPE_POLYLINE;

                        case FShape.POLYGON:

                                return FConstant.SHAPE_TYPE_POLYGON;

                        case FShape.MULTI:

                                return 0; //TODO falta aclarar cosas aqu?.

                }

                return FConstant.SHAPE_TYPE_NULL;

        }

        /**

         * DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        public SelectableDataSource getSDS() {

                return sds;

        }

}