qffmpegencoderthread_p.h

Go to the documentation of this file.

// 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