要转换为UIImage的CMSampleBuffer扩展名:内存太贵

时间:2017-07-06 19:43:53

标签: ios swift uiimage cmsamplebuffer

我正在尝试使用Swift 3.0将CMSampleBuffer转换为UIImage。一种流行的解决方案是为CMSampleBuffer类编写扩展,并添加一个getter来将缓冲区转换为图像。这就是它的样子:

import Foundation
import AVFoundation

extension CMSampleBuffer {
    @available(iOS 9.0, *)
    var uiImage: UIImage? {
    guard let imageBuffer = CMSampleBufferGetImageBuffer(self) else { return nil }

    let ciimage: CIImage = CIImage(cvImageBuffer: imageBuffer)


    let image:UIImage = UIImage(ciImage: ciimage)


    return image
    }

}

它工作正常,但它占用了大量内存,占应用内存总量的40%。是否有更高效的内存解决方案?

修改

我已经更改了我的代码,它看起来像这样:

var uiImage: UIImage? {
    guard let imageBuffer = CMSampleBufferGetImageBuffer(self) else { return nil }

    CVPixelBufferLockBaseAddress(imageBuffer, CVPixelBufferLockFlags(rawValue: 0))
    let baseAddress = CVPixelBufferGetBaseAddress(imageBuffer)
    let bytesPerRow = CVPixelBufferGetBytesPerRow(imageBuffer)
    let width = CVPixelBufferGetWidth(imageBuffer)
    let height = CVPixelBufferGetHeight(imageBuffer)
    let colorSpace = CGColorSpaceCreateDeviceRGB()
    let bitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.noneSkipFirst.rawValue | CGBitmapInfo.byteOrder32Little.rawValue)


    var image: UIImage?
    autoreleasepool(invoking: {() -> () in
        guard let context = CGContext(data: baseAddress,
        width: width,
        height: height,
        bitsPerComponent: 8,
        bytesPerRow: bytesPerRow,
        space: colorSpace,
        bitmapInfo: bitmapInfo.rawValue) else { return }
        guard let cgImage = context.makeImage() else { return }
        image = UIImage(cgImage: cgImage)
    })    
    CVPixelBufferUnlockBaseAddress(imageBuffer,CVPixelBufferLockFlags(rawValue: 0));



    return image
}

内存泄漏与CGContext有关。除了使用自动释放池之外,还有其他方法可以释放/释放/解除分配吗?

0 个答案:

没有答案