Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.cache.buffer.impl.rocache;

import java.io.IOException;

import junit.framework.TestCase;

import org.gvsig.jgdal.GdalException;

import org.gvsig.raster.cache.buffer.Buffer;

import org.gvsig.raster.cache.buffer.BufferDataSource;

import org.gvsig.raster.cache.buffer.exception.OperationNotSupportedException;

import org.gvsig.raster.cache.buffer.exception.ProcessInterruptedException;

import org.gvsig.raster.cache.buffer.impl.io.BufferDataSourceImpl;

import org.gvsig.raster.cache.buffer.impl.io.GdalRead;

import org.gvsig.raster.cache.buffer.impl.memory.RasterMemoryBuffer;

/**

 * This test read a RGB image and remove one band from the read buffer. Then compare 

 * the bands that have not been removed with the source bands. As a result of this test a raster

 * of two bands is created and saved in the hard disk

 * 

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

 */

public class TestRemoveBand extends TestCase {

        private String rasterIn = "./src/test/resources/image/001m09_1_0.tif";

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void setUp() {

                System.err.println("ROCache TestRemoveBand running...");

        }

        public void testStack() {

                long t1 = System.currentTimeMillis();

                GdalRead input = null;

                try {

                        //Read input data

                        input = new GdalRead(rasterIn);

                        Object dataIn = input.readBlock(0, 0, input.getWidth(), input.getHeight());

                        //Create Buffer

                        Buffer buf1 = new RasterMemoryBuffer(input.getRasterBufTypeFromGdalType(input.getDataType()), 

                                        input.getWidth(), 

                                        input.getHeight(),

                                        input.getBandCount(),

                                        true);

                        //Set data to buffer

                        setLines(dataIn, buf1);

                        BufferDataSource ds = null;

                        try {

                                ds = new BufferDataSourceImpl(rasterIn);

                        } catch (IOException e) {

                                e.printStackTrace();

                        }

                        Buffer buf2 = new RasterReadOnlyBuffer(ds);

                        //Remove bands

                        buf2.removeBand(1);

                        buf1.removeBand(1);

                        compareIRasterBuffer(buf1, buf2);

                } catch (GdalException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                } catch (ProcessInterruptedException e) {

                        e.printStackTrace();

                } catch (OperationNotSupportedException e) {

                        e.printStackTrace();

                }

                long t2 = System.currentTimeMillis();

                System.out.println("Tiempo ROCache TestRemoveBand: " + (t2 - t1) + " milisegundos");

        }

        private void setLines(Object dataIn, Buffer buf) throws OperationNotSupportedException {

                if(dataIn instanceof byte[][][]) {

                    byte[][][] d = (byte[][][])dataIn;

                    for (int iBand = 0; iBand < d.length; iBand++) {

                                for (int row = 0; row < d[iBand].length; row++) {

                                        buf.setLineInBandByte(d[iBand][row], row, iBand);

                                }

                        }

            }

        }

        public void compareIRasterBuffer(Buffer b1, Buffer b2) {

                for (int iBand = 0; iBand < b1.getBandCount(); iBand++) {

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

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

                                byte b = b1.getElemByte(i, j, iBand);

                                byte c = b2.getElemByte(i, j, iBand);

                                if(b != c)

                                        System.out.println("B:" + iBand + "" + i + " " + j);

                                assertEquals(c, b);

                        }

                }

                }

        }

}