Java中从RGBA字节数组创建图片遇内存溢出问题求解

一、手动构建PNG文件（无需第三方库） #

PNG结构可手动拼接，已知宽高和RGBA原始数据时，直接补全文件头和块信息即可生成合法PNG：

• 步骤：
  1. 写入PNG固定签名：89 50 4E 47 0D 0A 1A 0A
  2. 构建IHDR块：填入宽、高、位深度（8）、颜色类型（6，带Alpha的RGB）等元数据
  3. 对RGBA数据添加行过滤字节（每行开头加0）后，用Deflate压缩
  4. 构建IDAT块封装压缩后的数据
  5. 写入固定结构的IEND块收尾

示例Java代码：

import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.Deflater;

public class RgbaToPng {
    public static void writeRgbaToPng(int width, int height, byte[] rgbaData, String outputPath) throws IOException {
        try (FileOutputStream fos = new FileOutputStream(outputPath)) {
            // 写入PNG签名
            fos.write(new byte[]{(byte) 0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A});

            // 构建并写入IHDR块
            byte[] ihdr = new byte[13];
            // 宽（大端字节序）
            ihdr[0] = (byte) (width >> 24);
            ihdr[1] = (byte) (width >> 16);
            ihdr[2] = (byte) (width >> 8);
            ihdr[3] = (byte) width;
            // 高（大端字节序）
            ihdr[4] = (byte) (height >> 24);
            ihdr[5] = (byte) (height >> 16);
            ihdr[6] = (byte) (height >> 8);
            ihdr[7] = (byte) height;
            ihdr[8] = 8;    // 位深度
            ihdr[9] = 6;    // 带Alpha的RGB颜色类型
            ihdr[10] = 0;   // 压缩方法
            ihdr[11] = 0;   // 过滤方法
            ihdr[12] = 0;   // 非隔行扫描
            writeChunk(fos, "IHDR", ihdr);

            // 给每行数据添加0过滤字节
            byte[] filteredData = new byte[rgbaData.length + height];
            int offset = 0;
            for (int y = 0; y < height; y++) {
                filteredData[offset++] = 0;
                System.arraycopy(rgbaData, y * width * 4, filteredData, offset, width * 4);
                offset += width * 4;
            }

            // 压缩数据并写入IDAT块
            Deflater deflater = new Deflater(Deflater.DEFAULT_COMPRESSION);
            deflater.setInput(filteredData);
            deflater.finish();
            byte[] compressed = new byte[filteredData.length];
            int compLen = deflater.deflate(compressed);
            byte[] actualCompressed = new byte[compLen];
            System.arraycopy(compressed, 0, actualCompressed, 0, compLen);
            writeChunk(fos, "IDAT", actualCompressed);

            // 写入IEND块
            writeChunk(fos, "IEND", new byte[0]);
        }
    }

    private static void writeChunk(FileOutputStream fos, String type, byte[] data) throws IOException {
        // 写入块长度（大端）
        byte[] lenBytes = new byte[4];
        lenBytes[0] = (byte) (data.length >> 24);
        lenBytes[1] = (byte) (data.length >> 16);
        lenBytes[2] = (byte) (data.length >> 8);
        lenBytes[3] = (byte) data.length;
        fos.write(lenBytes);

        // 写入块类型
        fos.write(type.getBytes());

        // 写入块数据
        fos.write(data);

        // 计算并写入CRC校验值
        int crc = 0xFFFFFFFF;
        for (byte b : type.getBytes()) crc = updateCrc(crc, b);
        for (byte b : data) crc = updateCrc(crc, b);
        crc ^= 0xFFFFFFFF;
        byte[] crcBytes = new byte[4];
        crcBytes[0] = (byte) (crc >> 24);
        crcBytes[1] = (byte) (crc >> 16);
        crcBytes[2] = (byte) (crc >> 8);
        crcBytes[3] = (byte) crc;
        fos.write(crcBytes);
    }

    private static int updateCrc(int crc, byte b) {
        crc ^= b & 0xFF;
        for (int i = 0; i < 8; i++) {
            crc = (crc & 1) != 0 ? (crc >> 1) ^ 0xEDB88320 : crc >> 1;
        }
        return crc;
    }

    public static void main(String[] args) throws IOException {
        // 测试用RGBA数据：100x100半透明红色
        int w = 100, h = 100;
        byte[] rgba = new byte[w * h * 4];
        for (int i = 0; i < rgba.length; i += 4) {
            rgba[i] = (byte)255; rgba[i+1] = 0; rgba[i+2] = 0; rgba[i+3] = (byte)127;
        }
        writeRgbaToPng(w, h, rgba, "output.png");
    }
}

二、替换stb_image的内存分配方式 #

如果坚持用stb_image，问题出在MemoryStack是栈内存，容量受限。改用堆分配的直接ByteBuffer即可解决OOM：

// 用堆外内存分配Buffer，替代MemoryStack
ByteBuffer buffer = ByteBuffer.allocateDirect(rgbaData.length);
buffer.put(rgbaData);
buffer.flip();
// 调用stb_image_write写入图片，以PNG为例
stbi_write_png("output.png", width, height, 4, buffer, width * 4);

三、用Java ImageIO快速生成图片 #

借助BufferedImage包装RGBA数据，让ImageIO自动处理文件头，支持PNG、BMP等格式（注意JPEG不支持Alpha通道）：

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import javax.imageio.ImageIO;

public class RgbaToImage {
    public static void writeRgbaToImage(int width, int height, byte[] rgba, String path, String format) throws IOException {
        BufferedImage img = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
        int[] pixels = new int[width * height];
        for (int i = 0; i < pixels.length; i++) {
            int r = rgba[i*4] & 0xFF;
            int g = rgba[i*4+1] & 0xFF;
            int b = rgba[i*4+2] & 0xFF;
            int a = rgba[i*4+3] & 0xFF;
            pixels[i] = (a << 24) | (r << 16) | (g << 8) | b;
        }
        img.setRGB(0, 0, width, height, pixels, 0, width);
        ImageIO.write(img, format, new File(path));
    }

    public static void main(String[] args) throws IOException {
        int w = 100, h = 100;
        byte[] rgba = new byte[w * h * 4];
        for (int i = 0; i < rgba.length; i += 4) {
            rgba[i] = (byte)255; rgba[i+1] = 0; rgba[i+2] = 0; rgba[i+3] = (byte)127;
        }
        writeRgbaToImage(w, h, rgba, "output.png", "PNG");
    }
}

内容的提问来源于stack exchange，提问作者v22