Application: gvSIG desktop » gvSIG data providers based on gdal

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

import java.awt.Color;

import java.util.ArrayList;

import java.util.List;

import org.gdal.gdal.Band;

import org.gdal.gdal.Dataset;

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

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

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.api.colortable.ColorTable;

import org.gvsig.raster.lib.legend.api.colortable.colortableclass.ColorTableClass;

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;

/**

 * Dynamic method to add a get Color Table funcionality

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalGetColorTable implements DynMethod {

    private static Integer code = null;

    static void register(DynClass storeClass) {

        if (code != null) {

            return;

        }

        code = ToolsLocator.getDynObjectManager()

                .registerDynMethod(storeClass, new RasterGdalGetColorTable());

    }

    @Override

    public int getCode() throws DynMethodNotSupportedException {

        return code;

    }

    @Override

    public String getDescription() {

        return "Raster GDAL Color Table";

    }

    @Override

    public String getName() {

        return RasterStore.DYNMETHOD_GETCOLORTABLE_NAME;

    }

    @Override

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

        throws DynMethodException {

        ColorTable colorTable = null;

        try {

            AbstractRasterGdalStoreProvider rasterGdalProvider=(AbstractRasterGdalStoreProvider)self;

            RasterLegendManager legendManager=RasterLegendLocator.getRasterLegendManager();

            Dataset gdalDataSet = rasterGdalProvider.getGdalDataSet();

            ColorInterpretation colorInterpretation =(ColorInterpretation)rasterGdalProvider.

                invokeDynMethod(RasterStore.DYNMETHOD_GETCOLORINTERPRETATION_NAME, null);

            for (int bandNumber = 1; bandNumber <= gdalDataSet.getRasterCount(); bandNumber++) {

                //Band in GDAL begins in 1, meanwhile we begin in 0

                int indexColorInterpretation=bandNumber-1;

                if (ColorInterpretation.PALETTE_BAND.equals(colorInterpretation.get(indexColorInterpretation))){

                    Band gdalBand=gdalDataSet.GetRasterBand(bandNumber);

                    org.gdal.gdal.ColorTable gdalBandColorTable = gdalBand.GetColorTable();

                    if (gdalBandColorTable!=null){

                        List<ColorTableClass> colorTableClasses=new ArrayList<ColorTableClass>();

                        for (int i=0;i<gdalBandColorTable.GetCount();i++){

                            String className=i+"";

                            double value=((byte)i);

                            double interpolation=50.0;

                            Color color=gdalBandColorTable.GetColorEntry(i);

                            ColorTableClass colorTableClass = legendManager.createColorTableClass(className, value, interpolation, color);

                            colorTableClasses.add(colorTableClass);

                        }

                        String colorTableName = rasterGdalProvider.getName() + "_color_table";

                        colorTable = legendManager.createColorTable(colorTableName,

                            colorTableClasses, true);

                        //FIXME: De momento devolvemos el primero solo.

                        return colorTable;

                    }

                    else {

                        List<ColorTableClass> colorTableClasses=new ArrayList<ColorTableClass>();

                        double[] minMax=new double[2];

                        gdalBand.ComputeRasterMinMax(minMax);

                        Double increment=(minMax[1]-minMax[0])/256;

                        for (int i=0;i<255;i++){

                            String className=i+"";

                            double value=((minMax[0]+(i*increment)));

                            double interpolation=50.0;

                            int intARGB = ((i & 0xFF) << 24) | // alpha

                                ((i & 0xFF) << 16) | // red

                                ((i & 0xFF) << 8) | // green

                                ((i & 0xFF) << 0); // blue

                            Color color=new Color(intARGB);

                            ColorTableClass colorTableClass = legendManager.createColorTableClass(className, value, interpolation, color);

                            colorTableClasses.add(colorTableClass);

                        }

                        String colorTableName = rasterGdalProvider.getName() + "_color_table";

                        colorTable = legendManager.createColorTable(colorTableName,

                            colorTableClasses, true);

                        //FIXME: De momento devolvemos el primero solo.

                        return colorTable;

                    }

                }

            }

       } catch (DynMethodNotSupportedException e) {

           //FIXME

           throw new RuntimeException(e);

       }

       return null;

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        return super.clone();

    }

}