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.HashMap;

import java.util.Iterator;

import java.util.Map;

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

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

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

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

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

/**

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

 * FeatureType and attrName or by index name.

 * @author jyarza

 */

public class DefaultDataIndexes implements DataIndexes {

        // Access by FeatureType and attribute name

        private Map featureTypes = new HashMap();

        // Access by DataIndex name

        private Map names = new HashMap();

        /**

         * Creates an empty DataIndexes for the given FeatureStore

         * 

         * @param store

         *            FeatureStore to whom this belongs

         * @throws DataException

         */

        public DefaultDataIndexes(FeatureStore store) throws DataException {

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

                while (it.hasNext()) {

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

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

                }

        }

        /* (non-Javadoc)

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

         */

        public DataIndex getDataIndex(FeatureType fType, String attrName) {

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

                if (indexes != null) {

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

                }

                return null;

        }

        /* (non-Javadoc)

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

         */

        public DataIndex getDataIndex(String name) {

                return (DataIndex) 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(FeatureType fType, String attrName, DataIndex index) {

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

                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(fType, indexes);

                }

                indexes.put(attrName, index);

                // By name

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

        }

        /* (non-Javadoc)

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

         */

        public boolean contains(FeatureType fType, String attrName) {

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

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

        }

        public boolean contains(String name) {

                return names.containsKey(name);

        }

}