如何快速检测NSImage是否为模板图像的最优方法

假设你拥有一个带NSBitmapImageRep（光栅图像）的16x16像素NSImage，它既可以是彩色图像，也可能仅包含带Alpha通道的黑色像素。当图像仅含黑色像素时，我们可为该NSImage设置.isTemplate属性并进行相应处理。目前的核心问题是：检测给定图像是否为模板图像（仅由黑色像素构成）的最快方式是什么？

我目前的实现方案虽可行，但需要逐个遍历所有像素进行检查。即使是16x16尺寸的图像，处理10-20张也约需一秒钟，因此正在寻找更优化的实现方式：

+ (BOOL)detectImageIsTemplate:(NSImage *)image
{
    BOOL result = NO;
    
    if (image)
    {
        // 先假设是模板图像，直到找到非黑色像素
        result = YES;
        
        NSSize imageSize = image.size;
        NSRect imageRect = NSMakeRect(0, 0, imageSize.width, imageSize.height);
        CGColorSpaceRef colorSpace = CGColorSpaceCreateWithName(kCGColorSpaceGenericRGB);
        
        CGContextRef ctx = CGBitmapContextCreate(NULL,
                                                 imageSize.width,
                                                 imageSize.height,
                                                 8,
                                                 0,
                                                 colorSpace,
                                                 kCGImageAlphaPremultipliedLast);
        
        NSGraphicsContext* gctx = [NSGraphicsContext graphicsContextWithCGContext:ctx flipped:NO];
        [NSGraphicsContext setCurrentContext:gctx];
        [image drawInRect:imageRect];
        
        // ......................................................
        
        size_t width = CGBitmapContextGetWidth(ctx);
        size_t height = CGBitmapContextGetHeight(ctx);

        uint32_t* pixel = (uint32_t*)CGBitmapContextGetData(ctx);

        for (unsigned y = 0; y < height; y++)
        {
            for (unsigned x = 0; x < width; x++)
            {
                uint32_t rgba = *pixel;

                uint8_t red   = (rgba & 0x000000ff) >> 0;
                uint8_t green = (rgba & 0x0000ff00) >> 8;
                uint8_t blue  = (rgba & 0x00ff0000) >> 16;
                
                if (red != 0 || green != 0 || blue != 0)
                {
                    result = NO;
                    break;
                }

                pixel++; // 下一个像素
            }
            
            if (result == NO) break;
        }
        
        // ......................................................
        
        [NSGraphicsContext setCurrentContext:nil];
        CGContextRelease(ctx);
        CGColorSpaceRelease(colorSpace);
    }
    
    return result;
}

优化方案 #

1. 直接复用NSImage已有的NSBitmapImageRep，避免重复绘制 #

现有代码将图像重新绘制到CGContext的操作会带来额外开销，可直接从NSImage中提取已有的NSBitmapImageRep，直接读取像素数据：

+ (BOOL)detectImageIsTemplate:(NSImage *)image
{
    if (!image) return NO;
    
    // 获取图像最优的BitmapRep
    NSBitmapImageRep *bitmapRep = [image bestRepresentationForRect:NSMakeRect(0,0,image.size.width,image.size.height) context:nil hints:nil];
    if (!bitmapRep) return NO;
    
    BOOL isTemplate = YES;
    size_t width = bitmapRep.pixelsWide;
    size_t height = bitmapRep.pixelsHigh;
    size_t bytesPerPixel = bitmapRep.bitsPerPixel / 8;
    unsigned char *pixelData = bitmapRep.bitmapData;
    
    for (size_t y = 0; y < height; y++) {
        for (size_t x = 0; x < width; x++) {
            size_t byteIndex = (y * bitmapRep.bytesPerRow) + (x * bytesPerPixel);
            // 这里假设为RGBA格式，若为BGRA需调整通道索引
            uint8_t red = pixelData[byteIndex];
            uint8_t green = pixelData[byteIndex + 1];
            uint8_t blue = pixelData[byteIndex + 2];
            
            if (red != 0 || green != 0 || blue != 0) {
                isTemplate = NO;
                goto endCheck; // 直接跳出所有循环，减少嵌套判断开销
            }
        }
    }
    
endCheck:
    return isTemplate;
}

2. 利用SIMD指令批量检查像素 #

针对16x16的小图像，可使用Apple平台的NEON SIMD指令一次性批量检查多个像素，大幅提升处理速度：

#import <Accelerate/Accelerate.h>

+ (BOOL)detectImageIsTemplateWithSIMD:(NSImage *)image
{
    if (!image) return NO;
    
    NSBitmapImageRep *bitmapRep = [image bestRepresentationForRect:NSMakeRect(0,0,image.size.width,image.size.height) context:nil hints:nil];
    if (!bitmapRep) return NO;
    
    size_t width = bitmapRep.pixelsWide;
    size_t height = bitmapRep.pixelsHigh;
    size_t bytesPerRow = bitmapRep.bytesPerRow;
    unsigned char *pixelData = bitmapRep.bitmapData;
    
    // 按RGBA格式处理，一次批量检查8个像素
    const size_t pixelsPerBatch = 8;
    const size_t bytesPerBatch = pixelsPerBatch * 4;
    
    // 创建全0的对比向量
    uint8x8x3_t zeroVector = {
        .val[0] = vdup_n_u8(0),
        .val[1] = vdup_n_u8(0),
        .val[2] = vdup_n_u8(0)
    };
    
    for (size_t y = 0; y < height; y++) {
        unsigned char *rowStart = pixelData + (y * bytesPerRow);
        size_t remainingPixels = width;
        
        // 批量处理8个像素的批次
        while (remainingPixels >= pixelsPerBatch) {
            uint8x8x3_t rgbPixels = vld3_u8(rowStart);
            
            // 对比RGB通道是否全为0
            uint8x8_t redEqual = vceqq_u8(rgbPixels.val[0], zeroVector.val[0]);
            uint8x8_t greenEqual = vceqq_u8(rgbPixels.val[1], zeroVector.val[1]);
            uint8x8_t blueEqual = vceqq_u8(rgbPixels.val[2], zeroVector.val[2]);
            
            uint8x8_t allEqual = vandq_u8(vandq_u8(redEqual, greenEqual), blueEqual);
            
            // 只要有一个像素非黑，直接返回NO
            if (vmaxv_u8(allEqual) != 0xff) {
                return NO;
            }
            
            rowStart += bytesPerBatch;
            remainingPixels -= pixelsPerBatch;
        }
        
        // 处理剩余不足8个的像素
        for (size_t x = 0; x < remainingPixels; x++) {
            size_t byteIndex = x * 4;
            uint8_t red = rowStart[byteIndex];
            uint8_t green = rowStart[byteIndex + 1];
            uint8_t blue = rowStart[byteIndex + 2];
            
            if (red != 0 || green != 0 || blue != 0) {
                return NO;
            }
        }
    }
    
    return YES;
}

3. 额外注意事项 #

• 需根据NSBitmapImageRep的色彩空间调整RGB通道的索引（比如BGRA格式下，蓝色通道在第一个位置），可通过bitmapRep.colorSpace判断。
• 模板图像仅要求RGB通道为0，Alpha通道不影响判断。

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