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

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.impl.DefaultNoData;

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 LinearStretchEnhancementOperation extends AbstractSpecifiedBandsOperation{

    static public String STATISTICS_PARAM = "statistics";

    static public String REMOVE_ENDS_PARAM = "remove_ends";

    static public String TAIL_TRIM_PARAM = "tail_trim";

    static public String TAIL_TRIM_PERCENT_PARAM = "tail_trim_percent";

    private Statistics statistics;

    private boolean removeEnds;

    private boolean tailTrim;

    private double tailTrimPercent;

    private RowProcessor[] rowProcessors;

    /**

     * @param factory

     *

     */

    public LinearStretchEnhancementOperation(OperationFactory factory) {

        super(factory);

    }

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        BufferManager manager = BufferLocator.getBufferManager();

        statistics = (Statistics) this.getParameter(STATISTICS_PARAM, null);

        if (statistics == null) {

            statistics = this.getInputBuffer().getStatistics(null);

        }

        removeEnds = (boolean) this.getParameter(REMOVE_ENDS_PARAM, false);

        tailTrim = (boolean) this.getParameter(TAIL_TRIM_PARAM, false);

        tailTrimPercent = (double) this.getParameter(TAIL_TRIM_PERCENT_PARAM, 0);

        tailTrimPercent = (tailTrimPercent>100)?100:tailTrimPercent;

        tailTrimPercent = (tailTrimPercent<0)?0:tailTrimPercent;

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

        rowProcessors = new RowProcessor[bands];

        int [] bandTypes = new int[bands];

        //FIXME: Falta la gesti?n del par?metro copyUnprocessedBands, de momento se copian sin tenerlo en cuenta

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

            if(this.getBandsToProcess().contains(i)){

                bandTypes[i] = BufferManager.TYPE_BYTE;

            } else {

                bandTypes[i] = this.getInputBuffer().getBandTypes()[i];

            }

        }

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

        NoData[] resultNoData = new NoData[noData.length];

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

            }

            if(noData[band].isDefined()){

                resultNoData[band] = new DefaultNoData((byte)0);

                resultNoData[band] = manager.createNoData((byte)0,(byte)0);

            } else {

                resultNoData[band]=manager.createNoData(null, null);

            }

        }

        try {

            this.setOutputBuffer( manager.createBuffer(

                this.getInputBuffer().getRows(),

                this.getInputBuffer().getColumns(),

                bandTypes,

                resultNoData,

                this.getInputBuffer().getProjection(),

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

        } catch (LocatorException | BufferException | CreateEnvelopeException e) {

            throw new ProcessingOperationException(e);

        }

    }

    @Override

    public void process() throws ProcessingOperationException {

        super.process();

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

        int rowCount = this.getInputBuffer().getRows();

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

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

                this.getTaskStatus().setRangeOfValues(0, rowCount);

                this.getTaskStatus().setCurValue(0);

                this.getTaskStatus().message("LinearStretch band "+band+"/"+bandCount);

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

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

                for (int row = 0; row < rowCount; row++) {

                    this.getTaskStatus().setCurValue(row);

                    Object rowBuffer = bufferBand.createRowBuffer();

                    bufferBand.fetchRow(row, rowBuffer);

                    Object outputRowBuffer = outputBufferBand.createRowBuffer();

                    outputBufferBand.fetchRow(row, outputRowBuffer);

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

                    outputBufferBand.putRow(row, outputRowBuffer);

                }

            } else {

                try {

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

                } catch (BandException e) {

                    throw new ProcessingOperationException(e);

                }

            }

        }

        this.getTaskStatus().terminate();

    }

    @Override

    public void postProcess()  throws BufferOperationException {

        super.postProcess();

    }

    interface RowProcessor {

        void processRow(Object inputRow, Object outputRow);

        byte processValue(Object value);

    };

    private abstract class AbstractRowProcessor implements RowProcessor {

        int band;

        double minValue;

        double maxValue;

        double maxResult = 255;

        double minResult = 0;

        NoData noData;

        public AbstractRowProcessor(int band) {

            this.band = band;

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

            if(noData.isDefined()) {

                minResult = (byte)1;

            }

            minValue = statistics.getMin()[band];

            maxValue = statistics.getMax()[band];

            if(removeEnds) {

                minValue = statistics.getSecondMin()[band];

                maxValue = statistics.getSecondMax()[band];

            }

            if(tailTrim) {

                double[][] tailTrim = statistics.getTailTrimValue(tailTrimPercent);

                minValue = tailTrim[band][0];

                maxValue = tailTrim[band][1];

            }

        }

    }

    private class ByteRowProcessor extends AbstractRowProcessor {

        public ByteRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

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

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

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

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

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

                return (byte)0;

            }

            int iValue = 0xFF & ((Byte) value);

            Double dValue = new Double(iValue);

            double result;

            if(dValue < minValue){

                result = minResult;

            } else if (dValue > maxValue){

                result = maxResult;

            } else {

                double ratio = (maxResult - minResult) / (maxValue - minValue);

                result = (dValue-minValue) * ratio + minResult;

            }

            return (byte)result;

        }

    }

    private class ShortRowProcessor extends AbstractRowProcessor {

        public ShortRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

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

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

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

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

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

                return (byte)0;

            }

            int iValue = ((Short) value);

            Double dValue = ((Number) iValue).doubleValue();

            double result;

            if(dValue < minValue){

                result = minResult;

            } else if (dValue > maxValue){

                result = maxResult;

            } else {

                double ratio = (maxResult - minResult) / (maxValue - minValue);

                result = (dValue-minValue) * ratio + minResult;

            }

            return (byte)result;

        }

    }

    private class UShortRowProcessor extends AbstractRowProcessor {

        public UShortRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

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

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

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

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

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

                return (byte)0;

            }

            //FIXME ???:

            int iValue = 0xFFFF & ((Short) value);

            Double dValue = ((Number) iValue).doubleValue();

            double result;

            if(dValue < minValue){

                result = minResult;

            } else if (dValue > maxValue){

                result = maxResult;

            } else {

                double ratio = (maxResult - minResult) / (maxValue - minValue);

                result = (dValue-minValue) * ratio + minResult;

            }

            return (byte)result;

        }

    }

    private class IntRowProcessor extends AbstractRowProcessor {

        public IntRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

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

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

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

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

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

                return (byte)0;

            }

            Double dValue = ((Number) value).doubleValue();

            double result;

            if(dValue < minValue){

                result = minResult;

            } else if (dValue > maxValue){

                result = maxResult;

            } else {

                double ratio = (maxResult - minResult) / (maxValue - minValue);

                result = (dValue-minValue) * ratio + minResult;

            }

            return (byte)result;

        }

    }

    private class FloatRowProcessor extends AbstractRowProcessor {

        public FloatRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

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

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

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

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

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

                return (byte)0;

            }

            Double dValue = ((Number) value).doubleValue();

            double result;

            if(dValue < minValue){

                result = minResult;

            } else if (dValue > maxValue){

                result = maxResult;

            } else {

                double ratio = (maxResult - minResult) / (maxValue - minValue);

                result = (dValue-minValue) * ratio + minResult;

            }

            return (byte)result;

        }

    }

    private class DoubleRowProcessor extends AbstractRowProcessor {

        public DoubleRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

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

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

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

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

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

                return (byte)0;

            }

            Double dValue = ((Number) value).doubleValue();

            double result;

            if(dValue < minValue){

                result = minResult;

            } else if (dValue > maxValue){

                result = maxResult;

            } else {

                double ratio = (maxResult - minResult) / (maxValue - minValue);

                result = (dValue-minValue) * ratio + minResult;

            }

            return (byte)result;

        }

    }

}