qt+ffmpeg设计的录屏与推流软件完整源码.zip

#include "ffmpeg_EncodeThread.h" //关闭流 void ffmpeg_EncodeThread::ffmpeg_close_stream(OutputStream *ost) { avcodec_free_context(&ost->enc); av_frame_free(&ost->frame); av_frame_free(&ost->tmp_frame); sws_freeContext(ost->sws_ctx); swr_free(&ost->swr_ctx); } //添加输出流 void ffmpeg_EncodeThread::ffmpeg_add_stream(OutputStream *ost, AVFormatContext *oc,AVCodec **codec,enum AVCodecID codec_id) { AVCodecContext *c; *codec = avcodec_find_encoder(codec_id); ost->st = avformat_new_stream(oc,nullptr); ost->st->id = oc->nb_streams-1; c=avcodec_alloc_context3(*codec); ost->enc = c; switch ((*codec)->type) { case AVMEDIA_TYPE_AUDIO: //设置数据格式 c->sample_fmt = FFMPEG_AudioFormat; c->bit_rate = AUDIO_BIT_RATE_SET; //设置码率 c->sample_rate = AUDIO_RATE_SET; //音频采样率 //设置采样通道 c->channels= av_get_channel_layout_nb_channels(c->channel_layout); c->channel_layout = AUDIO_CHANNEL_SET; //AV_CH_LAYOUT_MONO 单声道 AV_CH_LAYOUT_STEREO 立体声 c->channels=av_get_channel_layout_nb_channels(c->channel_layout); //通道数 ost->st->time_base={1,c->sample_rate}; //c->strict_std_compliance = FF_COMPLIANCE_EXPERIMENTAL; //允许使用实验性AAC编码器-增加的 break; case AVMEDIA_TYPE_VIDEO: c->codec_id = codec_id; //码率：影响体积，与体积成正比：码率越大，体积越大；码率越小，体积越小。 c->bit_rate = m_RecordConfig->video_bit_rate; //设置码率400000 400kps 500000 /*分辨率必须是2的倍数。 */ c->width = m_RecordConfig->ImageWidth; c->height = m_RecordConfig->ImageHeight; /*时基：这是基本的时间单位，以秒为单位*/ ost->st->time_base={1,m_RecordConfig->video_frame}; //按帧率计算时间基准 c->framerate = {m_RecordConfig->video_frame,1}; //设置帧率 c->time_base = ost->st->time_base; c->gop_size = 10; // 关键帧间隔 /* 最多每十二帧发射一帧内帧 */ c->pix_fmt = AV_PIX_FMT_YUV420P; //固定设置为420P c->max_b_frames = 0; // 不使用b帧 if(c->codec_id == AV_CODEC_ID_MPEG1VIDEO) { c->mb_decision = 2; } // 预设：快速 av_opt_set(c->priv_data,"preset", "superfast", 0); break; default: break; } /* 某些格式希望流头分开*/ if(oc->oformat->flags & AVFMT_GLOBALHEADER)c->flags |= AV_CODEC_FLAG_GLOBAL_HEADER; } AVFrame *ffmpeg_EncodeThread::ffmpeg_alloc_picture(enum AVPixelFormat pix_fmt, int width, int height) { AVFrame *picture; picture=av_frame_alloc(); picture->format = pix_fmt; picture->width = width; picture->height = height; /*为帧数据分配缓冲区*/ av_frame_get_buffer(picture,32); return picture; } void ffmpeg_EncodeThread::ffmpeg_open_video(AVCodec *codec, OutputStream *ost, AVDictionary *opt_arg) { AVCodecContext *c = ost->enc; AVDictionary *opt = nullptr; av_dict_copy(&opt, opt_arg, 0); //H264编码器的一些参数设置 c->qmin = 10; c->qmax = 51; //Optional Param c->max_b_frames = 0; #if 1 //下面设置两个参数影响编码延时，如果不设置，编码器默认会缓冲很多帧 // Set H264 preset and tune // av_dict_set(&opt, "preset", "fast", 0); // av_dict_set(&opt, "tune", "zerolatency", 0); #else /** * ultrafast,superfast, veryfast, faster, fast, medium * slow, slower, veryslow, placebo. 注意：这是x264编码速度的选项， 设置该参数可以降低编码延时 */ av_opt_set(c->priv_data,"preset","superfast",0); #endif avcodec_open2(c, codec, &opt); av_dict_free(&opt); ost->frame = ffmpeg_alloc_picture(c->pix_fmt, c->width,c->height); ost->tmp_frame = nullptr; /* 将流参数复制到多路复用器 */ avcodec_parameters_from_context(ost->st->codecpar,c); } /* 准备图像数据 YUV422占用内存空间 = w * h * 2 YUV420占用内存空间 = width*height*3/2 */ void ffmpeg_EncodeThread::fill_yuv_image(AVFrame *pict, int frame_index,int width, int height) { unsigned int y_size=width*height; m_RecordConfig->video_encode_mutex.lock(); m_RecordConfig->video_WaitConditon.wait(&m_RecordConfig->video_encode_mutex); memcpy(m_RecordConfig->video_yuv420p_buff_temp, m_RecordConfig->video_yuv420p_buff, m_RecordConfig->yuv420p_size); m_RecordConfig->video_encode_mutex.unlock(); //将YUV数据拷贝到缓冲区 y_size=wXh memcpy(pict->data[0],m_RecordConfig->video_yuv420p_buff_temp,y_size); memcpy(pict->data[1],m_RecordConfig->video_yuv420p_buff_temp+y_size,y_size/4); memcpy(pict->data[2],m_RecordConfig->video_yuv420p_buff_temp+y_size+y_size/4,y_size/4); } //获取一帧视频 AVFrame *ffmpeg_EncodeThread::get_video_frame(OutputStream *ost) { AVCodecContext *c = ost->enc; /* 检查我们是否要生成更多帧---判断是否结束录制 */ // if(av_compare_ts(ost->next_pts, c->time_base,STREAM_DURATION, (AVRational){ 1, 1 }) >= 0) // return nullptr; /*当我们将帧传递给编码器时，它可能会保留对它的引用 *内部； 确保我们在这里不覆盖它*/ if (av_frame_make_writable(ost->frame) < 0) return nullptr; //获取图像 //DTS（解码时间戳）和PTS（显示时间戳） fill_yuv_image(ost->frame, ost->next_pts, c->width, c->height); ost->frame->pts = ost->next_pts++; //视频帧添加水印 return ost->frame; } int ffmpeg_EncodeThread::ffmpeg_write_frame(AVFormatContext *fmt_ctx, const AVRational *time_base, AVStream *st, AVPacket *pkt) { /*将输出数据包时间戳值从编解码器重新调整为流时基 */ av_packet_rescale_ts(pkt, *time_base, st->time_base); pkt->stream_index = st->index; /*将压缩的帧写入媒体文件*/ return av_interleaved_write_frame(fmt_ctx, pkt); } /* *编码一个视频帧并将其发送到多路复用器 */ int ffmpeg_EncodeThread::ffmpeg_write_video_frame(AVFormatContext *oc, OutputStream *ost) { int ret; AVCodecContext *c; AVFrame *frame; int got_packet = 0; AVPacket pkt; c=ost->enc; //获取一帧数据 frame = get_video_frame(ost); if(frame==nullptr) { emit LogSend(-1,"视频帧制作失败...\n"); return -1; } av_init_packet(&pkt); //编码图像 ret=avcodec_encode_video2(c, &pkt, frame, &got_packet); if(ret < 0) { emit LogSend(-1,"视频帧编码出错...\n"); return ret; } if(got_packet) { ret=ffmpeg_write_frame(oc, &c->time_base, ost->st, &pkt); } else { ret = 0; } if(ret < 0) { emit LogSend(-1,"写入视频帧时出错...\n"); return ret; } return 0; } /*获取一帧音频数据 */ AVFrame *ffmpeg_EncodeThread::get_audio_frame(OutputStream *ost) { AVFrame *frame = ost->tmp_frame; int16_t *q = (int16_t*)frame->data[0]; /* 检查我们是否要生成更多帧----用于判断是否结束*/ // if(av_compare_ts(ost->next_pts, ost->enc->time_base, // STREAM_DURATION, (AVRational){ 1, 1 }) >= 0) // { // // if(videoaudioencode.audio_data_queue.isEmpty()) // return nullptr; // } // qDebug("frame->nb_samples=%d\n",frame->nb_samples); //2048 // qDebug("ost->enc->channels=%d\n",ost->enc->channels); //1 //判断缓冲区的数据是