Freeswitch的mod_av模块优化

最近在解决一个Freeswitch录制视频后出现视频文件播放花屏问题，大概梳理mod_av模块的录制流程，并做了一些流程上的优化。

mod_av的录制流程如下：

收到的RTP音视频包-> 解码 -> 然后写文件（首先是编码-> 然后调用ffmpeg的音视频混合接口输出为mp4文件）

首先做的流程优化，去掉画中横线的两个流程，直接将接收到的编码帧写文件：

收到的RTP音视频包-> 解码 -> 然后写文件（首先是~~编码-~~> 调用ffmpeg的音视频混合接口输出为mp4文件）

然后把录制出来的文件播放，发现还是花屏，多次尝试后发现，是在调用av_packet_rescale_ts后，才调用av_interleaved_write_frame，而av_packet_rescale_ts函数则是把AVCodecContext 的time_base转换为AVStream的time_base，先看看time_base的理解：ffmpeg存在多个时间基准(time_base)，对应不同的阶段(结构体)，每个time_base具体的值不一样，ffmpeg提供函数在各个time_base中进行切换。

AVStream->time_base比AVCodecContext->time_base精度要高(数值要小)，比如AVStream->time_base为1/90000，而AVCodecContext->time_base为1/30(假设frame_rate为30)；同样的pts和dts，以AVStream->time_base为单位，数值要比以AVCodecContext->time_base为单位要大。

编码后，pkt.pts、pkt.dts使用AVCodecContext->time_base为单位，后通过调用"av_packet_rescale_ts"转换为AVStream->time_base为单位。

首先修改time_base的初始化赋值：

原来的值：

		mst->st->time_base.den = 1000;
		mst->st->time_base.num = 1;
		c->time_base.den = 1000;
		c->time_base.num = 1;

修改为：

		mst->st->time_base.den = 90000;
		mst->st->time_base.num = 1;
		c->time_base.den = 25;
		c->time_base.num = 1;

从网络接收到的rtp包是编码后的包，首先需要把多个rtp包组成一个完整的frame的包，然后pts需要重设，代码：

        	       if (frame->m) {
                		AVPacket *pkt = (AVPacket *)malloc(sizeof(AVPacket));
                		uint32_t size = switch_buffer_inuse(context_temp->nalu_buffer);
                            if (size > 0){
                                uint8_t *temp_data; 	    
                                static uint8_t ff_input_buffer_padding[FF_INPUT_BUFFER_PADDING_SIZE] = { 0};

			                    switch_buffer_write(context_temp->nalu_buffer, ff_input_buffer_padding, sizeof(ff_input_buffer_padding));  //lyz delete

                                switch_buffer_peek_zerocopy(context_temp->nalu_buffer, (const void **)&temp_data);

                                av_init_packet(pkt);
                                av_new_packet(pkt, size);
                                pkt->size = size;

                                if (switch_test_flag(frame, SFF_WAIT_KEY_FRAME)){
                                        pkt->flags        |= AV_PKT_FLAG_KEY;
                                }

                                pkt->pts = context_temp->pts++;//context_temp->pts + 40;//((frame->timestamp - context_temp->last_received_timestamp)/90);
                                pkt->dts = pkt->pts; 
                                //pkt->duration = 40;

                                //context_temp->pts = pkt->pts;
                                
                                memcpy((unsigned char *)pkt->data, (unsigned char *)temp_data, size);
                                /*
                                switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "write video stream,pkt:%08x, data:%08x, pkt->pts:%d, data[0]:%02x, data[1]:%02x, data[2]:%02x, data[3]:%02x, data[4]:%02x\n", 
                                    pkt, 
                                    pkt->data, 
                                    pkt->pts, 
                                    pkt->data[0],
                                    pkt->data[1],
                                    pkt->data[2],
                                    pkt->data[3],
                                    pkt->data[4]);
                                    */

                        	    context_temp->last_received_timestamp = frame->timestamp;
                        	    context_temp->last_received_complete_picture = frame->m ? SWITCH_TRUE : SWITCH_FALSE;

                                switch_queue_push(context->eh.video_queue, pkt);
                            }
                        
                		switch_buffer_zero(context_temp->nalu_buffer);
        	       }

注释掉mod_av原来设置pts的代码，

if (switch_queue_pop(eh->video_queue, &pop) == SWITCH_STATUS_SUCCESS) {
                     switch_img_free(&img);

			if (!pop) {
				goto endfor;
			}

                    pkt = (AVPacket *)pop;



            ret = write_frame(eh->fc, &eh->video_st->st->codec->time_base, eh->video_st->st, pkt);

后来发现，不修改录制流程，只修改编码时pts的值也能解决这个问题，只需要把eh->video_st->frame->pts每次累加即可。

Freeswitch的mod_av模块优化

猜你喜欢