|
|
@@ -0,0 +1,137 @@
|
|
|
+//
|
|
|
+// AVAssetImageDataProvider.swift
|
|
|
+// Kingfisher
|
|
|
+//
|
|
|
+// Created by onevcat on 2020/08/09.
|
|
|
+//
|
|
|
+// Copyright (c) 2020 Wei Wang <onevcat@gmail.com>
|
|
|
+//
|
|
|
+// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
+// of this software and associated documentation files (the "Software"), to deal
|
|
|
+// in the Software without restriction, including without limitation the rights
|
|
|
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
+// copies of the Software, and to permit persons to whom the Software is
|
|
|
+// furnished to do so, subject to the following conditions:
|
|
|
+//
|
|
|
+// The above copyright notice and this permission notice shall be included in
|
|
|
+// all copies or substantial portions of the Software.
|
|
|
+//
|
|
|
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
+// THE SOFTWARE.
|
|
|
+
|
|
|
+#if !os(watchOS)
|
|
|
+
|
|
|
+import Foundation
|
|
|
+import AVKit
|
|
|
+
|
|
|
+#if canImport(MobileCoreServices)
|
|
|
+import MobileCoreServices
|
|
|
+#else
|
|
|
+import CoreServices
|
|
|
+#endif
|
|
|
+
|
|
|
+/// A data provider to provide thumbnail data from a given AVKit asset.
|
|
|
+public struct AVAssetImageDataProvider: ImageDataProvider {
|
|
|
+
|
|
|
+ /// The possible error might be caused by the `AVAssetImageDataProvider`.
|
|
|
+ /// - userCancelled: The data provider process is cancelled.
|
|
|
+ /// - invalidImage: The retrieved image is invalid.
|
|
|
+ public enum AVAssetImageDataProviderError: Error {
|
|
|
+ case userCancelled
|
|
|
+ case invalidImage(_ image: CGImage?)
|
|
|
+ }
|
|
|
+
|
|
|
+ /// The asset image generator bound to `self`.
|
|
|
+ public let assetImageGenerator: AVAssetImageGenerator
|
|
|
+
|
|
|
+ /// The time at which the image should be generate in the asset.
|
|
|
+ public let time: CMTime
|
|
|
+
|
|
|
+ private var internalKey: String {
|
|
|
+ return (assetImageGenerator.asset as? AVURLAsset)?.url.absoluteString ?? UUID().uuidString
|
|
|
+ }
|
|
|
+
|
|
|
+ /// The cache key used by `self`.
|
|
|
+ public var cacheKey: String {
|
|
|
+ return "\(internalKey)_\(time.seconds)"
|
|
|
+ }
|
|
|
+
|
|
|
+ /// Creates an asset image data provider.
|
|
|
+ /// - Parameters:
|
|
|
+ /// - assetImageGenerator: The asset image generator controls data providing behaviors.
|
|
|
+ /// - time: At which time in the asset the image should be generated.
|
|
|
+ public init(assetImageGenerator: AVAssetImageGenerator, time: CMTime) {
|
|
|
+ self.assetImageGenerator = assetImageGenerator
|
|
|
+ self.time = time
|
|
|
+ }
|
|
|
+
|
|
|
+ /// Creates an asset image data provider.
|
|
|
+ /// - Parameters:
|
|
|
+ /// - assetURL: The URL of asset for providing image data.
|
|
|
+ /// - time: At which time in the asset the image should be generated.
|
|
|
+ ///
|
|
|
+ /// This method uses `assetURL` to create an `AVAssetImageGenerator` object and calls
|
|
|
+ /// the `init(assetImageGenerator:time:)` initializer.
|
|
|
+ ///
|
|
|
+ public init(assetURL: URL, time: CMTime) {
|
|
|
+ let asset = AVAsset(url: assetURL)
|
|
|
+ let generator = AVAssetImageGenerator(asset: asset)
|
|
|
+ self.init(assetImageGenerator: generator, time: time)
|
|
|
+ }
|
|
|
+
|
|
|
+ /// Creates an asset image data provider.
|
|
|
+ ///
|
|
|
+ /// - Parameters:
|
|
|
+ /// - assetURL: The URL of asset for providing image data.
|
|
|
+ /// - seconds: At which time in seconds in the asset the image should be generated.
|
|
|
+ ///
|
|
|
+ /// This method uses `assetURL` to create an `AVAssetImageGenerator` object, uses `seconds` to create a `CMTime`,
|
|
|
+ /// and calls the `init(assetImageGenerator:time:)` initializer.
|
|
|
+ ///
|
|
|
+ public init(assetURL: URL, seconds: TimeInterval) {
|
|
|
+ let time = CMTime(seconds: seconds, preferredTimescale: 600)
|
|
|
+ self.init(assetURL: assetURL, time: time)
|
|
|
+ }
|
|
|
+
|
|
|
+ public func data(handler: @escaping (Result<Data, Error>) -> Void) {
|
|
|
+ assetImageGenerator.generateCGImagesAsynchronously(forTimes: [NSValue(time: time)]) {
|
|
|
+ (requestedTime, image, imageTime, result, error) in
|
|
|
+ if let error = error {
|
|
|
+ handler(.failure(error))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if result == .cancelled {
|
|
|
+ handler(.failure(AVAssetImageDataProviderError.userCancelled))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ guard let cgImage = image, let data = cgImage.jpegData else {
|
|
|
+ handler(.failure(AVAssetImageDataProviderError.invalidImage(image)))
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ handler(.success(data))
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+extension CGImage {
|
|
|
+ var jpegData: Data? {
|
|
|
+ guard let mutableData = CFDataCreateMutable(nil, 0),
|
|
|
+ let destination = CGImageDestinationCreateWithData(mutableData, kUTTypeJPEG, 1, nil)
|
|
|
+ else {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ CGImageDestinationAddImage(destination, self, nil)
|
|
|
+ guard CGImageDestinationFinalize(destination) else { return nil }
|
|
|
+ return mutableData as Data
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+#endif
|
Wei Wang