Merge pull request #26 from iwbh15/dev_ph0.1

Setting jpeg quality in JPEGCodec
ome · Apr 6, 2023 · 7c676b5 · 7c676b5
2 parents 2666ade + 0cae3d5
commit 7c676b5
Showing 1 changed file with 42 additions and 6 deletions.
diff --git a/src/main/java/ome/codecs/JPEGCodec.java b/src/main/java/ome/codecs/JPEGCodec.java
@@ -52,6 +52,14 @@
 import loci.common.RandomAccessInputStream;
 import ome.codecs.gui.AWTImageTools;
 
+//::phaub 09.02.23
+import javax.imageio.IIOImage;
+import javax.imageio.ImageWriteParam;
+import javax.imageio.ImageWriter;
+import javax.imageio.stream.ImageOutputStream;
+import javax.imageio.IIOException;
+
+
 /**
  * This class implements JPEG compression and decompression.
  */
@@ -88,13 +96,41 @@ public byte[] compress(byte[] data, CodecOptions options)
       options.bitsPerSample / 8, false, options.littleEndian, options.signed);
 
     try {
-    	ImageOutputStream stream = new MemoryCacheImageOutputStream(out);
-    	Iterator<ImageWriter> iterator = ImageIO.getImageWritersByFormatName("jpeg");
+      //::phaub 09.02.23   (Adjustable jpeg quality)
+      // How to use:
+      // Set jpegquality using CodecOptions in the calling object (e.g. QuPath using OMEPyramidWriter()):
+      //   CodecOptions options = new CodecOptions();
+      //   options.quality = "0.95";
+      //   writer.setCodecOptions(options)
+
+      double jpegquality = 0.75;
+      if (options.quality > 0) {
+        jpegquality = options.quality;
+      }
+      jpegquality = Math.max(0.25, Math.min(1.0, jpegquality));
+
+      ImageWriter jpgWriter = ImageIO.getImageWritersByFormatName("jpg").next();
+      if (jpgWriter == null) {
+        return null;
+      }
+      ImageWriteParam jpgWriteParam = jpgWriter.getDefaultWriteParam();
+      jpgWriteParam.setCompressionMode(ImageWriteParam.MODE_EXPLICIT);
+      jpgWriteParam.setCompressionQuality((float) jpegquality);
+
+      ImageOutputStream stream = new MemoryCacheImageOutputStream(out);
+      try {
+        Iterator<ImageWriter> iterator = ImageIO.getImageWritersByFormatName("jpeg");
     	if (iterator.hasNext()) {
-	        ImageWriter writer = iterator.next();
-	        writer.setOutput(stream);
-	        writer.write(img);
-    	}
+	  ImageWriter writer = iterator.next();
+	  writer.setOutput(stream);
+          IIOImage outputImage = new IIOImage(img, null, null);
+	  writer.write(null, outputImage, jpgWriteParam);
+        }
+      } finally {
+        jpgWriter.dispose();
+        stream.flush(); 	      
+        stream.close();
+      }          
     }
     catch (IOException e) {
       throw new CodecException("Could not write JPEG data", e);