iOS - 将 UIImage 转换为 vImage 内存处理

Question

我有一个函数可以将UIImage 转换为vImage（用于iOS 的Accelerate.framework 中的一些vImage 方法。

我的方法是：

-(vImage_Buffer)convertImage:(UIImage *)image {
    CGImageRef sourceRef = [image CGImage];
    NSUInteger sourceWidth = CGImageGetWidth(sourceRef);
    NSUInteger sourceHeight = CGImageGetHeight(sourceRef);
    CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
    unsigned char *sourceData = (unsigned char*)calloc(sourceHeight * sourceWidth * 4, sizeof(unsigned char));
    NSUInteger bytesPerPixel = 4;
    NSUInteger sourceBytesPerRow = bytesPerPixel * sourceWidth;
    NSUInteger bitsPerComponent = 8;
    CGContextRef context = CGBitmapContextCreate(sourceData, sourceWidth, sourceHeight,
                                                       bitsPerComponent, sourceBytesPerRow, colorSpace,
                                                       kCGImageAlphaPremultipliedFirst | kCGBitmapByteOrder32Big);
    CGContextDrawImage(context, CGRectMake(0, 0, sourceWidth, sourceHeight), sourceRef);
    CGContextRelease(context);
    vImage_Buffer v_image = {
        .data = sourceData,
        .height = sourceHeight,
        .width = sourceWidth,
        .rowBytes = sourceBytesPerRow
    };

    return v_image;
}

我通过在网络上混合和匹配一些代码 sn-ps 来构建它。

我的问题是，我有一个calloc 调用，它为sourceData 分配空间。但是如何以及在哪里释放这些内存？

鉴于我可能会多次调用此方法，看起来像原样，会有巨大的内存泄漏。我不认为我会在这里发布它，因为它为我的vImage_Buffer 变量提供了支持.data。但是我会在我的调用函数中释放它吗？或者将v_image 设置为nil（或在调用方法中命名的任何返回变量）最终会释放calloc 行分配的内存？

有人指点一下吗？

Answer 1

首先你还需要通过调用CGColorSpaceRelease(colorSpace)释放colorSpace

calloc 分配的内存稍后可以在调用函数中使用free(v_image.data)（或调用方法中命名的任何返回变量）释放。

您还可以更改您的方法实现以摆脱 CGContext 并使用 CGDataProviderRef 从 CGImage 获取字节，这将类似于：

-(vImage_Buffer)convertImage:(UIImage *)image
{
    CGImageRef sourceRef = [image CGImage];
    NSUInteger sourceWidth = CGImageGetWidth(sourceRef);
    NSUInteger sourceHeight = CGImageGetHeight(sourceRef);

    CGDataProviderRef provider = CGImageGetDataProvider(sourceRef);
    CFDataRef bitmapData = CGDataProviderCopyData(provider);

    unsigned char *sourceData = (unsigned char*)calloc(sourceHeight * sourceWidth * 4, sizeof(unsigned char));
    NSUInteger bytesPerPixel = 4;
    NSUInteger sourceBytesPerRow = bytesPerPixel * sourceWidth;

    CFDataGetBytes(bitmapData, CFRangeMake(0, CFDataGetLength(bitmapData)), sourceData);

    vImage_Buffer v_image = {
        .data = (void *)sourceData,
        .height = sourceHeight,
        .width = sourceWidth,
        .rowBytes = sourceBytesPerRow
    };

    CFRelease(bitmapData);

    return v_image;
}

但是我没有检查性能。

我还建议将您的方法重命名为 vImageCreateFromImage 之类的名称。稍后你会知道你负责清理内存。

Answer 2

首先，也是最重要的，这在 iOS7.0 / OSX 10.9 中得到了极大的简化，引入了 vImageBuffer_InitWithCGImage 函数（在 vImage_Utilities.h 中声明，标题中包含相当广泛的文档）。如果您不需要支持较旧的操作系统，则应考虑使用此新功能。

其次，无论您使用现有代码还是InitWithCGImage，一旦您完成使用通过任一方式创建的vImage_Buffer 对象，您应该释放与数据字段相关联的内存：free(buffer.data);。