GTFeatureIterator.java

/* (c) 2014 Open Source Geospatial Foundation - all rights reserved
 * (c) 2001 - 2013 OpenPlans
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.wps.sextante;

import java.util.NoSuchElementException;

import org.geotools.feature.FeatureCollection;
import org.geotools.feature.FeatureIterator;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;

import com.vividsolutions.jts.geom.Geometry;

import es.unex.sextante.dataObjects.FeatureImpl;
import es.unex.sextante.dataObjects.IFeature;
import es.unex.sextante.dataObjects.IFeatureIterator;
import es.unex.sextante.exceptions.IteratorException;

public class GTFeatureIterator
         implements
            IFeatureIterator {

   private final FeatureIterator<SimpleFeature> m_Iter;


   public GTFeatureIterator(final FeatureCollection<SimpleFeatureType, SimpleFeature> fc) {


      m_Iter = fc.features();

   }


   public boolean hasNext() {

      if (m_Iter != null) {
         return m_Iter.hasNext();
      }
      else {
         return false;
      }

   }


   public IFeature next() throws NoSuchElementException, IteratorException {

      if (m_Iter != null) {
         if (!m_Iter.hasNext()) {
            throw new NoSuchElementException();
         }
         final SimpleFeature gtFeat = m_Iter.next();
         final Object values[] = new Object[gtFeat.getAttributeCount() - 1];
         for (int i = 1; i < gtFeat.getAttributeCount(); i++) {
            values[i - 1] = gtFeat.getAttribute(i);
         }
         final IFeature feat = new FeatureImpl((Geometry) gtFeat.getDefaultGeometry(), values);
         return feat;
      }
      else {
         throw new IteratorException();
      }

   }


   public void close() {

      m_Iter.close();

   }


}