| gvSIG desktop 1

/*

 * Created on 11-abr-2007

 *

 * 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

 */

/* CVS MESSAGES:

*

* $Id: DefaultFeatureIterator.java 11285 2007-04-19 17:27:58Z azabala $

* $Log$

* Revision 1.1  2007-04-19 17:27:58  azabala

* first version in cvs

*

*

*/

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

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

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

import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileReadException;

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

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

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

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

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

/**

 * 

 * Iterator over all the features of a vectorial adapter.

 * It is thinked for data sources which dont have capabilities

 * of querying or reprojection.

 * 

 * 

 * @author Alvaro Zabala

 * */

public class DefaultFeatureIterator implements IFeatureIterator {

        /**

         * Projection of the layer on which this iterator iterates.

         * TODO Move projection from layer to adapter or driver

         * */

        protected IProjection sourceProjection;

    /**

     * If its setted, all features returned by this iterator

     * will be previously reprojected to this target projection

     */

        protected IProjection targetProjection;

        /**

         * If its setted, returned features only will have these alphanumeric attributes

         */

        protected String[] fieldNames;

        /**

         * vectorial data source. It reads geometries

         * and has the recordset

         */

        protected ReadableVectorial source;

        /**

         * recordset, it reads alphanumeric attributes

         */

        protected SelectableDataSource recordset;

        /**

         * index of the next feature that will be returned by this

         * iterator

         */

        protected int currentFeature;

        /**

         * Default constructor. 

         * Creates an iterator which will return features in the data source projection

         * (without reprojection) and with all the alphanumeric attributes

         * @throws ReadDriverException 

         *

         */

        public DefaultFeatureIterator( ReadableVectorial source) throws ReadDriverException{

                this.source = source;

                this.recordset = source.getRecordset();

                currentFeature = 0;

        }

        /**

         * Constructor.

         * The iterator will reproject the geometry of the features to the specified target projection,

         * and with the specified attribute fields.

         * */

        public DefaultFeatureIterator(ReadableVectorial source, 

                                                                        IProjection sourceProj, 

                                                                        IProjection targetProj, 

                                                                        String[] fieldNames) throws ReadDriverException{

                this(source);

                this.sourceProjection = sourceProj;

                this.targetProjection = targetProj;

                this.fieldNames = fieldNames;

        }

        public boolean hasNext() throws ReadDriverException,

                        ExpansionFileReadException {

                boolean bMore = (currentFeature < source.getShapeCount());

                return bMore;

        }

        public IFeature next() throws ReadDriverException {

                try {

                        IGeometry geom = getGeometry(currentFeature);

                        Value[] regAtt = getValues(currentFeature);

                        IFeature feat  = new DefaultFeature(geom, regAtt, currentFeature + "");

                        currentFeature++;

                        return feat;

                } catch (ExpansionFileReadException e) {

                        throw new ReadDriverException("",e);

                } 

        }

        public void closeIterator() throws ReadDriverException {

        }

        public String[] getFieldNames() {

                return fieldNames;

        }

        public void setFieldNames(String[] fieldNames) {

                this.fieldNames = fieldNames;

        }

        public IProjection getTargetProjection() {

                return targetProjection;

        }

        public void setTargetProjection(IProjection targetProjection) {

                this.targetProjection = targetProjection;

        }

        public IProjection getSourceProjection() {

                return sourceProjection;

        }

        public void setSourceProjection(IProjection sourceProjection) {

                this.sourceProjection = sourceProjection;

        }

        protected IGeometry getGeometry(int geomIdx) throws ExpansionFileReadException, 

                                                                                                                                ReadDriverException{

                IGeometry geom = source.getShape(geomIdx); 

                if(this.targetProjection != null && this.sourceProjection != null){

                        ICoordTrans trans = sourceProjection.getCT(targetProjection);

                        geom.reProject(trans);

                }

                return geom;

        }

        protected Value[] getValues(int featureIdx) throws ReadDriverException{

                Value[] regAtt = null;

                if(fieldNames == null){

                        regAtt = new Value[recordset.getFieldCount()];

                        for (int fieldId = 0; fieldId < recordset.getFieldCount(); fieldId++) {

                                regAtt[fieldId] = recordset.getFieldValue(featureIdx, fieldId);

                        }

                }else{

                        regAtt = new Value[fieldNames.length];

                        for (int fieldId = 0; fieldId < fieldNames.length; fieldId++) {

                                int fieldCode = recordset.getFieldIndexByName(fieldNames[fieldId]);

                                regAtt[fieldId] = recordset.getFieldValue(featureIdx, fieldCode);

                        }

                }

                return regAtt;

        }

}