305 lines
8.8 KiB
C++
305 lines
8.8 KiB
C++
#pragma once
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <deque>
|
|
#include <string>
|
|
|
|
#include "bitfields.h"
|
|
#include "bitstream.h"
|
|
#include "nal.h"
|
|
|
|
namespace h264{
|
|
|
|
std::deque<nalu::nalData> analysePackets(const char *data, size_t len);
|
|
|
|
/// Struct containing pre-calculated metadata of an SPS nal unit. Width and height in pixels, fps
|
|
/// in Hz
|
|
struct SPSMeta{
|
|
uint32_t width;
|
|
uint32_t height;
|
|
double fps;
|
|
uint8_t profile;
|
|
uint8_t level;
|
|
bool sep_col_plane;
|
|
uint8_t cnt_type;
|
|
bool gaps; ///< Gaps in frame num allowed flag
|
|
bool mbs_only; ///< MBS only flag
|
|
uint16_t log2_max_frame_num;
|
|
uint16_t log2_max_order_cnt;
|
|
uint16_t max_ref_frames; ///< Maximum number of reference frames
|
|
};
|
|
|
|
/// Class for analyzing generic nal units
|
|
class NAL{
|
|
public:
|
|
NAL();
|
|
NAL(std::string &InputData);
|
|
bool ReadData(std::string &InputData, bool raw = false);
|
|
std::string AnnexB(bool LongIntro = false);
|
|
std::string SizePrepended();
|
|
int Type();
|
|
std::string getData();
|
|
|
|
protected:
|
|
uint32_t chroma_format_idc; ///< the value of chroma_format_idc
|
|
std::string MyData; ///< The h264 nal unit data
|
|
};
|
|
// NAL class
|
|
|
|
/// Special instance of NAL class for analyzing SPS nal units
|
|
class SPS : public NAL{
|
|
public:
|
|
SPS() : NAL(){}
|
|
SPS(std::string &InputData, bool raw = false);
|
|
SPSMeta getCharacteristics();
|
|
void analyzeSPS();
|
|
};
|
|
|
|
/// Special instance of NAL class for analyzing PPS nal units
|
|
class PPS : public NAL{
|
|
public:
|
|
PPS() : NAL(){}
|
|
PPS(std::string &InputData) : NAL(InputData){}
|
|
void analyzePPS();
|
|
};
|
|
|
|
class sequenceParameterSet{
|
|
public:
|
|
sequenceParameterSet(const char *_data = NULL, size_t _dataLen = 0);
|
|
void fromDTSCInit(const std::string &dtscInit);
|
|
SPSMeta getCharacteristics() const;
|
|
|
|
private:
|
|
const char *data;
|
|
size_t dataLen;
|
|
};
|
|
|
|
bool isKeyframe(const char *data, uint32_t len);
|
|
|
|
class nalUnit{
|
|
public:
|
|
nalUnit(const char *data, size_t len) : payload(data, len){}
|
|
virtual ~nalUnit(){}
|
|
|
|
uint8_t getType(){return payload[0] & 0x1F;}
|
|
size_t getSize(){return payload.size();}
|
|
virtual void toPrettyString(std::ostream &out){
|
|
out << "Nal unit of type " << (((uint8_t)payload[0]) & 0x1F) << ", " << payload.size()
|
|
<< " bytes long" << std::endl;
|
|
}
|
|
void write(std::ostream &out){
|
|
// always writes in annex_b style
|
|
out.write("\000\000\000\001", 4);
|
|
out.write(payload.data(), payload.size());
|
|
}
|
|
virtual std::string generate(){return "";}
|
|
virtual void setSPSNumber(size_t newNumber){}
|
|
virtual void setPPSNumber(size_t newNumber){}
|
|
|
|
protected:
|
|
std::string payload;
|
|
};
|
|
|
|
class hrd_parameters{
|
|
public:
|
|
hrd_parameters(){}
|
|
hrd_parameters(Utils::bitstream &bs);
|
|
void toPrettyString(std::ostream &out, size_t indent = 0);
|
|
|
|
uint64_t cpbCntMinus1;
|
|
uint8_t bitRateScale;
|
|
uint8_t cpbSizeScale;
|
|
};
|
|
|
|
class vui_parameters{
|
|
public:
|
|
vui_parameters(){}
|
|
vui_parameters(Utils::bitstream &bs);
|
|
void generate(Utils::bitWriter &bw);
|
|
void toPrettyString(std::ostream &out, size_t indent = 0);
|
|
|
|
bool aspectRatioInfoPresentFlag;
|
|
uint8_t aspectRatioIdc;
|
|
uint16_t sarWidth;
|
|
uint16_t sarHeight;
|
|
bool overscanInfoPresentFlag;
|
|
bool overscanAppropriateFlag;
|
|
bool videoSignalTypePresentFlag;
|
|
uint8_t videoFormat;
|
|
bool videoFullRangeFlag;
|
|
bool colourDescriptionPresentFlag;
|
|
uint8_t colourPrimaries;
|
|
uint8_t transferCharacteristics;
|
|
uint8_t matrixCoefficients;
|
|
bool chromaLocInfoPresentFlag;
|
|
uint64_t chromaSampleLocTypeTopField;
|
|
uint64_t chromaSampleLocTypeBottomField;
|
|
bool timingInfoPresentFlag;
|
|
uint32_t numUnitsInTick;
|
|
uint32_t timeScale;
|
|
bool fixedFrameRateFlag;
|
|
bool nalHrdParametersPresentFlag;
|
|
|
|
bool vclHrdParametersPresentFlag;
|
|
|
|
bool lowDelayHrdFlag;
|
|
bool picStructPresentFlag;
|
|
bool bitstreamRestrictionFlag;
|
|
bool motionVectorsOverPicBoundariesFlag;
|
|
uint64_t maxBytesPerPicDenom;
|
|
uint64_t maxBitsPerMbDenom;
|
|
uint64_t log2MaxMvLengthHorizontal;
|
|
uint64_t log2MaxMvLengthVertical;
|
|
uint64_t numReorderFrames;
|
|
uint64_t maxDecFrameBuffering;
|
|
};
|
|
|
|
class spsUnit : public nalUnit{
|
|
public:
|
|
spsUnit(const char *data, size_t len);
|
|
~spsUnit(){
|
|
if (scalingListPresentFlags != NULL){free(scalingListPresentFlags);}
|
|
}
|
|
std::string generate();
|
|
void toPrettyString(std::ostream &out);
|
|
void scalingList(uint64_t *scalingList, size_t sizeOfScalingList,
|
|
bool &useDefaultScalingMatrixFlag, Utils::bitstream &bs);
|
|
void setSPSNumber(size_t newNumber);
|
|
uint8_t profileIdc;
|
|
bool constraintSet0Flag;
|
|
bool constraintSet1Flag;
|
|
bool constraintSet2Flag;
|
|
bool constraintSet3Flag;
|
|
bool constraintSet4Flag;
|
|
bool constraintSet5Flag;
|
|
uint8_t levelIdc;
|
|
uint64_t seqParameterSetId;
|
|
|
|
uint64_t chromaFormatIdc;
|
|
bool separateColourPlaneFlag;
|
|
uint64_t bitDepthLumaMinus8;
|
|
uint64_t bitDepthChromaMinus8;
|
|
bool qpprimeYZeroTransformBypassFlag;
|
|
bool seqScalingMatrixPresentFlag;
|
|
// Here go scaling lists
|
|
uint8_t *scalingListPresentFlags;
|
|
|
|
uint64_t **scalingList4x4;
|
|
bool *useDefaultScalingMatrix4x4Flag;
|
|
uint64_t **scalingList8x8;
|
|
bool *useDefaultScalingMatrix8x8Flag;
|
|
|
|
uint64_t log2MaxFrameNumMinus4;
|
|
uint64_t picOrderCntType;
|
|
uint64_t log2MaxPicOrderCntLsbMinus4;
|
|
|
|
// Here go values for pic_order_cnt_type == 1
|
|
|
|
uint64_t maxNumRefFrames;
|
|
bool gapsInFrameNumValueAllowedFlag;
|
|
uint64_t picWidthInMbsMinus1;
|
|
uint64_t picHeightInMapUnitsMinus1;
|
|
bool frameMbsOnlyFlag;
|
|
bool mbAdaptiveFrameFieldFlag;
|
|
bool direct8x8InferenceFlag;
|
|
bool frameCroppingFlag;
|
|
uint64_t frameCropLeftOffset;
|
|
uint64_t frameCropRightOffset;
|
|
uint64_t frameCropTopOffset;
|
|
uint64_t frameCropBottomOffset;
|
|
bool vuiParametersPresentFlag;
|
|
|
|
vui_parameters vuiParams;
|
|
|
|
// DERIVATIVE VALUES
|
|
uint8_t derived_subWidthC;
|
|
uint8_t derived_subHeightC;
|
|
uint8_t derived_mbWidthC;
|
|
uint8_t derived_mbHeightC;
|
|
uint64_t derived_bitDepth_Y;
|
|
uint64_t derived_qpBdOffset_Y;
|
|
uint64_t derived_bitDepth_C;
|
|
uint64_t derived_qpBdOffset_C;
|
|
uint64_t derived_rawMbBits;
|
|
uint64_t derived_maxFrameNum;
|
|
uint64_t derived_maxPicOrderCntLsb;
|
|
uint64_t derived_picWidthInMbs;
|
|
uint64_t derived_picWidthInSamples_L;
|
|
uint64_t derived_picWidthInSamples_C;
|
|
uint64_t derived_picHeightInMapUnits;
|
|
uint64_t derived_picSizeInMapUnits;
|
|
uint64_t derived_frameHeightInMbs;
|
|
size_t derived_scalingListSize;
|
|
size_t derived_scalingList4x4Amount;
|
|
size_t derived_scalingList8x8Amount;
|
|
};
|
|
class ppsUnit : public nalUnit{
|
|
public:
|
|
ppsUnit(const char *data, size_t len, uint8_t chromaFormatIdc = 0);
|
|
~ppsUnit(){
|
|
if (picScalingMatrixPresentFlags != NULL){free(picScalingMatrixPresentFlags);}
|
|
}
|
|
void scalingList(uint64_t *scalingList, size_t sizeOfScalingList,
|
|
bool &useDefaultScalingMatrixFlag, Utils::bitstream &bs);
|
|
void setPPSNumber(size_t newNumber);
|
|
void setSPSNumber(size_t newNumber);
|
|
void toPrettyString(std::ostream &out);
|
|
std::string generate();
|
|
|
|
uint64_t picParameterSetId;
|
|
uint64_t seqParameterSetId;
|
|
bool entropyCodingModeFlag;
|
|
bool bottomFieldPicOrderInFramePresentFlag;
|
|
uint64_t numSliceGroupsMinus1;
|
|
uint64_t numrefIdx10DefaultActiveMinus1;
|
|
uint64_t numrefIdx11DefaultActiveMinus1;
|
|
bool weightedPredFlag;
|
|
uint8_t weightedBipredIdc;
|
|
int64_t picInitQpMinus26;
|
|
int64_t picInitQsMinus26;
|
|
int64_t chromaQpIndexOffset;
|
|
bool deblockingFilterControlPresentFlag;
|
|
bool constrainedIntraPredFlag;
|
|
bool redundantPicCntPresentFlag;
|
|
bool transform8x8ModeFlag;
|
|
bool picScalingMatrixPresentFlag;
|
|
// Here go scaling lists
|
|
uint8_t *picScalingMatrixPresentFlags;
|
|
|
|
uint64_t **scalingList4x4;
|
|
bool *useDefaultScalingMatrix4x4Flag;
|
|
uint64_t **scalingList8x8;
|
|
bool *useDefaultScalingMatrix8x8Flag;
|
|
|
|
int64_t secondChromaQpIndexOffset;
|
|
|
|
size_t derived_scalingListSize;
|
|
size_t derived_scalingList4x4Amount;
|
|
size_t derived_scalingList8x8Amount;
|
|
|
|
bool status_moreRBSP;
|
|
};
|
|
class codedSliceUnit : public nalUnit{
|
|
public:
|
|
codedSliceUnit(const char *data, size_t len);
|
|
void setPPSNumber(size_t newNumber);
|
|
void toPrettyString(std::ostream &out);
|
|
|
|
uint64_t firstMbInSlice;
|
|
uint64_t sliceType;
|
|
uint64_t picParameterSetId;
|
|
};
|
|
|
|
class seiUnit : public nalUnit{
|
|
public:
|
|
seiUnit(const char *data, size_t len);
|
|
void toPrettyString(std::ostream &out);
|
|
|
|
uint32_t payloadType;
|
|
uint32_t payloadSize;
|
|
uint32_t payloadOffset;
|
|
};
|
|
|
|
nalUnit *nalFactory(FILE *in, bool annexb = true);
|
|
nalUnit *nalFactory(const char *data, size_t len, size_t &offset, bool annexb = true);
|
|
}// namespace h264
|