Application: gvSIG desktop

package org.gvsig.data.datastores.vectorial.file.dbf;

import java.util.ArrayList;

import java.util.BitSet;

import java.util.Collection;

import java.util.Iterator;

import java.util.NoSuchElementException;

import org.gvsig.data.ComplexObservable;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.vectorial.AbstractFeatureCollection;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureID;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.expressionevaluator.FeatureFilter;

//FIXME: OJO Excepciones

public class DBFFeatureCollectionBitSet extends AbstractFeatureCollection {

        protected BitSet bitSet=new BitSet();

        protected ComplexObservable observable = new ComplexObservable();

//        protected Collection featureIDs;//<IFeatureID>

        protected IFeatureType featureType;

private DBFStore store;

private String filter;

private FeatureFilter parser;

private long driverFeatureCount;

        public DBFFeatureCollectionBitSet(DBFStore store,IFeatureType type, String filter) throws ReadException {

                this.featureType=type;

                this.filter=filter;

                if (this.filter!=null)

                        parser = new FeatureFilter(filter,this.featureType);

                driverFeatureCount=store.getFeatureCount();

                this.store=store;

                intilizeFeatureIDs();

        }

        protected void intilizeFeatureIDs() 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++) {

                                        DBFFeatureID featureID=(DBFFeatureID)createCurrentFeatureID(i);

                                        if (parser.match(featureID.getFeature(featureType))){

                                                bitSet.set(i);

                                        }

                                }

                        }

                }catch (Exception e) {

                        throw new ReadException("DBFFeatureCollectionBitSet",e);

                }

        }

        protected IFeatureID createCurrentFeatureID(long pos){

                if (pos<driverFeatureCount){

                        return new DBFFeatureID((DBFStore)store,pos);

                } else {

                        return null;

                }

        }

        public int size() {

                return bitSet.cardinality();

        }

        public boolean isEmpty() {

                return bitSet.isEmpty();

        }

        public boolean contains(Object o) {

                if (o instanceof IFeature){

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

                }

                return false;

        }

        public Iterator iterator() {

                DBFBitSetIterator dbfIter=new DBFBitSetIterator(bitSet);

                return dbfIter;

        }

        public Object[] toArray() {

                ArrayList features= new ArrayList();

                Iterator iterator= this.iterator();

                while(iterator.hasNext()){

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

                        features.add(feature);

                }

                return features.toArray();

        }

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

                ArrayList features= new ArrayList();

                Iterator iterator= this.iterator();

                while(iterator.hasNext()){

                        IFeature feature=(IFeature)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) {

                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.bs=bitSet;

                }

                public boolean hasNext() {

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

                }

                public Object next() {

                        DBFFeature feature=null;

                        if (!this.hasNext()){

                                throw new NoSuchElementException();

                        }

                        current = bs.nextSetBit(current+1);

                        try {

                                feature = new DBFFeature(featureType,store,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;

        }

}