qffmpegencoderthread_p.h

Enable keyboard shortcuts

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