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

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.util.List;

import java.util.Vector;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.fmap.dal.coverage.datastruct.Extent;

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

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

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

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

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

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.fmap.dal.coverage.store.RasterQuery;

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

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

import org.gvsig.raster.cache.tile.Tile;

import org.gvsig.raster.cache.tile.exception.TileGettingException;

import org.gvsig.raster.impl.DefaultRasterManager;

import org.gvsig.raster.impl.provider.RasterProvider;

import org.gvsig.raster.impl.provider.tile.BaseTileDownloader;

import org.gvsig.raster.util.DefaultProviderServices;

import org.gvsig.raster.wms.io.RemoteWMSStyle;

import org.gvsig.raster.wms.io.WMSConnector;

import org.gvsig.raster.wms.io.WMSDataParameters;

import org.gvsig.raster.wms.io.WMSProvider;

import org.gvsig.remoteclient.utils.Utilities;

import org.gvsig.remoteclient.wms.WMSStatus;

import org.gvsig.tools.locator.LocatorException;

/** 

 * Tile getter 

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class TileDownloaderForWMS extends BaseTileDownloader {

        private WMSConnector             connector  = null;

        private Extent                   lyrExtent  = null;

        public TileDownloaderForWMS(RasterDataStore store,

                        int tilePxWidth,

                        int tilePxHeight) throws RemoteServiceException {

                super(store, tilePxWidth, tilePxHeight);

                this.connector = ((WMSProvider)store.getProvider()).getConnector();

                lyrExtent = store.getExtent();

        }

        public synchronized Tile downloadTile(Tile tile) throws TileGettingException {

                try {

                        Rectangle2D r = new Rectangle2D.Double(tile.getExtent().getMinX(), 

                                        tile.getExtent().getMinY() - tile.getExtent().getHeight(), 

                                        tile.getExtent().getWidth(), 

                                        tile.getExtent().getHeight());

                        //Extent tileExtent = RasterLocator.getManager().getDataStructFactory().createExtent(r);

                        WMSDataParameters p = (WMSDataParameters)store.getParameters();

                        WMSStatus wmsStatus = new WMSStatus();

                        wmsStatus.setLayerNames(Utilities.createVector(p.getLayerQuery(), ","));

                        wmsStatus.setSrs(p.getSRSCode());

                        wmsStatus.setFormat(p.getFormat());

                        List<RemoteWMSStyle> listStyles = p.getStyles();

                        Vector<?> v = listStyles != null ? new Vector<RemoteWMSStyle>(listStyles) : null;

                        wmsStatus.setStyles(v);

                        wmsStatus.setDimensions(p.getDimensions());

                        wmsStatus.setTransparency(p.isWmsTransparent());

                        wmsStatus.setOnlineResource((String) p.getOnlineResource().get("GetMap"));

                        wmsStatus.setExtent(r);

                        wmsStatus.setHeight(tile.getHeightPx());

                        wmsStatus.setWidth(tile.getWidthPx());

                        IProjection proj = (IProjection)p.getSRS();

                        if(proj != null)

                                wmsStatus.setProjected(proj.isProjected());

                        connector.getMap(wmsStatus, tile.getCancelled(), tile.getFile());

                        //Si borramos el rmf no se puede leer la etiqueta Alpha. En caso de que se modifique jgdal para

                        //poder guardar esta etiqueta deberiamos borrar el rmf para ahorrar ficheros

                        //File rmf = new File(tile.getFile().getAbsolutePath() + ".rmf");

                        //if(rmf.exists())

                                //rmf.delete();

                } catch (RemoteServiceException e) {

                        throw new TileGettingException(e);

                } 

                readTileFromDisk(tile);

                return tile;

        }

        @SuppressWarnings("unused")

        private void fuseFiles(Extent dstExtent, int dstWidth, int dstHeight, File src, File dst, double pixelSize) 

                throws RasterDriverException, ProcessInterruptedException, ProviderNotRegisteredException, InitializeException, LocatorException, NotSupportedExtensionException, IOException, QueryException {

                RasterProvider provider = DefaultProviderServices.loadProvider(src);

                RasterQuery q = RasterLocator.getManager().createQuery();

                Buffer bufInput = store.query(q);

                Buffer bufDst = DefaultRasterManager.getInstance().createMemoryBuffer(provider.getDataType()[0], dstWidth, dstHeight, provider.getBandCount(), true);

                String extension = RasterLocator.getManager().getFileUtils().getExtensionFromFileName(src.getAbsolutePath());

                RasterLocator.getManager().getRasterUtils().copyToBuffer(

                                bufDst, 

                                dstExtent, 

                                bufInput, 

                                lyrExtent, 

                                pixelSize, 

                                provider.getColorInterpretation().hasAlphaBand());

                RasterLocator.getManager().getProviderServices().exportBufferToFile(

                                bufDst, 

                                provider.getCellSize(), 

                                extension, 

                                provider.getColorInterpretation().hasAlphaBand(), 

                                dst, 

                                dstExtent, 

                                provider.getColorInterpretation());

        }

}