java/copy-image-resizer/src/cz/frantovo/copyImageResizer/JavaSingleImageResizer.java
author František Kučera <franta-hg@frantovo.cz>
Mon, 17 Nov 2014 23:18:41 +0100
changeset 19 cda209705642
parent 17 505031965440
permissions -rw-r--r--
use smaller dimension
     1 /**
     2  * copy-image-resizer
     3  * Copyright © 2014 František Kučera (frantovo.cz)
     4  *
     5  * This program is free software: you can redistribute it and/or modify
     6  * it under the terms of the GNU General Public License as published by
     7  * the Free Software Foundation, either version 3 of the License, or
     8  * (at your option) any later version.
     9  *
    10  * This program is distributed in the hope that it will be useful,
    11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    13  * GNU General Public License for more details.
    14  *
    15  * You should have received a copy of the GNU General Public License
    16  * along with this program. If not, see <http://www.gnu.org/licenses/>.
    17  */
    18 package cz.frantovo.copyImageResizer;
    19 
    20 import java.awt.Graphics2D;
    21 import java.awt.image.BufferedImage;
    22 import java.io.IOException;
    23 import java.io.InputStream;
    24 import java.io.OutputStream;
    25 import java.util.logging.Level;
    26 import java.util.logging.Logger;
    27 import java.util.regex.Pattern;
    28 import javax.imageio.ImageIO;
    29 
    30 /**
    31  *
    32  * @author Ing. František Kučera (frantovo.cz)
    33  */
    34 public class JavaSingleImageResizer implements SingleImageResizer {
    35 
    36 	private static final Logger log = Logger.getLogger(SingleImageResizer.class.getName());
    37 
    38 	@Override
    39 	public boolean resize(InputStream input, OutputStream output, SizeSpecification size, ImageFormat outputFormat) throws ResizeException {
    40 		try {
    41 			BufferedImage image = readImage(input);
    42 			if (shouldResize(image, size)) {
    43 				BufferedImage resized = resize(image, size.getWidth(), size.getHeight(), image.getType());
    44 				ImageIO.write(resized, outputFormat.getFormat(), output);
    45 				return true;
    46 			} else {
    47 				return false;
    48 			}
    49 		} catch (IOException e) {
    50 			throw new ResizeException("Error while resizing image", e);
    51 		}
    52 	}
    53 
    54 	private BufferedImage resize(BufferedImage original, int maxWidth, int maxHeight, int type) {
    55 
    56 		int width;
    57 		int height;
    58 		float originalAspectRatio = (float) original.getWidth() / (float) original.getHeight();
    59 		float requestedAspectRatio = (float) maxWidth / (float) maxHeight;
    60 
    61 		if (originalAspectRatio > requestedAspectRatio) {
    62 			width = maxWidth;
    63 			height = Math.max(Math.round(maxWidth / originalAspectRatio), 1);
    64 		} else {
    65 			width = Math.max(Math.round(maxHeight * originalAspectRatio), 1);
    66 			height = maxHeight;
    67 		}
    68 
    69 		if (type == BufferedImage.TYPE_CUSTOM) {
    70 			log.log(Level.FINER, "Setting default image type: from TYPE_CUSTOM to TYPE_INT_ARGB");
    71 			type = BufferedImage.TYPE_INT_ARGB;
    72 		}
    73 
    74 		BufferedImage resized = new BufferedImage(width, height, type);
    75 		Graphics2D g = resized.createGraphics();
    76 		g.drawImage(original, 0, 0, width, height, null);
    77 		g.dispose();
    78 
    79 		return resized;
    80 	}
    81 
    82 	private static BufferedImage readImage(InputStream input) throws ResizeException {
    83 		try {
    84 			return ImageIO.read(input);
    85 		} catch (IOException e) {
    86 			throw new ResizeException("Unable to read image from stream", e);
    87 		}
    88 	}
    89 
    90 	private static boolean shouldResize(BufferedImage input, SizeSpecification requested) {
    91 		if (requested.isResizeSmaller()) {
    92 			return input.getHeight() != requested.getHeight() || input.getWidth() != requested.getWidth();
    93 		} else {
    94 			return input.getHeight() > requested.getHeight() || input.getWidth() > requested.getWidth();
    95 		}
    96 	}
    97 
    98 	public static enum ImageFormat {
    99 
   100 		JPEG("jpg", Pattern.compile(".*\\.(jpg|jpeg)$", Pattern.CASE_INSENSITIVE)),
   101 		PNG("png", Pattern.compile(".*\\.png$", Pattern.CASE_INSENSITIVE)),
   102 		GIF("gif", Pattern.compile(".*\\.gif$", Pattern.CASE_INSENSITIVE));
   103 
   104 		private final String format;
   105 		private final Pattern regex;
   106 
   107 		private ImageFormat(String format, Pattern regex) {
   108 			this.format = format;
   109 			this.regex = regex;
   110 		}
   111 
   112 		/**
   113 		 * @return format name for {@linkplain ImageIO#write(java.awt.image.RenderedImage, java.lang.String, java.io.File)
   114 		 */
   115 		public String getFormat() {
   116 			return format;
   117 		}
   118 
   119 		public static ImageFormat getMatching(String fileName) {
   120 			for (ImageFormat type : values()) {
   121 				if (type.regex.matcher(fileName).matches()) {
   122 					return type;
   123 				}
   124 			}
   125 			return null;
   126 		}
   127 	}
   128 }