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

	}

}