| 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.db.h2spatial.impl.H2SpatialNewStoreParameters;

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.dal.feature.impl.DefaultFeatureStore;

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

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

import org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters;

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+"gvsigcache"+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();

//                return "tablaprueba";

        }

        @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()));

                        }

                }

        }

        public void apply(

                        FeatureStoreProviderServices featureStoreProviderServices,

                        FeatureStoreProvider featureStoreProvider) throws DataException {

//                EditableFeatureType eft = featureStoreProviderServices.getDefaultFeatureType().getEditable();

//                if (featureType.getDefaultGeometryAttribute().getGeometryType() == TYPES.GEOMETRY){

//                        throw new NotSupportMultipleGeometriesException();

//                }

                super.apply(featureStoreProviderServices, featureStoreProvider);

        }

}