AVFoundation从视频中提取图片(八)

前言

从本文开始逐渐学习iOS自带的多媒体处理框架,例如AVFoundation,VideoToolbox,CoreMedia,CoreVideo实现多媒体的处理,并且将实现方式以及效果和ffmpeg的方式做对比

从一个视频文件中提取出指定时间或者多个指定时间处的视频帧然后转换为图片是很常见的需求,AVFoundation就提供了这样的接口。

本文的目的:
1、提取一个视频文件中10秒处的所有视频帧并且以JPG格式保存到本地
2、提取一个视频文件中某一个时刻的一帧视频并且以JPG格式保存到本地
3、提取一个视频文件中某一个时刻的一帧视频压缩到指定大小并且以JPG格式保存到本地

从视频中提取视频帧相关流程

image.png

上图介绍了AVFoundation框架中从视频中提取视频帧的相关对象关系图,可以看到使用AVFoundation从视频中提取视频帧还是相对比较简单的。这套流程同样适用于远程的文件

相关对象及函数介绍

  • 1、AVURLAsset
    容器对象,代表了要操作的容器。封装,解封装,音视频播放,以及音视频合并等等操作的基础都涉及到这个对象。

  • 2、AVAssetTrack
    音视频轨道对象,代表了文件中的一路音频流或者一路视频流,它作为每一个要被合并的音频或者视频流被添加到组合对象中最终进行合并

  • 3、AVAssetImageGenerator
    视频帧提取对象,它用于从本地或者远程视频中提取指定时间处的视频帧。支持对提取的视频进行压缩后输出,通过maximumSize属性来设定最后输出视频帧的分辨率,默认maximumSize为CGSizeZero代表不压缩按照原始大小输出

  • 4、-(nullable CGImageRef)copyCGImageAtTime:(CMTime)requestedTime actualTime:(nullable CMTime *)actualTime error:(NSError * _Nullable * _Nullable)outError CF_RETURNS_RETAINED;
    提取指定时刻的一帧视频

  • 5、-(void)generateCGImagesAsynchronouslyForTimes:(NSArray<NSValue *> *)requestedTimes completionHandler:(AVAssetImageGeneratorCompletionHandler)handler;
    当提取多个时间处的视频帧时用此方法比较效率

实现代码

#import <Foundation/Foundation.h>

@interface AVGetImage : NSObject

/** 实现功能:
 *  从视频文件中截图指定时间段的视频帧所有视频帧,并最终以JPG的格式保存到指定目录中
 */
- (void)getImageFromURL:(NSURL*)srURL saveDstPath:(NSString*)dstPath;
@end
#import "AVGetImage.h"
#import <AVFoundation/AVFoundation.h>
#import <UIKit/UIKit.h>

@implementation AVGetImage
{
    // 视频处理队列信号量
    dispatch_semaphore_t processSemaphore;
}
- (void)getImageFromURL:(NSURL*)srURL saveDstPath:(NSString*)dstPath
{
    processSemaphore = dispatch_semaphore_create(0);
    
    AVURLAsset *asset = [[AVURLAsset alloc] initWithURL:srURL options:nil];
    // 从视频中提取视频帧同样适用于远程文件
//    asset = [[AVURLAsset alloc] initWithURL:[NSURL URLWithString:@"https://images.flypie.net/test_1280x720_3.mp4"] options:nil];
    [asset loadValuesAsynchronouslyForKeys:@[@"tracks",@"duration"] completionHandler:^{
        
        NSError *error = nil;
        AVKeyValueStatus status = [asset statusOfValueForKey:@"tracks" error:&error];
        if (status != AVKeyValueStatusLoaded) {
            NSLog(@"key value load error %@",error);
            return;
        }
        
        [self processAsset:asset dst:dstPath];
        
        dispatch_semaphore_signal(self->processSemaphore);
        
    }];
    
    dispatch_semaphore_wait(processSemaphore, DISPATCH_TIME_FOREVER);
    NSLog(@"结束了。。。");
    
}

- (void)processAsset:(AVAsset*)asset dst:(NSString*)path
{
    NSArray *videoTracks = [asset tracksWithMediaType:AVMediaTypeVideo];
    if (!videoTracks) {
        return;
    }
    AVAssetTrack *track = videoTracks[0];
    // 获取视频的帧率信息
    CGFloat fps = track.nominalFrameRate;
    NSLog(@"formats %f",track.nominalFrameRate);
    
    /** AVAssetImageGenerator 用来从视频中提取图片的一个对象
     *  本地和远程文件都可以用该方法进行提取
     */
    AVAssetImageGenerator *imageGen = [[AVAssetImageGenerator alloc] initWithAsset:asset];
    // 定义请求指定时间上的误差范围,这里设置为0,代表尽量精确
    imageGen.requestedTimeToleranceAfter = kCMTimeZero;
    imageGen.requestedTimeToleranceBefore = kCMTimeZero;
    
    // 构建10秒处的所有视频帧然后转换成图像
    CMTime start = CMTimeMake(10000, 1000);
    if (CMTimeGetSeconds(start) > CMTimeGetSeconds(asset.duration)) {
        start = CMTimeMake((CMTimeGetSeconds(asset.duration) - 1) * 1000, 1000);
    }
    NSMutableArray *times = [NSMutableArray array];
    
    for (int i=0; i<fps; i++) {
        CMTime oneFps = CMTimeMake(i/fps * 1000, 1000);
        CMTime time = CMTimeAdd(start, oneFps);
        NSValue *timeValue = [NSValue valueWithCMTime:time];
        [times addObject:timeValue];
    }
    __block BOOL finish = NO;
    __block int num = 0;
    // 1、此方法用于请求多个视频帧,效率比较高,备注:返回的image由系统自动释放内存
    [imageGen generateCGImagesAsynchronouslyForTimes:times completionHandler:^(CMTime requestedTime, CGImageRef  _Nullable image, CMTime actualTime, AVAssetImageGeneratorResult result, NSError * _Nullable error) {
        NSLog(@"actual time %f",CMTimeGetSeconds(actualTime));
        num++;
        if (num == fps) {
            finish = YES;
        }
        
        @autoreleasepool {
            UIImage *imageObj = [[UIImage alloc] initWithCGImage:image];
            NSData *jpgData = UIImageJPEGRepresentation(imageObj, 1.0);
            NSString *spath = [path stringByAppendingFormat:@"/%d.JPG",num];
            [jpgData writeToFile:spath atomically:YES];
        }
    }];
    
    while (!finish) {
        usleep(1000);
    }
    NSLog(@"写入时间段结束了");
    
    // 2、提取指定时间的一个视频帧
    CMTime atime = CMTimeMake(5*1000, 1000);
    CMTime actime = kCMTimeZero;
    NSError *error = nil;
    CGImageRef cgimage = [imageGen copyCGImageAtTime:atime actualTime:&actime error:&error];
    NSLog(@"need time %f actual time %f",CMTimeGetSeconds(atime),CMTimeGetSeconds(actime));
    UIImage *uiimage = [[UIImage alloc] initWithCGImage:cgimage];
    NSData *jpgData = UIImageJPEGRepresentation(uiimage, 1.0);
    NSString *atPath = [path stringByAppendingPathComponent:@"at.JPG"];
    [jpgData writeToFile:atPath atomically:YES];
    CGImageRelease(cgimage);
    
    // 3、提取指定时间的一个视频帧,并且进行压缩后输出
    // 默认maximumSize为CGSizeZero代表不压缩按照原始大小输出
    imageGen.maximumSize = CGSizeMake(300, 150);
    cgimage = [imageGen copyCGImageAtTime:atime actualTime:&actime error:&error];
    NSLog(@"need time %f actual time %f",CMTimeGetSeconds(atime),CMTimeGetSeconds(actime));
    uiimage = [[UIImage alloc] initWithCGImage:cgimage];
    jpgData = UIImageJPEGRepresentation(uiimage, 1.0);
    atPath = [path stringByAppendingPathComponent:@"at_scale.JPG"];
    [jpgData writeToFile:atPath atomically:YES];
    CGImageRelease(cgimage);
    
}
@end

备注:这套从视频中提取视频帧的流程也同样适用于远程的视频文件,将上面中/ asset = [[AVURLAsset alloc] initWithURL:[NSURL URLWithString:@"https://images.flypie.net/test_1280x720_3.mp4"] options:nil];解注释 发现运行结果一样

遇到问题

项目地址

https://github.com/nldzsz/ffmpeg-demo

位于AVFoundation目录下文件AVGetImage.h/AVGetImage.m中

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,189评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,577评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,857评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,703评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,705评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,620评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,995评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,656评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,898评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,639评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,720评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,395评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,982评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,953评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,195评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,907评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,472评论 2 342