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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

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

 * {@link ColorTable} to Gimp gradients.

 * 

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

 *

 */

public class GimpGradientColorTableIO implements ColorTableIO {

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

    public static final String NAME = "GimpGradientColorTableIO";

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

    @Override

    public String getName() {

        return NAME;

    }

    @Override

    public String getDescription() {

        return DESCRIPTION;

    }

    @Override

    public ColorTable read(File file) throws ColorTableIOException {

        List<ColorTableClass> colorTableClasses = 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 (cont == 1) {

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

                } else if (cont > 1) {

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

                    if (strings.length < 13) {

                        continue;

                    }

                    colorTableClasses.add(parseColorItem(strings, 0));

                    colorTableClasses.add(parseColorItem(strings, 1));

                }

                cont++;

            }

            reader.close();

        } catch (IOException ex) {

            LOG.error("Error reading file to create color table", ex);

            throw new ColorTableIOException(ex);

        }

        colorTable.setClasses(colorTableClasses, true);

        return colorTable;

    }

    @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> colorTableClass = colorTable.getClasses();

        try {

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

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

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

            writer.write((colorTableClass.size() - 1) + "\n");

            String line;

            for (int i = 1; i < colorTableClass.size(); i++) {

                double min = colorTableClass.get(0).getValue();

                double max = colorTableClass.get(colorTableClass.size() - 1).getValue();

                line = "";

                ColorTableClass item1 = (ColorTableClass) colorTableClass.get(i - 1);

                ColorTableClass item2 = (ColorTableClass) colorTableClass.get(i);

                double pos1 = (item1.getValue() - min) / (max - min);

                double pos2 = (item2.getValue() - min) / (max - min);

                NumberFormat format = NumberFormat.getNumberInstance();

                format.setMaximumFractionDigits(6);

                format.setMinimumFractionDigits(6);

                line += format.format(pos1).replaceAll(",", ".") + " ";

                double interp = pos1 + (((pos2 - pos1) * item1.getInterpolated()) / 100.0);

                line += format.format(interp).replaceAll(",", ".") + " ";

                line += format.format(pos2).replaceAll(",", ".") + " ";

                Color color = item1.getColor();

                line += format.format(color.getRed() / 255.0).replaceAll(",", ".") + " ";

                line += format.format(color.getGreen() / 255.0).replaceAll(",", ".") + " ";

                line += format.format(color.getBlue() / 255.0).replaceAll(",", ".") + " ";

                line += format.format(color.getAlpha() / 255.0).replaceAll(",", ".") + " ";

                color = item2.getColor();

                line += format.format(color.getRed() / 255.0).replaceAll(",", ".") + " ";

                line += format.format(color.getGreen() / 255.0).replaceAll(",", ".") + " ";

                line += format.format(color.getBlue() / 255.0).replaceAll(",", ".") + " ";

                line += format.format(color.getAlpha() / 255.0).replaceAll(",", ".") + " ";

                line += "0 0\n";

                writer.write(line);

            }

            writer.close();

        } catch (IOException ex) {

            LOG.error("Error writting file to create Gimp gradient", ex);

            throw new ColorTableIOException(ex);

        }

    }

    private ColorTableClass parseColorItem(String[] strings, int pos) {

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

        double pos1 = Double.valueOf(strings[0]).doubleValue();

        double pos2 = Double.valueOf(strings[1]).doubleValue();

        double pos3 = Double.valueOf(strings[2]).doubleValue();

        if (pos == 0) {

            item.setValue(pos1 * 255);

        } else {

            item.setValue(pos3 * 255);

        }

        if (pos == 0) {

            item.setInterpolated((100 * (pos2 - pos1)) / (pos3 - pos1));

        }

        if (pos == 0) {

            item.setColor(new Color((int) (Double.valueOf(strings[3]).doubleValue() * 255.0),

                (int) (Double.valueOf(strings[4]).doubleValue() * 255.0), (int) (Double.valueOf(

                    strings[5]).doubleValue() * 255.0), (int) (Double.valueOf(strings[6])

                    .doubleValue() * 255.0)));

        } else {

            item.setColor(new Color((int) (Double.valueOf(strings[7]).doubleValue() * 255.0),

                (int) (Double.valueOf(strings[8]).doubleValue() * 255.0), (int) (Double.valueOf(

                    strings[9]).doubleValue() * 255.0), (int) (Double.valueOf(strings[10])

                    .doubleValue() * 255.0)));

        }

        return item;

    }

}