| gvSIG desktop 1

/* 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):

 * 2009 {Iver T.I.}   {Task}

 */

package org.gvsig.fmap.dal.cache.h2spatial;

import java.io.File;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import org.gvsig.compat.CompatLocator;

import org.gvsig.compat.lang.StringUtils;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.cache.AbstractFeatureCacheProvider;

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

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.geom.Geometry;

/**

 * @author Vicente Caballero Navarro

 */

public class FeatureCacheH2SpatialProvider extends AbstractFeatureCacheProvider {

        public static final String NAME = "H2SPATIAL.CACHE";

        public static final String DESCRIPTION = "Cache provider for H2Spatial";

        private static final String FIELD_SRS = "srs";

        private static StringUtils stringUtils = CompatLocator.getStringUtils();

        private Map fieldsPosition = null;

        public FeatureStore createDataStore(double scale) throws DataException {

                NewFeatureStoreParameters parameters = createDataStoreParameters(0);

                EditableFeatureType eft = featureStoreProviderServices

                                .getDefaultFeatureType().getEditable();

                EditableFeatureAttributeDescriptor efad = eft.add("IDPK",

                                DataTypes.LONG);

                efad.setIsPrimaryKey(true);

                efad.setIsAutomatic(true);

                parameters.setDefaultFeatureType(eft);

                parameters.setDynValue("pkfields", "IDPK");

                calculateFieldsPosition();

                dataServerExplorer.add(parameters, true);

                dataServerExplorer.dispose();

                try {

                        FeatureStore store = (FeatureStore) dataManager

                                        .createStore(parameters);

                        store.getParameters().setDynValue("pkfields", "IDPK");

                        return store;

                } catch (ValidateDataParametersException e) {

                        throw new CreateException("Cache for scale" + scale, e);

                }

        }

        private void calculateFieldsPosition() throws DataException {

                double d = 0;

                FeatureType featureType = featureStoreProviderServices

                                .getDefaultFeatureType().getEditable();

                Iterator it = featureType.iterator();

                fieldsPosition = new HashMap();

                int i = 0;

                while (it.hasNext()) {

                        FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) it

                                        .next();

                        if (attr.getDataType() != DataTypes.GEOMETRY) {

                                fieldsPosition.put(attr.getName(), i);

                                i++;

                        }

                }

        }

        private NewFeatureStoreParameters createDataStoreParameters(double scale)

                        throws DataException {

                NewFeatureStoreParameters parameters = (NewFeatureStoreParameters) dataServerExplorer

                                .getAddParameters("H2Spatial");

                parameters.setDynValue("dbname", System.getProperty("user.home")

                                + File.separator + "gvSIG"+File.separator+"cache" + File.separator + "cache");

                parameters.setDynValue("schema", "PUBLIC");

                parameters.setDynValue("table", getStoreIdByScale(scale));

                parameters.setDynValue(FIELD_SRS, sourceProjection);

                return parameters;

        }

        private String getStoreIdByScale(double scale) {

                String cacheName = featureStoreProvider.getSourceId().toString();

                cacheName = cacheName.substring(cacheName.lastIndexOf("/") + 1,

                                cacheName.length());

                cacheName = stringUtils.replaceAll(cacheName, "\\.", "_");

                cacheName = stringUtils.replaceAll(cacheName, ":", "_");

                return cacheName

                                + "_"

                                + stringUtils.replaceAll(String.valueOf(scale), "\\.", "_")

                                                .toUpperCase();

        }

        @Override

        protected void copyFeature(EditableFeature editableFeature,

                        Feature sourceFeature) {

                FeatureAttributeDescriptor[] attDescriptors = editableFeature.getType()

                                .getAttributeDescriptors();

                String geomField = editableFeature.getType()

                                .getDefaultGeometryAttributeName();

                FeatureAttributeDescriptor[] pks = editableFeature.getType()

                                .getPrimaryKey();

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

                        FeatureAttributeDescriptor attr = attDescriptors[i];

                        if (attr.getDataType() != DataTypes.GEOMETRY) {

                                if (attr.equals(pks[0])) {

                                        editableFeature.set(attr.getName(),

                                                        ((DefaultFeatureReference) sourceFeature

                                                                        .getReference()).getOID());

                                } else {

                                        Object obj = sourceFeature.get(attr.getName());

                                        if (obj != null) {

                                                editableFeature.set(attr.getName(), obj);

                                        }

                                }

                        } else {

                                editableFeature.setGeometry(geomField, (Geometry) sourceFeature

                                                .get(attr.getName()));

                        }

                }

        }

}