| gvSIG desktop 1

package org.gvsig.remoteClient.wms;

import java.awt.geom.Rectangle2D;

import java.util.TreeMap;

import java.util.Vector;

import org.gvsig.remoteClient.exceptions.WMSException;

import org.gvsig.remoteClient.utils.BoundaryBox;

import org.gvsig.remoteClient.wms.wms_1_1_1.WMSProtocolHandler1_1_1;

/**

 * <p>Represents the class the with the necessary logic to connect to a OGCWMS and interpretate the data </p>

 * 

 */

public class WMSClient extends org.gvsig.remoteClient.RasterClient {

    private org.gvsig.remoteClient.wms.WMSProtocolHandler handler;

    private TreeMap layers = new TreeMap();

    private WMSLayer rootLayer;

    /**

     * @return Returns the rootLayer.

     */

    public WMSLayer getRootLayer() {

        return rootLayer;

    }

    /**

     * @param rootLayer The rootLayer to set.

     */

    public void setRootLayer(WMSLayer rootLayer) {

        this.rootLayer = rootLayer;

    }

    /**

     * Constructor.

     * the parameter host, indicates the WMS host to connect.

     * */

    public WMSClient(String host)

    {

        //m_log = LogManager.getLogManager().getLogger(getClass().getName());

        setHost(host);

        /*

          The negotiation is performed in the connect method.

         try

        {

            //TODO: implement a correct negotiation algorithm

            //handler = WMSProtocolHandlerFactory.negotiate(new URL(host));

            handler = new WMSProtocolHandler1_1_1();

            handler.setHost(host);

        }

        catch(Exception e)

        {

            e.printStackTrace();

        }

        */

        handler = new WMSProtocolHandler1_1_1();

        handler.setHost(host);

    }

    public String getVersion()

    {

        return handler.getVersion();

    }

    /**

     * <p>One of the three interfaces that OGC WMS defines. Request a map.</p> 

     */

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

        return handler.getMap(status);

    } 

    /**

     * <p>One of the three interfaces defined by OGC WMS, it gets the service capabilities</p>

     * 

     */

    public void getCapabilities() {        

        handler.getCapabilities();

        layers = handler.layers;

        rootLayer = handler.rootLayer;

    } 

    /**

     * <p>One of the three interfaces defined by the OGC WMS, it gets the information about a feature requested</p>

     * @return 

     */

    public String getFeatureInfo(WMSStatus status, int x, int y, int featureCount) throws WMSException{        

        return handler.getFeatureInfo(status, x, y, featureCount);

    } 

    /**

     * <p> Reads from the WMS Capabilities, the layers available in the service</p>

     * @return a TreeMap with the available layers in the WMS 

     */

    public TreeMap getLayers() {        

        return layers;

    } 

    /**

     * <p>Reads from the WMS Capabilities the number if layers available in the service</p>

     * @return, number of layers available

     */

    public int getNumberOfLayers() {        

        if (layers != null)

        {

            return layers.size();

        }

        return 0;

    } 

    /**

     * <p>Gets the WMSLayer with this name</p>

     * 

     * @param _name, layer name

     * @return the layer with this name

     */

    public WMSLayer getLayer(String _name) {        

        if (layers.get(_name) != null)

        {

            return (WMSLayer)layers.get(_name);

        }

        return null;

    } 

    public String[] getLayerNames()

    {            

        WMSLayer[] lyrs;

        lyrs = (WMSLayer[])layers.values().toArray(new WMSLayer[0]);

        String[] names = new String[lyrs.length];

        for(int i = 0; i<lyrs.length; i++)

        {

            names[i] = ((WMSLayer)lyrs[i]).getName();

        }

        return names;

    }

//  /**

//  * Adds a Layer object to the vector of layers of this LayerSet

//  */

//  public void addLayer(WMSLayer layer)

//  {

//  String layerName = layer.getName();

//  

//  if (layerName == null)

//  {

//  throw new RuntimeException("Error: attempt to add a nameless layer");

//  }

//  

//  // We use the lower case version of the name as a key for the hash map.

//  layerName = layerName.toLowerCase();    

//  if (layers == null)

//  {

//  layers = new TreeMap();

//  }

//  layers.put(layerName, layer);

//  }

    /**

     * <p>Gets the image formats available in the Service to retrieve the maps</p>

     * @return a vector with all the available formats

     */

    public Vector getFormats() {        

        return handler.getServiceInformation().formats;         

    } 

    /**

     * <p>Gets the SRS availables in the Service, all the SRS the Service can reproject to.</p>

     * @return vector with all the available SRS.

     */

    public Vector getSrs() {        

        return handler.srs;

    } 

    public void close() {        

        // your code here

    } 

    /**

     * Returns the max extent that envolves the requested layers

     * */

    public Rectangle2D getLayersExtent(String[]layerNames, String srs)

    {

        try

        {

            BoundaryBox bbox;

            WMSLayer layer = getLayer(layerNames[0]);

            bbox = layer.getBbox(srs);

            double xmin = bbox.getXmin();

            double xmax = bbox.getXmax();

            double ymin = bbox.getYmin();

            double ymax = bbox.getYmax();

            for(int i=1; i<layerNames.length; i++)

            {

                layer = getLayer(layerNames[i]);

                bbox = layer.getBbox(srs);

                if (bbox.getXmin() < xmin)

                {

                    xmin = bbox.getXmin();

                }

                if (bbox.getYmin() < ymin)

                {

                    ymin = bbox.getYmin();

                }

                if (bbox.getXmax() > xmax)

                {

                    xmax = bbox.getXmax();

                }

                if (bbox.getYmax() > ymax)

                {

                    ymax = bbox.getYmax();

                }

            }        

            Rectangle2D extent = new Rectangle2D.Double(xmin,ymin,Math.abs(xmax-xmin),Math.abs(ymax-ymin));

            return extent;

        }

        catch(Exception e)

        {

            e.printStackTrace();

            return null;

        }

    }

    /**

     * Gets the Service information included in the Capabilities

     * */

    public WMSProtocolHandler.ServiceInformation getServiceInformation()

    {

        return handler.getServiceInformation();  

    }

    /**

     * <p>Checks the connection to de remote WMS and requests its capabilities.</p>

     * 

     */

    public boolean connect() 

    {

        try {

            getCapabilities();

            if (handler == null)

            {

                if (getHost().trim().length() > 0)

                {                                        

                    //TODO: Implement correctly the negotiate algorithm

                    //handler = WMSProtocolHandlerFactory.negotiate(new URL(getHost()));

                    handler = new WMSProtocolHandler1_1_1();

                    handler.setHost(getHost());

                }

                else

                {

                    //must to specify host first!!!!

                    return false;

                }

            }

            return true;

        } catch (Exception e) {

            e.printStackTrace();

            return false;

        }

    }

    //TODO Check this out: Always 1 layer at first level...

    public WMSLayer getLayersRoot() {

        return rootLayer;

    }

}