avpacket 的 data 不能直接保存为h264吗?
#include <thread>
#include <iostream>
#include <fstream>
extern "C"
{
#include <libavcodec/avcodec.h>
#include <libavcodec/bsf.h>
#include <libavformat/avformat.h>
}
#pragma comment(lib,"avcodec.lib")
#pragma comment(lib,"avdevice.lib")
#pragma comment(lib,"avfilter.lib")
#pragma comment(lib,"avformat.lib")
#pragma comment(lib,"avutil.lib")
#pragma comment(lib,"swresample.lib")
#pragma comment(lib,"swscale.lib")
int mp4toannexb(AVBSFContext* bsf_ctx, AVPacket* pkt, std::ofstream& out)
{
int ret;
ret = av_bsf_send_packet(bsf_ctx, pkt);
if (ret < 0) {
std::cout << "bsf send packet failed, errno:" << ret << std::endl;
return -1;
}
for (;;) {
ret = av_bsf_receive_packet(bsf_ctx, pkt);
if (AVERROR_EOF == ret || AVERROR(EAGAIN) == ret) {
return 0;
}
if (ret < 0) {
std::cout << "Could not receive packet, errno:" << ret << std::endl;
return -1;
}
out.write((const char*)pkt->data, pkt->size);
}
return 0;
}
int main()
{
AVOutputFormat* ofmt = NULL;
AVFormatContext* ifmt_ctx = NULL;
AVBSFContext* bsf_ctx = NULL;
AVPacket* pkt = NULL;
const char* in_filename = "http://219.151.31.38/liveplay-kk.rtxapp.com/live/program/live/hnwshd/4000000/mnf.m3u8";
int ret = -1;
int video_index = -1;
std::ofstream outh264("D:\\ConsoleApplication1\\Debug\\out.h264", std::ios::binary | std::ios::trunc);
// 初始化网络
avformat_network_init();
// 初始化输入格式上下文
if ((ifmt_ctx = avformat_alloc_context()) == NULL) {
std::cout << "avformat_alloc_context failed." << std::endl;
exit(1);
}
// 打开输入格式上下文
if ((ret = avformat_open_input(&ifmt_ctx, in_filename, 0, 0)) < 0) {
std::cout << "Could not open input file." << std::endl;
exit(1);
}
// 获取输入信息
if ((ret = avformat_find_stream_info(ifmt_ctx, 0)) < 0) {
std::cout << "Failed to retrieve input stream information" << std::endl;
return -1;
}
// 申请pkt
if (!(pkt = av_packet_alloc())) {
std::cout << "Could not allocate packet" << std::endl;
return -1;
}
video_index = av_find_best_stream(ifmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
if (video_index < 0) {
std::cout << "Could not find stream " << std::string(av_get_media_type_string(AVMEDIA_TYPE_VIDEO)) << std::endl;
return -1;
}
av_dump_format(ifmt_ctx, 0, in_filename, 0);
const AVBitStreamFilter* filter = av_bsf_get_by_name("h264_mp4toannexb");
if (NULL == filter) {
std::cout << "Could not create filter" << std::endl;
return -1;
}
ret = av_bsf_alloc(filter, &bsf_ctx);
if (ret < 0) {
printf("Could not alloc bitstream filter \n");
return -1;
}
// avcodec_parameters_from_context
ret = avcodec_parameters_copy(bsf_ctx->par_in, ifmt_ctx->streams[video_index]->codecpar);
if (ret < 0) {
printf("Parameter copy filed, errno: %d \n", ret);
return -1;
}
ret = av_bsf_init(bsf_ctx);
if (ret < 0) {
printf("BSF init failed, errno: %d \n", ret);
return -1;
}
while (av_read_frame(ifmt_ctx, pkt) >= 0) {
if (pkt->stream_index == video_index)
{
mp4toannexb(bsf_ctx, pkt, outh264);
}
av_packet_unref(pkt);
}
// 释放
outh264.close();
avformat_close_input(&ifmt_ctx);
av_packet_free(&pkt);
av_bsf_free(&bsf_ctx);
return 0;
}