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

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.net.ConnectException;

import java.net.ProtocolException;

import java.net.URL;

import java.util.ArrayList;

import java.util.TreeMap;

import java.util.Vector;

import javax.imageio.ImageIO;

import org.gvsig.remoteClient.exceptions.ServerErrorException;

import org.gvsig.remoteClient.wms.WMSClient;

import org.gvsig.remoteClient.wms.WMSDimension;

import org.gvsig.remoteClient.wms.WMSLayer;

import org.gvsig.remoteClient.wms.WMSStatus;

import org.gvsig.remoteClient.wms.WMSStyle;

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

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

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

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

//import com.iver.wmsclient.Capabilities;

//import com.iver.wmsclient.FeatureInfoQuery;

//import com.iver.wmsclient.MapQuery;

//import com.iver.wmsclient.UnsupportedVersionException;

//import com.iver.wmsclient.WMSClient;

//import com.iver.wmsclient.WMSClientFactory;

/**

 * Driver WMS.

 *

 * @author Vicente Caballero Navarro

 */

public class FMapWMSDriver implements WMSDriver {

        private WMSClient client;

    private WMSLayerNode fmapRootLayer;

    private TreeMap layers = new TreeMap();

        /**

         * @see com.iver.cit.gvsig.fmap.drivers.WMSDriver#getCapabilities(java.net.URL)

         */

        public void getCapabilities(URL server)

                throws WMSException {

                try {

                        getClient(server).connect();

                } catch (Exception e) {                        

                        throw new WMSException(e);

                }

        }

        /**

     * @deprecated

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

         */

        public Image _getMap(WMSStatus wmsstatus)

                throws WMSException, IOException, ProtocolException {

                BufferedImage image = new BufferedImage(wmsstatus.getWidth(),

                                wmsstatus.getHeight(), BufferedImage.TYPE_INT_ARGB);

                try {

                        byte[] res = client.getMap(wmsstatus);                        

                        ByteArrayInputStream inbytes = new ByteArrayInputStream(res);

                        image = ImageIO.read(inbytes);

                        return image;

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

            throw new WMSException(e);

        } catch (ServerErrorException e) {

            throw new WMSException(e);

        }

    }

    public byte[] getMap(WMSStatus status) throws WMSException {

        try {

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

        }

    }

        /**

         * Devuelve WMSClient a partir de su URL.

         *

         * @param url URL.

         *

         * @return WMSClient.

         * @throws IOException 

         * @throws ConnectException 

         *

         * @throws UnsupportedVersionException

         * @throws IOException

         */

        private WMSClient getClient(URL url) throws ConnectException, IOException {

                if (client == null) {

                        client = new WMSClient(url.toString());

                }

                return client;

        }

    /**

     * Creates a new instance of a WMSClient.

     * @param host

     * @throws IOException 

     * @throws ConnectException 

     */

    public void createClient(URL host) throws ConnectException, IOException{

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

    }

    /**

     * Establishes the connection.

     * @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() {

        return client.connect();

    }

    /**

     * @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;

    }

    /**

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

     */

    public Vector getFormats() {

        return client.getFormats();

    }

    /**

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

     */

    public WMSLayerNode getLayersTree() {

        if (fmapRootLayer == null){

            WMSLayer clientRoot = client.getLayersRoot();

            fmapRootLayer = parseTree(clientRoot, null);

        }

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

        myNode.setName(node.getName());

        myNode.setTitle(node.getTitle());

        myNode.setTransparency(node.hasTransparency());

        myNode.setSrs(node.getAllSrs());

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

        myNode.setParent(parentNode);

        myNode.setAbstract(node.getAbstract());

        if (node.getLatLonBox()!=null)

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

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

        }

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

        }

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

    }

}