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.impl.io.BufferDataSourceImpl;

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

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

/**

 * This test gets bands from a raster with the method getBufferWithOneBand() and compare

 * each band from original source with obtained bands 

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

 */

public class TestGetBufferWithOneBand extends TestCase {

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

        private String rasterOut1 = "/tmp/out-b1.tif";

        private String rasterOut2 = "/tmp/out-b2.tif";

        private String rasterOut3 = "/tmp/out-b3.tif";

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void setUp() {

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

        }

        public void testStack() {

                long t1 = System.currentTimeMillis();

                GdalRead input = null;

                try {

                        //Read input data

                        input = new GdalRead(rasterIn);

                        BufferDataSource ds = null;

                        try {

                                ds = new BufferDataSourceImpl(rasterIn);

                        } catch (IOException e) {

                                e.printStackTrace();

                        }

                        Buffer buf = new RasterReadOnlyBuffer(ds);

                        Buffer b0 = buf.getBufferWithOneBand(0);

                        Buffer b1 = buf.getBufferWithOneBand(1);

                        Buffer b2 = buf.getBufferWithOneBand(2);

                        //Write output file 

                        GdalWrite out1 = new GdalWrite(rasterOut1, 1, input.getDataType(), input.getWidth(), input.getHeight(), GdalWrite.COLOR_INTERP_GRAY);

                        out1.writeBands(b0.getBands());

                        out1.close();

                        GdalWrite out2 = new GdalWrite(rasterOut2, 1, input.getDataType(), input.getWidth(), input.getHeight(), GdalWrite.COLOR_INTERP_GRAY);

                        out2.writeBands(b1.getBands());

                        out2.close();

                        GdalWrite out3 = new GdalWrite(rasterOut3, 1, input.getDataType(), input.getWidth(), input.getHeight(), GdalWrite.COLOR_INTERP_GRAY);

                        out3.writeBands(b2.getBands());

                        out3.close();

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

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

                                        byte b_0 = buf.getElemByte(row, col, 0);

                                        byte b_1 = buf.getElemByte(row, col, 1);

                                        byte b_2 = buf.getElemByte(row, col, 2);

                                        assertEquals(b_0, b0.getElemByte(row, col, 0));

                                        assertEquals(b_1, b1.getElemByte(row, col, 0));

                                        assertEquals(b_2, b2.getElemByte(row, col, 0));

                                }

                        }

                        b0.free();

                        b1.free();

                        b2.free();

                        buf.free();

                } catch (GdalException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                } catch (OperationNotSupportedException e) {

                        e.printStackTrace();

                }

                long t2 = System.currentTimeMillis();

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

        }

}