Revision control
++ // 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 QFFMPEGENCODERTHREAD_P_H
++ #define QFFMPEGENCODERTHREAD_P_H
++
++ //
++ // W A R N I N G
++ // -------------
++ //
++ // This file is not part of the Qt API. It exists purely as an
++ // implementation detail. This header file may change from version to
++ // version without notice, or even be removed.
++ //
++ // We mean it.
++ //
++
++ #include <QtFFmpegMediaPluginImpl/private/qffmpegthread_p.h>
++
++ #include <QtCore/qfuture.h>
++ #include <QtCore/qpointer.h>
++ #include <QtCore/qpromise.h>
++ #include <QtMultimedia/private/qmediainputencoderinterface_p.h>
++
++ #include <atomic>
++
++ QT_BEGIN_NAMESPACE
++
++ namespace QFFmpeg {
++
++ class RecordingEngine;
++
++ class EncoderThread : public ConsumerThread, public QMediaInputEncoderInterface
++ {
++ Q_OBJECT
++ public:
++ EncoderThread(RecordingEngine &recordingEngine);
++
++ void setPaused(bool paused);
++
++ void setAutoStop(bool autoStop);
++
++ void setSource(QObject *source) { m_source = source; }
++
++ QObject *source() const { return m_source; }
++
++ bool canPushFrame() const override { return m_canPushFrame.load(std::memory_order_relaxed); }
++
++ void setEndOfSourceStream();
++
++ bool isEndOfSourceStream() const { return m_endOfSourceStream.load(std::memory_order_acquire); }
++
++ bool isInitialized() const { return m_initialized; }
++
++ QFuture<bool> resolvedFuture() const { return m_resolvedPromise.future(); }
++
++ protected:
++ bool init() override;
++
++ void updateCanPushFrame();
++
++ virtual bool checkIfCanPushFrame() const = 0;
++
++ void resetEndOfSourceStream() { m_endOfSourceStream.store(false, std::memory_order_release); }
++
++ auto lockLoopData()
++ {
++ return QScopeGuard([this, locker = ConsumerThread::lockLoopData()]() mutable {
++ const bool autoStopActivated = m_endOfSourceStream && m_autoStop;
++ const bool canPush = !autoStopActivated && !m_paused && checkIfCanPushFrame();
++ locker.unlock();
++ if (m_canPushFrame.exchange(canPush, std::memory_order_relaxed) != canPush)
++ Q_EMIT canPushFrameChanged();
++ });
++ }
++
++ void markResolved(bool succeeded);
++
++ virtual void handlePausedChanged(bool /*paused*/) { }
++
++ Q_SIGNALS:
++ void canPushFrameChanged();
++ void endOfSourceStream();
++
++ protected:
++ bool m_paused = false;
++ std::atomic_bool m_endOfSourceStream = false;
++ bool m_autoStop = false;
++ bool m_initialized = false;
++ bool m_encodingStarted = false;
++ std::atomic_bool m_canPushFrame = false;
++ RecordingEngine &m_recordingEngine;
++ QPointer<QObject> m_source;
++ QPromise<bool> m_resolvedPromise;
++ };
++
++ } // namespace QFFmpeg
++
++ QT_END_NAMESPACE
++
++ #endif