Swift 自定义图片选择器(二) -- 获取图片资源

上一篇简单的介绍了一下Photos，下面就开始介绍demo里面的业务类。

相册列表项

表示一个非空的相册，即非空的PHAsset资源的集合。主要用于显示图片列表，并且在选中某一列表时，显示该相册里面的图片资源。

  /// 相簿列表项
  struct AlbumItem {
  var title : String?     // 相册的标题
  var fetchResult : PHFetchResult<PHAsset>    // 资源集合

  init(title: String?, fetchResult: PHFetchResult<PHAsset>) {
    self.title = title
    self.fetchResult = fetchResult
}
}

重载运算符 +

AlbumItem 是自定义的数据类型，为了实现该类型的数组相加，需要重载运算符，然后就可以像其他数据类型一样直接使用符号 “+” 来合并两个数组。（重载运算符需要在全局域中，就是跟AlbumItem在同一层级下）

postfix operator +

func + (left: [AlbumItem]?, right: [AlbumItem]?) ->[AlbumItem]? {
if (left == nil) {
    return right
}

if (right == nil) {
    return left
}
var result = left!

for (_, item) in right!.enumerated() {
    result.append(item)
}

return result
}

PickerHelper 图片资源获取帮助类

所有的关于图片资源获取的方法都封装在PickerHelper里面，并且使用单例设计模式，设置成单例类。

/// 获取图片的资源的类
class PickerHelper: NSObject {
static let helperDefault : PickerHelper = {
    let helper = PickerHelper()
    return helper
}()

var targetSize = CGSize(width: 320, height: 675)    // 默认大图的大小
var maxCount = 9    // 默认最多可选9张图片
}

targetSize 和maxCount 公开可变，是让用户可以根据需要进行修改。

获取相册数组

/// 获取智能相册
///
/// - returns: 智能相册数组 PHFetchResult<PHAsset>
func getSmartAlbums() -> [AlbumItem]? {
    //获取所有的智能相册
    let smartAlbums = PHAssetCollection.fetchAssetCollections(with: .smartAlbum, subtype: .albumRegular, options: nil)
    // 获取资源选项设置，以创建时间升序排序
    let options = PHFetchOptions()
    options.sortDescriptors = [NSSortDescriptor.init(key: "creationDate", ascending: true)]
    let albumsArray = getAssetResultInCollection(smartAlbums, fetchOptions: options)
    
    return albumsArray
}

/// 从PHAssetCollection中获取相册，主要是用于相册列表
///
/// - parameter collection: 相册集合
/// - parameter options:    获取相册的相关设置参数
///
/// - returns: 相册数组
private func getAssetResultInCollection(_ collection: PHFetchResult<PHAssetCollection>, fetchOptions options: PHFetchOptions) ->[AlbumItem]? {
    var albumsArray = [AlbumItem]()
    
    for index in 0..<collection.count {
        //获取一个相册
        let assetCollection = collection[index];
        
        //然后从每一个智能相册中获取资源，获取到的是一个list
        let assetFecthResults = PHAsset.fetchAssets(in: assetCollection, options: nil);
        print("\(assetCollection.localizedTitle!)相册，共有照片数:\(assetFecthResults.count)")
        
        if assetFecthResults.count > 0 {
            //有照片的相册
            let album = AlbumItem.init(title: assetCollection.localizedTitle, fetchResult: assetFecthResults)
            
            albumsArray.append(album)
        }
    }
    
    return albumsArray
}

获取用户创建的相册数组

/// 获取用户创建的相册
///
/// - returns: 相册数组
func getUserAlbums() -> [AlbumItem]? {
    let userAlbums = PHCollectionList.fetchTopLevelUserCollections(with: nil) as! PHFetchResult<PHAssetCollection>
    let options = PHFetchOptions()
    options.sortDescriptors = [NSSortDescriptor.init(key: "creationDate", ascending: true)]
    
    let albumsArray = getAssetResultInCollection(userAlbums, fetchOptions: options)
    
    return albumsArray
}

PictureItem

表示一个图片资源，并且保存了获取到的大图、小图和该资源是否被选中

class PictureItem: NSObject {
/// 图片资源
var asset = PHAsset()
/// 显示的小图
var lowImage: UIImage?
/// 上传的大图
var hightImage: UIImage?
/// 是否选中的标志
var isSelected = false

init(asset: PHAsset, lowImage: UIImage?, hightImage: UIImage?, isSelected: Bool = false) {
    self.asset = asset
    self.lowImage = lowImage
    self.hightImage = hightImage
    self.isSelected = isSelected
}
}

获取相册的图片

/// 获取某个相册里面的图片资源
///
/// - parameter albumItem: 相册
///
/// - returns: 图片数组 [PictureItem]
func getPicturesInAlbumItem(_ albumItem: AlbumItem) -> [PictureItem] {
    var pictures = [PictureItem]()
    
    albumItem.fetchResult.enumerateObjects({ (asset, curIndex, nil) in
        pictures.append(PictureItem.init(asset: asset, lowImage: nil, hightImage: nil))
    })
    
    return pictures
}

/// 从图片资源PHAsset中获取指定大小的image对象
///
/// - parameter asset:           Asset图片资源
/// - parameter targetSize:      指定尺寸
/// - parameter finishedCallack: 获取后返回的block
func getImageWithAsset(_ asset: PHAsset, size targetSize: CGSize, finishedCallack: @escaping (_ image: UIImage) -> ()) {
    let options = PHImageRequestOptions()
    options.resizeMode = .exact // 返回图像与目标size保持一致
    options.deliveryMode = .highQualityFormat   // 只返回高像素的图像
    var size = targetSize
    
    // 如果没有目标大小，则返回原图
    if targetSize == .zero {
        size = CGSize(width: asset.pixelWidth, height: asset.pixelHeight)
    }
    
    PHImageManager.default().requestImage(for: asset, targetSize: size, contentMode: .default, options: options) { (image, info) in
        // 如果是image是nil 则直接返回
        guard let image = image else {
            return
        }
        
        // 拿到图片，则返回
        finishedCallack(image)
    }
}