Application: gvSIG desktop

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

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.apache.commons.io.IOUtils;

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;

import org.gvsig.tools.dispose.impl.AbstractDisposable;

import org.gvsig.tools.exception.BaseException;

public class BSQBandPageManager extends AbstractDisposable implements BandPageManager {

    private RandomAccessFile  raf;

    private final int rows;

    private final int columns;

    private final int dataType;

    private final File file;

    private final long startpos;

    /**

     * @param file

     * @param startpos

     * @param rows

     * @param columns

     * @param dataType

     * @throws IOException

     */

    public BSQBandPageManager(File file, long startpos, int rows, int columns, int dataType) throws IOException {

        this.rows = rows;

        this.columns = columns;

        this.dataType = dataType;

        this.file = file;

        this.startpos = startpos;

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

        // BufferManager bufferManager = BufferLocator.getBufferManager();

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

    }

    @Override

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

        long pos = row*this.columns + this.startpos;

        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.");

        }

    }

    @Override

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

        long pos = row*this.columns + this.startpos;

        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 false;

    }

    @Override

    public BandInfo getBandInfo() {

        return null;

    }

    @Override

    protected void doDispose() throws BaseException {

        IOUtils.closeQuietly(this.raf);

        this.raf = null;

    }

}