h264bitstream.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_PROAV
#include <cstdint>
#include <functional>
#include <promeki/function.h>
#include <promeki/namespace.h>
#include <promeki/buffer.h>
#include <promeki/bufferview.h>
#include <promeki/error.h>
#include <promeki/list.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class H264Bitstream {
        public:
                struct NalUnit {
                                BufferView view;        
                                uint8_t    header0 = 0; 
                                uint8_t    header1 = 0; 
                };
 
                using Visitor = Function<Error(const NalUnit &)>;
 
                static Error forEachAnnexBNal(const BufferView &in, const Visitor &visit);
 
                static Error forEachAvccNal(const BufferView &in, uint8_t lenSize, const Visitor &visit);
 
                static Error annexBToAvcc(const BufferView &in, uint8_t lenSize, Buffer &outBuf);
 
                static Error annexBToAvccFiltered(const BufferView &in, uint8_t lenSize,
                                                  const Function<bool(const NalUnit &)> &keep,
                                                  Buffer                                &outBuf);
 
                static Error avccToAnnexB(const BufferView &in, uint8_t lenSize, Buffer &outBuf);
 
                static Error wrapNalsAsAnnexB(const List<BufferView> &nals, Buffer &outBuf);
 
                struct SpsInfo {
                                uint32_t width = 0;
                                uint32_t height = 0;
                                uint8_t  chromaFormatIdc = 1;
                                uint8_t  bitDepthLumaMinus8 = 0;
                                uint8_t  bitDepthChromaMinus8 = 0;
                                uint8_t  profileIdc = 0; 
                                uint8_t  levelIdc = 0;   
                                bool     frameMbsOnly = true; 
                                uint16_t sarWidth = 0;
                                uint16_t sarHeight = 0;
                };
 
                static Error parseSpsResolution(const BufferView &sps, SpsInfo &out);
};
 
struct AvcDecoderConfig {
                uint8_t           configurationVersion = 1; 
                uint8_t           avcProfileIndication = 0; 
                uint8_t           profileCompatibility = 0; 
                uint8_t           avcLevelIndication = 0;   
                uint8_t           lengthSizeMinusOne = 3;   
                List<Buffer> sps; 
                List<Buffer> pps; 
 
                static Error fromAnnexB(const BufferView &au, AvcDecoderConfig &out);
 
                static Error parse(const BufferView &payload, AvcDecoderConfig &out);
 
                Error serialize(Buffer &outBuf) const;
 
                Error toAnnexB(Buffer &outBuf) const;
 
                static bool isIdrAnnexB(const BufferView &au);
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_PROAV