Application: gvSIG desktop » gvSIG raster

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

import java.util.Arrays;

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

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

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

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

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

import org.gvsig.raster.lib.buffer.api.Band.BandShort;

import org.gvsig.raster.lib.buffer.impl.exceptions.CopyFromBandException;

/**

 * @author fdiaz

 *

 */

public class MemoryBandShort extends AbstractBand implements BandShort {

    short[] data;

    /**

     * @param rows

     * @param columns

     */

    public MemoryBandShort(int rows, int columns) {

        this(rows, columns, BufferLocator.getBufferManager().createNoData(null, null));

    }

    /**

     * @param rows

     * @param columns

     * @param noData

     */

    public MemoryBandShort(int rows, int columns, NoData noData) {

        this.rows = rows;

        this.columns = columns;

        data = new short[rows * columns];

        if (noData == null) {

            this.noData = BufferLocator.getBufferManager().createNoData(null, null);

        } else {

            this.noData = noData;

        }

    }

    @Override

    public int getDataType() {

        return BufferManager.TYPE_SHORT;

    }

    @Override

    public Object get(int row, int column) {

        return getValue(row, column);

    }

    @Override

    protected void doSet(int row, int column, Object value) {

        setValue(row, column, (short) value);

    }

    @Override

    protected void doFill(Object value) {

        value = nullValueToNoData(value);

        if(value == null){

            value = 0;

        }

        Arrays.fill(data, (short) value);

    }

    short[] getData() {

        return data;

    }

    @Override

    protected void doCopyFrom(Band source) throws CopyFromBandException {

        if (this.getColumns() == source.getColumns() && this.getRows() == source.getRows()){

            if(source instanceof MemoryBandShort) {

                System.arraycopy(((MemoryBandShort) source).getData(), 0, this.data, 0, this.getRows()

                    * this.getColumns());

            } else {

                Object rowBuffer = source.createRowBuffer();

                for(int row=0; row<source.getRows(); row++){

                    source.fetchRow(row, rowBuffer);;

                    try {

                        System.arraycopy(rowBuffer, 0, this.data, row*this.getColumns(), this.getColumns());

                    } catch (Exception e){

                        throw new CopyFromBandException(source, this, e);

                    }

                }

            }

        } else {

            throw new CopyFromBandException(source, this);

        }

    }

    @Override

    protected void doCopyFrom(Band source, int row, int column) throws CopyFromBandException {

        if (this.getColumns() >= source.getColumns() + column && this.getRows() >= source.getRows() + row

            && source instanceof MemoryBandShort) {

            for(int r=0; r<Math.min(this.getRows()-row, source.getRows()); r++){

                System.arraycopy(((MemoryBandShort) source).getData(), r*source.getColumns(), this.data, (r+row)*this.getColumns()+column, Math.min(this.getColumns()-column, source.getColumns()));

            }

        } else {

            throw new CopyFromBandException(source, this);

        }

    }

    @Override

    public short getValue(int row, int column) {

        return data[row * getColumns() + column];

    }

    @Override

    public void setValue(int row, int column, short value) {

        doSetValue(row, column, value);

        notifyObservers(new DefaultBandNotification(BandNotification.SET, new Object[]{row, column, value}));

    }

    protected void doSetValue(int row, int column, short value) {

        data[row * getColumns() + column] = value;

    }

    @Override

    public short[] createRowBuffer() {

        return new short[getColumns()];

    }

    @Override

    public void fetchRow(int row, Object rowBuffer) {

        System.arraycopy(data, row * getColumns(), (short[]) rowBuffer, 0, getColumns());

    }

    @Override

    protected void doPutRow(int row, Object rowBuffer) {

        System.arraycopy((short[]) rowBuffer, 0, data, row * getColumns(), getColumns());

    }

}