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

import java.util.ArrayList;

import java.util.List;

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.raster.impl.grid.filter.RasterFilterListManagerImpl;

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

/**

 * Gestor de los filtros de brillo y contraste. Contiene m?todos para a?adir

 * ambos tipos de filtro en la pila.

 *

 * @author Miguel ?ngel Querol Carratal? 

 */

public class BrightnessContrastListManager extends AbstractRasterFilterManager {

        private static String   ID = "BrightnessContrast";

        /**

         * Default constructor. Sets the filter list.

         * @param filterList

         */

        public BrightnessContrastListManager(RasterFilterList filterList) {

                super(filterList);

        }

        public String getManagerID() {

                return ID;

        }

        public static void register() {

                AbstractRasterFilterManager.register(ID, BrightnessContrastListManager.class);

        }

        /**

         * Constructor. Asigna la lista de filtros y el manager.

         * @param filterListManager

         */

        public BrightnessContrastListManager(RasterFilterListManagerImpl filterListManager) {

                super(filterListManager.getFilterList());

        }

        public boolean isDataTypeSupported(int dataType) {

                return true;

        }

        /**

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

         * @param incrBrillo valor de brillo a aplicar

         * @param removeAll true si se desea eliminar cualquier filtro de brillo que

         * hubiera en la pila antes y false si se acumula sobre lo que haya

         * @throws FilterTypeException

         */

        public void addBrightnessFilter(int incrBrillo) throws FilterTypeException {

                RasterFilter filter = createBrightnessFilter(incrBrillo);

                if (filter != null)

                        getFilterList().add(filter);

        }

        public Class<?> getFilterClassByID(String id) {

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

                        return BrightnessFilter.class;

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

                        return ContrastFilter.class;

                return null;

        }

        public static RasterFilter createBrightnessFilter(int incrBrillo) {

                RasterFilter filter = new BrightnessByteFilter();

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

                if (filter != null) {

                        filter.addParam("incrBrillo", new Integer(incrBrillo));

                }

                return filter;

        }

        /**

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

         * @param incrBrillo

         * @throws FilterTypeException

         */

        public void addContrastFilter(int incrContraste) throws FilterTypeException {

                RasterFilter filter = createContrastFilter(incrContraste);

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

                if (filter != null)

                        getFilterList().add(filter);

        }

        public static RasterFilter createContrastFilter(int incrContraste) {

                RasterFilter filter = new ContrastByteFilter();

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

                if (filter != null)

                        filter.addParam("incrContraste", new Integer(incrContraste));

                return filter;

        }

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

                if ((rf instanceof BrightnessFilter) || (rf instanceof ContrastFilter)) {

                        filterList.add("filter.brightCont.active=true");

                }

                if (rf instanceof BrightnessFilter) {

                        BrightnessFilter bright = (BrightnessFilter) rf;

                        filterList.add("filter.brightness.incrBrillo=" + bright.incrBrillo);

                } else if (rf instanceof ContrastFilter) {

                        ContrastFilter contrast = (ContrastFilter) rf;

                        filterList.add("filter.contrast.incrContraste=" + contrast.incrContraste);

                }

                return filterList;

        }

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

                if ((fil.startsWith("filter.brightCont.active")) && (RasterFilterListManagerImpl.getValue(fil).equals("true"))) {

                        int incrBrillo = 0;

                        int incrContraste = 0;

                        filters.remove(0);

                        for (int prop = 0; prop < filters.size(); prop++) {

                                String elem = (String) filters.get(prop);

                                if (elem.startsWith("filter.brightness.incrBrillo")) {

                                        incrBrillo = Integer.parseInt(RasterFilterListManagerImpl.getValue(elem));

                                        addBrightnessFilter(incrBrillo);

                                        filters.remove(prop);

                                        prop--;

                                }

                                if (elem.startsWith("filter.contrast.incrContraste")) {

                                        incrContraste = Integer.parseInt(RasterFilterListManagerImpl.getValue(elem));

                                        addContrastFilter(incrContraste);

                                        filters.remove(prop);

                                        prop--;

                                }

                        }

                        filteri = -1;

                }

                return filteri;

        }

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

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

                filters.add(BrightnessFilter.class);

                filters.add(ContrastFilter.class);

                return filters;

        }

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

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

                        int contrast = 0;

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

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

                                        contrast = ((Integer) ((ParamImpl)params.getParam(i)).getDefaultValue()).intValue();

                        }

                        addContrastFilter(contrast);

                }

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

                        int brightness = 0;

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

                                if (((ParamImpl)params.getParam(i)).getId().equals("Brightness") && ((ParamImpl)params.getParam(i)).getDefaultValue() instanceof Integer)

                                        brightness = ((Integer) ((ParamImpl)params.getParam(i)).getDefaultValue()).intValue();

                        }

                        addBrightnessFilter(brightness);

                }

        }

        public void addFilter(Params params) throws FilterTypeException {

                addFilter(ContrastFilter.class, params);

        }

        public RasterFilter createFilter(Params params) {

                if(        params.getParamById("Brightness") != null) {

                        int brightness = ((Integer) params.getParamById("Brightness").getDefaultValue()).intValue();

                        return createBrightnessFilter(brightness);

                }

                if(        params.getParamById("Contrast") != null) {

                        int contrast = ((Integer) params.getParamById("Contrast").getDefaultValue()).intValue();

                        return createContrastFilter(contrast);

                }

                return null;

        }

}