Application: gvSIG desktop

package org.gvsig.fmap.dal.feature.impl.featureSet;

import java.util.Comparator;

import java.util.Iterator;

import org.gvsig.fmap.dal.exceptions.DataEvaluatorRuntimeException;

import org.gvsig.fmap.dal.feature.FeatureQueryOrder;

import org.gvsig.fmap.dal.feature.FeatureQueryOrder.FeatureQueryOrderMember;

import org.gvsig.fmap.dal.feature.impl.DefaultFeature;

import org.gvsig.fmap.dal.feature.impl.DefaultFeatureStore;

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

import org.gvsig.tools.evaluator.EvaluatorException;

/*

*

* Compara dos Feature o FeatureData.

*/

public class FeatureDataComparator implements Comparator {

        private FeatureQueryOrder order;

        private DefaultFeature f1;

        private DefaultFeature f2;

        private DefaultFeatureStore store;

        public FeatureDataComparator(DefaultFeatureStore store, FeatureQueryOrder order) {

                this.order = order;

                this.store = store;

                this.f1 = new DefaultFeature(this.store);

                this.f2 = new DefaultFeature(this.store);

        }

        public int compare(Object arg0, Object arg1) {

                FeatureQueryOrderMember order;

                Comparable v1;

                Comparable v2;

                int v;

                f1.setData((FeatureData) arg1);

                f2.setData((FeatureData) arg1);

                Iterator iterator = this.order.iterator();

                while (iterator.hasNext()) {

                        order = (FeatureQueryOrderMember)iterator.next();

                        if( order.hasEvaluator() ) {

                                try {

                                        v1 = (Comparable) order.getEvaluator().evaluate(f1);

                                        v2 = (Comparable) order.getEvaluator().evaluate(f2);

                                } catch (EvaluatorException e) {

                                        throw new DataEvaluatorRuntimeException(e);

                                }

                        } else {

                                v1 = (Comparable) f1.get(order.getAttributeName());

                                v2 = (Comparable) f2.get(order.getAttributeName());

                        }

                        v = v1.compareTo(v2);

                        if (v != 0) {

                                if (order.getAscending()) {

                                        return v;

                                } else {

                                        return -v;

                                }

                        }

                }

                return 0;

        }

}