Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.lib.buffer.impl.statistics;

import org.gvsig.raster.lib.buffer.api.Band;

import org.gvsig.raster.lib.buffer.api.statistics.HistogramBand;

import org.gvsig.raster.lib.buffer.api.statistics.StatisticsBand;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.task.SimpleTaskStatus;

/**

 * @author fdiaz

 *

 */

public class DefaultStatisticsBand implements StatisticsBand {

    private Band band;

    private boolean calculated;

    private double mean;

    private double max;

    private double secondMax;

    private double min;

    private double secondMin;

    private double variance;

    private HistogramBand histogramBand;

    /**

     * @param band

     */

    public DefaultStatisticsBand(Band band) {

        this.band = band;

        this.calculated = false;

    }

    @Override

    public boolean isCalculated() {

        return calculated;

    }

    @Override

    public void calculate(SimpleTaskStatus status) {

        boolean isMyStatus = false;

        if (status == null) {

            status = ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus("Statistics");

            status.add();

            isMyStatus = true;

        } else {

            status.push();

        }

        try {

            calculated = false;

            status.setRangeOfValues(0, band.getRows() * 2);

            status.message("Calculating statistics band");

            long cont = 0;

            MeanOperation meanOperation = new MeanOperation();

            MaxOperation maxOperation = new MaxOperation();

            MinOperation minOperation = new MinOperation();

            SecondMaxOperation secondMaxOperation = new SecondMaxOperation();

            SecondMinOperation secondMinOperation = new SecondMinOperation();

            VarianceOperation varianceOperation = new VarianceOperation();

            // TODO: CovarianceOperation

            meanOperation.pre();

            maxOperation.pre();

            minOperation.pre();

            secondMaxOperation.pre();

            secondMinOperation.pre();

            varianceOperation.pre();

            for (int row = 0; row < band.getRows(); row++) {

                status.setCurValue(cont++);

                if (status.isCancelled()) {

                    status.abort();

                    return;

                }

                for (int column = 0; column < band.getColumns(); column++) {

                    Object obj = band.get(row, column);

                    if (obj instanceof Number) {

                        double value = ((Number) obj).doubleValue();

                        meanOperation.addValue(value);

                        maxOperation.addValue(value);

                        minOperation.addValue(value);

                        secondMaxOperation.addValue(value);

                        secondMinOperation.addValue(value);

                        varianceOperation.addValue(value);

                    }

                }

            }

            meanOperation.post();

            maxOperation.post();

            minOperation.post();

            secondMaxOperation.post();

            secondMinOperation.post();

            varianceOperation.post();

            this.mean = ((Number) meanOperation.getResult()).doubleValue();

            this.max = ((Number) maxOperation.getResult()).doubleValue();

            this.min = ((Number) minOperation.getResult()).doubleValue();

            this.secondMax = ((Number) secondMaxOperation.getResult()).doubleValue();

            this.secondMin = ((Number) secondMinOperation.getResult()).doubleValue();

            this.variance = ((Number) varianceOperation.getResult()).doubleValue();

            HistogramOperation histogramOperation = new HistogramOperation(this, this.band);

            histogramOperation.pre();

            for (int row = 0; row < band.getRows(); row++) {

                status.setCurValue(cont++);

                if (status.isCancelled()) {

                    status.abort();

                    return;

                }

                for (int column = 0; column < band.getColumns(); column++) {

                    Object obj = band.get(row, column);

                    if (obj instanceof Number) {

                        double value = ((Number) obj).doubleValue();

                        histogramOperation.addValue(value);

                    }

                }

            }

            histogramOperation.post();

            this.histogramBand = ((HistogramBand) histogramOperation.getResult());

            if(isMyStatus){

                status.terminate();

            } else {

                status.pop();

            }

            calculated = true;

        } catch (Exception e) {

            status.abort();

            throw e;

        }

    }

    @Override

    public int getDataType() {

        return this.band.getDataType();

    }

    @Override

    public long getBandLenght() {

        return band.getRows() * band.getColumns();

    }

    @Override

    public double getSecondMax() {

        // TODO ??? y si no est? calculado???

        return this.secondMax;

    }

    @Override

    public double getSecondMin() {

        return this.secondMin;

    }

    @Override

    public double getMaximum() {

        return this.max;

    }

    @Override

    public double getMinimum() {

        return this.min;

    }

    @Override

    public double getMean() {

        return this.mean;

    }

    @Override

    public double getVariance() {

        return this.variance;

    }

    @Override

    public Object getTailTrimValue(double percent) {

        return getHistogramBand().getTailTrimValue(percent);

    }

    @Override

    public Object getTailTrimValue(int pos) {

        return getHistogramBand().getTailTrimValue(pos);

    }

    @Override

    public HistogramBand getHistogramBand() {

        return this.histogramBand;

    }

    @Override

    public double getCovariance(StatisticsBand statisticBand) {

        Band otherBand = ((DefaultStatisticsBand) statisticBand).getBand();

        long otherBandLenght = statisticBand.getBandLenght();

        if (getBandLenght() != otherBandLenght) {

            return 0;

        }

        MeanOperation productMeanOperation = new MeanOperation();

        productMeanOperation.pre();

        for (int row = 0; row < band.getRows(); row++) {

            for (int column = 0; column < band.getColumns(); column++) {

                Object obj = band.get(row, column);

                Object obj2 = otherBand.get(row, column);

                double value = ((Number) obj).doubleValue() * ((Number) obj2).doubleValue();

                productMeanOperation.addValue(value);

            }

        }

        productMeanOperation.post();

        return ((Number) productMeanOperation.getResult()).doubleValue() - (this.getMean() * statisticBand.getMean());

    }

    @Override

    public double getMedian() {

        return getHistogramBand().getMedian();

    }

    protected Band getBand() {

        return this.band;

    }

    @Override

    public long getNoDataValuesCount() {

        // TODO Auto-generated method stub

        return 0;

    }

}