Application: gvSIG desktop

package org.gvsig.data.datastores.vectorial.driver.jdbc.h2;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Comparator;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import java.util.TreeSet;

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

import org.gvsig.data.vectorial.IFeatureID;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.expansionadapter.FeatureManager;

import org.gvsig.data.vectorial.filter.FeatureFilterParser;

import org.gvsig.data.vectorial.order.FeatureComparator;

public class H2FeatureCollectionWithFeatureID extends AbstractFeatureCollection {

        protected ComplexObservable observable = new ComplexObservable();

        protected Collection featureIDs;//<IFeatureID>

        protected IFeatureType featureType;

        public H2FeatureCollectionWithFeatureID(FeatureManager fm,H2Driver driver,IFeatureType type, String filter,String order) throws ReadException {

                this.featureType=type;

                intilizeFeatureIDs(fm, driver, filter, order);

        }

        protected void intilizeFeatureIDs(FeatureManager fm,H2Driver driver, String filter,String order) throws ReadException {

                FeatureFilterParser parser = null;

                IFeature feature;

                Collection tmpFeatures;

                if (order == null){

                        if (filter == null){

                                tmpFeatures = new ArrayList(); // Si queremos optimizar deberiamos inicializar

                                // el tama?o

                        } else {

                                tmpFeatures = new ArrayList();

                        }

                } else {

                        Comparator comparator = new FeatureComparator(this.featureType,order);

                        tmpFeatures = new TreeSet(comparator);

                }

                if (filter != null && filter != ""){

                        parser = new FeatureFilterParser(filter,featureType);

                }

                for (int i=0;i<fm.getNum();i++){

                        feature = fm.getFeature(i);

                        if (fm.isDeleted(feature)){

                                continue;

                        }

                        if (parser == null || parser.match(feature)){

                                if (order != null){

                                        tmpFeatures.add(feature);

                                } else {

                                        tmpFeatures.add(feature.getID());

                                }

                        }

                }

                IFeatureCollection coll = new H2FeatureCollection(driver,this.featureType,filter,order);

                Iterator iter = coll.iterator();

                while (iter.hasNext()){

                        feature =(IFeature)iter.next();

                        if (!fm.isDeleted(feature)){

                                if (order != null){

                                        tmpFeatures.add(feature);

                                } else {

                                        tmpFeatures.add(feature.getID());

                                }

                        }

                }

                if (order != null){

                        featureIDs = new ArrayList();

                        iter = tmpFeatures.iterator();

                        while (iter.hasNext()){

                                featureIDs.add(((IFeature)iter.next()).getID());

                        }

                } else {

                        featureIDs = tmpFeatures;

                }

        }

        public int size() {

                return featureIDs.size();

        }

        public boolean isEmpty() {

                return featureIDs.isEmpty();

        }

        public boolean contains(Object o) {

                if (o instanceof IFeature){

                        featureIDs.contains(((IFeature)o).getID());

                }

                return featureIDs.contains(o);

        }

        public Iterator iterator() {

                H2FIDIterator dbfIter=new H2FIDIterator(featureIDs.iterator());

                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 H2FIDIterator implements Iterator{

                private Iterator iter;

                public H2FIDIterator(Iterator iter){

                        this.iter=iter;

                }

                public boolean hasNext() {

                        return iter.hasNext();

                }

                public Object next() {

                        IFeatureID fid= (IFeatureID)iter.next();

                        IFeature feature;

                        try {

                                feature = fid.getFeature(featureType);

                        } catch (ReadException e) {

                                throw new RuntimeException(e);

                        }

                        return feature;

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

        }

}