1155 lines
45 KiB
C++
1155 lines
45 KiB
C++
#ifndef _GNU_SOURCE
|
|
#define _GNU_SOURCE
|
|
#endif
|
|
#include "h264.h"
|
|
#include <cmath>
|
|
#include <cstring>
|
|
#include <iomanip>
|
|
#include "bitfields.h"
|
|
#include "bitstream.h"
|
|
#include "defines.h"
|
|
|
|
namespace h264 {
|
|
|
|
///Helper function to determine if a H264 NAL unit is a keyframe or not
|
|
bool isKeyframe(const char * data, uint32_t len){
|
|
uint8_t nalType = (data[0] & 0x1F);
|
|
if (nalType == 0x05){return true;}
|
|
if (nalType != 0x01){return false;}
|
|
Utils::bitstream bs;
|
|
for (size_t i = 1; i < 10 && i < len; ++i) {
|
|
if (i + 2 < len && (memcmp(data + i, "\000\000\003", 3) == 0)) { //Emulation prevention bytes
|
|
bs.append(data + i, 2);
|
|
i += 2;
|
|
} else {
|
|
bs.append(data + i, 1);
|
|
}
|
|
}
|
|
bs.getExpGolomb();//Discard first_mb_in_slice
|
|
uint64_t sliceType = bs.getUExpGolomb();
|
|
//Slice types:
|
|
// 0: P - Predictive slice (at most 1 reference)
|
|
// 1: B - Bi-predictive slice (at most 2 references)
|
|
// 2: I - Intra slice (no external references)
|
|
// 3: SP - Switching predictive slice (at most 1 reference)
|
|
// 4: SI - Switching intra slice (no external references)
|
|
// 5-9: 0-4, but all in picture of same type
|
|
if (sliceType == 2 || sliceType == 4 || sliceType == 7 || sliceType == 9){
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::deque<nalu::nalData> analysePackets(const char * data, unsigned long len){
|
|
std::deque<nalu::nalData> res;
|
|
|
|
int offset = 0;
|
|
//Make sure entire packet is within len
|
|
while (offset+5 < len && Bit::btohl(data + offset)+offset+4 <= len){
|
|
nalu::nalData entry;
|
|
entry.nalSize = Bit::btohl(data + offset);
|
|
entry.nalType = (data + offset)[4] & 0x1F;
|
|
res.push_back(entry);
|
|
offset += entry.nalSize + 4;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
sequenceParameterSet::sequenceParameterSet(const char * _data, size_t _dataLen) : data(_data), dataLen(_dataLen) {}
|
|
|
|
//DTSC Initdata is the payload for an avcc box. init[8+] is data, init[6-7] is a network-encoded length
|
|
void sequenceParameterSet::fromDTSCInit(const std::string & dtscInit){
|
|
data = dtscInit.data() + 8;
|
|
dataLen = Bit::btohs(dtscInit.data() + 6);
|
|
}
|
|
|
|
void skipScalingList(Utils::bitstream & bs, size_t listSize){
|
|
size_t lastScale = 8;
|
|
size_t nextScale = 8;
|
|
for (size_t i = 0; i < listSize; i++){
|
|
if (nextScale){
|
|
uint64_t deltaScale = bs.getExpGolomb();
|
|
nextScale = (lastScale + deltaScale + 256) % 256;
|
|
}
|
|
lastScale = (nextScale ? nextScale : lastScale);
|
|
}
|
|
}
|
|
|
|
SPSMeta sequenceParameterSet::getCharacteristics() const {
|
|
SPSMeta result;
|
|
result.sep_col_plane = false;
|
|
|
|
//For calculating width
|
|
unsigned int widthInMbs = 0;
|
|
unsigned int cropHorizontal = 0;
|
|
|
|
//For calculating height
|
|
unsigned int heightInMapUnits = 0;
|
|
unsigned int cropVertical = 0;
|
|
|
|
//Fill the bitstream
|
|
Utils::bitstream bs;
|
|
for (unsigned int i = 1; i < dataLen; i++) {
|
|
if (i + 2 < dataLen && (memcmp(data + i, "\000\000\003", 3) == 0)){//Emulation prevention bytes
|
|
//Yes, we increase i here
|
|
bs.append(data + i, 2);
|
|
i += 2;
|
|
} else {
|
|
//No we don't increase i here
|
|
bs.append(data + i, 1);
|
|
}
|
|
}
|
|
|
|
char profileIdc = bs.get(8);
|
|
result.profile = profileIdc;
|
|
//Start skipping unused data
|
|
bs.skip(8);
|
|
result.level = bs.get(8);
|
|
bs.getUExpGolomb();
|
|
if (profileIdc == 100 || profileIdc == 110 || profileIdc == 122 || profileIdc == 244 || profileIdc == 44 || profileIdc == 83 || profileIdc == 86 || profileIdc == 118 || profileIdc == 128) {
|
|
//chroma format idc
|
|
char chromaFormatIdc = bs.getUExpGolomb();
|
|
if (chromaFormatIdc == 3) {
|
|
result.sep_col_plane = (bs.get(1) == 1);
|
|
}
|
|
bs.getUExpGolomb();//luma
|
|
bs.getUExpGolomb();//chroma
|
|
bs.skip(1);//transform bypass
|
|
if (bs.get(1)) {//Scaling matrix is present
|
|
char listSize = (chromaFormatIdc == 3 ? 12 : 8);
|
|
for (size_t i = 0; i < listSize; i++){
|
|
bool thisListPresent = bs.get(1);
|
|
if (thisListPresent){
|
|
if (i < 6){
|
|
skipScalingList(bs, 16);
|
|
}else{
|
|
skipScalingList(bs, 64);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
result.log2_max_frame_num = bs.getUExpGolomb() + 4;
|
|
result.cnt_type = bs.getUExpGolomb();
|
|
if (!result.cnt_type) {
|
|
result.log2_max_order_cnt = bs.getUExpGolomb() + 4;
|
|
} else if (result.cnt_type == 1) {
|
|
DEBUG_MSG(DLVL_DEVEL, "This part of the implementation is incomplete(2), to be continued. If this message is shown, contact developers immediately.");
|
|
}
|
|
result.max_ref_frames = bs.getUExpGolomb();//max_num_ref_frames
|
|
result.gaps = (bs.get(1) == 1);//gaps in frame num allowed
|
|
//Stop skipping data and start doing useful stuff
|
|
|
|
|
|
widthInMbs = bs.getUExpGolomb() + 1;
|
|
heightInMapUnits = bs.getUExpGolomb() + 1;
|
|
|
|
result.mbs_only = (bs.get(1) == 1);//Gets used in height calculation
|
|
if (!result.mbs_only) {
|
|
bs.skip(1);
|
|
}
|
|
bs.skip(1);
|
|
//cropping flag
|
|
if (bs.get(1)) {
|
|
cropHorizontal = bs.getUExpGolomb();//leftOffset
|
|
cropHorizontal += bs.getUExpGolomb();//rightOffset
|
|
cropVertical = bs.getUExpGolomb();//topOffset
|
|
cropVertical += bs.getUExpGolomb();//bottomOffset
|
|
}
|
|
|
|
//vuiParameters
|
|
result.fps = 0;
|
|
if (bs.get(1)) {
|
|
//Skipping all the paramters we dont use
|
|
if (bs.get(1)) {
|
|
if (bs.get(8) == 255) {
|
|
bs.skip(32);
|
|
}
|
|
}
|
|
if (bs.get(1)) {
|
|
bs.skip(1);
|
|
}
|
|
if (bs.get(1)) {
|
|
bs.skip(4);
|
|
if (bs.get(1)) {
|
|
bs.skip(24);
|
|
}
|
|
}
|
|
if (bs.get(1)) {
|
|
bs.getUExpGolomb();
|
|
bs.getUExpGolomb();
|
|
}
|
|
|
|
//Decode timing info
|
|
if (bs.get(1)) {
|
|
unsigned int unitsInTick = bs.get(32);
|
|
unsigned int timeScale = bs.get(32);
|
|
result.fps = (double)timeScale / (2 * unitsInTick);
|
|
bs.skip(1);
|
|
}
|
|
}
|
|
|
|
result.width = (widthInMbs * 16) - (cropHorizontal * 2);
|
|
result.height = ((result.mbs_only ? 1 : 2) * heightInMapUnits * 16) - (cropVertical * 2);
|
|
return result;
|
|
}
|
|
|
|
void spsUnit::scalingList(uint64_t * scalingList, size_t sizeOfScalingList, bool & useDefaultScalingMatrixFlag, Utils::bitstream & bs){
|
|
int lastScale = 8;
|
|
int nextScale = 8;
|
|
for (int i = 0; i < sizeOfScalingList; i++){
|
|
if (nextScale){
|
|
int64_t deltaScale = bs.getExpGolomb();
|
|
nextScale = (lastScale + deltaScale + 256) % 256;
|
|
useDefaultScalingMatrixFlag = (i == 0 && nextScale == 0);
|
|
}
|
|
scalingList[i] = (nextScale == 0 ? lastScale : nextScale);
|
|
lastScale = scalingList[i];
|
|
}
|
|
}
|
|
|
|
spsUnit::spsUnit(const char * data, size_t len) : nalUnit(data, len) {
|
|
scalingListPresentFlags = NULL;
|
|
scalingList4x4 = NULL;
|
|
useDefaultScalingMatrix4x4Flag = NULL;
|
|
scalingList8x8 = NULL;
|
|
useDefaultScalingMatrix8x8Flag = NULL;
|
|
derived_subWidthC = 0;
|
|
derived_subHeightC = 0;
|
|
derived_mbWidthC = 0;
|
|
derived_mbHeightC = 0;
|
|
derived_scalingList4x4Amount = 0;
|
|
derived_scalingList8x8Amount = 0;
|
|
|
|
//Fill the bitstream
|
|
Utils::bitstream bs;
|
|
for (size_t i = 1; i < len; i++) {
|
|
if (i + 2 < len && (memcmp(data + i, "\000\000\003", 3) == 0)){//Emulation prevention bytes
|
|
//Yes, we increase i here
|
|
bs.append(data + i, 2);
|
|
i += 2;
|
|
} else {
|
|
//No we don't increase i here
|
|
bs.append(data + i, 1);
|
|
}
|
|
}
|
|
profileIdc = bs.get(8);
|
|
constraintSet0Flag = bs.get(1);
|
|
constraintSet1Flag = bs.get(1);
|
|
constraintSet2Flag = bs.get(1);
|
|
constraintSet3Flag = bs.get(1);
|
|
constraintSet4Flag = bs.get(1);
|
|
constraintSet5Flag = bs.get(1);
|
|
bs.skip(2);
|
|
levelIdc = bs.get(8);
|
|
seqParameterSetId = bs.getUExpGolomb();
|
|
switch (profileIdc){
|
|
case 100:
|
|
case 110:
|
|
case 122:
|
|
case 244:
|
|
case 44:
|
|
case 83:
|
|
case 86:
|
|
case 118:
|
|
case 128:
|
|
chromaFormatIdc = bs.getUExpGolomb();
|
|
if (chromaFormatIdc == 3){
|
|
separateColourPlaneFlag = bs.get(1);
|
|
}
|
|
if (chromaFormatIdc == 1){
|
|
derived_subWidthC = 2;
|
|
derived_subHeightC = 2;
|
|
}
|
|
if (chromaFormatIdc == 2){
|
|
derived_subWidthC = 2;
|
|
derived_subHeightC = 1;
|
|
}
|
|
if (chromaFormatIdc == 3 && !separateColourPlaneFlag){
|
|
derived_subWidthC = 1;
|
|
derived_subHeightC = 1;
|
|
}
|
|
if (derived_subWidthC){
|
|
derived_mbWidthC = 16 / derived_subWidthC;
|
|
derived_mbHeightC = 16 / derived_subHeightC;
|
|
}
|
|
|
|
bitDepthLumaMinus8 = bs.getUExpGolomb();
|
|
derived_bitDepth_Y = 8 + bitDepthLumaMinus8;
|
|
derived_qpBdOffset_Y = 6 * bitDepthLumaMinus8;
|
|
bitDepthChromaMinus8 = bs.getUExpGolomb();
|
|
derived_bitDepth_C = 8 + bitDepthChromaMinus8;
|
|
derived_qpBdOffset_C = 6 * bitDepthChromaMinus8;
|
|
derived_rawMbBits = 256 * derived_bitDepth_Y + 2 * derived_mbWidthC * derived_mbHeightC * derived_bitDepth_C;
|
|
qpprimeYZeroTransformBypassFlag = bs.get(1);
|
|
seqScalingMatrixPresentFlag = bs.get(1);
|
|
if (seqScalingMatrixPresentFlag){
|
|
derived_scalingListSize = (chromaFormatIdc == 3 ? 12 : 8);
|
|
scalingListPresentFlags = (uint8_t*)malloc(derived_scalingListSize * sizeof(uint8_t));
|
|
|
|
derived_scalingList4x4Amount = 6;
|
|
scalingList4x4 = (uint64_t **)malloc(derived_scalingList4x4Amount * sizeof(uint64_t*));
|
|
useDefaultScalingMatrix4x4Flag = (bool*)malloc(derived_scalingList4x4Amount * sizeof(bool));
|
|
for (int i = 0; i < derived_scalingList4x4Amount; i++){
|
|
scalingList4x4[i] = NULL;
|
|
useDefaultScalingMatrix4x4Flag[i] = false;
|
|
}
|
|
|
|
|
|
derived_scalingList8x8Amount = derived_scalingListSize - 6;
|
|
scalingList8x8 = (uint64_t **)malloc(derived_scalingList8x8Amount * sizeof(uint64_t*));
|
|
useDefaultScalingMatrix8x8Flag = (bool*)malloc(derived_scalingList8x8Amount * sizeof(bool));
|
|
for (int i = 0; i < derived_scalingList8x8Amount; i++){
|
|
scalingList8x8[i] = NULL;
|
|
useDefaultScalingMatrix8x8Flag[i] = false;
|
|
}
|
|
|
|
for (size_t i = 0; i < derived_scalingListSize; i++){
|
|
scalingListPresentFlags[i] = bs.get(1);
|
|
if (scalingListPresentFlags[i]){
|
|
if (i < 6){
|
|
scalingList4x4[i] = (uint64_t*)malloc(16 * sizeof(uint64_t));
|
|
scalingList(scalingList4x4[i], 16, useDefaultScalingMatrix4x4Flag[i], bs);
|
|
}else{
|
|
scalingList8x8[i-6] = (uint64_t*)malloc(64 * sizeof(uint64_t));
|
|
scalingList(scalingList8x8[i-6], 64, useDefaultScalingMatrix8x8Flag[i-6], bs);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
log2MaxFrameNumMinus4 = bs.getUExpGolomb();
|
|
derived_maxFrameNum = pow(2, log2MaxFrameNumMinus4 + 4);
|
|
picOrderCntType = bs.getUExpGolomb();
|
|
if (!picOrderCntType){
|
|
log2MaxPicOrderCntLsbMinus4 = bs.getUExpGolomb();
|
|
derived_maxPicOrderCntLsb = pow(2, log2MaxPicOrderCntLsbMinus4 + 4);
|
|
}else if (picOrderCntType == 1){
|
|
WARN_MSG("picOrderCntType == 1 encountered, which has not-yet implemented parameters");
|
|
return;
|
|
}
|
|
maxNumRefFrames = bs.getUExpGolomb();
|
|
gapsInFrameNumValueAllowedFlag = bs.get(1);
|
|
picWidthInMbsMinus1 = bs.getUExpGolomb();
|
|
derived_picWidthInMbs = picWidthInMbsMinus1 + 1;
|
|
derived_picWidthInSamples_L = derived_picWidthInMbs * 16;
|
|
derived_picWidthInSamples_C = derived_picWidthInMbs * derived_mbWidthC;
|
|
picHeightInMapUnitsMinus1 = bs.getUExpGolomb();
|
|
derived_picHeightInMapUnits = picHeightInMapUnitsMinus1 + 1;
|
|
derived_picSizeInMapUnits = derived_picWidthInMbs * derived_picHeightInMapUnits;
|
|
frameMbsOnlyFlag = bs.get(1);
|
|
derived_frameHeightInMbs = (2 - (frameMbsOnlyFlag ? 1 : 0)) * derived_picHeightInMapUnits;
|
|
if (!frameMbsOnlyFlag){
|
|
mbAdaptiveFrameFieldFlag = bs.get(1);
|
|
}
|
|
direct8x8InferenceFlag = bs.get(1);
|
|
frameCroppingFlag = bs.get(1);
|
|
if (frameCroppingFlag){
|
|
frameCropLeftOffset = bs.getUExpGolomb();
|
|
frameCropRightOffset = bs.getUExpGolomb();
|
|
frameCropTopOffset = bs.getUExpGolomb();
|
|
frameCropBottomOffset = bs.getUExpGolomb();
|
|
}
|
|
vuiParametersPresentFlag = bs.get(1);
|
|
if (vuiParametersPresentFlag){
|
|
vuiParams = vui_parameters(bs);
|
|
}
|
|
}
|
|
|
|
void spsUnit::setSPSNumber(size_t newNumber){
|
|
//for now, can only convert from 0 to 16
|
|
if (seqParameterSetId != 0){
|
|
return;
|
|
}
|
|
seqParameterSetId = 16;
|
|
payload.insert(4, 1, 0x08);
|
|
}
|
|
|
|
void spsUnit::toPrettyString(std::ostream & out) {
|
|
out << "Nal unit of type " << (((uint8_t)payload[0]) & 0x1F) << " [Sequence Parameter Set] , " << payload.size() << " bytes long" << std::endl;
|
|
out << " profile_idc: 0x" << std::setw(2) << std::setfill('0') << std::hex << (int)profileIdc << std::dec << " (" << (int)profileIdc << ")" << std::endl;
|
|
out << " contraints: " << (constraintSet0Flag ? "0 " : "") << (constraintSet1Flag ? "1 " : "") << (constraintSet2Flag ? "2 " : "") << (constraintSet3Flag ? "3 " : "") << (constraintSet4Flag ? "4 " : "") << (constraintSet5Flag ? "5" : "") << std::endl;
|
|
out << " level_idc: 0x" << std::setw(2) << std::setfill('0') << std::hex << (int)levelIdc << std::dec << " (" << (int)profileIdc << ")" << std::endl;
|
|
out << " seq_parameter_set_id: " << seqParameterSetId << (seqParameterSetId >= 32 ? " INVALID" : "") << std::endl;
|
|
switch (profileIdc){
|
|
case 100:
|
|
case 110:
|
|
case 122:
|
|
case 244:
|
|
case 44:
|
|
case 83:
|
|
case 86:
|
|
case 118:
|
|
case 128:
|
|
out << " chroma_format_idc: " << chromaFormatIdc << (chromaFormatIdc >= 4 ? " INVALID" : "") << std::endl;
|
|
if (chromaFormatIdc == 3){
|
|
out << " separate_colour_plane_flag: " << (separateColourPlaneFlag ? 1 : 0) << std::endl;
|
|
}
|
|
out << " bit_depth_luma_minus_8: " << bitDepthLumaMinus8 << (bitDepthLumaMinus8 >= 7 ? " INVALID": "") << std::endl;
|
|
out << " -> BitDepth_Y: " << derived_bitDepth_Y << std::endl;
|
|
out << " -> QpBdOffset_Y: " << derived_qpBdOffset_Y << std::endl;
|
|
out << " bit_depth_chroma_minus_8: " << bitDepthChromaMinus8 << (bitDepthChromaMinus8 >= 7 ? " INVALID": "") << std::endl;
|
|
out << " -> BitDepth_C: " << derived_bitDepth_C << std::endl;
|
|
out << " -> QpBdOffset_C: " << derived_qpBdOffset_C << std::endl;
|
|
out << " -> RawMbBits: " << derived_rawMbBits << std::endl;
|
|
out << " qpprime_y_zero-transform_bypass_flag: " << (qpprimeYZeroTransformBypassFlag ? 1 : 0) << std::endl;
|
|
out << " seq_scaling_matrix_present_flag: " << (seqScalingMatrixPresentFlag ? 1 : 0) << std::endl;
|
|
if (seqScalingMatrixPresentFlag){
|
|
for (int i = 0; i < derived_scalingListSize; i++){
|
|
out << " seq_scaling_list_present_flag[" << i << "]: " << (scalingListPresentFlags[i] ? 1 : 0) << std::endl;
|
|
if (scalingListPresentFlags[i]){
|
|
if (i < 6){
|
|
for (int j = 0; j < 16; j++){
|
|
out << " scalingMatrix4x4[" << i << "][" << j << "]: " << scalingList4x4[i][j] << std::endl;
|
|
}
|
|
out << " useDefaultScalingMatrix4x4Flag[" << i << "]: " << (useDefaultScalingMatrix4x4Flag[i] ? 1 : 0) << std::endl;
|
|
}else{
|
|
for (int j = 0; j < 64; j++){
|
|
out << " scalingMatrix8x8[" << i-6 << "][" << j << "]: " << scalingList8x8[i-6][j] << std::endl;
|
|
}
|
|
out << " useDefaultScalingMatrix8x8Flag[" << i - 6 << "]: " << (useDefaultScalingMatrix8x8Flag[i - 6] ? 1 : 0) << std::endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
out << " log2_max_frame_num_minus4: " << log2MaxFrameNumMinus4 << (log2MaxFrameNumMinus4 >= 13 ? " INVALID" : "") << std::endl;
|
|
out << " -> MaxFrameNum: " << derived_maxFrameNum << std::endl;
|
|
out << " pic_order_cnt_type: " << picOrderCntType << (picOrderCntType >= 3 ? " INVALID" : "") << std::endl;
|
|
if (!picOrderCntType){
|
|
out << " log2_max_pic_order_cnt_lsb_minus4: " << log2MaxPicOrderCntLsbMinus4 << (log2MaxPicOrderCntLsbMinus4 >= 13 ? " INVALID" : "") << std::endl;
|
|
out << " -> MaxPicOrderCntLsb: " << derived_maxPicOrderCntLsb << std::endl;
|
|
}
|
|
out << " max_num_ref_frames: " << maxNumRefFrames << std::endl;
|
|
out << " gaps_in_frame_num_value_allowed_flag: " << (gapsInFrameNumValueAllowedFlag ? 1 : 0) << std::endl;
|
|
out << " pic_width_in_mbs_minus_1: " << picWidthInMbsMinus1 << std::endl;
|
|
out << " -> PicWidthInMbs: " << derived_picWidthInMbs << std::endl;
|
|
out << " -> PicWidthInSamples_L: " << derived_picWidthInSamples_L << std::endl;
|
|
out << " -> PicWidthInSamples_C: " << derived_picWidthInSamples_C << std::endl;
|
|
out << " pic_height_in_map_units_minus_1: " << picHeightInMapUnitsMinus1 << std::endl;
|
|
out << " -> PicHeightInMapUnits: " << derived_picHeightInMapUnits << std::endl;
|
|
out << " -> PicSizeInMapUnits: " << derived_picSizeInMapUnits << std::endl;
|
|
out << " frame_mbs_only_flag: " << frameMbsOnlyFlag << std::endl;
|
|
out << " -> FrameHeightInMbs: " << derived_frameHeightInMbs << std::endl;
|
|
if (!frameMbsOnlyFlag){
|
|
out << " mb_adaptive_frame_field_flag: " << mbAdaptiveFrameFieldFlag << std::endl;
|
|
}
|
|
out << " direct_8x8_inference_flag: " << direct8x8InferenceFlag << std::endl;
|
|
out << " frame_cropping_flag: " << frameCroppingFlag << std::endl;
|
|
if (frameCroppingFlag){
|
|
out << " frame_crop_left_offset: " << frameCropLeftOffset << std::endl;
|
|
out << " frame_crop_right_offset: " << frameCropRightOffset << std::endl;
|
|
out << " frame_crop_top_offset: " << frameCropTopOffset << std::endl;
|
|
out << " frame_crop_bottom_offset: " << frameCropBottomOffset << std::endl;
|
|
}
|
|
out << " vui_parameter_present_flag: " << vuiParametersPresentFlag << std::endl;
|
|
if (vuiParametersPresentFlag){
|
|
vuiParams.toPrettyString(out, 2);
|
|
}
|
|
}
|
|
|
|
std::string spsUnit::generate() {
|
|
Utils::bitWriter bw;
|
|
bw.append(0x07, 8);
|
|
bw.append(profileIdc, 8);
|
|
bw.append(constraintSet0Flag ? 1 : 0, 1);
|
|
bw.append(constraintSet1Flag ? 1 : 0, 1);
|
|
bw.append(constraintSet2Flag ? 1 : 0, 1);
|
|
bw.append(constraintSet3Flag ? 1 : 0, 1);
|
|
bw.append(constraintSet4Flag ? 1 : 0, 1);
|
|
bw.append(constraintSet5Flag ? 1 : 0, 1);
|
|
bw.append(0x00, 2);
|
|
bw.append(levelIdc, 8);
|
|
bw.appendUExpGolomb(seqParameterSetId);
|
|
switch (profileIdc){
|
|
case 100:
|
|
case 110:
|
|
case 122:
|
|
case 244:
|
|
case 44:
|
|
case 83:
|
|
case 86:
|
|
case 118:
|
|
case 128:
|
|
bw.appendUExpGolomb(chromaFormatIdc);
|
|
if (chromaFormatIdc == 3){
|
|
bw.append(separateColourPlaneFlag ? 1 : 0, 1);
|
|
}
|
|
bw.appendUExpGolomb(bitDepthLumaMinus8);
|
|
bw.appendUExpGolomb(bitDepthChromaMinus8);
|
|
bw.append(qpprimeYZeroTransformBypassFlag, 1);
|
|
bw.append(seqScalingMatrixPresentFlag, 1);
|
|
if (seqScalingMatrixPresentFlag){
|
|
for (int i = 0; i < derived_scalingListSize; i++){
|
|
bw.append(0, 1);
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
bw.appendUExpGolomb(log2MaxFrameNumMinus4);
|
|
bw.appendUExpGolomb(picOrderCntType);
|
|
if (picOrderCntType == 0){
|
|
bw.appendUExpGolomb(log2MaxPicOrderCntLsbMinus4);
|
|
}
|
|
bw.appendUExpGolomb(maxNumRefFrames);
|
|
bw.append(gapsInFrameNumValueAllowedFlag ? 1 : 0, 1);
|
|
bw.appendUExpGolomb(picWidthInMbsMinus1);
|
|
bw.appendUExpGolomb(picHeightInMapUnitsMinus1);
|
|
bw.append(frameMbsOnlyFlag ? 1 : 0, 1);
|
|
if (!frameMbsOnlyFlag){
|
|
bw.append(mbAdaptiveFrameFieldFlag ? 1 : 0, 1);
|
|
}
|
|
bw.append(direct8x8InferenceFlag ? 1 : 0, 1);
|
|
bw.append(frameCroppingFlag ? 1 : 0, 1);
|
|
if (frameCroppingFlag){
|
|
bw.appendUExpGolomb(frameCropLeftOffset);
|
|
bw.appendUExpGolomb(frameCropRightOffset);
|
|
bw.appendUExpGolomb(frameCropTopOffset);
|
|
bw.appendUExpGolomb(frameCropBottomOffset);
|
|
}
|
|
bw.append(vuiParametersPresentFlag ? 1 : 0, 1);
|
|
if (vuiParametersPresentFlag){
|
|
vuiParams.generate(bw);
|
|
}
|
|
bw.append(1,1);
|
|
std::string tmp = bw.str();
|
|
std::string res;
|
|
for (int i = 0; i < tmp.size(); i++){
|
|
if (res.size() > 2 && res[res.size() - 1] == 0x00 && res[res.size() - 2] == 0x00){
|
|
if (tmp[i] == 0x00 || tmp[i] == 0x01 || tmp[i] == 0x02 || tmp[i] == 0x03){
|
|
res += (char)0x03;
|
|
}
|
|
}
|
|
res += tmp[i];
|
|
}
|
|
return res;
|
|
}
|
|
|
|
bool more_rbsp_data(Utils::bitstream & bs){
|
|
if (bs.size() < 8){
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void ppsUnit::scalingList(uint64_t * scalingList, size_t sizeOfScalingList, bool & useDefaultScalingMatrixFlag, Utils::bitstream & bs){
|
|
int lastScale = 8;
|
|
int nextScale = 8;
|
|
for (int i = 0; i < sizeOfScalingList; i++){
|
|
if (nextScale){
|
|
int64_t deltaScale = bs.getExpGolomb();
|
|
nextScale = (lastScale + deltaScale + 256) % 256;
|
|
useDefaultScalingMatrixFlag = (i == 0 && nextScale == 0);
|
|
}
|
|
scalingList[i] = (nextScale == 0 ? lastScale : nextScale);
|
|
lastScale = scalingList[i];
|
|
}
|
|
}
|
|
ppsUnit::ppsUnit(const char * data, size_t len, uint8_t chromaFormatIdc) : nalUnit(data, len) {
|
|
picScalingMatrixPresentFlags = NULL;
|
|
Utils::bitstream bs;
|
|
for (size_t i = 1; i < len; i++) {
|
|
if (i + 2 < len && (memcmp(data + i, "\000\000\003", 3) == 0)){//Emulation prevention bytes
|
|
//Yes, we increase i here
|
|
bs.append(data + i, 2);
|
|
i += 2;
|
|
} else {
|
|
//No we don't increase i here
|
|
bs.append(data + i, 1);
|
|
}
|
|
}
|
|
picParameterSetId = bs.getUExpGolomb();
|
|
seqParameterSetId = bs.getUExpGolomb();
|
|
entropyCodingModeFlag = bs.get(1);
|
|
bottomFieldPicOrderInFramePresentFlag = bs.get(1);
|
|
numSliceGroupsMinus1 = bs.getUExpGolomb();
|
|
if (numSliceGroupsMinus1 > 0){
|
|
WARN_MSG("num_slice_groups_minus1 > 0, unimplemented structure");
|
|
return;
|
|
}
|
|
numrefIdx10DefaultActiveMinus1 = bs.getUExpGolomb();
|
|
numrefIdx11DefaultActiveMinus1 = bs.getUExpGolomb();
|
|
weightedPredFlag = bs.get(1);
|
|
weightedBipredIdc = bs.get(2);
|
|
picInitQpMinus26 = bs.getExpGolomb();
|
|
picInitQsMinus26 = bs.getExpGolomb();
|
|
chromaQpIndexOffset = bs.getExpGolomb();
|
|
deblockingFilterControlPresentFlag = bs.get(1);
|
|
constrainedIntraPredFlag = bs.get(1);
|
|
redundantPicCntPresentFlag = bs.get(1);
|
|
if (!more_rbsp_data(bs)){
|
|
status_moreRBSP = false;
|
|
return;
|
|
}
|
|
status_moreRBSP = true;
|
|
transform8x8ModeFlag = bs.get(1);
|
|
picScalingMatrixPresentFlag = bs.get(1);
|
|
if (picScalingMatrixPresentFlag){
|
|
derived_scalingListSize = 6 + (chromaFormatIdc ? ((chromaFormatIdc == 3 ? 6 : 2) * transform8x8ModeFlag) : 0);
|
|
picScalingMatrixPresentFlags = (uint8_t*)malloc(derived_scalingListSize * sizeof(uint8_t));
|
|
|
|
|
|
derived_scalingList4x4Amount = 6;
|
|
scalingList4x4 = (uint64_t **)malloc(derived_scalingList4x4Amount * sizeof(uint64_t*));
|
|
useDefaultScalingMatrix4x4Flag = (bool*)malloc(derived_scalingList4x4Amount * sizeof(bool));
|
|
for (int i = 0; i < derived_scalingList4x4Amount; i++){
|
|
scalingList4x4[i] = NULL;
|
|
useDefaultScalingMatrix4x4Flag[i] = false;
|
|
}
|
|
|
|
|
|
derived_scalingList8x8Amount = derived_scalingListSize - 6;
|
|
scalingList8x8 = (uint64_t **)malloc(derived_scalingList8x8Amount * sizeof(uint64_t*));
|
|
useDefaultScalingMatrix8x8Flag = (bool*)malloc(derived_scalingList8x8Amount * sizeof(bool));
|
|
for (int i = 0; i < derived_scalingList8x8Amount; i++){
|
|
scalingList8x8[i] = NULL;
|
|
useDefaultScalingMatrix8x8Flag[i] = false;
|
|
}
|
|
|
|
for (size_t i = 0; i < derived_scalingListSize; i++){
|
|
picScalingMatrixPresentFlags[i] = bs.get(1);
|
|
if (picScalingMatrixPresentFlags[i]){
|
|
if (i < 6){
|
|
scalingList4x4[i] = (uint64_t*)malloc(16 * sizeof(uint64_t));
|
|
scalingList(scalingList4x4[i], 16, useDefaultScalingMatrix4x4Flag[i], bs);
|
|
}else{
|
|
scalingList8x8[i-6] = (uint64_t*)malloc(64 * sizeof(uint64_t));
|
|
scalingList(scalingList8x8[i-6], 64, useDefaultScalingMatrix8x8Flag[i-6], bs);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
secondChromaQpIndexOffset = bs.getExpGolomb();
|
|
}
|
|
|
|
void ppsUnit::setPPSNumber(size_t newNumber){
|
|
//for now, can only convert from 0 to 16
|
|
picParameterSetId = newNumber;
|
|
}
|
|
|
|
void ppsUnit::setSPSNumber(size_t newNumber){
|
|
//for now, can only convert from 0 to 16
|
|
if (seqParameterSetId != 0 || picParameterSetId != 16){
|
|
return;
|
|
}
|
|
seqParameterSetId = 16;
|
|
payload.insert(2, 1, 0x84);
|
|
payload[3] &= 0x7F;
|
|
}
|
|
|
|
void ppsUnit::toPrettyString(std::ostream & out) {
|
|
out << "Nal unit of type " << (((uint8_t)payload[0]) & 0x1F) << " [Picture Parameter Set] , " << payload.size() << " bytes long" << std::endl;
|
|
out << " pic_parameter_set_id: " << picParameterSetId << (picParameterSetId >= 256 ? " INVALID" : "") << std::endl;
|
|
out << " seq_parameter_set_id: " << seqParameterSetId << (seqParameterSetId >= 32 ? " INVALID" : "") << std::endl;
|
|
out << " entropy_coding_mode_flag: " << (entropyCodingModeFlag ? 1 : 0) << std::endl;
|
|
out << " bottom_field_pic_order_in_frame_present_flag: " << (bottomFieldPicOrderInFramePresentFlag ? 1 : 0) << std::endl;
|
|
out << " num_slice_groups_minus1: " << numSliceGroupsMinus1 << std::endl;
|
|
if (numSliceGroupsMinus1 > 0){
|
|
return;
|
|
}
|
|
out << " num_ref_idx_10_default_active_minus1: " << numrefIdx10DefaultActiveMinus1 << std::endl;
|
|
out << " num_ref_idx_11_default_active_minus1: " << numrefIdx11DefaultActiveMinus1 << std::endl;
|
|
out << " weighted_pred_flag: " << (weightedPredFlag ? 1 : 0) << std::endl;
|
|
out << " weighted_bipred_idc: " << (uint32_t)weightedBipredIdc << std::endl;
|
|
out << " pic_init_qp_minus26: " << picInitQpMinus26 << std::endl;
|
|
out << " pic_init_qs_minus26: " << picInitQsMinus26 << std::endl;
|
|
out << " chroma_qp_index_offset: " << chromaQpIndexOffset << std::endl;
|
|
out << " deblocking_filter_control_present_flag: " << deblockingFilterControlPresentFlag << std::endl;
|
|
out << " constrained_intra_pred_flag: " << constrainedIntraPredFlag << std::endl;
|
|
out << " redundant_pic_cnt_present_flag: " << redundantPicCntPresentFlag << std::endl;
|
|
if (status_moreRBSP){
|
|
out << " transform_8x8_mode_flag: " << transform8x8ModeFlag << std::endl;
|
|
out << " pic_scaling_matrix_present_flag: " << picScalingMatrixPresentFlag << std::endl;
|
|
if (picScalingMatrixPresentFlag){
|
|
for (int i = 0; i < derived_scalingListSize; i++){
|
|
out << " pic_scaling_matrix_present_flag[" << i << "]: " << (picScalingMatrixPresentFlags[i] ? 1 : 0) << std::endl;
|
|
if (picScalingMatrixPresentFlags[i]){
|
|
if (i < 6){
|
|
for (int j = 0; j < 16; j++){
|
|
out << " scalingMatrix4x4[" << i << "][" << j << "]: " << scalingList4x4[i][j] << std::endl;
|
|
}
|
|
out << " useDefaultScalingMatrix4x4Flag[" << i << "]: " << (useDefaultScalingMatrix4x4Flag[i] ? 1 : 0) << std::endl;
|
|
}else{
|
|
for (int j = 0; j < 64; j++){
|
|
out << " scalingMatrix8x8[" << i-6 << "][" << j << "]: " << scalingList8x8[i-6][j] << std::endl;
|
|
}
|
|
out << " useDefaultScalingMatrix8x8Flag[" << i - 6 << "]: " << (useDefaultScalingMatrix8x8Flag[i - 6] ? 1 : 0) << std::endl;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
out << " second_chroma_qp_index_offset: " << secondChromaQpIndexOffset << std::endl;
|
|
}
|
|
}
|
|
|
|
std::string ppsUnit::generate() {
|
|
Utils::bitWriter bw;
|
|
bw.append(0x08, 8);
|
|
bw.appendUExpGolomb(picParameterSetId);
|
|
bw.appendUExpGolomb(seqParameterSetId);
|
|
bw.append(entropyCodingModeFlag ? 1 : 0, 1);
|
|
bw.append(bottomFieldPicOrderInFramePresentFlag ? 1 : 0, 1);
|
|
if(numSliceGroupsMinus1 > 0){
|
|
INFO_MSG("Forcing to numSliceGroupsMinus1 == 0");
|
|
}
|
|
bw.appendUExpGolomb(0);//numSliceGroupsMinus1
|
|
bw.appendUExpGolomb(numrefIdx10DefaultActiveMinus1);
|
|
bw.appendUExpGolomb(numrefIdx11DefaultActiveMinus1);
|
|
bw.append(weightedPredFlag ? 1 : 0, 1);
|
|
bw.append(weightedBipredIdc, 2);
|
|
bw.appendExpGolomb(picInitQpMinus26);
|
|
bw.appendExpGolomb(picInitQsMinus26);
|
|
bw.appendExpGolomb(chromaQpIndexOffset);
|
|
bw.append(deblockingFilterControlPresentFlag ? 1 : 0, 1);
|
|
bw.append(constrainedIntraPredFlag ? 1 : 0, 1);
|
|
bw.append(redundantPicCntPresentFlag ? 1 : 0, 1);
|
|
|
|
if (status_moreRBSP){
|
|
bw.append(transform8x8ModeFlag ? 1 : 0, 1);
|
|
bw.append(picScalingMatrixPresentFlag, 1);
|
|
if (picScalingMatrixPresentFlag){
|
|
for (int i = 0; i < derived_scalingListSize; i++){
|
|
bw.append(0, 1);//picScalingMatrixPresnetFlags[i]
|
|
}
|
|
}
|
|
bw.appendExpGolomb(secondChromaQpIndexOffset);
|
|
}
|
|
bw.append(1,1);
|
|
|
|
std::string tmp = bw.str();
|
|
std::string res;
|
|
for (int i = 0; i < tmp.size(); i++){
|
|
if (res.size() > 2 && res[res.size() - 1] == 0x00 && res[res.size() - 2] == 0x00){
|
|
if (tmp[i] == 0x00 || tmp[i] == 0x01 || tmp[i] == 0x02 || tmp[i] == 0x03){
|
|
res += (char)0x03;
|
|
}
|
|
}
|
|
res += tmp[i];
|
|
}
|
|
return res;
|
|
}
|
|
|
|
codedSliceUnit::codedSliceUnit(const char * data, size_t len) : nalUnit(data, len) {
|
|
Utils::bitstream bs;
|
|
for (size_t i = 1; i < len; i++) {
|
|
if (i + 2 < len && (memcmp(data + i, "\000\000\003", 3) == 0)){//Emulation prevention bytes
|
|
//Yes, we increase i here
|
|
bs.append(data + i, 2);
|
|
i += 2;
|
|
} else {
|
|
//No we don't increase i here
|
|
bs.append(data + i, 1);
|
|
}
|
|
}
|
|
firstMbInSlice = bs.getUExpGolomb();
|
|
sliceType = bs.getUExpGolomb();
|
|
picParameterSetId = bs.getUExpGolomb();
|
|
}
|
|
|
|
void codedSliceUnit::setPPSNumber(size_t newNumber){
|
|
//for now, can only convert from 0 to 16
|
|
if (picParameterSetId != 0){
|
|
return;
|
|
}
|
|
size_t bitOffset = 0;
|
|
bitOffset += Utils::bitstream::bitSizeUExpGolomb(firstMbInSlice);
|
|
bitOffset += Utils::bitstream::bitSizeUExpGolomb(sliceType);
|
|
size_t byteOffset = bitOffset / 8;
|
|
bitOffset -= (byteOffset * 8);
|
|
INFO_MSG("Offset for this value: %d bytes and %d bits", byteOffset, bitOffset);
|
|
size_t firstBitmask = ((1 << bitOffset) - 1) << (8-bitOffset);
|
|
size_t secondBitmask = (1 << (8-bitOffset)) -1;
|
|
INFO_MSG("Bitmasks: %0.2X, %0.2X", firstBitmask, secondBitmask);
|
|
char toCopy = payload[1+byteOffset];
|
|
payload.insert(1+byteOffset, 1, toCopy);
|
|
payload[1+byteOffset] &= firstBitmask;
|
|
payload[1+byteOffset] |= (0x08 >> bitOffset);
|
|
payload[2+byteOffset] &= secondBitmask;
|
|
payload[2+byteOffset] |= (0x08 << (8-bitOffset));
|
|
INFO_MSG("Translated %0.2X to %0.2X %0.2X", toCopy, payload[1+byteOffset], payload[2+byteOffset]);
|
|
}
|
|
|
|
void codedSliceUnit::toPrettyString(std::ostream & out) {
|
|
std::string strSliceType = "Unknown";
|
|
switch (sliceType){
|
|
case 5:
|
|
case 0: strSliceType = "P - Predictive slice (at most 1 reference)"; break;
|
|
case 6:
|
|
case 1: strSliceType = " B - Bi-predictive slice (at most 2 references)"; break;
|
|
case 7:
|
|
case 2: strSliceType = " I - Intra slice (no external references)"; break;
|
|
case 8:
|
|
case 3: strSliceType = " SP - Switching predictive slice (at most 1 reference)"; break;
|
|
case 9:
|
|
case 4: strSliceType = " SI - Switching intra slice (no external references)"; break;
|
|
}
|
|
out << "Nal unit of type " << (((uint8_t)payload[0]) & 0x1F) << " [Coded Slice] , " << payload.size() << " bytes long" << std::endl;
|
|
out << " first_mb_in_slice: " << firstMbInSlice << std::endl;
|
|
out << " slice_type " << sliceType << ": " << strSliceType << std::endl;
|
|
out << " pic_parameter_set_id: " << picParameterSetId << (picParameterSetId >= 256 ? " INVALID" : "") << std::endl;
|
|
}
|
|
|
|
|
|
seiUnit::seiUnit(const char * data, size_t len) : nalUnit(data, len) {
|
|
Utils::bitstream bs;
|
|
payloadOffset = 1;
|
|
for (size_t i = 1; i < len; i++) {
|
|
if (i + 2 < len && (memcmp(data + i, "\000\000\003", 3) == 0)){//Emulation prevention bytes
|
|
//Yes, we increase i here
|
|
bs.append(data + i, 2);
|
|
i += 2;
|
|
} else {
|
|
//No we don't increase i here
|
|
bs.append(data + i, 1);
|
|
}
|
|
}
|
|
uint8_t tmp = bs.get(8);
|
|
++payloadOffset;
|
|
payloadType = 0;
|
|
while (tmp == 0xFF){
|
|
payloadType += tmp;
|
|
tmp = bs.get(8);
|
|
++payloadOffset;
|
|
}
|
|
payloadType += tmp;
|
|
|
|
tmp = bs.get(8);
|
|
++payloadOffset;
|
|
payloadSize = 0;
|
|
while (tmp == 0xFF){
|
|
payloadSize += tmp;
|
|
tmp = bs.get(8);
|
|
++payloadOffset;
|
|
}
|
|
payloadSize += tmp;
|
|
}
|
|
|
|
void seiUnit::toPrettyString(std::ostream & out) {
|
|
out << "Nal unit of type " << (((uint8_t)payload[0]) & 0x1F) << " [Supplemental Enhancement Unit] , " << payload.size() << " bytes long" << std::endl;
|
|
switch (payloadType){
|
|
case 5:{//User data, unregistered
|
|
out << " Type 5: User data, unregistered." << std::endl;
|
|
std::stringstream uuid;
|
|
for (uint32_t i = payloadOffset; i < payloadOffset+16; ++i){
|
|
uuid << std::setw(2) << std::setfill('0') << std::hex << (int)(payload.data()[i]);
|
|
}
|
|
if (uuid.str() == "dc45e9bde6d948b7962cd820d923eeef"){
|
|
uuid.str("x264 encoder configuration");
|
|
}
|
|
out << " UUID: " << uuid.str() << std::endl;
|
|
out << " Payload: " << std::string(payload.data()+payloadOffset+16, payloadSize - 17) << std::endl;
|
|
}
|
|
break;
|
|
default:
|
|
out << " Message of type " << payloadType << ", " << payloadSize << " bytes long" << std::endl;
|
|
}
|
|
}
|
|
|
|
nalUnit * nalFactory(const char * _data, size_t _len, size_t & offset, bool annexb) {
|
|
if (annexb){
|
|
//check if we have a start marker at the beginning, if so, move the offset over
|
|
if (_len > offset && !_data[offset]){
|
|
for (size_t i = offset+1; i < _len; ++i){
|
|
if (_data[i] > 1){
|
|
FAIL_MSG("Encountered bullshit AnnexB data..?");
|
|
return 0;
|
|
}
|
|
if (_data[i] == 1){offset = i+1; break;}
|
|
}
|
|
}
|
|
//now we know we're starting at real data. Yay!
|
|
}
|
|
if (_len < offset + 4){
|
|
WARN_MSG("Not at least 4 bytes available - cancelling");
|
|
return 0;
|
|
}
|
|
uint32_t pktLen = 0;
|
|
if (!annexb){
|
|
//read the 4b size in front
|
|
pktLen = Bit::btohl(_data+offset);
|
|
if (_len - offset < 4 + pktLen){
|
|
WARN_MSG("Not at least 4+%lu bytes available - cancelling", pktLen);
|
|
return 0;
|
|
}
|
|
offset += 4;
|
|
}
|
|
const char * data = _data + offset;
|
|
size_t len = _len - offset;
|
|
if (annexb){
|
|
//search for the next start marker
|
|
for (size_t i = 1; i < len-2; ++i){
|
|
if (data[i] == 0 && data[i+1] == 0 && data[i+2] == 1){
|
|
while (i && !data[i]){--i;}
|
|
pktLen = i+1;
|
|
offset += pktLen;
|
|
break;
|
|
}
|
|
}
|
|
}else{
|
|
offset += pktLen;
|
|
}
|
|
if (!pktLen){
|
|
WARN_MSG("Cannot determine packet length - cancelling");
|
|
return 0;
|
|
}
|
|
switch (data[0] & 0x1F){
|
|
case 1:
|
|
case 5:
|
|
case 19:
|
|
return new codedSliceUnit(data, pktLen);
|
|
case 6:
|
|
return new seiUnit(data, pktLen);
|
|
case 7:
|
|
return new spsUnit(data, pktLen);
|
|
case 8:
|
|
return new ppsUnit(data, pktLen);
|
|
default:
|
|
return new nalUnit(data, pktLen);
|
|
}
|
|
}
|
|
|
|
nalUnit * nalFactory(FILE * in, bool annexb) {
|
|
nalUnit * result = NULL;
|
|
char size[4];
|
|
if (fread(size, 4, 1, in)){
|
|
if (annexb){
|
|
size_t curPos = ftell(in);
|
|
if (size[2] == 0x01){
|
|
curPos --;
|
|
}
|
|
fseek(in, curPos, SEEK_SET);
|
|
char * data= (char*)malloc(1024 * 1024 * sizeof(char));//allocate 1MB in size
|
|
size_t len = fread(data, 1, 1024 * 1024, in);
|
|
if (len){
|
|
std::string str(data, len);
|
|
size_t nextPos = str.find("\000\000\001", 0, 3);
|
|
if (nextPos == std::string::npos && feof(in)){
|
|
nextPos = len;
|
|
}
|
|
if (nextPos != std::string::npos){
|
|
if (str[nextPos - 1] == 0x00){
|
|
nextPos --;
|
|
}
|
|
switch (data[0] & 0x1F){
|
|
case 1:
|
|
case 5:
|
|
case 19:
|
|
result = new codedSliceUnit(data, nextPos);
|
|
break;
|
|
case 6:
|
|
result = new seiUnit(data, nextPos);
|
|
break;
|
|
case 7:
|
|
result = new spsUnit(data, nextPos);
|
|
break;
|
|
case 8:
|
|
result = new ppsUnit(data, nextPos);
|
|
break;
|
|
default:
|
|
result = new nalUnit(data, nextPos);
|
|
break;
|
|
}
|
|
fseek(in, curPos + nextPos, SEEK_SET);
|
|
}else{
|
|
FAIL_MSG("NAL Unit of over 1MB, unexpected behaviour until next AnnexB boundary in file");
|
|
}
|
|
}
|
|
free(data);
|
|
}else{
|
|
uint32_t len = Bit::btohl(size);
|
|
char * data = (char*)malloc(len * sizeof(char));
|
|
if (fread(data, len, 1, in)){
|
|
switch (data[0] & 0x1F){
|
|
case 7:
|
|
result = new spsUnit(data, len);
|
|
break;
|
|
default:
|
|
result = new nalUnit(data, len);
|
|
break;
|
|
}
|
|
}
|
|
free(data);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
vui_parameters::vui_parameters(Utils::bitstream & bs) {
|
|
aspectRatioInfoPresentFlag = bs.get(1);
|
|
if (aspectRatioInfoPresentFlag){
|
|
aspectRatioIdc = bs.get(8);
|
|
if (aspectRatioIdc == 255){
|
|
sarWidth = bs.get(16);
|
|
sarHeight = bs.get(16);
|
|
}
|
|
}
|
|
overscanInfoPresentFlag = bs.get(1);
|
|
if (overscanInfoPresentFlag){
|
|
overscanAppropriateFlag = bs.get(1);
|
|
}
|
|
videoSignalTypePresentFlag = bs.get(1);
|
|
if (videoSignalTypePresentFlag){
|
|
videoFormat = bs.get(3);
|
|
videoFullRangeFlag = bs.get(1);;
|
|
colourDescriptionPresentFlag = bs.get(1);
|
|
if (colourDescriptionPresentFlag){
|
|
colourPrimaries = bs.get(8);
|
|
transferCharacteristics = bs.get(8);
|
|
matrixCoefficients = bs.get(8);
|
|
}
|
|
}
|
|
chromaLocInfoPresentFlag = bs.get(1);
|
|
if (chromaLocInfoPresentFlag){
|
|
chromaSampleLocTypeTopField = bs.getUExpGolomb();
|
|
chromaSampleLocTypeBottomField = bs.getUExpGolomb();
|
|
}
|
|
timingInfoPresentFlag = bs.get(1);
|
|
if (timingInfoPresentFlag){
|
|
numUnitsInTick = bs.get(32);
|
|
timeScale = bs.get(32);
|
|
fixedFrameRateFlag = bs.get(1);
|
|
}
|
|
nalHrdParametersPresentFlag = bs.get(1);
|
|
//hrd param nal
|
|
vclHrdParametersPresentFlag = bs.get(1);
|
|
//hrd param vcl
|
|
if (nalHrdParametersPresentFlag || vclHrdParametersPresentFlag){
|
|
lowDelayHrdFlag = bs.get(1);
|
|
}
|
|
picStructPresentFlag = bs.get(1);
|
|
bitstreamRestrictionFlag = bs.get(1);
|
|
if (bitstreamRestrictionFlag){
|
|
motionVectorsOverPicBoundariesFlag = bs.get(1);
|
|
maxBytesPerPicDenom = bs.getUExpGolomb();
|
|
maxBitsPerMbDenom = bs.getUExpGolomb();
|
|
log2MaxMvLengthHorizontal = bs.getUExpGolomb();
|
|
log2MaxMvLengthVertical = bs.getUExpGolomb();
|
|
numReorderFrames = bs.getUExpGolomb();
|
|
maxDecFrameBuffering = bs.getUExpGolomb();
|
|
}
|
|
}
|
|
|
|
void vui_parameters::generate(Utils::bitWriter & bw){
|
|
bw.append(aspectRatioInfoPresentFlag ? 1 : 0, 1);
|
|
if (aspectRatioInfoPresentFlag){
|
|
bw.append(aspectRatioIdc, 8);
|
|
if (aspectRatioIdc == 0xFF){
|
|
bw.append(sarWidth, 16);
|
|
bw.append(sarHeight, 16);
|
|
}
|
|
}
|
|
bw.append(overscanInfoPresentFlag ? 1 : 0, 1);
|
|
if (overscanInfoPresentFlag){
|
|
bw.append(overscanAppropriateFlag ? 1 : 0, 1);
|
|
}
|
|
bw.append(videoSignalTypePresentFlag ? 1 : 0, 1);
|
|
if (videoSignalTypePresentFlag){
|
|
bw.append(videoFormat, 3);
|
|
bw.append(videoFullRangeFlag, 1);
|
|
bw.append(colourDescriptionPresentFlag ? 1 : 0, 1);
|
|
if (colourDescriptionPresentFlag){
|
|
bw.append(colourPrimaries, 8);
|
|
bw.append(transferCharacteristics, 8);
|
|
bw.append(matrixCoefficients, 8);
|
|
}
|
|
}
|
|
bw.append(chromaLocInfoPresentFlag ? 1 : 0, 1);
|
|
if (chromaLocInfoPresentFlag){
|
|
bw.appendUExpGolomb(chromaSampleLocTypeTopField);
|
|
bw.appendUExpGolomb(chromaSampleLocTypeBottomField);
|
|
}
|
|
bw.append(timingInfoPresentFlag ? 1 : 0, 1);
|
|
if (timingInfoPresentFlag){
|
|
bw.append(numUnitsInTick, 32);
|
|
bw.append(timeScale, 32);
|
|
bw.append(fixedFrameRateFlag ? 1 : 0, 1);
|
|
}
|
|
bw.append(nalHrdParametersPresentFlag ? 1 : 0, 1);
|
|
if (nalHrdParametersPresentFlag){
|
|
|
|
}
|
|
bw.append(vclHrdParametersPresentFlag ? 1 : 0, 1);
|
|
if (vclHrdParametersPresentFlag){
|
|
|
|
}
|
|
if (nalHrdParametersPresentFlag || vclHrdParametersPresentFlag){
|
|
bw.append(lowDelayHrdFlag ? 1 : 0, 1);
|
|
}
|
|
bw.append(picStructPresentFlag ? 1 : 0, 1);
|
|
bw.append(bitstreamRestrictionFlag ? 1 : 0, 1);
|
|
if (bitstreamRestrictionFlag){
|
|
bw.append(motionVectorsOverPicBoundariesFlag ? 1 : 0, 1);
|
|
bw.appendUExpGolomb(maxBytesPerPicDenom);
|
|
bw.appendUExpGolomb(maxBitsPerMbDenom);
|
|
bw.appendUExpGolomb(log2MaxMvLengthHorizontal);
|
|
bw.appendUExpGolomb(log2MaxMvLengthVertical);
|
|
bw.appendUExpGolomb(numReorderFrames);
|
|
bw.appendUExpGolomb(maxDecFrameBuffering);
|
|
}
|
|
}
|
|
|
|
void vui_parameters::toPrettyString(std::ostream & out, size_t indent){
|
|
out << std::string(indent, ' ') << "Vui parameters" << std::endl;
|
|
out << std::string(indent + 2, ' ') << "aspect_ratio_info_present_flag: " << aspectRatioInfoPresentFlag << std::endl;
|
|
if (aspectRatioInfoPresentFlag){
|
|
out << std::string(indent + 2, ' ') << "aspect_ratio_idc: " << (int32_t)aspectRatioIdc << std::endl;
|
|
if (aspectRatioIdc == 255){
|
|
out << std::string(indent + 2, ' ') << "sar_width: " << sarWidth << std::endl;
|
|
out << std::string(indent + 2, ' ') << "sar_height: " << sarHeight << std::endl;
|
|
}
|
|
}
|
|
out << std::string(indent + 2, ' ') << "overscan_info_present_flag: " << overscanInfoPresentFlag << std::endl;
|
|
if (overscanInfoPresentFlag){
|
|
out << std::string(indent + 2, ' ') << "overscan_appropriate_present_flag: " << overscanAppropriateFlag << std::endl;
|
|
}
|
|
out << std::string(indent + 2, ' ') << "video_signal_type_present_flag: " << videoSignalTypePresentFlag << std::endl;
|
|
if (videoSignalTypePresentFlag){
|
|
out << std::string(indent + 2, ' ') << "video_format" << videoFormat << std::endl;
|
|
out << std::string(indent + 2, ' ') << "video_full_range_flag" << videoFullRangeFlag << std::endl;
|
|
out << std::string(indent + 2, ' ') << "colour_description_present_flag" << colourDescriptionPresentFlag << std::endl;
|
|
if (colourDescriptionPresentFlag){
|
|
out << std::string(indent + 2, ' ') << "colour_primaries" << colourPrimaries << std::endl;
|
|
out << std::string(indent + 2, ' ') << "transfer_characteristics" << transferCharacteristics << std::endl;
|
|
out << std::string(indent + 2, ' ') << "matrix_coefficients" << matrixCoefficients << std::endl;
|
|
}
|
|
}
|
|
out << std::string(indent + 2, ' ') << "chroma_loc_info_present_flag: " << chromaLocInfoPresentFlag << std::endl;
|
|
if (chromaLocInfoPresentFlag){
|
|
out << std::string(indent + 2, ' ') << "chroma_sample_loc_type_top_field" << chromaSampleLocTypeTopField << std::endl;
|
|
out << std::string(indent + 2, ' ') << "chroma_sample_loc_type_bottom_field" << chromaSampleLocTypeBottomField << std::endl;
|
|
}
|
|
out << std::string(indent + 2, ' ') << "timing_info_present_flag: " << timingInfoPresentFlag << std::endl;
|
|
if (timingInfoPresentFlag){
|
|
out << std::string(indent + 2, ' ') << "num_units_in_tick: " << numUnitsInTick << std::endl;
|
|
out << std::string(indent + 2, ' ') << "time_scale: " << timeScale << std::endl;
|
|
out << std::string(indent + 2, ' ') << "fixed_frame_rate_flag: " << fixedFrameRateFlag << std::endl;
|
|
}
|
|
out << std::string(indent + 2, ' ') << "nal_hrd_parameters_present_flag: " << nalHrdParametersPresentFlag << std::endl;
|
|
|
|
out << std::string(indent + 2, ' ') << "vcl_hrd_parameters_present_flag: " << vclHrdParametersPresentFlag << std::endl;
|
|
if (nalHrdParametersPresentFlag || vclHrdParametersPresentFlag){
|
|
out << std::string(indent + 2, ' ') << "low_delay_hrd_flag: " << lowDelayHrdFlag << std::endl;
|
|
}
|
|
out << std::string(indent + 2, ' ') << "pic_struct_present_flag: " << picStructPresentFlag << std::endl;
|
|
out << std::string(indent + 2, ' ') << "bitstream_restiction_flag: " << bitstreamRestrictionFlag << std::endl;
|
|
if (bitstreamRestrictionFlag){
|
|
out << std::string(indent + 2, ' ') << "motion_vectors_over_pic_boundaries_flag: " << motionVectorsOverPicBoundariesFlag << std::endl;
|
|
out << std::string(indent + 2, ' ') << "max_bytes_per_pic_denom: " << maxBytesPerPicDenom << std::endl;
|
|
out << std::string(indent + 2, ' ') << "max_bits_per_mb_denom: " << maxBitsPerMbDenom << std::endl;
|
|
out << std::string(indent + 2, ' ') << "log2_max_mv_length_horizontal: " << log2MaxMvLengthHorizontal << std::endl;
|
|
out << std::string(indent + 2, ' ') << "log2_max_mv_length_vertical: " << log2MaxMvLengthVertical << std::endl;
|
|
out << std::string(indent + 2, ' ') << "num_reorder_frames: " << numReorderFrames << std::endl;
|
|
out << std::string(indent + 2, ' ') << "max_dec_frame_buffering: " << maxDecFrameBuffering << std::endl;
|
|
}
|
|
}
|
|
}
|
|
|