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.impl.provider.tile;

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.datastruct.BandList;

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

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

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

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

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.fmap.dal.coverage.store.props.ColorInterpretation;

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.cache.tile.provider.Downloader;

import org.gvsig.raster.impl.DefaultRasterManager;

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

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

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

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

import org.gvsig.raster.util.DefaultProviderServices;

/** 

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

        }

        /**

         * Gets the size of a tile in pixels 

         * @return Array of two integers. The first integer is the width and the second the height in pixels

         */

        public int[] getTileSize() {

                return new int[]{tilePxWidth, 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()) {

                        tile.setCorrupt(true);

                        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();

                try {

                        DefaultRasterProvider provider = DefaultProviderServices.loadProvider(tile.getFile());

                        //int mallocNBands = prov.getDataType()[0] == Buffer.TYPE_BYTE ? provider.getBandCount() - 1 : provider.getBandCount();

                        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);

                        //buf = provider.getWindow((int)0, (int)0, createBandListFromProvider(provider), buf);

                        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 (ProviderNotRegisteredException e) {

                        throw new TileGettingException(e);

                } catch (InitializeException e) {

                        throw new TileGettingException(e);

                }

        }

        /**

         * Builds a BandList structure using all bands of a raster provider

         * @param prov

         * @return

         */

        @SuppressWarnings("unused")

        private BandList createBandListFromProvider(RasterProvider prov) {

                BandList bandList = new BandListImpl();

                bandList.clear();

                int nBands = prov.getDataType()[0] == Buffer.TYPE_BYTE ? prov.getBandCount() - 1 : prov.getBandCount();

                int[] drawableBands = new int[nBands];

                for(int i = 0; i < nBands; i++) {

                        try {

                                int dataType = prov.getDataType()[i];

                                DatasetBand band = new DatasetBandImpl(prov.getURIByBand(i), 

                                                                                                        prov.getBandPositionByProvider(i), 

                                                                                                        dataType, 

                                                                                                        nBands);

                                bandList.addBand(band, i);

                                drawableBands[i] = i;

                        } catch(BandNotFoundInListException ex) {

                                //No a?adimos la banda

                        }

                }

                bandList.setDrawableBands(drawableBands);

                return bandList;

        }

        /**

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

                        Tile tile, 

                        Extent tileExtent,

                        ColorInterpretation colorInterpretation) 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, tile.getFile().getAbsolutePath(),

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

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

                if(colorInterpretation != null) {

                        String[] values = colorInterpretation.getValues();

                        if(alphaBand) {

                                String[] newValues = values;

                                boolean exists = false;

                                for (int i = 0; i < values.length; i++) {

                                        if(values[i].compareTo("Alpha") == 0)

                                                exists = true;

                                }

                                if(!exists) {

                                        newValues = new String[values.length + 1];

                                        for (int i = 0; i < values.length; i++) {

                                                newValues[i] = values[i];

                                        }

                                        newValues[newValues.length - 1] = "Alpha";

                                }

                                rw.setColorBandsInterpretation(newValues);

                        }

                } else {

                        if(alphaBand) {

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

                                if(bufResult.getBandCount() == 4) {

                                        ci[0] = "Red";

                                        ci[1] = "Green";

                                        ci[2] = "Blue";

                                        ci[3] = "Alpha";

                                } else {

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

                                                ci[i] = "Gray";

                                        }

                                }

                                rw.setColorBandsInterpretation(ci);

                        }

                }

                rw.dataWrite();

                rw.writeClose();

                tile.setCorrupt(false);

        }

}