Application: gvSIG desktop

/*

 * Created on 11-mar-2005

 *

 * 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

 */

package com.iver.cit.gvsig.fmap.drivers.arcsde;

import java.awt.geom.GeneralPath;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.sql.Types;

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

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

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

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

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

import com.esri.sde.sdk.geom.SePoint;

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

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

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

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

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

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

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

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

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

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

/**

 * @author FJP

 *

 * TODO To change the template for this generated type comment go to

 * Window - Preferences - Java - Code Generation - Code and Comments

 */

public class ArcSdeFeatureIterator implements IFeatureIterator {

    IGeometry geom;

    int numColumns;

    private SeQuery query = null;

    private boolean bFirst;

    Value[] regAtt;

    SeRow row;

    public static IGeometry getGeometry( SeShape shape ) {

        try {

            /*

             *   Retrieve the shape type.

             */

            int type = -1;

            type = shape.getType();

            // Display the X and Y values

            /* for( int partNo = 0 ; partNo < points.length ; partNo++, System.out.println("") ) 

                for( int subPartNo = 0 ; subPartNo < points[partNo].length ; subPartNo++, System.out.println("") ) 

                    for( int pointNo = 0 ; pointNo < points[partNo][subPartNo].length ; pointNo+=2) 

                        System.out.println("X: " + points[partNo][subPartNo][pointNo] + "\tY: "

                                                 + points[partNo][subPartNo][(pointNo+1)] ); */            

            switch( type )

            {

                case SeShape.TYPE_POINT:

                    double[][][] points = shape.getAllCoords();

                    FPoint2D p =  new FPoint2D(points[0][0][0],points[0][0][1]);

                    return ShapeFactory.createPoint2D(p);

                case SeShape.TYPE_LINE:

                case SeShape.TYPE_MULTI_LINE:

                case SeShape.TYPE_MULTI_SIMPLE_LINE:

                case SeShape.TYPE_SIMPLE_LINE:

                    GeneralPathX gpx = new GeneralPathX(shape.toGeneralPath());

                    return ShapeFactory.createPolyline2D(gpx);

                case SeShape.TYPE_MULTI_POINT:                    

                    break;

                case SeShape.TYPE_NIL:

                    return new FNullGeometry();

                case SeShape.TYPE_POLYGON:

                case SeShape.TYPE_MULTI_POLYGON:                    

                    GeneralPathX gpx2 = new GeneralPathX(shape.toGeneralPath());

                    return ShapeFactory.createPolygon2D(gpx2);

            } // End switch

        }

        catch (SeException e)

        {

            e.printStackTrace();

        }

        return new FNullGeometry();

    }     

    /**

     * @throws SQLException

     * 

     */

    public ArcSdeFeatureIterator(SeQuery query) {

        // Debe ser forward only

        this.query = query;

        try {

            row = query.fetch();

            numColumns = row.getNumColumns();

            regAtt = new Value[numColumns-1];

            bFirst = true;

        } catch (SeException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.drivers.jdbc.GeometryIterator#hasNext()

     */

    public boolean hasNext() throws SQLException {

        try {

            if (!bFirst)

                bFirst = false;

            else

                row = query.fetch();

            if (row == null)

            {

                query.close();

                return false;

            }

            return true;

        }

        catch (SeException e) {            

            e.printStackTrace();            

            // throw new SQLException(e.getMessage());

        }

        return false;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.drivers.jdbc.GeometryIterator#next()

     */

    public IFeature next() throws SQLException {

        SeShape spVal = null;

        SeColumnDefinition[] colDefs = row.getColumns(); 

        IFeature feat = null;

        try

        {            

            if ( row != null ) 

            { 

                for (int colNum = 0; colNum < colDefs.length; colNum++)

                {                

                    SeColumnDefinition colDef = colDefs[colNum];

                    int dataType = colDef.getType();

                    if ( row.getIndicator((short)colNum) != SeRow.SE_IS_NULL_VALUE)

                    {

                        switch( dataType )

                        {

                            case SeColumnDefinition.TYPE_SMALLINT:

                                regAtt[colNum-1] =  ValueFactory.createValue(row.getShort(colNum).intValue());

                                break;                                

                            case SeColumnDefinition.TYPE_DATE:

                                regAtt[colNum-1] =  ValueFactory.createValue(row.getDate(colNum));

                                break;

                            case SeColumnDefinition.TYPE_INTEGER:

                                regAtt[colNum-1] =  ValueFactory.createValue(row.getInteger(colNum).intValue());

                                break;

                            case SeColumnDefinition.TYPE_FLOAT:

                                regAtt[colNum-1] =  ValueFactory.createValue(row.getFloat(colNum).floatValue());

                                break;

                            case SeColumnDefinition.TYPE_DOUBLE:

                                regAtt[colNum-1] =  ValueFactory.createValue(row.getDouble(colNum).doubleValue());

                                break;

                            case SeColumnDefinition.TYPE_STRING:

                                regAtt[colNum-1] =  ValueFactory.createValue(row.getString(colNum));

                                break;

                            case SeColumnDefinition.TYPE_SHAPE:

                                spVal = row.getShape(colNum);                               

                                geom = getGeometry(spVal);

                                break;

                        } // End switch

                    } // End if

                } // for

                // System.out.println("Dentro de next(): " + spVal.getFeatureId().toString() + " " + regAtt[0]);

                feat = new DefaultFeature(geom, regAtt, "" + spVal.getFeatureId().longValue());

            } // if

        } catch (SeException e)

        {

            e.printStackTrace();

        }

        return feat;

    }

}