videoencodermediaio.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_PROAV
#include <promeki/namespace.h>
#include <promeki/sharedthreadmediaio.h>
#include <promeki/mediaiofactory.h>
#include <promeki/mediaconfig.h>
#include <promeki/videoencoder.h>
#include <promeki/string.h>
#include <promeki/videocodec.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class VideoEncoderMediaIO : public SharedThreadMediaIO {
                PROMEKI_OBJECT(VideoEncoderMediaIO, SharedThreadMediaIO)
        public:
                static inline const MediaIOStats::ID StatsFramesEncoded{"FramesEncoded"};
 
                static inline const MediaIOStats::ID StatsPacketsOut{"PacketsOut"};
 
                VideoEncoderMediaIO(ObjectBase *parent = nullptr);
                ~VideoEncoderMediaIO() override;
 
                Error describe(MediaIODescription *out) const override;
                Error proposeInput(const MediaDesc &offered, MediaDesc *preferred) const override;
                Error proposeOutput(const MediaDesc &requested, MediaDesc *achievable,
                                    MediaConfig *configDelta = nullptr) const override;
                int   pendingInternalWrites() const override;
 
        protected:
                Error executeCmd(MediaIOCommandOpen &cmd) override;
                Error executeCmd(MediaIOCommandClose &cmd) override;
                Error executeCmd(MediaIOCommandRead &cmd) override;
                Error executeCmd(MediaIOCommandWrite &cmd) override;
                Error executeCmd(MediaIOCommandStats &cmd) override;
                void  configChanged(const MediaConfig &delta) override;
 
        private:
                // Drains the underlying encoder's ready Frames into
                // @c _outputQueue.  The encoder is responsible for
                // echoing the source Frame's audio / ANC / metadata
                // through onto each emitted Frame via the base
                // @ref VideoEncoder::buildOutputFrame helper.
                void drainEncoderInto();
 
                MediaConfig        _config;
                VideoCodec         _codec;
                VideoEncoder::UPtr _encoder;
                int                _capacity = 8;
                Frame::List        _outputQueue;
                FrameCount         _frameCount{0};
                int64_t            _readCount = 0;
                FrameCount         _framesEncoded{0};
                int64_t            _packetsOut = 0;
                bool               _capacityWarned = false;
                bool               _closed = false;
};
 
class VideoEncoderFactory : public MediaIOFactory {
        public:
                VideoEncoderFactory() = default;
 
                String name() const override { return String("VideoEncoder"); }
                String displayName() const override { return String("Video Encoder"); }
                String description() const override {
                        return String("Encodes uncompressed video frames into a registered VideoCodec bitstream");
                }
                bool canBeTransform() const override { return true; }
 
                Config::SpecMap configSpecs() const override;
                bool            bridge(const MediaDesc &from, const MediaDesc &to, Config *outConfig,
                                       int *outCost) const override;
                MediaIO        *create(const Config &config, ObjectBase *parent = nullptr) const override;
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_PROAV