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

import java.awt.geom.Rectangle2D;

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.ObjectInputStream;

import java.nio.ByteBuffer;

import java.nio.channels.FileChannel;

import com.hardcode.gdbms.engine.data.driver.DriverException;

import com.hardcode.gdbms.engine.data.driver.ObjectDriver;

import com.hardcode.gdbms.engine.values.Value;

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

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

import com.iver.cit.gvsig.fmap.drivers.DriverAttributes;

import com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver;

public class GVSDriver implements VectorialFileDriver, ObjectDriver {

        private File file;

        private FileChannel channel;

        private File indexFile;

        private FileChannel indexChannel;

        private ByteBuffer byteCountBuffer = ByteBuffer.allocate(4);

        private int fieldCount;

        private int rowCount; 

        private Rectangle2D extent;

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#open(java.io.File)

         */

        public void open(File f) throws IOException {

                //Abrimos el fichero de ?ndice

                String name = f.getName().substring(0, f.getName().length() - 4) + ".gvi";

                indexFile = new File(name);

                FileInputStream fis = new FileInputStream(indexFile);

                indexChannel = fis.getChannel();

                //leemos el n?mero de geometr?as y de campos

                ByteBuffer indexBuffer = ByteBuffer.allocate(4);

                indexChannel.read(indexBuffer);

                fieldCount = indexBuffer.getInt();

                indexChannel.read(indexBuffer);

                rowCount = indexBuffer.getInt();

                //Leemos el extent

                ByteBuffer extentBuffer = ByteBuffer.allocate(8 * 4);

                indexChannel.read(extentBuffer);

                extent = new Rectangle2D.Double(extentBuffer.getDouble(),

                                extentBuffer.getDouble(),

                                extentBuffer.getDouble(),

                                extentBuffer.getDouble());

                //Abrimos el fichero de datos

                name = f.getName();

                file = new File(name);

                fis = new FileInputStream(file);

                channel = fis.getChannel();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#close()

         */

        public void close() throws IOException {

                channel.close();

                indexChannel.close();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#initialize()

         */

        public void initialize() throws IOException {

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#accept(java.io.File)

         */

        public boolean accept(File f) {

                return (f.getName().toLowerCase().endsWith("gvs"));

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver#getShape(int)

         */

        public IGeometry getShape(int index) throws IOException {

                //Obtenemos el ?ndice del fichero de ?ndices

                int fileIndex = getFileIndex(index, -1);

                //Leemos la geometr?a

                channel.position((long)fileIndex);

                //Leemos el tama?o en bytes

                channel.read(byteCountBuffer);

                int geomBytes = byteCountBuffer.getInt();

                //Leemos la geometr?a

                ByteBuffer byteBuffer = ByteBuffer.allocate(geomBytes);

                channel.read(byteBuffer);

                byte[] byteArray = new byte[geomBytes];

                byteBuffer.get(byteArray);

                ByteArrayInputStream bais = new ByteArrayInputStream(byteArray);

                ObjectInputStream ois = new ObjectInputStream(bais);

                try {

                        return (IGeometry) ois.readObject();

                } catch (ClassNotFoundException e) {

                        throw new IOException(e.getMessage());

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getShapeType()

         */

        public int getShapeType() {

                return FShape.MULTI;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getShapeCount()

         */

        public int getShapeCount() throws IOException {

                return rowCount;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getDriverAttributes()

         */

        public DriverAttributes getDriverAttributes() {

                return null;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getFullExtent()

         */

        public Rectangle2D getFullExtent() throws IOException {

                return extent;

        }

        /**

         * @see com.hardcode.driverManager.Driver#getName()

         */

        public String getName() {

                return "gvs driver";

        }

        private int getFileIndex(int rowIndex, int columnIndex){

                //Obtenemos la fila en el fichero de ?ndices

                int indexFileIndex = 2*4 + 8 * 4;

                indexFileIndex += (fieldCount + 1) * rowIndex;

                columnIndex++;

                indexFileIndex += columnIndex * 4;

                return indexFileIndex;

        }

        /**

         * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getFieldValue(long, int)

         */

        public Value getFieldValue(long rowIndex, int fieldIndex) throws DriverException {

                int fileIndex = getFileIndex((int) rowIndex, fieldIndex);

                try {

                        //Leemos el tama?o en bytes

                        channel.position((long)fileIndex);

                        channel.read(byteCountBuffer);

                        int geomBytes = byteCountBuffer.getInt();

                        //Leemos el value

                        ByteBuffer byteBuffer = ByteBuffer.allocate(geomBytes);

                        channel.read(byteBuffer);

                        byte[] byteArray = new byte[geomBytes];

                        byteBuffer.get(byteArray);

                        ByteArrayInputStream bais = new ByteArrayInputStream(byteArray);

                        ObjectInputStream ois = new ObjectInputStream(bais);

                        return (Value) ois.readObject();

                } catch (IOException e1) {

                        throw new DriverException(e1);

                } catch (ClassNotFoundException e) {

                        throw new DriverException(e);

                }

        }

        /**

         * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getFieldCount()

         */

        public int getFieldCount() throws DriverException {

                return fieldCount;

        }

        /**

         * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getFieldName(int)

         */

        public String getFieldName(int arg0) throws DriverException {

                return "geom";

        }

        /**

         * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getRowCount()

         */

        public long getRowCount() throws DriverException {

                return rowCount;

        }

        /**

         * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getFieldType(int)

         */

        public int getFieldType(int arg0) throws DriverException {

                // TODO Auto-generated method stub

                return 0;

        }

    /**

     * @see com.hardcode.gdbms.engine.data.driver.ObjectDriver#getPrimaryKeys()

     */

    public int[] getPrimaryKeys() throws DriverException {

        // TODO Auto-generated method stub

        return null;

    }

}