Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

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

import java.util.ArrayList;

import java.util.Comparator;

import java.util.Iterator;

import java.util.List;

import org.gvsig.compat.CompatLocator;

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

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

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

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

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

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

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

import org.gvsig.tools.evaluator.Evaluator;

import org.gvsig.tools.evaluator.EvaluatorException;

/*

 *

 * Compara dos Feature o FeatureProvider.

 */

public class FeatureProviderComparator implements Comparator {

        private FeatureQueryOrder order;

        private DefaultFeature f1;

        private DefaultFeature f2;

        private DefaultFeatureStore store;

        private List<OrderElement> elements = null;

        private class OrderElement {

                FeatureQueryOrderMember order = null;

                int fieldIndex = -1;

        }

        public FeatureProviderComparator(DefaultFeatureStore store, FeatureQueryOrder order) {

                this.order = order;

                this.store = store;

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

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

        }

        private void initElements(FeatureType type) {

                elements = new ArrayList<>();

                OrderElement cur;

                Iterator iter = order.iterator();

                while (iter.hasNext()) {

                    FeatureQueryOrderMember orderMember = (FeatureQueryOrderMember) iter.next();

                    int index = type.getIndex(orderMember.getAttributeName());

                    if(index >= 0) {

                        cur = new OrderElement();

                        cur.order = orderMember;

                        cur.fieldIndex = index;

                        elements.add(cur);

                    } else if(orderMember.hasEvaluator()){

                        cur = new OrderElement();

                        cur.order = orderMember;

                        cur.fieldIndex = index;

                        elements.add(cur);

                    }

                }

        }

        public int compare(Object arg0, Object arg1) {

                FeatureQueryOrderMember order;

                Comparable v1;

                Comparable v2;

                Object o1;

                Object o2;

                int v;

                int fieldIndex;

                f1.setData((FeatureProvider) arg0);

                f2.setData((FeatureProvider) arg1);

                if (elements == null){

                        initElements(f1.getType());

                }

                for (OrderElement element : elements) {

                        if (element.order.hasEvaluator()) {

                                Evaluator evaluator = element.order.getEvaluator();

                                try {

                                        o1 = evaluator.evaluate(f1);

                                        o2 = evaluator.evaluate(f2);

                                } catch (EvaluatorException e) {

                                        throw new DataEvaluatorRuntimeException(e);

                                }

                        } else {

                                o1 = f1.get(element.fieldIndex);

                                o2 = f2.get(element.fieldIndex);

                        }

                        if (o1 == null) {

                                if (o2 == null) {

                                        return 0;

                                } else {

                                        v = 1;

                                }

                        } else {

                                if (o1 instanceof String && o2 instanceof String) {

                                        v = CompatLocator.getStringUtils().compare((String) o1,

                                                        (String) o2, true);

                                }

                                else if (o1 instanceof Comparable && o2 instanceof Comparable) {

                                        v1 = (Comparable) o1;

                                        v2 = (Comparable) o2;

                                        v = v1.compareTo(v2);

                                } else {

                                        v = -1;

                                }

                        }

                        if (v != 0) {

                                if (element.order.getAscending()) {

                                        return v;

                                } else {

                                        return -v;

                                }

                        }

                }

                return 0;

        }

}