Application: gvSIG desktop

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

import java.awt.Color;

import java.awt.image.BufferedImage;

import java.awt.image.ColorModel;

import java.awt.image.ComponentColorModel;

import java.awt.image.IndexColorModel;

import java.awt.image.WritableRaster;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.file.jpg.JpgStoreProvider;

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

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

/**

 * @author dmartinezizquierdo

 *

 */

public class JpgGetColorInterpretation implements DynMethod {

    private static Integer code = null;

    static void register(DynClass storeClass) {

        if (code != null) {

            return;

        }

        code = ToolsLocator.getDynObjectManager()

                .registerDynMethod(storeClass, new JpgGetColorInterpretation());

    }

    @Override

    public int getCode() throws DynMethodNotSupportedException {

        return code;

    }

    @Override

    public String getDescription() {

        return "JPG Color Interpretation";

    }

    @Override

    public String getName() {

        return RasterStore.DYNMETHOD_GETCOLORINTERPRETATION_NAME;

    }

    @Override

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

        ColorInterpretation colorInterpretation = null;

        // try {

        JpgStoreProvider provider = (JpgStoreProvider) self;

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        BufferedImage image = provider.getImage();

        ColorModel colorModel = image.getColorModel();

        if (colorModel instanceof ComponentColorModel) { //DirectColorModel) {

            WritableRaster raster = image.getRaster();

            if (raster.getNumBands() == 3) {

                colorInterpretation = legendManager.createColorInterpretation(ColorInterpretation.RGB);

            } else if (raster.getNumBands() == 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 = provider.getName() + "_color_table";

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

            colorInterpretation.setPalette(colorTable);

            colorInterpretation.setPaletteBand(0);

        }

        return colorInterpretation;

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        return super.clone();

    }

}