Application: gvSIG desktop

package org.gvsig.fmap.dal.fileutils.impl;

import org.gvsig.fmap.dal.fileutils.CLRFile;

import java.awt.Color;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

import static org.gvsig.fmap.dal.fileutils.PRJFile.FILE_EXTENSION;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultCLRFile implements CLRFile {

    private static final Logger logger = LoggerFactory.getLogger(DefaultCLRFile.class);

    private File source;

    private ColorTable colorTable;

    public DefaultCLRFile() {

        this.source = null;

        this.colorTable = null;

    }

    @SuppressWarnings("OverridableMethodCallInConstructor")

    public DefaultCLRFile(File file) {

        this();

        this.read(file);

    }

    @Override

    public File getFile(File file) {

        File f = new File(FilenameUtils.removeExtension(file.getAbsolutePath()) + "." + FILE_EXTENSION);

        return f;

    }

    @Override

    public File getFile() {

        return source;

    }

    @Override

    public ColorTable getColorTable() {

        return this.colorTable;

    }

    @Override

    public void setColorTable(ColorTable colorTable) {

        this.colorTable = colorTable;

    }

    @Override

    public void read(File file) {

        File f = this.getFile(file);

        if (f.exists()) {

            try {

                RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

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

                List<String> lines = FileUtils.readLines(f);

                if (lines != null) {

                    this.source = f.getAbsoluteFile();

                    int lineno = 1;

                    double interpolation = 50.0;

                    for (String line : lines) {

                        line = line.trim();

                        if (!Character.isDigit(line.charAt(0))) {

                            continue;

                        }

                        try {

                            String[] words = StringUtils.split(line);

                            if (words.length < 4) {

                                continue;

                            }

                            int value = Integer.parseUnsignedInt(words[0]);

                            int r = Integer.parseUnsignedInt(words[1]);

                            int g = Integer.parseUnsignedInt(words[2]);

                            int b = Integer.parseUnsignedInt(words[3]);

                            Color color = new Color(r, g, b);

                            ColorTableClass colorTableClass = legendManager.createColorTableClass(

                                    String.valueOf(value), value, interpolation, color);

                            colors.add(colorTableClass);

                        } catch (Exception e) {

                            logger.warn("Can't parse line '" + line

                                    + "' (lineno=+" + lineno

                                    + ", file=" + f.getAbsoluteFile()

                                    + ").", e);

                        }

                    }

                }

                if( !colors.isEmpty() ) {

                    String colorTableName = FilenameUtils.getBaseName(f.getName());

                    this.colorTable = legendManager.createColorTable(colorTableName, colors, true);

                }

            } catch (IOException e) {

                logger.warn("Couldn't read " + FILE_EXTENSION + " file (" + f.getAbsoluteFile() + ")", e);

            }

        }

    }

    @Override

    public void write(File file) throws IOException {

        File f = this.getFile(file);

        // TODO: falta implementar guardar la CLR

    }

}