Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.lib.buffer.impl;

import java.io.File;

import java.io.IOException;

import java.io.RandomAccessFile;

import java.nio.Buffer;

import java.nio.ByteBuffer;

import java.nio.DoubleBuffer;

import java.nio.FloatBuffer;

import java.nio.IntBuffer;

import java.nio.ShortBuffer;

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

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

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

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

/**

 * @author fdiaz

 *

 */

public class SimpleBandPageManager implements BandPageManager {

    private RandomAccessFile  raf;

    private int rows;

    private int columns;

    private int dataType;

    private File file;

    /**

     * @param file

     * @param rows

     * @param columns

     * @param dataType

     * @throws IOException

     */

    public SimpleBandPageManager(File file, int rows, int columns, int dataType) throws IOException {

        this.rows = rows;

        this.columns = columns;

        this.dataType = dataType;

        this.file = file;

        if(this.file==null){

            this.file = File.createTempFile("bandPageManager", "tmp");

        }

        this.raf = new RandomAccessFile(this.file, "rw");

        BufferManager bufferManager = BufferLocator.getBufferManager();

        raf.setLength(rows*columns*bufferManager.getDataTypeSize(dataType));

    }

    public void load(Buffer buffer, int row, int rows, int dataType) throws IOException {

        int pos = row*this.columns;

        raf.seek(pos);

        BufferManager bufferManager = BufferLocator.getBufferManager();

        int dataTypeSize = bufferManager.getDataTypeSize(dataType);

        byte[] byteArray = new byte[buffer.capacity() * dataTypeSize];

        raf.read(byteArray);

        buffer.rewind();

        switch (dataType) {

        case BufferManager.TYPE_BYTE:

            //FIXME: Elegir cual de las dos siguientes maneras de hacerlo es la id?nea

            ((ByteBuffer) buffer).put(byteArray, 0, byteArray.length);

//            System.arraycopy(byteArray, 0, ((ByteBuffer) buffer).array(), 0, byteArray.length);

            break;

        case BufferManager.TYPE_SHORT:

        case BufferManager.TYPE_USHORT:

            for( int cell =0 ; cell<byteArray.length ; cell++ ) {

                ((ShortBuffer) buffer).put((int)(cell/dataTypeSize), byteArray[cell]);

            }

            break;

        case BufferManager.TYPE_INT:

            for( int cell =0 ; cell<byteArray.length ; cell++ ) {

                ((IntBuffer) buffer).put((int)(cell/dataTypeSize), byteArray[cell]);

            }

            break;

        case BufferManager.TYPE_FLOAT:

            for( int cell =0 ; cell<byteArray.length ; cell++ ) {

                ((FloatBuffer) buffer).put((int)(cell/dataTypeSize), byteArray[cell]);

            }

            break;

        case BufferManager.TYPE_DOUBLE:

            for( int cell =0 ; cell<byteArray.length ; cell++ ) {

                ((DoubleBuffer) buffer).put((int)(cell/dataTypeSize), byteArray[cell]);

            }

            break;

        default:

            throw new IllegalArgumentException("Data Type " + dataType + " not supported.");

        }

    }

    public void save(Buffer buffer, int row, int rows, int dataType) throws IOException {

        int pos = row*this.columns;

        raf.seek(pos);

        BufferManager bufferManager = BufferLocator.getBufferManager();

        ByteBuffer byteBuffer = ByteBuffer.allocate(buffer.capacity() * bufferManager.getDataTypeSize(dataType));

        switch (dataType) {

        case BufferManager.TYPE_BYTE:

            byteBuffer.put(((ByteBuffer)buffer).array());

            break;

        case BufferManager.TYPE_SHORT:

        case BufferManager.TYPE_USHORT:

            ShortBuffer byteAsShortBuffer = byteBuffer.asShortBuffer();

            byteAsShortBuffer.put((short[]) buffer.array());

            break;

        case BufferManager.TYPE_INT:

            IntBuffer byteAsIntBuffer = byteBuffer.asIntBuffer();

            byteAsIntBuffer.put((int[]) buffer.array());

            break;

        case BufferManager.TYPE_FLOAT:

            FloatBuffer byteAsFloatBuffer = byteBuffer.asFloatBuffer();

            byteAsFloatBuffer.put((float[]) buffer.array());

            break;

        case BufferManager.TYPE_DOUBLE:

            DoubleBuffer byteAsDoubleBuffer = byteBuffer.asDoubleBuffer();

            byteAsDoubleBuffer.put((double[]) buffer.array());

            break;

        default:

            throw new IllegalArgumentException("Data Type " + dataType + " not supported.");

        }

        raf.write(byteBuffer.array());

    }

    @Override

    public boolean isSupportedSave() {

        return true;

    }

    @Override

    public BandInfo getBandInfo() {

        return null;

    }

}