| 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.IGeometry;

import com.iver.cit.gvsig.fmap.drivers.shp.DbaseFileHeaderNIO;

import com.iver.cit.gvsig.fmap.drivers.shp.DbaseFileWriterNIO;

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

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 zoom a lo seleccionado.

 *

 * @author Vicente Caballero Navarro

 */

public class DBFFromSelected {

        private IGeometry[] fgs = null;

        private boolean hasdbf = false;

        private URL dbfURL = null;

        private int temp = 0;

        private DbaseFileWriterNIO dbfWrite;

        private Integer[] enteros;

        private SelectableDataSource sds;

        private Object[] record;

        //private DbaseFileNIO m_FichDbf = new DbaseFileNIO();

        /**

         * Inserta el fichero.

         *

         * @param f Fichero.

         */

        public void setFile(File f) {

                try {

                        String strFichDbf = f.getAbsolutePath().replaceAll("\\.shp", ".dbf");

                        strFichDbf = strFichDbf.replaceAll("\\.SHP", ".DBF");

                        dbfURL = new URL("file://" + strFichDbf);

                } catch (MalformedURLException e) {

                        e.printStackTrace();

                }

        }

        /**

         * Inserta true si se quiere crear el dbf vacio y no copiar el contenido de otro dbf.

         *

         * @param b True si se quiere crear vacio el dbf.

         */

        public void setIsNewDBF(boolean b) {

                hasdbf = b;

        }

        /**

         * Inserta las geometr?as.

         *

         * @param g Geometr?as.

         */

        public void setGeometries(IGeometry[] g) {

                fgs = g;

        }

        /**

         * Inicializa.

         *

         * @param sds Capa.

         */

        public void start(SelectableDataSource sds) {

                //if (layer instanceof AlphanumericData) {

                try {

                        this.sds = sds;

                        if (!hasdbf) {

                                DbaseFileHeaderNIO myHeader = DbaseFileHeaderNIO.createNewDbaseHeader();

                                myHeader.setNumRecords(fgs.length);

                                dbfWrite = new DbaseFileWriterNIO(myHeader,

                                                (FileChannel) getWriteChannel(getStorageURL(dbfURL)));

                                enteros = new Integer[1];

                        } else {

                                //VectorialFileAdapter vfa=(VectorialFileAdapter)((SingleLayer)lv).getSource();

                                DbaseFileHeaderNIO myHeader;

                                myHeader = DbaseFileHeaderNIO.createDbaseHeader(sds);

                                myHeader.setNumRecords(fgs.length);

                                dbfWrite = new DbaseFileWriterNIO(myHeader,

                                                (FileChannel) getWriteChannel(getStorageURL(dbfURL)));

                                record = new Object[sds.getFieldCount()];

                        }

                } catch (IOException e) {

                        e.printStackTrace();

                        ///} catch (DriverException e1) {

                        //        e1.printStackTrace();

                } catch (com.hardcode.gdbms.engine.data.DriverException e2) {

                        e2.printStackTrace();

                }

                //return true;

                //}

                //return false;

        }

        /**

         * Finaliza.

         */

        public void stop() {

                System.out.println("Acabado el DBF con NIO escritura");

        }

        /**

         * Rellena los registros del dbf.

         */

        public void createdbfRow() {

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

                        try {

                                if (!hasdbf) {

                                        enteros[0] = Integer.valueOf(String.valueOf(i));

                                        dbfWrite.write(enteros);

                                } else {

                                        for (int r = 0; r < sds.getFieldCount(); r++) {

                                                record[r] = sds.getFieldValue(i, r);

                                        }

                                        dbfWrite.write(record);

                                }

                        } catch (com.hardcode.gdbms.engine.data.DriverException e) {

                                e.printStackTrace();

                        } catch (IOException e1) {

                                e1.printStackTrace();

                        }

                }

        }

        protected URL getStorageURL(URL url) throws java.net.MalformedURLException {

                return (temp == 0) ? url : getStorageFile(url).toURL();

        }

        protected File getStorageFile(URL url) {

                String f = url.getFile();

                f = temp + f.substring(f.lastIndexOf("/") + 1);

                File tf = new File(System.getProperty("java.io.tmpdir"), f);

                return tf;

        }

        private WritableByteChannel getWriteChannel(URL url)

                throws IOException {

                WritableByteChannel channel;

                if (url.getProtocol().equals("file")) {

                        File f = new File(url.getFile());

                        if (!f.exists() && !f.createNewFile()) {

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

                        }

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

                        channel = raf.getChannel();

                } else {

                        OutputStream out = url.openConnection().getOutputStream();

                        channel = Channels.newChannel(out);

                }

                return channel;

        }

}