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

import java.util.ArrayList;

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.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 abstract class AbstractColorBalanceOperation  extends AbstractColoredOperation {

    /**

     * @param factory

     *

     */

    public AbstractColorBalanceOperation(OperationFactory factory) {

        super(factory);

    }

    abstract protected void initialiceParams();

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        BufferManager manager = BufferLocator.getBufferManager();

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        initialiceParams();

        try {

            if (!(getInputColorInterpretation().isGray() || getInputColorInterpretation().isRGB())) {

                throw new UnsupportedOperationException(

                    "The color interpretation of input buffer isn't RGB nor GRAYSCALE");

            }

            int[] inputBandTypes = getInputBuffer().getBandTypes();

            if (getInputColorInterpretation().isGray()) {

                if (inputBandTypes[getInputColorInterpretation().getBand(ColorInterpretation.GRAY_BAND)] != BufferManager.TYPE_BYTE) {

                    throw new UnsupportedOperationException("The type of GRAY band isn't BYTE");

                }

            } else if (getInputColorInterpretation().isRGB()) {

                if (inputBandTypes[getInputColorInterpretation().getBand(ColorInterpretation.RED_BAND)] != BufferManager.TYPE_BYTE

                    || inputBandTypes[getInputColorInterpretation().getBand(ColorInterpretation.GREEN_BAND)] != BufferManager.TYPE_BYTE

                    || inputBandTypes[getInputColorInterpretation().getBand(ColorInterpretation.BLUE_BAND)] != BufferManager.TYPE_BYTE) {

                    throw new UnsupportedOperationException("The type of RGB bands isn't BYTE");

                }

            } else {

                throw new UnsupportedOperationException(

                    "The color interpretation of input buffer isn't RGB nor GRAYSCALE");

            }

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

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

            int[] sourceTypes = this.getInputBuffer().getBandTypes();

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

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

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

            colorInterpretations.add(ColorInterpretation.RED_BAND);

            colorInterpretations.add(ColorInterpretation.GREEN_BAND);

            colorInterpretations.add(ColorInterpretation.BLUE_BAND);

            types.add(BufferManager.TYPE_BYTE);

            types.add(BufferManager.TYPE_BYTE);

            types.add(BufferManager.TYPE_BYTE);

            noDatas.add(null);

            noDatas.add(null);

            noDatas.add(null);

            if (getInputColorInterpretation().hasAlphaBand()) {

                int alphaBand = getInputColorInterpretation().getAlphaBand();

                if(sourceTypes[alphaBand]!=BufferManager.TYPE_BYTE){

                    throw new UnsupportedOperationException("The type of ALPHA band isn't BYTE");

                }

                colorInterpretations.add(ColorInterpretation.ALPHA_BAND);

                types.add(sourceTypes[alphaBand]);

                noDatas.add(sourceNoDatas[alphaBand]);

            }

            if (mustCopyUnprocessedBands()) {

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

                    if (!isProcessableBand(band) && !getInputColorInterpretation().isAlphaInterpretation(band)) {

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

                        noDatas.add(sourceNoDatas[band]);

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

                    }

                }

            }

            setOutputColorInterpretation(

                legendManager.createColorInterpretation(colorInterpretations));

            this.setParameter(OUTPUT_COLOR_INTERPRETATION_PARAM, getOutputColorInterpretation());

            this.setOutputBuffer(

                manager.createBuffer(

                        this.getInputBuffer().getRows(), 

                        this.getInputBuffer().getColumns(), 

                        this.getTypesAsArray(types),

                        this.getNoDatasAsArray(noDatas), 

                        this.getInputBuffer().getProjection(), 

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

        } catch (LocatorException | BufferException | CreateEnvelopeException e) {

            throw new BufferOperationException(e);

        }

    }

    @Override

    public void process() throws ProcessingOperationException {

        super.process();

        if (getInputColorInterpretation().hasAlphaBand()) {

            try {

                getOutputBuffer().getBand(3).copyFrom(this.getInputBuffer().getBand(getInputColorInterpretation().getAlphaBand()));

            } catch (BandException e) {

                throw new ProcessingOperationException(e);

            }

        }

        Object[] rowBandsBuffer = new Object[3];

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

            if(getInputColorInterpretation().isGray()){

                Band bufferBandGray = getInputBuffer().getBand(getInputColorInterpretation().getBand(ColorInterpretation.GRAY_BAND));

                rowBandsBuffer[0] = bufferBandGray.createRowBuffer();

                bufferBandGray.fetchRow(row, rowBandsBuffer[0]);

                rowBandsBuffer[1] = bufferBandGray.createRowBuffer();

                bufferBandGray.fetchRow(row, rowBandsBuffer[0]);

                rowBandsBuffer[2] = bufferBandGray.createRowBuffer();

                bufferBandGray.fetchRow(row, rowBandsBuffer[0]);

            } else if (getInputColorInterpretation().isRGB()){

                Band bufferBandRed = getInputBuffer().getBand(getInputColorInterpretation().getBand(ColorInterpretation.RED_BAND));

                rowBandsBuffer[0] = bufferBandRed.createRowBuffer();

                bufferBandRed.fetchRow(row, rowBandsBuffer[0]);

                Band bufferBandGreen = getInputBuffer().getBand(getInputColorInterpretation().getBand(ColorInterpretation.GREEN_BAND));

                rowBandsBuffer[1] = bufferBandGreen.createRowBuffer();

                bufferBandGreen.fetchRow(row, rowBandsBuffer[1]);

                Band bufferBandBlue = getInputBuffer().getBand(getInputColorInterpretation().getBand(ColorInterpretation.BLUE_BAND));

                rowBandsBuffer[2] = bufferBandBlue.createRowBuffer();

                bufferBandBlue.fetchRow(row, rowBandsBuffer[2]);

            } else { //No deber?a pasar por aqu?

                throw new UnsupportedOperationException("The color interpretation of input buffer isn't RGB nor GRAYSCALE");

            }

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

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

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

                Object outputRowBuffer = outputBufferBand.createRowBuffer();

                outputRowBuffers.add(outputRowBuffer);

            }

            processRow(rowBandsBuffer, outputRowBuffers);

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

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

                outputBufferBand.putRow(row, outputRowBuffers.get(band));

            }

        }

    }

    abstract protected void processRow(Object[] inputRows, List outputRows);

    @Override

    public void postProcess() throws BufferOperationException {

        super.postProcess();

    }

    /**

     * @param band

     * @return

     */

    @Override

    protected boolean isProcessableBand(int band) {

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

    }

    protected boolean isRGBorGrayBand(int band) {

        String bandColorInterpretation = getInputColorInterpretation().get(band);

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

            bandColorInterpretation.equals(ColorInterpretation.GREEN_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.BLUE_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GRAY_BAND));

    }

}