/* * Copyright (c) 2003 Sun Microsystems, Inc. All Rights Reserved. * Copyright (c) 2010 JogAmp Community. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are * met: * * - Redistribution of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * - Redistribution in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * Neither the name of Sun Microsystems, Inc. or the names of * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * * This software is provided "AS IS," without a warranty of any kind. ALL * EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, * INCLUDING ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A * PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE HEREBY EXCLUDED. SUN * MICROSYSTEMS, INC. ("SUN") AND ITS LICENSORS SHALL NOT BE LIABLE FOR * ANY DAMAGES SUFFERED BY LICENSEE AS A RESULT OF USING, MODIFYING OR * DISTRIBUTING THIS SOFTWARE OR ITS DERIVATIVES. IN NO EVENT WILL SUN OR * ITS LICENSORS BE LIABLE FOR ANY LOST REVENUE, PROFIT OR DATA, OR FOR * DIRECT, INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL OR PUNITIVE * DAMAGES, HOWEVER CAUSED AND REGARDLESS OF THE THEORY OF LIABILITY, * ARISING OUT OF THE USE OF OR INABILITY TO USE THIS SOFTWARE, EVEN IF * SUN HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES. * * You acknowledge that this software is not designed or intended for use * in the design, construction, operation or maintenance of any nuclear * facility. * * Sun gratefully acknowledges that this software was originally authored * and developed by Kenneth Bradley Russell and Christopher John Kline. */ package com.jogamp.common.nio; import java.nio.*; import com.jogamp.common.util.ValueConv; /** * Utility methods allowing easy {@link java.nio.Buffer} manipulations. * * @author Kenneth Russel * @author Sven Gothel * @author Michael Bien */ public class Buffers { public static final int SIZEOF_BYTE = 1; public static final int SIZEOF_SHORT = 2; public static final int SIZEOF_CHAR = 2; public static final int SIZEOF_INT = 4; public static final int SIZEOF_FLOAT = 4; public static final int SIZEOF_LONG = 8; public static final int SIZEOF_DOUBLE = 8; protected Buffers() {} /** * Allocates a new direct ByteBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static ByteBuffer newDirectByteBuffer(int numElements) { return nativeOrder(ByteBuffer.allocateDirect(numElements)); } public static ByteBuffer newDirectByteBuffer(byte[] values, int offset, int lenght) { return (ByteBuffer)newDirectByteBuffer(lenght).put(values, offset, lenght).rewind(); } public static ByteBuffer newDirectByteBuffer(byte[] values, int offset) { return newDirectByteBuffer(values, offset, values.length-offset); } public static ByteBuffer newDirectByteBuffer(byte[] values) { return newDirectByteBuffer(values, 0); } /** * Allocates a new direct DoubleBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static DoubleBuffer newDirectDoubleBuffer(int numElements) { return newDirectByteBuffer(numElements * SIZEOF_DOUBLE).asDoubleBuffer(); } public static DoubleBuffer newDirectDoubleBuffer(double[] values, int offset, int lenght) { return (DoubleBuffer)newDirectDoubleBuffer(lenght).put(values, offset, lenght).rewind(); } public static DoubleBuffer newDirectDoubleBuffer(double[] values, int offset) { return newDirectDoubleBuffer(values, offset, values.length - offset); } public static DoubleBuffer newDirectDoubleBuffer(double[] values) { return newDirectDoubleBuffer(values, 0); } /** * Allocates a new direct FloatBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static FloatBuffer newDirectFloatBuffer(int numElements) { return newDirectByteBuffer(numElements * SIZEOF_FLOAT).asFloatBuffer(); } public static FloatBuffer newDirectFloatBuffer(float[] values, int offset, int lenght) { return (FloatBuffer)newDirectFloatBuffer(lenght).put(values, offset, lenght).rewind(); } public static FloatBuffer newDirectFloatBuffer(float[] values, int offset) { return newDirectFloatBuffer(values, offset, values.length - offset); } public static FloatBuffer newDirectFloatBuffer(float[] values) { return newDirectFloatBuffer(values, 0); } /** * Allocates a new direct IntBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static IntBuffer newDirectIntBuffer(int numElements) { return newDirectByteBuffer(numElements * SIZEOF_INT).asIntBuffer(); } public static IntBuffer newDirectIntBuffer(int[] values, int offset, int lenght) { return (IntBuffer)newDirectIntBuffer(lenght).put(values, offset, lenght).rewind(); } public static IntBuffer newDirectIntBuffer(int[] values, int offset) { return newDirectIntBuffer(values, offset, values.length - offset); } public static IntBuffer newDirectIntBuffer(int[] values) { return newDirectIntBuffer(values, 0); } /** * Allocates a new direct LongBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static LongBuffer newDirectLongBuffer(int numElements) { return newDirectByteBuffer(numElements * SIZEOF_LONG).asLongBuffer(); } public static LongBuffer newDirectLongBuffer(long[] values, int offset, int lenght) { return (LongBuffer)newDirectLongBuffer(lenght).put(values, offset, lenght).rewind(); } public static LongBuffer newDirectLongBuffer(long[] values, int offset) { return newDirectLongBuffer(values, offset, values.length - offset); } public static LongBuffer newDirectLongBuffer(long[] values) { return newDirectLongBuffer(values, 0); } /** * Allocates a new direct ShortBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static ShortBuffer newDirectShortBuffer(int numElements) { return newDirectByteBuffer(numElements * SIZEOF_SHORT).asShortBuffer(); } public static ShortBuffer newDirectShortBuffer(short[] values, int offset, int lenght) { return (ShortBuffer)newDirectShortBuffer(lenght).put(values, offset, lenght).rewind(); } public static ShortBuffer newDirectShortBuffer(short[] values, int offset) { return newDirectShortBuffer(values, offset, values.length - offset); } public static ShortBuffer newDirectShortBuffer(short[] values) { return newDirectShortBuffer(values, 0); } /** * Allocates a new direct CharBuffer with the specified number of * elements. The returned buffer will have its byte order set to * the host platform's native byte order. */ public static CharBuffer newDirectCharBuffer(int numElements) { return newDirectByteBuffer(numElements * SIZEOF_SHORT).asCharBuffer(); } public static CharBuffer newDirectCharBuffer(char[] values, int offset, int lenght) { return (CharBuffer)newDirectCharBuffer(lenght).put(values, offset, lenght).rewind(); } public static CharBuffer newDirectCharBuffer(char[] values, int offset) { return newDirectCharBuffer(values, offset, values.length - offset); } public static CharBuffer newDirectCharBuffer(char[] values) { return newDirectCharBuffer(values, 0); } /** * Calls slice on the specified buffer while maintaining the byteorder. * @see #slice(java.nio.Buffer, int, int) */ @SuppressWarnings("unchecked") public static B slice(B buffer) { if (buffer instanceof ByteBuffer) { final ByteBuffer bb = (ByteBuffer) buffer; return (B) bb.slice().order(bb.order()); // slice and duplicate may change byte order } else if (buffer instanceof IntBuffer) { return (B) ((IntBuffer) buffer).slice(); } else if (buffer instanceof ShortBuffer) { return (B) ((ShortBuffer) buffer).slice(); } else if (buffer instanceof FloatBuffer) { return (B) ((FloatBuffer) buffer).slice(); } else if (buffer instanceof DoubleBuffer) { return (B) ((DoubleBuffer) buffer).slice(); } else if (buffer instanceof LongBuffer) { return (B) ((LongBuffer) buffer).slice(); } else if (buffer instanceof CharBuffer) { return (B) ((CharBuffer) buffer).slice(); } throw new IllegalArgumentException("unexpected buffer type: " + buffer.getClass()); } /** * Slices the specified buffer with offset as position and offset+size as limit * while maintaining the byteorder. * Concurrency warning: this method changes the buffers position and limit but * will restore it before return. */ public static B slice(B buffer, int offset, int size) { int pos = buffer.position(); int limit = buffer.limit(); B slice = null; try { buffer.position(offset).limit(offset+size); slice = slice(buffer); } finally { buffer.position(pos).limit(limit); } return slice; } /** * Slices a ByteBuffer or a primitive float array to a FloatBuffer at the given position with the given size * in float-space. *
* The returned sliced buffer's start position is not necessarily zero, * but the float position within the host ByteBuffer. *
** The returned sliced buffer is {@link FloatBuffer#mark() marked} at it's starting position. Hence * {@link FloatBuffer#reset()} will rewind it to start after applying relative operations like {@link FloatBuffer#get()}. *
** Using a ByteBuffer as the source guarantees * keeping the source native order programmatically. * This works around Honeycomb / Android 3.0 Issue 16434. * This bug is resolved at least in Android 3.2. *
* * @param buf source Buffer, maybe ByteBuffer (recommended) or FloatBuffer ornull
.
* Buffer's position is ignored and floatPos is being used.
* @param backing source float array or null
* @param floatPos {@link Buffers#SIZEOF_FLOAT} position
* @param floatSize {@link Buffers#SIZEOF_FLOAT} size
* @return FloatBuffer w/ native byte order as given ByteBuffer
*/
public static final FloatBuffer slice2Float(Buffer buf, float[] backing, int floatPos, int floatSize) {
final int pos;
final int limit;
if(null != buf) {
pos = buf.position();
limit = buf.limit();
} else {
pos = 0;
limit = 0;
}
final FloatBuffer res;
try {
if(buf instanceof ByteBuffer) {
ByteBuffer bb = (ByteBuffer) buf;
bb.position( floatPos * Buffers.SIZEOF_FLOAT );
bb.limit( (floatPos + floatSize) * Buffers.SIZEOF_FLOAT );
res = bb.slice().order(bb.order()).asFloatBuffer(); // slice and duplicate may change byte order
} else if(null != backing) {
res = FloatBuffer.wrap(backing, floatPos, floatSize);
} else if(buf instanceof FloatBuffer) {
FloatBuffer fb = (FloatBuffer) buf;
fb.position( floatPos );
fb.limit( floatPos + floatSize );
res = fb.slice(); // slice and duplicate may change byte order
} else {
throw new InternalError("Buffer not ByteBuffer, nor FloarBuffer, nor backing array given");
}
} finally {
if(null != buf) {
buf.position(pos).limit(limit);
}
}
res.mark();
return res;
}
/**
* Helper routine to set a ByteBuffer to the native byte order, if
* that operation is supported by the underlying NIO
* implementation.
*/
public static ByteBuffer nativeOrder(ByteBuffer buf) {
return buf.order(ByteOrder.nativeOrder());
}
/**
* Returns the size of a single element of this buffer in bytes.
*/
public static int sizeOfBufferElem(Buffer buffer) {
if (buffer == null) {
return 0;
}
if (buffer instanceof ByteBuffer) {
return SIZEOF_BYTE;
} else if (buffer instanceof IntBuffer) {
return SIZEOF_INT;
} else if (buffer instanceof ShortBuffer) {
return SIZEOF_SHORT;
} else if (buffer instanceof FloatBuffer) {
return SIZEOF_FLOAT;
} else if (buffer instanceof DoubleBuffer) {
return SIZEOF_DOUBLE;
} else if (buffer instanceof LongBuffer) {
return SIZEOF_LONG;
} else if (buffer instanceof CharBuffer) {
return SIZEOF_CHAR;
}
throw new RuntimeException("Unexpected buffer type " + buffer.getClass().getName());
}
/**
* Helper routine to tell whether a buffer is direct or not. Null
* pointers are considered direct.
*/
public static boolean isDirect(Object buf) {
if (buf == null) {
return true;
}
if (buf instanceof Buffer) {
return ((Buffer) buf).isDirect();
} else if (buf instanceof PointerBuffer) {
return ((PointerBuffer) buf).isDirect();
}
throw new IllegalArgumentException("Unexpected buffer type " + buf.getClass().getName());
}
/**
* Helper routine to get the Buffer byte offset by taking into
* account the Buffer position and the underlying type. This is
* the total offset for Direct Buffers.
*/
public static int getDirectBufferByteOffset(Object buf) {
if (buf == null) {
return 0;
}
if (buf instanceof Buffer) {
int pos = ((Buffer) buf).position();
if (buf instanceof ByteBuffer) {
return pos;
} else if (buf instanceof FloatBuffer) {
return pos * SIZEOF_FLOAT;
} else if (buf instanceof IntBuffer) {
return pos * SIZEOF_INT;
} else if (buf instanceof ShortBuffer) {
return pos * SIZEOF_SHORT;
} else if (buf instanceof DoubleBuffer) {
return pos * SIZEOF_DOUBLE;
} else if (buf instanceof LongBuffer) {
return pos * SIZEOF_LONG;
} else if (buf instanceof CharBuffer) {
return pos * SIZEOF_CHAR;
}
} else if (buf instanceof NativeBuffer) {
final NativeBuffer> nb = (NativeBuffer>) buf;
return nb.position() * nb.elementSize() ;
}
throw new IllegalArgumentException("Disallowed array backing store type in buffer " + buf.getClass().getName());
}
/**
* Helper routine to return the array backing store reference from
* a Buffer object.
* @throws UnsupportedOperationException if the passed Object does not have an array backing store
* @throws IllegalArgumentException if the passed Object is neither of type {@link java.nio.Buffer} or {@link NativeBuffer}.
*/
public static Object getArray(Object buf) throws UnsupportedOperationException, IllegalArgumentException {
if (buf == null) {
return null;
}
if (buf instanceof Buffer) {
return ((Buffer) buf).array();
} else if (buf instanceof NativeBuffer) {
return ((NativeBuffer>) buf).array();
}
throw new IllegalArgumentException("Disallowed array backing store type in buffer " + buf.getClass().getName());
}
/**
* Helper routine to get the full byte offset from the beginning of
* the array that is the storage for the indirect Buffer
* object. The array offset also includes the position offset
* within the buffer, in addition to any array offset.
*/
public static int getIndirectBufferByteOffset(Object buf) {
if (buf == null) {
return 0;
}
if (buf instanceof Buffer) {
int pos = ((Buffer) buf).position();
if (buf instanceof ByteBuffer) {
return (((ByteBuffer) buf).arrayOffset() + pos);
} else if (buf instanceof FloatBuffer) {
return (SIZEOF_FLOAT * (((FloatBuffer) buf).arrayOffset() + pos));
} else if (buf instanceof IntBuffer) {
return (SIZEOF_INT * (((IntBuffer) buf).arrayOffset() + pos));
} else if (buf instanceof ShortBuffer) {
return (SIZEOF_SHORT * (((ShortBuffer) buf).arrayOffset() + pos));
} else if (buf instanceof DoubleBuffer) {
return (SIZEOF_DOUBLE * (((DoubleBuffer) buf).arrayOffset() + pos));
} else if (buf instanceof LongBuffer) {
return (SIZEOF_LONG * (((LongBuffer) buf).arrayOffset() + pos));
} else if (buf instanceof CharBuffer) {
return (SIZEOF_CHAR * (((CharBuffer) buf).arrayOffset() + pos));
}
} else if (buf instanceof NativeBuffer) {
final NativeBuffer> nb = (NativeBuffer>) buf;
return nb.elementSize() * ( nb.arrayOffset() + nb.position() );
}
throw new IllegalArgumentException("Unknown buffer type " + buf.getClass().getName());
}
//----------------------------------------------------------------------
// Copy routines (type-to-type)
//
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed ByteBuffer into
* a newly-allocated direct ByteBuffer. The returned buffer will
* have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static ByteBuffer copyByteBuffer(ByteBuffer orig) {
ByteBuffer dest = newDirectByteBuffer(orig.remaining());
dest.put(orig);
dest.rewind();
return dest;
}
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed FloatBuffer
* into a newly-allocated direct FloatBuffer. The returned buffer
* will have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static FloatBuffer copyFloatBuffer(FloatBuffer orig) {
return copyFloatBufferAsByteBuffer(orig).asFloatBuffer();
}
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed IntBuffer
* into a newly-allocated direct IntBuffer. The returned buffer
* will have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static IntBuffer copyIntBuffer(IntBuffer orig) {
return copyIntBufferAsByteBuffer(orig).asIntBuffer();
}
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed ShortBuffer
* into a newly-allocated direct ShortBuffer. The returned buffer
* will have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static ShortBuffer copyShortBuffer(ShortBuffer orig) {
return copyShortBufferAsByteBuffer(orig).asShortBuffer();
}
//----------------------------------------------------------------------
// Copy routines (type-to-ByteBuffer)
//
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed FloatBuffer
* into a newly-allocated direct ByteBuffer. The returned buffer
* will have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static ByteBuffer copyFloatBufferAsByteBuffer(FloatBuffer orig) {
ByteBuffer dest = newDirectByteBuffer(orig.remaining() * SIZEOF_FLOAT);
dest.asFloatBuffer().put(orig);
dest.rewind();
return dest;
}
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed IntBuffer into
* a newly-allocated direct ByteBuffer. The returned buffer will
* have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static ByteBuffer copyIntBufferAsByteBuffer(IntBuffer orig) {
ByteBuffer dest = newDirectByteBuffer(orig.remaining() * SIZEOF_INT);
dest.asIntBuffer().put(orig);
dest.rewind();
return dest;
}
/**
* Copies the remaining elements (as defined by
* limit() - position()
) in the passed ShortBuffer
* into a newly-allocated direct ByteBuffer. The returned buffer
* will have its byte order set to the host platform's native byte
* order. The position of the newly-allocated buffer will be zero,
* and the position of the passed buffer is unchanged (though its
* mark is changed).
*/
public static ByteBuffer copyShortBufferAsByteBuffer(ShortBuffer orig) {
ByteBuffer dest = newDirectByteBuffer(orig.remaining() * SIZEOF_SHORT);
dest.asShortBuffer().put(orig);
dest.rewind();
return dest;
}
//----------------------------------------------------------------------
// Conversion routines
//
/**
* @param source the source array
* @param soffset the offset
* @param dest the target array, if null, a new array is being created with size len.
* @param doffset the offset in the dest array
* @param len the payload of elements to be copied, if len < 0
then len = source.length - soffset
* @return the passed or newly created target array
*/
public static float[] getFloatArray(double[] source, int soffset, float[] dest, int doffset, int len) {
if(0>len) {
len = source.length - soffset;
}
if(len > source.length - soffset) {
throw new IllegalArgumentException("payload ("+len+") greater than remaining source bytes [len "+source.length+", offset "+soffset+"]");
}
if(null==dest) {
dest = new float[len];
doffset = 0;
}
if(len > dest.length - doffset) {
throw new IllegalArgumentException("payload ("+len+") greater than remaining dest bytes [len "+dest.length+", offset "+doffset+"]");
}
for(int i=0; ilen < 0
then len = source.length - soffset
* @return the passed or newly created target array
*/
public static double[] getDoubleArray(float[] source, int soffset, double[] dest, int doffset, int len) {
if(0>len) {
len = source.length - soffset;
}
if(len > source.length - soffset) {
throw new IllegalArgumentException("payload ("+len+") greater than remaining source bytes [len "+source.length+", offset "+soffset+"]");
}
if(null==dest) {
dest = new double[len];
doffset = 0;
}
if(len > dest.length - doffset) {
throw new IllegalArgumentException("payload ("+len+") greater than remaining dest bytes [len "+dest.length+", offset "+doffset+"]");
}
for(int i=0; inull
for unformatted output.
* Note: Caller is responsible to match the format string w/ the data type as expected in the given buffer.
* @param buffer Any valid Buffer instance
* @return the modified StringBuilder containing the Buffer details
*/
public static StringBuilder toString(StringBuilder sb, String f, Buffer buffer) {
if(null == sb) {
sb = new StringBuilder();
}
sb.append(buffer.getClass().getSimpleName());
sb.append("[pos ").append(buffer.position()).append(", lim ").append(buffer.limit()).append(", cap ").append(buffer.capacity());
sb.append(", remaining ").append(buffer.remaining());
sb.append("; array ").append(buffer.hasArray()).append(", direct ").append(buffer.isDirect());
sb.append(", r/w ").append(!buffer.isReadOnly()).append(": ");
if (buffer instanceof ByteBuffer) {
final ByteBuffer b = (ByteBuffer)buffer;
for(int i=0; i