| 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

 */

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

 */

/* 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 org.gvsig.data.datastores.vectorial.db.sde;

import java.awt.geom.PathIterator;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import java.util.Date;

import java.util.Iterator;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datastores.vectorial.ISelectiveWriter;

import org.gvsig.data.datastores.vectorial.db.DBAttributeDescriptor;

import org.gvsig.data.datastores.vectorial.db.DBFeatureType;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.DataException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.InitializeWriterException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.exception.WriteException;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureAttributeDescriptor;

import org.gvsig.data.vectorial.IFeatureStore;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.INewFeatureStoreParameters;

import com.esri.sde.sdk.client.SDEPoint;

import com.esri.sde.sdk.client.SeColumnDefinition;

import com.esri.sde.sdk.client.SeConnection;

import com.esri.sde.sdk.client.SeCoordinateReference;

import com.esri.sde.sdk.client.SeDelete;

import com.esri.sde.sdk.client.SeException;

import com.esri.sde.sdk.client.SeInsert;

import com.esri.sde.sdk.client.SeLayer;

import com.esri.sde.sdk.client.SeRegisteredColumn;

import com.esri.sde.sdk.client.SeRow;

import com.esri.sde.sdk.client.SeShape;

import com.esri.sde.sdk.client.SeTable;

import com.esri.sde.sdk.client.SeUpdate;

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

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

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

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

class SDEFeaturesWriter implements ISelectiveWriter {

    DBFeatureType featureType;

    boolean bCreateTable = false;

    SDEStoreParameters parameters;

    protected SeConnection conex;

    protected SDEStore store;

    protected boolean previousAutocommit;

    public void postProcess() throws OpenException, WriteException {

        try {

            conex.commitTransaction();

        } catch (SeException e) {

            throw new WriteException(this.store.getName(),e);

        }

    }

    public void cancelProcess() throws WriteException {

        try {

            conex.rollbackTransaction();

        } catch (SeException e) {

            throw new WriteException(this.store.getName(),e);

        }

    }

    SDEFeaturesWriter() {

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.IFeaturesWriter#init(org.gvsig.data.vectorial.IFeatureStore)

     */

    public void init(IFeatureStore store) throws InitializeWriterException {

        this.store = (SDEStore)store;

        try {

                        this.conex = this.store.getWriterConnection();

                } catch (ReadException e) {

                        throw new InitializeWriterException(this.getClass().getName(),e);

                }

        SDEStore sdeStore = (SDEStore) store;

        this.parameters = sdeStore.getParametersSDE();

        this.featureType = (DBFeatureType) this.store.getDefaultFeatureType();

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.IFeaturesWriter#preProcess()

     */

    public void preProcess() throws WriteException, ReadException {

        try {

            conex.startTransaction();

        } catch (SeException e) {

            throw new WriteException(this.store.getName(),e);

        }

        if (bCreateTable) {

            SDEExplorer explorer = (SDEExplorer) this.store.getExplorer();

            IDataStoreParameters params = this.store.getParameters();

            explorer.remove(params, conex);

            INewFeatureStoreParameters newParam = (INewFeatureStoreParameters) explorer.createNewDataStoreParameter();

            newParam.setFeatureType(this.featureType);

            explorer.add(newParam, conex);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.ISelectiveWriter#deleteFeature(org.gvsig.data.vectorial.IFeature)

     */

    public void deleteFeature(IFeature feature) throws WriteException {

        try {

            SeDelete delete = new SeDelete(conex);

            String id = getFliterForIDForPStatement(featureType);

            String where = id + " = '" + feature.get(id) + "'";

            delete.fromTable(featureType.getTableID(),where);

            delete.close();

        } catch (SeException e) {

            throw new WriteException(this.store.getName(), e);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.IFeaturesWriter#insertFeature(org.gvsig.data.vectorial.IFeature)

     */

    public void insertFeature(IFeature feature) throws WriteException {

        try {

            SeLayer lay = null;

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

                SeLayer selay = (SeLayer) conex.getLayers().elementAt(i);

                if (featureType.getTableID().toLowerCase().endsWith(selay.getName().toLowerCase())) {

                    lay = (SeLayer) conex.getLayers().elementAt(i);

                    break;

                }

            }

            SeInsert insert = new SeInsert(conex);

            String[] names=featureType.getFields();

            String[] newNames=new String[names.length-1];

            int j=0;

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

                    if (!names[i].equals(featureType.getFieldsId()[0])){

                            newNames[j]=names[i];

                            j++;

                    }

            }

            insert.intoTable(lay.getName(),newNames);

            insert.setWriteMode(true);

            SeRow row = insert.getRowToSet();

            Iterator iterator = featureType.iterator();

            int jObjectID=0;

            while (iterator.hasNext()) {

                IFeatureAttributeDescriptor fad = (IFeatureAttributeDescriptor) iterator.next();

                if (fad.getName().equals(featureType.getFieldsId()[0])){

                    jObjectID++;

                    continue;

                }

                Object value=feature.get(fad.ordinal());

                if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_GEOMETRY) {

                    IGeometry geom = (IGeometry) feature.get(fad.ordinal());

                    writeGeometry(geom, row, lay, fad.ordinal()-jObjectID);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_STRING) {

                    if (value==null){

                        value="";

                    }

                    row.setString(fad.ordinal()-jObjectID,

                        (String)value);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_INT) {

                    if (value==null){

                        value=new Integer(0);

                    }

                    row.setInteger(fad.ordinal()-jObjectID,

                        (Integer)value);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_LONG) {

                    if (value==null){

                        value=new Long(0);

                    }

                    row.setLong(fad.ordinal()-jObjectID, (Long) value);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_FLOAT) {

                    if (value==null){

                        value=new Float(0);

                    }

                    row.setFloat(fad.ordinal()-jObjectID,

                        (Float) value);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_DOUBLE) {

                    if (value==null){

                        value=new Double(0);

                    }

                    row.setDouble(fad.ordinal()-jObjectID,

                        (Double) value);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_DATE) {

                    if (value==null){

                        value=new Date();

                    }

                    row.setDate(fad.ordinal()-jObjectID, (Date) value);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_BOOLEAN) {

                    boolean b = false;

                    if (value==null){

                        b=false;

                    }else{

                    b = feature.getBoolean(fad.ordinal());

                    }

                    if (b) {

                        row.setInteger(fad.ordinal()-jObjectID, new Integer(1));

                    } else {

                        row.setInteger(fad.ordinal()-jObjectID, new Integer(0));

                    }

                }

            }

            insert.execute();

            insert.close();

        } catch (SeException e) {

            throw new WriteException(this.getClass().getName(), e);

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @param geom DOCUMENT ME!

     * @param row DOCUMENT ME!

     * @param lay DOCUMENT ME!

     * @param ordinal DOCUMENT ME!

     *

     * @throws SeException DOCUMENT ME!

     */

    private void writeGeometry(IGeometry geom, SeRow row, SeLayer lay,

        int ordinal) throws SeException {

        SeCoordinateReference scr = lay.getCoordRef();

        if (geom==null || geom.getGeometryType()==0){

                SeShape exp = new SeShape(scr);

                exp.makeNil();

                row.setShape(ordinal,exp);

        }else if (geom.getGeometryType() == FShape.POINT) {

            SeShape exp = new SeShape(scr);

            SDEPoint[] vec = new SDEPoint[1];

            FPoint2D point = (FPoint2D) geom.getInternalShape();

            SDEPoint sdv = new SDEPoint(point.getX(), point.getY());

            vec[0] = sdv;

            exp.generatePoint(1, vec);

            row.setShape(ordinal, exp);

        } else if ((geom.getGeometryType() == FShape.LINE) ||

                (geom.getGeometryType() == FShape.POLYGON)) {

            ArrayList points = null;

            ArrayList partsOffset = null;

            PathIterator theIterator = geom.getPathIterator(null);

            int theType;

            int numParts = 0;

            //                 Use this array to store segment coordinate data

            double[] theData = new double[6];

            while (!theIterator.isDone()) {

                //while not done

                theType = theIterator.currentSegment(theData);

                //Populate a segment of the new

                // GeneralPathX object.

                //Process the current segment to populate a new

                // segment of the new GeneralPathX object.

                switch (theType) {

                case PathIterator.SEG_MOVETO:

                    if (points == null) {

                        points = new ArrayList();

                    }

                    partsOffset.add(new Integer(points.size()));

                    numParts++;

                    points.add(new Point2D.Double(theData[0], theData[1]));

                    break;

                case PathIterator.SEG_LINETO:

                    points.add(new Point2D.Double(theData[0], theData[1]));

                    break;

                case PathIterator.SEG_QUADTO:

                    System.out.println("Not supported here");

                    break;

                case PathIterator.SEG_CUBICTO:

                    System.out.println("Not supported here");

                    break;

                case PathIterator.SEG_CLOSE:

                    // A?adimos el primer punto para cerrar.

                    Point2D firstPoint = (Point2D) points.get(0);

                    points.add(new Point2D.Double(firstPoint.getX(),

                            firstPoint.getY()));

                    break;

                } //end switch

                theIterator.next();

            } //end while loop

            SDEPoint[] vec = new SDEPoint[points.size()];

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

                Point2D p = (Point2D) points.get(i);

                vec[i] = new SDEPoint(p.getX(), p.getY());

            }

            int[] offsetParts = new int[partsOffset.size()];

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

                offsetParts[i] = ((Integer) partsOffset.get(i)).intValue();

            }

            SeShape exp = new SeShape(scr);

            if (geom.getGeometryType() == FShape.LINE) {

                exp.generateLine(vec.length, offsetParts.length, offsetParts,

                    vec);

            } else if (geom.getGeometryType() == FShape.POLYGON) {

                exp.generatePolygon(vec.length, offsetParts.length,

                    offsetParts, vec);

            }

            row.setShape(ordinal, exp);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.IFeaturesWriter#updateFeatureType(org.gvsig.data.vectorial.IFeatureType)

     */

    public void updateFeatureType(IFeatureType featureType) {

        this.featureType = (DBFeatureType) featureType;

    }

    /**

     * DOCUMENT ME!

     *

     * @param createTable The bCreateTable to set.

     */

    public void setCreateTable(boolean createTable) {

        bCreateTable = createTable;

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.ISelectiveWriter#updateFeature(org.gvsig.data.vectorial.IFeature, org.gvsig.data.vectorial.IFeature)

     */

    public void updateFeature(IFeature oldFeature, IFeature feature)

        throws WriteException, ReadException {

        try {

            SeLayer lay = null;

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

                SeLayer selay = (SeLayer) conex.getLayers().elementAt(i);

                if (featureType.getTableID().toLowerCase().endsWith(selay.getName().toLowerCase())) {

                    lay = (SeLayer) conex.getLayers().elementAt(i);

                }

            }

            String[] names=featureType.getFields();

            String[] newNames=new String[names.length-1];

            int j=0;

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

                    if (!names[i].equals(featureType.getFieldsId()[0])){

                            newNames[j]=names[i];

                            j++;

                    }

            }

            SeUpdate update = new SeUpdate(conex);

            String id = getFliterForIDForPStatement(featureType);

            String where = id + " = '" + feature.get(id) + "'";

            update.toTable(featureType.getTableID(),

                newNames, where);

            update.setWriteMode(true);

            SeRow row = update.getRowToSet();

            Iterator iterator = featureType.iterator();

            int jObjectID=0;

            while (iterator.hasNext()) {

                IFeatureAttributeDescriptor fad = (IFeatureAttributeDescriptor) iterator.next();

                if (fad.getName().equals(featureType.getFieldsId()[0])){

                    jObjectID++;

                    continue;

                }

                Object value=feature.get(fad.ordinal());

                if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_GEOMETRY) {

                    IGeometry geom = (IGeometry) feature.get(fad.ordinal());

                    writeGeometry(geom, row, lay, fad.ordinal()-jObjectID);

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_STRING) {

                    row.setString(fad.ordinal()-jObjectID,

                        feature.getString(fad.ordinal()));

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_INT) {

                    row.setInteger(fad.ordinal()-jObjectID,

                        (Integer) feature.get(fad.ordinal()));

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_LONG) {

                    row.setLong(fad.ordinal()-jObjectID, (Long) feature.get(fad.ordinal()));

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_FLOAT) {

                    row.setFloat(fad.ordinal()-jObjectID,

                        (Float) feature.get(fad.ordinal()));

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_DOUBLE) {

                    row.setDouble(fad.ordinal()-jObjectID,

                        (Double) feature.get(fad.ordinal()));

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_DATE) {

                    row.setDate(fad.ordinal()-jObjectID, (Date) feature.get(fad.ordinal()));

                } else if (fad.getDataType() == IFeatureAttributeDescriptor.TYPE_BOOLEAN) {

                        boolean b = false;

                    if (value==null){

                        b=false;

                    }else{

                    b = feature.getBoolean(fad.ordinal());

                    }

                    if (b) {

                        row.setInteger(fad.ordinal()-jObjectID, new Integer(1));

                    } else {

                        row.setInteger(fad.ordinal()-jObjectID, new Integer(0));

                    }

                }

            }

            update.execute();

            update.close();

        } catch (SeException e) {

           throw new WriteException(this.getClass().getName(),e);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.ISelectiveWriter#deleteAttribute(org.gvsig.data.vectorial.IFeatureAttributeDescriptor)

     */

    public void deleteAttribute(IFeatureAttributeDescriptor attribute)

        throws WriteException {

        try {

            SeTable table = new SeTable(conex,featureType.getTableID());

            table.dropColumn(attribute.getName());

        } catch (SeException e) {

            throw new WriteException(this.store.getName(),e);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.ISelectiveWriter#updateAttribute(org.gvsig.data.vectorial.IFeatureAttributeDescriptor, org.gvsig.data.vectorial.IFeatureAttributeDescriptor)

     */

    public void updateAttribute(IFeatureAttributeDescriptor oldAttribute,

        IFeatureAttributeDescriptor attribute)

        throws WriteException, ReadException {

        try {

            SeTable table = new SeTable(conex,featureType.getTableID());

            SeRegisteredColumn column=null;//new SeRegisteredColumn(conex);

            SeRegisteredColumn[] columns=table.getColumnList();

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

                if(columns[i].getName().equals(oldAttribute.getName())){

                    column=columns[i];

                }

            }

            DBAttributeDescriptor dbAttribute=(DBAttributeDescriptor)attribute;

            column.setAllowsNulls(dbAttribute.isAllowNull());

            column.setDecimalDigits(dbAttribute.getPrecision());

            column.setSize(dbAttribute.getSize());

            column.setType(SDETypes.fieldTypeToInt(dbAttribute.getDataType()));

            table.alterColumn(column);

        } catch (SeException e) {

            throw new WriteException(this.store.getName(),e);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.ISelectiveWriter#insertAttribute(org.gvsig.data.vectorial.IFeatureAttributeDescriptor)

     */

    public void insertAttribute(IFeatureAttributeDescriptor attribute)

        throws WriteException {

        try {

            SeTable table = new SeTable(conex,featureType.getTableID());

            DBAttributeDescriptor dbAttribute=(DBAttributeDescriptor)attribute;

            SeColumnDefinition column=new SeColumnDefinition(dbAttribute.getName(),

                    SDETypes.fieldTypeToInt(dbAttribute.getDataType()),

                    dbAttribute.getSize(),

                    dbAttribute.getPrecision(),

                    dbAttribute.isAllowNull());

            table.addColumn(column);

        } catch (SeException e) {

            throw new WriteException(this.store.getName(),e);

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @param gvSIGgeometryType DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean canWriteGeometry(int gvSIGgeometryType) {

        switch (gvSIGgeometryType) {

        case FShape.POINT:

            return true;

        case FShape.LINE:

            return true;

        case FShape.POLYGON:

            return true;

        case FShape.ARC:

            return false;

        case FShape.ELLIPSE:

            return false;

        case FShape.MULTIPOINT:

            return true;

        case FShape.TEXT:

            return false;

        }

        return false;

    }

    /**

     * DOCUMENT ME!

     *

     * @param parameters DOCUMENT ME!

     * @param featureType DOCUMENT ME!

     *

     * @throws InitializeWriterException DOCUMENT ME!

     * @throws InitializeException DOCUMENT ME!

     */

//    public static void create(SDEStoreParameters parameters,

//        IFeatureType featureType)

//        throws InitializeWriterException, InitializeException {

//        SeConnection con = SDEUtils.getConnection(parameters.getHost(),parameters.getPort(),parameters.getDb(),

//                parameters.getUser(), parameters.getPassw());

//    }

    protected static String getFliterForIDForPStatement(DBFeatureType fType) {

        if (fType.getFieldsId().length != 1) {

            throw new UnsupportedOperationException("ID fields > 1");

        }

        String id = fType.getFieldsId()[0];

        return id;

    }

    /*

     *  (non-Javadoc)

     * @see org.gvsig.data.datastores.vectorial.IFeaturesWriter#dispose()

     */

        public void dispose() throws DataException {

                try {

                        this.conex.close();

                } catch (SeException e) {

                        throw new CloseException(this.store.getName(),e);

                }

                this.conex = null;

        }

}