Application: gvSIG desktop » gvSIG raster

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

import java.util.Arrays;

import java.util.Iterator;

import java.util.List;

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.Statistics;

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.task.SimpleTaskStatus;

/**

 * @author fdiaz

 *

 */

public class DefaultStatistics implements Statistics {

    StatisticsBand[] statisticsBands;

    boolean calculated;

    double[][] varCov;

    /**

     * @param bands

     *

     */

    public DefaultStatistics( List<Band> bands) {

        statisticsBands = new StatisticsBand[bands.size()];

        for (int i = 0; i < statisticsBands.length; i++) {

            statisticsBands[i] = new DefaultStatisticsBand(bands.get(i));

        }

        varCov = new double[statisticsBands.length][statisticsBands.length];

        calculated = false;

    }

    @Override

    public long[] getNumberOfValues() {

        long[] result = new long[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getBandLenght();

        }

        return result;

    }

    @Override

    public double[] getMax() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getMaximum();

        }

        return result;

    }

    @Override

    public double[] getSecondMax() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getSecondMax();

        }

        return result;

    }

    @Override

    public double[] getSecondMin() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getSecondMin();

        }

        return result;

    }

    @Override

    public double getMaximun() {

        double result = Double.NEGATIVE_INFINITY;

        for (int i = 0; i < statisticsBands.length; i++) {

            double max = statisticsBands[i].getMaximum();

            if(result<max){

                result = max;

            }

        }

        return result;

    }

    @Override

    public double getMinimun() {

        double result = Double.POSITIVE_INFINITY;

        for (int i = 0; i < statisticsBands.length; i++) {

            double min = statisticsBands[i].getMinimum();

            if(result>min){

                result = min;

            }

        }

        return result;

    }

    @Override

    public double[] getMean() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getMean();

        }

        return result;

    }

    @Override

    public double[] getMedian() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getMedian();

        }

        return result;

    }

    @Override

    public double[] getMin() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getMinimum();

        }

        return result;

    }

    @Override

    public double[] getVariance() {

        double[] result = new double[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]= statisticsBands[i].getVariance();

        }

        return result;

    }

    @Override

    public int getBandCount() {

        return statisticsBands.length;

    }

    @Override

    public boolean isCalculated() {

        return calculated;

    }

    @Override

    public void calculate(SimpleTaskStatus status) {

        calculated = false;

        boolean isMyStatus = false;

        if (status == null) {

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

            status.add();

            isMyStatus = true;

        } else {

            status.push();

        }

        try {

            status.setRangeOfValues(0, statisticsBands.length * 3);

            status.message("Calculating statistics");

            long cont = 0;

            for (int i = 0; i < statisticsBands.length; i++) {

                status.setCurValue(cont++);

                if (status.isCancelled()) {

                    status.abort();

                    return;

                }

                statisticsBands[i].calculate(status);

            }

            for (int i = 0; i < statisticsBands.length; i++) {

                status.setCurValue(cont++);

                if (status.isCancelled()) {

                    status.abort();

                    return;

                }

                for (int j = i; j < statisticsBands.length; j++) {

                    this.varCov[i][j] = statisticsBands[i].getCovariance(statisticsBands[j]);

                }

            }

            for (int i = 0; i < statisticsBands.length; i++) {

                status.setCurValue(cont++);

                if (status.isCancelled()) {

                    status.abort();

                    return;

                }

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

                    varCov[i][j] = varCov[j][i];

                }

            }

            calculated = true;

            if (isMyStatus) {

                status.terminate();

            } else {

                status.pop();

            }

            calculated = true;

        } catch (Exception e) {

            status.abort();

            throw e;

        }

    }

    @Override

    public double[][] getVarianceCovarianceMatrix() {

        return this.varCov;

    }

    @Override

    public HistogramBand[] getHistogram() {

        HistogramBand[] histogramBands = new HistogramBand[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            histogramBands[i]= statisticsBands[i].getHistogramBand();

        }

        return histogramBands;

    }

    @Override

    public Object getTailTrimValue(double percent) {

        Object[] result = new Object[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]=statisticsBands[i].getTailTrimValue(percent);

        }

        return result;

    }

    @Override

    public Object[] getTailTrimValue(int pos) {

        Object[] result = new Object[statisticsBands.length];

        for (int i = 0; i < statisticsBands.length; i++) {

            result[i]=statisticsBands[i].getTailTrimValue(pos);

        }

        return result;

    }

    @Override

    public Iterator<StatisticsBand> iterator() {

        return Arrays.asList(statisticsBands).iterator();

    }

}