Application: gvSIG desktop » gvSIG raster

import java.awt.Color;

import java.util.ArrayList;

import java.util.List;

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;

    public Object invoke(DynObject self, Object[] args) throws DynMethodException {

        ColorInterpretation colorInterpretation = null;

        // try {

        JimiRasterStoreProvider jimiProvider = (JimiRasterStoreProvider) self;

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        MemoryImage image = jimiProvider.getMemoryImage();

        ColorModel colorModel = image.colorModel;

        if (colorModel instanceof DirectColorModel) {

            if (image.bands == 3) {

                colorInterpretation = legendManager.createColorInterpretation(ColorInterpretation.RGB);

            } else if (image.bands == 4) {

                colorInterpretation = legendManager.createColorInterpretation(ColorInterpretation.ARGB);

            }

        } else if (colorModel instanceof IndexColorModel) {

            colorInterpretation = legendManager.createColorInterpretation(ColorInterpretation.PALETTE);

            ColorTable colorTable = null;

            IndexColorModel indexColorModel = (IndexColorModel) colorModel;

            int[] rgbs = new int[indexColorModel.getMapSize()];

            boolean hasalpha = indexColorModel.hasAlpha();

            indexColorModel.getRGBs(rgbs);

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

            for (int i = 0; i < rgbs.length; i++) {

                String className = i + "";

                double value = ((byte) i);

                double interpolation = 50.0;

                Color color = new Color(rgbs[i], hasalpha);

                ColorTableClass colorTableClass =

                    legendManager.createColorTableClass(className, value, interpolation, color);

                colorTableClasses.add(colorTableClass);

            }

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

            colorTable = legendManager.createColorTable(colorTableName, colorTableClasses, true);

            colorInterpretation.setPalette(colorTable);

            colorInterpretation.setPaletteBand(0);

        }