Application: gvSIG desktop

package com.iver.wmsclient.wms_1_3_0;

import com.iver.utiles.StringUtilities;

import com.iver.wmsclient.Capabilities;

import com.iver.wmsclient.FeatureInfoQuery;

import com.iver.wmsclient.MapQuery;

import com.iver.wmsclient.UnsupportedVersionException;

import com.iver.wmsclient.WMSException;

import com.iver.wmsclient.WMSUtilities;

import com.iver.wmsclient.WMSVersionClient;

import com.iver.wmsclient.wms_1_3_0.capabilities.Format;

import com.iver.wmsclient.wms_1_3_0.capabilities.BoundingBox;

import com.iver.wmsclient.wms_1_3_0.capabilities.DCPType;

import com.iver.wmsclient.wms_1_3_0.capabilities.GetFeatureInfo;

import com.iver.wmsclient.wms_1_3_0.capabilities.Layer;

import com.iver.wmsclient.wms_1_3_0.capabilities.WMS_Capabilities;

import com.iver.wmsclient.wms_1_3_0.exception.ServiceException;

import com.iver.wmsclient.wms_1_3_0.exception.ServiceExceptionReport;

import org.apache.log4j.Logger;

import org.exolab.castor.xml.MarshalException;

import org.exolab.castor.xml.Unmarshaller;

import org.exolab.castor.xml.ValidationException;

import java.awt.geom.Rectangle2D;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.net.URL;

import java.util.ArrayList;

/**

 * Cliente para la versi?n 1.3.0

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class WMS_1_3_0 implements WMSVersionClient {

    private static Logger logger = Logger.getLogger(WMS_1_3_0.class.getName());

    private String[] mapURLs;

    private String[] infoURLs;

    private WMS_Capabilities root;

    /**

     * @see com.iver.cit.gvsig.fmap.wms.wmsclient.WMSClient#getCapabilities(java.net.URL)

     */

    public Capabilities getCapabilities(URL host)

        throws UnsupportedVersionException, IllegalStateException, IOException {

        InputStream in = null;

        String response = null;

        try {

            logger.debug("Obteniendo capabilities...");

            String hostString = host.toString();

            if (!hostString.endsWith("?")) {

                hostString += "?";

            }

            String request = getQueryHeader() + "SERVICE=WMS&request=getCapabilities";

            //String request = "REQUEST=GetCapabilities&SERVICE=wms";

            URL query = new URL(hostString + request);

            in = query.openStream();

            ByteArrayOutputStream outbytes = new ByteArrayOutputStream();

            WMSUtilities.serializar(in, outbytes);

            response = new String(outbytes.toByteArray());

            logger.debug(response);

            logger.debug("Parseando la respuesta...");

            try {

                Unmarshaller umn = new Unmarshaller(WMS_Capabilities.class);

                umn.setIgnoreExtraElements(true);

                umn.setIgnoreExtraAttributes(true);

                root = (WMS_Capabilities) umn.unmarshal(new InputStreamReader(

                            new ByteArrayInputStream(WMSUtilities.eliminarDTD(

                                    outbytes.toByteArray(), "WMS_Capabilities"))));

            } catch (MarshalException e) {

                Capabilities c = new Capabilities();

                c.setVersion(null);

                return c;

            }

            //                        LocalConfiguration.getInstance().getProperties().setProperty("org.exolab.castor.xml.strictelements", "false");

            /*                        WMT_MS_Capabilities root = (WMT_MS_Capabilities) WMT_MS_Capabilities.unmarshal(new InputStreamReader(

               new ByteArrayInputStream(WMSUtilities.eliminarDTD(

                               outbytes.toByteArray()))));

             */

            DCPType[] DCPs = root.getCapability().getRequest().getGetMap()

                                 .getDCPType();

            mapURLs = new String[DCPs.length];

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

                mapURLs[i] = DCPs[i].getHTTP().getGet().getOnlineResource()

                                    .getHref();

                if (!mapURLs[i].endsWith("?")) {

                    mapURLs[i] = mapURLs[i] + "?";

                }

            }

            GetFeatureInfo fi = root.getCapability().getRequest()

                                    .getGetFeatureInfo();

            if (fi != null) {

                DCPs = fi.getDCPType();

                infoURLs = new String[DCPs.length];

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

                    infoURLs[i] = DCPs[i].getHTTP().getGet().getOnlineResource()

                                         .getHref();

                    if (!infoURLs[i].endsWith("?")) {

                        infoURLs[i] = infoURLs[i] + "?";

                    }

                }

            }

            Capabilities ret = new Capabilities();

            ret.setRoot(root);

            ret.setVersion(root.getVersion());

            return ret;

        } catch (IOException e) {

            logger.error("Error", e);

            throw e;

        } catch (Exception e) {

            Capabilities ret = new Capabilities();

            if (response == null) {

                ret.setRoot(null);

                ret.setVersion(null);

                return ret;

            } else {

                ret.setRoot(null);

                ret.setVersion(StringUtilities.substringDelimited(response,

                        "version=\"", "\"", response.indexOf("bilities")));

                return ret;

            }

        } finally {

            if (in != null) {

                in.close();

            }

        }

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.wmsclient.WMSClient#doMapQuery(com.iver.cit.gvsig.fmap.wms.wmsclient.MapQuery)

     */

    public byte[] doMapQuery(MapQuery queryInfo)

        throws IOException, NoSuchFieldException, ValidationException, 

            WMSException {

        throw new RuntimeException("No debio llegar aqu?");

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.wmsclient.WMSClient#getVersion()

     */

    public String getVersion() {

        return "1.3.0";

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.wmsclient.WMSClient#getBoundingBox(javax.swing.tree.TreePath,

     *      java.lang.String)

     */

    public Rectangle2D getBoundingBox(String[] t, String srs) {

        Layer[] layer = getLayerByPath(t);

        BoundingBox[] bbox = getBoundingBox(layer);

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

            if (bbox[i].getCRS().equals(srs)) {

                return new Rectangle2D.Double(bbox[i].getMinx(),

                    bbox[i].getMiny(), bbox[i].getMaxx() - bbox[i].getMinx(), bbox[i].getMaxy() - bbox[i].getMiny());

            }

        }

        /*

           LatLonBoundingBox[] llbbox = getLatLonBoundingBox(layer);

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

                           return new Rectangle2D.Double(Double.parseDouble(

                           llbbox[i].getMinx()),

                                   Double.parseDouble(llbbox[i].getMiny()),

                                   Double.parseDouble(llbbox[i].getMaxx()),

                                   Double.parseDouble(llbbox[i].getMaxy()));

           }

         */

        return null;

    }

    /**

     * Dado un array de layers devuelve el array de boundingbox de  todos ellos

     *

     * @param layer Array de layers

     *

     * @return array de boundingBox

     */

    private BoundingBox[] getBoundingBox(Layer[] layer) {

        ArrayList globalBbox = new ArrayList();

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

            Layer l = (Layer) layer[i];

            BoundingBox[] bboxes = l.getBoundingBox();

            if (bboxes != null) {

                for (int j = 0; j < bboxes.length; j++) {

                    globalBbox.add(bboxes[j]);

                }

            }

        }

        return (BoundingBox[]) globalBbox.toArray(new BoundingBox[0]);

    }

    /*

     * Dado un array de layers devuelve el array de boundingbox de

     * todos ellos

     *

     * @param layers Array de layers

     *

     * @return array de boundingBox

     *

           private LatLonBoundingBox[] getLatLonBoundingBox(Layer[] layer) {

                   ArrayList globalBbox = new ArrayList();

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

                           Layer l = (Layer) layer[i];

                           LatLonBoundingBox bbox = l.getLatLonBoundingBox();

                           if (bbox != null) globalBbox.add(bbox);

                   }

                   return (LatLonBoundingBox[]) globalBbox.toArray(new LatLonBoundingBox[0]);

           }

     */

    /**

     * Obtiene dado un TreePath con las capas devuelve un array con los objetos

     * Layer desde el inicio al final del path

     *

     * @param t TreePath

     *

     * @return Array de layers

     */

    private Layer[] getLayerByPath(String[] t) {

        ArrayList ret = new ArrayList();

        Layer raiz = root.getCapability().getLayer();

        ret.add(raiz);

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

            for (int j = 0; j < raiz.getLayerCount(); j++) {

                    if (raiz.getLayer(j).getTitle() == null) continue;

                if (raiz.getLayer(j).getTitle().equals(t[i])) {

                    raiz = raiz.getLayer(j);

                    ret.add(raiz);

                    break;

                }

            }

        }

        return (Layer[]) ret.toArray(new Layer[0]);

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.wmsclient.WMSClient#doFeatureInfo(com.iver.cit.gvsig.fmap.wms.wmsclient.FeatureInfoQuery)

     */

    public byte[] doFeatureInfo(FeatureInfoQuery query)

        throws ValidationException, IOException, WMSException, 

            NoSuchFieldException {

        throw new RuntimeException("No debio llegar aqu?");

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.WMSVersionClient#getMapURLs()

     */

    public String[] getMapURLs() {

        return mapURLs;

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.WMSVersionClient#parseException(byte[])

     */

    public String parseException(byte[] bytes) {

        try {

            String error = "";

            ServiceExceptionReport ex = (ServiceExceptionReport) ServiceExceptionReport.unmarshal(new InputStreamReader(

                        new ByteArrayInputStream(bytes)));

            ServiceException[] exs = ex.getServiceException();

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

                if (exs[i].getCode() != null) {

                    error = error + exs[i].getCode() + ": ";

                }

                error += (exs[i].getContent() + "\n");

            }

            return error;

        } catch (Exception e) {

            return null;

        }

    }

    /**

     * @see com.iver.cit.gvsig.fmap.wms.WMSVersionClient#getInfoURLs()

     */

    public String[] getInfoURLs() {

        return infoURLs;

    }

        /**

         * @see com.iver.wmsclient.WMSClient#getInfoFormats()

         */

        public String[] getInfoFormats() {

                Format[] formats = root.getCapability().getRequest().getGetFeatureInfo().getFormat();

                String[] ret = new String[formats.length];

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

                        ret[i] = formats[i].getContent();                        

                }

                return ret;

        }

        /**

         * @see com.iver.wmsclient.WMSVersionClient#getQueryHeader()

         */

        public String getQueryHeader() {

                return "VERSION=1.3.0&";

        }

        /**

         * @see com.iver.wmsclient.WMSClient#createQuery()

         */

        public MapQuery createQuery() {

        throw new RuntimeException("No debi? llegar nunca a esta linea");

        }

}