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.legend.impl.operations.contrast;

import java.util.ArrayList;

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.spi.exceptions.ProcessingOperationException;

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

import org.gvsig.raster.lib.legend.api.RasterLegendLocator;

import org.gvsig.raster.lib.legend.api.RasterLegendManager;

import org.gvsig.raster.lib.legend.api.colorinterpretation.ColorInterpretation;

import org.gvsig.raster.lib.legend.spi.AbstractColoredOperation;

import org.gvsig.tools.locator.LocatorException;

/**

 * @author fdiaz

 *

 */

public class ContrastOperation extends AbstractColoredOperation{

    static public String CONTRAST_PARAM = "contrast";

    private int contrast;

    private RowProcessor rowProcessor;

    /**

     * @param factory

     *

     */

    public ContrastOperation(OperationFactory factory) {

        this.factory = factory;

    }

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        BufferManager manager = BufferLocator.getBufferManager();

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        contrast = (Integer) this.parameters.getDynValue(CONTRAST_PARAM);

        int bands = this.buffer.getBandCount();

        NoData[] noData;

        if (copyUnprocessedBands) {

            this.parameters.setDynValue(OUTPUT_COLOR_INTERPRETATION_PARAM, colorInterpretation);

            noData = this.buffer.getBandNoData();

            try {

                this.outputBuffer =

                    manager.createBuffer(this.buffer.getRows(), this.buffer.getColumns(), this.buffer.getBandTypes(),

                        this.buffer.getBandNoData(), this.buffer.getProjection(), this.buffer.getEnvelope());

            } catch (LocatorException | BufferException | CreateEnvelopeException e) {

                throw new ProcessingOperationException(e);

            }

        } else {

            List<String> colorInterpretations = new ArrayList<String>();

            List<NoData> noDatas = new ArrayList<NoData>();

            List<Integer> types = new ArrayList<Integer>();

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

                if (isProcessableBand(band)) {

                    colorInterpretations.add(colorInterpretation.get(band));

                    noDatas.add(this.buffer.getBandNoData()[band]);

                    types.add(this.buffer.getBandTypes()[band]);

                }

            }

            if (colorInterpretation.hasAlphaBand()) {

                colorInterpretations.add(ColorInterpretation.ALPHA_BAND);

            }

            outputColorInterpretation = legendManager.createColorInterpretation(colorInterpretations.toArray(new String[0]));

            this.parameters.setDynValue(OUTPUT_COLOR_INTERPRETATION_PARAM, outputColorInterpretation);

            int[] typesInt = new int[types.size()];

            for (Iterator iterator = types.iterator(); iterator.hasNext();) {

                int i = 0;

                Integer type = (Integer) iterator.next();

                typesInt[i] = type.intValue();

            }

            try {

                this.outputBuffer =

                    manager.createBuffer(this.buffer.getRows(), this.buffer.getColumns(), typesInt,

                        noDatas.toArray(new NoData[0]), this.buffer.getProjection(), this.buffer.getEnvelope());

            } catch (LocatorException | BufferException | CreateEnvelopeException e) {

                throw new ProcessingOperationException(e);

            }

        }

    }

    @Override

    public void process() throws ProcessingOperationException {

        super.process();

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

            rowProcessor = new ByteRowProcessor(band);

            if (isProcessableBand(band)) {

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

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

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

                    Object rowBuffer = bufferBand.createRowBuffer();

                    bufferBand.fetchRow(row, rowBuffer);

                    Object outputRowBuffer = outputBufferBand.createRowBuffer();

                    outputBufferBand.fetchRow(row, outputRowBuffer);

                    rowProcessor.processRow(rowBuffer, outputRowBuffer);

                    outputBufferBand.putRow(row, outputRowBuffer);

                }

            } else if (copyUnprocessedBands) {

                try {

                    this.outputBuffer.getBand(band).copyFrom(this.buffer.getBand(band));

                } catch (BandException e) {

                    throw new ProcessingOperationException(e);

                }

            } else if (colorInterpretation.isAlphaInterpretation(band)) {

                try {

                    this.outputBuffer.getBand(outputColorInterpretation.getAlphaBand()).copyFrom(this.buffer.getBand(band));

                } catch (BandException e) {

                    throw new ProcessingOperationException(e);

            }

        }

    }

    }

    /**

     * @param band

     * @return

     */

    private boolean isProcessableBand(int band) {

        return isRGBorGrayBand(band) && this.buffer.getBandTypes()[band]==BufferManager.TYPE_BYTE;

    }

    private boolean isRGBorGrayBand(int band) {

        String bandColorInterpretation = colorInterpretation.get(band);

        return (bandColorInterpretation.equals(ColorInterpretation.RED_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GREEN_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.BLUE_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GRAY_BAND));

    }

    @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;

        int maxResult = 255;

        int minResult = 0;

        NoData noData;

        public AbstractRowProcessor(int band) {

//            this.band = band;

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

            if(noData.isDefined()) {

                minResult = (byte)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)value;

            }

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

            //FIXME: Copiado del raster viejo. No entiendo los c?lculos

            int result;

            int diferencia = 127 - iValue;

            if (contrast >= 0) {

                result = (int) (iValue - (0.02 * diferencia * contrast)); // ((5.0 * 0.1) / 25) = 0.02

            } else {

                result = (int) (iValue - (0.004 * diferencia * contrast)); // (0.1 / 25) = 0.004;

                if (iValue < 127) {

                    if (result > 127) result = 127;

                } else {

                    if (result < 127) result = 127;

                }

            }

            if(result>maxResult){

                result = maxResult;

            }

            if(result<minResult){

                result = minResult;

            }

            return (byte)result;

        }

    }

}