Application: gvSIG desktop

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

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

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

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

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

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.core.ShapeFactory;

import com.iver.cit.gvsig.fmap.core.v02.FConverter;

import com.iver.cit.gvsig.fmap.core.v02.FLabel;

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

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

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

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

import com.vividsolutions.jts.geom.Envelope;

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

public class AnnotationEditableAdapter extends VectorialEditableAdapter{

        private ArrayList labels;

        private MappingAnnotation mapping;

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

                return super.doAddRow(feat);

        }

        public int doModifyRow(int calculatedIndex, IRow feat) throws IOException, DriverIOException {

                FLabel label=(FLabel)labels.get(calculatedIndex);

                Value value=feat.getAttribute(mapping.getColumnText());

                Rectangle2D rLabelAnt=(Rectangle2D)label.getBoundBox().clone();

                Rectangle2D rLabel=(Rectangle2D)label.getBoundBox();

                label.setString(value.toString());

                if (mapping.getColumnRotate()==mapping.getColumnText()){

                        label.setRotation(((NumericValue)value).doubleValue());

                } //Aqu? hay que comprobar, si se quiere, el resto de columnas, si son iguales a la de Texto.

                int posAnteriorInExpansionFile = -1;

                Integer integer = new Integer(calculatedIndex);

                IFeature featAnt = null;

                System.err.println("Modifica una Row en la posici?n: "

                                + calculatedIndex);

                // Si la geometr?a no ha sido modificada

                if (!relations.containsKey(integer)) {

                        int newPosition = expansionFile.addRow(feat);

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

                        // Se actualiza el ?ndice espacial

                        try {

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

                        } catch (DriverException e) {

                                throw new DriverIOException(e);

                        }

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

                        Rectangle2D rAnt=label.getBoundBox();

                        rAnt.setFrame(rAntAux.getX(),rAntAux.getY(),rAnt.getWidth(),rAnt.getHeight());

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

                        double difX=r.getX()-rAnt.getX();

                        double difY=r.getY()-rAnt.getY();

                        Point2D p=new Point2D.Double(rLabel.getX()+difX,rLabel.getY()+difY);

                        label.setBoundBox(new Rectangle2D.Double(p.getX(),p.getY(),rLabel.getWidth(),rLabel.getHeight()));

                        label.setOrig(p);

                        index.remove(new Envelope(rLabelAnt.getX(), rLabelAnt.getX()

                                        + rLabelAnt.getWidth(), rLabelAnt.getY(), rLabelAnt.getY()

                                        + rLabelAnt.getHeight()), new Integer(calculatedIndex));

                        index.insert(new Envelope(rLabel.getX(), rLabel.getX() + rLabel.getWidth(), rLabel

                                        .getY(), rLabel.getY() + rLabel.getHeight()), new Integer(

                                        calculatedIndex));

                } else {

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

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

                        posAnteriorInExpansionFile = 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();

                        double difX=r.getX()-rAnt.getX();

                        double difY=r.getY()-rAnt.getY();

                        Point2D p=new Point2D.Double(rLabel.getX()+difX,rLabel.getY()+difY);

                        label.setBoundBox(new Rectangle2D.Double(p.getX(),p.getY(),rLabel.getWidth(),rLabel.getHeight()));

                        label.setOrig(p);

                        index.remove(new Envelope(rLabelAnt.getX(), rLabelAnt.getX()

                                        + rLabelAnt.getWidth(), rLabelAnt.getY(), rLabelAnt.getY()

                                        + rLabelAnt.getHeight()), new Integer(calculatedIndex));

                        index.insert(new Envelope(rLabel.getX(), rLabel.getX() + rLabel.getWidth(), rLabel

                                        .getY(), rLabel.getY() + rLabel.getHeight()), new Integer(

                                        calculatedIndex));

                }

                return posAnteriorInExpansionFile;

        }

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

                return super.doRemoveRow(index);

        }

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

                super.undoAddRow(calculatedIndex);

        }

        public void undoModifyRow(int calculatedIndex, int previousExpansionFileIndex) throws IOException, DriverIOException {

                super.undoModifyRow(calculatedIndex, previousExpansionFileIndex);

                Value value=this.getRow(calculatedIndex).getAttribute(mapping.getColumnText());

                ((FLabel)labels.get(calculatedIndex)).setString(value.toString());

                if (mapping.getColumnRotate()==mapping.getColumnText()){

                        ((FLabel)labels.get(calculatedIndex)).setRotation(((NumericValue)value).doubleValue());

                }

        }

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

                super.undoRemoveRow(index);

        }

        public AnnotationEditableAdapter(ArrayList labels,MappingAnnotation mapping) {

                super();

                this.labels=labels;

                this.mapping=mapping;

        }

        public IRowEdited[] getFeatures(Rectangle2D r, String strEPSG) throws DriverException {

                // En esta clase suponemos random access.

                // Luego tendremos otra clase que sea VectorialEditableDBAdapter

                // que reescribir? este m?todo.

        Envelope e = FConverter.convertRectangle2DtoEnvelope(r);

        List l = index.query(e);

        IRowEdited[] feats = new IRowEdited[l.size()];

        try {

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

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

                    int inverse = getInversedIndex(i.intValue());

                                feats[index] = (IRowEdited) getRow(inverse);

                }

                } catch (DriverIOException e1) {

                        throw new DriverException(e1);

                } catch (IOException e1) {

                        throw new DriverException(e1);

                }

                return feats;

        }

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

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

            DefaultRowEdited edRow=null;

        if (!relations.containsKey(integer)) {

                        try {

                                IFeature f=ova.getFeature(calculatedIndex);

                                //IGeometry g=f.getGeometry().cloneGeometry();

                                IGeometry geom=getGeometry(((FLabel)labels.get(index)).getBoundBox());

                                f=new DefaultFeature(geom,f.getAttributes());

                                edRow = new DefaultRowEdited(f,

                                        DefaultRowEdited.STATUS_ORIGINAL, index);

                                System.out.println("Piden la feature con ID= " + index);

                                // Exception e = new Exception();

                                // e.printStackTrace();

                        } catch (DriverException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

            return edRow;

        } else {

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

            IRowEdited aux = expansionFile.getRow(num);

            IFeature f=(IFeature)aux.getLinkedRow().cloneRow();

            IGeometry geom=getGeometry(((FLabel)labels.get(index)).getBoundBox());

            f=new DefaultFeature(geom,f.getAttributes());

            edRow = new DefaultRowEdited(f, aux.getStatus(), index);

            return edRow;

        }

    }

    private IGeometry getGeometry(Rectangle2D r){

            GeneralPathX resul = new GeneralPathX();

                Point2D[] vs=new Point2D[5];

                vs[0]=new Point2D.Double(r.getX(),r.getY());

            vs[1]=new Point2D.Double(r.getMaxX(),r.getY());

            vs[2]=new Point2D.Double(r.getMaxX(),r.getMaxY());

            vs[3]=new Point2D.Double(r.getX(),r.getMaxY());

            vs[4]=new Point2D.Double(r.getX(),r.getY());

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

                        if (i == 0) {

                                resul.moveTo(vs[i].getX(),vs[i].getY());

                        } else {

                                resul.lineTo(vs[i].getX(),vs[i].getY());

                        }

                }

                return ShapeFactory.createPolygon2D(resul);

    }

    public void startEdition() throws EditionException {

            isEditing = true;

                try {

                        expansionFile.open();

                                index = new Quadtree();

                                for (int i = 0; i < labels.size(); i++) {

                                        Rectangle2D r=((FLabel)labels.get(i)).getBoundBox();

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

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

                                                                        + r.getHeight());

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

                                        if (fullExtent == null) {

                                                fullExtent = r;

                                        } else {

                                                fullExtent = fullExtent.createUnion(r);

                                        }

                                }

                } catch (IOException e) {

                        throw new EditionException(e);

                }

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

                                + index.queryAll().size() + " labels");

        }

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

                }

        }

}