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

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 ColorBalanceHSLOperation extends AbstractColorBalanceOperation {

    static public String COLOR_INTERPRETATION_PARAM = "color_interpretation";

    static public String HUE_PARAM = "hue";

    static public String LIGHTNESS_PARAM = "lightness";

    static public String SATURATION_PARAM = "saturation";

    static public String COPY_UNPROCESSED_BANDS_PARAM = "copy_unprocessed_bands";

    private ColorInterpretation colorInterpretation;

    private double hue = 0;

    private double lightness = 0;

    private double saturation = 0;

    private boolean copyUnprocessedBands;

    /**

     * @param factory

     *

     */

    public ColorBalanceHSLOperation(OperationFactory factory) {

        this.factory = factory;

    }

    @Override

    protected void initialiceParams() {

        hue = (Double) this.parameters.getDynValue(HUE_PARAM);

        lightness = (Double) this.parameters.getDynValue(LIGHTNESS_PARAM);

        saturation = (Double) this.parameters.getDynValue(SATURATION_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];

        }

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

            hsl[0] = ((hsl[0] + hue) % 360);

            hsl[1] = Math.max(Math.min(hsl[1] + (saturation / 100.0), 1D), 0.001);

            hsl[2] = Math.max(Math.min(hsl[2] + (lightness / 100.0), 1D), 0);

            int[] rgb = colorManager.HSLtoRGB(hsl[0], hsl[1], hsl[2]);

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

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

            }

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

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

            }

        }

    }

}