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

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

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

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 insert one band at the end of the layer. The data of this new band is a copy 

 * of the band in position zero. Finally data of band zero and band three are compared

 * checking if both have the same values. 

 *  

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

 */

public class TestReplicateBand extends TestCase {

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

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

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void setUp() {

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

        }

        public void testStack() {

                long t1 = System.currentTimeMillis();

                GdalRead input1 = null;

                GdalWrite out = null;

                try {

                        //Read input data

                        input1 = new GdalRead(rasterIn);

                        BufferDataSource ds = null;

                        try {

                                ds = new BufferDataSourceImpl(rasterIn);

                        } catch (IOException e) {

                                e.printStackTrace();

                        }

                        Buffer buf = new RasterReadOnlyBuffer(ds);

                        buf.replicateBand(0, 3);

                        //Write output file 

                        out = new GdalWrite(rasterOut, 4, input1.getDataType(), input1.getWidth(), input1.getHeight(), GdalWrite.COLOR_INTERP_GRAY);

                        out.writeBands(buf.getBands());

                        out.close();        

                        Band band1 = buf.getBand(0);

                        Band band2 = buf.getBand(3);

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

                                }

                        }

                        buf.free();

                } catch (GdalException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

                long t2 = System.currentTimeMillis();

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

        }

}