mpegtsfilemediaio.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_PROAV
#include <cstdint>
#include <promeki/namespace.h>
#include <promeki/buffer.h>
#include <promeki/file.h>
#include <promeki/frame.h>
#include <promeki/framerate.h>
#include <promeki/mediaiofactory.h>
#include <promeki/sharedthreadmediaio.h>
#include <promeki/uniqueptr.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class MpegTsFramer;
 
class MpegTsFileMediaIO : public SharedThreadMediaIO {
                PROMEKI_OBJECT(MpegTsFileMediaIO, SharedThreadMediaIO)
        public:
                static inline const MediaIOStats::ID StatsPacketsWritten{"PacketsWritten"};
 
                static inline const MediaIOStats::ID StatsBytesWritten{"BytesWritten"};
 
                static inline const MediaIOStats::ID StatsFramesWritten{"FramesWritten"};
 
                static inline const MediaIOStats::ID StatsPacketsRead{"PacketsRead"};
 
                static inline const MediaIOStats::ID StatsBytesRead{"BytesRead"};
 
                static inline const MediaIOStats::ID StatsFramesRead{"FramesRead"};
 
                static inline const MediaIOStats::ID StatsContinuityErrors{"ContinuityErrors"};
 
                static inline const MediaIOStats::ID StatsBytesDiscarded{"BytesDiscarded"};
 
                MpegTsFileMediaIO(ObjectBase *parent = nullptr);
                ~MpegTsFileMediaIO() override;
 
                Error describe(MediaIODescription *out) const override;
                Error proposeInput(const MediaDesc &offered, MediaDesc *preferred) const override;
 
        protected:
                Error executeCmd(MediaIOCommandOpen &cmd) override;
                Error executeCmd(MediaIOCommandClose &cmd) override;
                Error executeCmd(MediaIOCommandWrite &cmd) override;
                Error executeCmd(MediaIOCommandRead &cmd) override;
                Error executeCmd(MediaIOCommandStats &cmd) override;
 
        private:
                Error openSink(const MediaIOCommandOpen &cmd);
                Error openSource(const MediaIOCommandOpen &cmd);
                Error pumpReader();
                void  applyFramerConfig(const MediaIO::Config &cfg);
 
                File   _file;
                String _filename;
                bool   _isWrite = false;
                bool   _isOpen = false;
                bool   _eof = false;
 
                UniquePtr<MpegTsFramer> _framer;
                Frame::List             _readQueue;
 
                int64_t _packetsWritten = 0;
                int64_t _bytesWritten = 0;
                int64_t _framesWritten = 0;
                int64_t _packetsRead = 0;
                int64_t _bytesRead = 0;
                int64_t _framesRead = 0;
};
 
class MpegTsFileFactory : public MediaIOFactory {
        public:
                MpegTsFileFactory() = default;
 
                String name() const override { return String("MpegTsFile"); }
                String displayName() const override { return String("MPEG-2 TS file"); }
                String description() const override {
                        return String("MPEG-2 Transport Stream file read / write "
                                      "backend (H.264 / HEVC + AAC).");
                }
                StringList extensions() const override {
                        return {String("ts"), String("m2ts"), String("mts")};
                }
                bool canBeSink() const override { return true; }
                bool canBeSource() const override { return true; }
 
                Config::SpecMap configSpecs() const override;
                MediaIO        *create(const Config &config, ObjectBase *parent = nullptr) const override;
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_PROAV