将FloatBuffer转换为ByteBuffer

时间:2014-12-15 20:02:25

标签: java nio

ByteBuffer提供asFloatBuffer()功能。但是,没有等效的FloatBuffer.asByteBuffer()

我正在尝试:

float[] array = ...
try( ByteChannel channel = Files.newByteChannel( path, WRITE, CREATE, TRUNCATE_EXISTING ) ) {
    channel.write (FloatBuffer.wrap (array) .asByteBuffer());
}

有没有办法有效地执行此操作,或者我是否必须按照以下方式分配ByteBuffer

ByteBuffer buffer = ByteBuffer.allocate( array.length * 4 );
buffer .asFloatBuffer() .put (array);
channel.write (buffer);

1 个答案:

答案 0 :(得分:0)

对于由HeapFloatBufferFloatBuffer.allocate创建的FloatBuffer.wrap,没有简单的解决方案。需要编写一个扩展ByteBuffer的自定义类。

对于HotSpot 8中的直接缓冲区,这将适用于简单的情况:

FloatBuffer floatBuffer = ByteBuffer.allocateDirect (...).asFloatBuffer();
ByteBuffer byteBuffer = (ByteBuffer) ((sun.nio.ch.DirectBuffer)floatBuffer).attachment();

对于其他情况,请使用以下方法。请注意,此类在包java.nio中声明。这也可能仅适用于HotSpot 8。

package java.nio;

/**
 *
 * @author Aleksandr Dubinsky
 */
public class BufferUtils {

      public static ByteBuffer
    asByteBuffer (FloatBuffer floatBuffer) {

            if (floatBuffer instanceof DirectFloatBufferU)
            {
                DirectFloatBufferU buffer = (DirectFloatBufferU) floatBuffer;
                return (ByteBuffer) new DirectByteBuffer (buffer.address(), floatBuffer.capacity() * Float.BYTES, buffer)
                                            .position (floatBuffer.position() * Float.BYTES)
                                            .limit (floatBuffer.limit() * Float.BYTES);
            }
            else if (floatBuffer instanceof DirectFloatBufferS)
            {
                DirectFloatBufferS buffer = (DirectFloatBufferS) floatBuffer;
                return (ByteBuffer) new DirectByteBuffer (buffer.address(), floatBuffer.capacity() * Float.BYTES, buffer)
                                            .position (floatBuffer.position() * Float.BYTES)
                                            .limit (floatBuffer.limit() * Float.BYTES);
            }
            else if (floatBuffer instanceof ByteBufferAsFloatBufferB)
            {
                ByteBufferAsFloatBufferB buffer = (ByteBufferAsFloatBufferB)floatBuffer;
                return (ByteBuffer) ((ByteBuffer) buffer.bb
                                                        .duplicate()
                                                        .position (buffer.offset)
                                                        .limit (buffer.offset + buffer.capacity() * Float.BYTES))
                                                        .slice()
                                                        .position (buffer.position() * Float.SIZE)
                                                        .limit (buffer.limit() * Float.BYTES);
            }
            else if (floatBuffer instanceof ByteBufferAsFloatBufferL)
            {
                ByteBufferAsFloatBufferL buffer = (ByteBufferAsFloatBufferL)floatBuffer;
                return (ByteBuffer) ((ByteBuffer) buffer.bb
                                                        .duplicate()
                                                        .position (buffer.offset)
                                                        .limit (buffer.offset + buffer.capacity() * Float.BYTES))
                                                        .slice()
                                                        .position (buffer.position() * Float.SIZE)
                                                        .limit (buffer.limit() * Float.BYTES);
            }
            else
                throw new IllegalArgumentException ("Unsupported implementing class " + floatBuffer.getClass().getName());
        }

}