Application: gvSIG desktop » gvSIG raster

package org.gvsig.fmap.dal.file.jimi.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.gvsig.fmap.dal.file.jimi.JimiRasterStoreProvider;

import org.gvsig.fmap.dal.file.jimi.MemoryImage;

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

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

import org.gvsig.raster.lib.legend.api.colortable.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 JimiGetColorTable implements DynMethod {

    private static Integer code = null;

    static void register(DynClass storeClass) {

        if (code != null) {

            return;

        }

        code = ToolsLocator.getDynObjectManager()

                .registerDynMethod(storeClass, new JimiGetColorTable());

    }

    @Override

    public int getCode() throws DynMethodNotSupportedException {

        return code;

    }

    @Override

    public String getDescription() {

        return "Jimi 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 {

             JimiRasterStoreProvider jimiProvider=(JimiRasterStoreProvider)self;

             RasterLegendManager legendManager=RasterLegendLocator.getRasterLegendManager();

             MemoryImage image=jimiProvider.getMemoryImage();

             ColorModel colorModel=image.colorModel;

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

             }

//        } catch (DynMethodNotSupportedException e) {

//            //FIXME

//            throw new RuntimeException(e);

//        }

        return colorTable;

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        return super.clone();

    }

}