diff options
author | Sven Gothel <[email protected]> | 2023-05-04 01:04:53 +0200 |
---|---|---|
committer | Sven Gothel <[email protected]> | 2023-05-04 01:04:53 +0200 |
commit | b32e378ec80488c5ffbd0d9bb356217e6db0245f (patch) | |
tree | 1eae94088800a2eb7924822cf7d23561a403357e /src/java | |
parent | 6e67a7b54af82aa4e4ec28f50ff08a26f9d80627 (diff) |
IOUtil.copyStreamChunk2ByteBuffer(..): Added new method to copy a chunk (segment) of the input stream (skipBytes, byteCount)
This method is inspired by Bug 1280, <https://github.com/sgothel/joal/pull/16>,
'copy only needed bytes' for JOAL's com.jogamp.openal.util.WAVData.loadFromStream(..).
This method is a revised version of the proposed IOHelpers.copyFromStream2ByteBuffer(..),
see <https://github.com/OndrejSpanel/joal/commit/1616659e98904270af4faca25b770d0983609735>
Diffstat (limited to 'src/java')
-rw-r--r-- | src/java/com/jogamp/common/util/IOUtil.java | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/java/com/jogamp/common/util/IOUtil.java b/src/java/com/jogamp/common/util/IOUtil.java index 499fa1b..e68c3b3 100644 --- a/src/java/com/jogamp/common/util/IOUtil.java +++ b/src/java/com/jogamp/common/util/IOUtil.java @@ -337,6 +337,39 @@ public class IOUtil { return data; } + /** + * Copy the specified input stream chunk to a NIO ByteBuffer w/ native byte order, which is being returned. + * + * @param stream input stream, which will be wrapped into a BufferedInputStream, if not already done. + * @param skipBytes initial bytes to skip from input stream. + * @param byteCount bytes to copy starting after skipBytes. + */ + public static ByteBuffer copyStreamChunk2ByteBuffer(InputStream stream, int skipBytes, int byteCount) throws IOException { + if( !(stream instanceof BufferedInputStream) ) { + stream = new BufferedInputStream(stream); + } + final MachineDataInfo machine = Platform.getMachineDataInfo(); + final ByteBuffer data = Buffers.newDirectByteBuffer( machine.pageAlignedSize( byteCount ) ); + final byte[] chunk = new byte[machine.pageSizeInBytes()]; + int numRead = 1; // EOS: -1 == numRead, EOF maybe reached earlier w/ 0 == numRead + while ( numRead > 0 && skipBytes > 0 ) { + final int chunk2Read = Math.min(machine.pageSizeInBytes(), skipBytes); + numRead = stream.read(chunk, 0, chunk2Read); + skipBytes -= numRead; + } + while ( numRead > 0 && byteCount > 0 ) { + final int chunk2Read = Math.min(machine.pageSizeInBytes(), byteCount); + numRead = stream.read(chunk, 0, chunk2Read); + if (numRead > 0) { + data.put(chunk, 0, numRead); + } + byteCount -= numRead; + } + + data.flip(); + return data; + } + /*** * * RESOURCE / FILE NAME STUFF |