| gvSIG desktop 1

/*

 * Created on 19-jul-2006

 *

 * To change the template for this generated file go to

 * Window>Preferences>Java>Code Generation>Code and Comments

 */

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.raster.dataset;

import junit.framework.TestCase;

import org.gvsig.raster.RasterLibrary;

import org.gvsig.raster.buffer.BufferFactory;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.dataset.io.MemoryRasterDriverParam;

/**

 * 

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

 *

 */

public class TestIOMemoryRasterDriver extends TestCase {

        private String baseDir = "./test-images/";

        private String path = baseDir + "miniRaster25x24.tif";        

        private RasterDataset f = null;

        private RasterDataset f1 = null;

        private BufferFactory ds = null;

        private BandList bandList = null;

        public void start() {

                this.setUp();

                this.testStack();

        }

        static {

                RasterLibrary.wakeUp();

        }

        public void setUp() {

                System.err.println("TestIOMemoryRasterDriver running...");

                try {

                        f1 = RasterDataset.open(null, path);

                } catch (NotSupportedExtensionException e1) {

                        e1.printStackTrace();

                } catch (RasterDriverException e1) {

                        e1.printStackTrace();

                }

                int[] drawableBands = {0, 1, 2};

                ds = new BufferFactory(f1);

                ds.addDrawableBands(drawableBands);

                ds.setAreaOfInterest(0, 0, 25, 24);

                IBuffer buf = ds.getRasterBuf();

                try {

                        MemoryRasterDriverParam param = new MemoryRasterDriverParam(buf, ds.getGeoRasterMultiFile().getExtent());

                        f = RasterDataset.open(null, param);

                } catch (NotSupportedExtensionException e1) {

                        e1.printStackTrace();

                } catch (RasterDriverException e1) {

                        e1.printStackTrace();

                }

                bandList = new BandList();

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

                        try{

                                Band band = new Band(f.getFName(), i, f.getDataType());

                                band.setPositionToDrawInBuffer(i);

                                bandList.addBand(band, i);

                        }catch(BandFoundInListException ex){

                                //No a?adimos la banda

                        }

                }

        }

        public void testStack(){

                //**************************************

                //Prueba getWindowRaster(int x, int y, int w, int h, BandList bandList, IBuffer rasterBuf)

                //COMPLETA

                IBuffer buffer = RasterBuffer.getBuffer(f.getDataType(), 25, 24, 3, true, null);

                f.getWindowRaster(0, 0, 25, 24, bandList, buffer);

                dataTest1(buffer);

                //ZOOM

                buffer = RasterBuffer.getBuffer(f.getDataType(), 2, 2, 3, true, null);

                f.getWindowRaster(10, 10, 2, 2, bandList, buffer);

                dataTest2(buffer);

                //**************************************                

                //Prueba de getWindowRaster(double minX, double minY, double maxX, double maxY, int bufWidth, int bufHeight, BandList bandList, IBuffer rasterBuf, boolean adjustToExtent)

                //EXTENT COMPLETO

                buffer = RasterBuffer.getBuffer(f.getDataType(), 25, 24, 3, true, null);

                f.getWindowRaster(f.getExtent().minX(),

                                                        f.getExtent().minY(),

                                                        f.getExtent().maxX(),

                                                        f.getExtent().maxY(),

                                                        25, 24, bandList, buffer, true);

                dataTest1(buffer);

                //ZOOM + subsampling

                buffer = RasterBuffer.getBuffer(f.getDataType(), 8, 8, 3, true, null);

                f.getWindowRaster(        Math.ceil(f.getExtent().minX()),

                                                        Math.ceil(f.getExtent().minY()),

                                                        Math.floor(f.getExtent().maxX()),

                                                        Math.floor(f.getExtent().maxY()),

                                                        8, 8, bandList, buffer, true);

                dataTest3(buffer);

                //ZOOM + supersampling

                buffer = RasterBuffer.getBuffer(f.getDataType(), 20, 20, 3, true, null);

                f.getWindowRaster(        Math.ceil(f.getExtent().minX()),

                                                        Math.ceil(f.getExtent().minY()),

                                                        Math.floor(f.getExtent().minX() + 10),

                                                        Math.floor(f.getExtent().minY() + 10),

                                                        20, 20, bandList, buffer, true);

                //**************************************

                //Prueba de getWindowRaster(double x, double y, double w, double h, BandList bandList, IBuffer rasterBuf, boolean adjustToExtent)

                //COMPLETA

                buffer = RasterBuffer.getBuffer(f.getDataType(), 25, 24, 3, true, null);

                f.getWindowRaster(        f.getExtent().minX(),

                                                        f.getExtent().maxY(),

                                                        f.getExtent().width(),

                                                        f.getExtent().height(),

                                                        bandList, buffer, true);

                dataTest1(buffer);

                //**************************************

        }

        private void dataTest1(IBuffer raster){

                //Upper Left

                assertEquals((int)(raster.getElemByte(0, 0, 0) & 0xff), 14);

                assertEquals((int)(raster.getElemByte(0, 0, 1) & 0xff), 14);

                assertEquals((int)(raster.getElemByte(0, 0, 2) & 0xff), 0);

                //Upper Right

                assertEquals((int)(raster.getElemByte(0, 24, 0) & 0xff), 68);

                assertEquals((int)(raster.getElemByte(0, 24, 1) & 0xff), 90);

                assertEquals((int)(raster.getElemByte(0, 24, 2) & 0xff), 52);

                //Lower Left

                assertEquals((int)(raster.getElemByte(23, 0, 0) & 0xff), 129);

                assertEquals((int)(raster.getElemByte(23, 0, 1) & 0xff), 122);

                assertEquals((int)(raster.getElemByte(23, 0, 2) & 0xff), 106);

                //Lower Right

                assertEquals((int)(raster.getElemByte(23, 24, 0) & 0xff), 145);

                assertEquals((int)(raster.getElemByte(23, 24, 1) & 0xff), 140);

                assertEquals((int)(raster.getElemByte(23, 24, 2) & 0xff), 134);

                assertEquals((int)(raster.getElemByte(6, 6, 0) & 0xff), 21);

                assertEquals((int)(raster.getElemByte(6, 6, 1) & 0xff), 37);

                assertEquals((int)(raster.getElemByte(6, 6, 2) & 0xff), 10);

                assertEquals((int)(raster.getElemByte(6, 23, 0) & 0xff), 91);

                assertEquals((int)(raster.getElemByte(6, 23, 1) & 0xff), 105);

                assertEquals((int)(raster.getElemByte(6, 23, 2) & 0xff), 92);

                assertEquals((int)(raster.getElemByte(23, 6, 0) & 0xff), 153);

                assertEquals((int)(raster.getElemByte(23, 6, 1) & 0xff), 133);

                assertEquals((int)(raster.getElemByte(23, 6, 2) & 0xff), 122);

                assertEquals((int)(raster.getElemByte(9, 14, 0) & 0xff), 63);

                assertEquals((int)(raster.getElemByte(9, 14, 1) & 0xff), 69);

                assertEquals((int)(raster.getElemByte(9, 14, 2) & 0xff), 55);

                assertEquals((int)(raster.getElemByte(6, 13, 0) & 0xff), 70);

                assertEquals((int)(raster.getElemByte(6, 13, 1) & 0xff), 78);

                assertEquals((int)(raster.getElemByte(6, 13, 2) & 0xff), 55);

        }

        private void dataTest2(IBuffer raster){

                //Upper Left

                assertEquals((int)(raster.getElemByte(0, 0, 0) & 0xff), 80);

                assertEquals((int)(raster.getElemByte(0, 0, 1) & 0xff), 96);

                assertEquals((int)(raster.getElemByte(0, 0, 2) & 0xff), 60);

                //Upper Right

                assertEquals((int)(raster.getElemByte(0, 1, 0) & 0xff), 80);

                assertEquals((int)(raster.getElemByte(0, 1, 1) & 0xff), 96);

                assertEquals((int)(raster.getElemByte(0, 1, 2) & 0xff), 60);

                //Lower Left

                assertEquals((int)(raster.getElemByte(1, 0, 0) & 0xff), 82);

                assertEquals((int)(raster.getElemByte(1, 0, 1) & 0xff), 97);

                assertEquals((int)(raster.getElemByte(1, 0, 2) & 0xff), 68);

                //Lower Right

                assertEquals((int)(raster.getElemByte(1, 1, 0) & 0xff), 82);

                assertEquals((int)(raster.getElemByte(1, 1, 1) & 0xff), 97);

                assertEquals((int)(raster.getElemByte(1, 1, 2) & 0xff), 68);

        }

        private void dataTest3(IBuffer raster){

                //Upper Left

                assertEquals((int)(raster.getElemByte(0, 0, 0) & 0xff), 14);

                assertEquals((int)(raster.getElemByte(0, 0, 1) & 0xff), 14);

                assertEquals((int)(raster.getElemByte(0, 0, 2) & 0xff), 0);

                //Upper Right

                assertEquals((int)(raster.getElemByte(0, 7, 0) & 0xff), 75);

                assertEquals((int)(raster.getElemByte(0, 7, 1) & 0xff), 84);

                assertEquals((int)(raster.getElemByte(0, 7, 2) & 0xff), 57);

                //Lower Left

                assertEquals((int)(raster.getElemByte(7, 0, 0) & 0xff), 97);

                assertEquals((int)(raster.getElemByte(7, 0, 1) & 0xff), 106);

                assertEquals((int)(raster.getElemByte(7, 0, 2) & 0xff), 87);

                //Lower Right

                assertEquals((int)(raster.getElemByte(7, 7, 0) & 0xff), 215);

                assertEquals((int)(raster.getElemByte(7, 7, 1) & 0xff), 215);

                assertEquals((int)(raster.getElemByte(7, 7, 2) & 0xff), 215);

        }

        /**

         * Imprime todos los pixels de la fuente de datos en RGB

         */

        private void print(IBuffer raster){

                for(int line = 0; line < raster.getHeight(); line++){

                        for(int col = 0; col < raster.getWidth(); col++)

                                System.out.print("(" + (int)(raster.getElemByte(line, col, 0) & 0xff) + " " + (int)(raster.getElemByte(line, col, 1) & 0xff) + " " + (int)(raster.getElemByte(line, col, 2) & 0xff) + ")");

                        System.out.println();

                }

        }

}