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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
// Copyright (C) 2021 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
#include "playbackengine/qffmpegdemuxer_p.h"
QT_BEGIN_NAMESPACE
namespace QFFmpeg::PlaybackEngineInternal {
Demuxer::Demuxer(AVFormatContext *context, qint64 seekPos, const StreamIndexes &streamIndexes)
: m_context(context), m_seekPos(seekPos)
{
Q_ASSERT(m_context);
for (auto i = 0; i < QPlatformMediaPlayer::NTrackTypes; ++i) {
if (streamIndexes[i] >= 0)
m_streams[streamIndexes[i]] = { static_cast<QPlatformMediaPlayer::TrackType>(i) };
}
}
void Demuxer::doNextStep()
{
ensureSeeked();
Packet packet(av_packet_alloc());
if (av_read_frame(m_context, packet.avPacket()) < 0) {
setAtEnd(true);
return;
}
const auto streamIndex = packet.avPacket()->stream_index;
auto it = m_streams.find(streamIndex);
if (it != m_streams.end()) {
it->second.dataSize += packet.avPacket()->size;
it->second.duration += packet.avPacket()->duration;
switch (it->second.trackType) {
case QPlatformMediaPlayer::TrackType::VideoStream:
emit requestProcessVideoPacket(packet);
break;
case QPlatformMediaPlayer::TrackType::AudioStream:
emit requestProcessAudioPacket(packet);
break;
case QPlatformMediaPlayer::TrackType::SubtitleStream:
emit requestProcessSubtitlePacket(packet);
break;
default:
Q_ASSERT(!"Unknown track type");
}
}
scheduleNextStep(false);
}
void Demuxer::onPacketProcessed(Packet packet)
{
if (packet.isValid()) {
const auto streamIndex = packet.avPacket()->stream_index;
auto it = m_streams.find(streamIndex);
if (it != m_streams.end()) {
it->second.dataSize -= packet.avPacket()->size;
it->second.duration -= packet.avPacket()->duration;
Q_ASSERT(it->second.dataSize >= 0);
Q_ASSERT(it->second.duration >= 0);
}
}
scheduleNextStep();
}
bool Demuxer::canDoNextStep() const
{
if (!PlaybackEngineObject::canDoNextStep() || isAtEnd() || m_streams.empty())
return false;
const bool hasSmallDuration =
std::any_of(m_streams.begin(), m_streams.end(),
[](const auto &s) { return s.second.duration < 200; });
if (hasSmallDuration)
return true;
const auto dataSize =
std::accumulate(m_streams.begin(), m_streams.end(), 0,
[](quint64 value, const auto &s) { return value + s.second.dataSize; });
if (dataSize > MaxQueueSize)
return false;
return true;
}
void Demuxer::ensureSeeked()
{
if (std::exchange(m_seeked, true))
return;
const qint64 seekPos = m_seekPos * AV_TIME_BASE / 1000000;
auto err = av_seek_frame(m_context, -1, seekPos, AVSEEK_FLAG_BACKWARD);
if (err < 0) {
qDebug() << err2str(err);
emit error(err, err2str(err));
return;
}
setAtEnd(false);
scheduleNextStep();
}
Demuxer::RequestingSignal Demuxer::signalByTrackType(QPlatformMediaPlayer::TrackType trackType)
{
switch (trackType) {
case QPlatformMediaPlayer::TrackType::VideoStream:
return &Demuxer::requestProcessVideoPacket;
case QPlatformMediaPlayer::TrackType::AudioStream:
return &Demuxer::requestProcessAudioPacket;
case QPlatformMediaPlayer::TrackType::SubtitleStream:
return &Demuxer::requestProcessSubtitlePacket;
default:
Q_ASSERT(!"Unknown track type");
}
return nullptr;
}
}
QT_END_NAMESPACE
#include "moc_qffmpegdemuxer_p.cpp"
|
