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: WFSGeotoolsDriver.java 5291 2006-05-19 12:48:21Z jorpiell $

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

        }

}