| gvSIG desktop 1

package org.gvsig.fmap.dal.store.lidar;

import java.util.Iterator;

import java.util.NoSuchElementException;

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

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

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

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

import org.gvsig.fmap.dal.feature.spi.FeatureData;

import org.gvsig.fmap.dal.feature.spi.FeatureSetProvider;

public class LiDARSetProvider implements FeatureSetProvider {

        private LiDARStoreProvider store;

        private FeatureQuery query;

        public LiDARSetProvider(LiDARStoreProvider liDARStoreProvider,

                        FeatureQuery query) {

                this.store = liDARStoreProvider;

                this.query = query;

        }

        public boolean canFilter() {

                return false;

        }

        public boolean canIterateFromIndex() {

                return true;

        }

        public boolean canOrder() {

                return false;

        }

        public Iterator fastIterator(long index) throws DataException {

                return new FastLiDARIterator(this.store, this.query.getFeatureType(),

                                index);

        }

        public Iterator fastIterator() throws DataException {

                return this.fastIterator(0);

        }

        public long getSize() throws DataException {

                return this.store.getFeatureCount();

        }

        public boolean isEmpty() throws DataException {

                return this.store.getFeatureCount() > 0;

        }

        public Iterator iterator() throws DataException {

                return this.iterator(0);

        }

        public Iterator iterator(long index) throws DataException {

                return new LiDARIterator(this.store, this.query.getFeatureType(), index);

        }

        private class LiDARIterator implements Iterator {

                protected long curIndex;

                protected LiDARStoreProvider store;

                protected FeatureType featureType;

                private long count;

                public LiDARIterator(LiDARStoreProvider store, FeatureType featureType,

                                long index) throws DataException {

                        this.store = store;

                        this.featureType = featureType;

                        this.curIndex = index;

                        this.count = this.store.getFeatureCount();

                }

                public boolean hasNext() {

                        return curIndex < count;

                }

                public Object next() {

                        if (!hasNext()){

                                throw new NoSuchElementException();

                        }

                        try{

                                FeatureData data = this.createFeature();

                                curIndex++;

                                return data;

                        } catch (DataException e){

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

                        }

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

                protected FeatureData createFeature() throws DataException {

                        return this.store.getFeatureDataByIndex(curIndex, featureType);

                }

        }

        private class FastLiDARIterator extends LiDARIterator {

                public FastLiDARIterator(LiDARStoreProvider store,

                                FeatureType featureType, long index) throws DataException {

                        super(store, featureType, index);

                        this.curData = this.store.createFeatureData(featureType);

                }

                private FeatureData curData;

                protected FeatureData createFeature() throws DataException {

                        this.store.loadFeatureDataByIndex(curData, curIndex, featureType);

                        return curData;

                }

        }

}