基于C语言ARM NEON的RGB转灰度图异常问题求助

针对你用ARM NEON实现RGB转灰度BMP时出现的「一半黑色+另一半三条纹」问题，核心原因大概率是内存指针操作错误、BMP格式规则未遵守或NEON向量处理逻辑错位，以下是具体排查和修复方案：

1. 修正BMP行对齐处理 #

BMP要求每行像素数据必须4字节对齐，忽略填充字节会导致读写错位：

• 计算源图像每行实际字节数：src_stride = ((width * 3) + 3) & ~3;
• 计算目标灰度图像每行实际字节数：dst_stride = ((width * 1) + 3) & ~3;
• 循环处理每行时，必须基于stride而非width*3/width*1移动指针，避免读取下一行错误数据或写入越界。

2. 修复NEON向量的内存操作逻辑 #

你的现象大概率是指针移动错误导致：原本每3字节RGB生成1字节灰度，但代码可能错误地让目标指针每次移动3字节，只使用了1/3的目标图像空间，剩余区域为默认黑色，写入区域则因通道重复形成条纹。

正确的NEON处理代码（针对BGR转灰度） #

// 定点权重：对应Y=0.299R+0.587G+0.114B，转10位整数运算避免浮点开销
const uint16x8_t coeff_b = vdupq_n_u16(117);  // 0.114*1024≈117
const uint16x8_t coeff_g = vdupq_n_u16(601);  // 0.587*1024≈601
const uint16x8_t coeff_r = vdupq_n_u16(306);  // 0.299*1024≈306

int remaining_pixels = width * height;
uint8_t* src_ptr = src_image_data;
uint8_t* dst_ptr = dst_image_data;

// 处理开头未对齐的像素（NEON要求16字节对齐）
while ((uintptr_t)src_ptr % 16 != 0 && remaining_pixels > 0) {
    uint8_t b = *src_ptr++;
    uint8_t g = *src_ptr++;
    uint8_t r = *src_ptr++;
    uint8_t gray = (r * 306 + g * 601 + b * 117) >> 10;
    *dst_ptr++ = gray;
    remaining_pixels--;
}

// 批量处理对齐的像素块（每次8个像素）
while (remaining_pixels >= 8) {
    // 加载8个像素的B、G、R通道（BMP存储顺序是BGR）
    uint8x8x3_t bgr = vld3_u8(src_ptr);
    
    // 转换为16位整数防止溢出
    uint16x8_t b = vmovl_u8(bgr.val[0]);
    uint16x8_t g = vmovl_u8(bgr.val[1]);
    uint16x8_t r = vmovl_u8(bgr.val[2]);
    
    // 计算灰度并转回8位
    uint16x8_t gray_16 = vmlaq_u16(vmlaq_u16(vmulq_u16(b, coeff_b), g, coeff_g), r, coeff_r);
    uint8x8_t gray_8 = vshrn_n_u16(gray_16, 10);
    
    // 写入灰度数据
    vst1_u8(dst_ptr, gray_8);
    
    // 指针移动：源指针走8*3=24字节，目标指针走8字节
    src_ptr += 24;
    dst_ptr += 8;
    remaining_pixels -= 8;
}

// 处理末尾剩余的像素
while (remaining_pixels > 0) {
    uint8_t b = *src_ptr++;
    uint8_t g = *src_ptr++;
    uint8_t r = *src_ptr++;
    uint8_t gray = (r * 306 + g * 601 + b * 117) >> 10;
    *dst_ptr++ = gray;
    remaining_pixels--;
}

3. 验证灰度BMP文件头配置 #

确保生成的灰度BMP文件头正确：

• 位深度设为8位；
• 生成256级灰度调色板（每个项的R/G/B值相同）；
• 正确计算文件大小、图像数据偏移量等字段，否则图像显示会错位。

调色板生成示例 #

typedef struct {
    uint8_t blue;
    uint8_t green;
    uint8_t red;
    uint8_t reserved;
} BMPPaletteEntry;

BMPPaletteEntry palette[256];
for (int i = 0; i < 256; i++) {
    palette[i].red = i;
    palette[i].green = i;
    palette[i].blue = i;
    palette[i].reserved = 0;
}

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