#include <iostream>
#include <thread>
#include <stdio.h>
extern "C"
{
#include "libavutil/opt.h"
#include "libavutil/channel_layout.h"
#include "libavutil/common.h"
#include "libavutil/imgutils.h"
#include "libavutil/mathematics.h"
#include "libavutil/samplefmt.h"
#include "libavutil/time.h"
#include "libavutil/fifo.h"
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavformat/avio.h"
#include "libavfilter/avfilter.h"
#include "libavfilter/buffersink.h"
#include "libavfilter/buffersrc.h"
#include "libswscale/swscale.h"
#include "libswresample/swresample.h"
}
#ifdef _WIN32
#pragma comment(lib, "avcodec.lib")
#pragma comment(lib, "avfilter.lib")
#pragma comment(lib, "avutil.lib")
#pragma comment(lib, "avdevice.lib")
#pragma comment(lib, "postproc.lib")
#pragma comment(lib, "swresample.lib")
#pragma comment(lib, "swscale.lib")
#pragma comment(lib, "avformat.lib")
#endif
/******************************
功能:编码YUV序列为h264视频文件
FFmpeg:4.0.2
******************************/
#include <iostream>
extern "C"
{
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libavutil/imgutils.h>
}
using namespace std;
static void encode(AVCodecContext* enc_ctx, AVFrame* frame, AVPacket* pkt, FILE* outfile)
{
int ret;
if (frame)
printf("Send frame %3lld\n", frame->pts);
ret = avcodec_send_frame(enc_ctx, frame);
if (ret < 0)
{
fprintf(stderr, "Error sending a frame for encoding\n");
exit(1);
}
while (ret >= 0)
{
ret = avcodec_receive_packet(enc_ctx, pkt);
if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
return;
else if (ret < 0)
{
fprintf(stderr, "Error during encoding\n");
exit(1);
}
printf("Write packet %3lld(size=%5d)\n", pkt->pts, pkt->size);
fflush(stdout);
fwrite(pkt->data, 1, pkt->size, outfile);
av_packet_unref(pkt);
}
}
int EncodeYUVToH264(const char* input_fileName, const char* out_fileName, int in_w, int in_h, int framenum)
{
// 打开输入YUV序列
FILE* in_file = NULL;
fopen_s(&in_file, input_fileName, "rb");
if (!in_file)
{
cout << "can not open file!" << endl;
return -1;
}
// 打开输出视频文件
FILE* out_file = NULL;
fopen_s(&out_file, out_fileName, "wb");
// 初始化AVFormatContext结构体,根据文件名获取到合适的封装格式
AVFormatContext* pFormatCtx = avformat_alloc_context();
avformat_alloc_output_context2(&pFormatCtx, NULL, NULL, out_fileName);
AVOutputFormat* fmt = pFormatCtx->oformat;
// 初始化视频码流
AVStream* video_st = avformat_new_stream(pFormatCtx, 0);
if (video_st == NULL)
{
printf("failed allocating output stream\n");
return -1;
}
video_st->time_base.num = 1;
video_st->time_base.den = 25;
// 编码器Context设置参数
AVCodecContext* pCodecCtx = avcodec_alloc_context3(NULL);
avcodec_parameters_to_context(pCodecCtx, pFormatCtx->streams[0]->codecpar);
pCodecCtx->codec_id = fmt->video_codec;
pCodecCtx->codec_type = AVMEDIA_TYPE_VIDEO;
pCodecCtx->pix_fmt = AV_PIX_FMT_YUV420P;
pCodecCtx->width = in_w;
pCodecCtx->height = in_h;
pCodecCtx->time_base.num = 1;
pCodecCtx->time_base.den = 25;
pCodecCtx->bit_rate = 1024*1024;
pCodecCtx->gop_size = 12;
pCodecCtx->flags |= AV_CODEC_FLAG2_LOCAL_HEADER;
if (pCodecCtx->codec_id == AV_CODEC_ID_H264)
{
pCodecCtx->qmin = 10;
pCodecCtx->qmax = 51;
pCodecCtx->qcompress = 0.6;
}
if (pCodecCtx->codec_id == AV_CODEC_ID_MPEG2VIDEO)
pCodecCtx->max_b_frames = 2;
if (pCodecCtx->codec_id == AV_CODEC_ID_MPEG1VIDEO)
pCodecCtx->mb_decision = 2;
// 寻找编码器
AVCodec* pCodec = avcodec_find_encoder(AV_CODEC_ID_H264);
if (!pCodec)
{
cout << "no right encoder!" << endl;
return -1;
}
// 打开编码器
if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0)
{
cout << "open encoder fail!" << endl;
return -1;
}
// 输出格式信息
av_dump_format(pFormatCtx, 0, out_fileName, 1);
// 初始化帧
AVFrame* pictureFrame = av_frame_alloc();
pictureFrame->width = pCodecCtx->width;
pictureFrame->height = pCodecCtx->height;
pictureFrame->format = pCodecCtx->pix_fmt;
// ffmpeg4.0
int size = av_image_get_buffer_size(pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, 1);
uint8_t* picture_buf = (uint8_t*)av_malloc(size);
av_image_fill_arrays(pictureFrame->data, pictureFrame->linesize, picture_buf, AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height, 1);
// 写头文件
avformat_write_header(pFormatCtx, NULL);
// 创建已编码帧
AVPacket* pkt = av_packet_alloc();
if (!pkt)
{
return -1;
}
// 编码器Context大小
int y_size = pCodecCtx->width * pCodecCtx->height;
//循环每一帧
for (int i = 0; i < framenum; i++)
{
// 读入YUV
if (fread(picture_buf, 1, y_size * 3 / 2, in_file) <= 0)
{
cout << "read file fail!" << endl;
return -1;
}
else if (feof(in_file))
{
break;
}
pictureFrame->data[0] = picture_buf; // Y
pictureFrame->data[1] = picture_buf + y_size; // U
pictureFrame->data[2] = picture_buf + y_size * 5 / 4;// V
pictureFrame->pts = i;
// encode the image
encode(pCodecCtx, pictureFrame, pkt, out_file);
}
// flush the encoder
encode(pCodecCtx, NULL, pkt, out_file);
av_write_trailer(pFormatCtx);
// 释放内存
av_frame_free(&pictureFrame);
av_packet_free(&pkt);
avcodec_free_context(&pCodecCtx);
av_free(picture_buf);
avformat_free_context(pFormatCtx);
fclose(out_file);
fclose(in_file);
}
int main(int argc, char* argv[])
{
// 输入YUV序列
const char* input_fileName = "./out.yuv";
// 输入YUV宽度和高度、帧数
int in_w = 1920, in_h = 768;
int framenum = 999;
// 输出H264文件
const char* out_fileName = "./test_out.h264";
if (EncodeYUVToH264(input_fileName, out_fileName, in_w, in_h, framenum) < 0)
cout << "encode failed!" << endl;
cout << "encode finished!" << endl;
return getchar();
}
ffmpeg YUV420P保存H264文件
最新推荐文章于 2024-05-28 11:05:48 发布