HEVC Test Model (HM)  HM-16.3
NALread.cpp
Go to the documentation of this file.
1 /* The copyright in this software is being made available under the BSD
2  * License, included below. This software may be subject to other third party
3  * and contributor rights, including patent rights, and no such rights are
4  * granted under this license.
5  *
6  * Copyright (c) 2010-2015, ITU/ISO/IEC
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions are met:
11  *
12  * * Redistributions of source code must retain the above copyright notice,
13  * this list of conditions and the following disclaimer.
14  * * Redistributions in binary form must reproduce the above copyright notice,
15  * this list of conditions and the following disclaimer in the documentation
16  * and/or other materials provided with the distribution.
17  * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18  * be used to endorse or promote products derived from this software without
19  * specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31  * THE POSSIBILITY OF SUCH DAMAGE.
32  */
33 
40 #include <vector>
41 #include <algorithm>
42 #include <ostream>
43 
44 #include "NALread.h"
45 #include "TLibCommon/NAL.h"
47 #if RExt__DECODER_DEBUG_BIT_STATISTICS
49 #endif
50 
51 using namespace std;
52 
55 static Void convertPayloadToRBSP(vector<uint8_t>& nalUnitBuf, TComInputBitstream *bitstream, Bool isVclNalUnit)
56 {
57  UInt zeroCount = 0;
58  vector<uint8_t>::iterator it_read, it_write;
59 
60  UInt pos = 0;
62  for (it_read = it_write = nalUnitBuf.begin(); it_read != nalUnitBuf.end(); it_read++, it_write++, pos++)
63  {
64  assert(zeroCount < 2 || *it_read >= 0x03);
65  if (zeroCount == 2 && *it_read == 0x03)
66  {
67  bitstream->pushEmulationPreventionByteLocation( pos );
68  pos++;
69  it_read++;
70  zeroCount = 0;
71 #if RExt__DECODER_DEBUG_BIT_STATISTICS
73 #endif
74  if (it_read == nalUnitBuf.end())
75  {
76  break;
77  }
78  assert(*it_read <= 0x03);
79  }
80  zeroCount = (*it_read == 0x00) ? zeroCount+1 : 0;
81  *it_write = *it_read;
82  }
83  assert(zeroCount == 0);
84 
85  if (isVclNalUnit)
86  {
87  // Remove cabac_zero_word from payload if present
88  Int n = 0;
89 
90  while (it_write[-1] == 0x00)
91  {
92  it_write--;
93  n++;
94  }
95 
96  if (n > 0)
97  {
98  printf("\nDetected %d instances of cabac_zero_word\n", n/2);
99  }
100  }
101 
102  nalUnitBuf.resize(it_write - nalUnitBuf.begin());
103 }
104 
106 {
107  TComInputBitstream& bs = *nalu.m_Bitstream;
108 
109  Bool forbidden_zero_bit = bs.read(1); // forbidden_zero_bit
110  assert(forbidden_zero_bit == 0);
111  nalu.m_nalUnitType = (NalUnitType) bs.read(6); // nal_unit_type
112  nalu.m_nuhLayerId = bs.read(6); // nuh_layer_id
113  nalu.m_temporalId = bs.read(3) - 1; // nuh_temporal_id_plus1
114 #if RExt__DECODER_DEBUG_BIT_STATISTICS
116 #endif
117 
118  // only check these rules for base layer
119  if (nalu.m_nuhLayerId == 0)
120  {
121  if ( nalu.m_temporalId )
122  {
129  && nalu.m_nalUnitType != NAL_UNIT_VPS
130  && nalu.m_nalUnitType != NAL_UNIT_SPS
131  && nalu.m_nalUnitType != NAL_UNIT_EOS
132  && nalu.m_nalUnitType != NAL_UNIT_EOB );
133  }
134  else
135  {
140  }
141  }
142 }
147 Void read(InputNALUnit& nalu, vector<uint8_t>& nalUnitBuf)
148 {
149  /* perform anti-emulation prevention */
150  TComInputBitstream *pcBitstream = new TComInputBitstream(NULL);
151  convertPayloadToRBSP(nalUnitBuf, pcBitstream, (nalUnitBuf[0] & 64) == 0);
152 
153  nalu.m_Bitstream = new TComInputBitstream(&nalUnitBuf);
155  delete pcBitstream;
156  readNalUnitHeader(nalu);
157 }
TComInputBitstream * m_Bitstream
Definition: NALread.h:60
void Void
Definition: TypeDef.h:285
reading funtionality for NAL units
#define NULL
Definition: CommonDef.h:100
Void readNalUnitHeader(InputNALUnit &nalu)
Definition: NALread.cpp:105
unsigned int UInt
Definition: TypeDef.h:297
NalUnitType m_nalUnitType
nal_unit_type
Definition: NAL.h:49
STL namespace.
Void setEmulationPreventionByteLocation(std::vector< UInt > vec)
static Void convertPayloadToRBSP(vector< uint8_t > &nalUnitBuf, TComInputBitstream *bitstream, Bool isVclNalUnit)
Definition: NALread.cpp:55
Void pushEmulationPreventionByteLocation(UInt pos)
Void read(InputNALUnit &nalu, vector< uint8_t > &nalUnitBuf)
Definition: NALread.cpp:147
bool Bool
Definition: TypeDef.h:286
class for handling bitstream (header)
std::vector< UInt > getEmulationPreventionByteLocation()
UInt m_nuhLayerId
nuh_layer_id
Definition: NAL.h:51
Void read(UInt uiNumberOfBits, UInt &ruiBits)
int Int
Definition: TypeDef.h:296
NalUnitType
Definition: CommonDef.h:218
UInt m_temporalId
temporal_id
Definition: NAL.h:50
static Void IncrementStatisticEP(const TComCodingStatisticsClassType &stat, const Int numBits, const Int value)
Void clearEmulationPreventionByteLocation()