| gvSIG desktop 1

/*

 * Created on 9-ago-2006

 *

 * To change the template for this generated file go to

 * Window>Preferences>Java>Code Generation>Code and Comments

 */

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

 *

 * Copyright (C) 2004 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 tmp;

import junit.framework.TestCase;

import org.gvsig.raster.RasterLibrary;

import org.gvsig.raster.dataaccess.DataSource;

import org.gvsig.raster.dataaccess.buffer.RasterBufferInvalidAccessException;

import org.gvsig.raster.dataaccess.buffer.RasterBufferInvalidException;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.GridExtent;

import org.gvsig.raster.grid.GridInterpolated;

import org.gvsig.raster.grid.OutOfGridException;

import org.gvsig.raster.shared.Extent;

/**

 * Este test prueba el acceso a datos a traves de un grid

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

 *

 */

public class TestGrid extends TestCase{

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

        private String path = baseDir + "03AUG23153350-M2AS-000000122423_01_P001-BROWSE.jpg";

        private RasterDataset f1 = null;

        private DataSource ds = null;

        /**

         * Casos de prueba para el grid

         * caso 0: Inicializaci?n de grid a partir de un rasterbuffer creado con coordenadas pixel de la imagen completa

         * caso 1: Inicializaci?n de grid a partir de una fuente de datos y un extent de imagen completo en coordenadas reales

         * caso 2: Grabado de datos sobre un grid y uso de operaciones

         * caso 3: Acceso a datos interpolados

         */

        private int caso = 2;

        static{

                RasterLibrary.wakeUp();

        }

        public void setUp() {

                System.out.println("TestGrid running...");

                try {

                        f1 = RasterDataset.openFile(null, path);

                } catch (NotSupportedExtensionException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

                ds = new DataSource(f1);

        }

        public void testStack(){

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

                switch(caso){

                case 0:        ds.addDrawableBands(drawableBands);

                                ds.setAreaOfInterest(0, 0, f1.getWidth(), f1.getHeight());

                                try {

                                        Grid g = new Grid(ds, drawableBands);

                                        System.out.println (g.getDataType());

                                        System.out.println (g.getNY() + " " + g.getNX());

                                        testCellBuffer(g);

                                } catch (RasterBufferInvalidException e1) {

                                        e1.printStackTrace();

                                } catch (RasterBufferInvalidAccessException e2) {

                                        e2.printStackTrace();

                                }

                                break;

                case 1:        GridExtent ge = new GridExtent(ds.getExtent(), ds.getXCellSize());

                                ds.addDrawableBands(drawableBands);

                                Extent e = f1.getExtent();

                                double minX = e.getMin().getX();

                                double minY = e.getMin().getY();

                                double maxX = e.getMax().getX();

                                double maxY = e.getMax().getY();

                                ds.setAreaOfInterest(minX, maxY, maxX - minX, maxY - minY, true);

                                try {

                                        Grid g = new Grid(ds, drawableBands, ge);

                                        System.out.println (g.getDataType());

                                        System.out.println (g.getNY() + " " + g.getNX());

                                        testCellBuffer(g);

                                } catch (RasterBufferInvalidException e1) {

                                        e1.printStackTrace();

                                } catch (RasterBufferInvalidAccessException e2) {

                                        e2.printStackTrace();

                                }

                                break;

                case 2: ds.addDrawableBands(drawableBands);

                                ds.setAreaOfInterest(0, 0, f1.getWidth(), f1.getHeight());

                                try {

                                        Grid g = new Grid(ds, drawableBands);

                                        for(int i = 400; i < 410; i ++)

                                                System.out.print("("+g.getCellValueAsByte(i, 400)+","+ds.getRasterBuf().getElemByte(400, i, 0)+") ");

                                        System.out.println();

                                        g.setCellValue(400, 400, (byte)10);

                                        g.addToCellValue(401, 400, (byte)2);

                                        g.multiply(0.5);

                                        for(int i = 400; i < 410; i ++)

                                                System.out.print("("+g.getCellValueAsByte(i, 400)+","+ds.getRasterBuf().getElemByte(400, i, 0)+") ");

                                        System.out.println();

                                        g.assign((byte)3);

                                        for(int i = 400; i < 410; i ++)

                                                System.out.print("("+g.getCellValueAsByte(i, 400)+","+ds.getRasterBuf().getElemByte(400, i, 0)+") ");

                                } catch (RasterBufferInvalidException e1) {

                                        e1.printStackTrace();

                                } catch (RasterBufferInvalidAccessException e2) {

                                        e2.printStackTrace();

                                } catch (OutOfGridException e2) {

                                        e2.printStackTrace();

                                }

                                break;

                case 3: //TODO: TEST: Probar acceso datos interpolados

                                ds.addDrawableBands(drawableBands);

                                ds.setAreaOfInterest(0, 0, f1.getWidth(), f1.getHeight());

                                try {

                                        Grid g = new Grid(ds, drawableBands);

                                        g.setInterpolationMethod(GridInterpolated.INTERPOLATION_Bilinear);

                                        for(int i = 400; i < 410; i ++)

                                                System.out.print("("+g.getCellValueAsByte(i, 400)+","+ds.getRasterBuf().getElemByte(400, i, 0)+") ");

                                        System.out.println();

                                } catch (RasterBufferInvalidException e1) {

                                        e1.printStackTrace();

                                } catch (RasterBufferInvalidAccessException e2) {

                                        e2.printStackTrace();

                                }

                                break;

                }

        }

        /**

         * Compara todos los elementos del buffer con los del grid para comprobar

         * la correcci?n de la lectura.

         * @param g

         * @throws RasterBufferInvalidAccessException

         */

        private void testCellBuffer(Grid g) throws RasterBufferInvalidAccessException{

                int nCols = f1.getWidth();

                int nRows = f1.getHeight();

                for(int i = 0; i < nRows; i ++){

                        for(int j = 0; j < nCols; j ++){

                                assertEquals(g.getCellValueAsByte(j, i), ds.getRasterBuf().getElemByte(i, j, 0));

                                //System.out.print("("+g.getCellValueAsByte(i, 400)+","+ds.getRasterBuf().getElemByte(400, i, 0)+") ");

                        }

                        //System.out.println();

                }

        }

}