FEAT: Use memory mapped stream for files < 2M.

openpreserve · Jul 7, 2020 · da6f2fb · da6f2fb
1 parent 4444f3b
commit da6f2fb
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/src/main/java/com/easyinnova/tiff/io/MappedByteInputStream.java b/src/main/java/com/easyinnova/tiff/io/MappedByteInputStream.java
@@ -32,6 +32,7 @@
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.RandomAccessFile;
 import java.lang.reflect.Method;
 import java.nio.BufferUnderflowException;
 import java.nio.MappedByteBuffer;
@@ -66,6 +67,17 @@ public MappedByteInputStream(File file) throws FileNotFoundException {
     }
   }
 
+  public MappedByteInputStream(RandomAccessFile raf) throws FileNotFoundException {
+    path = "";
+    ch = raf.getChannel();
+    try {
+      mbsize = ch.size();
+      mb = ch.map(FileChannel.MapMode.READ_ONLY, 0L, mbsize);
+    } catch (Exception ex) {
+      ex.printStackTrace();
+    }
+  }
+
   public String getPath() {
     return path;
   }

diff --git a/src/main/java/com/easyinnova/tiff/io/TiffInputStream.java b/src/main/java/com/easyinnova/tiff/io/TiffInputStream.java
@@ -99,12 +99,26 @@ public TiffInputStream(File file) throws FileNotFoundException {
   }
 
   public TiffInputStream(RandomAccessFile raf) throws FileNotFoundException {
+    long mbsize = 0;
+    try {
+      mbsize = raf.getChannel().size();
+    } catch (Exception ex) {
+
+    }
+
     internalFile = null;
-    internalFileBig = new RandomAccessFileInputStream(raf);
+    internalFileBig = null;
+
+    if (mbsize > 2147483647L) {
+      internalFileBig = new RandomAccessFileInputStream(raf);
+    } else {
+      internalFile = new MappedByteInputStream(raf);
+    }
     byteOrder = ByteOrder.BIG_ENDIAN;
     fileOffset = 0;
     buffer = new PagedInputBuffer(this);
   }
+
 
   /**
    * Gets the byte order.