Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.raster.impl.grid.filter.band;

import java.awt.Color;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.coverage.datastruct.ColorItem;

import org.gvsig.fmap.dal.coverage.datastruct.Params;

import org.gvsig.fmap.dal.coverage.exception.FilterTypeException;

import org.gvsig.fmap.dal.coverage.grid.AbstractRasterFilterManager;

import org.gvsig.fmap.dal.coverage.grid.RasterFilter;

import org.gvsig.fmap.dal.coverage.grid.RasterFilterList;

import org.gvsig.fmap.dal.coverage.grid.filter.BaseRasterFilter;

import org.gvsig.fmap.dal.coverage.store.props.ColorTable;

import org.gvsig.raster.impl.datastruct.ColorItemImpl;

import org.gvsig.raster.impl.grid.filter.RasterFilterListManagerImpl;

import org.gvsig.raster.impl.store.ParamImpl;

import org.gvsig.raster.impl.store.properties.DataStoreColorTable;

/**

 * Gestor del filtro de aplicaci?n de tablas de color sobre un raster.

 *

 * @version 06/06/2007

 * @author Nacho Brodin (nachobrodin@gmail.com)

 *

 */

public class ColorTableListManager  extends AbstractRasterFilterManager {

        private static String   ID = "ColorTable";

        /**

         * Default constructor. Sets the filter list.

         * @param filterList

         */

        public ColorTableListManager(RasterFilterList filterList) {

                super(filterList);

        }

        public String getManagerID() {

                return ID;

        }

        public static void register() {

                AbstractRasterFilterManager.register(ID, ColorTableListManager.class);

        }

        public boolean isDataTypeSupported(int dataType) {

                return true;

        }

        public Class<?> getFilterClassByID(String id) {

                if( id.compareTo("colortable") == 0)

                        return ColorTableFilter.class;

                return null;

        }

        /**

         * Constructor.

         * Asigna la lista de filtros y el managener global.

         *

         * @param filterListManager

         */

        public ColorTableListManager(RasterFilterListManagerImpl filterListManager) {

                super(filterListManager.getFilterList());

        }

        /**

         * A?ade un filtro de tabla de color a la pila de filtros.

         * @param ladoVentana

 * @throws FilterTypeException

         */

        public void addColorTableFilter(ColorTable palette) throws FilterTypeException {

                BaseRasterFilter filter = new ColorTableByteFilter();

                //Cuando el filtro esta creado, tomamos los valores y lo a?adimos a la pila

                if (filter != null) {

                        filter.addParam("colorTable", palette);

                        getFilterList().add(filter);

                }

        }

        public List<Class<?>> getRasterFilterList() {

                List<Class<?>> filters = new ArrayList<Class<?>>();

                filters.add(ColorTableFilter.class);

                return filters;

        }

        public void addFilter(Class<?> classFilter, Params params) throws FilterTypeException {

                if (ColorTableFilter.class.isAssignableFrom(classFilter)) {

                        ColorTable colorTable = null;

                        for (int i = 0; i < params.getNumParams(); i++) {

                                if (((ParamImpl)params.getParam(i)).getId().equals("colorTable"))

                                        colorTable = (ColorTable) ((ParamImpl)params.getParam(i)).getDefaultValue();

                        }

                        addColorTableFilter(colorTable);

                }

        }

        /**

         * Devuelve el color si lo encuentra en el arraylist y lo elimina, en caso

         * contrario devuelve null

         * @param list

         * @param value

         * @return

         */

        private static ColorItem getColorItem(List<ColorItem> list, double value) {

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

                        if (((ColorItem) list.get(i)).getValue() == value) {

                                return (ColorItem) list.remove(i);

                        }

                }

                return null;

        }

        @SuppressWarnings("unchecked")

        public static ColorTable createColorTableFromArray(List<String> lines) {

                String pkgBase = "filter.colortable.";

                List<String> linesCloned = null;

                if(lines instanceof ArrayList) {

                        linesCloned = (List<String>)((ArrayList<?>) lines).clone();

                } else {

                        linesCloned = new ArrayList<String>();

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

                                linesCloned.add(lines.get(i));

                        }

                }

                String paletteName = "";

                int color = 0;

                int alpha = 0;

                List<ColorItem> rows = new ArrayList<ColorItem>();

                ColorItem colorItem = new ColorItemImpl();

                boolean interpolated = false;

                while (linesCloned.size() > 0) {

                        String elem = (String) linesCloned.get(0);

                        if (!elem.startsWith(pkgBase)) {

                                linesCloned.remove(0);

                                continue;

                        }

                        if (elem.startsWith(pkgBase + "name"))

                                paletteName = RasterFilterListManagerImpl.getValue(elem);

                        if (elem.startsWith(pkgBase + "interpolated"))

                                interpolated = Boolean.parseBoolean(RasterFilterListManagerImpl.getValue(elem));

                        if (elem.startsWith(pkgBase + "color" + color)) {

                                if (elem.startsWith(pkgBase + "color" + color + ".value"))

                                        colorItem.setValue(Double.parseDouble(RasterFilterListManagerImpl.getValue(elem)));

                                if (elem.startsWith(pkgBase + "color" + color + ".name"))

                                        colorItem.setNameClass(RasterFilterListManagerImpl.getValue(elem));

                                if (elem.startsWith(pkgBase + "color" + color + ".rgb")) {

                                        String rgb = RasterFilterListManagerImpl.getValue(elem);

                                        int r = Integer.valueOf(rgb.substring(0, rgb.indexOf(","))).intValue();

                                        int g = Integer.valueOf(rgb.substring(rgb.indexOf(",") + 1, rgb.lastIndexOf(","))).intValue();

                                        int b = Integer.valueOf(rgb.substring(rgb.lastIndexOf(",") + 1, rgb.length())).intValue();

                                        colorItem.setColor(new Color(r, g, b));

                                }

                                if (elem.startsWith(pkgBase + "color" + color + ".interpolated"))

                                        colorItem.setInterpolated(Double.parseDouble(RasterFilterListManagerImpl.getValue(elem)));

                                if ((linesCloned.size() <= 1) || (!((String) linesCloned.get(1)).startsWith(pkgBase + "color" + color))) {

                                        rows.add(colorItem);

                                        color++;

                                        colorItem = new ColorItemImpl();

                                }

                        }

                        if (elem.startsWith(pkgBase + "alpha" + alpha)) {

                                if (elem.startsWith(pkgBase + "alpha" + alpha + ".value")) {

                                        ColorItem aux = getColorItem(rows, Double.parseDouble(RasterFilterListManagerImpl.getValue(elem)));

                                        if (aux != null) {

                                                colorItem = aux;

                                                colorItem.setNameClass(aux.getNameClass());

                                                colorItem.setInterpolated(aux.getInterpolated());

                                                colorItem.setColor(new Color(aux.getColor().getRed(), aux.getColor().getGreen(), aux.getColor().getBlue(), colorItem.getColor().getAlpha()));

                                        }

                                        colorItem.setValue(Double.parseDouble(RasterFilterListManagerImpl.getValue(elem)));

                                }

                                if (elem.startsWith(pkgBase + "alpha" + alpha + ".a")) {

                                        Color c = colorItem.getColor();

                                        colorItem.setColor(new Color(c.getRed(), c.getGreen(), c.getBlue(),Integer.parseInt(RasterFilterListManagerImpl.getValue(elem))));

                                }

                                if (elem.startsWith(pkgBase + "alpha" + alpha + ".interpolated"))

                                        colorItem.setInterpolated(Double.parseDouble(RasterFilterListManagerImpl.getValue(elem)));

                                if ((linesCloned.size() <= 1) || (!((String) linesCloned.get(1)).startsWith(pkgBase + "alpha" + alpha))) {

                                        rows.add(colorItem);

                                        alpha++;

                                        colorItem = new ColorItemImpl();

                                }

                        }

                        linesCloned.remove(0);

                }

                ColorTable colorTable = new DataStoreColorTable();

                colorTable = new DataStoreColorTable();

                colorTable.setName(paletteName);

                colorTable.createPaletteFromColorItems(rows, false);

                colorTable.setInterpolated(interpolated);

                return colorTable;

        }

        public int createFilterListFromStrings(List<String> filters, String fil, int filteri) throws FilterTypeException {

                String pkgBase = "filter.colortable.";

                if (fil.startsWith(pkgBase + "active")) {

                        boolean exec = true;

                        if ((RasterFilterListManagerImpl.getValue(fil).equals("false")))

                                exec = false;

                        filters.remove(0);

                        ColorTable colorTable = createColorTableFromArray(filters);

                        getFilterList().remove(ColorTableFilter.class);

                        addColorTableFilter(new DataStoreColorTable(colorTable));

                        ColorTableFilter ct = (ColorTableFilter) getFilterList().getFilterByBaseClass(ColorTableFilter.class);

                        ct.setExec(exec);

                }

                return filteri;

        }

        public List<String> getStringsFromFilterList(List<String> filterList, RasterFilter rf) {

                if (rf instanceof ColorTableFilter) {

                        String pkgBase = "filter.colortable.";

                        ColorTableFilter colorTableFilter = (ColorTableFilter) rf;

                        ColorTable colorTable = (ColorTable) colorTableFilter.getParam("colorTable");

                        if (colorTable != null) {

                                if (colorTableFilter.isExec())

                                        filterList.add(pkgBase + "active=true");

                                else

                                        filterList.add(pkgBase + "active=false");

                                filterList.add(pkgBase + "name=" + colorTable.getName());

                                filterList.add(pkgBase + "interpolated=" + colorTable.isInterpolated());

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

                                        ColorItem colorItem = (ColorItem) colorTable.getColorItems().get(i);

                                        filterList.add(pkgBase + "color" + i + ".value=" + colorItem.getValue());

                                        filterList.add(pkgBase + "color" + i + ".name=" + colorItem.getNameClass());

                                        Color c = colorItem.getColor();

                                        filterList.add(pkgBase + "color" + i + ".rgb=" + c.getRed() + "," + c.getGreen() + "," + c.getBlue());

                                        filterList.add(pkgBase + "color" + i + ".interpolated=" + colorItem.getInterpolated());

                                }

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

                                        ColorItem colorItem = (ColorItem) colorTable.getColorItems().get(i);

                                        filterList.add(pkgBase + "alpha" + i + ".value=" + colorItem.getValue());

                                        Color c = colorItem.getColor();

                                        filterList.add(pkgBase + "alpha" + i + ".a=" + c.getAlpha());

                                        filterList.add(pkgBase + "alpha" + i + ".interpolated=" + colorItem.getInterpolated());

                                }

                        }

                }

                return filterList;

        }

        public void addFilter(Params params) throws FilterTypeException {

                addFilter(ColorTableFilter.class, params);

        }

        public RasterFilter createFilter(Params params) {

                ColorTable colorTable = ((ColorTable) params.getParamById("colorTable").getDefaultValue());

                RasterFilter filter = new ColorTableByteFilter();

                filter.addParam("colorTable", colorTable);

                return filter;

        }

}