Application: gvSIG desktop » gvSIG raster

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2016 gvSIG Association

 *

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.wms.provider;

import java.io.IOException;

import java.net.ConnectException;

import java.net.URL;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Set;

import java.util.TreeMap;

import java.util.Vector;

import javax.swing.tree.TreeNode;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.NewDataStoreParameters;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.exception.InitializeException;

import org.gvsig.fmap.dal.spi.AbstractDataServerExplorer;

import org.gvsig.fmap.dal.spi.DataServerExplorerProvider;

import org.gvsig.fmap.dal.spi.DataServerExplorerProviderServices;

import org.gvsig.raster.lib.buffer.api.statistics.StatisticsBand;

import org.gvsig.remoteclient.wms.WMSClient;

import org.gvsig.remoteclient.wms.WMSLayer;

import org.gvsig.remoteclient.wms.WMSStatus;

import org.gvsig.remoteclient.wms.WMSStyle;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.util.InformationBuilder;

import org.gvsig.tools.util.ToolsUtilLocator;

/**

 * @author fdiaz

 *

 */

public class WMSRasterServerExplorer extends AbstractDataServerExplorer implements DataServerExplorerProvider {

    private static final Logger logger =

        LoggerFactory.getLogger(WMSRasterServerExplorer.class);

    public static final String NAME = "WMS";

    public static String DESCRIPTION = "WMS Server";

    private URL service = null;

    private boolean override = false;

    //WMS Parameters

    private WMSStatus status = null;

    private WMSClient wmsClient = null;

    protected WMSRasterServerExplorer(DataServerExplorerParameters parameters,

        DataServerExplorerProviderServices providerServices) throws InitializeException {

        super(parameters, providerServices);

        // TODO Auto-generated constructor stub

        this.service = (URL)(parameters.getDynValue(WMSRasterServerExplorerParameters.WMS_SERVICE_PARAMETER_NAME));

        this.override = (Boolean)(parameters.getDynValue(WMSRasterServerExplorerParameters.WMS_OVERRIDE_CAPABILITIES_PARAMETER_NAME));

        if (wmsClient == null){

            try {

                wmsClient = new WMSClient(service.toString());

                if (status == null){

                    status = new WMSStatus();

                }

                wmsClient.getCapabilities(status, this.override, null);

            } catch (ConnectException e) {

                throw new InitializeException("Not possible to connect with " + service, e);

            } catch (IOException e) {

                throw new InitializeException("Not possible to connect with " + service, e);

            }

        }

    }

    @Override

    public String getProviderName() {

        return WMSRasterProvider.NAME;

    }

    @Override

    public boolean canAdd() {

        return false;

    }

    @Override

    public boolean canAdd(String storeName) throws DataException {

        return false;

    }

    @Override

    public List<WMSRasterProviderParameters> list() throws DataException {

        List<WMSRasterProviderParameters> result = new ArrayList<WMSRasterProviderParameters>();

        @SuppressWarnings("unchecked")

        TreeMap<String, WMSLayer> wmsLayersMap = wmsClient.getLayers();

        Collection<WMSLayer> values = wmsLayersMap.values();

        for (Iterator<WMSLayer> iterator = values.iterator(); iterator.hasNext();) {

            WMSLayer wmsLayer = (WMSLayer) iterator.next();

            WMSRasterProviderParameters params =  new WMSRasterProviderParameters();

            params.setService(this.service);

            List<String> paramLayers = new ArrayList<String>();

            paramLayers.add(wmsLayer.getName());

            params.setLayers(paramLayers);

            @SuppressWarnings("unchecked")

            ArrayList<WMSStyle> styles = wmsLayer.getStyles();

            if (styles != null && styles.size() > 0) {

                List<String> paramStyles = new ArrayList<String>();

                paramStyles.add(((WMSStyle) (styles.get(0))).getName());

                params.setStyles(paramStyles);

            }

            @SuppressWarnings("unchecked")

            Vector<String> formats = wmsClient.getFormats();

            if(formats!=null && formats.size()>0){

                params.setFormat((String)(formats.get(0)));

            }

            @SuppressWarnings("unchecked")

            Vector<String> infoFormats = wmsClient.getInfoFormats();

            if(infoFormats!=null && infoFormats.size()>0){

                params.setInfoFormat((String)infoFormats.get(0));

            }

            @SuppressWarnings("unchecked")

            Vector<String> allSrs = wmsLayer.getAllSrs();

            if(allSrs!=null && allSrs.size()>0){

                int i=0;

                while(i<allSrs.size()){

                    String srs = allSrs.get(i);

                    if(isValidSrs(srs)){

                        params.setCRS(srs);

                        break;

                    }

                    i++;

                }

                if(i>=allSrs.size()){

                    throw new IllegalArgumentException(this.getProviderName()

                            + ": The server does not provide any valid srs.'");

                }

            }

            result.add(params);

        }

        return result;

    }

    @Override

    public List<WMSRasterProviderParameters> list(int mode) throws DataException {

        if(mode!=DataServerExplorer.MODE_RASTER){

            return null;

        }

        return list();

    }

    @Override

    public boolean add(String provider, NewDataStoreParameters parameters, boolean overwrite) throws DataException {

        return false;

    }

    @Override

    public void remove(DataStoreParameters parameters) throws DataException {

        throw new UnsupportedOperationException("Can't remove a WMS provider.");

    }

    @Override

    public NewDataStoreParameters getAddParameters(String storeName) throws DataException {

        return null;

    }

    @Override

    public List<String> getDataStoreProviderNames() {

        List<String> x = new ArrayList<String>();

        x.add(WMSRasterProvider.NAME);

        return x;

    }

    /**

     * @param params

     * @return formats

     */

    @SuppressWarnings("unchecked")

    public List<String> getFormats(){

        return new ArrayList<String>(wmsClient.getFormats());

    }

    /**

     * @return info formats

     */

    @SuppressWarnings("unchecked")

    public List<String> getInfoFormats(){

        return new ArrayList<String>(wmsClient.getInfoFormats());

    }

    /**

     * @param params

     * @return list of available srs for the params

     */

    @SuppressWarnings("unchecked")

    public List<String> getSrss(WMSRasterProviderParameters params){

        List<String> layers = params.getLayers();

        Set<String> setSrs = new HashSet<String>();

        for (Iterator<String> iterator = layers.iterator(); iterator.hasNext();) {

            String name = (String) iterator.next();

            WMSLayer wmsLayer = wmsClient.getLayer(name);

            Vector<String> allSrs = wmsLayer.getAllSrs();

            for (Iterator<String> iterator2 = allSrs.iterator(); iterator2.hasNext();) {

                String srs = iterator2.next();

                if(isValidSrs(srs)){

                    setSrs.add(srs);

                }

            }

        }

        return new ArrayList<String>(setSrs);

    }

    private boolean isValidSrs(String srs){

        try{

            IProjection crs=CRSFactory.getCRS(srs);

            if(crs!=null){

                return true;

            }

        }catch(Exception e){

            logger.info("Can't get crs from: '"+srs+"' code", e);

            return false;

        }

        return false;

    }

//    public WMSLayer getRootLayer(){

//        return this.wmsClient.getRootLayer();

//    }

    public TreeNode getTreeNode(){

        LayerNode node = new LayerNode(this.wmsClient.getRootLayer(), null);

        return node;

    }

    public String getInfoString(){

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        InformationBuilder builder = ToolsUtilLocator.getToolsUtilManager().createInformationBuilder();

        builder.title().labelkey(i18nManager.getTranslation("_service_information"));

        builder.property().labelkey("_server").value(service.toString());

        builder.property().labelkey("_server_type").value("WMS "+ wmsClient.getVersion());

        builder.property().labelkey("_server_title").value(wmsClient.getServiceInformation().title);

        builder.property().labelkey("_server_sumary").value(wmsClient.getServiceInformation().abstr);

        return builder.toString();

    }

}