[小白]FFMPEG小白-day04(动手写个小程序,将文件转成h264 yuv)

原文地址: https://blog.csdn.net/JerryWu145/article/details/79380185


昨天插了一条html的东西,今天补档ffmpeg的编程,通过ffmpeg编写一个最简单的程序,今天的程序主要的功能是

打印视频文件的相关信息

输出H.264文件

输出YUV文件

不过在编程之前,再来将前天的ffmpeg结构体框架图拿出来



ok这张图能够对下面的编程有所帮助,你可以根据这张图去知道哪个结构体或者数组是属于那个大的结构体的。

再来介绍下该结构体中的一些重要的信息

AVStream(在AVFormatContext结构体下方)

* id:序号
* codec:该流对应的AVCodecContext
* time_base:该流的时基(用来为视频或者音频设定播放位置的)
* r_frame_rate:该流的帧率(一秒钟有多少个画面)

AVCodecContext(编解码器的一些信息)

* codec:编解码器的AVCodec
* width,height:图像的宽高(只针对视频)
* pix_fmt:像素格式(只针对视频)
* sample_rate:采样率(只针对音频)
* channels:声道数(只针对音频)
* sample_fmt:采样格式(只针对音频)

AVCodec

* name:编解码器名称
* long_name:编解码器长名称
* type:编解码器类型(音频或者视频)
* id:编解码器ID
* 一些编解码的接口函数

AVPacket(可以理解成装H.264的结构体)

* pts:显示时间戳(即视频应该几分,几秒的时候显示到界面上,他的单位是整数的,将来会和上文中的时基进行换算,换算成为我们常用的时分秒这样的单位)
* dts:解码时间戳
注意:视频的码流存储的次序和播放视频的次序并不是顺序的,所以有两个时间戳,即一个显示的一个解码的。
* data:压缩编码数据(h.264数据)
* size:压缩编码数据大小
* stream_index:所属的AVStream(标识是属于视频流还是音频流)

AVFrame(可以理解成装YUV的结构体)

* data:解码后的图像像素数据(音频采样数据即yuv数据)
* linesize:对视频来说是图像中的一行像素的大小;而对于音频来说是整个音频帧的大小
* width,height:图像的宽高(只针对视频)
* key_frame:是否为关键帧(只针对视频,红颜色的帧是关键帧)
* pict_type:帧类型(只针对视频)。例如I,P,B。

下文中要用到的小知识:解码后的数据需要经过sws_scale()函数处理

 解码后yuv格式的视频像素数据保存在AVFrame的data[0]、data[1]、data[2]中。
 但是这些像素值并不是连续存储的,每行有效像素之后存储了一些无效像素,以
 高度Y数据为例,data[0]中一共包含了linesize[0]*height个数据。但是出于优化等方面
 的考虑,linesize[0]实际上并不等于宽度width,而是一个比宽度大一些的值。因此需要
 使用ses_scale()进行转换。转换后去除了无效数据,width和linesize[0]就取值相同了。
 
 最后用大白话概括一下吧,就是说如果不用这个sws_scale()函数的话,直接播放的yuv数据是
 带有黑边的,只有经过sws_scale()函数裁剪完了之后,才能正常的播放视频。

该函数效果如下图



ok,先把大体的代码 up 上来,我这里做了必要的注释,应该都能看懂,看不懂的话可以留言联系我

 // simple_sdl2_player.cpp : 定义控制台应用程序的入口点。
//

include "stdio.h"
define __STDC_CONSTANT_MACROS
include "stdafx.h"
//引入ffmpeg中的相关头文件
extern "C" {

include "libavcodec\avcodec.h"
include "libavformat/avformat.h"
include "libswscale/swscale.h"
}

int main(int argc, char* argv[])
{
//声明了AVFormatContext视频数据指针
AVFormatContext *pFormatCtx;
//声明变量i和视频的位置索引
int i, videoindex;
//解码器信息指针
AVCodecContext *pCodecCtx;
//解码器指针
AVCodec *pCodec;
//像素数据指针
AVFrame *pFrame, *pFrameYUV;
//可以理解成缓冲数组
uint8_t *out_buffer;
//h.264数据指针
AVPacket *packet;
int y_size;
int ret, got_picture;
struct SwsContext *img_convert_ctx;
//输入文件路径
char filepath[] = "Titanic.ts";
int frame_cnt;
//注册所有相关组件
av_register_all();
//初始化网络
avformat_network_init();
//分配空间
pFormatCtx = avformat_alloc_context();
//打开视频流
//如果没有找到avformat_open_input()方法如果返回0表示获取成功,反之则失败
if (avformat_open_input(&pFormatCtx, filepath, NULL, NULL) != 0) {
    printf("Couldn't open input stream.\n");
    return -1;
}
//获取视频流信息 如果avformat_find_stream_info()结果大于等于0表示获取流成功
if (avformat_find_stream_info(pFormatCtx, NULL)<0) {
    printf("Couldn't find stream information.\n");
    return -1;
}
//ok 打开了视频并且获取了视频流 ,设置视频索引值默认值
videoindex = -1;
//循环遍历输入视频的av_Stream个数,依次判断av_Stream中的codec_type类型,如果是视频类型,进行记录,一般来说,这个值一般为0
for (i = 0; i<pFormatCtx->nb_streams; i++)
    if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
        videoindex = i;
        break;
    }
//如果没有找到视频的索引,说明并不是一个视频文件
if (videoindex == -1) {
    printf("Didn't find a video stream.\n");
    return -1;
}
//取得视频的解码器信息
pCodecCtx = pFormatCtx->streams[videoindex]->codec;
//得到的解码器
pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
//解码器未找到
if (pCodec == NULL) {
    printf("Codec not found.\n");
    return -1;
}
//解码器和解码器信息的话,开始解码,如果返回值为0,说明无法开始解码
if (avcodec_open2(pCodecCtx, pCodec, NULL)<0) {
    printf("Could not open codec.\n");
    return -1;
}
/*
* 在此处添加输出视频信息的代码
* 取自于pFormatCtx,使用fprintf()
*/
 
//初始化yuv容器,并且初始化内存空间
pFrame = av_frame_alloc();
pFrameYUV = av_frame_alloc();
 
out_buffer = (uint8_t *)av_malloc(avpicture_get_size(PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));
//设置图像内容
avpicture_fill((AVPicture *)pFrameYUV, out_buffer, PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);
//初始化h.264容器
packet = (AVPacket *)av_malloc(sizeof(AVPacket));
//Output Info-----------------------------
printf("--------------- File Information ----------------\n");
av_dump_format(pFormatCtx, 0, filepath, 0);
printf("-------------------------------------------------\n");
//上文说的对图形进行宽度上方的裁剪,以便于显示的更好
img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt,
    pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);
 
frame_cnt = 0;
//如果读流成功
while (av_read_frame(pFormatCtx, packet) >= 0) {
    if (packet->stream_index == videoindex) {
        /*
        * 在此处添加输出H264码流的代码
        * 取自于packet,使用fwrite()
        */
        ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
        if (ret < 0) {
            printf("Decode Error.\n");
            return -1;
        }
        if (got_picture) {
            //上文说的对图形进行宽度上方的裁剪,以便于显示的更好
            sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, 0, pCodecCtx->height,
                pFrameYUV->data, pFrameYUV->linesize);
            printf("Decoded frame index: %d\n", frame_cnt);
 
            /*
            * 在此处添加输出YUV的代码
            * 取自于pFrameYUV,使用fwrite()
            */
            //计数
            frame_cnt++;
 
        }
    }
    //释放packet
    av_free_packet(packet);
}
//释放相关资源
sws_freeContext(img_convert_ctx);
av_frame_free(&pFrameYUV);
av_frame_free(&pFrame);
avcodec_close(pCodecCtx);
avformat_close_input(&pFormatCtx);
 
return 0;

ok,注释都加上了,然后就是要做一些输出了,我们先把一个ts视频文件的时长 格式打印一下。



ok,现在,我们先尝试下把这些信息输出到一个文件里面去



ok,现在剩下输出h264文件和yuv文件了,不过这里先不帖出来了,下篇文章会贴出来的,有跟着学的同学们可以自己试一下,如何输出h264文件和yuv文件~~

持续更新,欢迎关注~~~

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

推荐阅读更多精彩内容