音视频开发——ffmpeg解码（四）

iOS音视频开发相关文章：

音视频开发——概述（一）

音视频开发——流媒体数据传输RTSP（二）

音视频开发——流媒体数据传输RTP（三）

音视频开发——ffmpeg解码（四）

音视频最强大的开源库非ffmpeg莫属，网上对ffmpeg总结的最好的是雷神的博客（http://blog.csdn.net/leixiaohua1020/article/details/15811977），本文简单介绍下ffmpeg视频解码的使用。

1、ffmpeg初始化

- (void)videoDecoder_init {

avcodec_register_all();
//    _videoFrame = av_frame_alloc();
AVCodec *codec = avcodec_find_decoder(AV_CODEC_ID_H264);
_videoCodecCtx = avcodec_alloc_context3(codec);
int ret = avcodec_open2(_videoCodecCtx, codec, nil);
if (ret != 0){
NSLog(@"open codec failed :%d",ret);
}

_videoFrame = av_frame_alloc();
av_init_packet(&_packet);
}

其中，

_videoCodecCtx是ffmpeg编解码对象；

_videoFrame是解码后的图像帧，可从中生成image图像；

_packet是解码前的数据帧，包括I帧、P帧等
2、解码操作

- (CGSize)videoDecoder_decodeToImage:(uint8_t *)nalBuffer size:(int)inSize {

_packet.size = inSize;
_packet.data = nalBuffer;

CGSize frameSize = {0, 0};

while (inSize > 0) {

int gotframe = 0;
int len = avcodec_decode_video2(_videoCodecCtx,
_videoFrame,
&gotframe,
&_packet);

if (len < 0) {
NSLog(@"decode video error, skip packet");
return frameSize;
}

inSize -= len;
}
frameSize.width = _videoCodecCtx->width;
frameSize.height = _videoCodecCtx->height;

_outputWidth = _videoCodecCtx->width;
self.outputHeight = _videoCodecCtx->height;

return frameSize;
}

3、获取解码后的图像

- (UIImage *)currentImage {

if (!_videoFrame->data[0]) {
return nil;
}

[self convertFrameToRGB];
return [self imageFromAVPicture:_picture width:_outputWidth height:_outputHeight];
}

- (void)convertFrameToRGB {

sws_scale(_img_convert_ctx, (const uint8_t * const*)_videoFrame->data, _videoFrame->linesize, 0, _videoCodecCtx->height, _picture.data, _picture.linesize);
}

- (UIImage *)imageFromAVPicture:(AVPicture)pict width:(int)width height:(int)height {

CGBitmapInfo bitmapInfo = kCGBitmapByteOrderDefault;
CFDataRef data = CFDataCreateWithBytesNoCopy(kCFAllocatorDefault, pict.data[0], pict.linesize[0] * height,kCFAllocatorNull);
CGDataProviderRef provider = CGDataProviderCreateWithCFData(data);
CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
CGImageRef cgImage = CGImageCreate(width,
height,
8,
24,
pict.linesize[0],
colorSpace,
bitmapInfo,
provider,
NULL,
YES,
kCGRenderingIntentDefault);
CGColorSpaceRelease(colorSpace);
UIImage *image = [[UIImage alloc]initWithCGImage:cgImage];

CGImageRelease(cgImage);
CGDataProviderRelease(provider);
CFRelease(data);

return image;
}

对解码后的图像传入UIImageView，即可进行视频播放



附上之前参考的ffmpeg解码播放的例子：ffmpeg解码播放：https://github.com/durfu/DFURTSPPlayer

另外，欢迎大家加入iOS音视频开发的QQ群：331753091