Application: gvSIG desktop

/* 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.lib.buffer.impl;

import org.gvsig.raster.lib.buffer.api.Kernel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.Persistent;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

/**

 * Clase que representa un kernel de NxN p?xeles para realizar operaciones sobre

 * un pixel.

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

 * @author fdiaz

 */

public class DefaultKernel implements Persistent, Kernel {

        private double[][] kernel        = null;

        private double        divisor = 0;

        /**

         * This constructor is useful only for persistence

         */

        public DefaultKernel() {

        }

        /**

         * Constructor. Crea la matriz de datos para el kernel.

         * @param k datos del kernel

         */

        public DefaultKernel(double[][] k) {

                this.kernel = k;

                for (int i = 0; i < kernel.length; i++)

                        for (int j = 0; j < kernel[0].length; j++)

                                divisor = divisor + kernel[i][j];

        }

        /**

         * Constructor. Crea la matriz de datos para el kernel.

         * @param k datos del kernel

         */

        public DefaultKernel(double[][] k, double divisor) {

                this.kernel = k;

                this.divisor = divisor;

        }

        @Override

        public double[][] getKernel() {

            return this.kernel;

        }

        /* (non-Javadoc)

         * @see org.gvsig.raster.lib.buffer.impl.operations.Kernel#getDivisor()

         */

        @Override

        public double getDivisor() {

            return divisor;

        }

        /* (non-Javadoc)

         * @see org.gvsig.raster.lib.buffer.impl.operations.Kernel#setDivisor(double)

         */

        @Override

        public void setDivisor(double divisor) {

            this.divisor = divisor;

        }

        /* (non-Javadoc)

         * @see org.gvsig.raster.lib.buffer.impl.operations.Kernel#getLado()

         */

        @Override

        public int getSide() {

            return kernel.length;

        }

        /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.impl.operations.Kernel#kernelOperation(org.gvsig.raster.lib.buffer.impl.operations.DefaultKernel)

     */

        @Override

    public double kernelOperation(Kernel k) {

                double res = 0;

                for (int i = 0; i < kernel.length; i++)

                        for (int j = 0; j < kernel[0].length; j++)

                                res += kernel[i][j] * k.getKernel()[i][j];

                return res;

        }

        /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.impl.operations.Kernel#convolution(org.gvsig.raster.lib.buffer.impl.operations.DefaultKernel)

     */

        @Override

    public double convolution(Kernel k) {

                double res = this.kernelOperation(k);

                if (this.divisor != 0)

                        res = res / divisor;

                return res;

        }

        /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.impl.operations.Kernel#rgbNormalization()

     */

        @Override

    public void rgbNormalization() {

                for (int i = 0; i < kernel.length; i++)

                        for (int j = 0; j < kernel[0].length; j++)

                                kernel[i][j] = ((byte)kernel[i][j]) & 0xff;

        }

        public void loadFromState(PersistentState state)

                        throws PersistenceException {

                int side = state.getInt("side");

                kernel = new double[side][side];

                if(side >= 3) {

                        kernel[0] = (double[])state.getDoubleArray("row_0");

                        kernel[1] = (double[])state.getDoubleArray("row_1");

                        kernel[2] = (double[])state.getDoubleArray("row_2");

                }

                if(side >= 5) {

                        kernel[3] = (double[])state.getDoubleArray("row_3");

                        kernel[4] = (double[])state.getDoubleArray("row_4");

                }

                if(side >= 7) {

                        kernel[5] = (double[])state.getDoubleArray("row_5");

                        kernel[6] = (double[])state.getDoubleArray("row_6");

                }

        }

        public void saveToState(PersistentState state) throws PersistenceException {

                if(kernel == null)

                        return;

                state.set("row_0", kernel[0]);

                state.set("row_1", kernel[1]);

                state.set("row_2", kernel[2]);

                if(kernel.length == 3) {

                        state.set("side", 3);

                        return;

                }

                state.set("row_3", kernel[3]);

                state.set("row_4", kernel[4]);

                if(kernel.length == 5) {

                        state.set("side", 5);

                        return;

                }

                state.set("row_5", kernel[5]);

                state.set("row_6", kernel[6]);

                if(kernel.length == 7) {

                        state.set("side", 7);

                        return;

                }

        }

        public static void registerPersistence() {

                PersistenceManager manager = ToolsLocator.getPersistenceManager();

                DynStruct definition = manager.getDefinition("Kernel_Persistent");

                if( definition == null ) {

                        definition = manager.addDefinition(

                                        DefaultKernel.class,

                                        "Kernel_Persistent",

                                        "Kernel Persistency",

                                        null,

                                        null

                        );

                }

                definition.addDynFieldInt("side").setMandatory(false);

                definition.addDynFieldList("row_0").setClassOfItems(double.class).setMandatory(false);

                definition.addDynFieldList("row_1").setClassOfItems(double.class).setMandatory(false);

                definition.addDynFieldList("row_2").setClassOfItems(double.class).setMandatory(false);

                definition.addDynFieldList("row_3").setClassOfItems(double.class).setMandatory(false);

                definition.addDynFieldList("row_4").setClassOfItems(double.class).setMandatory(false);

                definition.addDynFieldList("row_5").setClassOfItems(double.class).setMandatory(false);

                definition.addDynFieldList("row_6").setClassOfItems(double.class).setMandatory(false);

        }

}