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.Band;

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

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

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

/**

 * This test swap several bands of a raster and save the result in other file.

 * Finally it compares the output bands with the input bands.

 * <P>

 * The array to test is {2, 0, 1}. That means the band 0 goes to the position two, the

 * band one goes to the position zero and the band two goes to the position one. 

 * When the swap operation is executed, this test will check comparing the result 

 * bands with the input bands 

 * </P>

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

 */

public class TestSwapBands extends TestCase {

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

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

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void setUp() {

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

        }

        public void testStack() {

                long t1 = System.currentTimeMillis();

                GdalRead input1 = null;

                GdalWrite out = null;

                try {

                        //Read input data

                        input1 = new GdalRead(rasterIn);

                        Object dataIn1 = input1.readBlock(0, 0, input1.getWidth(), input1.getHeight());

                        //Create Buffer

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

                                                                                                                input1.getWidth(), 

                                                                                                                input1.getHeight(),

                                                                                                                input1.getBandCount(),

                                                                                                                true);

                        //Set data to buffer

                        setBytes(dataIn1, buf1);

                        Buffer buf2 = new RasterMemoryBuffer(input1.getRasterBufTypeFromGdalType(input1.getDataType()), 

                                                                                                                input1.getWidth(), 

                                                                                                                input1.getHeight(),

                                                                                                                input1.getBandCount(),

                                                                                                                true);

                        //Set data to buffer

                        setBytes(dataIn1, buf2);

                        try {

                                buf2.swapBands(new int[]{2, 0, 1});

                        } catch (WrongParameterException e) {

                                e.printStackTrace();

                        }

                        //Write output file 

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

                        out.writeBands(buf2.getBands());

                        out.close();        

                        Band band1 = buf1.getBand(0);

                        Band band2 = buf2.getBand(1);

                        compareBands(band1, band2);

                        band1 = buf1.getBand(1);

                        band2 = buf2.getBand(2);

                        compareBands(band1, band2);

                        band1 = buf1.getBand(2);

                        band2 = buf2.getBand(0);

                        compareBands(band1, band2);

                        buf1.free();

                        buf2.free();

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

        }

        public void compareBands(Band band1, Band band2) {

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

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

                                byte b1 = band1.getElemByte(i, j);

                                byte b2 = band2.getElemByte(i, j);

                                assertEquals(b1, b2);

                        }

                }

        }

//        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);

//                                }

//                        }

//            }

//        }

        private void setBytes(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++) {

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

                                                buf.setElem(row, col, iBand, d[iBand][row][col]);

                                        }

                                }

                        }

            }

        }

}