Application: gvSIG desktop

/*

 * Cresques Mapping Suite. Graphic Library for constructing mapping applications.

 * 

 * Copyright (C) 2004-5. 

 *

 * 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.

 *

 * For more information, contact:

 * 

 * cresques@gmail.com

 */

package org.cresques.io;

import java.awt.Dimension;

import java.awt.Graphics2D;

import java.awt.image.BufferedImage;

import java.awt.image.WritableRaster;

import org.cresques.geo.ViewPortData;

import org.cresques.px.Extent;

import org.cresques.px.PxLayerList;

/**

 * Sirve datos solicitados por los drivers que salvan a raster. Implementa

 * IDataWriter que obliga al m?todo readData que es el que ser? llamado

 * desde el driver cada vez que vacie el buffer y necesite m?s datos.

 * @author Nacho Brodin (brodin_ign@gva.es)

 */

public class Rasterizer implements IDataWriter {

        private ViewPortData                 viewPort = null;

        private ViewPortData                 viewPortBlock = null;

        protected BufferedImage                image;

        private PxLayerList                 layerList;

        protected int                                 sizeBlock = 0;

        protected int                                lastBlock = 0;

        protected WritableRaster        raster = null;

        protected int                                 contBlocks = 1;

        protected double                        wcIntervalo = 0;

        protected int[]                         rasterData = null;

        protected Dimension                 dimension = null;

        protected int[]                         buffer; 

        protected double                         wcAlto = 0;

        protected Graphics2D                 g;

        /**

         * Calculo del viewPort de cada bloque que vamos salvando a raster

         * @param vp        viewPort

         */

        private void calcViewPort(ViewPortData vp){

                Extent ext = null;

                if(viewPortBlock == null)//Primer bloque

                        ext = new Extent(        Math.round(vp.getExtent().minX()),

                                                                Math.round(vp.getExtent().maxY()),

                                                                Math.round(vp.getExtent().maxX()),

                                                                Math.round(vp.getExtent().maxY()-wcIntervalo));

                else        //Bloques siguientes

                        ext = new Extent(        Math.round(vp.getExtent().minX()),

                                                                Math.round(vp.getExtent().minY()),

                                                                Math.round(vp.getExtent().maxX()),

                                                                Math.round(vp.getExtent().minY()-wcIntervalo));

                viewPortBlock = new ViewPortData(        vp.getProjection(),

                                                                                        ext,

                                                                                        dimension);

                viewPortBlock.zoom(ext);

        }

        /**

         * Constructor para las clases que hereden de esta

         * y redefinan la funcionalidad que sirve los datos. Estas 

         * tampoco usaran el viewPort ya que se supone que disponen

         * de su propia estructura para la seleccion de la vista

         */

        public Rasterizer(int sizeBlock){

                this.sizeBlock = sizeBlock;

        }

        /**

         * Constructor

         * @param layerList

         * @param vp

         * @param sizeBlock

         */

        public Rasterizer(PxLayerList layerList, ViewPortData vp, int sizeBlock){

                viewPort = (ViewPortData) vp.clone();

                //Calculo del viewPort del primer bloque

                wcAlto = viewPort.getExtent().maxY() - viewPort.getExtent().minY();

                wcIntervalo = (sizeBlock*wcAlto)/viewPort.getHeight();

                dimension = new Dimension((int)viewPort.getWidth(), sizeBlock);

                calcViewPort(viewPort);

                //Tama?o de ?ltimo bloque

                int nBlocks = (int)(vp.getHeight()/sizeBlock); 

                lastBlock = (int)(vp.getHeight()-(nBlocks*sizeBlock));

                this.layerList = layerList;

                this.sizeBlock = sizeBlock;

                //System.out.println("==>Iniciando servidor datos raster "+viewPort.getHeight()+" "+viewPort.getWidth()+" "+wcIntervalo +"..."+viewPortBlock.getExtent().maxX()+" - "+viewPortBlock.getExtent().maxY()+" - "+viewPortBlock.getExtent().minX()+" - "+viewPortBlock.getExtent().minY());

        }

        /* (non-Javadoc)

         * @see org.cresques.io.IDataWriter#readData(int, int, int)

         */

        public int[] readData(int sX, int sY, int nBand){

                if(nBand == 0){ //Con nBand==0 se devuelven las 3 bandas

                        //image = new BufferedImage((int)this.viewPortBlock.getWidth(), (int)this.viewPortBlock.getHeight(), BufferedImage.TYPE_INT_RGB);

                        image = new BufferedImage(sX, sY, BufferedImage.TYPE_INT_RGB);

                        g = (Graphics2D) image.getGraphics();

                        layerList.draw(g, viewPortBlock);        

                        rasterData = image.getRGB(0, 0, sX, sY, rasterData, 0, sX);                                

                        //System.out.println("==>Cargando bloque ..."+(int)this.viewPortBlock.getWidth()+" * "+(int)this.viewPortBlock.getHeight()+" * "+viewPortBlock.getExtent().maxX()+" - "+viewPortBlock.getExtent().maxY()+" - "+viewPortBlock.getExtent().minX()+" - "+viewPortBlock.getExtent().minY());

                        //System.out.println("==>Escribiendo linea ...0 - 0 - "+sX+" - "+sY+" - "+contBlocks);

                        //Calculamos el viewPort del sgte bloque

                        if(((contBlocks+1)*sizeBlock)<=viewPort.getHeight()){

                                dimension = new Dimension(sX, sY);

                        }

                        else{ //Calculo de la altura del ?ltimo bloque

                                dimension = new Dimension(sX, (int)(viewPort.getHeight()-(contBlocks*sizeBlock)));

                                wcIntervalo = (lastBlock*wcAlto)/viewPort.getHeight();

                        }

                        calcViewPort(viewPortBlock);                        

                        contBlocks++;

                        return rasterData;

                }

                return null;

        }

}