Application: gvSIG desktop

/*

 * Created on 22-feb-2005

 */

package org.cresques.io;

import java.awt.Point;

import java.awt.image.DataBuffer;

/**

 * Rectangulo de pixeles. parecido a java.awt.image.Raster

 * @author Luis W. Sevilla (sevilla_lui@gva.es)

 */

public class RasterBuf {

        private byte  [][][] byteBuf;

        private short [][][] shortBuf;

        private int   [][][] intBuf;

        private int width, height, bandNr;

        private int dataType;

        public RasterBuf(int dataType, int width, int height, int bandNr, Point orig) {

                this.dataType = dataType;

                this.width = width;

                this.height = height;

                this.bandNr = bandNr;

                if (dataType == DataBuffer.TYPE_BYTE)

                        byteBuf  = new byte[height][width][bandNr];

                else if (dataType == DataBuffer.TYPE_SHORT | dataType == DataBuffer.TYPE_USHORT)

                        shortBuf = new short[height][width][bandNr];

                else if (dataType == DataBuffer.TYPE_INT)

                        intBuf   = new int[height][width][bandNr];

        }

        public int getWidth() { return width; }

        public int getHeight() { return height; }

        public int getDataType() { return dataType; }

        public int getDataSize() {

                if (dataType == DataBuffer.TYPE_BYTE)

                        return 1;

                else if (dataType == DataBuffer.TYPE_SHORT | dataType == DataBuffer.TYPE_USHORT)

                        return 2;

                else if (dataType == DataBuffer.TYPE_INT)

                        return 4;

                return 0;

        }

        public int sizeof() {

                return getDataSize()*width*height*bandNr;

        }

        public byte [][] getLineByte(int y) {        return byteBuf[y]; }

        public short [][] getLineShort(int y) {        return shortBuf[y]; }

        public int [][] getLineInt(int y) {        return intBuf[y]; }

        public void setPixelShort(int x, int y, int [] px) {

                for (int i=0; i<bandNr; i++) shortBuf[y][x][i] = (short) px[i];

        }

        public void getPixelShort(int x, int y, int [] px) {

                for (int i=0; i<bandNr; i++) px[i] = shortBuf[y][x][i];

        }

        public void setPixelInt(int x, int y, int [] px) {

                for (int i=0; i<bandNr; i++) intBuf[y][x][i] = px[i];

        }

        public void getPixelInt(int x, int y, int [] px) {

                for (int i=0; i<bandNr; i++) px[i] = intBuf[y][x][i];

        }

}