Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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 2

* 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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2008 {{Company}}   {{Task}}

*/

package org.gvsig.fmap.data.feature.impl;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import org.gvsig.fmap.data.exceptions.DataException;

import org.gvsig.fmap.data.feature.FeatureIndex;

import org.gvsig.fmap.data.feature.FeatureIndexes;

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

import org.gvsig.fmap.data.feature.spi.FeatureSetProvider;

import org.gvsig.fmap.data.feature.spi.index.FeatureIndexProviderServices;

import org.gvsig.tools.evaluator.Evaluator;

import org.gvsig.tools.evaluator.EvaluatorFieldValue;

/**

 * This class provides access to a FeatureStore local indexes either by

 * FeatureType and attrName or by index name.

 * @author jyarza

 */

public class DefaultFeatureIndexes implements FeatureIndexes {

        // Access by FeatureType and attribute name

        private Map featureTypes = new HashMap();

        // Access by DataIndex name

        private Map names = new HashMap();

        private DefaultFeatureStore store;

        /**

         * Creates an empty DataIndexes for the given FeatureStore

         *

         * @param store

         *            FeatureStore to whom this belongs

         * @throws DataException

         */

        public DefaultFeatureIndexes(DefaultFeatureStore store)

                        throws DataException {

                this.store = store;

                Iterator it = store.getFeatureTypes().iterator();

                while (it.hasNext()) {

                        FeatureType type = (FeatureType) it.next();

                        featureTypes.put(type, new HashMap());

                }

        }

        /**

         * @see org.gvsig.fmap.data.index.DataIndexes#getDataIndex(org.gvsig.fmap.data.feature.FeatureType,

         *      java.lang.String)

         * @deprecated

         */

        public FeatureIndex getFeatureIndex(FeatureType fType, String attrName) {

                Map indexes = (Map) featureTypes.get(fType);

                if (indexes != null) {

                        return (FeatureIndex) (indexes.get(attrName));

                }

                return null;

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.index.DataIndexes#getDataIndex(java.lang.String)

         */

        public FeatureIndex getFeatureIndex(String name) {

                return (FeatureIndex) names.get(name);

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.index.DataIndexes#addIndex(org.gvsig.fmap.data.feature.FeatureType, java.lang.String, org.gvsig.fmap.data.feature.DataIndex)

         */

        public void addIndex(FeatureIndexProviderServices index) {

                Map indexes = (Map) featureTypes.get(index.getFeatureType());

                if (indexes == null) {

                        // This would mean that a new feature type has been added to the FeatureStore since this Indexes was created

                        indexes = new HashMap();

                        featureTypes.put(index.getFeatureType(), indexes);

                }

                // FIXME Taking into account only the first attribute

                indexes.put(index.getAttributeNames()[0], index);

                // By name

                names.put(index.getName(), index);

        }

        /**

         * @see org.gvsig.fmap.data.index.DataIndexes#contains(org.gvsig.fmap.data.feature.FeatureType,

         *      java.lang.String)

         * @deprecated

         */

        public boolean contains(FeatureType fType, String attrName) {

                Map map = (Map) featureTypes.get(fType);

                return map == null ? false : map.containsKey(attrName);

        }

        /**

         * @deprecated

         */

        public boolean contains(String name) {

                return names.containsKey(name);

        }

        public Iterator iterator() {

                return names.values().iterator();

        }

        /**

         * Obtain a FeatureSetProvider over an index with the attributes used in the

         * evaluator or null if can't apply any index.

         * 

         * @param evaluator

         * @return FeatureSetProvider or null if not has index availables.

         * 

         */

        public FeatureSetProvider getFeatureSet(Evaluator evaluator) {

                class ApplyIndex {

                        DefaultFeatureIndex index;

                        EvaluatorFieldValue[] data;

                        ApplyIndex(DefaultFeatureIndex index, EvaluatorFieldValue[] data) {

                                this.index = index;

                                this.data = data;

                        }

                }

                try {

                        List applyIndex = new ArrayList();

                        Iterator indexes = this.iterator();

                        while (indexes.hasNext()) {

                                DefaultFeatureIndex index = (DefaultFeatureIndex) indexes

                                                .next();

                                String[] attrs = index.getAttributeNames();

                                for (int i = 0; i < attrs.length; i++) {

                                        String attrname = attrs[i];

                                        EvaluatorFieldValue[] values = evaluator

                                                        .getFieldValues(attrname);

                                        if (values != null) {

                                                applyIndex.add(new ApplyIndex(index, values));

                                                break;

                                        }

                                }

                        }

                        switch (applyIndex.size()) {

                        case 0:

                                return null;

                        default:

                                // FIXME: hay mas de un indice aplicable, que hacer? de momento

                                // usa el primero.

                                this.store

                                                .getLogger()

                                                .info(

                                                                "Mas de un indice applica para la consulta. Se escoge arbitrariamente el primero.");

                        case 1:

                                ApplyIndex x = (ApplyIndex) applyIndex.get(0);

                                switch (x.data[0].getType()) {

                                case EvaluatorFieldValue.MATCH:

                                        return x.index.getMatchFeatureSet(x.data[0].getValue());

                                case EvaluatorFieldValue.RANGE:

                                        return x.index.getRangeFeatureSet(x.data[0].getValue1(),

                                                        x.data[0].getValue2());

                                case EvaluatorFieldValue.NEAREST:

                                        if (x.data[0].getTolerance() == -1) {

                                                return x.index.getNearestFeatureSet(x.data[0]

                                                                .getCount(), x.data[0].getValue());

                                        } else {

                                                return x.index.getNearestFeatureSet(x.data[0]

                                                                .getCount(), x.data[0].getTolerance(),

                                                                x.data[0].getValue());

                                        }

                                }

                        }

                } catch (Exception e) {

                        this.store.getLogger().error("Error buscando indice que aplicar.",

                                        e);

                        return null;

                }

                return null;

        }

}