| gvSIG desktop 1

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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., 51 Franklin Street, Fifth Floor, Boston,

* MA  02110-1301, USA.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2008 {{Company}}   {{Task}}

*/

package org.gvsig.fmap.dal.feature.impl;

import java.util.Iterator;

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.feature.spi.index.FeatureIndexProviderServices;

public class IndexFeatureSet implements FeatureSetProvider {

        LongList featureReferences = null;

        FeatureStoreProvider store = null;

        FeatureIndexProviderServices index = null;

        public class IndexIterator implements Iterator {

                Iterator it = null;

                public IndexIterator(Iterator it) {

                        this.it = it;

                }

                public boolean hasNext() {

                        return it.hasNext();

                }

                public Object next() {

                        Object oid = it.next();

                        Feature f = null;

                        try {

                                FeatureReference ref = new DefaultFeatureReference(store

                                                .getStoreServices(), oid);

                                f = index.getFeatureStore().getFeatureByReference(ref);

                        } catch (DataException e) {

                                throw new ReadRuntimeException(index.getFeatureStore().getName(), e);

                        }

                        return f;

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

        }

        public class FastIndexIterator implements Iterator {

                Iterator it = null;

                FeatureData data = null;

                Feature feat = null;

                public FastIndexIterator(Iterator it) throws DataException {

                        this.it = it;

                        data = store.createFeatureData(index.getFeatureType());

                        feat = index.getFeatureStore().createFeature(data);

                }

                public boolean hasNext() {

                        return it.hasNext();

                }

                public Object next() {

                        Object oid = it.next();

                        try {

                                FeatureReference ref = new DefaultFeatureReference(store

                                                .getStoreServices(), oid);

                                data = store

                                                .getFeatureDataByReference((FeatureReferenceProviderServices) ref);

                        } catch (DataException e) {

                                throw new ReadRuntimeException(index.getFeatureStore().getName(), e);

                        }

                        return data;

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

        }

        public IndexFeatureSet(FeatureIndexProviderServices index, LongList featureReferences) {

                this.featureReferences = featureReferences;

                this.store = index.getFeatureStore().getProvider();

                this.index = index;

        }

        public boolean canFilter() {

                return false;

        }

        public boolean canIterateFromIndex() {

                return true;

        }

        public boolean canOrder() {

                return false;

        }

        public Iterator fastIterator(long index) throws DataException {

                return new FastIndexIterator(this.featureReferences.iterator((int)index));

        }

        public Iterator fastIterator() throws DataException {

                return new FastIndexIterator(this.featureReferences.iterator());

        }

        public long getSize() throws DataException {

                return featureReferences.getSize();

        }

        public boolean isEmpty() throws DataException {

                return featureReferences.isEmpty();

        }

        public Iterator iterator() throws DataException {

                return new IndexIterator(this.featureReferences.iterator());

        }

        public Iterator iterator(long index) throws DataException {

                return new FastIndexIterator(this.featureReferences.iterator((int)index));

        }

}