Application: gvSIG desktop » gvSIG raster

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

import java.awt.Color;

import java.io.File;

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

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

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

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

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

/**

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

 *

 */

public class gvSIGColorTableIOTest extends AbstractLibraryAutoInitTestCase {

    private ColorTableIO colorTableIO;

    @Override

    protected void doSetUp() throws Exception {

        RasterLegendManager rasterLegendManager = RasterLegendLocator.getRasterLegendManager();

        ColorTableIOFactory colorTableIOFactory =

            rasterLegendManager

                .getColorTableIOFactory(org.gvsig.raster.lib.legend.impl.io.gvSIGColorTableIO.NAME);

        colorTableIO = colorTableIOFactory.create();

    }

    public void testReadFile() {

        File file =

            new File(this.getClass().getClassLoader()

                .getResource("colorTables/Forest-Soil-(100).xml").getFile());

        ColorTable colorTable = null;

        try {

            colorTable = colorTableIO.read(file);

        } catch (ColorTableIOException e) {

            fail("Can not read file to get Color table");

        }

        assertEquals(100, colorTable.getClasses().size());

        assertEquals(new Color(12, 122, 12), colorTable.getClasses().get(0).getColor());

        assertEquals(new Color(204, 232, 62), colorTable.getClasses().get(50).getColor());

        assertEquals(new Color(110, 50, 18), colorTable.getClasses().get(99).getColor());

    }

    public void testReadColorTableFile() {

        File file =

            new File(this.getClass().getClassLoader().getResource("colorTables/Forest-Soil-(100).xml")

                .getFile());

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

        try {

            colorTableIO.read(colorTable, file);

        } catch (ColorTableIOException e) {

            fail("Can not read file to get Color table");

        }

        assertEquals(100, colorTable.getClasses().size());

        assertEquals(new Color(12, 122, 12), colorTable.getClasses().get(0).getColor());

        assertEquals(new Color(204, 232, 62), colorTable.getClasses().get(50).getColor());

        assertEquals(new Color(110, 50, 18), colorTable.getClasses().get(99).getColor());

    }

    public void testWrite() {

        File file = null;

        try {

            file = File.createTempFile("testGimpPalette", ".gpl");

        } catch (IOException e) {

            fail("Can not create temp file");

        }

        RasterLegendManager rasterLegendManager = RasterLegendLocator.getRasterLegendManager();

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

        colorTableClasses.add(rasterLegendManager.createColorTableClass("0", 0, 50, Color.BLACK));

        colorTableClasses.add(rasterLegendManager.createColorTableClass("1", 20, 50, Color.CYAN));

        colorTableClasses.add(rasterLegendManager.createColorTableClass("2", 40, 50, Color.RED));

        colorTableClasses.add(rasterLegendManager.createColorTableClass("3", 60, 50, Color.BLUE));

        colorTableClasses.add(rasterLegendManager.createColorTableClass("4", 80, 50, Color.WHITE));

        colorTableClasses.add(rasterLegendManager.createColorTableClass("5", 100, 50, Color.GRAY));

        ColorTable colorTable =

            rasterLegendManager.createColorTable("testColorTable", colorTableClasses, false);

        try {

            colorTableIO.write(colorTable, file);

            colorTableIO.read(colorTable, file);

        } catch (ColorTableIOException e) {

            fail("Can not read file to get Color table");

        }

        assertEquals(6, colorTable.getClasses().size());

        assertEquals(Color.BLACK, colorTable.getClasses().get(0).getColor());

        assertEquals(Color.CYAN, colorTable.getClasses().get(1).getColor());

        assertEquals(Color.RED, colorTable.getClasses().get(2).getColor());

        assertEquals(Color.BLUE, colorTable.getClasses().get(3).getColor());

        assertEquals(Color.WHITE, colorTable.getClasses().get(4).getColor());

        assertEquals(Color.GRAY, colorTable.getClasses().get(5).getColor());

    }

}