| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2006 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.rastertools.cutting;

import org.gvsig.gui.beans.incrementabletask.IIncrementable;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.IDataWriter;

import org.gvsig.raster.hierarchy.ICancellable;

import com.iver.andami.PluginServices;

/**

 * <code>WriterBufferServer</code> sirve los datos desde un IBuffer para el

 * driver de escritura. El driver de escritura que va a volcar un

 * <code>IBuffer</code> a imagen en disco va solicitando los datos por bloques

 * a medida que va salvando los anteriores. Esta clase es la encargada de hacer

 * la divisi?n en bloques del buffer a volcar e ir sirviendolos a medida que el

 * driver pide m?s datos.

 * </P>

 * <P>

 * Implementa el interfaz <code>IDataWriter</code> que es el que define

 * m?todos necesarios para el driver. Adem?s es una tarea incrementable por lo

 * que deber? implementar el interfaz <code>IIncrementable</code> para poder

 * mostrar el dialogo de incremento de tarea.

 * </P>

 *

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

 * @author BorSanZa - Borja S?nchez Zamorano (borja.sanchez@iver.es)

 * @version 27/04/2007

 */

public class WriterBufferServer implements IDataWriter, IIncrementable, ICancellable {

        private IBuffer                        buffer = null;

        private int                         row = 0;

        /**

         * Tama?o de bloque. Este se asignar? en la primera petici?n

         */

        private int                                block = 0;

        /**

         * N?mero de bloques del proceso

         */

        private int                         nblocks = 0;

        private double                        percent = 0;

        private double                         increment = 0;

        private boolean                        canceled = false;

        private int                                nBand = 0;

        private boolean                        initProcess = false;

        private StringBuffer        logs = new StringBuffer();

        /**

         * Crea un nuevo <code>WriterBufferServer</code>

         */

        public WriterBufferServer() {

        }

        /**

         * Crea un nuevo <code>WriterBufferServer</code> con el buffer de datos.

         * @param buffer

         */

        public WriterBufferServer(IBuffer buffer) {

                setBuffer(buffer, -1);

        }

        /**

         * Asigna el buffer de datos e inicializa variables de

         * @param buffer

         * @param nband Si es menor que cero sirve datos de todas las bandas. Si es

         * mayor que cero sirve datos de la banda indicada por el valor.

         */

        public void setBuffer(IBuffer buffer, int nband) {

                this.buffer = buffer;

                nBand = nband;

                row = 0;

                block = 0;

                percent = 0;

                increment = 0;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.IDataWriter#readARGBData(int, int, int)

         */

        public int[] readARGBData(int sizeX, int sizeY, int nBand) {

                return null;

        }

        /**

         * Acciones de inicializaci?n para la lectura de de un bloque

         * @param sizeY Tama?o en Y del bloque

         */

        private void initRead(int sizeY) {

                //Si es la primera linea se asigna el tama?o de bloque y el incremento

                if (row == 0) {

                        block = sizeY;

                        nblocks = (int)Math.ceil(((double) buffer.getHeight() / (double) sizeY));

                        increment = 100D / ((double) buffer.getHeight() / (double) sizeY);

                }

                //Linea para el log

                int b = 0;

                if (block != 0)

                        b = (int) (row / block);

                logs.append(PluginServices.getText(this, "generando_bloque") + "..." + b + "/" + nblocks + "\n");

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.IDataWriter#readByteData(int, int)

         */

        public byte[][] readByteData(int sizeX, int sizeY) {

                initRead(sizeY);

                percent += increment;

                int len = buffer.getWidth() * sizeY;

                int nbands = (nBand < 0) ? buffer.getBandCount() : 1;

                byte[][] b = new byte[nbands][len];

                if(nBand < 0) {

                        for (int iBand = 0; iBand < nbands; iBand++)

                                for (int j = row; j < (row + sizeY); j++)

                                        for (int i = 0; i < buffer.getWidth(); i++)

                                                b[iBand][(j % block) * buffer.getWidth() + i] = buffer.getElemByte(j, i, iBand);

                } else {

                        for (int j = row; j < (row + sizeY); j++)

                                for (int i = 0; i < buffer.getWidth(); i++)

                                        b[0][(j % block) * buffer.getWidth() + i] = buffer.getElemByte(j, i, nBand);

                }

                row += sizeY;

                return b;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.IDataWriter#readShortData(int, int)

         */

        public short[][] readShortData(int sizeX, int sizeY) {

                initRead(sizeY);

                percent += increment;

                int len = buffer.getWidth() * sizeY;

                short[][] b = new short[buffer.getBandCount()][len];

                for (int iBand = 0; iBand < buffer.getBandCount(); iBand++)

                        for (int j = row; j < (row + sizeY); j++)

                                for (int i = 0; i < buffer.getWidth(); i++)

                                        b[iBand][(j % block) * buffer.getWidth() + i] = buffer.getElemShort(j, i, iBand);

                row += sizeY;

                return b;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.IDataWriter#readIntData(int, int)

         */

        public int[][] readIntData(int sizeX, int sizeY) {

                initRead(sizeY);

                percent += increment;

                int len = buffer.getWidth() * sizeY;

                int[][] b = new int[buffer.getBandCount()][len];

                for (int iBand = 0; iBand < buffer.getBandCount(); iBand++)

                        for (int j = row; j < (row + sizeY); j++)

                                for (int i = 0; i < buffer.getWidth(); i++)

                                        b[iBand][(j % block) * buffer.getWidth() + i] = buffer.getElemInt(j, i, iBand);

                row += sizeY;

                return b;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.IDataWriter#readFloatData(int, int)

         */

        public float[][] readFloatData(int sizeX, int sizeY) {

                initRead(sizeY);

                percent += increment;

                int len = buffer.getWidth() * sizeY;

                float[][] b = new float[buffer.getBandCount()][len];

                for (int iBand = 0; iBand < buffer.getBandCount(); iBand++)

                        for (int j = row; j < (row + sizeY); j++)

                                for (int i = 0; i < buffer.getWidth(); i++)

                                        b[iBand][(j % block) * buffer.getWidth() + i] = buffer.getElemFloat(j, i, iBand);

                row += sizeY;

                return b;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.IDataWriter#readDoubleData(int, int)

         */

        public double[][] readDoubleData(int sizeX, int sizeY) {

                initRead(sizeY);

                percent += increment;

                int len = buffer.getWidth() * sizeY;

                double[][] b = new double[buffer.getBandCount()][len];

                for (int iBand = 0; iBand < buffer.getBandCount(); iBand++)

                        for (int j = row; j < (row + sizeY); j++)

                                for (int i = 0; i < buffer.getWidth(); i++)

                                        b[iBand][(j % block) * buffer.getWidth() + i] = buffer.getElemDouble(j, i, iBand);

                row += sizeY;

                return b;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getLabel()

         */

        public String getLabel() {

                return PluginServices.getText(this, "generando_recorte") + "...";

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getLog()

         */

        public String getLog() {

                if(!initProcess) {

                        initProcess = true;

                        logs.append(PluginServices.getText(this, "leyendo_raster") + "\n");

                }

                return logs.toString();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getPercent()

         */

        public int getPercent() {

                return Math.min((int)percent, 100);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getTitle()

         */

        public String getTitle() {

                return PluginServices.getText(this, "incremento_recorte");

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.util.ICancellable#isCanceled()

         */

        public boolean isCanceled(int process) {

                return canceled;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.util.ICancellable#setCanceled(boolean)

         */

        public void setCanceled(boolean value, int process) {

                logs.append(PluginServices.getText(this, "cancelando") + "...");

                canceled = value;

        }

}