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

import java.util.List;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferOperationException;

import org.gvsig.raster.lib.buffer.api.operations.OperationFactory;

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

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.impl.operations.colorbalancergb.AbstractColorBalanceOperation;

/**

 * @author fdiaz

 *

 */

public class ColorBalanceCMYOperation extends AbstractColorBalanceOperation {

    static public String CYAN_PARAM = "cyan";

    static public String MAGENTA_PARAM = "magenta";

    static public String YELLOW_PARAM = "yellow";

    static public String LUMINOSITY_PARAM = "luminosity";

    static public String COPY_UNPROCESSED_BANDS_PARAM = "copy_unprocessed_bands";

    private int cyan;

    private int magenta;

    private int yellow;

    private boolean luminosity;

    private boolean copyUnprocessedBands;

    /**

     * @param factory

     *

     */

    public ColorBalanceCMYOperation(OperationFactory factory) {

        this.factory = factory;

    }

    @Override

    protected void initialiceParams() {

        cyan = (Integer) this.parameters.getDynValue(CYAN_PARAM);

        magenta = (Integer) this.parameters.getDynValue(MAGENTA_PARAM);

        yellow = (Integer) this.parameters.getDynValue(YELLOW_PARAM);

        luminosity = (Boolean) this.parameters.getDynValue(LUMINOSITY_PARAM);

    }

    @Override

    public void postProcess() throws BufferOperationException {

        super.postProcess();

    }

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

        ColorManager colorManager = ((RasterLegendManagerServices)RasterLegendLocator.getRasterLegendManager()).getColorManager();

        byte[][] inputByteRows = new byte[inputRows.length][((byte[]) inputRows[0]).length];

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

            inputByteRows[i] = (byte[]) inputRows[i];

        }

//        byte[][] outputByteRows = (byte[][]) outputRows;

        for (int i = 0; i < inputByteRows[0].length; i++) {

            byte[] value = new byte[4];

            value[0] = inputByteRows[0][i];

            value[1] = inputByteRows[1][i];

            value[2] = inputByteRows[2][i];

            double[] cmyk = colorManager.RGBtoCMYK(value[0] & 0xff, value[1] & 0xff, value[2] & 0xff, 1);

            double lum = colorManager.getLuminosity(value[0] & 0xff, value[1] & 0xff, value[2] & 0xff);

            cmyk[0] = Math.min(cmyk[0] + (cyan / 300.0), 1D); //FIXME: ?Por qu? 300 y no 255?

            cmyk[1] = Math.min(cmyk[1] + (magenta / 300.0), 1D);

            cmyk[2] = Math.min(cmyk[2] + (yellow / 300.0), 1D);

            double[] rgb = colorManager.CMYKtoRGB(cmyk[0], cmyk[1], cmyk[2], cmyk[3]);

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

                value[band] = (byte)(rgb[band] * 255);

            }

            if(luminosity) {

                double[] hsl = colorManager.RGBtoHSL(value[0] & 0xff, value[1] & 0xff, value[2] & 0xff);

                hsl[0] = (int)(255.0 * hsl[0] / 360.0 + 0.5);

                hsl[1] = (int) (hsl[1] * 255. + 0.5);

                hsl[2] = (int) (lum * 255. + 0.5);

                int[] v1 = colorManager.HSLtoRGB(

                    (int)(((int)hsl[0]) & 0xffffffff),

                    (int)(((int)hsl[1]) & 0xffffffff),

                    (int)(((int)hsl[2]) & 0xffffffff));

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

                    value[band] = new Integer(v1[band]).byteValue();

                }

            }

            for (int band = 0; band < inputRows.length; band++) {

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

            }

        }

    }

}