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

import java.util.ArrayList;

import java.util.Collections;

import java.util.Comparator;

import java.util.Iterator;

import java.util.List;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

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

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;

import org.gvsig.tools.locator.LocatorException;

/**

 * @author fdiaz

 *

 */

public class MedianOperation extends AbstractSpecifiedBandsOperation{

    static public String SIDE_WINDOW_PARAM = "side_window";

    private Statistics statistics;

    private int sideWindow;

    private RowProcessor[] rowProcessors;

    private int halfSideWindow;

    /**

     * @param factory

     *

     */

    public MedianOperation(OperationFactory factory) {

        super(factory);

    }

    @SuppressWarnings("unchecked")

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        BufferManager manager = BufferLocator.getBufferManager();

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

        halfSideWindow = (int)(sideWindow/2);

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

        NoData[] noData = this.getInputBuffer().getBandNoData();

        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 (LocatorException | BufferException | CreateEnvelopeException 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 (LocatorException | BufferException | CreateEnvelopeException e) {

                throw new ProcessingOperationException(e);

            }

        }

        rowProcessors = new RowProcessor[bands];

        for (int band = 0; band < noData.length; 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<>();

                    for(int r=Math.max(row-halfSideWindow,0); r<=Math.min(row+halfSideWindow,this.getInputBuffer().getRows()-1);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;

            }

            Collections.sort(kernel,new Comparator<Number>() {

                @Override

                public int compare(Number o1, Number o2) {

                    return o1.intValue()-o2.intValue();

                }

            });

            int n = kernel.size();

            int temp = n / 2;

            int pos;

            double temp0;

            if ( n % 2 == 0 ) {

                pos = (int)temp;

                temp0 = (double)(kernel.get(pos).intValue() + kernel.get(pos + 1).intValue())/2;

            } else { // if ( n % 2 == 1 )

                pos = (int)(temp + 0.5);

                temp0 = (double)(kernel.get(pos).intValue());

            }

          return (byte)(new Double(temp0).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;

            }

            Collections.sort(kernel,new Comparator<Number>() {

                @Override

                public int compare(Number o1, Number o2) {

                    return ((Short)o1).compareTo((Short)o2);

                }

            });

            int n = kernel.size();

            int temp = n / 2;

            int pos;

            double temp0;

            if ( n % 2 == 0 ) {

                pos = (int)temp;

                temp0 = (double)(kernel.get(pos).intValue() + kernel.get(pos + 1).intValue())/2;

            } else { // if ( n % 2 == 1 )

                pos = (int)(temp + 0.5);

                temp0 = (double)(kernel.get(pos).intValue());

            }

          return (short)(new Double(temp0).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;

            }

            Collections.sort(kernel,new Comparator<Number>() {

                @Override

                public int compare(Number o1, Number o2) {

                    return o1.intValue()-o2.intValue();

                }

            });

            int n = kernel.size();

            int temp = n / 2;

            int pos;

            double temp0;

            if ( n % 2 == 0 ) {

                pos = (int)temp;

                temp0 = (double)(kernel.get(pos).intValue() + kernel.get(pos + 1).intValue())/2;

            } else { // if ( n % 2 == 1 )

                pos = (int)(temp + 0.5);

                temp0 = (double)(kernel.get(pos).intValue());

            }

          return (short)(new Double(temp0).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;

            }

            Collections.sort(kernel,new Comparator<Number>() {

                @Override

                public int compare(Number o1, Number o2) {

                    return ((Integer)o1).compareTo((Integer)o2);

                }

            });

            int n = kernel.size();

            int temp = n / 2;

            int pos;

            double temp0;

            if ( n % 2 == 0 ) {

                pos = (int)temp;

                temp0 = (double)(kernel.get(pos).intValue() + kernel.get(pos + 1).intValue())/2;

            } else { // if ( n % 2 == 1 )

                pos = (int)(temp + 0.5);

                temp0 = (double)(kernel.get(pos).intValue());

            }

          return new Double(temp0).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;

            }

            Collections.sort(kernel,new Comparator<Number>() {

                @Override

                public int compare(Number o1, Number o2) {

                    return ((Double)o1).compareTo((Double)o2);

                }

            });

            int n = kernel.size();

            int temp = n / 2;

            int pos;

            double temp0;

            if ( n % 2 == 0 ) {

                pos = (int)temp;

                temp0 = (double)(kernel.get(pos).doubleValue() + kernel.get(pos + 1).doubleValue())/2;

            } else { // if ( n % 2 == 1 )

                pos = (int)(temp + 0.5);

                temp0 = (double)(kernel.get(pos).doubleValue());

            }

          return new Double(temp0).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;

            }

            Collections.sort(kernel,new Comparator<Number>() {

                @Override

                public int compare(Number o1, Number o2) {

                    return ((Double)o1).compareTo((Double)o2);

                }

            });

            int n = kernel.size();

            int temp = n / 2;

            int pos;

            double temp0;

            if ( n % 2 == 0 ) {

                pos = (int)temp;

                temp0 = (double)(kernel.get(pos).doubleValue() + kernel.get(pos + 1).doubleValue())/2;

            } else { // if ( n % 2 == 1 )

                pos = (int)(temp + 0.5);

                temp0 = (double)(kernel.get(pos).doubleValue());

            }

          return temp0;

        }

    }

}