基于Qt和FFmpeg开发的一款简易录屏软件源码+项目说明.zip

#include "cscreenrecorder.h" #include <QPainter> CScreenRecorder::CScreenRecorder() { } void CScreenRecorder::Init(const QVariantMap &_map) { m_strFilePath = _map["filePath"].toString(); m_nWidth = _map["width"].toInt(); m_nHeight = _map["height"].toInt(); m_nFps = _map["fps"].toInt(); m_nAudioBitrate = _map["audioBitrate"].toInt(); m_strMicrophoneName = _map["MicrophoneName"].toString(); } void CScreenRecorder::Start() { if (m_enState == EnRecordState::EN_STOP) { cout << "start record"; m_enState = EnRecordState::EN_START; std::thread muxThread(&CScreenRecorder::muxThread, this); muxThread.detach(); } else if (m_enState == EnRecordState::EN_PAUSE) { cout << "continue record"; m_enState = EnRecordState::EN_START; m_cvNotPause.notify_one(); } } void CScreenRecorder::Pause() { m_enState = EnRecordState::EN_PAUSE; } void CScreenRecorder::Stop() { EnRecordState state = m_enState; m_enState = EnRecordState::EN_STOP; if (state == EnRecordState::EN_PAUSE) m_cvNotPause.notify_one(); } void CScreenRecorder::muxThread() { cout << " start"; int nRet = -1; bool bFinish = false; int nVideoFrameIndex = 0; int nAudioFrameIndex = 0; avdevice_register_all(); if( openVideo() == ERR ||openAudio() == ERR || openOutput() == ERR) { release(); cout << " open device failed, exit..."; return; } initVideoBuffer(); initAudioBuffer(); std::thread screenRecord(&CScreenRecorder::screenRecordThread, this); std::thread soundRecord(&CScreenRecorder::soundRecordThread, this); screenRecord.detach(); soundRecord.detach(); QThread::msleep(200); m_nVideoCurPts = m_nAudioCurPts = 0; while(true) { AVPacket pkt; av_init_packet(&pkt); if(m_enState == EnRecordState::EN_STOP && !bFinish) bFinish = true; if(bFinish) { std::unique_lock<std::mutex> vBufLock(m_mtxVBuf, std::defer_lock); std::unique_lock<std::mutex> aBufLock(m_mtxABuf, std::defer_lock); std::lock(vBufLock, aBufLock); if (av_fifo_size(m_pVideoFifoBuf) < m_nVideoOutFrameSize && av_audio_fifo_size(m_pAudioFifoBuf) < m_nbSamples ) { cout << "video buff size:" << av_fifo_size(m_pVideoFifoBuf) << " audio buff size:" << av_audio_fifo_size(m_pAudioFifoBuf) << " both video and audio fifo buf are empty, break"; break; } cout << "video buff size:" << av_fifo_size(m_pVideoFifoBuf) << " audio buff size:" << av_audio_fifo_size(m_pAudioFifoBuf); if(av_fifo_size(m_pVideoFifoBuf) >= m_nVideoOutFrameSize && m_nVideoCurPts == INT_MAX) m_nVideoCurPts = 0; if(av_audio_fifo_size(m_pAudioFifoBuf) >= m_nbSamples && m_nAudioCurPts == INT_MAX) m_nAudioCurPts = 0; } // 比较音视频pts，大于0表示视频帧在前，音频需要连续编码。 // 小于0表示，音频帧在前，应该至少编码一帧视频 if (av_compare_ts(m_nVideoCurPts, m_pOutFmtCtx->streams[m_nVideoOutIndex]->time_base, m_nAudioCurPts, m_pOutFmtCtx->streams[m_nAudioOutIndex]->time_base) <= 0) { if (bFinish) { std::lock_guard<std::mutex> lk(m_mtxVBuf); if (av_fifo_size(m_pVideoFifoBuf) < m_nVideoOutFrameSize) { cout << "video wirte done"; m_nVideoCurPts = INT_MAX; continue; } } else { std::unique_lock<std::mutex> lk(m_mtxVBuf); m_cvVBufNotEmpty.wait(lk, [this] { return av_fifo_size(m_pVideoFifoBuf) >= m_nVideoOutFrameSize; }); } av_fifo_generic_read(m_pVideoFifoBuf, m_pVideoOutFrameBuf, m_nVideoOutFrameSize, nullptr); m_cvVBufNotFull.notify_one(); m_pVideoOutFrame->pts = nVideoFrameIndex++; m_pVideoOutFrame->format = m_pVideoEncodeCtx->pix_fmt; m_pVideoOutFrame->width = m_pVideoEncodeCtx->width; m_pVideoOutFrame->height = m_pVideoEncodeCtx->height; nRet = avcodec_send_frame(m_pVideoEncodeCtx, m_pVideoOutFrame); if(nRet != OK) { cout << "video avcodec_send_frame failed:" << getAvErrorMsg(nRet); continue; } nRet = avcodec_receive_packet(m_pVideoEncodeCtx, &pkt); if(nRet != OK) { cout << "video avcodec_receive_packet failed:" << getAvErrorMsg(nRet); av_packet_unref(&pkt); continue; } pkt.stream_index = m_nVideoIndex; av_packet_rescale_ts(&pkt, m_pVideoEncodeCtx->time_base, m_pOutFmtCtx->streams[m_nVideoOutIndex]->time_base); m_nVideoCurPts = pkt.pts; nRet = av_interleaved_write_frame(m_pOutFmtCtx, &pkt); if(nRet != OK) { cout << "video av_interleaved_write_frame failed:" << nRet; } cout << "video index:" << nVideoFrameIndex; av_packet_unref(&pkt); } else { if (bFinish) { std::lock_guard<std::mutex> lk(m_mtxABuf); if (av_audio_fifo_size(m_pAudioFifoBuf) < m_nbSamples) { cout << "audio write done"; m_nAudioCurPts = INT_MAX; continue; } } else { std::unique_lock<std::mutex> lk(m_mtxABuf); m_cvABufNotEmpty.wait(lk, [this] { return av_audio_fifo_size(m_pAudioFifoBuf) >= m_nbSamples; }); } int nRet = -1; AVFrame *pFrame = av_frame_alloc(); pFrame->nb_samples = m_nbSamples; pFrame->channel_layout = m_pAudioEncodeCtx->channel_layout; pFrame->format = m_pAudioEncodeCtx->sample_fmt; pFrame->sample_rate = m_pAudioEncodeCtx->sample_rate; pFrame->pts = m_nbSamples * nAudioFrameIndex++; //分配data buf nRet = av_frame_get_buffer(pFrame, 0); if(av_audio_fifo_read(m_pAudioFifoBuf, (void **)pFrame->data, m_nbSamples) < m_nbSamples) { cout << "av_audio_fifo_read failed"; continue; } m_cvABufNotFull.notify_one(); nRet = avcodec_send_frame(m_pAudioEncodeCtx, pFrame); if (nRet != OK) { cout << "audio avcodec_send_frame failed, ret: " << nRet; av_frame_free(&pFrame); continue; } nRet = avcodec_receive_packet(m_pAudioEncodeCtx, &pkt); if (nRet != OK) { cout << "audio avcodec_receive_packet failed, ret: " << nRet; av_frame_free(&pFrame); av_packet_unref(&pkt); continue; } pkt.stream_index = m_nAudioOutIndex; av_packet_rescale_ts(&pkt, m_pAudioEncodeCtx->time_base, m_pOutFmtCtx->streams[m_nAudioOutIndex]->time_base); m_nAudioCurPts = pkt.pts; nRet = av_interleaved_write_frame(m_pOutFmtCtx, &pkt); if (nRet != OK) cout << "audio av_interleaved_write_frame failed, ret: " << nRet; cout << "audio index:" << nAudioFrameIndex; av_frame_free(&pFrame); av_packet_unref(&pkt); } } flushEncoders(); av_write_trailer(m_pOutFmtCtx); cout << " exit"; release(); } void CScreenRecorder::screenRecordThread() { cout << " start"; int nRet = -1; AVPacket pkt; av_init_packet(&pkt); int y_size = m_nWidth * m_nHe