Application: gvSIG desktop » gvSIG raster

/* 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.colortable;

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

import org.gvsig.raster.lib.legend.impl.RasterLegendManagerServices;

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

import org.gvsig.tools.locator.LocatorException;

/**

 * @author fdiaz

 *

 */

public class ColorTableOperation extends AbstractColoredOperation {

    private RowProcessor rowProcessor;

    /**

     * @param factory

     *

     */

    public ColorTableOperation(OperationFactory factory) {

        this.factory = factory;

    }

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        BufferManager manager = BufferLocator.getBufferManager();

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        try {

            if (!colorInterpretation.isPalette()) {

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

            }

            int bandType = this.buffer.getBand(colorInterpretation.getPaletteBand()).getDataType();

            switch (bandType) {

            case BufferManager.TYPE_BYTE:

                rowProcessor = new ByteRowProcessor();

                break;

            case BufferManager.TYPE_USHORT:

                rowProcessor = new UShortRowProcessor();

                break;

            case BufferManager.TYPE_SHORT:

                rowProcessor = new ShortRowProcessor();

                break;

            case BufferManager.TYPE_INT:

                rowProcessor = new IntRowProcessor();

                break;

            case BufferManager.TYPE_FLOAT:

                rowProcessor = new FloatRowProcessor();

                break;

            case BufferManager.TYPE_DOUBLE:

                rowProcessor = new DoubleRowProcessor();

                break;

            default:

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

            }

            int sourceBands = this.buffer.getBandCount();

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

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

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

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

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

            colorInterpretations.add(ColorInterpretation.RED_BAND);

            colorInterpretations.add(ColorInterpretation.GREEN_BAND);

            colorInterpretations.add(ColorInterpretation.BLUE_BAND);

            colorInterpretations.add(ColorInterpretation.ALPHA_BAND);

            types.add(BufferManager.TYPE_BYTE);

            types.add(BufferManager.TYPE_BYTE);

            types.add(BufferManager.TYPE_BYTE);

            types.add(BufferManager.TYPE_BYTE);

            noDatas.add(null);

            noDatas.add(null);

            noDatas.add(null);

            noDatas.add(null);

//            if (colorInterpretation.hasAlphaBand()) {

//                int alphaBand = colorInterpretation.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 (copyUnprocessedBands) {

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

                    if (!isProcessableBand(band)) {

                        colorInterpretations.add(ColorInterpretation.UNDEFINED_BAND);

                        noDatas.add(sourceNoDatas[band]);

                        types.add(this.buffer.getBandTypes()[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<Integer> iterator = types.iterator(); iterator.hasNext();) {

                int i = 0;

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

                typesInt[i] = type.intValue();

            }

            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();

        try {

            // List<Integer> bandsToProcess = new ArrayList<Integer>();

            if (copyUnprocessedBands) {

                int bands = this.buffer.getBandCount();

                int outBand = 4; //colorInterpretation.hasAlphaBand() ? 4 : 3;

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

                    if (!isProcessableBand(band)) {

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

                        outBand++;

                    }

                }

            }

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

                Band bufferPaletteBand = buffer.getBand(colorInterpretation.getPaletteBand());

                Object rowBandBuffer = bufferPaletteBand.createRowBuffer();

                bufferPaletteBand.fetchRow(row, rowBandBuffer);

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

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

                    Band outputBufferBand = outputBuffer.getBand(band);

                    Object outputRowBuffer = outputBufferBand.createRowBuffer();

                    outputRowBuffers.add(outputRowBuffer);

                }

                rowProcessor.processRow(rowBandBuffer, outputRowBuffers);

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

                    Band outputBufferBand = outputBuffer.getBand(band);

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

                }

            }

        } catch (Exception e) {

            throw new ProcessingOperationException(e);

        }

    }

    @Override

    public void postProcess() throws BufferOperationException {

        super.postProcess();

    }

    /**

     * @param band

     * @return

     */

    private boolean isProcessableBand(int band) {

        return isPaletteBand(band);

    }

    private boolean isPaletteBand(int band) {

        return (colorInterpretation.getPaletteBand() == band);

    }

    interface RowProcessor {

        void processRow(Object inputRow, List outputRows);

        byte[] processValue(Object value);

    };

    private abstract class AbstractRowProcessor implements RowProcessor {

        NoData noData;

        public AbstractRowProcessor() {

            noData = buffer.getBand(colorInterpretation.getPaletteBand()).getNoData();

        }

    }

    private class ByteRowProcessor extends AbstractRowProcessor {

        public ByteRowProcessor() {

            super();

        }

        @Override

        public void processRow(Object inputRow, List outputRows) {

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

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

                byte[] rgb = processValue(inputByteRow[i]);

                for (int band = 0; band < outputRows.size(); band++) {

                    ((byte[])(outputRows.get(band)))[i] = rgb[band];

                }

            }

        }

        @Override

        public byte[] processValue(Object value) {

            if (noData.isDefined()) {

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

                    byte result[] = { 0, 0, 0, 0 };

                    return result;

                }

            }

            return colorInterpretation.getPalette().getRGBA(0xFF & (byte)value);

        }

    }

    private class ShortRowProcessor extends AbstractRowProcessor {

        public ShortRowProcessor() {

            super();

        }

        @Override

        public void processRow(Object inputRow, List outputRows) {

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

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

                byte[] rgb = processValue(inputByteRow[i]);

                for (int band = 0; band < outputRows.size(); band++) {

                    ((byte[])(outputRows.get(band)))[i] = rgb[band];

                }

            }

        }

        @Override

        public byte[] processValue(Object value) {

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

                byte result[] = {0,0,0,0};

                return result;

            }

            return colorInterpretation.getPalette().getRGBA(value);

        }

    }

    private class UShortRowProcessor extends AbstractRowProcessor {

        public UShortRowProcessor() {

            super();

        }

        @Override

        public void processRow(Object inputRow, List outputRows) {

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

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

                byte[] rgb = processValue(inputByteRow[i]);

                for (int band = 0; band < outputRows.size(); band++) {

                    ((byte[])(outputRows.get(band)))[i] = rgb[band];

                }

            }

        }

        @Override

        public byte[] processValue(Object value) {

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

                byte result[] = {0,0,0,0};

                return result;

            }

            return colorInterpretation.getPalette().getRGBA(0xFFFF & (byte)value);

        }

    }

    private class IntRowProcessor extends AbstractRowProcessor {

        public IntRowProcessor() {

            super();

        }

        @Override

        public void processRow(Object inputRow, List outputRows) {

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

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

                byte[] rgb = processValue(inputByteRow[i]);

                for (int band = 0; band < outputRows.size(); band++) {

                    ((byte[])(outputRows.get(band)))[i] = rgb[band];

                }

            }

        }

        @Override

        public byte[] processValue(Object value) {

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

                byte result[] = {0,0,0,0};

                return result;

            }

            return colorInterpretation.getPalette().getRGBA(value);

        }

    }

    private class FloatRowProcessor extends AbstractRowProcessor {

        public FloatRowProcessor() {

            super();

        }

        @Override

        public void processRow(Object inputRow, List outputRows) {

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

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

                byte[] rgb = processValue(inputByteRow[i]);

                for (int band = 0; band < outputRows.size(); band++) {

                    ((byte[])(outputRows.get(band)))[i] = rgb[band];

                }

            }

        }

        @Override

        public byte[] processValue(Object value) {

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

                byte result[] = {0,0,0,0};

                return result;

            }

            return colorInterpretation.getPalette().getRGBA(value);

        }

    }

    private class DoubleRowProcessor extends AbstractRowProcessor {

        public DoubleRowProcessor() {

            super();

        }

        @Override

        public void processRow(Object inputRow, List outputRows) {

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

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

                byte[] rgb = processValue(inputByteRow[i]);

                for (int band = 0; band < outputRows.size(); band++) {

                    byte[] bs = (byte[])(outputRows.get(band));

                    bs[i] = rgb[band];

                }

            }

        }

        @Override

        public byte[] processValue(Object value) {

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

                byte result[] = {0,0,0,0};

                return result;

            }

            return colorInterpretation.getPalette().getRGBA(value);

        }

    }

}