Application: gvSIG desktop

package org.gvsig.fmap.dal.file.imageio.legend;

import java.awt.Color;

import java.awt.image.ColorModel;

import java.awt.image.IndexColorModel;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.file.imageio.ImageIOStoreProvider;

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

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

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

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;

public class ImageIOGetColorTable implements DynMethod {

    private static Integer code = null;

    static public void register(DynClass storeClass) {

        if (code != null) {

            return;

        }

        code = ToolsLocator.getDynObjectManager()

                .registerDynMethod(storeClass, new ImageIOGetColorTable());

    }

    @Override

    public int getCode() throws DynMethodNotSupportedException {

        return code;

    }

    @Override

    public String getDescription() {

        return "ImageIO based 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 {

        ImageIOStoreProvider provider = (ImageIOStoreProvider) self;

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        ColorModel colorModel = provider.getImage().getColorModel();

        if (colorModel instanceof IndexColorModel) {

            IndexColorModel indexColorModel = (IndexColorModel) colorModel;

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

            boolean hasalpha = indexColorModel.hasAlpha();

            indexColorModel.getRGBs(rgbs);

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

            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 = provider.getName() + "_color_table";

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

        }

        return colorTable;

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        return super.clone();

    }

}