| gvSIG desktop 1

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

 *

 * Copyright (C) 2007 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.raster.dataset.io;

import java.awt.geom.NoninvertibleTransformException;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import org.gvsig.raster.dataset.BandList;

import org.gvsig.raster.dataset.FileNotOpenException;

import org.gvsig.raster.dataset.GeoInfo;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.InvalidSetViewException;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.dataset.io.rmf.ParsingException;

import org.gvsig.raster.datastruct.Extent;

import org.gvsig.raster.datastruct.Transparency;

import org.gvsig.raster.util.extensionPoints.ExtensionPoints;

import org.gvsig.raster.util.extensionPoints.ExtensionPointsSingleton;

import es.gva.cit.jmrsid.MrSIDException;

/**

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

 *

 * Clase encargada del acceso a los datos y repintado de imagenes MrSID. Estos

 * son registrados con la extensi?n sid

 */

public class MrSidDriver extends RasterDataset {

    public final static int BAND_HEIGHT = 64;

    protected MrSidNative file = null;

    private Extent viewRequest = null;

        /**

         * Estado de transparencia del raster.

         */

        protected Transparency                  fileTransparency = null;

    public static void register() {

                ExtensionPoints extensionPoints = ExtensionPointsSingleton.getInstance();

                extensionPoints.add("RasterReader", "sid", MrSidDriver.class);

        }

    /**

     * Contructor. Abre el fichero mrsid

     * @param proj        Proyecci?n

     * @param fName        Nombre del fichero mrsid

     */

    public MrSidDriver(IProjection proj, Object param) {

            super(proj, ((String)param));

            setParam(param);

        extent = new Extent();

        try {

            file = new MrSidNative(((String)param));

            load();

            bandCount = file.nbands;

            setDataType(IBuffer.TYPE_BYTE);

            super.init();

            try {

                             loadFromRmf(getRmfBlocksManager());

                     } catch (ParsingException e) {

                             //No lee desde rmf

                     }        

        } catch (Exception e) {

            System.out.println("Error en constructor de MrSID");

            e.printStackTrace();

            file = null;

        }

    }

        /**

         * Obtenemos o calculamos el extent de la imagen.

         */

    public GeoInfo load() {

                   extent = new Extent(file.esq.minX, file.esq.minY, file.esq.maxX, file.esq.maxY);

                   requestExtent = extent;        

        return this;

    }

    /**

     * Libera el objeto que ha abierto el fichero

     */

    public void close() {

            if(file != null){

                    file.close();

                    file = null;

            }

    }

    /**

     * Asigna el extent de la vista

     */

    public void setView(Extent e) {

            if(rmfExists) {

                        Point2D.Double petInit = null, petEnd = null;

                        try {

                                petInit = new Point2D.Double(e.minX(), e.maxY());

                                petEnd = new Point2D.Double(e.maxX(), e.minY());

                                transformRMF.inverseTransform(petInit, petInit);

                                transformRMF.inverseTransform(petEnd, petEnd);

                                transformTFW.transform(petInit, petInit);

                                transformTFW.transform(petEnd, petEnd);

                        } catch(NoninvertibleTransformException ex) {

                        }

                        viewRequest = new Extent(petInit.getX(), petInit.getY(), petEnd.getX(), petEnd.getY());

                } else

                        viewRequest = new Extent(e);

    }

    /**

     * Obtiene el Extent de la vista

     */

    public Extent getView() {

        return viewRequest;

    }

    /**

     * Obtiene el ancho de la imagen

     */

    public int getWidth() {

        return file.width;

    }

    /**

     * Obtiene el alto de la imagen

     */

    public int getHeight() {

        return file.height;

    }

    public void reProject(ICoordTrans rp) {

        // TODO Auto-generated method stub        

    }

    /**

     * Asigna al objeto Image los valores con los dato de la imagen contenidos en el

     * vector de enteros.

     * @param image        imagen con los datos actuales

     * @param startX        inicio de la posici?n en X dentro de la imagen

     * @param startY        inicio de la posici?n en X dentro de la imagen

     * @param w        Ancho de la imagen

     * @param h        Alto de la imagen

     * @param rgbArray        vector que contiene la banda que se va a sustituir

     * @param offset        desplazamiento

     * @param scansize        tama?o de imagen recorrida por cada p

     */

    protected void setRGBLine(BufferedImage image, int startX, int startY,

                              int w, int h, int[] rgbArray, int offset,

                              int scansize) {

        image.setRGB(startX, startY, w, h, rgbArray, offset, scansize);

    }

    /* (non-Javadoc)

     * @see org.cresques.io.GeoRasterFile#getData(int, int, int)

     */

    public Object getData(int x, int y, int band)throws InvalidSetViewException, FileNotOpenException, RasterDriverException {

            if(file != null){

                        if(x < 0 || y < 0 || x >= file.width || y >= file.height)

                                throw new InvalidSetViewException("Request out of grid");

                        Object[] data = file.getData(x, y);

                        return data[band];

                }

                throw new FileNotOpenException("GdalNative not exist");

    }

    /**

     * Devuelve el tama?o de bloque

     * @return Tama?o de bloque

     */

    public int getBlockSize() {

        return file.blocksize;

    }

         /**

         * Calcula la transformaci?n que se produce sobre la vista cuando la imagen tiene un fichero .rmf

         * asociado. En Gdal el origen de coordenadas en Y es el valor m?nimo y crece hasta el m?ximo. De la

         * misma forma calcula la matriz de transformaci?n de la cabecera del fichero o del world file asociado

         * @param originX Origen de la imagen en la coordenada X

         * @param originY Origen de la imagen en la coordenada Y

         */

        public void setExtentTransform(double originX, double originY, double psX, double psY) {                

                transformRMF.setToTranslation(originX, originY);

                transformRMF.scale(psX, psY);

                transformTFW.setToTranslation(file.esq.minX, file.esq.maxY);

                double pixSixeX = (file.esq.maxX - file.esq.minX) / file.width;

                double pixSixeY = (file.esq.minY - file.esq.maxY) / file.height;

                transformTFW.scale(pixSixeX, pixSixeY);

        }

    /**

         * Informa de si el driver ha supersampleado en el ?ltimo dibujado. Es el driver el que colocar?

         * el valor de esta variable cada vez que dibuja. 

         * @return true si se ha supersampleado y false si no se ha hecho.

         */

        public boolean isSupersampling() {

                return file.isSupersampling;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.RasterDataset#rasterToWorld(java.awt.geom.Point2D)

         */

        public Point2D rasterToWorld(Point2D pt) {

                return file.rasterToWorld(pt);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.RasterDataset#worldToRaster(java.awt.geom.Point2D)

         */

        public Point2D worldToRaster(Point2D pt) {

                return file.worldToRaster(pt);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.RasterDataset#getWindowRaster(double, double, double, double, org.gvsig.raster.dataset.BandList, org.gvsig.raster.dataset.IBuffer, boolean)

         */

        public IBuffer getWindowRaster(double x, double y, double w, double h, BandList bandList, IBuffer rasterBuf, boolean adjustToExtent) {

                int width = rasterBuf.getWidth();

                int height = rasterBuf.getHeight();

                //Impedimos que los valores de ancho y alto de la im?gen sean menores que 1

        if (width <= 0) 

                width = 1;

        if (height <= 0) 

                height = 1;

        setView(new Extent(x, y, x + w, y - h));

                file.setView(viewRequest.minX(), viewRequest.maxY(), viewRequest.maxX(), viewRequest.minY(), width, height);

                int[] pRGBArray = new int[width * height];

        try {

                file.readScene(pRGBArray);

                int wBuf = rasterBuf.getWidth();

                for (int row = 0; row < rasterBuf.getHeight(); row++) {

                    for(int col = 0; col < wBuf; col ++) {

                            rasterBuf.setElem(row, col, 0, (byte)((pRGBArray[(row * wBuf) + col] & 0x00ff0000) >> 16));

                            rasterBuf.setElem(row, col, 1, (byte)((pRGBArray[(row * wBuf) + col] & 0x0000ff00) >> 8));

                            rasterBuf.setElem(row, col, 2, (byte)(pRGBArray[(row * wBuf) + col] & 0x000000ff));

                    }

                }

        } catch (Exception e) {

                e.printStackTrace();

        }

        return rasterBuf;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.RasterDataset#getWindowRaster(double, double, double, double, int, int, org.gvsig.raster.dataset.BandList, org.gvsig.raster.dataset.IBuffer, boolean)

         */

        public IBuffer getWindowRaster(double minX, double minY, double maxX, double maxY, int bufWidth, int bufHeight, BandList bandList, IBuffer rasterBuf, boolean adjustToExtent) {

                //Impedimos que los valores de ancho y alto de la im?gen sean menores que 1

        if (bufWidth <= 0) 

                bufWidth = 1;

        if (bufHeight <= 0) 

                bufHeight = 1;

        setView(new Extent(minX, maxY, maxX, minY));

                file.setView(viewRequest.minX(), viewRequest.maxY(), viewRequest.maxX(), viewRequest.minY(), bufWidth, bufHeight);

        int[] pRGBArray = new int[bufWidth * bufHeight];

        try {

                file.readScene(pRGBArray);

                int w = rasterBuf.getWidth();

                for (int row = 0; row < rasterBuf.getHeight(); row++) {

                    for(int col = 0; col < w; col ++) {

                            rasterBuf.setElem(row, col, 0, (byte)((pRGBArray[(row * w) + col] & 0x00ff0000) >> 16));

                            rasterBuf.setElem(row, col, 1, (byte)((pRGBArray[(row * w) + col] & 0x0000ff00) >> 8));

                            rasterBuf.setElem(row, col, 2, (byte)(pRGBArray[(row * w) + col] & 0x000000ff));

                    }

                }

        } catch (Exception e) {

                e.printStackTrace();

        }

        return rasterBuf;

        }

        public IBuffer getWindowRaster(int x, int y, int w, int h, int bufWidth, int bufHeight, BandList bandList, IBuffer rasterBuf) {

                //TODO: FUNCIONALIDAD: Acabar getWindowRaster MrSID

                return null;

        }

        public Object readBlock(int pos, int blockHeight) throws InvalidSetViewException, FileNotOpenException, RasterDriverException {

                if(pos < 0)

                        throw new InvalidSetViewException("Request out of grid");

                if((pos + blockHeight) > file.height)

                        blockHeight = Math.abs(file.height - pos);

                Point2D begin = rasterToWorld(new Point2D.Double(0, pos));

                Point2D end = rasterToWorld(new Point2D.Double(file.width, pos + blockHeight));

                //TODO: FUNCIONALIDAD: Acabar lectura de bloque MrSID

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.RasterDataset#readCompleteLine(int, int)

         */

        public Object readCompleteLine(int line, int band) throws InvalidSetViewException, FileNotOpenException, RasterDriverException {

                if(line > this.getHeight() || band > this.getBandCount())

                        throw new InvalidSetViewException("Request out of grid");

                try{

                        Point2D pt = rasterToWorld(new Point2D.Double(extent.minX(), line));

                        file.setView(extent.minX(), pt.getY(), extent.maxX(), pt.getY(), getWidth(), 1);

                        int[] pRGBArray = new int[getWidth()];

                        file.readScene(pRGBArray);

                        return pRGBArray;

                }catch(MrSIDException e){

                        throw new RasterDriverException("Error reading data from MrSID library");

                }

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.dataset.RasterDataset#getWindowRaster(int, int, int, int, org.gvsig.raster.dataset.BandList, org.gvsig.raster.dataset.IBuffer)

         */

        public IBuffer getWindowRaster(int x, int y, int w, int h, BandList bandList, IBuffer rasterBuf) {

                try {

                        file.readWindow(rasterBuf, bandList, x, y, w, h);

                } catch (Exception e) {

                        e.printStackTrace();

                        return null;

                }

                return rasterBuf;

        }

    /**

         * Obtiene el objeto que contiene el estado de la transparencia

         */

        public Transparency getTransparencyDatasetStatus() {

                if(fileTransparency == null)

                        fileTransparency = new Transparency();

                return fileTransparency;

        }

}