Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.lib.buffer.impl.operations.mode;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

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

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

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

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

import org.gvsig.raster.lib.buffer.api.exceptions.BandException;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferOperationException;

import org.gvsig.raster.lib.buffer.api.operations.OperationFactory;

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

import org.gvsig.raster.lib.buffer.spi.exceptions.ProcessingOperationException;

import org.gvsig.raster.lib.buffer.spi.operations.AbstractSpecifiedBandsOperation;

/**

 * @author fdiaz

 *

 */

public class ModeOperation extends AbstractSpecifiedBandsOperation{

//    static public String STATISTICS_PARAM = "statistics";

    static public String SIDE_WINDOW_PARAM = "side_window";

    private Statistics statistics;

    private int sideWindow;

    private RowProcessor[] rowProcessors;

    private int halfSideWindow;

    /**

     * @param factory

     *

     */

    public ModeOperation(OperationFactory factory) {

        super(factory);

    }

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        BufferManager manager = BufferLocator.getBufferManager();

        sideWindow = (int) this.getParameter(SIDE_WINDOW_PARAM, 3);

        halfSideWindow = (int)(sideWindow/2);

        if (mustCopyUnprocessedBands()) {

            try {

                this.setOutputBuffer(

                    manager.createBuffer(

                            this.getInputBuffer().getRows(), 

                            this.getInputBuffer().getColumns(), 

                            this.getInputBuffer().getBandTypes(),

                            this.getInputBuffer().getBandNoData(), 

                            this.getInputBuffer().getProjection(), 

                            this.getInputBuffer().getEnvelope()));

            } catch (Exception e) {

                throw new ProcessingOperationException(e);

            }

        } else {

            try {

                this.setOutputBuffer(

                    manager.createBuffer(

                            this.getInputBuffer().getRows(), 

                            this.getInputBuffer().getColumns(), 

                            this.getProcessableBandTypesAsArray(),

                            this.getProcessableBandNoDatasAsArray(), 

                            this.getInputBuffer().getProjection(),

                            this.getInputBuffer().getEnvelope()));

            } catch (Exception e) {

                throw new ProcessingOperationException(e);

            }

        }

        int bands = this.getInputBuffer().getBandCount();

        rowProcessors = new RowProcessor[bands];

        for (int band = 0; band < bands; band++) {

            if (isProcessableBand(band)) {

            int bandType = this.getInputBuffer().getBand(band).getDataType();

            switch (bandType) {

            case BufferManager.TYPE_BYTE:

                rowProcessors[band] = new ByteRowProcessor(band);

                break;

            case BufferManager.TYPE_USHORT:

                rowProcessors[band] = new UShortRowProcessor(band);

                break;

            case BufferManager.TYPE_SHORT:

                rowProcessors[band] = new ShortRowProcessor(band);

                break;

            case BufferManager.TYPE_INT:

                rowProcessors[band] = new IntRowProcessor(band);

                break;

            case BufferManager.TYPE_FLOAT:

                rowProcessors[band] = new FloatRowProcessor(band);

                break;

            case BufferManager.TYPE_DOUBLE:

                rowProcessors[band] = new DoubleRowProcessor(band);

                break;

            default:

                throw new IllegalArgumentException("Unknow type of band '"+band+"'");

            }

            }

        }

    }

    @Override

    public void process() throws ProcessingOperationException {

        super.process();

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

            if (getBandsToProcess().contains(band)) {

                Band bufferBand = this.getInputBuffer().getBand(band);

                Band outputBufferBand = this.getOutputBuffer().getBand(band);

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

                    Object rowBuffer = bufferBand.createRowBuffer();

                    bufferBand.fetchRow(row, rowBuffer);

                    List<Object> bundle = new ArrayList<>();

                    int maxr = Math.min(row+halfSideWindow,this.getInputBuffer().getRows()-1);

                    for(int r=Math.max(row-halfSideWindow,0); r<=maxr;r++){

                        Object bundleRow = bufferBand.createRowBuffer();

                        bufferBand.fetchRow(row, bundleRow);

                        bundle.add(bundleRow);

                    }

                    Object outputRowBuffer = outputBufferBand.createRowBuffer();

//                    outputBufferBand.fetchRow(row, outputRowBuffer);

                    rowProcessors[band].processRow(rowBuffer, bundle, outputRowBuffer);

                    outputBufferBand.putRow(row, outputRowBuffer);

                }

            } else {

                if(mustCopyUnprocessedBands()){

                try {

                    this.getOutputBuffer().getBand(band).copyFrom(this.getInputBuffer().getBand(band));

                } catch (BandException e) {

                    throw new ProcessingOperationException(e);

                }

            }

        }

    }

    }

    @Override

    public void postProcess()  throws BufferOperationException {

        super.postProcess();

    }

    interface RowProcessor {

        void processRow(Object inputRow, List<Object> bundleRow, Object outputRow);

        Number processValue(Number value, List<Number> kernel);

    };

    private abstract class AbstractRowProcessor implements RowProcessor {

        int band;

        NoData noData;

        public AbstractRowProcessor(int band) {

            this.band = band;

            noData = getInputBuffer().getBand(band).getNoData();

            }

            }

    private class ByteRowProcessor extends AbstractRowProcessor {

        public ByteRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, List<Object> bundleRow, Object outputRow) {

            byte[] inputByteRow = (byte[])inputRow;

            byte[] outputByteRow = (byte[])outputRow;

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

                List<Number> kernel = new ArrayList<>();

                for (Iterator<Object> iterator = bundleRow.iterator(); iterator.hasNext();) {

                    byte[] row = (byte[]) iterator.next();

                    for(int c=Math.max(i-halfSideWindow,0); c<=Math.min(i+halfSideWindow,inputByteRow.length-1);c++){

                        Integer value = 0xFF & ((Byte) row[c]);

                        if(noData.isDefined() && noData.getValue().equals(value)){

                            continue;

                        } else {

                            kernel.add(value);

                        }

            }

        }

                outputByteRow[i] = processValue(inputByteRow[i], kernel).byteValue();

            }

        }

        @Override

        public Number processValue(Number value, List<Number> kernel) {

            if (noData.isDefined() && noData.getValue().equals(value)) {

                return (byte) value;

            }

            return (byte) (mode(value, kernel).intValue() & 0xFF);

        }

    }

    private class ShortRowProcessor extends AbstractRowProcessor {

        public ShortRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, List<Object> bundleRow, Object outputRow) {

            Number[] inputShortRow = (Number[])inputRow;

            short[] outputShortRow = (short[])outputRow;

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

                List<Number> kernel = new ArrayList<Number>();

                for (Iterator<Object> iterator = bundleRow.iterator(); iterator.hasNext();) {

                    short[] row = (short[]) iterator.next();

                    for(int c=Math.max(i-halfSideWindow,0); c<=Math.min(i+halfSideWindow,inputShortRow.length-1);c++){

                        Short value = ((Short) row[c]).shortValue();

                        if(noData.isDefined() && noData.getValue().equals(value)){

                            continue;

                        } else {

                            kernel.add(value);

                        }

                        kernel.add(value);

                    }

                }

                outputShortRow[i] = processValue(inputShortRow[i], kernel).shortValue();

            }

        }

        @Override

        public Number processValue(Number value, List<Number> kernel) {

            if(noData.isDefined() && noData.getValue().equals(value)){

                return (short)value;

            }

          return (short)(mode(value, kernel).intValue());

        }

    }

    private class UShortRowProcessor extends AbstractRowProcessor {

        public UShortRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, List<Object> bundleRow, Object outputRow) {

            Number[] inputShortRow = (Number[])inputRow;

            short[] outputShortRow = (short[])outputRow;

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

                List<Number> kernel = new ArrayList<Number>();

                for (Iterator<Object> iterator = bundleRow.iterator(); iterator.hasNext();) {

                    short[] row = (short[]) iterator.next();

                    for(int c=Math.max(i-halfSideWindow,0); c<=Math.min(i+halfSideWindow,inputShortRow.length-1);c++){

                        Integer value = 0xFFFF & ((Short) row[c]).shortValue();

                        if(noData.isDefined() && noData.getValue().equals(value)){

                            continue;

                        } else {

                            kernel.add(value);

            }

                        kernel.add(value);

        }

                }

                outputShortRow[i] = processValue(inputShortRow[i], kernel).shortValue();

            }

        }

        @Override

        public Number processValue(Number value, List<Number> kernel) {

            if(noData.isDefined() && noData.getValue().equals(value)){

                return (short)value;

            }

          return (short)(mode(value, kernel).intValue() & 0xFFFF);

        }

    }

    private class IntRowProcessor extends AbstractRowProcessor {

        public IntRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, List<Object> bundleRow, Object outputRow) {

            Number[] inputIntRow = (Number[])inputRow;

            int[] outputIntRow = (int[])outputRow;

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

                List<Number> kernel = new ArrayList<Number>();

                for (Iterator<Object> iterator = bundleRow.iterator(); iterator.hasNext();) {

                    int[] row = (int[]) iterator.next();

                    for(int c=Math.max(i-halfSideWindow,0); c<=Math.min(i+halfSideWindow,inputIntRow.length-1);c++){

                        Integer value = ((Integer) row[c]).intValue();

                        if(noData.isDefined() && noData.getValue().equals(value)){

                            continue;

                        } else {

                            kernel.add(value);

                    }

                        kernel.add(value);

                    }

                }

                outputIntRow[i] = processValue(inputIntRow[i], kernel).intValue();

            }

        }

        @Override

        public Number processValue(Number value, List<Number> kernel) {

            if(noData.isDefined() && noData.getValue().equals(value)){

                return (int)value;

            }

          return mode(value, kernel).intValue();

        }

    }

    private class FloatRowProcessor extends AbstractRowProcessor {

        public FloatRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, List<Object> bundleRow, Object outputRow) {

            Number[] inputFloatRow = (Number[])inputRow;

            float[] outputFloatRow = (float[])outputRow;

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

                List<Number> kernel = new ArrayList<Number>();

                for (Iterator<Object> iterator = bundleRow.iterator(); iterator.hasNext();) {

                    float[] row = (float[]) iterator.next();

                    for(int c=Math.max(i-halfSideWindow,0); c<=Math.min(i+halfSideWindow,inputFloatRow.length-1);c++){

                        Float value = ((Float) row[c]).floatValue();

                        if(noData.isDefined() && noData.getValue().equals(value)){

                            continue;

                        } else {

                            kernel.add(value);

                    }

                        kernel.add(value);

                }

                }

                outputFloatRow[i] = processValue(inputFloatRow[i], kernel).floatValue();

            }

        }

        @Override

        public Number processValue(Number value, List<Number> kernel) {

            if (noData.isDefined() && noData.getValue().equals(value)) {

                return (float) value;

            }

            return mode(value, kernel).floatValue();

        }

    }

    private class DoubleRowProcessor extends AbstractRowProcessor {

        public DoubleRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, List<Object> bundleRow, Object outputRow) {

            Number[] inputDoubleRow = (Number[])inputRow;

            double[] outputDoubleRow = (double[])outputRow;

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

                List<Number> kernel = new ArrayList<Number>();

                for (Iterator<Object> iterator = bundleRow.iterator(); iterator.hasNext();) {

                    double[] row = (double[]) iterator.next();

                    for(int c=Math.max(i-halfSideWindow,0); c<=Math.min(i+halfSideWindow,inputDoubleRow.length-1);c++){

                        Double value = ((Double) row[c]).doubleValue();

                        if(noData.isDefined() && noData.getValue().equals(value)){

                            continue;

                        } else {

                            kernel.add(value);

                        }

                        kernel.add(value);

                    }

                }

                outputDoubleRow[i] = processValue(inputDoubleRow[i], kernel).floatValue();

            }

        }

        @Override

        public Number processValue(Number value, List<Number> kernel) {

            if (noData.isDefined() && noData.getValue().equals(value)) {

                return (double) value;

            }

            return mode(value, kernel).doubleValue();

        }

    }

    private Number mode(Number value, List<Number> kernel) {

        Number maxValue = null;

        int maxCount = 0;

        Number[] kernelArray = kernel.toArray(new Number[0]);

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

            int count = 0;

            for (int j = 0; j < kernelArray.length; ++j) {

                if (kernelArray[j] == kernelArray[i]) ++count;

            }

            if (count > maxCount) {

                maxCount = count;

                maxValue = kernelArray[i];

            }

        }

        return maxValue;

    }

}