Application: gvSIG desktop

package org.gvsig.data.datastores.vectorial.driver;

import java.io.File;

import junit.framework.TestCase;

import org.gvsig.data.DataSourceManager;

import org.gvsig.data.IDataSource;

import org.gvsig.data.IDataSourceParameters;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datastores.vectorial.driver.dbf.DBFDataSource;

import org.gvsig.data.datastores.vectorial.driver.dbf.DBFDataSourceParameters;

import org.gvsig.data.datastores.vectorial.driver.shp.ShpDataSource;

import org.gvsig.data.datastores.vectorial.driver.shp.ShpDataSourceParameters;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureStore;

public class DataSourceTest extends TestCase {

	private File dbffile = new File(DataStoreTest.class.getResource("data/prueba.dbf").getFile());

	private File shpfile = new File(DataStoreTest.class.getResource("data/prueba.shp").getFile());

	public void testDBFSources(){

		org.gvsig.data.datastores.vectorial.driver.dbf.Register.selfRegister();

		org.gvsig.data.datastores.vectorial.driver.Register.selfRegister();

		DataSourceManager dsm=DataSourceManager.getManager();

		IDataSourceParameters dsp = dsm.createDataSourceParameters(

				DBFDataSource.DATASOURCE_NAME);

		((DBFDataSourceParameters)dsp).setSource(new File(dbffile.getParent()));

		IDataSource src = dsm.createDataSource(dsp);

		IDataStoreParameters[] params = null;

		try {

			params = src.getSources();

		} catch (ReadException e) {

			e.printStackTrace();

			fail();

		}

		assertEquals(1, params.length);

		IFeatureStore fs;

		try {

			fs = (IFeatureStore) dsm.createDataStore(params[0]);

			fs.open();

			IFeatureCollection fc = (IFeatureCollection) fs.getDataCollection();

			assertEquals(9, fc.size());

			fs.close();

		} catch (Exception e) {

			e.printStackTrace();

			fail();

		}

	}

	public void testSHPSources(){

		org.gvsig.data.datastores.vectorial.driver.shp.Register.selfRegister();

		org.gvsig.data.datastores.vectorial.driver.Register.selfRegister();

		DataSourceManager dsm=DataSourceManager.getManager();

		IDataSourceParameters dsp = dsm.createDataSourceParameters(

				ShpDataSource.DATASOURCE_NAME);

		((ShpDataSourceParameters)dsp).setSource(new File(shpfile.getParent()));

		IDataSource src = dsm.createDataSource(dsp);

		IDataStoreParameters[] params = null;

		try {

			params = src.getSources();

		} catch (ReadException e) {

			e.printStackTrace();

			fail();

		}

		assertEquals(1, params.length);

		IFeatureStore fs;

		try {

			fs = (IFeatureStore) dsm.createDataStore(params[0]);

			fs.open();

			IFeatureCollection fc = (IFeatureCollection) fs.getDataCollection();

			assertEquals(9, fc.size());

			fs.close();

		} catch (Exception e) {

			e.printStackTrace();

			fail();

		}

	}

}

package org.gvsig.data.datastores.vectorial.driver.shp;

import org.gvsig.data.datasource.driver.file.DataSourceFileDriverParameters;

public class ShpDataSourceParameters extends DataSourceFileDriverParameters {

	public static String DATASOURCE_NAME="ShpDataSource";

	public String getDataSourceName() {

		return DATASOURCE_NAME;

	}

}

		dsm.registerDriver(ShpDriver.DRIVER_NAME,

				ShpDriver.class,

		dsm.registerDataSource(

				ShpDataSource.DATASOURCE_NAME,

				ShpDataSource.class,

				ShpDataSourceParameters.class

		);

package org.gvsig.data.datastores.vectorial.driver.shp;

import java.io.File;

import org.gvsig.data.IDataSourceParameters;

import org.gvsig.data.datastores.vectorial.driver.IDriverParameters;

import org.gvsig.data.datastores.vectorial.driver.dbf.DBFDataSource;

import org.gvsig.data.datastores.vectorial.driver.shp.fileshp.SHP;

public class ShpDataSource extends DBFDataSource {

	public static String DATASOURCE_NAME="ShpDataSource";

	protected ShpDataSourceParameters parameters;

	protected boolean isValid(File file) {

		if (!file.exists()){

			return false;

		}

		if (!(file.getName().toLowerCase().endsWith(".shp"))){

			return false;

		}

		File shx = SHP.getShxFile(file);

		if (!shx.exists()){

			return false;

		}

		File dbf = SHP.getDbfFile(file);

		return super.isValid(dbf);

	}

	protected IDriverParameters newParameter(File file){

		ShpDriverParameters param = new ShpDriverParameters();

		param.setSHPFile(file);

		param.setSHXFile(SHP.getShxFile(file));

		param.setDBFFile(SHP.getDbfFile(file));

		return param;

	}

	public void init(IDataSourceParameters parameters) {

		this.parameters = (ShpDataSourceParameters)parameters;

		this.path = this.parameters.getSource();

	}

}

		dsm.registerDriver(DBFDriver.DRIVER_NAME,

				DBFDriver.class,

		dsm.registerDataSource(

				DBFDataSource.DATASOURCE_NAME,

				DBFDataSource.class,

				DBFDataSourceParameters.class

		);

package org.gvsig.data.datastores.vectorial.driver.dbf;

import java.io.File;

import java.util.ArrayList;

import org.gvsig.data.DataSourceManager;

import org.gvsig.data.IDataSourceParameters;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datasource.driver.file.AbstractDataSourceFileDriver;

import org.gvsig.data.datastores.vectorial.driver.DriverStore;

import org.gvsig.data.datastores.vectorial.driver.IDriverParameters;

import org.gvsig.data.datastores.vectorial.driver.IDriverStoreParameters;

import org.gvsig.data.datastores.vectorial.driver.exception.FileNotFoundDriverException;

import org.gvsig.data.exception.ReadException;

public class DBFDataSource extends AbstractDataSourceFileDriver {

    public static String DATASOURCE_NAME="DBFDataSource";

	protected DBFDataSourceParameters parameters;

	protected File path;

	public IDataStoreParameters[] getSources() throws ReadException {

		if (!this.path.exists()){

			new FileNotFoundDriverException(this.getName()+": Path not found '"+this.path+"'",new Exception());

		}

		DataSourceManager dsm=DataSourceManager.getManager();

		String files[] = this.path.list();

		int i;

		File theFile;

		ArrayList fileList = new ArrayList();

		IDriverParameters param;

		IDataStoreParameters dsp = null;

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

			theFile = new File(this.path,files[i]);

			if (this.isValid(theFile)){

				param= this.newParameter(theFile);

				dsp=dsm.createDataStoreParameters(DriverStore.DATASTORE_NAME);

				((IDriverStoreParameters)dsp).setDriverParameters(param);

				fileList.add(dsp);

			}

		}

		IDataStoreParameters[] x = new IDataStoreParameters[1];

		x[0]=dsp;

		return (IDataStoreParameters[])fileList.toArray(x);

	}

	protected IDriverParameters newParameter(File file){

		DBFDriverParameters param = new DBFDriverParameters();

		param.setDBFFile(file);

		return param;

	}

	protected boolean isValid(String file){

		return this.isValid(new File(file));

	}

	protected boolean isValid(File file){

		if (!file.exists()){

			return false;

		}

		return (file.getName().toLowerCase().endsWith(".dbf"));

	}

	public String getName() {

		return this.DATASOURCE_NAME;

	}

	public void init(IDataSourceParameters parameters) {

		this.parameters = (DBFDataSourceParameters)parameters;

		this.path = this.parameters.getSource();

	}

}

package org.gvsig.data.datastores.vectorial.driver.dbf;

import java.io.File;

import org.gvsig.data.datasource.driver.file.DataSourceFileDriverParameters;

public class DBFDataSourceParameters extends DataSourceFileDriverParameters {

	private static String DATASOURCE_NAME="DBFDataSource";

	public String getDataSourceName() {

		return DATASOURCE_NAME;

	}

}

package org.gvsig.data.datasource.driver.file;

import java.net.URI;

import org.gvsig.data.IDataSourceParameters;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.vectorial.AbstractDataSourceFile;

public abstract class AbstractDataSourceFileDriver extends AbstractDataSourceFile{

	static String DATASOURCE_NAME = "DataSourceFileDriver";

}

import java.io.File;

import java.net.URI;

public abstract class DataSourceFileDriverParameters implements IDataSourceParameters{

	protected File source;

	public void setSource(File source){

		this.source = source;

	}

	public File getSource(){

		return this.source;

		dsm.registerDataSource(AbstractDataSourceFileDriver.DATASOURCE_NAME,

				AbstractDataSourceFileDriver.class,