qt切换页面小练习更新2

#include "ffmpegapisavevideo.h" ffmpegApiSaveVideo::ffmpegApiSaveVideo(QObject *parent) : QObject(parent) { } ffmpegApiSaveVideo::~ffmpegApiSaveVideo() { } void ffmpegApiSaveVideo::insertFrame(AVFrame *frame) { if(m_FrameVector.length()>MaxFrameNum){ m_FrameVector.pop_front(); } m_FrameVector.append(frame); } void ffmpegApiSaveVideo::init(QString filePath, AVCodecContext *pCodecCtx, ffmpegApiSaveVideo::VideoInfo videoInfo) { m_videoInfo = videoInfo; m_ICodecCtx = pCodecCtx; initOutput(filePath.toUtf8()); uint8_t *pIBuffer; //开辟存储像素点的存储地址 int pixSize = av_image_get_buffer_size(m_videoInfo.pixFormat,m_ICodecCtx->width, m_ICodecCtx->height,16); //创建保存空间,底层使用malloc进行内存空间的开辟。 pIBuffer = static_cast<uint8_t *>(av_malloc(static_cast<size_t>(pixSize))); //创建图像转换之后的帧 m_pIFrameYUV = av_frame_alloc(); av_image_fill_arrays(m_pIFrameYUV->data, m_pIFrameYUV->linesize, pIBuffer, m_videoInfo.pixFormat, 1280, 720, 16); m_imgYUV_convert_ctx = sws_getContext(m_ICodecCtx->width, m_ICodecCtx->height, m_ICodecCtx->pix_fmt, 1280, 720, m_videoInfo.pixFormat, SWS_BICUBIC, nullptr, nullptr, nullptr); issave = true; } void ffmpegApiSaveVideo::stopWrite() { issave = false; } void ffmpegApiSaveVideo::writeVideo() { while(issave){ int ret; int got_packet=0; AVPacket pkt; if(m_FrameVector.isEmpty()){ continue; } AVFrame *pIFrame = m_FrameVector.front(); int length = m_FrameVector.length(); m_FrameVector.pop_front(); if(pIFrame == nullptr){ continue; } sws_scale(m_imgYUV_convert_ctx, static_cast<uint8_t const * const *>(pIFrame->data), pIFrame->linesize,0, m_ICodecCtx->height, m_pIFrameYUV->data, m_pIFrameYUV->linesize); m_pIFrameYUV->pts = m_next_pts++; av_init_packet(&pkt); /* 编码图像*/ ret = avcodec_send_frame(m_pCodecCtx, m_pIFrameYUV); if (ret < 0) { qDebug()<<"Error sending the frame to the encoder"; return; } ret = avcodec_receive_packet(m_pCodecCtx, &pkt); if (ret < 0) { qDebug()<<"Error encoding audio frame"; return; } ret = write_frame(m_pOFormatCtx, &m_pCodecCtx->time_base, m_pStream, &pkt); QThread::msleep(10); av_packet_unref(&pkt); } av_write_trailer(m_pOFormatCtx); avcodec_free_context(&m_pCodecCtx); /*关闭输出文件*/ if (!(m_pOutFmt->flags & AVFMT_NOFILE)) avio_closep(&m_pOFormatCtx->pb); /*释放流*/ avformat_free_context(m_pOFormatCtx); } void ffmpegApiSaveVideo::initOutput(QByteArray filepath) { int ret; //根据文件路径判断获取编码格式，写入到oc中。向m_pOFormatCtx中写入数据 avformat_alloc_output_context2(&m_pOFormatCtx, nullptr, nullptr, filepath.data());//为输出格式分配一个AVFormatContext。 if(!m_pOFormatCtx) { printf("无法从文件扩展名推断出输出格式：使用h264。\n"); avformat_alloc_output_context2(&m_pOFormatCtx, nullptr, m_videoInfo.packageFormat.toUtf8(), filepath.data()); } //添加流 向m_pOutFmt和m_pCodecCtx m_pCodec写入数据。 addStreamToftc(m_pOFormatCtx); /*现在所有参数都设置好了，我们可以打开音频和视频编解码器并分配必要的编码缓冲器*/ AVDictionary *opt = nullptr;//AVDictionary是一个健值对存储工具 open_video(m_pCodec,opt); av_dump_format(m_pOFormatCtx,0,filepath.data(),1);// TODO： /* 打开输出文件(如果需要) */ if(!(m_pOFormatCtx->flags & AVFMT_NOFILE)) { //创建并初始化AVIOContext以访问url指示的资源。 ret=avio_open(&m_pOFormatCtx->pb,filepath.data(),AVIO_FLAG_WRITE); if(ret<0) { qDebug()<<QString("打不开'%1': %2").arg(filepath.data()).arg(av_err2str(ret)); } } /* 编写流头(如果有)*/ ret=avformat_write_header(m_pOFormatCtx, &opt); if(ret<0) { fprintf(stderr, "打开输出文件时发生错误: %s\n",av_err2str(ret)); } // 创建 缓存区 } //*添加输出流。 */ void ffmpegApiSaveVideo::addStreamToftc(AVFormatContext *fct) { int i; /* find the encoder */ m_pOutFmt = m_pOFormatCtx->oformat; m_pOutFmt->video_codec = m_videoInfo.codecId;//设置编码格式 m_pCodec = avcodec_find_encoder(m_pOutFmt->video_codec); if (!(m_pCodec)) { qDebug()<<"Could not find encoder for :" << avcodec_get_name(m_pOutFmt->video_codec); return; } //给媒体文件添加一个流 m_pStream = avformat_new_stream(fct, nullptr); if (!m_pStream) { qDebug()<<"Could not allocate stream"; return; } m_pStream->id = static_cast<int>(fct->nb_streams-1); //创建 编码 上下文。 m_pCodecCtx = avcodec_alloc_context3(m_pCodec); av_opt_set(m_pCodecCtx->priv_data, "tune", "zerolatency", 0);//解决avcodec_receive_packet返回为-11的问题 if (!m_pCodecCtx) { qDebug()<<"Could not alloc an encoding context"; return; } switch((m_pCodec)->type) { case AVMEDIA_TYPE_AUDIO:{ m_pCodecCtx->sample_fmt = (m_pCodec)->sample_fmts ?(m_pCodec)->sample_fmts[0] : AV_SAMPLE_FMT_FLTP; m_pCodecCtx->bit_rate = 64000; m_pCodecCtx->sample_rate = 44100; if ((m_pCodec)->supported_samplerates) { m_pCodecCtx->sample_rate = (m_pCodec)->supported_samplerates[0]; for (i = 0; (m_pCodec)->supported_samplerates[i]; i++) { if ((m_pCodec)->supported_samplerates[i] == 44100) m_pCodecCtx->sample_rate = 44100; } } m_pCodecCtx->channels = av_get_channel_layout_nb_channels(m_pCodecCtx->channel_layout); m_pCodecCtx->channel_layout = AV_CH_LAYOUT_STEREO; if ((m_pCodec)->channel_layouts) { m_pCodecCtx->channel_layout = (m_pCodec)->channel_layouts[0]; for (i = 0; (m_pCodec)->channel_layouts[i]; i++) { if ((m_pCodec)->channel_layouts[i] == AV_CH_LAYOUT_STEREO) m_pCodecCtx->channel_layout = AV_CH_LAYOUT_STEREO; } } m_pCodecCtx->channels = av_get_channel_layout_nb_channels(m_pCodecCtx->channel_layout); m_pStream->time_base = (AVRational){ 1, m_pCodecCtx->sample_rate }; break; } case AVMEDIA_TYPE_VIDEO:{ m_pCodecCtx->codec_id = m_pOutFmt->video_codec; m_pCodecCtx->bit_rate = m_videoInfo.kbps; //平均比特率,例子代码默认值是400000 /* 分辨率必须是2的倍数。*/ m_pCodecCtx->width = m_videoInfo.wideth; m_pCodecCtx->height = m_videoInfo.hight; /*时基：这是基本的时间单位（以秒为单位）表示其中的帧时间戳。 对于固定fps内容，时基应为1 /framerate，时间戳增量应为等于1。*/ m_pStream->time_base = (AVRational){1,m_videoInfo.fps}; //帧率设置 帧率为25; m_pCodecCtx->time_base = m_pStr