| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2005 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.project.documents.gui;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.sql.Types;

import java.text.ParseException;

import java.util.HashMap;

import java.util.Iterator;

import javax.swing.JComponent;

import javax.swing.JOptionPane;

import org.cresques.cts.ICoordTrans;

import com.hardcode.driverManager.Driver;

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

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

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

import com.iver.andami.PluginServices;

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

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.edition.DefaultRowEdited;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.edition.IWriter;

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

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

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.rendering.VectorialLegend;

import com.iver.utiles.PostProcessSupport;

import com.iver.utiles.swing.threads.AbstractMonitorableTask;

/**

 * Task to create the annotation layer.

 *

 * @author Vicente Caballero Navarro

 */

public class Annotation_TaskCreate extends AbstractMonitorableTask {

    Annotation_Layer lyrVect;

    IWriter writer;

    int rowCount;

    ReadableVectorial va;

    SelectableDataSource sds;

    FBitSet bitSet;

    MapContext mapContext;

    VectorialDriver reader;

    private Annotation_Mapping mapping;

    private String duplicate;

    private HashMap mapGeom = new HashMap();

    private HashMap mapAttri = new HashMap();

    public Annotation_TaskCreate(MapContext mapContext, Annotation_Layer lyr,

        IWriter writer, Driver reader, String duplicate)

        throws DriverIOException, com.iver.cit.gvsig.fmap.DriverException {

        this.duplicate = duplicate;

        this.mapping = lyr.getAnnotatonMapping();

        this.mapContext = mapContext;

        this.lyrVect = lyr;

        this.writer = writer;

        this.reader = (VectorialDriver) reader;

        setInitialStep(0);

        setDeterminatedProcess(true);

        setStatusMessage(PluginServices.getText(this, "exporting_") + ": " +

            PluginServices.getText(this, "annotations"));

        va = lyrVect.getSource();

        sds = lyrVect.getRecordset();

        bitSet = sds.getSelection();

        if (bitSet.cardinality() == 0) {

            rowCount = va.getShapeCount();

        } else {

            rowCount = bitSet.cardinality();

        }

        setFinalStep(rowCount);

    }

    /**

     * DOCUMENT ME!

     *

     * @throws Exception DOCUMENT ME!

     */

    public void run() throws Exception {

        if (duplicate.equals(PluginServices.getText(this, "centered"))) {

            duplicate();

        }

        // va.start();

        ICoordTrans ct = lyrVect.getCoordTrans();

        DriverAttributes attr = va.getDriverAttributes();

        boolean bMustClone = false;

        if (attr != null) {

            if (attr.isLoadedInMemory()) {

                bMustClone = attr.isLoadedInMemory();

            }

        }

        SHPLayerDefinition lyrDef = (SHPLayerDefinition) writer.getTableDefinition();

        lyrDef.setShapeType(FShape.POINT);

        FieldDescription[] fields = getFieldDescriptions();

        lyrDef.setFieldsDesc(fields);

        writer.initialize(lyrDef);

        // Creamos la tabla.

        writer.preProcess();

        if (duplicate.equals(PluginServices.getText(this, "centered"))) {

            int pos = 0;

            Iterator iter = mapGeom.keySet().iterator();

            while (iter.hasNext()) {

                String textValue = (String) iter.next();

                Rectangle2D rectangle = (Rectangle2D) mapGeom.get(textValue);

                IGeometry geom = ShapeFactory.createPoint2D(rectangle.getCenterX(),

                        rectangle.getCenterY());

                if (ct != null) {

                    if (bMustClone) {

                        geom = geom.cloneGeometry();

                    }

                    geom.reProject(ct);

                }

                reportStep();

                setNote(PluginServices.getText(this, "exporting_") + " " + pos +

                    "    " + mapGeom.size());

                Value[] values = (Value[]) mapAttri.get(textValue);

                IFeature feat = new DefaultFeature(geom, values, "" + pos);

                DefaultRowEdited edRow = new DefaultRowEdited(feat,

                        DefaultRowEdited.STATUS_ADDED, pos);

                writer.process(edRow);

                pos++;

            }

        } else if (duplicate.equals(PluginServices.getText(this, "none"))) {

            va.start();

            if (bitSet.cardinality() == 0) {

                rowCount = va.getShapeCount();

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

                    IGeometry geom = va.getShape(i);

                    if (geom.getGeometryType() != FShape.POINT) {

                        Point2D p = FLabel.createLabelPoint((FShape) geom.getInternalShape());

                        geom = ShapeFactory.createPoint2D(p.getX(), p.getY());

                    }

                    if (ct != null) {

                        if (bMustClone) {

                            geom = geom.cloneGeometry();

                        }

                        geom.reProject(ct);

                    }

                    reportStep();

                    setNote(PluginServices.getText(this, "exporting_") + " " +

                        i + "    " + rowCount);

                    if (isCanceled()) {

                        break;

                    }

                    if (geom != null) {

                        Value[] valuesAnnotation = getValuesAnnotation(i);

                        IFeature feat = new DefaultFeature(geom,

                                valuesAnnotation, "" + i);

                        DefaultRowEdited edRow = new DefaultRowEdited(feat,

                                DefaultRowEdited.STATUS_ADDED, i);

                        writer.process(edRow);

                    }

                }

            } else {

                int counter = 0;

                for (int i = bitSet.nextSetBit(0); i >= 0;

                        i = bitSet.nextSetBit(i + 1)) {

                    IGeometry geom = va.getShape(i);

                    if (geom.getGeometryType() != FShape.POINT) {

                        Point2D p = FLabel.createLabelPoint((FShape) geom.getInternalShape());

                        geom = ShapeFactory.createPoint2D(p.getX(), p.getY());

                    }

                    if (ct != null) {

                        if (bMustClone) {

                            geom = geom.cloneGeometry();

                        }

                        geom.reProject(ct);

                    }

                    reportStep();

                    setNote(PluginServices.getText(this, "exporting_") + " " +

                        counter + "    " + bitSet.cardinality());

                    if (isCanceled()) {

                        break;

                    }

                    if (geom != null) {

                        Value[] valuesAnnotation = getValuesAnnotation(i);

                        IFeature feat = new DefaultFeature(geom,

                                valuesAnnotation, "" + i);

                        DefaultRowEdited edRow = new DefaultRowEdited(feat,

                                DefaultRowEdited.STATUS_ADDED, i);

                        writer.process(edRow);

                    }

                    va.stop();

                }

            }

        }

        writer.postProcess();

        //

        // if (bitSet.cardinality() == 0) {

        // rowCount = va.getShapeCount();

        //

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

        // IGeometry geom = va.getShape(i);

        //

        // if (geom.getGeometryType() != FShape.POINT) {

        // Point2D p = FLabel.createLabelPoint((FShape)

        // geom.getInternalShape());

        // geom = ShapeFactory.createPoint2D(p.getX(), p.getY());

        // }

        //

        // if (ct != null) {

        // if (bMustClone) {

        // geom = geom.cloneGeometry();

        // }

        //

        // geom.reProject(ct);

        // }

        //

        // reportStep();

        // setNote(PluginServices.getText(this, "exporting_") +" "+ i+ "

        // "+rowCount);

        //

        // if (isCanceled()) {

        // break;

        // }

        //

        // if (geom != null) {

        // Value[] valuesAnnotation = getValuesAnnotation(i);

        //

        // IFeature feat = new DefaultFeature(geom, valuesAnnotation,

        // "" + i);

        // DefaultRowEdited edRow = new DefaultRowEdited(feat,

        // DefaultRowEdited.STATUS_ADDED, i);

        // writer.process(edRow);

        // }

        // }

        // } else {

        // int counter = 0;

        //

        // for (int i = bitSet.nextSetBit(0); i >= 0;

        // i = bitSet.nextSetBit(i + 1)) {

        // IGeometry geom = va.getShape(i);

        //

        // if (geom.getGeometryType() != FShape.POINT) {

        // Point2D p = FLabel.createLabelPoint((FShape)

        // geom.getInternalShape());

        // geom = ShapeFactory.createPoint2D(p.getX(), p.getY());

        // }

        //

        // if (ct != null) {

        // if (bMustClone) {

        // geom = geom.cloneGeometry();

        // }

        //

        // geom.reProject(ct);

        // }

        //

        // reportStep();

        // setNote(PluginServices.getText(this, "exporting_") +" "+ counter+ "

        // "+bitSet.cardinality());

        //

        // if (isCanceled()) {

        // break;

        // }

        //

        // if (geom != null) {

        // Value[] valuesAnnotation = getValuesAnnotation(i);

        // IFeature feat = new DefaultFeature(geom, valuesAnnotation,

        // "" + i);

        // DefaultRowEdited edRow = new DefaultRowEdited(feat,

        // DefaultRowEdited.STATUS_ADDED, i);

        //

        // writer.process(edRow);

        // }

        // }

        // }

        //

        // writer.postProcess();

        // va.stop();

        mapAttri.clear();

        mapGeom.clear();

        if (reader != null) {

            int res = JOptionPane.showConfirmDialog((JComponent) PluginServices.getMDIManager()

                                                                               .getActiveWindow(),

                    PluginServices.getText(this,

                        "insertar_en_la_vista_la_capa_creada"),

                    PluginServices.getText(this, "insertar_capa"),

                    JOptionPane.YES_NO_OPTION);

            if (res == JOptionPane.YES_OPTION) {

                PostProcessSupport.executeCalls();

                lyrDef = (SHPLayerDefinition) writer.getTableDefinition();

                FLayer newLayer = LayerFactory.createLayer(lyrDef.getName(),

                        reader, mapContext.getProjection());

                Annotation_Layer la = new Annotation_Layer();

                LayerListener[] layerListeners = newLayer.getLayerListeners();

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

                    la.addLayerListener(layerListeners[i]);

                }

                la.setSource(((FLyrVect) newLayer).getSource());

                la.setProjection(newLayer.getProjection());

                la.setName(newLayer.getName());

                la.setLegend((VectorialLegend) lyrVect.getLegend());

                Annotation_Mapping.addAnnotationMapping(la);

                mapContext.getLayers().addLayer(la);

            }

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    private FieldDescription[] getFieldDescriptions() {

        SHPLayerDefinition lyrDef = (SHPLayerDefinition) writer.getTableDefinition();

        FieldDescription[] fields = new FieldDescription[Annotation_Mapping.NUMCOLUMNS]; // lyrDef.getFieldsDesc();

        int posText = mapping.getColumnText();

        int posTypeFont = mapping.getColumnTypeFont();

        int posStyleFont = mapping.getColumnStyleFont();

        int posColor = mapping.getColumnColor();

        int posHeight = mapping.getColumnHeight();

        int posRotate = mapping.getColumnRotate();

        FieldDescription[] fieldsDescriptions = lyrDef.getFieldsDesc();

        setField(Annotation_Mapping.TEXT, fieldsDescriptions, fields, posText, 0);

        setField(Annotation_Mapping.TYPEFONT, fieldsDescriptions, fields,

            posTypeFont, 1);

        setField(Annotation_Mapping.STYLEFONT, fieldsDescriptions, fields,

            posStyleFont, 2);

        setField(Annotation_Mapping.COLOR, fieldsDescriptions, fields,

            posColor, 3);

        setField(Annotation_Mapping.HEIGHT, fieldsDescriptions, fields,

            posHeight, 4);

        setField(Annotation_Mapping.ROTATE, fieldsDescriptions, fields,

            posRotate, 5);

        return fields;

    }

    /**

     * DOCUMENT ME!

     *

     * @param name DOCUMENT ME!

     * @param antFields DOCUMENT ME!

     * @param fields DOCUMENT ME!

     * @param pos DOCUMENT ME!

     * @param i DOCUMENT ME!

     */

    private void setField(String name, FieldDescription[] antFields,

        FieldDescription[] fields, int pos, int i) {

        int type = Annotation_Mapping.getType(name);

        if (pos != -1) {

            fields[i] = antFields[pos];

        } else {

            fields[i] = new FieldDescription();

            if (type != Types.VARCHAR) {

                if (name.equals("Color")) {

                    fields[i].setFieldLength(10);

                }

            }

            //                        else {

            //                                fields[i].setFieldLength(10);

            //                                fields[i].setFieldDecimalCount(3);

            //                        }

        }

        fields[i].setFieldType(type);

        fields[i].setFieldName(name);

    }

    /**

     * DOCUMENT ME!

     *

     * @param i DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws DriverException DOCUMENT ME!

     * @throws EditionException DOCUMENT ME!

     */

    private Value[] getValuesAnnotation(int i)

        throws DriverException, EditionException {

        int posText = mapping.getColumnText();

        int posTypeFont = mapping.getColumnTypeFont();

        int posStyleFont = mapping.getColumnStyleFont();

        int posColor = mapping.getColumnColor();

        int posHeight = mapping.getColumnHeight();

        int posRotate = mapping.getColumnRotate();

        Value[] values = sds.getRow(i);

        Value[] valuesAnnotation = new Value[Annotation_Mapping.NUMCOLUMNS];

        setValue(Annotation_Mapping.TEXT, values, valuesAnnotation, posText, 0);

        setValue(Annotation_Mapping.TYPEFONT, values, valuesAnnotation,

            posTypeFont, 1);

        setValue(Annotation_Mapping.STYLEFONT, values, valuesAnnotation,

            posStyleFont, 2);

        setValue(Annotation_Mapping.COLOR, values, valuesAnnotation, posColor, 3);

        setValue(Annotation_Mapping.HEIGHT, values, valuesAnnotation,

            posHeight, 4);

        setValue(Annotation_Mapping.ROTATE, values, valuesAnnotation,

            posRotate, 5);

        return valuesAnnotation;

    }

    /**

     * DOCUMENT ME!

     *

     * @param name DOCUMENT ME!

     * @param antValues DOCUMENT ME!

     * @param values DOCUMENT ME!

     * @param pos DOCUMENT ME!

     * @param i DOCUMENT ME!

     */

    private void setValue(String name, Value[] antValues, Value[] values,

        int pos, int i) {

        int type = Annotation_Mapping.getType(name);

        if (pos != -1) {

            try {

                values[i] = ValueFactory.createValueByType(antValues[pos].toString(),

                        type);

            } catch (ParseException e) {

                e.printStackTrace();

            }

        } else {

            // if (type == Types.VARCHAR) {

            // values[i] = ValueFactory.createNullValue();

            // } else {

            try {

                if (name == Annotation_Mapping.TEXT) {

                    values[i] = ValueFactory.createValueByType(String.valueOf(

                                Annotation_Mapping.DEFAULTTEXT), type);

                } else if (name == Annotation_Mapping.COLOR) {

                    values[i] = ValueFactory.createValueByType(String.valueOf(

                                Annotation_Mapping.DEFAULTCOLOR), type);

                } else if (name == Annotation_Mapping.HEIGHT) {

                    values[i] = ValueFactory.createValueByType(String.valueOf(

                                Annotation_Mapping.DEFAULTHEIGHT), type);

                } else if (name == Annotation_Mapping.TYPEFONT) {

                    values[i] = ValueFactory.createValueByType(String.valueOf(

                                Annotation_Mapping.DEFAULTTYPEFONT), type);

                } else if (name == Annotation_Mapping.STYLEFONT) {

                    values[i] = ValueFactory.createValueByType(String.valueOf(

                                Annotation_Mapping.DEFAULTSTYLEFONT), type);

                } else if (name == Annotation_Mapping.ROTATE) {

                    values[i] = ValueFactory.createValueByType(String.valueOf(

                                Annotation_Mapping.DEFAULTROTATE), type);

                }

            } catch (ParseException e) {

                e.printStackTrace();

            }

            // }

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @throws DriverIOException DOCUMENT ME!

     * @throws DriverException DOCUMENT ME!

     * @throws EditionException DOCUMENT ME!

     */

    private void duplicate()

        throws DriverIOException, DriverException, EditionException {

        va.start();

        if (bitSet.cardinality() == 0) {

            rowCount = va.getShapeCount();

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

                IGeometry geom = va.getShape(i);

                modifyMap(geom, i);

                // reportStep();

                // setNote(PluginServices.getText(this, "exporting_") +" "+ i+ "

                // "+rowCount);

                if (isCanceled()) {

                    break;

                }

            }

        } else {

            for (int i = bitSet.nextSetBit(0); i >= 0;

                    i = bitSet.nextSetBit(i + 1)) {

                IGeometry geom = va.getShape(i);

                modifyMap(geom, i);

                // reportStep();

                // setNote(PluginServices.getText(this, "exporting_") +" "+

                // counter+ " "+bitSet.cardinality());

                if (isCanceled()) {

                    break;

                }

            }

        }

        va.stop();

    }

    /**

     * DOCUMENT ME!

     *

     * @param geom DOCUMENT ME!

     * @param index DOCUMENT ME!

     *

     * @throws DriverException DOCUMENT ME!

     * @throws EditionException DOCUMENT ME!

     */

    private void modifyMap(IGeometry geom, int index)

        throws DriverException, EditionException {

        if (geom.getGeometryType() != FShape.POINT) {

            Point2D p = FLabel.createLabelPoint((FShape) geom.getInternalShape());

            geom = ShapeFactory.createPoint2D(p.getX(), p.getY());

        }

        if (geom != null) {

            Value[] valuesAnnotation = getValuesAnnotation(index);

            String textValue = valuesAnnotation[0].toString();

            if (!mapAttri.containsKey(textValue)) {

                mapAttri.put(textValue, valuesAnnotation);

            }

            if (mapGeom.containsKey(textValue)) {

                IGeometry geometry2 = geom;

                Rectangle2D rectangle1 = (Rectangle2D) mapGeom.get(textValue);

                // FPoint2D pointNew = new

                // FPoint2D((point1.getX()+point2.getX())/2,(point1.getY()+point2.getY())/2);

                mapGeom.put(textValue,

                    rectangle1.createUnion(geometry2.getBounds2D()));

                reportStep();

                setNote(PluginServices.getText(this, "exporting_"));

            } else {

                mapGeom.put(textValue, geom.getBounds2D());

            }

        }

    }

}