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;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.ArrayUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

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

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

import org.gvsig.fmap.dal.feature.exception.FeatureIndexException;

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

import org.gvsig.tools.evaluator.Evaluator;

import org.gvsig.tools.evaluator.EvaluatorFieldValue;

import org.gvsig.tools.evaluator.EvaluatorFieldValueMatch;

import org.gvsig.tools.evaluator.EvaluatorFieldValueNearest;

import org.gvsig.tools.evaluator.EvaluatorFieldValueRange;

/**

 * This class provides access to a FeatureStore local indexes and also decides

 * which index to use given an evaluator containing the filter expression.

 *

 * @author jyarza

 */

public class DefaultFeatureIndexes implements FeatureIndexes {

    private static final Logger LOG = LoggerFactory

        .getLogger(DefaultFeatureIndexes.class);

    // Access by index name

    private Map names;

    // Store to which this belongs

    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 {

        names = new HashMap();

        this.store = store;

    }

    /*

     * (non-Javadoc)

     *

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

     */

    public FeatureIndex getFeatureIndex(String name) {

        return (FeatureIndex) names.get(name);

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.gvsig.fmap.dal.index.DataIndexes#addIndex(org.gvsig.fmap.dal.feature

     * .FeatureType,

     * java.lang.String, org.gvsig.fmap.dal.feature.DataIndex)

     */

    public void addIndex(FeatureIndexProviderServices index) {

        // By name

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

    }

    public Iterator iterator() {

        return names.values().iterator();

    }

    /**

     * Using the given evaluator attributes, choose and use an appropriate index

     * to obtain a FeatureSet. If no index can be applied, then this method

     * returns null

     *

     * @param evaluator

     * @return FeatureSet or null if could not find any appropriate index.

     * @throws FeatureIndexException

     *

     */

    public FeatureSet getFeatureSet(Evaluator evaluator)

        throws FeatureIndexException {

        class ApplyIndex {

            DefaultFeatureIndex index;

            EvaluatorFieldValue[] data;

            ApplyIndex(DefaultFeatureIndex index, EvaluatorFieldValue[] data) {

                this.index = index;

                this.data = data;

            }

            /**

             * Checks whether the index supports the evaluator request

             *

             * @return

             */

            boolean isSupported() {

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

                case EvaluatorFieldValue.MATCH:

                    return index.getFeatureIndexProvider().isMatchSupported();

                case EvaluatorFieldValue.NEAREST:

                    return index.getFeatureIndexProvider().isNearestSupported();

                case EvaluatorFieldValue.RANGE:

                    return index.getFeatureIndexProvider().isRangeSupported();

                default:

                    return false;

                }

            }

            /**

             * Applies the index using the evaluator fields

             *

             * @return FeatureSet with the result

             * @throws FeatureIndexException

             */

            IndexFeatureSet apply() throws FeatureIndexException {

                EvaluatorFieldValueRange rangeField;

                EvaluatorFieldValueNearest nearestField;

                // Trick: we know DefaultIndexProvider returns an

                // IndexFeatureSet,

                // which implements both FeatureSetProvider and FeatureSet.

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

                case EvaluatorFieldValue.MATCH:

                    return (IndexFeatureSet) index

                        .getMatchFeatureSet(((EvaluatorFieldValueMatch) data[0])

                            .getValue());

                case EvaluatorFieldValue.RANGE:

                    rangeField = (EvaluatorFieldValueRange) data[0];

                    return (IndexFeatureSet) index.getRangeFeatureSet(

                        rangeField.getValue1(), rangeField.getValue2());

                case EvaluatorFieldValue.NEAREST:

                    nearestField = (EvaluatorFieldValueNearest) data[0];

                    if ((nearestField.getTolerance() == null)

                        || (!isSupported())) {

                        return (IndexFeatureSet) index.getNearestFeatureSet(

                            nearestField.getCount(), nearestField.getValue());

                    } else {

                        return (IndexFeatureSet) index.getNearestFeatureSet(

                            nearestField.getCount(), nearestField.getValue(),

                            nearestField.getTolerance());

                    }

                }

                return null;

            }

        }

        // Select applicable indexes

        List applyIndexes = new ArrayList();

        Iterator indexes = this.iterator();

        while (indexes.hasNext()) {

            DefaultFeatureIndex index = (DefaultFeatureIndex) indexes.next();

            if (index.isValid()) {

                String[] attrs =

                    (String[]) index.getAttributeNames().toArray(new String[0]);

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

                    String attrname = attrs[i];

                    EvaluatorFieldValue[] values = null;

                    if (evaluator.getFieldsInfo() != null) {

                        values =

                            evaluator.getFieldsInfo().getFieldValues(attrname);

                    }

                    if (ArrayUtils.isNotEmpty(values)) {

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

                        break;

                    }

                }

            }

        }

        // If there's not any applicable index, return null

        if (applyIndexes.size() == 0) {

            return null;

        }

        // Lookup an index with support for the requested function

        Iterator it = applyIndexes.iterator();

        ApplyIndex index = (ApplyIndex) it.next();

        while (it.hasNext() && (!index.isSupported())) {

            index = (ApplyIndex) it.next();

        }

        // If there is not any any index supporting the function, use the

        // first one

        if (!index.isSupported()) {

            LOG.info("No index support for the evaluator values. Using default index.");

            index = (ApplyIndex) applyIndexes.get(0);

        }

        // Apply index

        return index.apply();

    }

    public boolean areValid() {

        Iterator indexes = this.iterator();

        while (indexes.hasNext()) {

            FeatureIndex index = (FeatureIndex) indexes.next();

            if (!index.isValid()) {

                return false;

            }

        }

        return true;

    }

    public String toString() {

        StringBuffer text = new StringBuffer(super.toString());

        text.append(": [");

        Iterator indexes = this.iterator();

        while (indexes.hasNext()) {

            FeatureIndex index = (FeatureIndex) indexes.next();

            text.append(index);

            text.append("]");

            if (indexes.hasNext()) {

                text.append(", ");

            }

        }

        return text.toString();

    }

}