iOS 相册和网络图片的存取

﹏ヽ暗。殇╰゛Y 2022-06-17 12:26 184阅读 0赞

## 保存 UIImage 到相册 ##

### UIKit ###

UIKit 中一个古老的方法，Objective-C 的形式

void UIImageWriteToSavedPhotosAlbum(UIImage *image, id completionTarget, SEL completionSelector, void *contextInfo);

保存完成后，会调用 completionTarget 的 completionSelector。如果 completionTarget 不为空，completionTarget 必须实现以下方法

- (void)image:(UIImage *)image didFinishSavingWithError:(NSError *)error contextInfo:(void *)contextInfo;

Objective-C 的写法

- (void)saveImage:(UIImage *)image {
        UIImageWriteToSavedPhotosAlbum(image, self, @selector(image:didFinishSavingWithError:contextInfo:), nil); 
    }
    
    - (void)image:(UIImage *)image didFinishSavingWithError:(NSError *)error contextInfo:(void *)contextInfo {
        if (error) {
            // Fail
        } else {
            // Success
        }
    }

Swift 的写法

func saveImage(_ image: UIImage) {
        UIImageWriteToSavedPhotosAlbum(image, self, #selector(image(_:didFinishSavingWithError:contextInfo:)), nil)
    }
    
    func image(_ image: UIImage, didFinishSavingWithError error: NSError?, contextInfo: AnyObject) {
        if error == nil {
            // Success
        } else {
            // Fail
        }
    }

### Photos framework ###

iOS 8 开始，可以用 Photos framework。PHAssetChangeRequest 的类方法可以保存 UIImage

class func creationRequestForAsset(from image: UIImage) -> Self

编辑相册需要在 PHPhotoLibrary 的闭包中进行，有两种方法

func performChanges(_ changeBlock: @escaping () -> Void, completionHandler: ((Bool, Error?) -> Void)? = nil)

func performChangesAndWait(_ changeBlock: @escaping () -> Void) throws

以上两种方法，分别是异步和同步执行。一般用第一种异步执行的方法，不会阻塞主线程。

func saveImage(_ image: UIImage) {
        PHPhotoLibrary.shared().performChanges({ 
            PHAssetChangeRequest.creationRequestForAsset(from: image)
        }, completionHandler: { (success, error) in
            // NOT on main thread
            if success {
                // Success
            } else if let error = error {
                // Handle error
            }
        })
    }

**编辑相册的闭包 changeBlock 和完成的闭包 completionHandler，是在 serial queue 中执行，不在主线程**。需要更新 UI 的话，要切换到主线程中执行。

## 保存图片的 Data 到相册 ##

如果有图片的数据(Data 或 NSData)，可以用 Photos framework 的方法保存到相册。从 iOS 9 开始，可以使用 PHAssetCreationRequest 的方法

func addResource(with type: PHAssetResourceType, data: Data, options: PHAssetResourceCreationOptions?)

iOS 8 比较麻烦，需要把数据写入临时文件，用临时文件的 URL 作为参数，调用 PHAssetChangeRequest 的类方法

class func creationRequestForAssetFromImage(atFileURL fileURL: URL) -> Self?

以下是兼容 iOS 8 的写法

func saveImageData(_ data: Data) {
        if #available(iOS 9.0, *) {
            PHPhotoLibrary.shared().performChanges({
                PHAssetCreationRequest.forAsset().addResource(with: .photo, data: data, options: nil)
            }, completionHandler: { (success, error) in
                // NOT on main thread
                if success {
                    // Success
                } else if let error = error {
                    // Handle error
                }
            })
        } else {
            // Write image data to temp file
            let tempPath = NSTemporaryDirectory().appending("TempImageToSaveToPhoto.image")
            let tempUrl = URL(fileURLWithPath: tempPath)
            try? data.write(to: tempUrl)
            
            PHPhotoLibrary.shared().performChanges({
                PHAssetChangeRequest.creationRequestForAssetFromImage(atFileURL: tempUrl)
            }, completionHandler: { (success, error) in
                // NOT on main thread
                if success {
                    // Success
                } else if let error = error {
                    // Handle error
                }
                // Remove temp file
                try? FileManager.default.removeItem(at: tempUrl)
            })
        }
    }

## SDWebImage 缓存 UIImage、Data ##

SDWebImage (目前版本 4.0.0) 有两个方法可以使用。

SDWebImageManager 的方法

- (void)saveImageToCache:(nullable UIImage *)image forURL:(nullable NSURL *)url;

SDImageCache 的方法

- (void)storeImage:(nullable UIImage *)image
             imageData:(nullable NSData *)imageData
                forKey:(nullable NSString *)key
                toDisk:(BOOL)toDisk
            completion:(nullable SDWebImageNoParamsBlock)completionBlock;

![1089786-20170425194117631-1601821212.png][]

这个方法的 image、key 参数不能为空，否则直接执行 completionBlock 就返回。

## 从相册获取 UIImage、Data ##

UIImagePickerController 是常用的照片选取控制器。实现一个代理方法即可

optional func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [String : Any])

通过 info 字典，可以获取 UIImage 等信息。这里用来查询 info 字典的 key 有

UIImagePickerControllerOriginalImage // 原始 UIImage
    UIImagePickerControllerEditedImage // 编辑后的 UIImage
    UIImagePickerControllerReferenceURL // ALAsset 的 URL

通过 ALAsset 的 URL 可获取 PHAsset。通过 PHImageManager 的方法可以获得相册图片的 Data

func requestImageData(for asset: PHAsset, options: PHImageRequestOptions?, resultHandler: @escaping (Data?, String?, UIImageOrientation, [AnyHashable : Any]?) -> Void) -> PHImageRequestID

以下是代码示例

func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [String : Any]) {
        picker.dismiss(animated: true, completion: nil)
        
        if let image = info[UIImagePickerControllerOriginalImage] as? UIImage {
            // Get original image
        }
        
        if let url = info[UIImagePickerControllerReferenceURL] as? URL,
            let asset = PHAsset.fetchAssets(withALAssetURLs: [url], options: nil).firstObject {
            PHImageManager.default().requestImageData(for: asset, options: nil, resultHandler: { (imageData, _, _, _) in
                if let data = imageData {
                    // Get image data
                }
            })
        }
    }

## 从 SDWebImage 的缓存中获取 UIImage、Data ##

SDWebImage 给 UIImageView 提供了方法，方便获取、显示网络图片。如果需要获取下载的图片(进行保存到相册、上传至服务器等操作)，可以用以下方法

- (nullable id <SDWebImageOperation>)loadImageWithURL:(nullable NSURL *)url
                                                  options:(SDWebImageOptions)options
                                                 progress:(nullable SDWebImageDownloaderProgressBlock)progressBlock
                                                completed:(nullable SDInternalCompletionBlock)completedBlock;

Swift 的代码示例

SDWebImageManager.shared().loadImage(with: url, options: SDWebImageOptions(rawValue: 0), progress: nil, completed: { [weak self] (cachedImage, imageData, error, _, _, _) in
        guard self != nil else { return }
        
        if let image = cachedImage {
            // Get image
        }
        if let data = imageData {
            // Get image data
        }
        if error != nil {
            // Handle error
        }
    })

这个方法有个问题，对于静态图片，可能获取不到 Data。如果需要获取图片 Data 的话，不能直接这么写。查看源码可以找到原因。SDWebImageManager 的 loadImage: 方法会调用 SDImageCache 的 queryCacheOperationForKey: 方法

![1089786-20170425194137131-785031456.png][]

diskImageDataBySearchingAllPathsForKey: 方法用来获取 Disk 中图片的 Data。**当图片在 Memory 中，只有 GIF 图片才会提供 Data，静态图的 Data 为空**；当图片在 Disk 中，都会提供 Data。如果能在外部直接调用 diskImageDataBySearchingAllPathsForKey: 方法就很简单，但是不行，这是私有方法，只写在 .m 文件里，对外不可见。

改源码可以解决问题，将上图第一个箭头的 if 判断去掉，总是调用 diskImageDataBySearchingAllPathsForKey: 方法。然而，改第三方库源码不好，可能会有想不到的糟糕后果。

一种方法是，根据 diskImageExistsWithKey: 方法，获取 Disk 上的 Data。

![1089786-20170425194151787-1809376809.png][]

判断 Disk 的图片是否存在，就是查找两个路径。同样，拿到这两个路径的文件就可以获得 Data。以下是 Swift 代码示例

SDWebImageManager.shared().diskImageExists(for: imageUrl) { [weak self] (exist) in
        // Always on main thread
        guard self != nil else { return }
        if exist {
            // Find image data from disk
            var data: NSData?
            // Get cache key
            let key = SDWebImageManager.shared().cacheKey(for: imageUrl)
            // Get cache path
            if let path = SDImageCache.shared().defaultCachePath(forKey: key) {
                data = NSData(contentsOfFile: path)
                if data == nil {
                    data = NSData(contentsOfFile: (path as NSString).deletingPathExtension)
                }
            }
            if data != nil {
                // Get image data
            } else {
                // Fail getting image data
            }
        } else {
            // No disk image
        }
    }

这个方法缺点在于，代码复杂，可能会在 SDWebImage 版本升级后失效(例如，Disk 缓存路径改变)。

**推荐的方法是，将图片缓存从 Memory 中移除，然后调用 SDWebImageManager 的 loadImage: 方法。**

// Get cache key
    let key = SDWebImageManager.shared().cacheKey(for: imageUrl)
    // Remove memory cache
    SDImageCache.shared().removeImage(forKey: key, fromDisk: false, withCompletion: nil)
    // Load image and data
    SDWebImageManager.shared().loadImage(with: imageUrl, options: SDWebImageOptions(rawValue: 0), progress: nil) { [weak self] (_, data, _, _, _, _) in
        guard self != nil else { return }
        if data != nil {
            // Get image data
        } else {
            // Fail getting image data
        }
    }

这样写比较简洁。即使 SDWebImage 版本升级后改变 Disk 缓存路径，依然有效。以上代码执行之后，当前图片又会存在 Memory 中。

[1089786-20170425194117631-1601821212.png]: /images/20220617/b5914b6eb60543d381ea43f85c0d2473.png
[1089786-20170425194137131-785031456.png]: /images/20220617/dbe31ad8cd234ac182c7e3d29410f707.png
[1089786-20170425194151787-1809376809.png]: /images/20220617/1ed1fd7e9f234324ba4aae87c7bffedc.png