Application: gvSIG desktop

import java.awt.geom.Rectangle2D;

import java.net.MalformedURLException;

import java.net.URL;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import java.sql.Types;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Hashtable;

import java.util.Map;

import java.util.NoSuchElementException;

import org.geotools.data.DataStore;

import org.geotools.data.DefaultFIDReader;

import org.geotools.data.DefaultQuery;

import org.geotools.data.FeatureReader;

import org.geotools.data.FeatureSource;

import org.geotools.data.Query;

import org.geotools.data.Transaction;

import org.geotools.data.ows.BoundingBox;

import org.geotools.data.wfs.WFSDataStoreFactory;

import org.geotools.factory.FactoryConfigurationError;

import org.geotools.feature.AttributeType;

import org.geotools.feature.DefaultFeature;

import org.geotools.feature.Feature;

import org.geotools.feature.IllegalAttributeException;

import org.geotools.filter.BBoxExpressionImpl;

import org.geotools.filter.FidFilter;

import org.geotools.filter.FidFilterImpl;

import org.geotools.filter.Filter;

import org.geotools.filter.FilterFactory;

import org.geotools.filter.FilterFactoryImpl;

import org.geotools.filter.IllegalFilterException;

import org.postgis.PGbox3d;

import com.hardcode.gdbms.engine.data.driver.ObjectDriver;

import com.hardcode.gdbms.engine.values.StringValue;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.IFeature;

import com.iver.cit.gvsig.fmap.core.ShapeFactory;

import com.iver.cit.gvsig.fmap.core.v02.FConverter;

import com.iver.cit.gvsig.fmap.core.v02.FSymbol;

import com.iver.cit.gvsig.fmap.drivers.DriverAttributes;

import com.iver.cit.gvsig.fmap.drivers.IFeatureIterator;

import com.iver.cit.gvsig.fmap.drivers.jdbc.mysql.MySqlFeatureIterator;

import com.vividsolutions.jts.geom.Envelope;

public class WFSDriver implements IWFSDriver,ObjectDriver {

	private Rectangle2D fullExtent=null;

	private Rectangle2D extent=null;

	private DataStore ds;

	//private FeatureReader fr;

	private String urlS="http://localhost:8080/geoserver/wfs";

	private String versionS="version=1.0.0";

	private String requestS="request=getcapabilities";

	private String serviceS="service=wfs";

	private String userS="admin";

	private String pwdS="geoserver";

	private String[] layersS;

	private boolean protocolS=true;

	private int numFeaturesS=100;

	private int timeoutS=10000;

	private Hashtable hashRelate;

	///private Hashtable indexFID;

	protected int idFID_FieldName;

	private int numReg=-1;

	private int numField=-1;

	private AttributeType[] attributes;

	//private FeatureReader fr;

	///private ArrayList typeNames=new ArrayList();

	private ArrayList features=new ArrayList();

	public Rectangle2D getFullExtent() {

		return fullExtent;

	}

	public int getShapeType() {

		return FShape.MULTI;

	}

	private Filter createFidFilter(String[] fids) {

		 FidFilter filter = FilterFactory.createFilterFactory().createFidFilter();

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

	            filter.addFid(fids[i]);

	        }

        return filter;

    }

	public IGeometry getShape(int index) {

		return ((IFeature)features.get(index)).getGeometry();

	}

	public URL getHost() {

		try {

			return new URL(urlS+"?"+versionS+"&"+requestS+"&"+serviceS);

		} catch (MalformedURLException e) {

			e.printStackTrace();

		}

		return null;

	}

	public IFeatureIterator getFeatureIterator(Rectangle2D r, String strEPSG) throws DriverException {

		WFSFeatureIterator geomIterator = null;

		/*if (r!=null && extent.contains(r)){

			//extent=(Rectangle2D)r.clone();

			return new WFSFeatureIterator(features);

		}

		*/

	       // ArrayList shapes=new ArrayList();   

		/*	BBoxExpressionImpl RectangleFilter=null;

			try {

				RectangleFilter = (BBoxExpressionImpl)FilterFactoryImpl.createFilterFactory().createBBoxExpression(new Envelope(fullExtent.getX(),fullExtent.getY(),fullExtent.getMaxX(),fullExtent.getMaxY()));

			} catch (IllegalFilterException e2) {

				e2.printStackTrace();

			} catch (FactoryConfigurationError e2) {

				e2.printStackTrace();

			}

			*/

			   Query query = new DefaultQuery(layersS[0]);

			   FeatureReader fr=null;

				try {

					fr = ds.getFeatureReader(query,Transaction.AUTO_COMMIT);

					if (numField ==-1){

					numField=fr.getFeatureType().getAttributeCount();

					attributes=fr.getFeatureType().getAttributeTypes();

					}

				} catch (IOException e) {

					e.printStackTrace();

				}

				geomIterator=new WFSFeatureIterator(fr);

			/*   ResultSet rs = st.executeQuery(sql);

	            geomIterator = new MySqlFeatureIterator(rs);

	        } catch (SQLException e) {

	            e.printStackTrace();

	            throw new com.iver.cit.gvsig.fmap.DriverException(e);

	        }

	          */  

	        return geomIterator;

	}

	public void close() {

	}

	public void open() throws DriverException {

	}

	public int getRowIndexByFID(IFeature FID) {

		int resul;

	    Integer rowIndex = (Integer) hashRelate.get(FID.getID());

		resul = rowIndex.intValue();

		return resul;

	}

	public int getShapeCount() throws IOException {

		return numReg;

	}

	public DriverAttributes getDriverAttributes() {

		return null;

	}

	public String getName() {

		return "WFS Driver";

	}

	/**

     * Recorre el recordset creando una tabla Hash que usaremos para 

     * relacionar el n?mero de un registro con su identificador ?nico.

     * Debe ser llamado en el setData justo despu?s de crear el recorset

     * principal

     */

    protected void doRelateID_FID()

    {

		if (ds==null){

			URL url=null;

			try {

				url = new URL(urlS+"?"+versionS+"&"+requestS+"&"+serviceS);

			} catch (MalformedURLException e) {

				e.printStackTrace();

			}

			//int timeout=10000;

			//boolean protocol=true;

         Map m = new HashMap();

         m.put(WFSDataStoreFactory.URL.key,url);

         m.put(WFSDataStoreFactory.TIMEOUT.key,new Integer(timeoutS));

         m.put(WFSDataStoreFactory.PROTOCOL.key,new Boolean(protocolS));

		 m.put(WFSDataStoreFactory.PASSWORD.key,pwdS);

		 m.put(WFSDataStoreFactory.USERNAME.key,userS);

		 m.put(WFSDataStoreFactory.BUFFER_SIZE.key,new Integer(numFeaturesS));

        try {

			ds = (new WFSDataStoreFactory()).createNewDataStore(m);

		} catch (IOException e) {

			e.printStackTrace();

		}

		/*

		String[] names=null;

		try {

			names = ds.getTypeNames();

		} catch (IOException e1) {

			e1.printStackTrace();

		}

		System.out.println("Capas :");

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

			System.out.println("Capa "+i+" : "+names[i]);

			//Se selecciona la capa que se quiere abrir

			//if (true)//si esta seleccionada en la lista o arbol que se muestre en el wizard

			//typeNames.add(names[i]);

		}

		*/

	}

        hashRelate = new Hashtable();

		///indexFID = new Hashtable();

		Query query = new DefaultQuery(layersS[0]);

		//FeatureReader fr=null;

			//fr = ds.getFeatureReader(query,Transaction.AUTO_COMMIT);

			int index = 0;

            IFeatureIterator iter=null;

			try {

				iter = getFeatureIterator(null,null);

			} catch (DriverException e) {

				e.printStackTrace();

			}

			try {

				while(iter.hasNext()){

					IFeature feature=iter.next();

					features.add(feature);

					Rectangle2D rect=feature.getGeometry().getBounds2D();

					if (fullExtent==null){

						fullExtent=new Rectangle2D.Double(rect.getMinX(),rect.getMinY(),rect.getWidth(),rect.getHeight());

					}else{

						fullExtent.add(new Rectangle2D.Double(rect.getMinX(),rect.getMinY(),rect.getWidth(),rect.getHeight()));

					}

					hashRelate.put(feature.getID(), new Integer(index));

					///indexFID.put(new Integer(index),feature.getID());

					index++;

				}

			} catch (SQLException e) {

				e.printStackTrace();

			}

			numReg = index;

			extent=fullExtent;

    }

	public void setData(String url,String user,String pwd,String[] layers,boolean protocol,int numFeatures,int timeout,int id_FID_field) {

		idFID_FieldName=id_FID_field;

		urlS=url;

		userS=user;

		pwdS=pwd;

		layersS=layers;

		protocolS=protocol;

		numFeaturesS=numFeatures;

		timeoutS=timeout;

		doRelateID_FID();

	}

	public Value getFieldValue(long arg0, int arg1) throws com.hardcode.gdbms.engine.data.driver.DriverException {

		/*try {

			ds.getTypeNames();

		} catch (IOException e1) {

			e1.printStackTrace();

		}*/

		/*System.out.println("Filter");

		Filter fidFilter=createFilter(new String[]{(String)indexFID.get(new Integer((int)arg0))});

		System.out.println("Query");

		Query query = new DefaultQuery((String)typeNames.get(0),fidFilter);

		FeatureReader fr=null;

		try {

			System.out.println("getFeatureReader");

			fr = ds.getFeatureReader(query,Transaction.AUTO_COMMIT);

		} catch (IOException e) {

			e.printStackTrace();

		}

			try {

				return getValue(fr,arg1);

			} catch (NoSuchElementException e) {

				e.printStackTrace();

			} catch (IOException e) {

				e.printStackTrace();

			} catch (IllegalAttributeException e) {

				e.printStackTrace();

			}

			*/

		return ((IFeature)features.get((int)arg0)).getAttribute(arg1+1);

	}

	public int getFieldCount() throws com.hardcode.gdbms.engine.data.driver.DriverException {

		return numField-1;

	}

	public String getFieldName(int arg0) throws com.hardcode.gdbms.engine.data.driver.DriverException {

		return attributes[arg0].getName();		

	}

	public long getRowCount() throws com.hardcode.gdbms.engine.data.driver.DriverException {

		return numReg;

	}

	public int getFieldType(int arg0) throws com.hardcode.gdbms.engine.data.driver.DriverException {

		attributes[arg0].getType();

		return FIELD_TYPE_STRING;//fr.getFeatureType().getAttributeType(arg0).getType();

	}