Application: gvSIG desktop

/* 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.awt.geom.Rectangle2D;

import java.io.IOException;

import java.net.MalformedURLException;

import java.net.ProtocolException;

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.IGeometry;

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.VectorialDriver;

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

        }

}