圖像處理工具類

在實際項目中，咱們常常會遇處處理各類各樣的圖片問題。

好比：圖片的旋轉、縮放、圖片格式轉換、獲取圖片類型、驗證圖片大小、寫入圖片 等。 這裏咱們使用java.awt.Graphics2D來實現經常使用圖像處理的功能，造成咱們的圖像處理工具類。

package com.zhangsx.util.image;

import java.util.Iterator;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import javax.imageio.ImageIO;
import javax.imageio.ImageReader;
import javax.imageio.stream.ImageInputStream;

/**
 * 圖像處理工具類。
 *
 * @version 1.00 2010-1-15
 * @since 1.5
 * @author ZhangShixi
 */
public class ImageUtil {

    /**
     * 旋轉圖像。
     * @param bufferedImage 圖像。
     * @param degree 旋轉角度。
     * @return 旋轉後的圖像。
     */
    public static BufferedImage rotateImage(
            final BufferedImage bufferedImage, final int degree) {
        int width = bufferedImage.getWidth();
        int height = bufferedImage.getHeight();
        int type = bufferedImage.getColorModel().getTransparency();

        BufferedImage image = new BufferedImage(width, height, type);
        Graphics2D graphics2D = image.createGraphics();
        graphics2D.setRenderingHint(
                RenderingHints.KEY_INTERPOLATION,
                RenderingHints.VALUE_INTERPOLATION_BILINEAR);

        graphics2D.rotate(Math.toRadians(degree), width / 2, height / 2);
        graphics2D.drawImage(bufferedImage, 0, 0, null);

        try {
            return image;
        } finally {
            if (graphics2D != null) {
                graphics2D.dispose();
            }
        }
    }

    /**
     * 將圖像按照指定的比例縮放。
     * 好比須要將圖像放大20%，那麼調用時scale參數的值就爲20；若是是縮小，則scale值爲-20。
     * @param bufferedImage 圖像。
     * @param scale 縮放比例。
     * @return 縮放後的圖像。
     */
    public static BufferedImage resizeImageScale(
            final BufferedImage bufferedImage, final int scale) {
        if (scale == 0) {
            return bufferedImage;
        }

        int width = bufferedImage.getWidth();
        int height = bufferedImage.getHeight();

        int newWidth = 0;
        int newHeight = 0;

        double nowScale = (double) Math.abs(scale) / 100;
        if (scale > 0) {
            newWidth = (int) (width * (1 + nowScale));
            newHeight = (int) (height * (1 + nowScale));
        } else if (scale < 0) {
            newWidth = (int) (width * (1 - nowScale));
            newHeight = (int) (height * (1 - nowScale));
        }

        return resizeImage(bufferedImage, newWidth, newHeight);
    }

    /**
     * 將圖像縮放到指定的寬高大小。
     * @param bufferedImage 圖像。
     * @param width 新的寬度。
     * @param height 新的高度。
     * @return 縮放後的圖像。
     */
    public static BufferedImage resizeImage(
            final BufferedImage bufferedImage,
            final int width, final int height) {
        int type = bufferedImage.getColorModel().getTransparency();
        BufferedImage image = new BufferedImage(width, height, type);

        Graphics2D graphics2D = image.createGraphics();
        graphics2D.setRenderingHint(
                RenderingHints.KEY_INTERPOLATION,
                RenderingHints.VALUE_INTERPOLATION_BILINEAR);

        graphics2D.drawImage(bufferedImage, 0, 0, width, height, 0, 0,
                bufferedImage.getWidth(), bufferedImage.getHeight(), null);

        try {
            return image;
        } finally {
            if (graphics2D != null) {
                graphics2D.dispose();
            }
        }
    }

    /**
     * 將圖像水平翻轉。
     * @param bufferedImage 圖像。
     * @return 翻轉後的圖像。
     */
    public static BufferedImage flipImage(
            final BufferedImage bufferedImage) {
        int width = bufferedImage.getWidth();
        int height = bufferedImage.getHeight();
        int type = bufferedImage.getColorModel().getTransparency();

        BufferedImage image = new BufferedImage(width, height, type);
        Graphics2D graphics2D = image.createGraphics();
        graphics2D.drawImage(bufferedImage, 0, 0, width, height,
                width, 0, 0, height, null);

        try {
            return image;
        } finally {
            if (graphics2D != null) {
                graphics2D.dispose();
            }
        }
    }

    /**
     * 獲取圖片的類型。若是是 gif、jpg、png、bmp 之外的類型則返回null。
     * @param imageBytes 圖片字節數組。
     * @return 圖片類型。
     * @throws java.io.IOException IO異常。
     */
    public static String getImageType(final byte[] imageBytes)
            throws IOException {
        ByteArrayInputStream input = new ByteArrayInputStream(imageBytes);
        ImageInputStream imageInput = ImageIO.createImageInputStream(input);
        Iterator<ImageReader> iterator = ImageIO.getImageReaders(imageInput);
        String type = null;
        if (iterator.hasNext()) {
            ImageReader reader = iterator.next();
            type = reader.getFormatName().toUpperCase();
        }

        try {
            return type;
        } finally {
            if (imageInput != null) {
                imageInput.close();
            }
        }
    }

    /**
     * 驗證圖片大小是否超出指定的尺寸。未超出指定大小返回true，超出指定大小則返回false。
     * @param imageBytes 圖片字節數組。
     * @param width 圖片寬度。
     * @param height 圖片高度。
     * @return 驗證結果。未超出指定大小返回true，超出指定大小則返回false。
     * @throws java.io.IOException IO異常。
     */
    public static boolean checkImageSize(
            final byte[] imageBytes, final int width, final int height)
            throws IOException {
        BufferedImage image = byteToImage(imageBytes);
        int sourceWidth = image.getWidth();
        int sourceHeight = image.getHeight();
        if (sourceWidth > width || sourceHeight > height) {
            return false;
        } else {
            return true;
        }
    }

    /**
     * 將圖像字節數組轉化爲BufferedImage圖像實例。
     * @param imageBytes 圖像字節數組。
     * @return BufferedImage圖像實例。
     * @throws java.io.IOException IO異常。
     */
    public static BufferedImage byteToImage(
            final byte[] imageBytes) throws IOException {
        ByteArrayInputStream input = new ByteArrayInputStream(imageBytes);
        BufferedImage image = ImageIO.read(input);

        try {
            return image;
        } finally {
            if (input != null) {
                input.close();
            }
        }
    }

    /**
     * 將BufferedImage持有的圖像轉化爲指定圖像格式的字節數組。
     * @param bufferedImage 圖像。
     * @param formatName 圖像格式名稱。
     * @return 指定圖像格式的字節數組。
     * @throws java.io.IOException IO異常。
     */
    public static byte[] imageToByte(
            final BufferedImage bufferedImage, final String formatName)
            throws IOException {
        ByteArrayOutputStream output = new ByteArrayOutputStream();
        ImageIO.write(bufferedImage, formatName, output);

        try {
            return output.toByteArray();
        } finally {
            if (output != null) {
                output.close();
            }
        }
    }

    /**
     * 將圖像以指定的格式進行輸出，調用者需自行關閉輸出流。
     * @param bufferedImage 圖像。
     * @param output 輸出流。
     * @param formatName 圖片格式名稱。
     * @throws java.io.IOException IO異常。
     */
    public static void writeImage(final BufferedImage bufferedImage,
            final OutputStream output, final String formatName)
            throws IOException {
        ImageIO.write(bufferedImage, formatName, output);
    }

    /**
     * 將圖像以指定的格式進行輸出，調用者需自行關閉輸出流。
     * @param imageBytes 圖像的byte數組。
     * @param output 輸出流。
     * @param formatName 圖片格式名稱。
     * @throws java.io.IOException IO異常。
     */
    public static void writeImage(final byte[] imageBytes,
            final OutputStream output, final String formatName)
            throws IOException {
        BufferedImage image = byteToImage(imageBytes);
        ImageIO.write(image, formatName, output);
    }
}
http://zhangshixi.iteye.com/blog/641733