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

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.net.ConnectException;

import java.net.URL;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.Iterator;

import java.util.TreeMap;

import java.util.Vector;

import org.gvsig.remoteClient.exceptions.ServerErrorException;

import org.gvsig.remoteClient.taskplanning.retrieving.RetrieveEvent;

import org.gvsig.remoteClient.wms.WMSClient;

import org.gvsig.remoteClient.wms.WMSDimension;

import org.gvsig.remoteClient.wms.WMSEventListener;

import org.gvsig.remoteClient.wms.WMSLayer;

import org.gvsig.remoteClient.wms.WMSStatus;

import org.gvsig.remoteClient.wms.WMSStyle;

import org.gvsig.remoteClient.wms.WMSProtocolHandler.ServiceInformation;

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

import com.iver.cit.gvsig.fmap.layers.WMSLayerNode;

/**

 * Driver WMS.

 *

 * @author Jaume Dominguez Faus

 */

public class FMapWMSDriver implements WMSEventListener {

        private WMSClient client;

    private WMSLayerNode fmapRootLayer;

    private TreeMap layers = new TreeMap();

    private Vector listeners = new Vector();

    public void newEvent(int idRequest, RetrieveEvent event) {

            int eventType = event.getType();

            if (idRequest == WMSOperations.GET_CAPABILITIES) {

                    if (eventType == WMSOperations.REQUEST_FINISHED) {

                            WMSLayer clientRoot = client.getLayersRoot();

                            fmapRootLayer = parseTree(clientRoot, null);

                    }

                    for (int i = 0; i < listeners.size(); i++) {

                            WMSOperations l = (WMSOperations) listeners.get(i);

                            l.getCapabilities(eventType, event.getMessage());

                    }

            } else if (idRequest == WMSOperations.GET_MAP) {

                    String fileName = event.getFileName();

                    System.out.println("GetMap al driver");

                    for (int i = 0; i < listeners.size(); i++) {

                            System.out.println("getMap enviat al listener: "+i);

                            WMSOperations l = (WMSOperations) listeners.get(i);

                            l.getMap(eventType, new File(fileName), event.getMessage());

                    }

            }

    }

        FMapWMSDriver(URL host) {

                try {

                        client = new WMSClient(host.toString(), this);

                } catch (ConnectException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

        }

        /*

         *  (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.drivers.WMSDriver#getMap(org.gvsig.remoteClient.wms.WMSStatus)

         */

    public void getMap(WMSStatus status) throws WMSException {

        try {

            client.getMap(status);

        } catch (org.gvsig.remoteClient.exceptions.WMSException e) {

            throw new WMSException(e.getMessage());

        } catch (ServerErrorException e) {

            throw new WMSException("WMS Unexpected server error."+e.getMessage());      

        }

    }

    /**

     * Establishes the connection.

     * @param override, if true the previous downloaded data will be overridden

     * @return <b>true</b> if the connection was successful, or <b>false</b> if it was no possible

     * to establish the connection for any reason such as version negotiation.

     */

    public boolean connect(boolean override) {

        return client.connect(override);

    }

    public boolean connect() {

        return client.connect(false);

    }

    /**

     * @return the version of this client.

     */

    public String getVersion() {

        return client.getVersion();

    }

    /**

     * @return The title of the service offered by the WMS server.

     */

    public String getServiceTitle() {

        return client.getServiceInformation().title;

    }

    /**

     * Returns a Hash table containing the values for each online resource.

     * Using as key a String with name of the WMS request and the value returned

     * by the hash is another string containing the corresponding Url  

     * @return HashTable

     */

    public Hashtable getOnlineResources() {

            Hashtable onlineResources = new Hashtable();

            ServiceInformation si = client.getServiceInformation();

            Iterator it = si.operations.keySet().iterator();

            while (it.hasNext()) {

                    String key = (String) it.next();

                    String val = (String) si.operations.get(key);

                    if (val==null && (si.online_resource!=null || si.online_resource!= ""))

                            val = si.online_resource;

                    if (val!=null) {

                            onlineResources.put(key, val);

                    }

            }

            return onlineResources;

    }

    /**

     * @return <b>Vector</b> containing strings with the formats supported by this server.

     */

    public Vector getFormats() {

        return client.getFormats();

    }

    /**

     * @return <b>Boolean</b> returns if the WMS is queryable (suppports GetFeatureInfo)

     */

    public boolean isQueryable() {

        return client.isQueryable();

    }

    /**

     * @return A tree containing the info of all layers available on this server. 

     */

    public WMSLayerNode getLayersTree() {

        return fmapRootLayer;

    }

    /**

     * Parses the client's layer node and translates it to a fmap's layer node

     * @param WMSLayer

     * @return WMSLayerNode

     */

    private WMSLayerNode parseTree(WMSLayer node, WMSLayerNode parentNode) {

        WMSLayerNode myNode = new WMSLayerNode();

        // Name

        myNode.setName(node.getName());

        // Title

        myNode.setTitle(node.getTitle());

        // Transparency

        myNode.setTransparency(node.hasTransparency());

        // SRS

        myNode.setSrs(node.getAllSrs());

        // Queryable

        myNode.setQueryable(node.isQueryable() && client.getServiceInformation().isQueryable());

        // Parent layer

        myNode.setParent(parentNode);

        // Abstract

        myNode.setAbstract(node.getAbstract());

        // Fixed Size

        myNode.setFixedSize(node.getfixedWidth(), node.getfixedHeight());

        // LatLonBox

        if (node.getLatLonBox()!=null)

            myNode.setLatLonBox(node.getLatLonBox().toString());

        // Styles

        ArrayList styles = node.getStyles();

        for (int i = 0; i < styles.size(); i++) {

            WMSStyle style = (WMSStyle) styles.get(i);

            myNode.addStyle(style.getName(), style.getTitle(), style.getAbstract());

        }

        // Dimensions

        ArrayList dimensions = node.getDimensions();

        for (int i = 0; i < dimensions.size(); i++) {

            WMSDimension d = (WMSDimension) dimensions.get(i);

            myNode.addDimension(d.getName(), d.getUnits(), d.getUnitSymbol(), d.getDimensionExpression());

        }

        // Childs

        int children = node.getChildren().size();

        myNode.setChildren(new ArrayList());

        for (int i = 0; i < children; i++) {

            myNode.getChildren().add(parseTree((WMSLayer)node.getChildren().get(i), myNode));

        }

        if (myNode.getName()!=null)

            layers.put(myNode.getName(), myNode);

        return myNode;

    }

    /**

     * @return

     */

    public String getAbstract() {

        return client.getServiceInformation().abstr;

    }

    /**

     * @param layerName

     * @param srs

     * @return

     */

    public Rectangle2D getLayersExtent(String[] layerName, String srs) {

        return client.getLayersExtent(layerName, srs);

    }

    /**

     * @param string

     * @return

     */

    public WMSLayerNode getLayer(String layerName) {

        if (getLayers().get(layerName) != null)

        {

            return (WMSLayerNode)layers.get(layerName);

        }

        return null;

    }

    /**

     * @return

     */

    private TreeMap getLayers() {

        if (fmapRootLayer == null){

            fmapRootLayer = getLayersTree();

        }

        return layers;

    }

    /**

     * @param wmsStatus

     * @param i

     * @param j

     * @param max_value

     * @return

     * @throws WMSException 

     */

    public String getFeatureInfo(WMSStatus _wmsStatus, int i, int j, int max_value) throws WMSException {

        try {

            return client.getFeatureInfo(_wmsStatus, i, j, max_value);

        } catch (org.gvsig.remoteClient.exceptions.WMSException e) {

            throw new WMSException();

        }

    }

    public String getHost(){

        return client.getHost();

    }

    public void addListener(WMSOperations listener) {

            if (!listeners.contains(listener))

                    listeners.add(listener);

    }

        public void cancel(int opCode) {

                client.cancel(WMSOperations.GET_CAPABILITIES);

        }

}