summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/java/com/jogamp/common/util/IOUtil.java33
-rw-r--r--src/junit/com/jogamp/common/util/TestIOUtil01.java112
2 files changed, 145 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
diff --git a/src/junit/com/jogamp/common/util/TestIOUtil01.java b/src/junit/com/jogamp/common/util/TestIOUtil01.java
index 19bc8b0..c0e26a9 100644
--- a/src/junit/com/jogamp/common/util/TestIOUtil01.java
+++ b/src/junit/com/jogamp/common/util/TestIOUtil01.java
@@ -221,6 +221,118 @@ public class TestIOUtil01 extends SingletonJunitCase {
}
}
+ @Test
+ public void test21CopyStreamChunk01Buffer() throws IOException {
+ final URLConnection urlConn = IOUtil.getResource(tfilename, this.getClass().getClassLoader(), this.getClass());
+ Assert.assertNotNull(urlConn);
+ final BufferedInputStream bis = new BufferedInputStream( urlConn.getInputStream() );
+ final ByteBuffer bb;
+ final int skipBytes = 0;
+ final int byteCount = orig.length;
+ try {
+ bb = IOUtil.copyStreamChunk2ByteBuffer( bis, skipBytes, byteCount );
+ } finally {
+ IOUtil.close(bis, false);
+ }
+ Assert.assertTrue( machine.pageAlignedSize( byteCount ) >= byteCount );
+ Assert.assertEquals( machine.pageAlignedSize( byteCount ), bb.capacity() );
+ Assert.assertEquals(orig.length, bb.limit());
+ int i;
+ for(i=tsz-1; i>=0 && orig[i]==bb.get(i); i--) ;
+ Assert.assertTrue("Bytes not equal orig vs array", 0>i);
+ System.err.println(getTestMethodName()+" OK: ["+skipBytes+".."+(skipBytes+byteCount)+"): "+bb);
+ }
+
+ @Test
+ public void test22CopyStreamChunk02Buffer() throws IOException {
+ final URLConnection urlConn = IOUtil.getResource(tfilename, this.getClass().getClassLoader(), this.getClass());
+ Assert.assertNotNull(urlConn);
+ final BufferedInputStream bis = new BufferedInputStream( urlConn.getInputStream() );
+ final ByteBuffer bb;
+ final int skipBytes = 0;
+ final int byteCount = orig.length + machine.pageSizeInBytes(); // expect more data than exists, complete with actual data
+ try {
+ bb = IOUtil.copyStreamChunk2ByteBuffer( bis, skipBytes, byteCount );
+ } finally {
+ IOUtil.close(bis, false);
+ }
+ Assert.assertTrue( machine.pageAlignedSize( byteCount ) >= byteCount );
+ Assert.assertEquals( machine.pageAlignedSize( byteCount ), bb.capacity() );
+ Assert.assertEquals( orig.length, bb.limit() );
+ int i;
+ for(i=tsz-1; i>=0 && orig[i]==bb.get(i); i--) ;
+ Assert.assertTrue("Bytes not equal orig vs array", 0>i);
+ System.err.println(getTestMethodName()+" OK: ["+skipBytes+".."+(skipBytes+byteCount)+"): "+bb);
+ }
+
+ @Test
+ public void test23CopyStreamChunk03Buffer() throws IOException {
+ final URLConnection urlConn = IOUtil.getResource(tfilename, this.getClass().getClassLoader(), this.getClass());
+ Assert.assertNotNull(urlConn);
+ final BufferedInputStream bis = new BufferedInputStream( urlConn.getInputStream() );
+ final ByteBuffer bb;
+ final int skipBytes = 0;
+ final int byteCount = orig.length / 2; // take less data than exists
+ Assert.assertTrue( orig.length >= byteCount );
+ try {
+ bb = IOUtil.copyStreamChunk2ByteBuffer( bis, skipBytes, byteCount );
+ } finally {
+ IOUtil.close(bis, false);
+ }
+ Assert.assertTrue( machine.pageAlignedSize( byteCount ) >= byteCount );
+ Assert.assertEquals( machine.pageAlignedSize( byteCount ), bb.capacity() );
+ Assert.assertEquals( byteCount, bb.limit() );
+ int i;
+ for(i=byteCount-1; i>=0 && orig[i]==bb.get(i); i--) ;
+ Assert.assertTrue("Bytes not equal orig vs array", 0>i);
+ System.err.println(getTestMethodName()+" OK: ["+skipBytes+".."+(skipBytes+byteCount)+"): "+bb);
+ }
+
+ @Test
+ public void test24CopyStreamChunk04Buffer() throws IOException {
+ final URLConnection urlConn = IOUtil.getResource(tfilename, this.getClass().getClassLoader(), this.getClass());
+ Assert.assertNotNull(urlConn);
+ final BufferedInputStream bis = new BufferedInputStream( urlConn.getInputStream() );
+ final ByteBuffer bb;
+ final int skipBytes = orig.length / 2;
+ final int byteCount = orig.length / 4; // take less data than exists
+ Assert.assertTrue( orig.length >= skipBytes + byteCount );
+ Assert.assertTrue( orig.length >= byteCount );
+ try {
+ bb = IOUtil.copyStreamChunk2ByteBuffer( bis, skipBytes, byteCount );
+ } finally {
+ IOUtil.close(bis, false);
+ }
+ Assert.assertTrue( machine.pageAlignedSize( byteCount ) >= byteCount );
+ Assert.assertEquals( machine.pageAlignedSize( byteCount ), bb.capacity() );
+ Assert.assertEquals( byteCount, bb.limit() );
+ int i;
+ for(i=byteCount-1; i>=0 && orig[skipBytes+i]==bb.get(i); i--) ;
+ Assert.assertTrue("Bytes not equal orig vs array", 0>i);
+ System.err.println(getTestMethodName()+" OK: ["+skipBytes+".."+(skipBytes+byteCount)+"): "+bb);
+ }
+
+ @Test
+ public void test25CopyStreamChunk05Buffer() throws IOException {
+ final URLConnection urlConn = IOUtil.getResource(tfilename, this.getClass().getClassLoader(), this.getClass());
+ Assert.assertNotNull(urlConn);
+ final BufferedInputStream bis = new BufferedInputStream( urlConn.getInputStream() );
+ final ByteBuffer bb;
+ final int skipBytes = orig.length * 2; // skip all and more ..
+ final int byteCount = orig.length; // expect more data than left, won't take anything
+ Assert.assertTrue( orig.length < skipBytes + byteCount );
+ Assert.assertTrue( orig.length >= byteCount );
+ try {
+ bb = IOUtil.copyStreamChunk2ByteBuffer( bis, skipBytes, byteCount );
+ } finally {
+ IOUtil.close(bis, false);
+ }
+ Assert.assertTrue( machine.pageAlignedSize( byteCount ) >= byteCount );
+ Assert.assertEquals( machine.pageAlignedSize( byteCount ), bb.capacity() );
+ Assert.assertEquals( 0, bb.limit() );
+ System.err.println(getTestMethodName()+" OK: ["+skipBytes+".."+(skipBytes+byteCount)+"): "+bb);
+ }
+
public static void main(final String args[]) throws IOException {
final String tstname = TestIOUtil01.class.getName();
org.junit.runner.JUnitCore.main(tstname);