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

import java.awt.image.IndexColorModel;

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

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

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

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

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

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;

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 {

            RasterGdalStoreProvider rasterGdalProvider=(RasterGdalStoreProvider)self;

            RasterLegendManager legendManager=RasterLegendLocator.getRasterLegendManager();

            Dataset gdalDataSet = rasterGdalProvider.getGdalDataSet();

            ColorInterpretation colorInterpretation =(ColorInterpretation)rasterGdalProvider.

                invokeDynMethod(RasterStore.DYNMETHOD_GETCOLORINTERPRETATION_NAME, null);

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

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

                    // Bands begin in 1 instead of 0

                    int bandNumber = i + 1;

                    Band gdalBand=gdalDataSet.GetRasterBand(bandNumber);

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

                    if (gdalBandColorTable!=null){

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

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

                            String className=j+"";

                            double value=((byte)j);

                            double interpolation=50.0;

                            Color color=gdalBandColorTable.GetColorEntry(j);

                            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();

    }

}