将 CGImageRef 保存为 png 文件？

Question

在我的 Cocoa 应用程序中，我从磁盘加载一个 .jpg 文件，并对其进行操作。现在需要将其作为 .png 文件写入磁盘。你怎么能这么做呢？

感谢您的帮助！

Answer 1

使用 CGImageDestination 并传递 kUTTypePNG 是正确的方法。以下是一个快速片段：

@import MobileCoreServices; // or `@import CoreServices;` on Mac
@import ImageIO;

BOOL CGImageWriteToFile(CGImageRef image, NSString *path) {
    CFURLRef url = (__bridge CFURLRef)[NSURL fileURLWithPath:path];
    CGImageDestinationRef destination = CGImageDestinationCreateWithURL(url, kUTTypePNG, 1, NULL);
    if (!destination) {
        NSLog(@"Failed to create CGImageDestination for %@", path);
        return NO;
    }

    CGImageDestinationAddImage(destination, image, nil);

    if (!CGImageDestinationFinalize(destination)) {
        NSLog(@"Failed to write image to %@", path);
        CFRelease(destination);
        return NO;
    }

    CFRelease(destination);
    return YES;
}

您需要将 ImageIO 和 CoreServices（或 iOS 上的 MobileCoreServices）添加到项目中并包含标头。

---

如果您使用的是 iOS 并且不需要适用于 Mac 的解决方案，则可以使用更简单的方法：

// `image` is a CGImageRef
// `path` is a NSString with the path to where you want to save it
[UIImagePNGRepresentation([UIImage imageWithCGImage:image]) writeToFile:path atomically:YES];

在我的测试中，ImageIO 方法比 UIImage 方法快大约 10%我的 iPhone 5s。在模拟器中，UIImage 方法更快。如果您确实关心性能，那么可能值得针对您在设备上的特定情况进行测试。

Answer 2

这是 macOS 友好的 Swift 3 & 4 示例：

@discardableResult func writeCGImage(_ image: CGImage, to destinationURL: URL) -> Bool {
    guard let destination = CGImageDestinationCreateWithURL(destinationURL as CFURL, kUTTypePNG, 1, nil) else { return false }
    CGImageDestinationAddImage(destination, image, nil)
    return CGImageDestinationFinalize(destination)
}

Answer 3

创建一个CGImageDestination，传递kUTTypePNG作为要创建的文件类型。添加图像，然后最终确定目的地。

Answer 4

Swift 5+采用版本

import Foundation
import CoreGraphics
import CoreImage
import ImageIO
import MobileCoreServices

extension CIImage {
  
  public func convertToCGImage() -> CGImage? {
    let context = CIContext(options: nil)
    if let cgImage = context.createCGImage(self, from: self.extent) {
      return cgImage
    }
    return nil
  }
  
  public func data() -> Data? {
    convertToCGImage()?.pngData()
  }
}

extension CGImage {
  
  public func pngData() -> Data? {
    let cfdata: CFMutableData = CFDataCreateMutable(nil, 0)
    if let destination = CGImageDestinationCreateWithData(cfdata, kUTTypePNG as CFString, 1, nil) {
      CGImageDestinationAddImage(destination, self, nil)
      if CGImageDestinationFinalize(destination) {
        return cfdata as Data
      }
    }
    
    return nil
  }
}

Answer 5

提供的解决方案可能仍然可以正常工作，但有一些更新的 API CoreImage 中的 this 具有相同的功能，并且使用起来更加“Swifty”：

import CoreImage

func write(cgimage: CGImage, to url: URL) throws {
    let cicontext = CIContext()
    let ciimage = CIImage(cgImage: cgimage)
    try cicontext.writePNGRepresentation(of: ciimage, to: url, format: .RGBA8, colorSpace: ciimage.colorSpace!)
}