156 lines
11 KiB
C++
156 lines
11 KiB
C++
#include "call_video_encoder.h"
|
|
#include "debug_config.h"
|
|
#include "../../../src/call/call_video_rate.h"
|
|
#include "../../../src/call/call_video_bitstream.h"
|
|
#include <QElapsedTimer>
|
|
|
|
// Профиль и envelope задаются явно: автоматический выбор level при смене bitrate запрещён.
|
|
bool CallVideoEncoder::open(const call_video_format& format, uint32_t bitrate, const QString& backend) {
|
|
if (!call_video_format_valid(&format) || bitrate < CALL_VIDEO_RATE_MIN_BPS || bitrate > format.maximum_bps) {
|
|
m_error = "Invalid call video encoder configuration";
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call video open invalid generation=%llu codec=%u level=%u bitrate=%u cap=%u profile=%u",
|
|
(unsigned long long)format.generation, format.codec, format.level, bitrate, format.maximum_bps, format.profile); return false;
|
|
}
|
|
LiveVideoConfig config;
|
|
config.width = format.width; config.height = format.height; config.fps = format.fps; config.rotation = format.rotation;
|
|
config.codec = format.codec; config.level = format.level; config.vbvBits = 1000000;
|
|
config.bitrate = bitrate; config.references = 1; config.backend = backend;
|
|
config.profile = format.codec == CALL_VIDEO_H264 ? LiveVideoProfile::H264ConstrainedBaseline : LiveVideoProfile::H265Main;
|
|
auto candidate = std::make_unique<LiveVideoEncoder>();
|
|
if (!candidate->open(config, backend.isEmpty() ? LiveVideoEncoder::Backend::HardwareOnly : LiveVideoEncoder::Backend::Automatic)) {
|
|
m_error = candidate->error(); DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call video open failed: %s", qPrintable(m_error)); return false;
|
|
}
|
|
close(); m_encoder = std::move(candidate); m_config = config; m_config.backend = m_encoder->name(); m_format = format;
|
|
DEBUG_INFO(DEBUG_CATEGORY_CALL, "call video encoder opened generation=%llu codec=%d encoder=%s bitrate=%d level=%d",
|
|
(unsigned long long)format.generation, config.codec, qPrintable(m_config.backend), config.bitrate, config.level); return true;
|
|
}
|
|
|
|
void CallVideoEncoder::close() {
|
|
if (m_encoder) DEBUG_DEBUG(DEBUG_CATEGORY_CALL, "call video encoder close generation=%llu pending_request=%llu",
|
|
(unsigned long long)m_format.generation, (unsigned long long)m_request);
|
|
m_prepared = {}; // Завершение фоновой подготовки ждём только при остановке/смене режима.
|
|
m_replacement.reset(); m_encoder.reset(); m_config = {}; m_config.bitrate = 0; m_error.clear();
|
|
m_format = {}; m_request = m_lastRequest = m_lastPts = m_lastKey = 0;
|
|
m_lastOutputPts = m_switchPts = 0; m_backgroundMs = 0; m_switchInputs = 0;
|
|
m_pendingBitrate = 0; m_haveFrame = m_haveInput = m_keyPending = false;
|
|
}
|
|
|
|
// Ядро сериализует команды; повторные/чужие requests не заменяют уже ожидающую команду.
|
|
bool CallVideoEncoder::requestBitrate(uint64_t generation, uint64_t request, uint32_t bitrate) {
|
|
if (!m_encoder || generation != m_format.generation || !request || request <= m_lastRequest || m_request
|
|
|| bitrate < CALL_VIDEO_RATE_MIN_BPS || bitrate > m_format.maximum_bps) {
|
|
DEBUG_WARN(DEBUG_CATEGORY_CALL, "call video bitrate rejected generation=%llu/%llu request=%llu last=%llu pending=%llu bitrate=%u",
|
|
(unsigned long long)generation, (unsigned long long)m_format.generation, (unsigned long long)request,
|
|
(unsigned long long)m_lastRequest, (unsigned long long)m_request, bitrate); return false;
|
|
}
|
|
if (bitrate != (uint32_t)m_config.bitrate) {
|
|
LiveVideoConfig next = m_config; next.bitrate = bitrate;
|
|
try {
|
|
m_prepared = std::async(std::launch::async, [next] {
|
|
Prepared result; QElapsedTimer timer; timer.start();
|
|
try {
|
|
result.encoder = std::make_unique<LiveVideoEncoder>();
|
|
if (!result.encoder->open(next)) { result.error = result.encoder->error(); result.encoder.reset(); }
|
|
} catch (const std::exception& e) { result.error = QString::fromUtf8(e.what()); result.encoder.reset(); }
|
|
result.elapsed = timer.elapsed(); return result;
|
|
});
|
|
} catch (const std::exception& e) {
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call video prepare start failed request=%llu: %s", (unsigned long long)request, e.what());
|
|
return false;
|
|
}
|
|
}
|
|
m_request = m_lastRequest = request; m_pendingBitrate = bitrate;
|
|
DEBUG_DEBUG(DEBUG_CATEGORY_CALL, "call video bitrate queued generation=%llu request=%llu applied=%d target=%u last_key=%llu",
|
|
(unsigned long long)generation, (unsigned long long)request, m_config.bitrate, bitrate, (unsigned long long)m_lastKey); return true;
|
|
}
|
|
|
|
// Capture PTS продолжает идти при backpressure, поэтому запрет отправки не задерживает применение.
|
|
bool CallVideoEncoder::encode(const QImage& image, uint64_t pts, bool forceKey, QList<LiveEncodedFrame>& output,
|
|
CallVideoRateCompletion& completion) {
|
|
completion = {}; output.clear();
|
|
if (!m_encoder || image.isNull() || pts > INT64_MAX || (m_haveInput && pts <= m_lastPts)) {
|
|
m_error = "Invalid call video frame or non-monotonic PTS";
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call video encode invalid generation=%llu pts=%llu last=%llu active=%d image_null=%d",
|
|
(unsigned long long)m_format.generation, (unsigned long long)pts, (unsigned long long)m_lastPts,
|
|
m_encoder != nullptr, image.isNull()); return false;
|
|
}
|
|
bool key = !m_keyPending && (forceKey || !m_haveInput || (pts >= m_lastKey && pts - m_lastKey >= 1000000));
|
|
QList<LiveEncodedFrame> frames;
|
|
bool prepared = !m_prepared.valid() || m_prepared.wait_for(std::chrono::seconds(0)) == std::future_status::ready;
|
|
if (key && m_request && !prepared)
|
|
DEBUG_DEBUG(DEBUG_CATEGORY_CALL, "call video preparation pending request=%llu; keep current encoder", (unsigned long long)m_request);
|
|
if (key && m_request && prepared && !m_replacement) {
|
|
if (m_pendingBitrate == (uint32_t)m_config.bitrate) {
|
|
completion = {m_format.generation, m_request, m_pendingBitrate}; m_request = 0; m_pendingBitrate = 0;
|
|
}
|
|
else {
|
|
auto preparedEncoder = m_prepared.get(); m_backgroundMs = preparedEncoder.elapsed;
|
|
m_replacement = std::move(preparedEncoder.encoder); m_switchPts = pts; m_switchInputs = 0;
|
|
if (!m_replacement) {
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call GPU replacement open failed request=%llu: %s",
|
|
(unsigned long long)m_request, qPrintable(preparedEncoder.error));
|
|
completion = {m_format.generation, m_request, 0}; m_request = 0; m_pendingBitrate = 0;
|
|
}
|
|
}
|
|
}
|
|
QElapsedTimer timer; timer.start();
|
|
bool encoded = false;
|
|
if (m_replacement) {
|
|
++m_switchInputs;
|
|
encoded = m_replacement->encode(image, pts, m_switchInputs == 1, frames);
|
|
if (encoded && !frames.isEmpty()) {
|
|
const auto& first = frames.front();
|
|
encoded = first.key && first.pts == m_switchPts && !call_video_check_key(&m_format,
|
|
reinterpret_cast<const uint8_t*>(first.data.constData()), first.data.size());
|
|
if (encoded) {
|
|
const qint64 encodeUs = timer.nsecsElapsed() / 1000;
|
|
m_encoder = std::move(m_replacement); m_config.bitrate = m_pendingBitrate;
|
|
completion = {m_format.generation, m_request, m_pendingBitrate};
|
|
DEBUG_INFO(DEBUG_CATEGORY_CALL,
|
|
"call video bitrate committed generation=%llu request=%llu bitrate=%u pts=%llu background_ms=%lld "
|
|
"encode_us=%lld retire_us=%lld pipeline_frames=%u pipeline_pts_us=%llu bytes=%lld",
|
|
(unsigned long long)m_format.generation, (unsigned long long)m_request, completion.bitrate,
|
|
(unsigned long long)first.pts, (long long)m_backgroundMs, (long long)encodeUs,
|
|
(long long)(timer.nsecsElapsed() / 1000 - encodeUs), m_switchInputs,
|
|
(unsigned long long)(pts - first.pts), (long long)first.data.size());
|
|
m_request = 0; m_pendingBitrate = 0;
|
|
}
|
|
}
|
|
if (!encoded) {
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call GPU replacement failed request=%llu inputs=%u frames=%lld: %s",
|
|
(unsigned long long)m_request, m_switchInputs, (long long)frames.size(), qPrintable(m_replacement->error()));
|
|
completion = {m_format.generation, m_request, 0}; m_request = 0; m_pendingBitrate = 0;
|
|
m_replacement.reset(); frames.clear(); key = true; m_keyPending = false;
|
|
} else if (m_replacement && frames.isEmpty()) {
|
|
// До первого нового IDR продолжаем выдавать только старые кадры с PTS раньше границы замены.
|
|
// Уже выданный PTS не может обогнать ожидаемый IDR и сделать его устаревшим.
|
|
QList<LiveEncodedFrame> previous;
|
|
if (m_encoder->encode(image, pts, false, previous)) {
|
|
for (auto& frame : previous) if (frame.pts < m_switchPts) frames.push_back(std::move(frame));
|
|
} else DEBUG_WARN(DEBUG_CATEGORY_CALL, "old GPU failed while replacement is pending request=%llu: %s",
|
|
(unsigned long long)m_request, qPrintable(m_encoder->error()));
|
|
}
|
|
}
|
|
if (!encoded) encoded = m_encoder->encode(image, pts, key, frames);
|
|
m_haveInput = true; m_lastPts = pts; if (key) m_keyPending = true;
|
|
if (!encoded) {
|
|
m_error = m_encoder->error();
|
|
if (completion.request) completion.bitrate = 0;
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call video encoding failed generation=%llu: %s",
|
|
(unsigned long long)m_format.generation, qPrintable(m_error)); return false;
|
|
}
|
|
for (const auto& frame : frames) {
|
|
if ((m_haveFrame && frame.pts <= m_lastOutputPts) || frame.pts > pts || (!m_haveFrame && !frame.key)
|
|
|| (frame.key && call_video_check_key(&m_format,
|
|
reinterpret_cast<const uint8_t*>(frame.data.constData()), frame.data.size()))) {
|
|
m_error = "GPU produced an incompatible video frame";
|
|
if (completion.request) completion.bitrate = 0;
|
|
DEBUG_ERROR(DEBUG_CATEGORY_CALL, "call GPU output rejected pts=%llu last=%llu capture=%llu key=%d",
|
|
(unsigned long long)frame.pts, (unsigned long long)m_lastOutputPts, (unsigned long long)pts, frame.key); return false;
|
|
}
|
|
m_haveFrame = true; m_lastOutputPts = frame.pts;
|
|
if (frame.key) { m_lastKey = frame.pts; m_keyPending = false; }
|
|
}
|
|
output = std::move(frames); return true;
|
|
}
|