使用ffmpeg開源庫將h264封裝為mp4格式
阿新 • • 發佈:2019-02-07
#include "StdAfx.h"
#include "format.h"
ITAVFormat::ITAVFormat(void)
{
m_ifmt_ctx = NULL;
m_ofmt_ctx = NULL;
m_ofmt = NULL;
m_bufAvalloc = NULL;
m_fileName = NULL;
for(int i=0; i<BUF_UNIT_NUM; i++)
{
m_bufDeq.data[i].ptr = new unsigned char[UNIT_SIZE];
memset(m_bufDeq.data[i].ptr,0,UNIT_SIZE);
m_bufDeq.data[i].size = 0;
m_bufDeq.data[i].readsize = 0;
}
m_bufDeq.head = 0;
m_bufDeq.tail = 0;
m_countFill = 0;
m_frame_index = 0;
m_startCode = 0;
}
ITAVFormat::~ITAVFormat(void)
{
for(int i=0; i<BUF_UNIT_NUM; i++)
{
delete []m_bufDeq.data[i].ptr;
m_bufDeq.data[i].size = 0;
m_bufDeq.data[i].readsize = 0;
}
m_bufDeq.head = 0;
m_bufDeq.tail = 0;
}
int ITAVFormat::read_data(void *opaque, uint8_t *buf, int buf_size)
{
//printf("readData!!!!!!!!!!!!\n");
ITAVFormat* pThis = (ITAVFormat*)opaque;
if(pThis->m_bufDeq.head == pThis->m_bufDeq.tail)
{
//printf("*********head==tail********buf_size=%ld\n",buf_size);
return 0;
}
else
{
int head = pThis->m_bufDeq.head;
buf_size = FFMIN(buf_size, pThis->m_bufDeq.data[head].size);
printf("ptr:%p,size:%u,buf_size=%u\n",pThis->m_bufDeq.data[head].ptr, pThis->m_bufDeq.data[head].size,buf_size);
/* copy internal buffer data to buf */
memcpy(buf, pThis->m_bufDeq.data[head].ptr+pThis->m_bufDeq.data[head].readsize, buf_size);
pThis->m_bufDeq.data[head].readsize += buf_size;
pThis->m_bufDeq.data[head].size -= buf_size;
if(pThis->m_bufDeq.data[head].size <= 0)
{
pThis->m_bufDeq.data[head].readsize = 0;
pThis->m_bufDeq.head = (pThis->m_bufDeq.head+1)%BUF_UNIT_NUM;
}
return buf_size;
}
}
void ITAVFormat::init()
{
av_register_all();
}
int ITAVFormat::fillBuffer(unsigned char* buff,int len)
{
//buffer_data b
if(buff == NULL)
return TRUE;
if(len > UNIT_SIZE)
return FALSE;
int tail = m_bufDeq.tail;
memcpy(m_bufDeq.data[tail].ptr,buff,len);
m_bufDeq.data[tail].size = len;
m_bufDeq.data[tail].readsize = 0;
m_bufDeq.tail = (m_bufDeq.tail+1)%BUF_UNIT_NUM;
if(m_countFill == 0)
{
m_startCode = start();
if(m_startCode < 0)
return FALSE;
m_countFill++;
}
else
writeFrame();
return TRUE;
}
int ITAVFormat::writeFrame()
{
int ret = -1;
AVPacket pkt;
while(1)
{
AVStream *in_stream, *out_stream;
//Get an AVPacket
ret = av_read_frame(m_ifmt_ctx, &pkt);
//printf("ret=%d\n",ret);
if (ret < 0)
{
printf("ret = %d\n",ret);
break;
}
in_stream = m_ifmt_ctx->streams[pkt.stream_index];
out_stream = m_ofmt_ctx->streams[pkt.stream_index];
///如果沒有顯示時間戳自己加上時間戳並且將顯示時間戳賦值給解碼時間戳
if(pkt.pts==AV_NOPTS_VALUE){
//Write PTS
AVRational time_base1=in_stream->time_base;
//Duration between 2 frames (us)
int64_t calc_duration=(double)AV_TIME_BASE/av_q2d(in_stream->r_frame_rate);
//Parameters
pkt.pts=(double)(m_frame_index*calc_duration)/(double)(av_q2d(time_base1)*AV_TIME_BASE);
pkt.dts=pkt.pts;
pkt.duration=(double)calc_duration/(double)(av_q2d(time_base1)*AV_TIME_BASE);
m_frame_index++;
}
//Convert PTS/DTS
pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));
pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));
pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);
pkt.pos = -1;
//Write
if (av_interleaved_write_frame(m_ofmt_ctx, &pkt) < 0) {
printf( "Error muxing packet\n");
break;
}
//printf("Write %8d frames to output file\n",frame_index);
av_free_packet(&pkt);
m_frame_index++;
}
return TRUE;
}
// av_probe_input_buffer2
// AVPROBE_SCORE_MAX
int ITAVFormat::start()
{
int ret,i;
AVIOContext* pb = NULL;
AVInputFormat* piFmt = NULL;
m_bufAvalloc = (unsigned char*)av_malloc(32768);
pb = avio_alloc_context(m_bufAvalloc, 32768, 0, (void*)this, ITAVFormat::read_data, NULL, NULL);
if (av_probe_input_buffer(pb, &piFmt, "", NULL, 0, 0) < 0)
return -1;
else{
printf("format:%s[%s]\n", piFmt->name, piFmt->long_name);
}
m_ifmt_ctx = avformat_alloc_context();
m_ifmt_ctx->pb = pb;
//Input
if (avformat_open_input(&m_ifmt_ctx, "", piFmt, NULL) != 0){//iformat,priv_data賦值,pb, nbstreams,streams為null
printf("Couldn't open input stream.(無法開啟輸入流)\n");
return -2;
}
if ((ret = avformat_find_stream_info(m_ifmt_ctx, 0)) < 0) {
printf( "Failed to retrieve input stream information");
return -3;
}
//Output
avformat_alloc_output_context2(&m_ofmt_ctx, NULL, NULL, m_fileName);
if (!m_ofmt_ctx) {
printf( "Could not create output context\n");
return -4;
//goto end;
}
m_ofmt = m_ofmt_ctx->oformat;
for (i = 0; i < m_ifmt_ctx->nb_streams; i++) {
//Create output AVStream according to input AVStream
AVStream *in_stream = m_ifmt_ctx->streams[i];
AVStream *out_stream = avformat_new_stream(m_ofmt_ctx, in_stream->codec->codec);
if (!out_stream) {
printf( "Failed allocating output stream\n");
return -5;
}
//Copy the settings of AVCodecContext
if (avcodec_copy_context(out_stream->codec, in_stream->codec) < 0) {
printf( "Failed to copy context from input to output stream codec context\n");
return -6;
}
out_stream->codec->codec_tag = 0;
if (m_ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
{
out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
}
}
//Output information------------------
av_dump_format(m_ofmt_ctx, 0, m_fileName, 1);
//Open output file
if (!(m_ofmt->flags & AVFMT_NOFILE)) {
ret = avio_open(&m_ofmt_ctx->pb, m_fileName, AVIO_FLAG_WRITE);
if (ret < 0) {
printf( "Could not open output file '%s'", m_fileName);
return -7;
}
}
//Write file header
if (avformat_write_header(m_ofmt_ctx, NULL) < 0) {
printf( "Error occurred when opening output file\n");
return -8;
}
return TRUE;
}
void ITAVFormat::setFileName(char* fileName)
{
m_fileName = fileName;
}
void ITAVFormat::finish()
{
if(m_startCode < 0)
return;
//Write file trailer
av_write_trailer(m_ofmt_ctx);
//av_free(m_bufAvalloc);
avformat_close_input(&m_ifmt_ctx);
/* close output */
if (m_ofmt_ctx && !(m_ofmt->flags & AVFMT_NOFILE))
avio_close(m_ofmt_ctx->pb);
avformat_free_context(m_ofmt_ctx);
//av_free(m_bufAvalloc);
}