Swift中将RGB像素数组转为CGImage时行丢失问题排查

你的代码存在两个关键计算错误，导致CGImage解析像素数据时的步长和总长度匹配异常：

错误点1：bytesPerRow参数类型错误 #

CGImage初始化时的bytesPerRow要求传入每行的字节数，但你传入的是width * bitsPerPixel（这是每行的比特数）。正确计算应为每行像素数 × 单个像素的字节数：

bytesPerRow: width * MemoryLayout<PixelData>.stride

也可以等价写成width * bitsPerPixel / 8（32比特=4字节）。

错误点2：CGDataProvider的length参数单位错误 #

NSData(bytes: &data, length: ...)的length需要传入总字节数，但你传入的是data.count * bitsPerPixel（总比特数）。正确计算是像素总数 × 单个像素的字节数：

let providerRef = CGDataProvider(data: NSData(bytes: &data, length: data.count * MemoryLayout<PixelData>.stride))

修复后的pixeldata_to_image函数 #

func pixeldata_to_image(pixels: [PixelData], width: Int, height: Int)->CGImage
{
    assert(pixels.count == width * height)
    let bitsPerComponent: Int = 8
    let bitsPerPixel: Int = 32
    let rgbColorSpace = CGColorSpaceCreateDeviceRGB()
    let bitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedFirst.rawValue)

    var data = pixels
    let pixelStride = MemoryLayout<PixelData>.stride
    guard
        let providerRef = CGDataProvider(data: NSData(bytes: &data, length: data.count * pixelStride))
    else
    {
        fatalError("CGDataProvider failure")
    }
    guard
        let image = CGImage(
            width: width,
            height: height,
            bitsPerComponent: bitsPerComponent,
            bitsPerPixel: bitsPerPixel,
            bytesPerRow: width * pixelStride,
            space: rgbColorSpace,
            bitmapInfo: bitmapInfo,
            provider: providerRef,
            decode: nil,
            shouldInterpolate: true,
            intent: .defaultIntent
        )
    else
    {
        fatalError("CGImage failure")
    }
    return image
}

这种逐像素操作的场景和效率需要分情况判断：

适用场景 #

当你需要精确控制每个像素的颜色值（比如自定义图像处理算法、生成程序化纹理、实现特定像素级滤镜）时，逐像素操作是必要的选择。

效率优化方向 #

1. 减少内存拷贝开销：
   原代码中var data = pixels会拷贝整个像素数组，可改用withUnsafeBytes直接访问原数组内存，避免不必要的拷贝：

   pixels.withUnsafeBytes { buffer in
       guard let baseAddress = buffer.baseAddress else {
           fatalError("Pixel buffer empty")
       }
       let providerRef = CGDataProvider(data: NSData(bytes: baseAddress, length: buffer.count))
       // 后续CGImage初始化逻辑...
   }
   

2. 优先使用硬件加速API：
   如果你的需求是绘制线条、边框、形状等常规图形，直接用CGContext的strokePath()、fillPath()等方法更高效——这些API是GPU硬件加速的，性能远优于CPU逐像素计算，代码也更简洁。比如你示例中的对角线、边框、中心线，完全可以用路径绘制替代逐像素设置。

3. 大尺寸场景用Metal加速：
   处理大尺寸图像（如4K及以上）或需要实时像素处理时，使用Metal框架将计算逻辑放到GPU并行执行，效率会比CPU逐像素操作提升数倍甚至数十倍。

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