package com.atlassian.core.util.thumbnail.loader;
   
   import com.atlassian.core.util.ImageInfo;
   import com.google.common.base.Optional;
   import com.google.common.base.Throwables;
   import com.sun.media.jai.codec.SeekableStream;
   import org.apache.commons.io.IOUtils;
   
   import javax.media.jai.JAI;
  import javax.media.jai.OpImage;
  import javax.media.jai.PlanarImage;
  import javax.media.jai.RenderedOp;
  import java.awt.image.BufferedImage;
  import java.io.IOException;
  import java.io.InputStream;
  
  class JAIImageLoader implements ImageLoader
  {
  
      @Override
      public Optional<BufferedImage> loadImage(final InputStream inputStream, ImageInfo imageInfo) throws IOException
      {
          if(imageInfo != null && imageInfo.getFormat() == ImageInfo.FORMAT_PNG)
          {
              // JAI has some issues with PNG files(JRA-35816) and we use JAI, because of CMYK JPEGs (JRA-20072)
              // so it's safe to disable this loader for PNGs
              throw new IOException("PNG format is not supported by JAIImageLoader");
          }
  
          SeekableStream seekableStream = null;
          RenderedOp img = null;
          try
          {
              seekableStream = SeekableStream.wrapInputStream(inputStream, true);
              img = JAI.create("stream", seekableStream);
              if (img == null)
                  return Optional.absent();
              removeTileCache(img);
              return Optional.fromNullable(img.getAsBufferedImage());
          }
          catch (Exception e)
          {
              //JAI throws RuntimeExceptions when image is corrupted, we need to wrap them
  			Throwables.propagateIfInstanceOf(e, IOException.class);
              throw new IOException(e);
          }
          finally
          {
              IOUtils.closeQuietly(seekableStream);
              if(img != null)
              {
                  img.dispose();
              }
          }
      }
  
      /**
       * We don't want to cache image tiles in memory.
       */
      private void removeTileCache(final RenderedOp img)
      {
          final PlanarImage imgRendering = img.getRendering();
          if (imgRendering instanceof OpImage)
          {
              ((OpImage) imgRendering).setTileCache(null);
          }
      }
  }