Application: gvSIG desktop

/**

 *

 */

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

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.AbstractFeatureProviderIterator;

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

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

import org.gvsig.tools.exception.BaseException;

/**

 * @author jjdelcerro

 *

 */

public class DBFSetProvider extends AbstractFeatureSetProvider {

        public DBFSetProvider(DBFStoreProvider store, FeatureQuery query,

                        FeatureType featureType)

                        throws DataException {

                super(store, query, featureType);

        }

        public boolean canFilter() {

                return false;

        }

        public boolean canIterateFromIndex() {

                return true;

        }

        public boolean canOrder() {

                return false;

        }

        public long getSize() throws DataException {

                return getStore().getFeatureCount();

        }

        public boolean isEmpty() throws DataException {

                return getSize() == 0;

        }

        @Override

        protected void doDispose() throws BaseException {

        }

        protected AbstractFeatureProviderIterator createIterator(long index)

                        throws DataException {

                return new DBFIterator((DBFStoreProvider) getStore(), getFeatureType(),

                                index);

        }

        protected AbstractFeatureProviderIterator createFastIterator(long index)

                        throws DataException {

                return new FastDBFIterator((DBFStoreProvider) getStore(),

                                getFeatureType(), index);

        }

        protected class DBFIterator extends AbstractFeatureProviderIterator {

                protected long index;

                protected FeatureType type;

                protected long count;

                public DBFIterator(DBFStoreProvider store, FeatureType type,

                                long startOn) throws DataException {

                        super(store);

                        this.index = startOn;

                        this.type = type;

                        this.count = store.getFeatureCount();

                }

                protected boolean internalHasNext() {

                        return this.count > index;

                }

                protected Object internalNext() {

                        if (index >= this.count) {

                                throw new NoSuchElementException();

                        }

                        try {

                                FeatureProvider ret =

                                                getDBFStoreProvider().getFeatureProviderByIndex(index,

                                                                this.type);

                                index++;

                                return ret;

                        } catch (DataException e) {

                                throw new ReadRuntimeException(getDBFStoreProvider().getProviderName(),

                                                e);

                        }

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

                protected void internalDispose() {

                        type = null;

                }

                protected DBFStoreProvider getDBFStoreProvider() {

                        return (DBFStoreProvider) getFeatureStoreProvider();

                }

                @Override

                protected void doDispose() throws BaseException {

                        // Nothing to do

                }

        }

        protected class FastDBFIterator extends DBFIterator {

                protected FeatureProvider data;

                public FastDBFIterator(DBFStoreProvider store, FeatureType type,

                                long startOn) throws DataException {

                        super(store, type, startOn);

                        this.data = getFeatureStoreProvider().createFeatureProvider(type);

                }

                protected Object internalNext() {

                        if (index >= this.count) {

                                throw new NoSuchElementException();

                        }

                        try {

                                getDBFStoreProvider().initFeatureProviderByIndex(this.data,

                                                index, type);

                        } catch (DataException e) {

                                throw new ReadRuntimeException("next", e);

                        }

                        index++;

                        return this.data;

                }

                protected void internalDispose() {

                        super.internalDispose();

                        data = null;

                }

        }

}