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.wmts.provider;

import java.io.IOException;

import java.net.ConnectException;

import java.net.URL;

import java.util.ArrayList;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Set;

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

import org.gvsig.raster.wmts.ogc.WMTSClient;

import org.gvsig.raster.wmts.ogc.WMTSOGCLocator;

import org.gvsig.raster.wmts.ogc.struct.WMTSLayer;

import org.gvsig.raster.wmts.ogc.struct.WMTSStyle;

import org.gvsig.raster.wmts.ogc.struct.WMTSTheme;

import org.gvsig.raster.wmts.ogc.struct.WMTSThemes;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.util.InformationBuilder;

import org.gvsig.tools.util.ToolsUtilLocator;

import org.gvsig.wmts.provider.tree.DefaultWMTSRootNode;

import org.gvsig.wmts.provider.tree.WMTSNode;

import org.gvsig.wmts.provider.tree.WMTSRootNode;

/**

 * @author fdiaz

 *

 */

public class DefaultWMTSRasterServerExplorer extends AbstractDataServerExplorer implements WMTSRasterServerExplorer {

    private static final Logger logger =

        LoggerFactory.getLogger(DefaultWMTSRasterServerExplorer.class);

    private URL service = null;

    //WMTS Parameters

    private WMTSClient wmtsClient = null;

    private WMTSRootNode rootNode = null;

    protected DefaultWMTSRasterServerExplorer(DataServerExplorerParameters parameters,

        DataServerExplorerProviderServices providerServices) throws InitializeException {

        super(parameters, providerServices);

        // TODO Auto-generated constructor stub

        boolean override = false;

        this.service = (URL)(parameters.getDynValue(WMTSRasterServerExplorerParameters.WMTS_SERVICE_PARAMETER_NAME));

        if (wmtsClient == null){

            try {

                WMTSOGCLocator.getManager().createWMTSClient(service.toString());

                wmtsClient = WMTSOGCLocator.getManager().createWMTSClient(service.toString());

//                wmtsClient.setForceChangeAxisOrder(((WMTSRasterProviderParameters)getParameters()).getForceChangeAxisOrder());

                if (!wmtsClient.connect(override, null)) {

                    throw new ConnectException("Error connecting");

                }

//                if (status == null){

//                    status = wmtsClient.createStatus();

//                }

//

//                wmtsClient.getCapabilities(status, true, 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 WMTSRasterProvider.NAME;

    }

    @Override

    public boolean canAdd() {

        return false;

    }

    @Override

    public boolean canAdd(String storeName) throws DataException {

        return false;

    }

    private List<WMTSLayer> getLayers(WMTSTheme theme){

        List<WMTSLayer> wmtsLayers = new ArrayList<WMTSLayer>();

        if(theme.getLayer()!=null){

            wmtsLayers.add(theme.getLayer());

        }

        for (int i = 0; i < theme.getChildCount(); i++) {

            wmtsLayers.addAll(getLayers(theme.getChildren(i)));

        }

        return wmtsLayers;

    }

    @Override

    public List<WMTSRasterProviderParameters> list() throws DataException {

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

        @SuppressWarnings("unchecked")

        WMTSThemes themes = wmtsClient.getLayerListAsThemes();

        List<WMTSLayer> wmtsLayers = new ArrayList<WMTSLayer>();

        for (int i = 0; i < themes.getChildCount(); i++) {

            WMTSTheme theme = themes.getChildren(i);

            wmtsLayers.addAll(getLayers(theme));

        }

        for (Iterator<WMTSLayer> iterator = wmtsLayers.iterator(); iterator.hasNext();) {

            WMTSLayer wmtsLayer = (WMTSLayer) iterator.next();

            WMTSRasterProviderParameters params =  new WMTSRasterProviderParameters();

            params.setService(this.service);

            params.setLayer(wmtsLayer.getTitle());

            @SuppressWarnings("unchecked")

            List<WMTSStyle> styles = wmtsLayer.getStyle();

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

                params.setStyle(((WMTSStyle) (styles.get(0))).getIdentifier());

            }

            @SuppressWarnings("unchecked")

            List<String> formats = wmtsLayer.getFormat();

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

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

            }

            @SuppressWarnings("unchecked")

             List<String> infoFormats = wmtsLayer.getInfoFormat();

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

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

            }

            @SuppressWarnings("unchecked")

            List<String> allSrs = wmtsLayer.getSrsList();

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

                int i=0;

                while(i<allSrs.size()){

                    String srs = allSrs.get(i);

                    if(isValidSrs(srs)){

                        params.setCRS(srs);

                        IProjection crs=CRSFactory.getCRS(srs);

                        //FIXME: ?Qu? condiciones se deben cumplir para que este par?metro se deba poner a true?

//                        params.setForceChangeAxisOrder(false);

                        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<WMTSRasterProviderParameters> 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 WMTS provider.");

    }

    @Override

    public NewDataStoreParameters getAddParameters(String storeName) throws DataException {

        return null;

    }

    @Override

    public List<String> getDataStoreProviderNames() {

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

        x.add(WMTSRasterProvider.NAME);

        return x;

    }

    /* (non-Javadoc)

     * @see org.gvsig.wmts.provider.WMTSRasterServerExplorer#getSrss(org.gvsig.wmts.provider.WMTSRasterProviderParameters)

     */

    @Override

    @SuppressWarnings("unchecked")

    public List<String> getSrss(WMTSRasterProviderParameters params) {

        String layer = params.getLayer();

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

        WMTSLayer wmtsLayer = wmtsClient.getLayer(layer);

        List<String> allSrs = wmtsLayer.getSrsList();

        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;

    }

    /* (non-Javadoc)

     * @see org.gvsig.wmts.provider.WMTSRasterServerExplorer#getInfoString()

     */

    @Override

    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("WMTS "+ wmtsClient.getServiceIdentification().getServiceType());

        builder.property().labelkey("_server_title").value(wmtsClient.getServiceIdentification().getTitle());

        builder.property().labelkey("_server_sumary").value(wmtsClient.getServiceIdentification().getAbstract());

        return builder.toString();

    }

    @Override

    public boolean hasThemes() {

        return wmtsClient.getThemes().hasThemes();

    }

    @Override

    public WMTSNode getListAsTree() {

        return getListAsTree(false);

    }

    @Override

    public WMTSNode getListAsTree(boolean clone) {

        if (clone) {

            WMTSThemes wmtsThemes = wmtsClient.getThemes();

            if(!wmtsThemes.hasThemes()){

                wmtsThemes = wmtsClient.getLayerListAsThemes();

            }

            DefaultWMTSRootNode cloned = new DefaultWMTSRootNode(wmtsThemes, this.service);

            return cloned;

        } else {

            return getRootNode();

        }

    }

    private WMTSNode getRootNode() {

        if(rootNode  == null){

            WMTSThemes wmtsThemes = wmtsClient.getThemes();

            if(!wmtsThemes.hasThemes()){

                wmtsThemes = wmtsClient.getLayerListAsThemes();

            }

            rootNode = new DefaultWMTSRootNode(wmtsThemes, this.service);

        }

        return rootNode;

    }

}