| gvSIG desktop 1

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

	}

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

*

* Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

*

* 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

*

* For more information, contact:

*

*  Generalitat Valenciana

*   Conselleria d'Infraestructures i Transport

*   Av. Blasco Ib??ez, 50

*   46010 VALENCIA

*   SPAIN

*

*      +34 963862235

*   gvsig@gva.es

*      www.gvsig.gva.es

*

*    or

*

*   IVER T.I. S.A

*   Salamanca 50

*   46005 Valencia

*   Spain

*

*   +34 963163400

*   dac@iver.es

*/

package com.iver.cit.gvsig.fmap.drivers.wfs;

import java.io.IOException;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.Date;

import java.util.NoSuchElementException;

import org.geotools.data.FeatureReader;

import org.geotools.feature.DefaultFeature;

import org.geotools.feature.Feature;

import org.geotools.feature.IllegalAttributeException;

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

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

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

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

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

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

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

public class WFSFeatureIterator implements IFeatureIterator{

	//private IFeature[] features=null;

	private FeatureReader featureReader;

	//private ArrayList features;

	//private int index=-1;

	//private int pos=0;

	public WFSFeatureIterator(FeatureReader fr){

		featureReader=fr;

	}

	/*public WFSFeatureIterator(ArrayList l){

		features=l;

	}*/

	public boolean hasNext(){

		/*if (features !=null){

			if (index<features.size()-1)

				return true;

			else{

				index=-1;

				features=null;

				return false;

			}

		}*/

		try {

			if (featureReader.hasNext()){

				return true;

			}else{

				featureReader.close();

				return false;

			}

		} catch (IOException e) {

			e.printStackTrace();

		}

		return false;

	}

	public IFeature next() throws SQLException {

	/*	if (features !=null){

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

		}*/

		Feature feature=null;

		try {

			feature = featureReader.next();

		} catch (NoSuchElementException e) {

			e.printStackTrace();

		} catch (IOException e) {

			e.printStackTrace();

		} catch (IllegalAttributeException e) {

			e.printStackTrace();

		}

		Object[] attr=null;

		attr=feature.getAttributes(attr);

		Value[] values=getValues(attr);

		IGeometry g=ShapeFactory.createGeometry(FConverter.jts_to_java2d(feature.getDefaultGeometry()));

		return new com.iver.cit.gvsig.fmap.core.DefaultFeature(g,values,feature.getID());

	}

	private Value[] getValues(Object[] attr) {

		Value[] values=new Value[attr.length];

		for (int i=1;i<attr.length;i++){

			if (attr[i]!=null){

				if (attr[i] instanceof Double){

					values[i]=ValueFactory.createValue(((Double)attr[i]).doubleValue());

				}else if (attr[i] instanceof String){

					values[i]=ValueFactory.createValue(String.valueOf(attr[i]));

				}else if (attr[i] instanceof Long){

					values[i]=ValueFactory.createValue(((Long)attr[i]).longValue());

				}else if (attr[i] instanceof Integer){

					values[i]=ValueFactory.createValue(((Integer)attr[i]).intValue());

				}else if (attr[i] instanceof Float){

					values[i]=ValueFactory.createValue(((Float)attr[i]).floatValue());

				}else if (attr[i] instanceof Short){

					values[i]=ValueFactory.createValue(((Short)attr[i]).shortValue());

				}else if (attr[i] instanceof Boolean){

					values[i]=ValueFactory.createValue(((Boolean)attr[i]).booleanValue());

				}else if (attr[i] instanceof Date){

					values[i]=ValueFactory.createValue(((Date)attr[i]));

				}

			//	System.out.println("class = "+attr[i].getClass());

			}else{

				values[i]=ValueFactory.createValue("");

			}

		}

		return values;

	}

}