Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.lib.legend.impl.io;

import java.awt.Color;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileReader;

import java.io.FileWriter;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

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

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

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

import org.gvsig.raster.lib.legend.api.colortable.colortableclass.ColorTableClass;

import org.gvsig.raster.lib.legend.api.exceptions.ColorTableIOException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * {@link ColorTableIO} to read Gimp palettes to {@link ColorTable} and write

 * {@link ColorTable} to Gimp palettes.

 * 

 * @author <a href="mailto:lmarques@disid.com">Lluis Marques</a>

 *

 */

public class GimpPaletteColorTableIO implements ColorTableIO {

    private static final Logger LOG = LoggerFactory.getLogger(GimpPaletteColorTableIO.class);

    public static final String NAME = "GimpPaletteColorTableIO";

    public static final String DESCRIPTION = "ColorTableIO to read and write Gimp palettes";

    @Override

    public String getName() {

        return NAME;

    }

    @Override

    public String getDescription() {

        return DESCRIPTION;

    }

    @Override

    public ColorTable read(File file) throws ColorTableIOException {

        if (file.isDirectory()) {

            throw new IllegalArgumentException("File can not be a directory.");

        }

        if (!file.exists() || !file.canRead()) {

            throw new IllegalArgumentException("File does not exist or can be readed");

        }

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

        ColorTable colorTable = RasterLegendLocator.getRasterLegendManager().createColorTable();

        try {

            BufferedReader reader = new BufferedReader(new FileReader(file));

            String currentLine;

            int cont = 0;

            while ((currentLine = reader.readLine()) != null) {

                if (currentLine.charAt(0) == '#'){

                    continue;

                }

                if (cont == 1) {

                    colorTable.setName(currentLine.substring(6));

                } else if (cont > 1) {

                    String[] strings = (" " + currentLine).split("\\s+");

                    ColorTableClass colorTableClass = null;

                    if (strings.length >= 4) {

                        colorTableClass = parseColorItem(strings);

                    } else {

                        continue;

                    }

                    strings = (" " + currentLine).split("\\s+\\d+\\s+\\d+\\s+\\d+\\s+");

                    if (!strings[1].equals("Untitled")) {

                        colorTableClass.setName(strings[1]);

                    } else {

                        colorTableClass.setName("");

                    }

                    colorTableClass.setValue(cont - 2);

                    colorItems.add(colorTableClass);

                }

                cont++;

            }

            reader.close();

        } catch (IOException ex) {

            LOG.error("Can not read file to get gimp palette", ex);

            throw new ColorTableIOException(ex);

        }

        colorTable.setClasses(colorItems, false);

        return colorTable;

    }

    private ColorTableClass parseColorItem(String[] strings) {

        ColorTableClass item = RasterLegendLocator.getRasterLegendManager().createColorTableClass();

        item.setInterpolated(100);

        item.setColor(new Color(Integer.valueOf(strings[1]).intValue(), Integer.valueOf(strings[2])

            .intValue(), Integer.valueOf(strings[3]).intValue()));

        return item;

    }

    @Override

    public void read(ColorTable colorTable, File file) throws ColorTableIOException {

        ColorTable tmpColorTable = this.read(file);

        colorTable.copyFrom(tmpColorTable);

    }

    @Override

    public void write(ColorTable colorTable, File file) throws ColorTableIOException {

        List<ColorTableClass> colorTableClasses = colorTable.getClasses();

        BufferedWriter writer;

        try {

            writer = new BufferedWriter(new FileWriter(file));

            writer.write("GIMP Palette\n");

            writer.write("Name: " + colorTable.getName() + "\n");

            writer.write("#\n");

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

                String line = "";

                ColorTableClass item1 = colorTableClasses.get(i);

                Color color = item1.getColor();

                line += color.getRed() + " ";

                line += color.getGreen() + " ";

                line += color.getBlue() + "\t";

                if ((item1.getName() != null) && (item1.getName().length() > 0)) {

                    line += item1.getName() + "\n";

                } else {

                    line += "Untitled\n";

                }

                writer.write(line);

            }

            writer.close();

        } catch (IOException e) {

            LOG.error("Can not write to create new Gimp palette file from color table", e);

            throw new ColorTableIOException(e);

        }

    }

}