com.sksamuel.scrimage.nio.ImageIOReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scrimage-core Show documentation

JVM Image Library

There is a newer version: 4.2.0

package com.sksamuel.scrimage.nio;

import com.sksamuel.scrimage.ImmutableImage;

import javax.imageio.ImageIO;
import javax.imageio.ImageReadParam;
import javax.imageio.ImageTypeSpecifier;
import javax.imageio.stream.ImageInputStream;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.Iterator;
import java.util.Optional;

public class ImageIOReader implements ImageReader {

   private Optional tryLoad(javax.imageio.ImageReader reader, ImageInputStream iis, Rectangle rectangle) {
      try {
         reader.setInput(iis);

         ImageReadParam params = reader.getDefaultReadParam();
         Iterator imageTypes = reader.getImageTypes(0);
         if (imageTypes.hasNext()) {
            ImageTypeSpecifier imageTypeSpecifier = imageTypes.next();
            int bufferedImageType = imageTypeSpecifier.getBufferedImageType();
            params.setDestinationType(imageTypeSpecifier);
         }
         if (rectangle != null) {
            params.setSourceRegion(rectangle);
         }

         BufferedImage bufferedImage = reader.read(0, params);
         return Optional.of(ImmutableImage.wrapAwt(bufferedImage));
      } catch (IOException | InvalidImageParameterException e) {
         e.printStackTrace();
         return Optional.empty();
      }
   }

   @Override
   public ImmutableImage read(byte[] bytes, Rectangle rectangle) throws IOException {
      if (bytes == null)
         throw new IOException("bytes cannot be null");

      ImageInputStream iis = ImageIO.createImageInputStream(new ByteArrayInputStream(bytes));
      if (iis == null)
         throw new IOException("No ImageInputStream supported this image format");

      Iterator readers = ImageIO.getImageReaders(iis);
      while (readers.hasNext()) {
         Optional image = tryLoad(readers.next(), iis, rectangle);
         if (image.isPresent())
            return image.get();
      }

      throw new IOException("No ImageInputStream supported this image format");
   }
}