最近开发遇到一个点击按钮实现直播视频流截屏的功能，去网上查了一下资料，总结了一下iOS中截屏相关的知识，然后自己做了个demo。
demo主要实现了3种截屏方法，分别对应三种不同的应用场景。
1、imageView截图，这个截图方法可以用来实现截取项目中的控件，可截取的控件可包括UIImageView、UIView等。截图方法如下：

-(UIImage *)screenshotsWithView:(UIView *)view{UIGraphicsBeginImageContextWithOptions(view.frame.size, NO, 0.0);[view.layer renderInContext:UIGraphicsGetCurrentContext()];UIImage *image = UIGraphicsGetImageFromCurrentImageContext();UIGraphicsEndImageContext();return image;
}

2、MP4格式的视频流截图，即获取MP4格式视频流中的某一帧。截取方法如下：

-(UIImage *)screenshotsMP4WithCurrentTime:(CMTime)currentTime videoUrl:(NSString *)url{AVURLAsset * asset = [[AVURLAsset alloc] initWithURL:[NSURL URLWithString:url] options:nil];AVAssetImageGenerator * gen = [[AVAssetImageGenerator alloc] initWithAsset:asset];gen.appliesPreferredTrackTransform = YES;gen.requestedTimeToleranceAfter = kCMTimeZero;gen.requestedTimeToleranceBefore = kCMTimeZero;CMTime time = CMTimeMakeWithSeconds(CMTimeGetSeconds(currentTime), 600);NSError * error = nil;CMTime actualTime;CGImageRef imageRef = [gen copyCGImageAtTime:time actualTime:&actualTime error:&error];UIImage * image = [[UIImage alloc] initWithCGImage:imageRef];return image;
}

这个方法是不是仅限于MP4格式的视频我不是很清楚，但是可以确定的是，MP4截取某一帧的方法不适用于m3u8格式的视频流。
3、m3u8格式视频流截图。方法如下：

-(UIImage *)screenshotsm3u8WithCurrentTime:(CMTime)currentTime playerItemVideoOutput:(AVPlayerItemVideoOutput *)output{CVPixelBufferRef pixelBuffer = [output copyPixelBufferForItemTime:currentTime itemTimeForDisplay:nil];CIImage *ciImage = [CIImage imageWithCVPixelBuffer:pixelBuffer];CIContext *temporaryContext = [CIContext contextWithOptions:nil];CGImageRef videoImage = [temporaryContext createCGImage:ciImagefromRect:CGRectMake(0, 0,CVPixelBufferGetWidth(pixelBuffer),CVPixelBufferGetHeight(pixelBuffer))];UIImage *frameImg = [UIImage imageWithCGImage:videoImage];CGImageRelease(videoImage);CVBufferRelease(pixelBuffer);return frameImg;
}

需要特别指出来的那两个release方法，不调用的话会造成内存泄漏（可用Instruments调试）。
另外还有一点需要注意，m3u8格式的视频流在截取时需要传入一个AVPlayerItemVideoOutput对象，这个对象一定要在AVPlayerItem初始化的时候设置好，如果以临时变量的形式传入或导致截取失败。
在截屏事件里直接调用以上方法即可获取到截取的image。demo链接地址：https://github.com/zhanqin/screenshots