Application: gvSIG desktop

package org.gvsig.fmap.raster.tools;

import java.awt.Color;

import java.awt.Graphics2D;

import java.awt.Rectangle;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.gvsig.raster.datastruct.ColorItem;

import org.gvsig.raster.datastruct.ColorTable;

import org.gvsig.raster.datastruct.persistence.ColorTableLibraryPersistence;

import org.gvsig.gui.IColorTables;

import org.gvsig.gui.IColorTables.IColorTablePaint;

public class ColorTables implements IColorTables {

        /**

         * 

         */

        private static final long serialVersionUID = -8518160001050481010L;

        private List tables = null;

        ColorTables(){}

        public void load(String palettesPath, boolean interpolateValues) {

                tables = new ArrayList();

                ArrayList fileList = ColorTableLibraryPersistence.getPaletteFileList(palettesPath);

                for (int i = 0; i < fileList.size(); i++) {

                        ArrayList paletteItems = new ArrayList();

                        String paletteName = ColorTableLibraryPersistence.loadPalette(

                                                palettesPath,

                                                (String) fileList.get(i),

                                                paletteItems);

                        if (paletteItems.size() <= 0) {

                                continue;

                        }

                        ColorTable colorTable = new ColorTable();

                        colorTable.setName(paletteName);

                        colorTable.createPaletteFromColorItems(paletteItems, true);

                        colorTable.setInterpolated(interpolateValues);

                        this.tables.add(new ColorTablePaint(colorTable,paletteName));

                }

        }

        public IColorTablePaint get(int i){

                return (IColorTablePaint) this.tables.get(i);

        }

        public int size(){

                return this.tables.size();

        }

        public Iterator iterator(){

                return this.tables.iterator();

        }

        /**

         *

         * Class to paint a color palette in a box. It can be indicated if the palette

         * is selected and if it is painted with interpolations

         *

         * @version 30/07/2007

         * @author BorSanZa - Borja S?nchez Zamorano (borja.sanchez@iver.es)

         */

        public static class ColorTablePaint implements IColorTablePaint {

                private ColorTable colorTable;

                private String tableName;

                /**

                 * Build a ColorTablePaint with a color table (parameter or the method)

                 *

                 * @param colorTable

                 */

                public ColorTablePaint(ColorTable colorTable, String tableName) {

                        this.colorTable = colorTable;

                        this.tableName = tableName;

                }

                /**

                 * Defines if the values are interpolated between them or not.

                 *

                 * @param value

                 */

                public void setInterpolated(boolean value) {

                        colorTable.setInterpolated(value);

                }

                /**

                 * Obtains the color array of the color palette

                 *

                 * @return

                 */

                private ArrayList getColorItems() {

                        return colorTable.getColorItems();

                }

                public Color[] getColors() {

                        List colorItems = colorTable.getColorItems();

                        Color[] colors = new Color[colorItems.size()];

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

                                colors[i] = ((ColorItem) colorItems.get(i)).getColor();

                        }

                        return colors;

                }

                /**

                 * Obtains the ColorTable

                 * @return

                 */

                private ColorTable getColorTable() {

                        return colorTable;

                }

                /**

                 * Specifies the colors of the color table, defining it the values are

                 * interpolated and if the palette will be compressed or not.

                 *

                 * @param value

                 * @param interpolated

                 * @param compress

                 */

                private void setColorItems(ArrayList value, boolean interpolated, boolean compress) {

                        colorTable.createPaletteFromColorItems(value, compress);

                        setInterpolated(interpolated);

                }

                /**

                 * Method to paint the color table

                 *

                 * @param g

                 * @param isSelected

                 */

                public void paint(Graphics2D g, boolean isSelected, Rectangle bounds) {

                        Rectangle area = bounds;

                        area.y=0;

                        area.x=0;

                        int x1 = area.x;

                        int x2 = area.x + area.width - 1;

                        if (colorTable.getColorItems().size()>=1) {

                                double min = ((ColorItem) colorTable.getColorItems().get(0)).getValue();

                                double max = ((ColorItem) colorTable.getColorItems().get(colorTable.getColorItems().size()-1)).getValue();

                                for (int i = area.x; i < (area.x + area.width); i++) {

                                        double pos = min + (((max - min) * (i - area.x)) / (area.width - 2));

                                        byte[] col3 = colorTable.getRGBAByBand(pos);

                                        g.setColor(new Color(col3[0] & 0xff, col3[1] & 0xff, col3[2] & 0xff));

                                        g.drawLine(i, area.y, i, area.y + area.height);

                                }

                        } else {

                                g.setColor(new Color(224, 224, 224));

                                g.fillRect(x1, area.y, x2 - x1, area.height - 1);

                        }

                        if (isSelected) {

                                g.setColor(Color.black);

                        } else {

                                g.setColor(new Color(96, 96, 96));

                        }

                        g.drawRect(x1, area.y, x2 - x1, area.height - 1);

                }

                public String getTableName(){

                        return this.tableName;

                }

        }

}