Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.gdal.provider.legend;

import java.awt.Color;

import java.awt.image.ColorModel;

import java.awt.image.DirectColorModel;

import java.awt.image.IndexColorModel;

import org.gdal.gdal.Band;

import org.gdal.gdal.ColorTable;

import org.gdal.gdal.Dataset;

import org.gdal.gdal.gdal;

import org.gdal.gdalconst.gdalconst;

import org.gdal.gdalconst.gdalconstConstants;

import org.gdal.gdalconst.gdalconstJNI;

import org.gvsig.fmap.dal.raster.api.RasterStore;

import org.gvsig.raster.gdal.provider.RasterGdalStoreProvider;

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynClass;

import org.gvsig.tools.dynobject.DynMethod;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.exception.DynMethodException;

import org.gvsig.tools.dynobject.exception.DynMethodNotSupportedException;

/**

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalGetColorInterpretation implements DynMethod {

    private static Integer code = null;

    static void register(DynClass storeClass) {

        if (code != null) {

            return;

        }

        code = ToolsLocator.getDynObjectManager()

                .registerDynMethod(storeClass, new RasterGdalGetColorInterpretation());

    }

    @Override

    public int getCode() throws DynMethodNotSupportedException {

        return code;

    }

    @Override

    public String getDescription() {

        return "Raster GDAL Color Interpretation";

    }

    @Override

    public String getName() {

        return RasterStore.DYNMETHOD_GETCOLORINTERPRETATION_NAME;

    }

    @Override

    public Object invoke(DynObject self, Object[] args)

        throws DynMethodException {

        ColorInterpretation colorInterpretation = null;

        RasterGdalStoreProvider rasterGdalProvider =

            (RasterGdalStoreProvider) self;

        RasterLegendManager legendManager =

            RasterLegendLocator.getRasterLegendManager();

        Dataset gdalDataSet = rasterGdalProvider.getGdalDataSet();

        String[] bandColorInterpretations =

            new String[gdalDataSet.getRasterCount()];

        for (int i = 0; i < gdalDataSet.getRasterCount(); i++) {

            // Bands begin in 1 instead of 0

            int bandNumber = i + 1;

            Band gdalBand = gdalDataSet.GetRasterBand(bandNumber);

            String bandColorInterpretion =

                colorInterpretationFromGdal(gdalBand.GetColorInterpretation());

            bandColorInterpretations[i] = bandColorInterpretion;

        }

        colorInterpretation =

            legendManager.createColorInterpretation(bandColorInterpretations);

        return colorInterpretation;

    }

    /**

     * Obtiene la cadena que representa el tipo de banda de color. Los tipos posibles son:

     * <UL>

     *  <LI>0 = "Undefined" </LI>

     *  <LI>1 = "Gray";</LI>

     *  <LI>2 = "Palette";</LI>

     *  <LI>3 = "Red";</LI>

     *  <LI>4 = "Green";</LI>

     *  <LI>5 = "Blue";</LI>

     *  <LI>6 = "Alpha";</LI>

     *  <LI>7 = "Hue";</LI>

     *  <LI>8 = "Saturation";</LI>

     *  <LI>9 = "Lightness";</LI>

     *  <LI>10 = "Cyan";</LI>

     *  <LI>11 = "Magenta";</LI>

     *  <LI>12 = "Yellow";</LI>

     *  <LI>13 = "Black";</LI>

     *  <LI>14 = "YCbCr_Y";</LI>

     *  <LI>15 = "YCbCr_Cb";</LI>

     *  <LI>16 = "YCbCr_Cr";</LI>

     * </UL>

     * @return  Cadena con el nombre del tipo de banda de colorn

     */

    private String colorInterpretationFromGdal(int gdalCode){

        String gdalNameCI=gdal.GetColorInterpretationName(gdalCode);

        switch (gdalNameCI) {

        case "Undefined" :

            return ColorInterpretation.PALETTE_BAND;

        case "Gray" :

            return ColorInterpretation.GRAY_BAND;

        case "Palette" :

            return ColorInterpretation.PALETTE_BAND;

        case "Red" :

            return ColorInterpretation.RED_BAND;

        case "Green" :

            return ColorInterpretation.GREEN_BAND;

        case "Blue" :

            return ColorInterpretation.BLUE_BAND;

        case "Alpha" :

            return ColorInterpretation.ALPHA_BAND;

        case "Hue" :

            return ColorInterpretation.HUE_BAND;

        case "Saturation" :

            return ColorInterpretation.SATURATION_BAND;

        case "Lightness" :

            return ColorInterpretation.LIGHTNESS_BAND;

        case "Cyan" :

            return ColorInterpretation.CYAN_BAND;

        case "Magenta" :

            return ColorInterpretation.MAGENTA_BAND;

        case "Yellow" :

            return ColorInterpretation.YELLOW_BAND;

        case "Black" :

            return ColorInterpretation.BLACK_BAND;

        case "YCbCr_Y" :

            return ColorInterpretation.YCBCR_Y_BAND;

        case "YCbCr_Cb" :

            return ColorInterpretation.YCBCR_CB_BAND;

        case "YCbCr_Cr" :

            return ColorInterpretation.YCBCR_CR_BAND;

        default:

            return ColorInterpretation.UNDEFINED_BAND;

        }

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        return super.clone();

    }

}