Application: gvSIG desktop » gvSIG raster

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

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.exception.OperationNotSupportedException;

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

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

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

/**

 * This test reads a Tiff from disk with the class TIFFRead and load data in

 * a memory buffer. Then it saves those bytes in a new out file in the same format.

 * Finally it compares the intput data with the output data to test if there are some

 * diferences.

 * 

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

 */

public class TestRWTiffAndCompareData extends TestCase {

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

        private String rasterOut = "/tmp/out.tif";

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void setUp() {

                System.err.println("MemoryBuffer TestRWTiffAndCompareData running...");

        }

        public void testStack() {

                long t1 = System.currentTimeMillis();

                GdalRead input = null;

                GdalWrite out = null;

                try {

                        //Read input data

                        input = new GdalRead(rasterIn);

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

                        //Create Buffer

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

                                                                                                                input.getWidth(), 

                                                                                                                input.getHeight(),

                                                                                                                input.getBandCount(),

                                                                                                                true);

                        //Set data to buffer

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

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

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

                                        for (int j = 0; j < d[i].length; j++) {

                                                buf.setLineInBandByte(d[i][j], j, i);

                                        }

                                }

                    }

                        //Write output file 

                        out = new GdalWrite(rasterOut, input.getBandCount(), input.getDataType(), input.getWidth(), input.getHeight(), GdalWrite.COLOR_INTERP_RGB);

                        out.writeBands(buf.getBands());

                        out.close();        

                        //Open output file and compare

                        GdalRead newFile = new GdalRead(rasterOut);

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

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

                        byte[][][] d2 = (byte[][][])dataOut;

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

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

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

                                                if(d1[iBand][row][col] != d2[iBand][row][col])

                                                        System.out.println("BAND:" + iBand +" ROW:" + row + " COL:" + col + " " + d1[iBand][row][col] + " " + d2[iBand][row][col]);

                                                assertEquals(d1[iBand][row][col], d2[iBand][row][col]);

                                        }

                                }

                        }

                } 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 MemoryBuffer TestRWTiffAndCompareData: " + (t2 - t1) + " milisegundos");

        }

}