| gvSIG desktop 1

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

import java.util.ArrayList;

import java.util.Comparator;

import java.util.Iterator;

import java.util.TreeSet;

import org.gvsig.data.ReadException;

import org.gvsig.data.datastores.vectorial.file.dbf.DBFFeatureCollection;

import org.gvsig.data.datastores.vectorial.file.dbf.DBFFeatureCollectionWithFeatureID;

import org.gvsig.data.datastores.vectorial.file.dbf.DBFStore;

import org.gvsig.data.vectorial.FeatureManager;

import org.gvsig.data.vectorial.Feature;

import org.gvsig.data.vectorial.FeatureCollection;

import org.gvsig.data.vectorial.FeatureType;

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

import org.opengis.filter.Filter;

import org.opengis.filter.sort.SortBy;

public class ShpFeatureCollectionWithFeatureID extends

                DBFFeatureCollectionWithFeatureID {

        public ShpFeatureCollectionWithFeatureID(FeatureManager fm, SHPStore store, FeatureType type, Filter filter, SortBy[] Order) throws ReadException {

                super(fm, store, type, filter, Order);

                // TODO Auto-generated constructor stub

        }

        protected void intilizeFeatureIDs(FeatureManager fm,DBFStore store, Filter filter,SortBy[] order) throws ReadException {

                try{

                if (order == null){

                        if (filter == null){

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

                                // el tama?o

                        } else {

                                featureIDs = new ArrayList();

                        }

                } else {

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

                        featureIDs = new TreeSet(comparator);

                }

                FeatureCollection coll = new ShpFeatureCollection(fm,(SHPStore)store,this.featureType,filter);

                Iterator iter = coll.iterator();

                while (iter.hasNext()){

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

                        featureIDs.add(feature.getID());

                }

                }catch (Exception e) {

                        throw new ReadException("DBFFeatureCollection",e);

                }

        }

}