video-pc/AvRecorder/recorder/video_recorder.h

#ifndef __VIDEO_RECORDER_H__
#define __VIDEO_RECORDER_H__

#include "basic/timer.h"
#include "capturer/video/video_capturer.h"
#include "muxer/av_muxer.h"
#include <deque>
// #include <condition_variable>
// #include <queue>

class VideoRecorder {
public:
    bool Open(HWND srcHwnd, Encoder<MediaType::VIDEO>::Param& param, VideoCapturer::Method type);
    bool Open(int monitorIdx, Encoder<MediaType::VIDEO>::Param& param, VideoCapturer::Method type);
    bool LoadMuxer(AvMuxer& muxer);
    bool StartRecord();
    void StopRecord();
    auto GetCapturerType() { return _capturer.GetMethod(); }
    AVFrame* GetRenderFrame();
    // 停止录制
    void Close();
    void SetIsDrawCursor(bool isDraw)
    {
        _capturer.SetDrawCursor(isDraw);
    }
    bool IsCaptureOverload() const { return _captureTimer.IsOverload(); }
    // 返回滑动窗口丢帧率，窗口未满时返回-1
    double GetLossRate();

private:
    bool _Open(Encoder<MediaType::VIDEO>::Param& param);
    VideoCapturer _capturer;
    AvMuxer* _muxer = nullptr;
    bool _isRecord = false;
    int _streamIndex = -1;
    AVFrame* _encodeFrame = nullptr;
    AVFrame* _renderFrame = nullptr;
    Encoder<MediaType::VIDEO>::Param _param;
    Timer _captureTimer;
    Timer _muxTimer;
    std::mutex _renderMtx;
    uint64_t _totalPts = 0;
    uint64_t _lossPts = 0;
    // 滑动窗口丢帧统计
    static constexpr int LOSS_WINDOW = 100;
    std::deque<bool> _lossHistory;
};
#endif