Application: gvSIG desktop

/* 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 org.gvsig.raster.grid.filter;

import junit.framework.TestCase;

import org.gvsig.raster.RasterLibrary;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.IStatistics;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.Params;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.dataset.properties.DatasetStatistics;

import org.gvsig.raster.grid.filter.enhancement.BrightnessContrastListManager;

import org.gvsig.raster.grid.filter.enhancement.EnhancementListManager;

/**

 * Test para el control de tipos de los filtros que se insertan en la lista. 

 * Crea un RasterFilterList de prueba con un buffer e inserta filtros. Despu?s comprueba

 * que el tipo de dato de la salida de uno concuerda con el tipo de datos de la entrada

 * del siguiente.

 * 

 * Supone que el par?metro insertionMode es ADD_END.

 * 

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

 *

 */

public class TestControlTypesFilterADD_END extends TestCase{

        private RasterFilterList                list = new RasterFilterList();

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

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

        private RasterDataset f1 = null;

        static{

                RasterLibrary.wakeUp();        

        }

        public void setUp() {

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

                try {

                        f1 = RasterDataset.open(null, path1);

                } catch (NotSupportedExtensionException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

        }

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void testStack() {                

                IBuffer buf = RasterBuffer.getBuffer(IBuffer.TYPE_SHORT, 10, 10, 3, true);

                list.setInitRasterBuf(buf);

                RasterFilterListManager manager = new RasterFilterListManager(list);

                EnhancementListManager m0 = (EnhancementListManager)manager.getManagerByClass(EnhancementListManager.class);

                BrightnessContrastListManager m1 = (BrightnessContrastListManager)manager.getManagerByClass(BrightnessContrastListManager.class);

                //Test 1: Byte, Byte

                m0.addEnhancedFilter(false, f1.getStatistics(), 0, RasterFilterList.ADD_END, new int[]{0, 1, 2});

                assertEquals(list.getOutDataType(), IBuffer.TYPE_BYTE); 

                m1.addBrightnessFilter(10, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_BYTE);

                //Test 2: Short, Short, Byte

                list.clear();

                m1.addBrightnessFilter(10, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_SHORT);

                m1.addContrastFilter(12, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_SHORT);

                m0.addEnhancedFilter(false, f1.getStatistics(), 0, RasterFilterList.ADD_END, new int[]{0, 1, 2});

                assertEquals(list.getOutDataType(), IBuffer.TYPE_BYTE);

                //Test 3: Double, Double, Float, Float, Byte, Byte

                list.clear();

                buf = RasterBuffer.getBuffer(IBuffer.TYPE_DOUBLE, 10, 10, 3, true);

                list.setInitRasterBuf(buf);

                m1.addContrastFilter(12, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_DOUBLE);

                m1.addBrightnessFilter(10, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_DOUBLE);

                RasterFilterForTest f = new RasterFilterForTest();

                list.add(f);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_FLOAT);

                m1.addBrightnessFilter(10, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_FLOAT);

                m0.addEnhancedFilter(false, f1.getStatistics(), 0, RasterFilterList.ADD_END, new int[]{0, 1, 2});

                assertEquals(list.getOutDataType(), IBuffer.TYPE_BYTE);

                m1.addBrightnessFilter(10, RasterFilterList.ADD_END);

                assertEquals(list.getOutDataType(), IBuffer.TYPE_BYTE);

                //System.out.println(RasterUtilities.typesToString(list.getOutDataType()));

        }

        /**

         * Filtro para de prueba para el test TestRasterFilterList

         * 

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

         */

        class RasterFilterForTest extends RasterFilter {

                public void pre() {

                }

                public void post() {

                }

                public void process(int x, int y) {

                }

                public int getInRasterDataType() {

                        return IBuffer.TYPE_BYTE;

                }

                public int getOutRasterDataType() {

                        return IBuffer.TYPE_FLOAT;

                }

                public Object getResult(String name) {

                        return null;

                }

                public String getGroup() {

                        return null;

                }

                public Params getUIParams() {

                        Params params = new Params();

                        return params;

                }

                public void setUIParams(Params value) {

                }

                public Params getUIDefaultParams() {

                        Params params = new Params();

                        return params;

                }

        }

}