Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.sextante.app.extension.core;

import java.util.ArrayList;

import java.util.Iterator;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import es.unex.sextante.dataObjects.IRecord;

import es.unex.sextante.dataObjects.IRecordsetIterator;

import es.unex.sextante.dataObjects.RecordImpl;

public class gvRecordsetIterator implements IRecordsetIterator {

        private FeatureSet featureSet;

//        long m_iIndex;

        private DisposableIterator featureIterator;

        gvRecordsetIterator(FeatureSet obj) throws DataException {

                featureSet = obj;

                try {

                        featureIterator=featureSet.iterator();

                } catch (DataException e) {

                        throw e;

                }

//                m_iIndex = 0;

        }

        public boolean hasNext() {

                return featureIterator.hasNext();

        }

        public IRecord next() {

                Feature feature = (Feature) featureIterator.next();

                FeatureType fType=feature.getType();

                Iterator<FeatureAttributeDescriptor> descriptors=fType.iterator();

                ArrayList<Object> objects=new ArrayList<Object>();

                while (descriptors.hasNext()) {

                        FeatureAttributeDescriptor descriptor = descriptors.next();

                        objects.add(feature.get(descriptor.getName()));

                }

                IRecord record = new RecordImpl(objects.toArray(new Object[0]));

                return record;

        }

        public void close() {

                if (featureIterator != null) {

                        featureIterator.dispose();

                        featureIterator = null;

                }

                DisposeUtils.dispose(featureSet);

        }

}