Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.raster.wms.io;

import java.awt.geom.Rectangle2D;

import java.io.BufferedOutputStream;

import java.io.DataInputStream;

import java.io.DataOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.OutputStreamWriter;

import java.net.ConnectException;

import java.net.HttpURLConnection;

import java.net.URL;

import java.security.KeyManagementException;

import java.security.NoSuchAlgorithmException;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.List;

import java.util.TreeMap;

import java.util.Vector;

import java.util.prefs.Preferences;

import javax.management.RuntimeErrorException;

import javax.net.ssl.HttpsURLConnection;

import javax.net.ssl.SSLContext;

import javax.net.ssl.TrustManager;

import javax.net.ssl.X509TrustManager;

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import org.gvsig.compat.net.ICancellable;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DataParameters;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.coverage.exception.RemoteServiceException;

import org.gvsig.remoteclient.exceptions.ServerErrorException;

import org.gvsig.remoteclient.utils.BoundaryBox;

import org.gvsig.remoteclient.utils.Utilities;

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 org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class WMSConnector  {

        private WMSClient                     client;

    private WMSLayerNode                  fmapRootLayer;

    private TreeMap<String, WMSLayerNode> layers = new TreeMap<String, WMSLayerNode>();

    private Logger                        log    = LoggerFactory.getLogger(WMSConnector.class);

    public WMSConnector(URL url) throws ConnectException, IOException {

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

    }

    public String[] getLayerNames() {

            return client.getLayerNames();

    }

    public String[] getLayerTitles() {

            return client.getLayerTitles();

    }

        public void getCapabilities(URL server) throws RemoteServiceException {

//                try {

//                        client.connect();

//                } catch (Exception e) {

//                        throw new WMSException(e);

//                }

        }

    public File getMap(WMSStatus status, ICancellable cancel) throws RemoteServiceException {

        try {

                        return client.getMap(status, cancel);

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

            throw new RemoteServiceException(e.getMessage());

        } catch (ServerErrorException e) {

            throw new RemoteServiceException("WMS Unexpected server error." + e.getMessage(), e);

        }

    }

    /**

     * 

     * @param status

     * @param cancel

     * @param file

     * @throws RemoteServiceException

     */

    public void getMap(WMSStatus status, ICancellable cancel, File file) throws RemoteServiceException {

        try {

                        URL url = client.getGetMapURL(status, cancel);

                        downloadFile(url, file, cancel);

                        String exceptionMessage = client.getExceptionMessage(file);

                        if(exceptionMessage != null)

                                throw new RemoteServiceException(exceptionMessage);

                } catch(IOException e) {

                        throw new RemoteServiceException("WMS: error downloading the file. File:" + file.getAbsolutePath() + "...." + e.getMessage());

                } catch (ServerErrorException e) {

            throw new RemoteServiceException("WMS Unexpected server error."+e.getMessage());

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

            throw new RemoteServiceException(e.getMessage());

        }

    }

    public void downloadFile(URL url, File dstFile, ICancellable cancel) throws IOException {

                Preferences prefs = Preferences.userRoot().node( "gvsig.downloader" );

                // by default 1 minute (60000 milliseconds.

                int timeout = prefs.getInt("timeout", 60000);

                DataOutputStream dos;

                DataInputStream is;

                OutputStreamWriter os = null;

                HttpURLConnection connection = null;

                //If the used protocol is HTTPS

                if (url.getProtocol().equals("https")) {

                        try {

                                disableHttsValidation();

                        } catch (KeyManagementException e) {

                                e.printStackTrace();

                        } catch (NoSuchAlgorithmException e) {

                                e.printStackTrace();

                        }

                }

                connection = (HttpURLConnection)url.openConnection();

                connection.setConnectTimeout(timeout);

                is = new DataInputStream(url.openStream());

                dos = new DataOutputStream( new BufferedOutputStream(new FileOutputStream(dstFile)));

                byte[] buffer = new byte[1024 * 4];

                long readed = 0;

                for (int i = is.read(buffer); i > 0; i = is.read(buffer)){

                        dos.write(buffer, 0, i);

                        readed += i;

                        if(cancel != null && cancel.isCanceled())

                                return;

                }

                if(os != null) {

                        os.close();

                }

                dos.close();

                is.close();

                is = null;

                dos = null;

        }

        /**

         * This method disables the Https certificate validation.

         * @throws KeyManagementException

         * @throws NoSuchAlgorithmException

         */

        private void disableHttsValidation() throws KeyManagementException, NoSuchAlgorithmException{

                // Create a trust manager that does not validate certificate chains

                TrustManager[] trustAllCerts = new TrustManager[] {

                                new X509TrustManager() {

                                        public java.security.cert.X509Certificate[] getAcceptedIssuers() {

                                                return null;

                                        }

                                        public void checkClientTrusted(

                                                        java.security.cert.X509Certificate[] certs, String authType) {

                                        }

                                        public void checkServerTrusted(

                                                        java.security.cert.X509Certificate[] certs, String authType) {

                                        }

                                }

                };

                // Install the all-trusting trust manager

                SSLContext sc = SSLContext.getInstance("SSL");

                sc.init(null, trustAllCerts, new java.security.SecureRandom());

                HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory());

        }

    /**

     * Gets the legend graphic of one layer

     */

    public File getLegendGraphic(WMSStatus status, String layerName, ICancellable cancel) throws RemoteServiceException {

        try {

                File f = client.getLegendGraphic(status, layerName, cancel);

                        return f;

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

            //throw new RemoteServiceException(e.getMessage());

        } catch (ServerErrorException e) {

            throw new RemoteServiceException("WMS Unexpected server error."+e.getMessage());

        }

        return null;

    }

        /**

         * Devuelve WMSClient a partir de su URL.

         *

         * @param url URL.

         *

         * @return WMSClient.

         * @throws IOException

         * @throws ConnectException

         *

         * @throws UnsupportedVersionException

         * @throws IOException

         */

    /**

     * Establishes the connection.

     * @param override, if true the previous downloaded data will be overridden

     * @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(boolean override, ICancellable cancel) {

            if (override) {

                    fmapRootLayer = null;

                    layers.clear();

            }

                return client.connect(override, cancel);

    }

    /**

     * Establishes the connection.

     * @param override, if true the previous downloaded data will be overridden

     * @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(DataServerExplorerParameters p, boolean override, ICancellable cancel) {

            if (override) {

                    fmapRootLayer = null;

                    layers.clear();

            }

            WMSStatus status = new WMSStatus();

            if (p!=null && p instanceof WMSServerExplorerParameters) {

                    status.setXyAxisOrder(((WMSServerExplorerParameters) p).isXyAxisOrder());

            }

                return client.connect(status, override, cancel);

    }

    /**

     * 

     * @param cancel

     * @return

     * @deprecated Use {@link #connect(WMSServerExplorerParameters, boolean, ICancellable)}

     * instead

     */

    public boolean connect(ICancellable cancel) {

            return client.connect(false, cancel);

    }

    /**

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

    }

    /**

     * Returns a Hash table containing the values for each online resource.

     * Using as key a String with name of the WMS request and the value returned

     * by the hash is another string containing the corresponding Url

     * @return HashTable

     */

        public Hashtable getOnlineResources() {

            return client.getServiceInformation().getSupportedOperationsByName();

    }

    /**

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

     */

    public Vector getFormats() {

            return client.getFormats();

    }

    /**

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

     */

    public Vector getInfoFormats() {

            return client.getInfoFormats();

    }

    /**

     * @return <b>Boolean</b> returns if the WMS is queryable (suppports GetFeatureInfo)

     */

    public boolean isQueryable() {

        return client.isQueryable();

    }

    /**

     * @return <b>Boolean</b> returns if the WMS supports getLegendGraphic operation (suppports GetFeatureInfo)

     */

    public boolean hasLegendGraphic() {

        return client.hasLegendGraphic();

    }

    /**

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

     */

    public WMSLayerNode getLayersTree() {

        if (fmapRootLayer == null){

            WMSLayer clientRoot;

            if (client.getLayersRoot() == null) {

                    client.connect(false, null);

            }

            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

     */

    @SuppressWarnings("unchecked")

        private WMSLayerNode parseTree(WMSLayer node, WMSLayerNode parentNode) {

        WMSLayerNode myNode = new WMSLayerNode();

        // Name

        myNode.setName(node.getName());

        // Title

        myNode.setTitle(node.getTitle());

        // Transparency

        myNode.setTransparency(node.hasTransparency());

        // CRS

//        for (int i = 0; i < node.getAllSrs().size(); i++) {

//                String srs = (String)node.getAllSrs().get(i);

//                if(node.getBbox(srs) == null) {

//                        IProjection projSrc = CRSFactory.getCRS("EPSG:4326");

//                        try {

//                                IProjection projDst = CRSFactory.getCRS(srs);

//                                if(projDst != null) {

//                                        try {

//                                                ICoordTrans t = projSrc.getCT(projDst);

//                                                Rectangle2D r = t.convert(getRectangleFromBoundaryBox(node.getLatLonBox()));

//                                                BoundaryBox bbox = getBoundaryBoxFromRectangle(r);

//                                                bbox.setSrs(srs);

//                                                node.addBBox(bbox);

//                                        } catch (Exception e) {

//                                                log.info("I cannot get the transformation between EPSG:4326 and " + srs, e);

//                                        }

//                                }

//                        } catch (Exception e1) {

//                                log.info("I cannot get " + srs, e1);

//                                node.removeSrs(srs);

//                        }

//                }

//                }

        myNode.setSrs(node.getAllSrs());

        // Queryable

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

        // Parent layer

        myNode.setParent(parentNode);

        // Abstract

        myNode.setAbstract(node.getAbstract());

        // Fixed Size

        myNode.setFixedSize(node.getfixedWidth(), node.getfixedHeight());

        // LatLonBox

        if (node.getLatLonBox()!=null)

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

        // Keywords

        ArrayList keywords = node.getKeywords();

        for (int i = 0; i < keywords.size(); i++) {

                myNode.addKeyword((String) keywords.get(i));

                }

        // Styles

        ArrayList styles = node.getStyles();

        for (int i = 0; i < styles.size(); i++) {

            WMSStyle style = (WMSStyle) styles.get(i);

            myNode.addStyle(style);

        }

        // Dimensions

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

        }

        // Children

        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;

    }

    public Rectangle2D getRectangleFromBoundaryBox(BoundaryBox bbox) {

            return new Rectangle2D.Double(bbox.getXmin(),

                            bbox.getYmin(),

                            Math.abs(bbox.getXmax() - bbox.getXmin()),

                            Math.abs(bbox.getYmax() - bbox.getYmin()));

    }

    public BoundaryBox getBoundaryBoxFromRectangle(Rectangle2D r) {

            BoundaryBox bbox = new BoundaryBox();

            bbox.setXmin(r.getMinX());

            bbox.setXmax(r.getMaxX());

            bbox.setYmin(r.getMinY());

            bbox.setYmax(r.getMaxY());

            return bbox;

    }

    public String getAbstract() {

            return client.getServiceInformation().abstr;

    }

    public Rectangle2D getLayersExtent(String[] layerName, String srs) {

            return client.getLayersExtent(layerName, srs);

    }

    public WMSLayerNode getLayer(String layerName) {

        if (getLayers().get(layerName) != null) {

            return (WMSLayerNode)layers.get(layerName);

        }

        return null;

    }

        private TreeMap getLayers() {

        if (fmapRootLayer == null){

            fmapRootLayer = getLayersTree();

        }

        return layers;

    }

    public String getFeatureInfo(WMSStatus _wmsStatus, int i, int j, int max_value, ICancellable cancellable) throws RemoteServiceException {

        try {

            return client.getFeatureInfo(_wmsStatus, i, j, max_value, cancellable);

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

            throw new RemoteServiceException();

                }

    }

    public String getHost(){

            return client.getHost();

    }

}