Application: gvSIG desktop

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

    /**

     * Constructor.

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

     * */

        public WMSClient(String host)

        {

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

                setHost(host);

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

                }                

        }

        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;

    } 

/**

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

            }

            else

            {

                    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)

        {

                //TODO:

                //Implement logic to handle all the exceptions

                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 (WMSLayer)layers.values().toArray()[0];

    }

}