如何从Swift中的UIPickerViewController中获取视频的缩略图？

Question

This is how I access the video:

func imagePickerController(picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [NSObject : AnyObject]) {

    if let videoURL = info[UIImagePickerControllerMediaURL] as? NSURL {
        let video = NSData(contentsOfURL: videoURL)
    }
}

That video I would like to show the user in my controller (I do not mean UIDocumentInteractionController to preview this). I need a thumbnail of that video, and then assign this to UIImageView . Is it possible?

Answer 1

The simplest answer:

import AVFoundation

private func thumbnailForVideoAtURL(url: NSURL) -> UIImage? {

    let asset = AVAsset(URL: url)
    let assetImageGenerator = AVAssetImageGenerator(asset: asset)

    var time = asset.duration
    time.value = min(time.value, 2)

    do {
        let imageRef = try assetImageGenerator.copyCGImageAtTime(time, actualTime: nil)
        return UIImage(CGImage: imageRef)
    } catch {
        print("error")
        return nil
    }
}

Answer 2

Here's SWIFT 3.0 using extension

import Foundation
import UIKit
import MediaPlayer

import SystemConfiguration

extension AVAsset{
    var videoThumbnail:UIImage?{

        let assetImageGenerator = AVAssetImageGenerator(asset: self)
        assetImageGenerator.appliesPreferredTrackTransform = true

        var time = self.duration
        time.value = min(time.value, 2)

        do {
            let imageRef = try assetImageGenerator.copyCGImage(at: time, actualTime: nil)
            let thumbNail = UIImage.init(cgImage: imageRef)


            print("Video Thumbnail genertated successfuly".DubugSuccess())

            return thumbNail

        } catch {

            print("error getting thumbnail video".DubugError(),error.localizedDescription)
            return nil


        }

    }
}

Then simply from anywhere you have access to AVAsset/URL :

func imagePickerController(picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [NSObject : AnyObject]) {

    if let videoURL = info[UIImagePickerControllerMediaURL] as? NSURL {
        let video = NSData(contentsOfURL: videoURL)

        //Create AVAsset from url
        let ass = AVAsset(url:videoURL)

        if let videoThumbnail = ass.videoThumbnail{
          print("Success")
        }
    }
}

Answer 3

You can generate a thumbnail with this method:

import AVFoundation

func generateThumnail(url : NSURL) -> UIImage?{
    let asset: AVAsset = AVAsset(URL: url)
    let assetImgGenerate : AVAssetImageGenerator = AVAssetImageGenerator(asset: asset)
    assetImgGenerate.appliesPreferredTrackTransform = true
    let time        : CMTime = CMTimeMake(1, 30)
    let img         : CGImageRef
    do {
        try img = assetImgGenerate.copyCGImageAtTime(time, actualTime: nil)
        let frameImg: UIImage = UIImage(CGImage: img)
        return frameImg
    } catch {

    }
    return nil
}

Answer 4

Answer for AiOsN and the way how to fix image rotation. You just need to change this line of code:

return UIImage(CGImage: imageRef)

to this one:

return UIImage(CGImage: imageRef scale: 1.0 , orientation: UIImageOrientation.Right)

and the image will show you correct rotation.