Application: gvSIG desktop

package org.gvsig.fmap.data.feature.file.dbf;

import java.util.*;

import org.gvsig.fmap.data.DataManager;

import org.gvsig.fmap.data.DataStore;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.feature.AbstractFeatureCollection;

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

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

import org.gvsig.fmap.data.feature.expressionevaluator.Filter;

import org.gvsig.tools.observer.DefaultObservable;

//FIXME: OJO Excepciones

public class DBFFeatureCollectionBitSet extends AbstractFeatureCollection {

        protected BitSet bitSet=new BitSet();

        protected DefaultObservable observable = new DefaultObservable();

        protected FeatureType featureType;

        protected DBFStore store;

        protected String filter;

        protected Filter parser;

        protected long driverFeatureCount;

        protected DBFFeatureCollectionBitSet(DBFStore store,FeatureType type, String filter) throws ReadException {

                DataManager manager = DataManager.getManager();

                this.featureType=type;

                this.filter=filter;

                if (this.filter!=null) {

                        parser = manager.getExpressionParser().parseFilter(filter);

                }

                driverFeatureCount=store.getFeatureCount();

                this.store=store;

                intilizeBitSet();

        }

        protected void intilizeBitSet() throws ReadException {

                try{

                        bitSet.clear();

                        if (filter==null){

                                for (int i = 0; i < driverFeatureCount; i++) {

                                        bitSet.set(i);

                                }

                        }else{

                                for (int i = 0; i < driverFeatureCount; i++) {

                                        Feature feature = getFeature(i);

                                        if (parser.evaluate(feature)) {

                                                bitSet.set(i);

                                        }

                                }

                        }

                }catch (Exception e) {

                        throw new ReadException("DBFFeatureCollectionBitSet",e);

                }

        }

        public int size() {

                checkModified();

                return bitSet.cardinality();

        }

        public boolean isEmpty() {

                checkModified();

                return bitSet.isEmpty();

        }

        public boolean contains(Object o) {

                checkModified();

                if (o instanceof Feature){

                        return bitSet.get((int)((DBFFeatureID)((Feature)o).getID()).getIndex());

                }

                return false;

        }

        protected Iterator internalIterator(int index) {

        checkModified();

        DBFBitSetIterator dbfIter = new DBFBitSetIterator(bitSet, index);

        return dbfIter;

    }

        public Object[] toArray() {

                checkModified();

                ArrayList features= new ArrayList();

                Iterator iterator= this.iterator();

                while(iterator.hasNext()){

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

                        features.add(feature);

                }

                return features.toArray();

        }

        public Object[] toArray(Object[] a) {

                checkModified();

                ArrayList features= new ArrayList();

                Iterator iterator= this.iterator();

                while(iterator.hasNext()){

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

                        features.add(feature);

                }

                return features.toArray(a);

        }

        public boolean add(Object o) {

                throw new UnsupportedOperationException();

        }

        public boolean remove(Object o) {

                throw new UnsupportedOperationException();

        }

        public boolean containsAll(Collection c) {

                checkModified();

                Iterator iter = c.iterator();

                while (iter.hasNext()){

                        if (!this.contains(iter.next())){

                                return false;

                        }

                }

                return true;

        }

        public boolean addAll(Collection c) {

                throw new UnsupportedOperationException();

        }

        public boolean removeAll(Collection c) {

                throw new UnsupportedOperationException();

        }

        public boolean retainAll(Collection c) {

                throw new UnsupportedOperationException();

        }

        public void clear() {

                throw new UnsupportedOperationException();

        }

        private class DBFBitSetIterator implements Iterator{

                private BitSet bs;

                private int current=-1;

                public DBFBitSetIterator(BitSet bitSet){

                        this(bitSet, 0);

                }

        /**

         * Creates a new DBFBitSetIterator starting at the given position.

         * 

         * @param bitSet

         *            to iterate on

         * @param position

         *            to start the iteration

         */

        public DBFBitSetIterator(BitSet bitSet, int position) {

            this.bs = bitSet;

            this.current = position - 1;

        }

                public boolean hasNext() {

                        checkModified();

                        return (bs.nextSetBit(current+1) >= 0);

                }

                public Object next() {

                        checkModified();

                        Feature feature=null;

                        if (!this.hasNext()){

                                throw new NoSuchElementException();

                        }

                        current = bs.nextSetBit(current+1);

                        try {

                                feature = getFeature(current);

                        } catch (ReadException e) {

                                throw new RuntimeException(e);

                        }

                        return feature;

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

        }

        public void dispose() {

                observable.deleteObservers();

                this.observable = null;

                this.bitSet.clear();

                this.featureType = null;

        }

        public Feature getFeature(int index) throws ReadException {

                Feature feature=null;

                if (index<driverFeatureCount){

                        return store.getFeatureByIndex(index, featureType);

                }else{

                        feature = null;

                }

                return feature;

        }

        public boolean isFromStore(DataStore store) {

                return this.store.equals(store);

        }

        public FeatureType getFeatureType() {

                return this.featureType;

        }

}