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.fmap.dal.coverage.dataset.io.tile.downloader;

import java.awt.geom.AffineTransform;

import java.io.IOException;

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

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

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

import org.gvsig.fmap.dal.coverage.dataset.io.gdal.GdalDataParameters;

import org.gvsig.fmap.dal.coverage.dataset.io.gdal.GdalProvider;

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

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

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

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

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

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

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

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

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

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

import org.gvsig.raster.cache.tile.provider.Downloader;

import org.gvsig.raster.impl.DefaultRasterManager;

import org.gvsig.raster.impl.datastruct.BandListImpl;

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

/** 

 * Base class for downloaders  

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

 */

public abstract class BaseTileDownloader implements Downloader {

        protected int                       tilePxWidth  = 0;

        protected int                       tilePxHeight = 0;

        protected DefaultRasterProvider     prov         = null;

        protected byte                      byteNoData   = (byte)0x00;

        protected short                     shortNoData  = Short.MIN_VALUE;

        protected int                       intNoData    = Integer.MIN_VALUE;

        protected float                     floatNoData  = -99999;

        protected double                    doubleNoData = -99999;

        public BaseTileDownloader(DefaultRasterProvider prov, int tilePxWidth, int tilePxHeight) {

                this.prov = prov;

                this.tilePxWidth = tilePxWidth;

                this.tilePxHeight = tilePxHeight;

        }

        /**

         * Sets the nodata value

         * @param noData

         */

        public void setNoDataValue(Object noData) {

                switch (prov.getDataType()[0]) {

                case Buffer.TYPE_BYTE:

                        byteNoData = ((Byte)noData).byteValue();

                        break;

                case Buffer.TYPE_SHORT:

                        shortNoData = ((Short)noData).shortValue();

                        break;

                case Buffer.TYPE_INT:

                        intNoData = ((Integer)noData).intValue();

                        break;

                case Buffer.TYPE_FLOAT:

                        floatNoData = ((Float)noData).floatValue();

                        break;

                case Buffer.TYPE_DOUBLE:

                        doubleNoData = ((Double)noData).doubleValue();

                        break;

                }

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.cache.tile.provider.Downloader#readTile(org.gvsig.raster.cache.tile.Tile)

         */

        public Tile readTileFromDisk(Tile tile) throws TileGettingException {

                if(!tile.getFile().exists())

                        return null;

                BandList bandList = (BandList)tile.getDownloaderParams("BandList");

                for (int j = 0; j < bandList.getBandCount(); j++) {

                        bandList.getBand(j).setFileName(tile.getFile().getAbsolutePath());

                }

                int mallocNBands = 0;

                if(bandList.getDrawableBands() != null)

                        mallocNBands = bandList.getDrawableBands().length;

                else

                        mallocNBands = bandList.getDrawableBandsCount();

                Buffer buf = DefaultRasterManager.getInstance().createMemoryBuffer(prov.getDataType()[0], this.tilePxWidth, this.tilePxHeight, mallocNBands, true);

                Buffer transparency = DefaultRasterManager.getInstance().createMemoryBuffer(prov.getDataType()[0], this.tilePxWidth, this.tilePxHeight, 1, true);

                try {

                        GdalDataParameters p = new GdalDataParameters();

                        p.setFile(tile.getFile());

                        GdalProvider provider = new GdalProvider(p, null);

                        buf = provider.getWindow((int)0, (int)0, bandList, buf);

                        if(provider.getTransparency().getAlphaBandNumber() >= 0) {

                                //Lee la banda de transparencia para eliminar la zona del tile que est? fuera de la imagen

                                BandList newBandList = new BandListImpl(tile.getFile().getAbsolutePath(), provider.getBandCount(), provider.getDataType()[0]);

                                newBandList.clearDrawableBands();

                                newBandList.addDrawableBand(0, provider.getBandCount() - 1);

                                transparency = provider.getWindow((int)0, (int)0, newBandList, transparency);

                                tile.setData(new Object[]{buf, transparency, null});

                        } else

                                tile.setData(new Object[]{buf, null, provider.getColorTable()});

                        buf.setDataExtent(provider.getExtent().toRectangle2D());

                        provider.close();

                        return tile;

                } catch (ProcessInterruptedException e) {

                        throw new TileGettingException(e);

                } catch (RasterDriverException e) {

                        throw new TileGettingException(e);

                } catch (NotSupportedExtensionException e) {

                        throw new TileGettingException(e);

                }

        }

        /**

         * Sets a band to NODATA value 

         * @param bufResult

         * @param tileExtent

         * @param buf

         * @param ex

         * @param pixelSize

         */

        protected void clearMaskBuffer(Buffer buf, int nBand) {

                if(buf.getDataType() == Buffer.TYPE_BYTE) {

                        for (int i = 0; i < buf.getHeight(); i++) {

                                for (int j = 0; j < buf.getWidth(); j++) {

                                        buf.setElem(i, j, nBand, byteNoData);

                                }

                        }

                        return;

                }

                if(buf.getDataType() == Buffer.TYPE_SHORT) {

                        for (int i = 0; i < buf.getHeight(); i++) {

                                for (int j = 0; j < buf.getWidth(); j++) {

                                        buf.setElem(i, j, nBand, shortNoData);

                                }

                        }

                        return;

                }

                if(buf.getDataType() == Buffer.TYPE_INT) {

                        for (int i = 0; i < buf.getHeight(); i++) {

                                for (int j = 0; j < buf.getWidth(); j++) {

                                        buf.setElem(i, j, nBand, intNoData);

                                }

                        }

                        return;

                }

                if(buf.getDataType() == Buffer.TYPE_FLOAT) {

                        for (int i = 0; i < buf.getHeight(); i++) {

                                for (int j = 0; j < buf.getWidth(); j++) {

                                        buf.setElem(i, j, nBand, floatNoData);

                                }

                        }

                        return;

                }

                if(buf.getDataType() == Buffer.TYPE_DOUBLE) {

                        for (int i = 0; i < buf.getHeight(); i++) {

                                for (int j = 0; j < buf.getWidth(); j++) {

                                        buf.setElem(i, j, nBand, doubleNoData);

                                }

                        }

                        return;

                }

        }

        /**

         * Saves a buffer to disk

         * @param bufResult Buffer to save

         * @param pixelSize Pixel size

         * @param extension output file format

         * @param alphaBand true if it has alpha band

         * @param path Path to the new file

         * @param tileExtent Bounding box of the tile

         * @throws NotSupportedExtensionException

         * @throws RasterDriverException

         * @throws ProcessInterruptedException

         * @throws IOException

         */

        protected void saveTile(Buffer bufResult, 

                        double pixelSize, 

                        String extension, 

                        boolean alphaBand, 

                        String path, 

                        Extent tileExtent) throws NotSupportedExtensionException, RasterDriverException, ProcessInterruptedException, IOException {

                //Escritura en disco del tile

                RasterManager rManager = RasterLocator.getManager();

                DataServerWriter dataWriter = RasterLocator.getManager().createDataServerWriter();

                dataWriter.setBuffer(bufResult, -1);

                Params params = rManager.createWriter("_." + extension).getParams();

                AffineTransform affineTransform = new AffineTransform(pixelSize, 0, 0, -pixelSize,

                                tileExtent.getULX(),

                                tileExtent.getULY());

                RasterWriter rw = rManager.createWriter(dataWriter, path,

                                bufResult.getBandCount(), affineTransform, bufResult.getWidth(),

                                bufResult.getHeight(), bufResult.getDataType(), params, null);

                if(alphaBand) {

                        String[] ci = new String[bufResult.getBandCount()];

                        for (int i = 0; i < bufResult.getBandCount(); i++) {

                                if(i == bufResult.getBandCount() - 1)

                                        ci[i] = "Alpha";

                                else

                                        ci[i] = "Gray";

                        }

                        rw.setColorBandsInterpretation(ci);

                }

                rw.dataWrite();

                rw.writeClose();

        }

}