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

import java.awt.Graphics2D;

import java.awt.Point;

import java.awt.geom.AffineTransform;

import java.awt.geom.NoninvertibleTransformException;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.net.MalformedURLException;

import java.net.URL;

import javax.imageio.ImageIO;

import org.exolab.castor.xml.ValidationException;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.ViewPort;

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

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

import com.iver.cit.gvsig.fmap.layers.layerOperations.InfoByPoint;

import com.iver.cit.gvsig.fmap.operations.Cancellable;

import com.iver.utiles.StringUtilities;

import com.iver.utiles.XMLEntity;

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;

/**

 * Capa WMS.

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class FLyrWMS extends FLyrDefault implements InfoByPoint {

        boolean isPrinting = false;

        boolean mustTileDraw = false;

        boolean mustTilePrint = true;

        int maxTileDrawWidth = -1;

        int maxTileDrawHeight = -1;

        int maxTilePrintWidth = 1500;

        int maxTilePrintHeight = 1500;

        private String m_SRS;

        private String m_Format;

        private String layerQuery;

        private String infoLayerQuery;

        private URL host;

        private WMSClient wmsClient;

        private MapQuery lastMapQuery;

        private Rectangle2D fullExtent;

        /**

         * Inicializa una capa WMS con el driver que se le pasa como par?metro y

         * guard?ndose el nombre del fichero para realizar los accesos, la capa

         * tendr? asociada la proyecci?n que se pasa como parametro tambi?n

         *

         * @param layerName Nombre de la capa.

         * @param rect extent

         * @param host URL.

         * @param format Formato

         * @param query Consulta.

         * @param infoQuery inforamci?n de la consulta.

         * @param srs SRS.

         */

        public void init(String layerName, Rectangle2D rect,

                                URL host, String format, String query, String infoQuery, String srs) {

                FLyrWMS layer = this;

                layer.setHost(host);

                layer.setFullExtent(rect);

                layer.setFormat(format);

                layer.setLayerQuery(query);

                layer.setInfoLayerQuery(infoQuery);

                layer.setSRS(srs);

                layer.setName(layerName);

        }

        /**

         * Slecciona el formato del WMS.

         *

         * @return formato seleccionado.

         *

         * @throws WMSException

         * @throws IllegalStateException

         * @throws ValidationException

         * @throws UnsupportedVersionException

         * @throws IOException

         */

        private String selectFormat()

                throws WMSException, IllegalStateException, ValidationException, 

                        UnsupportedVersionException, IOException {

                String[] formats;

                formats = getWmsClient().getInfoFormats();

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

                        if (formats[i].equals("GML.1")) {

                                return formats[i];

                        }

                        if (formats[i].equals("GML.2")) {

                                return formats[i];

                        }

                        if (formats[i].equals("GML.3")) {

                                return formats[i];

                        }

                        if (formats[i].equals("application/vnd.ogc.gml")) {

                                return formats[i];

                        }

                        if (formats[i].indexOf("XML") != -1) {

                                return formats[i];

                        }

                }

                throw new WMSException("No format supported");

        }

        /**

         * Devuelve el XMLEntity con la informaci?n necesaria para reproducir la

         * capa.

         *

         * @return XMLEntity.

         * @throws XMLException

         */

        public XMLEntity getXMLEntity() throws XMLException {

                XMLEntity xml = super.getXMLEntity();

                xml.putProperty("fullExtent", StringUtilities.rect2String(fullExtent));

                xml.putProperty("host", host.toExternalForm());

                xml.putProperty("infoLayerQuery", infoLayerQuery);

                xml.putProperty("layerQuery", layerQuery);

                xml.putProperty("format", m_Format);

                xml.putProperty("srs", m_SRS);

                return xml;

        }

        /**

         * A partir del XMLEntity reproduce la capa.

         *

         * @param xml XMLEntity

         *

         * @throws XMLException

         * @throws DriverException

         * @throws DriverIOException

         */

        public void setXMLEntity03(XMLEntity xml)

                throws XMLException {

                super.setXMLEntity(xml);

                fullExtent = StringUtilities.string2Rect(xml.getStringProperty(

                                        "fullExtent"));

                try {

                        host = new URL(xml.getStringProperty("host"));

                } catch (MalformedURLException e) {

                        throw new XMLException(e);

                }

                infoLayerQuery = xml.getStringProperty("infoLayerQuery");

                layerQuery = xml.getStringProperty("layerQuery");

                m_Format = xml.getStringProperty("format");

                m_SRS = xml.getStringProperty("srs");

        }

        /**

         * A partir del XMLEntity reproduce la capa.

         *

         * @param xml XMLEntity

         *

         * @throws XMLException

         * @throws DriverException

         * @throws DriverIOException

         */

        public void setXMLEntity(XMLEntity xml)

                throws XMLException {

                super.setXMLEntity(xml);

                fullExtent = StringUtilities.string2Rect(xml.getStringProperty(

                                        "fullExtent"));

                try {

                        host = new URL(xml.getStringProperty("host"));

                } catch (MalformedURLException e) {

                        throw new XMLException(e);

                }

                infoLayerQuery = xml.getStringProperty("infoLayerQuery");

                layerQuery = xml.getStringProperty("layerQuery");

                m_Format = xml.getStringProperty("format");

                m_SRS = xml.getStringProperty("srs");

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.layerOperations.InfoByPoint#queryByPoint(com.iver.cit.gvsig.fmap.operations.QueriedPoint)

         */

        public String queryByPoint(Point p) throws DriverException {

                FeatureInfoQuery query = new FeatureInfoQuery(lastMapQuery);

                query.setFeatureCount(Integer.MAX_VALUE);

                query.setX((int) p.getX());

                query.setY((int) p.getY());

                query.setInfoQuery(infoLayerQuery);

                try {

                        query.setInfoFormat(selectFormat());

                        return new String(getWmsClient().doFeatureInfo(query));

                } catch (WMSException e) {

                        return "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?><exception>" +

                        e.getMessage() + "</exception>";

                } catch (ValidationException e) {

                        /*

                         * TODO Las traducciones en este m?todo han de ser

                         * las mismas que en el m?todo de dibujado

                         */

                        throw new DriverException("No se reconoce el formato de la respuesta",

                                e);

                } catch (UnsupportedVersionException e) {

                        throw new DriverException("Conflicto de versiones", e);

                } catch (IOException e) {

                        throw new DriverException("Error en la conexi?n", e);

                } catch (com.iver.wmsclient.WMSException e) {

                        throw new DriverException(e.getMessage(), e);

                } catch (NoSuchFieldException e) {

                        throw new RuntimeException(

                                "No se rellenaron todos los campos de la petici?n");

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.FLayer#getFullExtent()

         */

        public Rectangle2D getFullExtent() throws DriverException {

                return fullExtent;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.FLayer#draw(java.awt.image.BufferedImage,

         *                 java.awt.Graphics2D, com.iver.cit.gvsig.fmap.ViewPort,

         *                 com.iver.cit.gvsig.fmap.operations.Cancellable)

         */

        public void draw(BufferedImage image, Graphics2D g, ViewPort viewPort,

                Cancellable cancel,double scale) throws DriverException {

                if (isWithinScale(scale)){        

                try {

                        MapQuery mapQuery = getWmsClient().createQuery();

                        mapQuery.setBBOX(viewPort.getAdjustedExtent());

                        mapQuery.setFormat(m_Format);

                        mapQuery.setHeight(viewPort.getImageHeight());

                        // System.err.println("m_Mapa.getHeight() = " + m_Mapa.getHeight());

                        mapQuery.setLayers(layerQuery);

                        mapQuery.setSRS(m_SRS);

                        mapQuery.setStyles("");

                        mapQuery.setWidth(viewPort.getImageWidth());

                        mapQuery.setExceptions("application/vnd.ogc.se_xml");

                        byte[] bytes;

                        lastMapQuery = mapQuery;

                        bytes = getWmsClient().doMapQuery(lastMapQuery);

                        ByteArrayInputStream inbytes = new ByteArrayInputStream(bytes);

                        BufferedImage tempImg = ImageIO.read(inbytes);

                        // LWS Cambio de estrategia en el posicionado para la impresi?n.

                        Point2D p2=new Point2D.Double(viewPort.getAdjustedExtent().getX(), viewPort.getAdjustedExtent().getMaxY()); //viewPort.getOffset();

                        viewPort.getAffineTransform().transform(p2, p2);

                        g.drawImage(tempImg,(int)p2.getX(),(int)p2.getY(), null);

                } catch (ValidationException e) {

                        throw new DriverException("No se reconoce el formato de la respuesta",

                                e);

                } catch (UnsupportedVersionException e) {

                        throw new DriverException("Conflicto de versiones", e);

                } catch (IOException e) {

                        throw new DriverException("Error en la conexi?n", e);

                } catch (com.iver.wmsclient.WMSException e) {

                        throw new DriverException(e.getMessage(), e);

                } catch (NoSuchFieldException e) {

                        throw new RuntimeException(

                                "No se rellenaron todos los campos de la petici?n");

                }

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.FLayer#print(java.awt.Graphics2D,

         *                 com.iver.cit.gvsig.fmap.ViewPort,

         *                 com.iver.cit.gvsig.fmap.operations.Cancellable)

         */

        public void print(Graphics2D g, ViewPort viewPort, Cancellable cancel,double scale)

                throws DriverException {

                if (isVisible() && isWithinScale(scale)){        

                isPrinting = true;

                if (!mustTilePrint) {

                        draw(null, g, viewPort, cancel,scale);

                } else {

                // Para no pedir imagenes demasiado grandes, vamos

                // a hacer lo mismo que hace EcwFile: chunkear.

                // Llamamos a drawView con cuadraditos m?s peque?os

                // del BufferedImage ni caso, cuando se imprime viene con null

                        Tiling tiles = new Tiling(maxTilePrintWidth, maxTilePrintHeight, g.getClipRect());

                        tiles.setAffineTransform((AffineTransform) viewPort.getAffineTransform().clone());

                        for (int tileNr=0; tileNr < tiles.getNumTiles(); tileNr++) {

                            // Parte que dibuja

                            try {

                                ViewPort vp = tiles.getTileViewPort(viewPort, tileNr);

                                draw(null, g, vp, cancel,scale);

                                } catch (NoninvertibleTransformException e) {

                                        // TODO Auto-generated catch block

                                        e.printStackTrace();

                                }

                }

                }

            isPrinting = false;

                }

        }

        public void _print(Graphics2D g, ViewPort viewPort, Cancellable cancel,double scale)

                throws DriverException {

                draw(null, g, viewPort, cancel,scale);

        }

        /**

         * Devuelve el WMSClient.

         *

         * @return WMSClient

         *

         * @throws IllegalStateException

         * @throws ValidationException

         * @throws UnsupportedVersionException

         * @throws IOException

         */

        private WMSClient getWmsClient()

                throws IllegalStateException, ValidationException, 

                        UnsupportedVersionException, IOException {

                if (wmsClient == null) {

                        wmsClient = WMSClientFactory.getClient(host);

                }

                return wmsClient;

        }

        /**

         * Devuelve el URL.

         *

         * @return URL.

         */

        public URL getHost() {

                return host;

        }

        /**

         * Inserta el URL.

         *

         * @param host URL.

         */

        public void setHost(URL host) {

                this.host = host;

        }

        /**

         * Devuelve la informaci?n de la consulta.

         *

         * @return String.

         */

        public String getInfoLayerQuery() {

                return infoLayerQuery;

        }

        /**

         * Inserta la informaci?n de la consulta.

         *

         * @param infoLayerQuery String.

         */

        public void setInfoLayerQuery(String infoLayerQuery) {

                this.infoLayerQuery = infoLayerQuery;

        }

        /**

         * Devuelve la consulta.

         *

         * @return String.

         */

        public String getLayerQuery() {

                return layerQuery;

        }

        /**

         * Inserta la consulta.

         *

         * @param layerQuery consulta.

         */

        public void setLayerQuery(String layerQuery) {

                this.layerQuery = layerQuery;

        }

        /**

         * Devuelve el formato.

         *

         * @return Formato.

         */

        public String getFormat() {

                return m_Format;

        }

        /**

         * Inserta el formato.

         *

         * @param format Formato.

         */

        public void setFormat(String format) {

                m_Format = format;

        }

        /**

         * Devuelve el SRS.

         *

         * @return SRS.

         */

        public String getSRS() {

                return m_SRS;

        }

        /**

         * Inserta el SRS.

         *

         * @param m_srs SRS.

         */

        public void setSRS(String m_srs) {

                m_SRS = m_srs;

        }

        /**

         * Inserta la extensi?n total de la capa.

         *

         * @param fullExtent Rect?ngulo.

         */

        public void setFullExtent(Rectangle2D fullExtent) {

                this.fullExtent = fullExtent;

        }

}