Hardwire TIFFImageReaderSpi to avoid registering bug (again)

sismics · Nov 7, 2013 · 77f0368 · 77f0368
1 parent 596dd4d
commit 77f0368
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/docs-core/src/main/java/com/sismics/tess4j/ImageIOHelper.java b/docs-core/src/main/java/com/sismics/tess4j/ImageIOHelper.java
@@ -28,6 +28,8 @@
 
 import javax.imageio.IIOImage;
 import javax.imageio.ImageIO;
+import javax.imageio.ImageReadParam;
+import javax.imageio.ImageReader;
 import javax.imageio.ImageWriteParam;
 import javax.imageio.ImageWriter;
 import javax.imageio.metadata.IIOMetadata;
@@ -37,6 +39,7 @@
 import org.w3c.dom.NodeList;
 
 import com.sun.media.imageio.plugins.tiff.TIFFImageWriteParam;
+import com.sun.media.imageioimpl.plugins.tiff.TIFFImageReaderSpi;
 import com.sun.media.imageioimpl.plugins.tiff.TIFFImageWriterSpi;
 
 public class ImageIOHelper {
@@ -69,8 +72,20 @@ public static ByteBuffer getImageByteBuffer(IIOImage image) throws IOException {
         writer.setOutput(ios);
         writer.write(streamMetadata, new IIOImage(image.getRenderedImage(), null, null), tiffWriteParam);
         writer.dispose();
+
+        // Read the writed image
         ios.seek(0);
-        BufferedImage bi = ImageIO.read(ios);
+        ImageReader reader = new TIFFImageReaderSpi().createReaderInstance();
+        ImageReadParam param = reader.getDefaultReadParam();
+        reader.setInput(ios, true, true);
+        BufferedImage bi;
+        try {
+            bi = reader.read(0, param);
+        } finally {
+            reader.dispose();
+            ios.close();
+        }
+
         return convertImageData(bi);
     }