Application: gvSIG desktop » gvSIG data providers based on gdal

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

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

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

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.raster.lib.legend.api.colorinterpretation.ColorInterpretation;

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

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 Legend funcionality

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalGetLegend implements DynMethod {

    private static Integer code = null;

    static void register(DynClass storeClass) {

        if (code != null) {

            return;

        }

        code = ToolsLocator.getDynObjectManager()

                .registerDynMethod(storeClass, new RasterGdalGetLegend());

    }

    @Override

    public int getCode() throws DynMethodNotSupportedException {

        return code;

    }

    @Override

    public String getDescription() {

        return "Raster GDAL Legend";

    }

    @Override

    public String getName() {

        return RasterStore.DYNMETHOD_GETLEGEND_NAME;

    }

    @Override

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

            throws DynMethodException {

        try {

             AbstractRasterGdalStoreProvider rasterGdalProvider=(AbstractRasterGdalStoreProvider)self;

             RasterLegendManager legendManager=RasterLegendLocator.getRasterLegendManager();

             ColorInterpretation colorInterpretation =(ColorInterpretation)rasterGdalProvider.

                 invokeDynMethod(RasterStore.DYNMETHOD_GETCOLORINTERPRETATION_NAME, null);

             RasterLegend legend= legendManager.createLegend(colorInterpretation);

//             if ( !( colorInterpretation.isBGR() || colorInterpretation.isRGB() || colorInterpretation.isRGBA() ) ){

//                 ColorTable colorTable=(ColorTable)rasterGdalProvider.

//                     invokeDynMethod(RasterStore.DYNMETHOD_GETCOLORTABLE_NAME, null);

//                 legend.setColorTable(colorTable);

//             }

             return legend;

        } catch (DynMethodNotSupportedException e) {

            //FIXME

            throw new RuntimeException(e);

        }

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        return super.clone();

    }

}