起因

一個對試卷進行OCR識別需求，需要實作一個功能，一個章節下的題目圖片需要上下拼接合成一張大圖，起初寫了一個工具實作圖片的合并，程式一直很穩定的運行著，有一天用戶反饋合成的圖片方向不對，起扯訓疑是本身圖片方向有問題，但是用windows圖片查看器打開圖片方向是“正常”顯示的

定位

exif資訊

查閱相關資料，圖片資訊中有個exif標準，exif資訊如下：

圖蟲exif資訊查看器：https://exif.tuchong.com/
關注IFD0節點方向，Rotate 270 CW，意思圖片需要順時針旋轉270°方向正常，windows默認的圖片查看器已經幫我們自動旋轉展示了，我們在手機橫拍或者掃描儀、數碼相機輸出的圖片通常包含此類資訊，但是我們java讀取的是圖片的真實方向，所以在生成圖片方向自然也就不對了

代碼

添加依賴

<dependency>
    <groupId>com.drewnoakes</groupId>
    <artifactId>metadata-extractor</artifactId>
    <version>2.15.0</version>
</dependency>

自旋轉代碼

旋轉圖片

private static void rotateImage(List<String> stringList) {
    stringList.forEach(s -> {
        File file = new File(s);
        try {
            Metadata metadata = https://www.cnblogs.com/surging-dandelion/p/ImageMetadataReader.readMetadata(file);
            StringBuilder description = new StringBuilder();
            metadata.getDirectories().forEach(directory -> {
                directory.getTags().forEach(tag -> {
                    if (tag.getTagType() == ExifDirectoryBase.TAG_ORIENTATION) {
                        description.append(tag.getDescription().replaceAll(" ", ""));
                    }
                });
            });
            if (description.length() > 0) {
                int rotateIndex = description.indexOf("Rotate");
                int cwIndex = description.indexOf("CW");
                if (rotateIndex >= 0 && cwIndex > 0 && rotateIndex < cwIndex) {
                    int angel = Integer.valueOf(description.substring(rotateIndex + 6, cwIndex));
                    log.info("============圖片方向糾正，順時針旋轉{}°,圖片路徑：{}===========", angel, s);
                    BufferedImage oldImage = ImageIO.read(file);
                    BufferedImage newImage = RotateImage.Rotate(oldImage, angel);
                    ImageIO.write(newImage, "jpg", file);
                    newImage.getGraphics().dispose();
                    oldImage.getGraphics().dispose();
                }
            }
        } catch (ImageProcessingException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

    });
}

圖片旋轉工具類

import java.awt.*;
import java.awt.image.BufferedImage;

public class RotateImage {
    /**
     * 對圖片進行旋轉
     *
     * @param src   被旋轉圖片
     * @param angel 旋轉角度
     * @return 旋轉后的圖片
     */
    public static BufferedImage Rotate(Image src, int angel) {
        int srcWidth = src.getWidth(null);
        int srcHeight = src.getHeight(null);
        // 計算旋轉后圖片的尺寸
        Rectangle rect_des = CalcRotatedSize(new Rectangle(new Dimension(
            srcWidth, srcHeight)), angel);
        BufferedImage res = null;
        res = new BufferedImage(rect_des.width, rect_des.height,
            BufferedImage.TYPE_INT_RGB);
        Graphics2D g2 = res.createGraphics();
        // 進行轉換
        g2.translate((rect_des.width - srcWidth) / 2,
            (rect_des.height - srcHeight) / 2);
        g2.rotate(Math.toRadians(angel), srcWidth / 2, srcHeight / 2);

        g2.drawImage(src, null, null);
        return res;
    }

    /**
     * 計算旋轉后的圖片
     *
     * @param src   被旋轉的圖片
     * @param angel 旋轉角度
     * @return 旋轉后的圖片
     */
    public static Rectangle CalcRotatedSize(Rectangle src, int angel) {
        // 如果旋轉的角度大于90度做相應的轉換
        if (angel >= 90) {
            if (angel / 90 % 2 == 1) {
                int temp = src.height;
                src.height = src.width;
                src.width = temp;
            }
            angel = angel % 90;
        }

        double r = Math.sqrt(src.height * src.height + src.width * src.width) / 2;
        double len = 2 * Math.sin(Math.toRadians(angel) / 2) * r;
        double angelAlpha = (Math.PI - Math.toRadians(angel)) / 2;
        double angelDeltaWidth = Math.atan((double) src.height / src.width);
        double angelDeltaHeight = Math.atan((double) src.width / src.height);

        int lenDeltaWidth = (int) (len * Math.cos(Math.PI - angelAlpha
            - angelDeltaWidth));
        int lenDeltaHeight = (int) (len * Math.cos(Math.PI - angelAlpha
            - angelDeltaHeight));
        int desWidth = src.width + lenDeltaWidth * 2;
        int desHeight = src.height + lenDeltaHeight * 2;
        return new Rectangle(new Dimension(desWidth, desHeight));
    }
}

測驗代碼

public static void main(String[] args) throws ImageProcessingException, IOException {
    File file = new File("D:\\temp\\error_direction");
    List<String> stringList = Arrays.asList(file.listFiles()).stream().map(File::getAbsolutePath).collect(Collectors.toList());
    rotateImage(stringList);
}

控制臺日志如下：

18:52:40.066 [main] INFO ============圖片方向糾正，順時針旋轉270°,圖片路徑：D:\temp\error_direction\185.jpg===========
18:52:40.879 [main] INFO ============圖片方向糾正，順時針旋轉90°,圖片路徑：D:\temp\error_direction\186.jpg===========

至此，我們原始圖片的方向被糾正了，我們再來看下糾正之后的圖片的exif資訊，發現旋轉資訊已經沒有了

標籤：Java

上一篇：隨便聊聊 Java 8 的函式式編程

下一篇：想少踩坑？不可忽視的JAVA注釋