Application: gvSIG desktop » gvSIG raster

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2016 gvSIG Association

 *

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

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.dal.file.jimi;

import java.awt.image.ColorModel;

import java.awt.image.DirectColorModel;

import java.awt.image.IndexColorModel;

import java.nio.ByteBuffer;

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

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

import com.sun.jimi.core.ImageAccessException;

import com.sun.jimi.core.raster.ByteRasterImage;

import com.sun.jimi.core.raster.IntRasterImage;

import com.sun.jimi.core.raster.JimiRasterImage;

/**

 * Represents an image loaded in memory

 * @author dmartinezizquierdo

 *

 */

public class MemoryImage{

    public int rows;

    public int columns;

    public int bands;

    public byte buffer[][];

    public int dataType;

    private JimiRasterImage jimiRasterImage;

    public boolean loaded;

    public ColorModel colorModel;

    public MemoryImage(JimiRasterImage jimiRasterImage) throws ImageAccessException {

        this.jimiRasterImage=jimiRasterImage;

        rows=jimiRasterImage.getHeight();

        columns=jimiRasterImage.getWidth();

        colorModel = jimiRasterImage.getColorModel();

        if (colorModel instanceof DirectColorModel){

//            if( jimiRasterImage instanceof IntRasterImage ) {

//                dataType = BufferManager.TYPE_INT;

//            } else if( jimiRasterImage instanceof ByteRasterImage ) {

                dataType = BufferManager.TYPE_BYTE;

//            } else {

//                dataType = BufferManager.TYPE_INT;

//            }

            bands=3;

            if ( colorModel.hasAlpha() ){

                bands=4;

            }

            buffer = new byte[bands][columns*rows];

            for( int band=0; band<bands; band++) {

                switch (band) {

                case 0:

                    jimiRasterImage.getChannelRectangle(JimiRasterImage.CHANNEL_RED, 0, 0, columns, rows, buffer[band], 0, columns);

                    break;

                case 1:

                    jimiRasterImage.getChannelRectangle(JimiRasterImage.CHANNEL_GREEN, 0, 0, columns, rows, buffer[band], 0, columns);

                    break;

                case 2:

                    jimiRasterImage.getChannelRectangle(JimiRasterImage.CHANNEL_BLUE, 0, 0, columns, rows, buffer[band], 0, columns);

                    break;

                case 3:

                    jimiRasterImage.getChannelRectangle(JimiRasterImage.CHANNEL_ALPHA, 0, 0, columns, rows, buffer[band], 0, columns);

                    break;

                default:

                    break;

                }

            }

        }else if (colorModel instanceof IndexColorModel){

            bands=1;

            BufferManager bufferManager = BufferLocator.getBufferManager();

            int dataTypeSize = bufferManager.getDataTypeSize(dataType);

            buffer = new byte[bands][columns*rows* dataTypeSize];

            if( jimiRasterImage instanceof ByteRasterImage ) {

                dataType = BufferManager.TYPE_BYTE;

                ByteRasterImage byteRasterImage = ( ByteRasterImage )jimiRasterImage;

                byteRasterImage.getRectangle(0, 0, columns, rows, buffer[0], 0, columns);

            } else {

                dataType = BufferManager.TYPE_INT;

                IntRasterImage intRasterImage = ( IntRasterImage )jimiRasterImage;

                int[] intAuxArray=new int[columns*rows];

                intRasterImage.getRectangle(0, 0, columns, rows, intAuxArray, 0, columns);

                ByteBuffer byteBuffer=ByteBuffer.wrap(buffer[0]);

                for(int i=0;i<=intAuxArray.length;i=i){

                    byteBuffer.putInt(intAuxArray[i]);

                }

            }

        }

    }

    public byte getValue(int row, int column, int band) {

        return buffer[band][row*columns + column];

    }

}