Application: gvSIG desktop » gvSIG raster

package org.gvsig.wcs.provider;

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

 */

import java.io.IOException;

import java.net.ConnectException;

import java.net.URL;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Enumeration;

import java.util.Hashtable;

import java.util.List;

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.remoteclient.wcs.WCSClient;

import org.gvsig.remoteclient.wcs.WCSCoverage;

import org.gvsig.remoteclient.wcs.WCSCoverage.AxisDescription;

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 DefaultWCSRasterServerExplorer extends AbstractDataServerExplorer implements DataServerExplorerProvider, WCSRasterServerExplorer {

    private static final Logger logger =

        LoggerFactory.getLogger(DefaultWCSRasterServerExplorer.class);

    /**

     *

     */

    public static String DESCRIPTION = "WCS Raster Server Explorer";

    private URL service = null;

    //WCS Parameters

//    private WCSStatus status = null;

    private WCSClient wcsClient = null;

    private Boolean override;

    protected DefaultWCSRasterServerExplorer(DataServerExplorerParameters parameters,

        DataServerExplorerProviderServices providerServices) throws InitializeException {

        super(parameters, providerServices);

        // TODO Auto-generated constructor stub

        this.service = (URL)(parameters.getDynValue(WCSRasterServerExplorerParameters.WCS_SERVICE_PARAMETER_NAME));

        this.override = (Boolean)(parameters.getDynValue(WCSRasterServerExplorerParameters.WCS_OVERRIDE_CAPABILITIES_PARAMETER_NAME));

        if (wcsClient == null){

            try {

                wcsClient = new WCSClient(service.toString());

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

                    throw new ConnectException("Error connecting");

                }

//                if (status == null){

//                    status = new WCSStatus();

//                }

//                Rectangle2D extent = wcsClient.getExtent(

//                    (String)parameters.getDynValue(WCSRasterProviderParameters.WCS_COVERAGENAME_PARAMETER_NAME),

//                    (String)parameters.getDynValue(WCSRasterProviderParameters.WCS_CRS_PARAMETER_NAME));

//                status.setExtent(extent);

//

//                wcsClient.getCoverage(status, null); //.getCapabilities(status, true, null);

            } catch (IOException e) {

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

            }

        }

    }

    @Override

    public String getProviderName() {

        return WCSRasterProvider.NAME;

    }

    @Override

    public boolean canAdd() {

        return false;

    }

    @Override

    public boolean canAdd(String storeName) throws DataException {

        return false;

    }

    @Override

    public List<WCSRasterProviderParameters> list() throws DataException {

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

        @SuppressWarnings("unchecked")

        Hashtable<?, WCSCoverage> wcsCoverageList = wcsClient.getCoverageList();

        Collection<WCSCoverage> values = wcsCoverageList.values();

        for (WCSCoverage wcsCoverage : values) {

            WCSRasterProviderParameters params =  new WCSRasterProviderParameters();

            params.setService(this.service);

            String paramCoverage = wcsCoverage.getName();

            params.setCoverageName(paramCoverage);

            @SuppressWarnings("unchecked")

            List<String> formats = wcsCoverage.getFormats();

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

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

            }

            @SuppressWarnings("unchecked")

            List<String> allSrs = wcsCoverage.getAllSrs();

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

                boolean foundValidSrs = false;

                for (String srs : allSrs) {

                    if(isValidSrs(srs)){

                        params.setCRS(srs);

                        foundValidSrs = true;

                        break;

                    }

                }

                if(!foundValidSrs){

                    throw new IllegalArgumentException(this.getProviderName()

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

                }

            }

            result.add(params);

        }

        return result;

    }

    @Override

    public List<WCSRasterProviderParameters> 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 WCS provider.");

    }

    @Override

    public NewDataStoreParameters getAddParameters(String storeName) throws DataException {

        return null;

    }

    @Override

    public List<String> getDataStoreProviderNames() {

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

        x.add(WCSRasterProvider.NAME);

        return x;

    }

    @Override

    @SuppressWarnings("unchecked")

    public List<String> getFormats(){

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

    }

    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;

    }

    @Override

    public String getDescription() {

        return wcsClient.getDescription();

    }

    @Override

    @SuppressWarnings("unchecked")

    public List<String> getCoverageNames() {

        return new ArrayList<String>(wcsClient.getCoverageList().keySet());

    }

    @Override

    public List<WCSCoverageData> getCoverageList() {

         Hashtable<?, WCSCoverage> coverages = wcsClient.getCoverageList();

         Enumeration<WCSCoverage> elements = coverages.elements();

         List<WCSCoverageData> result = new ArrayList<WCSCoverageData>(coverages.size());

         while (elements.hasMoreElements()) {

            WCSCoverage coverage = (WCSCoverage) elements.nextElement();

            DefaultWCSCoverageData wcsCoverageData = new DefaultWCSCoverageData(coverage, wcsClient, service);

            result.add(wcsCoverageData);

        }

         return result;

    }

    @Override

    @SuppressWarnings("unchecked")

    public List<String> getFormats(WCSRasterProviderParameters params){

        String coverageName = params.getCoverageName();

        return ((WCSCoverage)wcsClient.getCoverageList().get(coverageName)).getFormats();

    }

    @Override

    @SuppressWarnings("unchecked")

    public List<String> getBands(WCSRasterProviderParameters params){

        ArrayList<String> bands = new ArrayList<String>();

        String coverageName = params.getCoverageName();

        WCSCoverage wcsCoverage = ((WCSCoverage)wcsClient.getCoverageList().get(coverageName));

        String rangeSetName = wcsCoverage.getRangeSetName();

        if (rangeSetName!=null && !rangeSetName.isEmpty()) {

            Hashtable<?, ?> axisPool = wcsCoverage.axisPool;

            AxisDescription axisDescription = (AxisDescription) axisPool.get(rangeSetName);

            if (axisDescription != null) {

                bands = axisDescription.getSingleValues();

            }

        }

        return bands;

    }

    @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_title").value(wcsClient.getServiceTitle());

        builder.property().labelkey("_server_type").value("WCS "+ wcsClient.getVersion());

        builder.property().labelkey("_server_description").value(wcsClient.getDescription());

        return builder.toString();

    }

}