Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.impl.store;

import java.awt.Rectangle;

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

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.store.RasterDataStore;

import org.gvsig.raster.impl.DefaultRasterManager;

import org.gvsig.raster.impl.buffer.DefaultRasterQuery;

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

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

/**

 * The requests to a <code>RasterDataStore</code> adjust the bounding box and the size in pixels 

 * to the source limits. The user can need a buffer that exceeds the raster dimension. 

 * In this case, the area that is out of the raster limits should be set to nodata values or 

 * transparent in RGB cases. 

 * 

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

 */

public class FramedBufferResampling {

        private DefaultRasterQuery           query            = null;

        private DefaultRasterQuery           originalQuery    = null;

        private RasterDataStore              store            = 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 FramedBufferResampling(DefaultRasterQuery q, RasterDataStore store) {

                this.query = q;

                this.store = store;

                this.originalQuery = (DefaultRasterQuery)query.clone();

                setNoDataValue(store.getNoDataValue());

        }

        public Buffer query(RasterProvider provider) throws QueryException, ProcessInterruptedException {

                adjustQueryToTheStore();

                query.calculateParameters(store);

                originalQuery.calculateParameters(store);

                Buffer bufResult = null;

                try {

                        Buffer buf = ((AbstractRasterProvider)provider).getDataSet(query);

                        bufResult = buildBuffer(buf.getBandCount(), originalQuery.getAdjustedBufWidth(), originalQuery.getAdjustedBufHeight());

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

                                        bufResult, 

                                        originalQuery.getAdjustedRequestBoundingBox(), 

                                        buf, 

                                        query.getAdjustedRequestBoundingBox(), 

                                        1, 

                                        store.getColorInterpretation().hasAlphaBand());

                        buf.dispose();

                } catch (RasterDriverException e) {

                        throw new QueryException("Error getting data", e);

                }

                return bufResult;

        }

        /**

         * Sets the nodata value

         * @param noData

         */

        private void setNoDataValue(NoData noData) {

                if(noData == null)

                        return;

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

                case Buffer.TYPE_BYTE:

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

                        break;

                case Buffer.TYPE_SHORT:

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

                        break;

                case Buffer.TYPE_INT:

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

                        break;

                case Buffer.TYPE_FLOAT:

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

                        break;

                case Buffer.TYPE_DOUBLE:

                        doubleNoData = ((NoData)noData).getValue().doubleValue();

                        break;

                }

        }

        /**

         * Builds a buffer using the specified number of bands and initialize it

         * @param nbands

         * @return

         */

        private Buffer buildBuffer(int nbands, int w, int h) {

                Buffer bufResult = DefaultRasterManager.getInstance().createBuffer(store.getDataType()[0], w, h, nbands, true);

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

                        clearMaskBuffer(bufResult, i);

                }

                return bufResult;

        }

        /**

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

                }

        }

        private double round(double value) {

                double a = (value - (int)value);

                return (a > 0.95 || a < 0.05) ? Math.round(value) : value;

        }

        private void adjustQueryToTheStore() {

                if(query.getType() == DefaultRasterQuery.TYPE_WCOORDS_SHIFT) {

                        Extent e = query.getAdjustedRequestBoundingBox().intersection(store.getExtent());

                        int w = (int)Math.ceil(round((query.getBufWidth() * e.width()) /  query.getAdjustedRequestBoundingBox().width()));

                        int h = (int)Math.ceil(round((query.getBufHeight() * e.height()) /  query.getAdjustedRequestBoundingBox().height()));

                        query.setAdjustedRequestBoundingBox(e);

                        query.setBufWidth(w);

                        query.setBufHeight(h);

                }

                if(query.getType() == DefaultRasterQuery.TYPE_PX_SHIFT) {

                        Rectangle r = adjustRectangleToStoreSize(query.getRequestPxWindow());

                        query.setRequestPixelWindow(r);

                }

                if(query.getType() == DefaultRasterQuery.TYPE_PX_RESCALED_SHIFT) {

                        Rectangle r = adjustRectangleToStoreSize(query.getRequestPxWindow());

                        query.setRequestPixelWindow(r);

                        int w = (int)r.getWidth() - (int)originalQuery.getRequestPxWindow().getWidth();

                        int h = (int)r.getHeight() - (int)originalQuery.getRequestPxWindow().getHeight();

                        query.setBufWidth(query.getBufWidth() - w);

                        query.setBufHeight(query.getBufHeight() - h);

                }

        }

        /**

         * Adjusts the limits of the rectangle to the store limits

         * @param r

         * @return

         */

        private Rectangle adjustRectangleToStoreSize(Rectangle r) {

                Rectangle newRect = (Rectangle)r.clone();  

                if(r.getX() < 0)

                        newRect.setBounds(0, (int)r.getY(), (int)(r.getWidth() + r.getX()), (int)r.getHeight());

                if(r.getY() < 0)

                        newRect.setBounds((int)r.getX(), 0, (int)r.getWidth(), (int)(r.getHeight() + r.getY()));

                if(newRect.getWidth() > store.getWidth())

                        newRect.setSize((int)store.getWidth(), (int)r.getHeight());

                if(newRect.getHeight() > store.getHeight())

                        newRect.setSize((int)r.getWidth(), (int)store.getHeight());

                return newRect;

        }

}