videodecodermediaio.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/videodecoder.h>
#include <promeki/pixelformat.h>
#include <promeki/string.h>
#include <promeki/videocodec.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class VideoDecoderMediaIO : public SharedThreadMediaIO {
                PROMEKI_OBJECT(VideoDecoderMediaIO, SharedThreadMediaIO)
        public:
                static inline const MediaIOStats::ID StatsPacketsDecoded{"PacketsDecoded"};
 
                static inline const MediaIOStats::ID StatsImagesOut{"ImagesOut"};
 
                VideoDecoderMediaIO(ObjectBase *parent = nullptr);
                ~VideoDecoderMediaIO() 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 every currently-available output Frame out of
                // the underlying decoder and pushes them onto
                // @c _outputQueue.  The decoder is responsible for
                // echoing the source Frame's audio / metadata through
                // onto each emitted Frame via the base
                // @ref VideoDecoder::buildOutputFrame helper.
                void  drainDecoderInto();
                Error createDecoder(const VideoCodec &codec);
 
                MediaConfig        _config;
                VideoCodec         _codec;
                VideoDecoder::UPtr _decoder;
                PixelFormat        _outputPixelFormat;
                bool               _outputPixelFormatSet = false;
                int                _capacity = 8;
                Frame::List        _outputQueue;
                FrameCount         _frameCount{0};
                int64_t            _readCount = 0;
                int64_t            _packetsDecoded = 0;
                int64_t            _imagesOut = 0;
                bool               _capacityWarned = false;
                bool               _closed = false;
};
 
class VideoDecoderFactory : public MediaIOFactory {
        public:
                VideoDecoderFactory() = default;
 
                String name() const override { return String("VideoDecoder"); }
                String displayName() const override { return String("Video Decoder"); }
                String description() const override {
                        return String("Decodes compressed video bitstreams into uncompressed frames");
                }
                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