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

 * 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.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.spi.FeatureStoreProvider;

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

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

//                eft.setDefaultGeometryAttributeName("GEOMETRY");

//                EditableFeatureType eft1=featureStoreProviderServices.createFeatureType();

//                FeatureAttributeDescriptor[] fads=eft.getAttributeDescriptors();

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

//                        EditableFeatureAttributeDescriptor efad=null;

//                        if (fads[i].getDataType()==DataTypes.GEOMETRY){

//                                efad=eft1.add("GEOMETRY", fads[i].getDataType());

//                                efad.setGeometryType(fads[i].getGeometryType());

//                                efad.setSRS(fads[i].getSRS());

//                        }else{

//                                efad=eft1.add(fads[i].getName(), fads[i].getDataType());

//                        }

//                        efad.setSize(fads[i].getSize());

//                }

                parameters.setDefaultFeatureType(eft);

                calculateFieldsPosition();

                dataServerExplorer.add(parameters, true);

                dataServerExplorer.dispose();

                try {

                        FeatureStore store=(FeatureStore)dataManager.createStore(parameters);

//                        FeatureSet setAll=store.getFeatureSet();

//                        DisposableIterator itAll=setAll.iterator();

//                        while (itAll.hasNext()) {

//                                Feature f = (Feature) itAll.next();

//                                System.err.println("All-createDataStore()= "+f.toString());

//                        }

//                

                        return store;                        

                } catch (ValidateDataParametersException e) {

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

                }                

        }        

        private void calculateFieldsPosition() throws DataException{

                FeatureType featureType = featureStoreProviderServices.getDefaultFeatureType();

                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");

//                File rootFile = ((FilesystemServerExplorer)dataServerExplorer).getCurrentPath();

                parameters.setDynValue("dbname",  System.getProperty("user.home")+File.separator+"gvsigcache"+File.separator+"cache.db");

//                parameters.setUrl(rootFile.getAbsolutePath());

                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("/"), cacheName.length());

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

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

//                return cacheName + "_" + stringUtils.replaceAll(String.valueOf(scale), "\\.", "_");

                return "tablaprueba";

        }

        @Override

        protected void copyFeature(EditableFeature editableFeature,

                        Feature sourceFeature) {

                 FeatureAttributeDescriptor[] attDescriptors = sourceFeature.getType().getAttributeDescriptors();                

                String geomField = editableFeature.getType().getDefaultGeometryAttributeName();

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

                        FeatureAttributeDescriptor attr = attDescriptors[i];

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

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

                                if (obj != null){

                                        editableFeature.set((Integer)fieldsPosition.get(attr.getName()), obj);

                                }

                        }else{

                                editableFeature.setGeometry(geomField, (Geometry)sourceFeature.get(attr.getIndex()));

                        }

                }

        }

        public void apply(

                        FeatureStoreProviderServices featureStoreProviderServices,

                        FeatureStoreProvider featureStoreProvider) throws DataException {

//                FeatureType featureType = featureStoreProviderServices.getFeatureStore().getDefaultFeatureType();

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

//                        throw new NotSupportMultipleGeometriesException();

//                }

                super.apply(featureStoreProviderServices, featureStoreProvider);

        }

}