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

import java.awt.geom.Rectangle2D;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.ObjectOutputStream;

import java.nio.ByteBuffer;

import java.nio.channels.FileChannel;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.data.DataSourceFactory;

import com.hardcode.gdbms.engine.data.NoSuchTableException;

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

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

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

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

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

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

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

/**

 * Adapta un driver de base de datos vectorial a la interfaz vectorial,

 * manteniendo adem?s el estado necesario por una capa vectorial de base de

 * datos (par?metros de la conexi?n)

 */

public class VectorialDisconnectedDBAdapter extends VectorialAdapter {

        private static final int REFRESH = 0;

        private static final int RECEIVING = 1;

        private static final int LOCAL = 2;

    private int numReg=-1;

    private DataSource ds = null;

    private int status = REFRESH;

    private VectorialDBAdapter connectedAdapter;

    private File dataFile;

    private File indexFile;

        /**

         * incrementa el contador de las veces que se ha abierto el fichero.

         * Solamente cuando el contador est? a cero pide al driver que conecte con

         * la base de datos

         */

        public void start() {

                switch (status){

                        case REFRESH:

                                Thread t = new Thread(new Receiver());

                                t.run();

                                break;

                        case RECEIVING:

                        case LOCAL:

                                break;

                }

        }

        /**

         * decrementa el contador de n?mero de aperturas y cuando llega a cero pide

         * al driver que cierre la conexion con el servidor de base de datos

         */

        public void stop() {

            ((VectorialDatabaseDriver)driver).close();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getShape(int)

         */

        public IGeometry getShape(int index) throws DriverIOException {

            return ((VectorialDatabaseDriver)driver).getShape(index);

        }

        private String getTableName(){

            return ((VectorialDatabaseDriver)driver).getTableName();

        }

        /**

         * @throws DriverException

         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getShapeCount()

         */

        public int getShapeCount() throws DriverIOException {

                if (status == LOCAL){

                        return 10;

                }else{

                    if (numReg == -1)

                    {

                        try

                    {

                            Statement s = ((VectorialDatabaseDriver)driver).getConnection().createStatement();                    

                            ResultSet r = s.executeQuery("SELECT COUNT(*) AS NUMREG FROM " + getTableName());

                            r.next();

                            numReg = r.getInt(1);

                            System.err.println("numReg = " + numReg);

                    }

                        catch (SQLException e)

                        {

                            throw new DriverIOException(e);

                        }

                    }

            return numReg;

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getFullExtent()

         */

        public Rectangle2D getFullExtent() {

            return ((VectorialDatabaseDriver)driver).getFullExtent();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getShapeType()

         */

        public int getShapeType() throws DriverIOException {

                return FShape.MULTI;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.VectorialAdapter#getRecordset()

         */

        public DataSource getRecordset() throws DriverLoadException {

            if (driver instanceof ObjectDriver)

            {

                        String name = LayerFactory.getDataSourceFactory().addDataSource((ObjectDriver)driver, null);

                        try {

                ds = LayerFactory.getDataSourceFactory().createRandomDataSource(name, DataSourceFactory.AUTOMATIC_MODE);

            } catch (NoSuchTableException e) {

                throw new RuntimeException(e);

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

                                throw new RuntimeException(e);

                        }

            }

                return ds;

        }

        public class Receiver implements Runnable{

                private ByteBuffer bb;

                private int currentCapacity = 0;

                private FileChannel channel;

                private ByteBuffer indexBuffer = ByteBuffer.allocate(4);

                private FileChannel indexChannel;

            private Rectangle2D extent = new Rectangle2D.Double();

                public ByteBuffer getBuffer(int capacity){

                        if (capacity > currentCapacity){

                                bb = ByteBuffer.allocate(capacity);

                                currentCapacity = capacity;

                        }

                        return bb;

                }

                private void writeObject(byte[] bytes) throws IOException{

                        //Se escribe al fichero de datos

                        bb = getBuffer(bytes.length + 4);

                        bb.clear();

                        bb.putInt(bytes.length);

                        bb.put(bytes);

                        bb.flip();

                        channel.write(bb);

                        //Se actualiza el fichero de ?ndice

                        indexBuffer.clear();

                        indexBuffer.putInt((int) channel.position());

                        indexBuffer.flip();

                        indexChannel.write(indexBuffer);

                }

                /**

                 * @see java.lang.Runnable#run()

                 */

                public void run() {

                        try {

                                //Abrimos el fichero de datos

                                dataFile = new File("/root/tirar/gvSIGtmp.gvs");

                                FileOutputStream fs = new FileOutputStream(dataFile);

                                channel = fs.getChannel();

                                //Abrimos el fichero de indice

                                indexFile = new File("/root/tirar/gvSIGindex.gvi");

                                FileOutputStream indexfs = new FileOutputStream(indexFile);

                                indexChannel = indexfs.getChannel();

                                //Creamos un adaptador conectado para bajarnos los datos

                                VectorialDatabaseDriver d = (VectorialDatabaseDriver) VectorialDisconnectedDBAdapter.this.driver;

                                connectedAdapter = new VectorialDBAdapter();

                                connectedAdapter.setDriver(d);

                                //Escribimos el n?mero de campos

                                indexBuffer.clear();

                                indexBuffer.putInt(connectedAdapter.getRecordset().getFieldCount());

                                indexBuffer.flip();

                                indexChannel.write(indexBuffer);

                                indexChannel.position(indexChannel.position() + 4);

                                //Reservamos espacio para el n?mero de campos

                                indexChannel.position(indexChannel.position() + 4 * 8);

                                connectedAdapter.start();

                                ByteArrayOutputStream bytes = new ByteArrayOutputStream();

                                ObjectOutputStream oos = new ObjectOutputStream(bytes);

                                int geom = 0;

                                for (int i = 0; i < connectedAdapter.getShapeCount(); i++) {

                                        geom++;

                                        IGeometry g = connectedAdapter.getShape(i);

                                        extent.add(g.getBounds2D());

                                        //Se obtienen los bytes del objeto a serializar

                                        bytes = new ByteArrayOutputStream();

                                        oos = new ObjectOutputStream(bytes);

                                        oos.writeObject(g);

                                        oos.close();

                                        //Se escribe al fichero

                                        writeObject(bytes.toByteArray());

                                        for (int j = 0; j < connectedAdapter.getRecordset().getFieldCount(); j++) {

                                                Value v = connectedAdapter.getRecordset().getFieldValue(i, j);

                                                //Se obtienen los bytes del objeto a serializar

                                                bytes = new ByteArrayOutputStream();

                                                oos = new ObjectOutputStream(bytes);

                                                oos.writeObject(v);

                                                oos.close();

                                                //Se escribe al fichero

                                                writeObject(bytes.toByteArray());

                                        }

                                }

                                //Escribimos el n?mero de geometr?as en el fichero de ?ndice

                                indexBuffer.clear();

                                indexBuffer.putInt(geom);

                                indexBuffer.flip();

                                indexChannel.position(4);

                                indexChannel.write(indexBuffer);

                                indexChannel.position(indexChannel.position() + 4);

                                //Escribimos el extent

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

                                extentBuffer.putDouble(extent.getMinX());

                                extentBuffer.putDouble(extent.getMinY());

                                extentBuffer.putDouble(extent.getMaxX());

                                extentBuffer.putDouble(extent.getMaxY());

                                extentBuffer.flip();

                                indexChannel.position(8);

                                indexChannel.write(extentBuffer);

                                //Cerramos

                                channel.close();

                                indexChannel.close();

                                connectedAdapter.stop();

                                status = LOCAL;

                        } catch (DriverIOException e) {

                                throw new RuntimeException(e);

                        } catch (IOException e) {

                                throw new RuntimeException(e);

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

                                throw new RuntimeException(e);

                        } catch (DriverLoadException e) {

                                throw new RuntimeException(e);

                        }

                }

        }

}