| gvSIG desktop 1

/*

 * Created on 12-ene-2006 by fjp

 *

 * 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

 */

/* CVS MESSAGES:

 *

 * $Id: VectorialEditableAdapter.java 3996 2006-02-08 15:21:08Z caballero $

 * $Log$

 * Revision 1.18  2006-02-08 15:18:45  caballero

 * control de las rows eliminadas

 *

 * Revision 1.17  2006/02/07 10:18:44  caballero

 * Con BoundedShape

 *

 * Revision 1.16  2006/02/06 12:01:41  caballero

 * driver del ova

 *

 * Revision 1.15  2006/02/03 14:09:32  fjp

 * Preparando edici?n

 *

 * Revision 1.14  2006/02/03 12:16:33  fjp

 * Preparando edici?n

 *

 * Revision 1.13  2006/02/03 11:54:12  caballero

 * tablas con vectorialEditableAdapter en edici?n

 *

 * Revision 1.11  2006/01/31 08:10:05  caballero

 * cambio de feature a row

 *

 * Revision 1.10  2006/01/30 08:18:14  caballero

 * m?todos para deshacer y rehacer

 *

 * Revision 1.9  2006/01/23 17:30:28  caballero

 * coger los datos del ova

 *

 * Revision 1.8  2006/01/23 16:16:16  caballero

 * getRowIndex

 *

 * Revision 1.7  2006/01/20 08:37:10  fjp

 * Preparando la edici?n

 *

 * Revision 1.6  2006/01/19 12:48:20  caballero

 * poder modificar su vectorial Adapter

 *

 * Revision 1.5  2006/01/19 09:28:11  fjp

 * Preparando la edici?n

 *

 * Revision 1.4  2006/01/17 10:24:02  fjp

 * Preparando edici?n

 *

 * Revision 1.3  2006/01/16 12:47:38  fjp

 * Preparando edici?n

 *

 * Revision 1.2  2006/01/16 11:23:00  fjp

 * Preparando edici?n

 *

 * Revision 1.1  2006/01/12 13:39:14  fjp

 * preaparar edicion

 *

 *

 */

package com.iver.cit.gvsig.fmap.edition;

import java.awt.geom.Rectangle2D;

import java.io.IOException;

import java.util.List;

import com.hardcode.driverManager.DriverLoadException;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

import com.vividsolutions.jts.geom.Envelope;

import com.vividsolutions.jts.index.quadtree.Quadtree;

/**

 * @author fjp

 *

 */

public class VectorialEditableAdapter extends EditableAdapter implements ReadableVectorial, BoundedShapes {

        private VectorialAdapter ova;

        //private VectorialDriver driver;

        private Quadtree index;

        public VectorialEditableAdapter(){

                super();

        }

        public void setOriginalVectorialAdapter(ReadableVectorial rv){

                ova=(VectorialAdapter)rv;

                try {

                        setOriginalDataSource(rv.getRecordset());

                } catch (DriverLoadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public void start() throws DriverIOException {

                ova.start();

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public void stop() throws DriverIOException {

                ova.stop();

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public IGeometry getShape(int rowIndex) throws DriverIOException {

                // Si no est? en el fichero de expansi?n

                int calculatedIndex=getCalculatedIndex(rowIndex);

                Integer integer = new Integer(calculatedIndex);

                if (!relations.containsKey(integer)) {

                        // Si ha sido eliminada

                        /*if (delRows.get(integer.intValue())) {

                                return null;

                        } else {*/

                                return ova.getShape(calculatedIndex);

                        //}

                } else {

                        int num = ((Integer) relations.get(integer)).intValue();

                        DefaultRowEdited feat;

                        try {

                                feat = (DefaultRowEdited) expansionFile.getRow(num);

                                if (feat.getStatus() == IRowEdited.STATUS_DELETED)

                                        return null;

                                return ((IFeature)feat.getLinkedRow()).getGeometry().cloneGeometry();//getGeometry();

                        } catch (IOException e) {

                                e.printStackTrace();

                                throw new DriverIOException(e);

                        }

                }

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public int getShapeType() throws DriverIOException {

                return ova.getShapeType();

        }

        public VectorialAdapter getOriginalAdapter(){

                return ova;

        }

        public VectorialDriver getDriver() {

                return ova.getDriver();

        }

        public void setDriver(VectorialDriver driver) {

                this.ova.setDriver(driver);

        }

        public DriverAttributes getDriverAttributes(){

                return ova.getDriverAttributes();

        }

         /**

     * DOCUMENT ME!

     *

     * @throws EditionException DOCUMENT ME!

     */

    public void startEdition() throws EditionException {

        super.startEdition();

        try {

            expansionFile.open();

            // TODO: Si la capa dispone de un ?ndice espacial, hacer

            // algo aqu? para que se use ese ?ndice espacial.

            index = new Quadtree();

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

                IGeometry g=null;

                                try {

                                        g = ((DefaultFeature) ova.getFeature(i)).getGeometry();

                                } catch (DriverException e1) {

                                        // TODO Auto-generated catch block

                                        e1.printStackTrace();

                                }

                if (g == null) {

                    continue;

                }

                Rectangle2D r = g.getBounds2D();

                Envelope e = new Envelope(r.getX(), r.getX() + r.getWidth(),

                        r.getY(), r.getY() + r.getHeight());

                index.insert(e, new Integer(i));

            }

        } catch (DriverIOException e) {

            throw new EditionException(e);

        } catch (IOException e) {

            throw new EditionException(e);

        }

        System.err.println("Se han metido en el ?ndice " +

            index.queryAll().size() + " geometr?as");

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.edition.IEditableSource#getRow(int)

     */

    public IRowEdited getRow(int index) throws DriverIOException, IOException {

        Integer integer = new Integer(index);

        //Si no est? en el fichero de expansi?n

        if (!relations.containsKey(integer)) {

            //Si ha sido eliminada

            if (delRows.get(index)) {

                return null;

            } else {

                DefaultRowEdited edRow=null;

                                try {

                                        edRow = new DefaultRowEdited(ova.getFeature(index),

                                                DefaultRowEdited.STATUS_ORIGINAL);

                                } catch (DriverException e) {

                                        // TODO Auto-generated catch block

                                        e.printStackTrace();

                                }

                return edRow;

            }

        } else {

            int num = ((Integer) relations.get(integer)).intValue();

            return expansionFile.getRow(num);

        }

    }

    /**

     * Elimina una geometria. Si es una geometr?a original de la capa en

     * edici?n se marca como eliminada (haya sido modificada o no). Si es una

     * geometr?a a?adida posteriormente se invalida en el fichero de

     * expansi?n, para que una futura compactaci?n termine con ella.

     *

     * @param index ?ndice de la geometr?a.

     *

     * @throws DriverIOException

     * @throws IOException

     */

    public void doRemoveRow(int index) throws DriverIOException, IOException {

        //Llega el calculatedIndex

            Integer integer = new Integer(index);

        IFeature feat = null;

        delRows.set(index, true);

        //Si la geometr?a no ha sido modificada

        if (!relations.containsKey(integer)) {

            try {

                                feat = (DefaultFeature) (ova.getFeature(index));

                        } catch (DriverException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

       } else {

                        int num = ((Integer) relations.get(integer)).intValue();

                        feat = (IFeature) expansionFile.getRow(num).getLinkedRow();

                        //expansionFile.invalidateRow(num);

                }

        System.err.println("Elimina una Row en la posici?n: " + index);

        //Se actualiza el ?ndice

        if (feat != null) {

            Rectangle2D r = feat.getGeometry().getBounds2D();

            boolean borrado=this.index.remove(new Envelope(r.getX(), r.getX() + r.getWidth(),

                    r.getY(), r.getY() + r.getHeight()), new Integer(index));

            System.out.println("Est? borrado : "+ borrado);

            System.out.println("Index.lenght : "+ this.index.size());

        }

    }

    /**

     * Si se intenta modificar una geometr?a original de la capa en edici?n se

     * a?ade al fichero de expansi?n y se registra la posici?n en la que se

     * a?adi?. Si se intenta modificar una geometria que se encuentra en el

     * fichero de expansi?n (por ser nueva o original pero modificada) se

     * invoca el m?todo modifyGeometry y se actualiza el ?ndice de la

     * geometria en el fichero.

     *

     * @param index DOCUMENT ME!

     * @param feat DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws IOException

     * @throws DriverIOException

     */

    public int doModifyRow(int index, IRow feat)

        throws IOException, DriverIOException {

        int pos = -1;

        Integer integer = new Integer(index);

        IFeature featAnt = null;

        System.err.println("Modifica una Row en la posici?n: " + index);

        //Si la geometr?a no ha sido modificada

        if (!relations.containsKey(integer)) {

            int expansionIndex = expansionFile.addRow(feat);

            relations.put(integer, new Integer(expansionIndex));

            //Se actualiza el ?ndice espacial

            try {

                                featAnt = (DefaultFeature) (ova.getFeature(index));

                        } catch (DriverException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

            Rectangle2D rAnt = featAnt.getGeometry().getBounds2D();

            Rectangle2D r = ((IFeature) feat).getGeometry().getBounds2D();

            this.index.remove(new Envelope(rAnt.getX(),

                    rAnt.getX() + rAnt.getWidth(), rAnt.getY(),

                    rAnt.getY() + rAnt.getHeight()), new Integer(index));

            this.index.insert(new Envelope(r.getX(), r.getX() + r.getWidth(),

                    r.getY(), r.getY() + r.getHeight()), new Integer(index));

        } else {

            //Obtenemos el ?ndice en el fichero de expansi?n

            int num = ((Integer) relations.get(integer)).intValue();

            pos = num;

            //Obtenemos la geometr?a para actualiza el ?ndice espacialposteriormente

            featAnt = (IFeature) expansionFile.getRow(num).getLinkedRow();

            /*

             * Se modifica la geometr?a y nos guardamos el ?ndice dentro del fichero

             * de expansi?n en el que se encuentra la geometr?a modificada

             */

            num = expansionFile.modifyRow(num, feat);

            /*

             * Actualiza la relaci?n del ?ndice de la geometr?a al ?ndice en el fichero

             * de expansi?n.

             */

            relations.put(integer, new Integer(num));

            //Se modifica el ?ndice espacial

            Rectangle2D rAnt = featAnt.getGeometry().getBounds2D();

            Rectangle2D r = ((IFeature) feat).getGeometry().getBounds2D();

            this.index.remove(new Envelope(rAnt.getX(),

                    rAnt.getX() + rAnt.getWidth(), rAnt.getY(),

                    rAnt.getY() + rAnt.getHeight()), new Integer(index));

            this.index.insert(new Envelope(r.getX(), r.getX() + r.getWidth(),

                    r.getY(), r.getY() + r.getHeight()), new Integer(index));

        }

        return pos;

    }

    /**

     * Actualiza en el mapa de ?ndices, la posici?n en la que estaba la

     * geometr?a antes de ser modificada. Se marca como v?lida, en caso de que

     * fuera una modificaci?n de una geometr?a que estuviese en el fichero de

     * expansi?n antes de ser modificada y se pone el puntero de escritura del

     * expansion file a justo despues de la penultima geometr?a

     *

     * @param geometryIndex ?ndice de la geometr?a que se quiere deshacer su

     *        modificaci?n

     * @param previousExpansionFileIndex ?ndice que ten?a antes la geometr?a en

     *        el expansionFile. Si vale -1 quiere decir que es una

     *        modificaci?n de una geometr?a original y por tanto no hay que

     *        actualizar el mapa de indices sino eliminar su entrada.

     *

     * @throws IOException

     * @throws DriverIOException

     */

    public void undoModifyRow(int geometryIndex, int previousExpansionFileIndex)

        throws IOException, DriverIOException {

       //Llega el CalculatedIndex

            /*

         * Si la acci?n de modificar se realiz? sobre una geometr?a original

         */

        if (previousExpansionFileIndex == -1) {

            //Se obtiene la geometr?a para actualizar el ?ndice

            IGeometry g = ((DefaultFeature) getRow(geometryIndex).getLinkedRow()).getGeometry();

            Rectangle2D r = g.getBounds2D();

            //Se elimina de las relaciones y del fichero de expansi?n

            relations.remove(new Integer(geometryIndex));

            expansionFile.deleteLastRow();

            //Se actualizan los ?ndices

            IGeometry gAnt = ((DefaultFeature) getRow(geometryIndex)

                                                   .getLinkedRow()).getGeometry();

            Rectangle2D rAnt = gAnt.getBounds2D();

            this.index.remove(new Envelope(r.getX(), r.getX() + r.getWidth(),

                    r.getY(), r.getY() + r.getHeight()),

                new Integer(geometryIndex));

            this.index.insert(new Envelope(rAnt.getX(),

                    rAnt.getX() + rAnt.getWidth(), rAnt.getY(),

                    rAnt.getY() + rAnt.getHeight()), new Integer(geometryIndex));

        } else {

            //Se obtiene la geometr?a para actualizar el ?ndice

            IGeometry g = null;

            g = ((DefaultFeature) getRow(geometryIndex).getLinkedRow()).getGeometry();

            Rectangle2D r = g.getBounds2D();

            this.index.remove(new Envelope(r.getX(), r.getX() + r.getWidth(),

                    r.getY(), r.getY() + r.getHeight()),

                new Integer(geometryIndex));

            //Se actualiza la relaci?n de ?ndices

            //Integer integer = new Integer(geometryIndex);

            relations.put(new Integer(geometryIndex),

                new Integer(previousExpansionFileIndex));

            //Se recupera la geometr?a

            //expansionFile.validateRow(previousExpansionFileIndex);

            //Se actualizan los ?ndices

            g = ((DefaultFeature) getRow(geometryIndex).getLinkedRow()).getGeometry();

            r = g.getBounds2D();

            this.index.insert(new Envelope(r.getX(), r.getX() + r.getWidth(),

                    r.getY(), r.getY() + r.getHeight()),

                new Integer(geometryIndex));

        }

    }

    /**

     * A?ade una geometria al fichero de expansi?n y guarda la correspondencia

     * en la tabla relations.

     *

     * @param feat geometr?a a guardar.

     *

     * @return DOCUMENT ME!

     *

     * @throws DriverIOException

     * @throws IOException

     */

    public int doAddRow(IRow feat) throws DriverIOException, IOException {

        int virtualIndex=super.doAddRow(feat);

                // Actualiza el ?ndice espacial

        IGeometry g = ((IFeature) feat).getGeometry();

        Rectangle2D r = g.getBounds2D();

        index.insert(new Envelope(r.getX(), r.getX() + r.getWidth(), r.getY(),

                r.getY() + r.getHeight()), new Integer(virtualIndex));

        return virtualIndex;

    }

    /**

     * Se desmarca como invalidada en el fichero de expansion o como eliminada

     * en el fichero original

     *

     * @param index DOCUMENT ME!

     *

     * @throws IOException

     * @throws DriverIOException

     */

    public void undoRemoveRow(int index) throws IOException, DriverIOException {

       super.undoRemoveRow(index);

        IGeometry g = null;

        g = ((IFeature) getRow(index).getLinkedRow()).getGeometry();

        Rectangle2D r = g.getBounds2D();

        this.index.insert(new Envelope(r.getX(), r.getX() + r.getWidth(),

                r.getY(), r.getY() + r.getHeight()), new Integer(index));

    }

    /**

     * Se elimina del final del fichero de expansi?n poniendo el puntero de

     * escritura apuntando al final de la pen?ltima geometr?a. Deber? quitar

     * la relaci?n del mapa de relaciones

     *

     * @param index ?ndice de la geometr?a que se a?adi?

     *

     * @throws DriverIOException

     * @throws IOException

     */

    public void undoAddRow(int index) throws DriverIOException, IOException {

        IGeometry g = ((IFeature) getRow(index).getLinkedRow()).getGeometry();

        Rectangle2D r = g.getBounds2D();

        this.index.remove(new Envelope(r.getX(), r.getX() + r.getWidth(),

                r.getY(), r.getY() + r.getHeight()), new Integer(index));

        super.undoAddRow(index);

    }

    /**

     * Obtiene las geometr?as que se encuentran en el rect?ngulo que se pasa

     * como par?metro haciendo uso del ?ndice espacial

     *

     * @param r Rect?ngulo indicando la porci?n del espacio para el cual se

     *        quiere saber los ?ndices de las geometr?as que se encuentran

     *        dentro de ?l

     *

     * @return Array de ?ndices para su uso con getGeometry, removeGeometry,

     *         ...

     */

    public int[] getRowsIndexes(Rectangle2D r) {

        Envelope e = new Envelope(r.getX(), r.getX() + r.getWidth(), r.getY(),

                r.getY() + r.getHeight());

        List l = index.query(e);

        int[] indexes = new int[l.size()];

        for (int index = 0; index < l.size(); index++) {

            Integer i = (Integer) l.get(index);

            indexes[index] = getInversedIndex(i.intValue());

        }

        return indexes;

    }

    /**

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

         */

        public int getShapeCount() throws DriverIOException {

                try {

                        return getRowCount();

                }  catch (IOException e) {

                        throw new DriverIOException(e);

                }

        }

        /**

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

         */

        public Rectangle2D getFullExtent() throws DriverIOException {

                // TODO: Recalcularlo cada vez que se a?ade un addRow

                return ova.getFullExtent();

        }

    /**

     * En la implementaci?n por defecto podemos hacer que cada

     * feature tenga ID = numero de registro.

     * En el DBAdapter podr?amos "overrride" este m?todo y poner

     * como ID de la Feature el campo ?nico escogido en

     * la base de datos.

     * @param numReg

     * @return

     * @throws DriverException

     */

    public IFeature getFeature(int numReg) throws DriverException

    {

        IGeometry geom;

        IFeature feat = null;

        try {

            geom = getShape(numReg);

            DataSource rs = getRecordset();

            Value[] regAtt = new Value[rs.getFieldCount()];

            for (int fieldId=0; fieldId < rs.getFieldCount(); fieldId++ )

            {

                regAtt[fieldId] =  rs.getFieldValue(numReg, fieldId);

            }

            feat = new DefaultFeature(geom, regAtt, numReg + "");

        } catch (DriverIOException e) {

            throw new DriverException(e);

        } catch (DriverLoadException e) {

            throw new DriverException(e);

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

            throw new DriverException(e);

        }

        return feat;

    }

        public void stopEdition(IWriter writer) throws EditionException {

                super.stopEdition(writer);

                try {

                        ova.getDriver().reLoad();

                } catch (IOException e) {

                        e.printStackTrace();

                        throw new EditionException(e);

                }

        }

        public Rectangle2D getShapeBounds(int index) throws IOException {

                //Solo se utiliza cuando el driver es BoundedShapes

//                 Si no est? en el fichero de expansi?n

                Integer integer = new Integer((int) index);

                if (!relations.containsKey(integer)) {

                        if (ova.getDriver() instanceof BoundedShapes){

                                BoundedShapes bs = (BoundedShapes) ova.getDriver();

                                return bs.getShapeBounds(index);

                        }else{

                                return ova.getDriver().getShape(index).getBounds2D();

                        }

                } else {

                        int num = ((Integer) relations.get(integer)).intValue();

                        DefaultRowEdited feat;

                        feat = (DefaultRowEdited) expansionFile.getRow(num);

                        if (feat.getStatus() == IRowEdited.STATUS_DELETED)

                                return null;

                        IGeometry geom = ((IFeature)feat.getLinkedRow()).getGeometry();

                        return geom.getBounds2D();//getGeometry();

                }

        }

        public int getShapeType(int index) {

                try {

                        return ova.getShapeType();

                } catch (DriverIOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                return FShape.MULTI;

        }

}