C++如何处理图像？STB库入门教程

如何用C++的STB库加载和处理图像？1.引入STB库需下载stb_image.h并定义STB_IMAGE_IMPLEMENTATION宏后包含；2.使用stbi_load()函数加载图像获取宽高通道及像素数据；3.图像数据以一维数组存储RGB或RGBA格式像素；4.处理完需调用stbi_image_free()释放内存；5.注意路径正确性跨平台特性及不支持写入等问题。

C++处理图像文件的方式有很多，其中使用第三方库是最常见的做法。STB库（特别是stb_image.h）是一个轻量级、单头文件的图像加载库，非常适合入门和快速开发。它支持多种常见图像格式（如PNG、JPEG、BMP等），而且使用起来非常方便。

下面我们就来一步步了解怎么用STB库在C++中加载和处理图像文件。

如何引入STB图像库？

要使用STB库，首先需要获取它的头文件。最常用的是 stb_image.h，你可以从 https://github.com/nothings/stb 下载。

使用时只需将 stb_image.h 文件放到你的项目目录中，在需要的地方包含它即可：

#define STB_IMAGE_IMPLEMENTATION
#include "stb_image.h"

注意：必须定义 STB_IMAGE_IMPLEMENTATION 宏一次，否则无法编译通过。通常放在主源文件或一个单独的实现文件里。

怎么加载一张图片？

加载图像的核心函数是 stbi_load()，它的基本用法如下：

int width, height, channels;
unsigned char* data = stbi_load("image.png", &width, &height, &channels, 0);

• "image.png" 是你要加载的图像路径。
• width 和 height 会返回图像的宽高。
• channels 表示图像的通道数（比如RGB是3，RGBA是4）。
• 最后一个参数可以指定输出的通道数，传0表示自动识别。

如果加载失败，data 会是 nullptr，记得判断一下：

if (!data) {
    std::cerr << "Failed to load image!" << std::endl;
    return -1;
}

加载成功后，data 就指向图像数据的首地址，可以用循环访问每个像素点。

图像数据是怎么存储的？

图像数据是以一维数组的形式保存的，每个像素由连续的几个字节表示。例如，对于RGB图像来说，每三个字节分别代表红、绿、蓝分量。

假设你加载了一张宽为 w、高为 h 的RGB图像：

for (int y = 0; y < h; ++y) {
    for (int x = 0; x < w; ++x) {
        int index = (y * w + x) * 3;
        unsigned char r = data[index];
        unsigned char g = data[index + 1];
        unsigned char b = data[index + 2];
        // 这里可以对像素进行处理
    }
}

如果你加载的是RGBA图像，那么每个像素占4个字节，顺序是R、G、B、A。

处理完图像之后，别忘了调用 stbi_image_free(data) 来释放内存：

stbi_image_free(data);

常见问题和注意事项

• 路径问题：确保传给 stbi_load() 的路径是正确的，相对路径或绝对路径都可以，但要注意当前工作目录的问题。
• 多线程安全：STB图像是线程安全的，但最好避免多个线程同时加载图像。
• 不支持写入：STB主要用于读取图像，如果你想保存处理后的图像，需要用到其他库，比如 stb_image_write.h。
• 跨平台兼容性好：STB是纯C代码，几乎可以在任何平台上编译使用。

基本上就这些内容了。STB库虽然简单，但在图像处理教学和小型项目中非常实用。不需要复杂的配置，也不依赖外部库，只要把头文件放进去就能直接用了。刚开始学的时候建议先试着加载图像并打印出宽高和通道信息，再逐步尝试修改像素颜色或者保存新图像。