Application: gvSIG desktop

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

import java.awt.geom.Rectangle2D;

import java.io.IOException;

import java.net.ConnectException;

import java.net.MalformedURLException;

import java.net.URL;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Hashtable;

import java.util.Map;

import org.geotools.data.DefaultQuery;

import org.geotools.data.FeatureReader;

import org.geotools.data.Query;

import org.geotools.data.Transaction;

import org.geotools.data.wfs.WFSDataStoreFactory;

import org.geotools.feature.AttributeType;

import org.gvsig.remoteClient.RemoteClient;

import org.gvsig.remoteClient.wfs.WFSStatus;

import org.xml.sax.SAXException;

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

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

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

/* 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

 */

/* CVS MESSAGES:

 *

 * $Id$

 * $Log$

 * Revision 1.1  2006-05-19 12:48:21  jorpiell

 * Driver que implementa la accesibilidad a geotools

 *

 *

 */

/**

 * @author Jorge Piera Llodr? (piera_jor@gva.es)

 */

public class WFSGeotoolsDriver extends RemoteClient{

	private DataStore dataStore = null;

	private ArrayList features = new ArrayList();

	private Rectangle2D fullExtent = null;

	private Hashtable hashRelate = new Hashtable();

	private WFSStatus currentStatus = null;

	private int numReg = -1;

	private int numField = -1;

	private AttributeType[] attributes;

	private WFSFeaturesIterator iterator = null;

	public WFSGeotoolsDriver(String host) throws ConnectException{

		this.setHost(host);

		if (!connect()){

			throw new ConnectException();	

		}		

	}

	/**

	 * The GetFeature operation allows retrieval of features from a 

	 * web feature service. A GetFeature request is processed by

	 * a WFS and when the value of the outputFormat attribute is 

	 * set to text/gml a GML instance document, containing the 

	 * result set, is returned to the client.

	 * @param status

	 * WFS client status. Contains all the information to create

	 * the query

	 * @return Object

	 * GML File

	 */

	public Object getFeature(WFSStatus status){

		try { 

			if (status != null){

				this.currentStatus = status;

			}

			Query query = new DefaultQuery(currentStatus.getFeatureName());

			FeatureReader featureReader;

			featureReader = dataStore.getFeatureReader(query,Transaction.AUTO_COMMIT);

			if (numField == -1) {

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

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

			}

			iterator = new WFSFeaturesIterator(featureReader);

			return iterator;

		} catch (IOException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} 

		return null;

	}

	/**

	 * Sets the retrieved features

	 * @param transformation

	 * @throws DriverException 

	 * @throws DriverException 

	 */

	public void setFeatures(WFSFeaturesIterator iterator) {

		int index = 0;

		while (iterator.hasNext()){

			IFeature feature;

			try {

				feature = iterator.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));

			} catch (DriverException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}			

			index++;

		}

		numReg = index;

	}

	/*

	 *  (non-Javadoc)

	 * @see org.gvsig.remoteClient.RemoteClient#connect()

	 */

	public boolean connect() {

		URL url;

		try {

			url = new URL(getCapabilitiesRequest());

			Map m = new HashMap();

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

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

			m.put(WFSDataStoreFactory.PROTOCOL.key,Boolean.TRUE);

			//dataStore = (new WFSDataStoreFactory()).createNewDataStore(m);

			dataStore = new DataStore(url, new Boolean(true), "", "", 1000000, 100);

			return true;

		} catch (MalformedURLException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (IOException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (SAXException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} 

		return false;         

	}

	/**

	 * @return the host name

	 */

    public String getCapabilitiesRequest() {        

    	StringBuffer req = new StringBuffer();

    	req.append(super.getHost()).append("?").append("REQUEST=GetCapabilities&SERVICE=WFS&");

    	req.append("&EXCEPTIONS=XML");

    	return req.toString();        

    } 

	/*

	 *  (non-Javadoc)

	 * @see org.gvsig.remoteClient.RemoteClient#close()

	 */

	public void close() {

		// TODO Auto-generated method stub

	}

	/**

	 * @return Returns the features.

	 */

	public ArrayList getFeatures() {

		return features;

	}

	/**

	 * @return Returns the hashRelate.

	 */

	public Hashtable getHashRelate() {

		return hashRelate;

	}

	/**

	 * @return Returns the fullExtent.

	 */

	public Rectangle2D getFullExtent() {

		return fullExtent;

	}

	/**

	 * @return Returns the numReg.

	 */

	public int getNumReg() {

		return numReg;

	}

	/**

	 * @return Returns the attributes.

	 */

	public AttributeType[] getAttributes() {

		return attributes;

	}

	/**

	 * @return Returns the numField.

	 */

	public int getNumField() {

		return numField;

	}

	/**

	 * @return Returns the iterator.

	 */

	public WFSFeaturesIterator getIterator() {

		return (WFSFeaturesIterator)getFeature(null);

	}

}