| gvSIG desktop 1

/* 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.dataset.properties;

import org.gvsig.raster.RasterLibrary;

import org.gvsig.raster.dataset.FileNotOpenException;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.InvalidSetViewException;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.util.Histogram;

/**

 * Clase para la gesti?n de histogramas de un raster. Es la encargada del calculo de un histograma

 * total o parcial de un raster a partir de los datos de disco. Adem?s tambi?n es la encargada de gestionar

 * salvar este histograma a .rmf. En caso de que solicite un histograma del raster completo este ir?

 * a buscarlo al fichero rmf asociado antes de calcularlo por si ya existise. Al realizar el calculo

 * del histograma de la imagen completa este ser? salvado en el fichero .rmf asociado al raster.

 * 

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

 */

public class DatasetHistogram {

        /**

         * Histograma de la imagen completa

         */

        private Histogram                                        histogram = null;

        /**

         * Dataset del cual se calcula el histograma

         */

        private RasterDataset                                dataset = null;

        private int percent = 0;

        private boolean canceled = false;

        /**

         * Constructor

         * @param dataset

         */

        public DatasetHistogram(RasterDataset dataset){

                this.dataset = dataset;

        }

        /**

         * Obtiene el minimo valor de las estadisticas de un histograma.

         * @return double

         */

        public double getMinimum() {

                return dataset.getStatistics().getMinimun();

        }

        /**

         * Obtiene el maximo valor de las estadisticas de un histograma.

         * @return double

         */

        public double getMaximum() {

                return dataset.getStatistics().getMaximun();

        }

        /**

         * Obtiene el histograma. Si puede conseguirlo del fichero rmf ir? all? a 

         * buscarlo sino lo calcular?.

         * @return histograma 

         */

        public Histogram getHistogram() throws FileNotOpenException, RasterDriverException {

                try {

                        if (dataset != null) {

                                dataset.getStatistics().calcFullStatistics();

                                if (isCanceled()) 

                                        return null;

                                if (dataset.getDataType() == IBuffer.TYPE_BYTE)

                                        histogram = new Histogram(dataset.getBandCount(), 255, dataset.getStatistics().getMinimun(), dataset.getStatistics().getMaximun());

                                else

                                        histogram = new Histogram(dataset.getBandCount(), RasterLibrary.defaultNumberOfClasses, dataset.getStatistics().getMinimun(), dataset.getStatistics().getMaximun());

                                return getHistogramByDataType();

                        }

                } catch (InvalidSetViewException e) {

                        //La vista se selecciona autom?ticamente no deber?a darse esta excepci?n

                }

                return null;

        }

        /**

         * Obtiene el histograma teniendo en cuenta la lista de clases

         * @return Histograma correspondiente a la lista de clases

         */

        private Histogram getHistogramByDataType() throws InvalidSetViewException, FileNotOpenException, RasterDriverException {

                percent = 0;

                for (int band = 0; band < dataset.getBandCount(); band++) {

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

                                Object obj = dataset.readCompleteLine(line, band);

                                switch(dataset.getDataType()) {

                                        case IBuffer.TYPE_BYTE:

                                                byte[] bLine = (byte[])obj;

                                                for (int k = 0; k < bLine.length; k++)

                                                        histogram.incrementPxValue(band, (double)(bLine[k] & 0xff));

                                                break;

                                        case IBuffer.TYPE_SHORT:

                                                short[] sLine = (short[])obj;

                                                for (int k = 0; k < sLine.length; k++)

                                                        histogram.incrementPxValue(band, (double)(sLine[k] & 0xffff));

                                                break;

                                        case IBuffer.TYPE_INT:

                                                int[] iLine = (int[])obj;

                                                for (int k = 0; k < iLine.length; k++)

                                                        histogram.incrementPxValue(band, (double)(iLine[k] & 0xffffffff));

                                                break;

                                        case IBuffer.TYPE_FLOAT:

                                                float[] fLine = (float[])obj;

                                                for (int k = 0; k < fLine.length; k++)

                                                        histogram.incrementPxValue(band, (double)fLine[k]);

                                                break;

                                        case IBuffer.TYPE_DOUBLE:

                                                double[] dLine = (double[])obj;

                                                for (int k = 0; k < dLine.length; k++)

                                                        histogram.incrementPxValue(band, (double)dLine[k]);

                                                break;

                                }

                                if (isCanceled()) 

                                        return null;

                                percent = ((band * dataset.getHeight() + line) * 100) / (dataset.getBandCount() * dataset.getHeight());

                        }

                }

                percent = 100;

                return histogram;

        }

        public void resetPercent() {

                percent=0;

        }

        public int getPercent() {

                return percent;

        }

        public void setCanceled(boolean value) {

                canceled = value;

                if (dataset != null)

                        dataset.getStatistics().setCanceled(value);

        }

        public boolean isCanceled() {

                return canceled;

        }

}