Java图片处理：实现图片居中插入与画布调整

import org.springframework.web.multipart.MultipartFile;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

public class ImageProcessor {

    /**
     * 将源图片居中插入到指定宽高比的新画布中。
     *
     * @param multipartFile 待处理的源图片文件
     * @param targetWidth 目标画布的宽度
     * @param targetHeight 目标画布的高度
     * @param backgroundColor 目标画布的背景颜色
     * @param outputPath 结果图片保存路径
     * @return 结果图片的文件路径，或在发生错误时返回"error"
     */
    public static String insertImageCentered(MultipartFile multipartFile, 
                                             int targetWidth, 
                                             int targetHeight, 
                                             Color backgroundColor, 
                                             String outputPath) {
        try {
            // 1. 加载源图片
            BufferedImage sourceImage = ImageIO.read(multipartFile.getInputStream());
            if (sourceImage == null) {
                System.err.println("无法读取源图片，请检查文件格式。");
                return "error";
            }

            // 获取源图片的尺寸
            int sourceWidth = sourceImage.getWidth();
            int sourceHeight = sourceImage.getHeight();

            // 2. 创建目标画布
            BufferedImage canvasImage = new BufferedImage(targetWidth, targetHeight, BufferedImage.TYPE_INT_RGB);

            // 获取Graphics2D对象，用于在画布上绘图
            Graphics2D g2d = canvasImage.createGraphics();

            try {
                // 填充背景色
                g2d.setColor(backgroundColor);
                g2d.fillRect(0, 0, targetWidth, targetHeight);

                // 3. 计算插入坐标
                // 计算源图片在新画布中居中放置时的左上角x坐标
                int x = (targetWidth - sourceWidth) / 2;
                // 计算源图片在新画布中居中放置时的左上角y坐标
                int y = (targetHeight - sourceHeight) / 2;

                // 4. 绘制源图片到画布上
                // drawImage(Image img, int x, int y, ImageObserver observer)
                // img: 要绘制的图像
                // x, y: 图像左上角的坐标
                // observer: 通常传入null即可，用于通知图像加载进度的对象
                g2d.drawImage(sourceImage, x, y, null);

            } finally {
                // 释放Graphics2D对象占用的系统资源，非常重要！
                g2d.dispose();
            }

            // 5. 保存结果图片
            File outputFile = new File(outputPath);
            // 确保父目录存在
            outputFile.getParentFile().mkdirs(); 
            ImageIO.write(canvasImage, "PNG", outputFile); // 可以根据需要修改输出格式，如"JPG"

            return outputPath;

        } catch (IOException e) {
            e.printStackTrace();
            System.err.println("图片处理过程中发生IO错误：" + e.getMessage());
            return "error";
        } catch (Exception e) {
            e.printStackTrace();
            System.err.println("发生未知错误：" + e.getMessage());
            return "error";
        }
    }

    public static void main(String[] args) throws IOException {
        // 模拟MultipartFile，实际应用中会从请求中获取
        // 这里为了演示，我们创建一个虚拟的MultipartFile
        // 假设有一个名为 "test.jpg" 的图片文件在项目根目录下
        // 请替换为实际的图片路径
        File testImageFile = new File("src/main/resources/test.jpg"); 
        if (!testImageFile.exists()) {
             System.err.println("请在 'src/main/resources/' 目录下放置一个名为 'test.jpg' 的测试图片。");
             return;
        }

        MultipartFile mockMultipartFile = new MockMultipartFile(
                "test.jpg", 
                "test.jpg", 
                "image/jpeg", 
                java.nio.file.Files.readAllBytes(testImageFile.toPath())
        );

        String resultPath = insertImageCentered(mockMultipartFile, 
                                                1920, 1080, 
                                                Color.BLACK, 
                                                "output/centered_image.png");

        if (!"error".equals(resultPath)) {
            System.out.println("图片成功处理并保存到：" + resultPath);
        } else {
            System.out.println("图片处理失败。");
        }
    }

    // 辅助类：模拟MultipartFile
    static class MockMultipartFile implements MultipartFile {
        private final String name;
        private final String originalFilename;
        private final String contentType;
        private final byte[] content;

        public MockMultipartFile(String name, String originalFilename, String contentType, byte[] content) {
            this.name = name;
            this.originalFilename = originalFilename;
            this.contentType = contentType;
            this.content = content;
        }

        @Override
        public String getName() { return name; }
        @Override
        public String getOriginalFilename() { return originalFilename; }
        @Override
        public String getContentType() { return contentType; }
        @Override
        public boolean isEmpty() { return content == null || content.length == 0; }
        @Override
        public long getSize() { return content.length; }
        @Override
        public byte[] getBytes() throws IOException { return content; }
        @Override
        public InputStream getInputStream() throws IOException { return new java.io.ByteArrayInputStream(content); }
        @Override
        public void transferTo(File dest) throws IOException, IllegalStateException {
            java.nio.file.Files.write(dest.toPath(), content);
        }
    }
}