I need to read the POC number from the HEVC bitstream using python BitStream. Currently I read the nal unit header. Is there an easy way to get it?
I have trace enabled HM14.0 but the EncTrace.txt does not include all my packets.
Any idea?
Edit: I attach my python code. This is my bitstream file .
import re
import collections
import bitstring
from bitstring import BitStream, BitArray, ConstBitStream, pack
from bitstring import ByteStore, offsetcopy
import unicodedata
text_file = open("nal_Output.txt", "wb")
s = BitStream(filename='new.str')
#Find number of packets
pcks = list(s.findall('0x000001', bytealigned=True))
s.pos=0
num_of_pcks = len(pcks)
num_of_POC_pcks =0
for x in range(0, num_of_pcks):
s.pos =pcks[x]+24
#print x
if x < num_of_pcks-1:
no_p = pcks[x+1]-pcks[x]-24
else:
no_p = 0
forbidden_zero_bit = s.read('uint:1')
nal_unit_type = s.read('uint:6')
nuh_layer_id = s.read('uint:6')
nuh_temporal_id_plus1 = s.read('uint:3')
#nal unit type 39 is for SEI messages: one byte message
if int(nal_unit_type) >31 :
#print 'nal=39'
#size_of_pck = (no_p+24+8)/8
packet = 0
elif int(nal_unit_type) <32:
#print int(nal_unit_type)
num_of_POC_pcks = num_of_POC_pcks+1
size_of_pck = (no_p+24)/8
text_file.write("NumBytesInNALunit: {0}\n".format(str(size_of_pck)))
s.read(no_p)
print num_of_POC_pcks
text_file.close()
Chapter 8.3.1 in the standard tells you how the PicOrderCntVal
is deriverad. In python, this is non-trivial to extract...
Using the reference SW , the slice header is parsed in function TDecCavlc::parseSliceHeader
in source/Lib/TLibDecoder/TDecCAVLC.cpp
that is also where the POC-value is read.
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.