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

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

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

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

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

/**

 * @author fdiaz

 *

 */

public class ColorBalanceRGBOperation extends AbstractColorBalanceOperation {

    static public String RED_PARAM = "red";

    static public String GREEN_PARAM = "green";

    static public String BLUE_PARAM = "blue";

    static public String LUMINOSITY_PARAM = "luminosity";

    private int red;

    private int green;

    private int blue;

    private boolean luminosity;

    /**

     * @param factory

     *

     */

    public ColorBalanceRGBOperation(OperationFactory factory) {

        super(factory);

    }

    @Override

    protected void initialiceParams() {

        red = (Integer) this.getParameter(RED_PARAM,0);

        green = (Integer) this.getParameter(GREEN_PARAM,0);

        blue = (Integer) this.getParameter(BLUE_PARAM,0);

        luminosity = (Boolean) this.getParameter(LUMINOSITY_PARAM,true);

    }

    @Override

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

        }

        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 lum = colorManager.getLuminosity(value[0] & 0xff, value[1] & 0xff, value[2] & 0xff);

            value[0] = (byte)Math.min((value[0] & 0xff) + (red >> 1), 255);

            value[1] = (byte)Math.min((value[1] & 0xff) + (green >> 1), 255);

            value[2] = (byte)Math.min((value[2] & 0xff) + (blue >> 1), 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];

            }

        }

    }

}