Application: gvSIG desktop

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

import java.awt.Image;

import java.io.IOException;

import java.util.HashMap;

import com.hardcode.driverManager.DriverLoadException;

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

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

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

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

import com.hardcode.gdbms.engine.data.edition.DataWare;

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

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

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

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

import com.iver.cit.gvsig.fmap.edition.commands.AddRowCommand;

import com.iver.cit.gvsig.fmap.edition.commands.Command;

import com.iver.cit.gvsig.fmap.edition.commands.CommandCollection;

import com.iver.cit.gvsig.fmap.edition.commands.CommandRecord;

import com.iver.cit.gvsig.fmap.edition.commands.MemoryCommandRecord;

import com.iver.cit.gvsig.fmap.edition.commands.ModifyRowCommand;

import com.iver.cit.gvsig.fmap.edition.commands.RemoveRowCommand;

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

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

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

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class EditableAdapter implements IEditableSource {

    protected boolean isEditing = false;

    private SelectableDataSource ds = null;

    protected FBitSet delRows = new FBitSet();

    private CommandRecord cr;

    private Image selectionImage;

    /**

     * Flag que indica que hay que tomar las siguientes operaciones como una

     * operaci?n at?mica

     */

    private boolean complex = false;

    private CommandCollection commands = null;

    /*

     * Establece una relaci?n entre los ?ndices de las geometr?as en el

     * EditableFeatureSource y los ?ndices en el fichero de expansi?n

     * FJP: CAMBIO: NECESITAMOS TRABAJAR CON FEATURE Y FEATUREITERATOR

     * PARA IR PREPARANDO EL CAMINO, GUARDAMOS EL FEATUREID (STRING)

     * COMO CLAVE, Y COMO VALOR, EL INDICE DENTRO DEL FICHERO

     * DE EXPANSION (Integer).

     * Lo de que FeatureId sea un String es por compatibilidad

     * con OGC. Seg?n OGC, una Feature tiene que tener un Id string

     * En el caso de los randomaccess, ser?n el id de registro

     * En los casos de base de datos espaciales, supongo

     * que siempre ser? num?rico tambi?n, pero lo tendremos

     * que convertir a string. Lo que est? claro es que

     * NO se puede confiar nunca en que sea algo correlativo

     * (1, 2, 3, 4, 5, ... => FALSO!!)

     */

    protected HashMap relations = new HashMap();

    /*

     * Fichero en el que se guardan las nuevas geometr?as, producto de adiciones

     * o de modificaciones

     */

    protected ExpansionFile expansionFile;

    protected int numAdd = 0;

    private ObjectDriver editingDriver = new myObjectDriver();

    private SelectableDataSource ods;

    /**

     * Crea un nuevo EditableAdapter.

     */

    public EditableAdapter() {

        expansionFile = new MemoryExpansionFile();

        cr = new MemoryCommandRecord();

    }

    /**

     * DOCUMENT ME!

     *

     * @param ds DOCUMENT ME!

     */

    public void setOriginalDataSource(SelectableDataSource ds) {

        this.ods = ds;

    }

    /**

     * DOCUMENT ME!

     *

     * @throws EditionException DOCUMENT ME!

     */

    public void startEdition() throws EditionException {

        isEditing = true;

    }

    /**

     * DOCUMENT ME!

     *

     * @param writer DOCUMENT ME!

     *

     * @throws EditionException DOCUMENT ME!

     */

    public void stopEdition(IWriter writer) throws EditionException {

        writer.preProcess();

        try {

            for (int i = 0; i < getRowCount(); i++) {

                IRowEdited rowEdited = getRow(i);

                if (rowEdited != null) {

                    writer.process(rowEdited);

                }

            }

        } catch (DriverIOException e) {

            e.printStackTrace();

            throw new EditionException(e);

        } catch (IOException e) {

            e.printStackTrace();

            throw new EditionException(e);

        }

        writer.postProcess();

        isEditing = false;

    }

    /**

     * DOCUMENT ME!

     *

     * @throws IOException DOCUMENT ME!

     */

    public void cancelEdition() throws IOException {

        isEditing = false;

        expansionFile.close();

    }

    /* (non-Javadoc)

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

     */

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

            int calculatedIndex=getCalculatedIndex(index);

            Integer integer = new Integer(calculatedIndex);

            DefaultRowEdited edRow = null;

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

        if (!relations.containsKey(integer)) {                

                try {

                    /* edRow = new DefaultRowEdited(new DefaultRow(ods.getRow(calculatedIndex), "" + index),

                            DefaultRowEdited.STATUS_ORIGINAL, index); */

                    edRow = new DefaultRowEdited(new DefaultRow(ods.getRow(calculatedIndex)),

                            DefaultRowEdited.STATUS_ORIGINAL, index);

                } catch (DriverException e) {

                    throw new DriverIOException(e);

                }

                return edRow;

        } else {

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

            // return expansionFile.getRow(num);

            IRowEdited rowFromExpansion = expansionFile.getRow(num); 

            // ?Habr?a que hacer aqu? setID(index + "")?

            edRow = new DefaultRowEdited(rowFromExpansion.getLinkedRow().cloneRow(), rowFromExpansion.getStatus(), index);

            return edRow; 

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws DriverIOException DOCUMENT ME!

     * @throws IOException DOCUMENT ME!

     */

    public int getRowCount() throws DriverIOException, IOException {

             try {

             return (int) (ods.getRowCount() + numAdd) - delRows.cardinality();// - expansionFile.getInvalidRows().cardinality();

         } catch (DriverException e) {

                 throw new DriverIOException(e);

         }

    }

    /**

     * DOCUMENT ME!

     *

     * @param row DOCUMENT ME!

     *

     * @throws DriverIOException DOCUMENT ME!

     * @throws IOException DOCUMENT ME!

     */

    public void addRow(IRow row,String descrip) throws DriverIOException, IOException {

        int calculatedIndex = doAddRow(row);

        Command command=new AddRowCommand(this, row, calculatedIndex);

        command.setDescription(descrip);

        if (complex) {

            commands.add(command);

        } else {

            cr.pushCommand(command);

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @throws DriverIOException DOCUMENT ME!

     * @throws IOException DOCUMENT ME!

     */

    public void undo() throws DriverIOException, IOException {

        //seleccion.clear();

        if (moreUndoCommands()) {

            cr.undoCommand();

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @throws DriverIOException DOCUMENT ME!

     * @throws IOException DOCUMENT ME!

     */

    public void redo() throws DriverIOException, IOException {

        //seleccion.clear();

        if (moreRedoCommands()) {

            cr.redoCommand();

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean moreUndoCommands() {

        return cr.moreUndoCommands();

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean moreRedoCommands() {

        return cr.moreRedoCommands();

    }

    /* (non-Javadoc)

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

     */

    public void removeRow(int index,String descrip) throws IOException, DriverIOException {

            int calculatedIndex = getCalculatedIndex(index);

            Command command=new RemoveRowCommand(this, calculatedIndex);

            command.setDescription(descrip);

            if (complex) {

            commands.add(command);

        } else {

            cr.pushCommand(command);

        }

        doRemoveRow(calculatedIndex);

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.edition.IEditableSource#modifyRow(int, com.iver.cit.gvsig.fmap.core.IRow)

     */

    public void modifyRow(int index, IRow row,String descrip)

        throws IOException, DriverIOException {

            int calculatedIndex = getCalculatedIndex(index);

        int pos = doModifyRow(calculatedIndex, row);

        Command command=new ModifyRowCommand(this, calculatedIndex, pos, row);

        command.setDescription(descrip);

        if (complex) {

            commands.add(command);

        } else {

            cr.pushCommand(command);

        }

    }

    /**

     * DOCUMENT ME!

     */

    public void compact() {

        expansionFile.compact(relations);

    }

    /**

     * DOCUMENT ME!

     *

     * @param i DOCUMENT ME!

     */

    public void setImage(Image i) {

        selectionImage = i;

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public Image getImage() {

        return selectionImage;

    }

    /**

     * DOCUMENT ME!

     */

    public void startComplexRow() {

        complex = true;

        commands = new CommandCollection();

    }

    /**

     * DOCUMENT ME!

     *

     * @throws IOException DOCUMENT ME!

     * @throws DriverIOException DOCUMENT ME!

     */

    public void endComplexRow() throws IOException, DriverIOException {

        cr.pushCommand(commands);

        complex = false;

    }

    /**

     * 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 {

        if (previousExpansionFileIndex == -1) {

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

            relations.remove(new Integer(geometryIndex));

            expansionFile.deleteLastRow();

        } else {

            //Se actualiza la relaci?n de ?ndices

            relations.put(new Integer(geometryIndex),

                new Integer(previousExpansionFileIndex));

        }

    }

    /**

     * 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 IRow doRemoveRow(int index) throws DriverIOException, IOException {

//Llega un calculatedIndex

            delRows.set(index, true);

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

            // TODO: Con tablas no es necesario devolver la anterior feature. Por ahora.

            return null;

    }

    /**

     * 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);

        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));

        } else {

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

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

            pos = num;

            /*

             * 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));

        }

        return pos;

    }

    /**

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

     * en la tabla relations.

     *

     * @param feat geometr?a a guardar.

     *

     * @return calculatedIndex

     *

     * @throws DriverIOException

     * @throws IOException

     */

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

        // A?ade la geometr?a

        // int virtualIndex = 0;

        int calculatedIndex = -1;

        try {

            calculatedIndex = (int) ods.getRowCount() + numAdd;

                // int externalIndex = getRowCount();

                // calculatedIndex = getCalculatedIndex(externalIndex);

        } catch (DriverException e) {

                throw new DriverIOException(e);

        } 

        int pos = expansionFile.addRow(feat);

        relations.put(new Integer(calculatedIndex), new Integer(pos));

        numAdd++;

        System.err.println("A?ade una Row en la posici?n: " + calculatedIndex);

        return calculatedIndex;

    }

    /**

     * 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 {

        delRows.set(index, false);

    }

    /**

     * 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 calculatedIndex) throws DriverIOException, IOException {

        expansionFile.deleteLastRow();

        relations.remove(new Integer(calculatedIndex));

        numAdd--;

    }

    /*

     * (non-Javadoc)

     *

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

     */

    public SelectableDataSource getRecordset() throws DriverLoadException {

        if (isEditing) {

            if (ds == null) {

                String name = LayerFactory.getDataSourceFactory().addDataSource((ObjectDriver) editingDriver);

                try {

                    ds = new SelectableDataSource(LayerFactory.getDataSourceFactory()

                                                              .createRandomDataSource(name,

                                DataSourceFactory.AUTOMATIC_OPENING));

                } catch (NoSuchTableException e) {

                    throw new RuntimeException(e);

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

                    throw new RuntimeException(e);

                }

            }

            return ds;

        } else {

            return ods;

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @param sds DOCUMENT ME!

     */

    public void setRecordSet(SelectableDataSource sds) {

        this.ods = sds;

    }

    /**

     * DOCUMENT ME!

     *

     * @return

     */

    public FBitSet getSelection() {

        /* try {

                        return getRecordset().getSelection();

                } catch (DriverLoadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } 

                return null; */

            return getRecordset().getSelection();

    }

    public void setSelection(FBitSet selection) {

            /* try {

                        getRecordset().setSelection(selection);

                } catch (DriverLoadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } */

            getRecordset().setSelection(selection);

        }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean isEditing() {

        return isEditing;

    }

    protected int getInversedIndex(long rowIndex) {

            int intervalNotDeleted=0;

        int antDeleted=-1;

        int idPedido = (int) rowIndex;

        int numNotDeleted=0;

        int numBorradosAnt = 0;

            for (int i = delRows.nextSetBit(0); i>=0; i = delRows.nextSetBit(i+1)) {

                    intervalNotDeleted = i-antDeleted-1;

                    numNotDeleted += intervalNotDeleted;

                    if (i > idPedido){

                            numNotDeleted = numNotDeleted + (i - idPedido);

                            break;

                    }

                    numBorradosAnt++;

                    antDeleted = i;

        }

                    numNotDeleted =  idPedido-numBorradosAnt;

                //System.out.println("Piden Viejo : "+ rowIndex + " y devuelvo como nuevo " + (numNotDeleted));

        return numNotDeleted;

    }

    /**

     * DOCUMENT ME!

     *

     * @param rowIndex DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    protected int getCalculatedIndex(long rowIndex) {

        int numNotDeleted=0;

            int intervalNotDeleted=0;

        int antDeleted=-1;

        int calculatedIndex;

        int idPedido = (int) rowIndex;

        int numBorradosAnt = 0;

            for (int i = delRows.nextSetBit(0); i>=0; i = delRows.nextSetBit(i+1)) {

                    intervalNotDeleted = i-antDeleted -1 ;

                    numNotDeleted += intervalNotDeleted;

                    if (numNotDeleted > idPedido)

                    {

                            break;

                    }

                    numBorradosAnt++;

                    antDeleted = i;

        }

             calculatedIndex = numBorradosAnt + idPedido;

            //System.out.println("Piden Registro : "+ rowIndex + " y devuelvo el " + (calculatedIndex));

        return calculatedIndex;

    }

    /**

     * DOCUMENT ME!

     *

     * @author Vicente Caballero Navarro

     */

    private class myObjectDriver implements ObjectDriver {

        /*

         * (non-Javadoc)

         *

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

         */

        public int[] getPrimaryKeys() throws DriverException {

                return ods.getPrimaryKeys();

        //        int[] pk=new int[1];

                /*for (int i=0;i<getRowCount();i++){

                        pk[i]=i;

                }*/

        //        pk[0]=1;

        //    return pk;

        }

        /*

         * (non-Javadoc)

         *

         * @see com.hardcode.gdbms.engine.data.driver.ObjectDriver#write(com.hardcode.gdbms.engine.data.edition.DataWare)

         */

        public void write(DataWare dataWare) throws DriverException {

                DataWare dataWareOrig = ods.getDataWare(DataSourceFactory.DATA_WARE_DIRECT_MODE);

                dataWareOrig.commitTrans();

        }

        /*

         * (non-Javadoc)

         *

         * @see com.hardcode.gdbms.engine.data.driver.GDBMSDriver#setDataSourceFactory(com.hardcode.gdbms.engine.data.DataSourceFactory)

         */

        public void setDataSourceFactory(DataSourceFactory dsf) {

           ods.setDataSourceFactory(dsf);

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public String getName() {

           return ods.getName();

        }

        /*

         * (non-Javadoc)

         *

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

         *      int)

         */

        public Value getFieldValue(long rowIndex, int fieldId)

            throws DriverException {

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

            Integer integer = new Integer(getCalculatedIndex(rowIndex));

            try {

                if (!relations.containsKey(integer)) {

                    return ods.getFieldValue(rowIndex, fieldId);

                } else {

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

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

                    if (feat == null) {

                        return null;

                    }

                    return feat.getAttribute(fieldId);

                }

            } catch (DriverException e) {

                e.printStackTrace();

                throw new DriverException(e);

            } catch (IOException e) {

                e.printStackTrace();

                throw new DriverException(e);

            }

            /**

             * try { if (!relations.containsKey(integer)) { // Si ha sido

             * eliminada if (delRows.get(integer.intValue())) { return null; }

             * else { return ods.getFieldValue(rowIndex, fieldId); }} else {

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

             * DefaultRowEdited feat = (DefaultRowEdited)

             * expansionFile.getRow(num); if (feat==null)return null; return

             * feat.getAttribute(fieldId); }} catch (DriverException e) {

             * e.printStackTrace(); throw new DriverException(e); } catch

             * (IOException e) { e.printStackTrace(); throw new

             * DriverException(e); }

             */

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public int getFieldCount() throws DriverException {

            // TODO Por ahora, no dejamos que se a?adan campos

            return ods.getFieldCount();

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public String getFieldName(int fieldId) throws DriverException {

            return ods.getFieldName(fieldId);

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public long getRowCount() {

                 try {

                 return (int) (ods.getRowCount() + numAdd) - delRows.cardinality();// - expansionFile.getInvalidRows().cardinality();

             } catch (DriverException e) {

                 // TODO Auto-generated catch block

                 e.printStackTrace();

             }

             return 0;

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public int getFieldType(int i) throws DriverException {

            return ods.getFieldType(i);

        }

    }

    public CommandRecord getCommandRecord(){

            return cr;

    }

}