Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2007 IVER T.I. and Generalitat Valenciana.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.raster.dataaccess;

import junit.framework.TestCase;

import org.gvsig.raster.dataaccess.DataSource;

import org.gvsig.raster.driver.FileNotOpenException;

import org.gvsig.raster.driver.IBuffer;

import org.gvsig.raster.driver.NotSupportedExtensionException;

import org.gvsig.raster.driver.RasterDataset;

import org.gvsig.raster.driver.RasterDriverException;

import org.gvsig.raster.shared.DataClass;

import org.gvsig.raster.shared.DataClassList;

import org.gvsig.raster.shared.RasterLibrary;

/**

 * Test a un histograma usando clases. El test se realiza sobre un raster en 

 * coma flotante de 28x25 pixels. Se crean clases de tres tipo de intervalos distintos y

 * se testean los valores obtenidos en el histograma resultante. El ?ltimo

 * test comprueba la generaci?n autom?tica de clases.

 * 

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

 */

public class TestGdalClassHistogram extends TestCase{

        private String baseDir = "./test-images/";

        private String path = baseDir + "miniRaster28x25F32.tif";        

        private RasterDataset f = null;

        private DataSource ds = null;

        public void start(){

                this.setUp();

                this.testStack();

        }

        public void setUp() {

                System.err.println("TestGdalClassHistogram running...");

        }

        static{

                RasterLibrary.wakeUp();

        }

        public void testStack(){

                int[] drawableBands = {0, 1, 2};

                try {

                        f = RasterDataset.openFile(null, path);

                } catch (NotSupportedExtensionException e) {

                        return;

                } catch (RasterDriverException e) {

                        return;

                }

                ds = new DataSource(f);

                ds.addDrawableBands(drawableBands);

                ds.setAreaOfInterest(0, 0, 28, 25);

                //printData();

                DataClassList classList = new DataClassList();

                DataClass classInterval1 = new DataClass(0D, 1000D, 0, "");

                DataClass classInterval2 = new DataClass(1000D, 2000D, 1, "");

                classList.addClass(classInterval1);

                classList.addClass(classInterval2);

                long[][] histogram = null;

                try {

                        histogram = ds.getHistogram(classList);

                } catch (FileNotOpenException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

                testHist1(histogram);

                classList.clear();

                classInterval1 = new DataClass(0D, 1200D, 0, "");

                classInterval2 = new DataClass(1200D, 2000D, 1, "");

                classList.addClass(classInterval1);

                classList.addClass(classInterval2);

                try {

                        histogram = ds.getHistogram(classList);

                } catch (FileNotOpenException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

                testHist2(histogram);

                classList.clear();

                classInterval1 = new DataClass(0D, 1150D, 0, "");

                classInterval2 = new DataClass(115D, 1175D, 1, "");

                DataClass classInterval3 = new DataClass(1175D, 1200D, 2, "");

                DataClass classInterval4 = new DataClass(1200D, 1210D, 3, "");

                DataClass classInterval5 = new DataClass(1210D, 2000D, 4, "");

                classList.addClass(classInterval1);

                classList.addClass(classInterval2);

                classList.addClass(classInterval3);

                classList.addClass(classInterval4);

                classList.addClass(classInterval5);

                try {

                        histogram = ds.getHistogram(classList);

                } catch (FileNotOpenException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

                testHist3(histogram);

                try {

                        histogram = ds.getHistogram(null);

                } catch (FileNotOpenException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

                testHist4(histogram);

                //print(histogram);

        }

        private void testHist1(long[][] histogram){

                assertEquals(histogram[0][0], 0);

                assertEquals(histogram[0][1], 700);

        } 

        private void testHist2(long[][] histogram){

                assertEquals(histogram[0][0], 180);

                assertEquals(histogram[0][1], 520);

        }

        private void testHist3(long[][] histogram){

                assertEquals(histogram[0][0], 4);

                assertEquals(histogram[0][1], 38);

                assertEquals(histogram[0][2], 138);

                assertEquals(histogram[0][3], 100);

                assertEquals(histogram[0][4], 420);

        }

        private void testHist4(long[][] histogram){

                assertEquals(histogram[0][0], 10);

                assertEquals(histogram[0][1], 17);

                assertEquals(histogram[0][2], 36);

                assertEquals(histogram[0][3], 72);

                assertEquals(histogram[0][4], 119);

                assertEquals(histogram[0][5], 129);

                assertEquals(histogram[0][6], 98);

                assertEquals(histogram[0][7], 93);

                assertEquals(histogram[0][8], 76);

                assertEquals(histogram[0][9], 49);

        }

        private void print(long[][] histogram){

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

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

                                System.out.print(histogram[i][j] + " ");

                        }

                        System.out.println("");

                }

        }

        /**

         * Imprime todos los pixels de la fuente de datos en RGB

         */

        private void printData(){

                IBuffer raster = ds.getRasterBuf();

                for(int line = 0; line < raster.getHeight(); line++){

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

                                System.out.print("(" + raster.getElemFloat(line, col, 0) + ")");

                        System.out.println();

                }

        }

}