Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.fmap.dal.coverage.buffer;

import org.gvsig.fmap.dal.coverage.BaseTestCase;

import org.gvsig.fmap.dal.coverage.datastruct.BufferHistogram;

import org.gvsig.fmap.dal.coverage.exception.HistogramException;

import org.gvsig.fmap.dal.coverage.exception.NotSupportedExtensionException;

import org.gvsig.fmap.dal.coverage.exception.ProcessInterruptedException;

import org.gvsig.fmap.dal.coverage.exception.RasterDriverException;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.raster.impl.datastruct.BufferHistogramImpl;

/**

 * Test a un histograma de una imagen de 3 bandas de un byte por dato leido con gdal. 

 * El test comprueba valores a lo largo de todo el histograma

 * 

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

 */

public class TestGdalByteHistogram extends BaseTestCase {

        private String path = baseDir + "histograma.bmp";        

        private RasterDataStore       f              = null;        

    public void start() throws Exception {

                this.setUp();

                this.testStack();

        }

    @Override

    protected void doSetUp() throws Exception {

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

                try {

                        super.doSetUp();

                } catch (Exception e) {

                        e.printStackTrace();

                }

        }

        public void testStack() {

                try {

                        f = manager.open(path);

                } catch (NotSupportedExtensionException e) {

                        return;

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

//                ds = new BufferFactory(f);

                BufferHistogram histogram = null;

                try {

                        histogram = ((RasterDataStore)f).getHistogramComputer().getBufferHistogram();

                        histogram = BufferHistogramImpl.convertHistogramToRGB(histogram);

                } catch (ProcessInterruptedException e) {

                        e.printStackTrace();

                } catch (HistogramException e) {

                        e.printStackTrace();

                }

                //print(histogram);

                resultTest(histogram);

        }

        private void resultTest(BufferHistogram histogram) {

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

                        switch (i) {

                                case 0:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 1);

                                        break;

                                case 1:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 2);

                                        break;

                                case 2:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 3);

                                        break;

                                case 126:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 4);

                                        break;

                                case 127:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 5);

                                        break;

                                case 128:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 6);

                                        break;

                                case 129:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 7);

                                        break;

                                case 253:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 8);

                                        break;

                                case 254:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 9);

                                        break;

                                case 255:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 10);

                                        break;

                                default:

                                        assertEquals((int) histogram.getHistogramValue(0, i), 0);

                                        break;

                        }

                }

        }

        public void print(BufferHistogram histogram) {

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

                        int cont = 0;

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

                                double var = histogram.getHistogramValue(i, j);

                                if (var != 0) {

                                        System.out.println(j + ": " + var);

                                        cont += var;

                                }

                        }

                        System.out.println("");

                        System.out.println(cont + " ");

                }

        }

}