blob: 8f9a943de5920415616e37d79cdf4011983b4574 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
#ifndef QFFMPEGVIDEOENCODER_P_H
#define QFFMPEGVIDEOENCODER_P_H
#include "qffmpegencoderthread_p.h"
#include "qffmpeg_p.h"
#include <qvideoframe.h>
#include <queue>
QT_BEGIN_NAMESPACE
class QVideoFrameFormat;
class QMediaEncoderSettings;
namespace QFFmpeg {
class VideoFrameEncoder;
class VideoEncoder : public EncoderThread
{
public:
VideoEncoder(RecordingEngine &recordingEngine, const QMediaEncoderSettings &settings,
const QVideoFrameFormat &format, std::optional<AVPixelFormat> hwFormat);
~VideoEncoder() override;
bool isValid() const;
void addFrame(const QVideoFrame &frame);
void setPaused(bool b) override
{
EncoderThread::setPaused(b);
if (b)
m_baseTime.storeRelease(-1);
}
private:
QVideoFrame takeFrame();
void retrievePackets();
void init() override;
void cleanup() override;
bool hasData() const override;
void processOne() override;
private:
std::queue<QVideoFrame> m_videoFrameQueue;
const size_t m_maxQueueSize = 10; // Arbitrarily chosen to limit memory usage (332 MB @ 4K)
std::unique_ptr<VideoFrameEncoder> m_frameEncoder;
QAtomicInteger<qint64> m_baseTime = std::numeric_limits<qint64>::min();
qint64 m_lastFrameTime = 0;
};
} // namespace QFFmpeg
QT_END_NAMESPACE
#endif
|