我目前在尝试将原始H264 NAL数据包封装到mp4容器中,但遇到了问题。不过,我希望将结果存储在内存中,而不是写入磁盘。我按照这个方法Raw H264 frames in mpegts container using libavcodec去做,但迄今为止还没有成功。
首先,这是写入内存的正确方法吗?我在头文件中有一个小结构体。
struct IOOutput {
uint8_t* outBuffer;
int bytesSet;
};
我在这里初始化缓冲区和字节设置。然后我初始化我的AVIOContext变量。
AVIOContext* pIOCtx = avio_alloc_context(pBuffer, iBufSize, 1, outptr, NULL, write_packet, NULL);
其中outptr是指向IOOutput输出的void指针,write_packet如下所示:
int write_packet (void *opaque, uint8_t *buf, int buf_size) {
IOOutput* out = reinterpret_cast<IOOutput*>(opaque);
memcpy(out->outBuffer+out->bytesSet, buf, buf_size);
out->bytesSet+=buf_size;
return buf_size;
}
我随后设置。
fc->pb = pIOCtx;
fc->flags = AVFMT_FLAG_CUSTOM_IO;
在我的AVFormatContext *fc变量上。
然后,每当我编码帧中的nal包时,我通过av_interleaved_write_frame将它们写入AVFormatContext,然后通过以下方法获取mp4内容:
void getBufferContent(char* buffer) {
memcpy(buffer, output.outBuffer, output.bytesSet);
output.bytesSet=0;
}
因此重置变量bytesSet,所以下一次写操作时将插入字节到缓冲区的开始。有更好的方法吗?这样做是否有效?如果我仅调用av_interleaved_write_frame和avformat_write_header以添加数据包,FFMPEG会执行任何读取操作吗?
非常感谢您的预先帮助!
编辑:
这是有关复用过程的代码 - 在我的编码函数中,我有类似以下的内容:
int frame_size = x264_encoder_encode(obj->mEncoder, &obj->nals, &obj->i_nals, obj->pic_in, obj->pic_out);
int total_size=0;
for(int i=0; i<obj->i_nals;i++)
{
if ( !obj->fc ) {
obj->create( obj->nals[i].p_payload, obj->nals[i].i_payload );
}
if ( obj->fc ) {
obj->write_frame( obj->nals[i].p_payload, obj->nals[i].i_payload);
}
}
// Here I get the output values
int currentBufferSize = obj->output.bytesSet;
char* mem = new char[currentBufferSize];
obj->getBufferContent(mem);
而创建和写入功能看起来像这样:
int create(void *p, int len) {
AVOutputFormat *of = av_guess_format( "mp4", 0, 0 );
fc = avformat_alloc_context();
// Add video stream
AVStream *pst = av_new_stream( fc, 0 );
vi = pst->index;
void* outptr = (void*) &output;
// Create Buffer
pIOCtx = avio_alloc_context(pBuffer, iBufSize, 1, outptr, NULL, write_packet, NULL);
fc->oformat = of;
fc->pb = pIOCtx;
fc->flags = AVFMT_FLAG_CUSTOM_IO;
pcc = pst->codec;
AVCodec c= {0};
c.type= AVMEDIA_TYPE_VIDEO;
avcodec_get_context_defaults3( pcc, &c );
pcc->codec_type = AVMEDIA_TYPE_VIDEO;
pcc->codec_id = codec_id;
pcc->bit_rate = br;
pcc->width = w;
pcc->height = h;
pcc->time_base.num = 1;
pcc->time_base.den = fps;
}
void write_frame( const void* p, int len ) {
AVStream *pst = fc->streams[ vi ];
// Init packet
AVPacket pkt;
av_init_packet( &pkt );
pkt.flags |= ( 0 >= getVopType( p, len ) ) ? AV_PKT_FLAG_KEY : 0;
pkt.stream_index = pst->index;
pkt.data = (uint8_t*)p;
pkt.size = len;
pkt.dts = AV_NOPTS_VALUE;
pkt.pts = AV_NOPTS_VALUE;
av_interleaved_write_frame( fc, &pkt );
}