Context navigation

TDecTop.h @ 852

Visit:

Last change on this file since 852 was 851, checked in by qualcomm, 10 years ago

R0223: Restrictions on POC reset-related syntax elements (Macro: POC_RESET_RESTRICTIONS)

Allow mixing of CRA and BLA pictures without requiring POC resetting (adoption) and other restrictions that were not implemented before.

From: Adarsh K. Ramasubramonian <aramasub@…>

Property svn:eol-style set to native

File size: 14.9 KB

Line
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-2014, 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
34	/** \file TDecTop.h
35	\brief decoder class (header)
36	*/
37
38	#ifndef __TDECTOP__
39	#define __TDECTOP__
40
41	#include "TLibCommon/CommonDef.h"
42	#include "TLibCommon/TComList.h"
43	#include "TLibCommon/TComPicYuv.h"
44	#include "TLibCommon/TComPic.h"
45	#include "TLibCommon/TComTrQuant.h"
46	#include "TLibCommon/SEI.h"
47	#if Q0048_CGS_3D_ASYMLUT
48	#include "TLibCommon/TCom3DAsymLUT.h"
49	#endif
50
51	#include "TDecGop.h"
52	#include "TDecEntropy.h"
53	#include "TDecSbac.h"
54	#include "TDecCAVLC.h"
55	#include "SEIread.h"
56
57	struct InputNALUnit;
58
59	//! \ingroup TLibDecoder
60	//! \{
61
62	// ====================================================================================================================
63	// Class definition
64	// ====================================================================================================================
65
66	#if Q0074_SEI_COLOR_MAPPING
67	class TDecColorMapping
68	{
69	Int m_colorMapId;
70	Bool m_colorMapCancelFlag;
71	Bool m_colorMapPersistenceFlag;
72	Bool m_colorMap_video_signal_type_present_flag;
73	Bool m_colorMap_video_full_range_flag;
74	Int m_colorMap_primaries;
75	Int m_colorMap_transfer_characteristics;
76	Int m_colorMap_matrix_coeffs;
77	Int m_colorMapModelId;
78
79	Int m_colour_map_coded_data_bit_depth;
80	Int m_colour_map_target_bit_depth;
81
82	Int m_num_input_pivots[3];
83	Int* m_coded_input_pivot_value[3];
84	Int* m_target_input_pivot_value[3];
85
86	Bool m_matrix_flag;
87	Int m_log2_matrix_denom;
88	Int m_matrix_coef[3][3];
89
90	Int m_num_output_pivots[3];
91	Int* m_coded_output_pivot_value[3];
92	Int* m_target_output_pivot_value[3];
93
94	Bool m_lut1d_computed[3];
95	Int* m_lut1d_input[3];
96	Int* m_lut1d_output[3];
97	TComPicYuv* m_pcColorMappingPic[2];
98
99	public:
100	TDecColorMapping();
101	~TDecColorMapping();
102
103	Bool getColorMappingFlag() { return(!m_colorMapCancelFlag);};
104
105	Void setColorMapping( SEIMessages m_SEIs );
106	Void setColorMapping( Int bitDepthY, Int bitDepthC );
107	TComPicYuv* getColorMapping( TComPicYuv* pPicYuvRec, Int iTop=0, Int curlayerId=0 );
108	};// END CLASS DEFINITION TDecColorMapping
109	#endif
110
111	/// decoder class
112	class TDecTop
113	{
114	private:
115	Int m_iMaxRefPicNum;
116
117	NalUnitType m_associatedIRAPType; ///< NAL unit type of the associated IRAP picture
118	Int m_pocCRA; ///< POC number of the latest CRA picture
119	Int m_pocRandomAccess; ///< POC number of the random access point (the first IDR or CRA picture)
120
121	TComList<TComPic*> m_cListPic; // Dynamic buffer
122	ParameterSetManagerDecoder m_parameterSetManagerDecoder; // storage for parameter sets
123	TComSlice* m_apcSlicePilot;
124
125	SEIMessages m_SEIs; ///< List of SEI messages that have been received before the first slice and between slices
126
127	// functional classes
128	TComPrediction m_cPrediction;
129	#if Q0048_CGS_3D_ASYMLUT
130	TCom3DAsymLUT m_c3DAsymLUTPPS;
131	TComPicYuv* m_pColorMappedPic;
132	#endif
133	TComTrQuant m_cTrQuant;
134	TDecGop m_cGopDecoder;
135	TDecSlice m_cSliceDecoder;
136	TDecCu m_cCuDecoder;
137	TDecEntropy m_cEntropyDecoder;
138	TDecCavlc m_cCavlcDecoder;
139	TDecSbac m_cSbacDecoder;
140	TDecBinCABAC m_cBinCABAC;
141	SEIReader m_seiReader;
142	TComLoopFilter m_cLoopFilter;
143	TComSampleAdaptiveOffset m_cSAO;
144
145	Bool isSkipPictureForBLA(Int& iPOCLastDisplay);
146	Bool isRandomAccessSkipPicture(Int& iSkipFrame, Int& iPOCLastDisplay);
147	TComPic* m_pcPic;
148	UInt m_uiSliceIdx;
149	#if !SVC_EXTENSION
150	Int m_prevPOC;
151	#endif
152	Bool m_bFirstSliceInPicture;
153	#if !SVC_EXTENSION
154	Bool m_bFirstSliceInSequence;
155	#endif
156	Bool m_prevSliceSkipped;
157	Int m_skippedPOC;
158	#if SETTING_NO_OUT_PIC_PRIOR
159	Bool m_bFirstSliceInBitstream;
160	Int m_lastPOCNoOutputPriorPics;
161	Bool m_isNoOutputPriorPics;
162	Bool m_craNoRaslOutputFlag; //value of variable NoRaslOutputFlag of the last CRA pic
163	#endif
164	#if Q0177_EOS_CHECKS
165	Bool m_isLastNALWasEos;
166	#endif
167	#if SVC_EXTENSION
168	static UInt m_prevPOC; // POC of the previous slice
169	static UInt m_uiPrevLayerId; // LayerId of the previous slice
170	static Bool m_bFirstSliceInSequence;
171	UInt m_layerId;
172	UInt m_numLayer;
173	TDecTop** m_ppcTDecTop;
174	#if AVC_BASE
175	fstream* m_pBLReconFile;
176	#if !REPN_FORMAT_IN_VPS
177	Int m_iBLSourceWidth;
178	Int m_iBLSourceHeight;
179	#endif
180	#endif
181	#if VPS_EXTN_DIRECT_REF_LAYERS
182	Int m_numDirectRefLayers;
183	Int m_refLayerId[MAX_VPS_LAYER_ID_PLUS1];
184	Int m_numSamplePredRefLayers;
185	Int m_samplePredRefLayerId[MAX_VPS_LAYER_ID_PLUS1];
186	Int m_numMotionPredRefLayers;
187	Int m_motionPredRefLayerId[MAX_VPS_LAYER_ID_PLUS1];
188	Bool m_samplePredEnabledFlag[MAX_VPS_LAYER_ID_PLUS1];
189	Bool m_motionPredEnabledFlag[MAX_VPS_LAYER_ID_PLUS1];
190	#endif
191	TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture
192	#endif
193	#if OUTPUT_LAYER_SET_INDEX
194	CommonDecoderParams* m_commonDecoderParams;
195	#endif
196	#if NO_CLRAS_OUTPUT_FLAG
197	Bool m_noClrasOutputFlag;
198	Bool m_layerInitializedFlag;
199	Bool m_firstPicInLayerDecodedFlag;
200	#endif
201	#if POC_RESET_IDC_DECODER
202	Int m_parseIdc;
203	Int m_lastPocPeriodId;
204	Int m_prevPicOrderCnt;
205	#endif
206	#if RESOLUTION_BASED_DPB
207	Int m_subDpbIdx; // Index to the sub-DPB that the layer belongs to.
208	// When new VPS is activated, this should be re-initialized to -1
209	#endif
210	public:
211	#if Q0074_SEI_COLOR_MAPPING
212	TDecColorMapping* m_ColorMapping;
213	#endif
214
215	#if POC_RESET_RESTRICTIONS
216	static Bool m_checkPocRestrictionsForCurrAu;
217	static Int m_pocResetIdcOrCurrAu;
218	static Bool m_baseLayerIdrFlag;
219	static Bool m_baseLayerPicPresentFlag;
220	static Bool m_baseLayerIrapFlag;
221	static Bool m_nonBaseIdrPresentFlag;
222	static Int m_nonBaseIdrType;
223	static Bool m_picNonIdrWithRadlPresentFlag;
224	static Bool m_picNonIdrNoLpPresentFlag;
225	#endif
226
227	TDecTop();
228	virtual ~TDecTop();
229
230	Void create ();
231	Void destroy ();
232
233	void setDecodedPictureHashSEIEnabled(Int enabled) { m_cGopDecoder.setDecodedPictureHashSEIEnabled(enabled); }
234
235	Void init();
236	#if SVC_EXTENSION
237	Bool decode(InputNALUnit& nalu, Int& iSkipFrame, Int& iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC);
238	#else
239	Bool decode(InputNALUnit& nalu, Int& iSkipFrame, Int& iPOCLastDisplay);
240	#endif
241
242	Void deletePicBuffer();
243
244
245	TComSPS* getActiveSPS() { return m_parameterSetManagerDecoder.getActiveSPS(); }
246
247
248	Void executeLoopFilters(Int& poc, TComList<TComPic>& rpcListPic);
249	#if SETTING_NO_OUT_PIC_PRIOR
250	Void checkNoOutputPriorPics (TComList<TComPic>& rpcListPic);
251	Bool getNoOutputPriorPicsFlag () { return m_isNoOutputPriorPics; }
252	Void setNoOutputPriorPicsFlag (Bool val) { m_isNoOutputPriorPics = val; }
253	#endif
254
255	#if SVC_EXTENSION
256	#if EARLY_REF_PIC_MARKING
257	Void earlyPicMarking(Int maxTemporalLayer, std::vector<Int>& targetDecLayerIdList);
258	#endif
259	#if POC_RESET_IDC_DECODER
260	Int getParseIdc() { return m_parseIdc;}
261	Void setParseIdc(Int x) { m_parseIdc = x;}
262	Void markAllPicsAsNoCurrAu();
263
264	Int getLastPocPeriodId() { return m_lastPocPeriodId; }
265	Void setLastPocPeriodId(Int x) { m_lastPocPeriodId = x; }
266
267	Int getPrevPicOrderCnt() { return m_prevPicOrderCnt; }
268	Void setPrevPicOrderCnt(Int const x) { m_prevPicOrderCnt = x; }
269	#endif
270	UInt getLayerId () { return m_layerId; }
271	Void setLayerId (UInt layer) { m_layerId = layer; }
272	UInt getNumLayer () { return m_numLayer; }
273	Void setNumLayer (UInt uiNum) { m_numLayer = uiNum; }
274	TComList<TComPic> getListPic() { return &m_cListPic; }
275	Void setLayerDec(TDecTop **p) { m_ppcTDecTop = p; }
276	TDecTop* getLayerDec(UInt layer) { return m_ppcTDecTop[layer]; }
277	#if VPS_EXTN_DIRECT_REF_LAYERS
278	TDecTop* getRefLayerDec(UInt refLayerIdc);
279	Int getNumDirectRefLayers () { return m_numDirectRefLayers; }
280	Void setNumDirectRefLayers (Int num) { m_numDirectRefLayers = num; }
281
282	Int getRefLayerId (Int i) { return m_refLayerId[i]; }
283	Void setRefLayerId (Int i, Int refLayerId) { m_refLayerId[i] = refLayerId; }
284
285	Int getNumSamplePredRefLayers () { return m_numSamplePredRefLayers; }
286	Void setNumSamplePredRefLayers (Int num) { m_numSamplePredRefLayers = num; }
287
288	Int getSamplePredRefLayerId (Int i) { return m_samplePredRefLayerId[i]; }
289	Void setSamplePredRefLayerId (Int i, Int refLayerId) { m_samplePredRefLayerId[i] = refLayerId; }
290
291	Int getNumMotionPredRefLayers () { return m_numMotionPredRefLayers; }
292	Void setNumMotionPredRefLayers (Int num) { m_numMotionPredRefLayers = num; }
293
294	Int getMotionPredRefLayerId (Int i) { return m_motionPredRefLayerId[i]; }
295	Void setMotionPredRefLayerId (Int i, Int refLayerId) { m_motionPredRefLayerId[i] = refLayerId; }
296
297	Bool getSamplePredEnabledFlag (Int i) { return m_samplePredEnabledFlag[i]; }
298	Void setSamplePredEnabledFlag (Int i,Bool flag) { m_samplePredEnabledFlag[i] = flag; }
299
300	Bool getMotionPredEnabledFlag (Int i) { return m_motionPredEnabledFlag[i]; }
301	Void setMotionPredEnabledFlag (Int i,Bool flag) { m_motionPredEnabledFlag[i] = flag; }
302
303	TDecTop* getSamplePredRefLayerDec ( UInt layerId );
304	TDecTop* getMotionPredRefLayerDec ( UInt layerId );
305
306	Void setRefLayerParams( TComVPS* vps );
307	#endif
308	#if AVC_BASE
309	Void setBLReconFile( fstream* pFile ) { m_pBLReconFile = pFile; }
310	fstream* getBLReconFile() { return m_pBLReconFile; }
311	#if !REPN_FORMAT_IN_VPS
312	Void setBLsize( Int iWidth, Int iHeight ) { m_iBLSourceWidth = iWidth; m_iBLSourceHeight = iHeight; }
313	Int getBLWidth() { return m_iBLSourceWidth; }
314	Int getBLHeight() { return m_iBLSourceHeight; }
315	#endif
316	#endif
317	#if REPN_FORMAT_IN_VPS
318	Void xInitILRP(TComSlice *slice);
319	#else
320	Void xInitILRP(TComSPS *pcSPS);
321	#endif
322	#if OUTPUT_LAYER_SET_INDEX
323	CommonDecoderParams* getCommonDecoderParams() { return m_commonDecoderParams; }
324	Void setCommonDecoderParams(CommonDecoderParams* x) { m_commonDecoderParams = x; }
325	Void checkValueOfTargetOutputLayerSetIdx(TComVPS *vps);
326	#endif
327	#if SCALINGLIST_INFERRING
328	ParameterSetManagerDecoder* getParameterSetManager() { return &m_parameterSetManagerDecoder; }
329	#endif
330	#if RESOLUTION_BASED_DPB
331	Void setSubDpbIdx(Int idx) { m_subDpbIdx = idx; }
332	Int getSubDpbIdx() { return m_subDpbIdx; }
333	Void assignSubDpbs(TComVPS *vps);
334	#endif
335	#endif //SVC_EXTENSION
336
337	protected:
338	Void xGetNewPicBuffer (TComSlice* pcSlice, TComPic*& rpcPic);
339	Void xCreateLostPicture (Int iLostPOC);
340
341	Void xActivateParameterSets();
342	#if SVC_EXTENSION
343	#if POC_RESET_FLAG
344	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int &iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC);
345	#else
346	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC);
347	#endif
348	#else
349	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay);
350	#endif
351	Void xDecodeVPS();
352	Void xDecodeSPS();
353	Void xDecodePPS(
354	#if Q0048_CGS_3D_ASYMLUT
355	TCom3DAsymLUT * pc3DAsymLUT
356	#endif
357	);
358	Void xDecodeSEI( TComInputBitstream* bs, const NalUnitType nalUnitType );
359
360	#if NO_CLRAS_OUTPUT_FLAG
361	Int getNoClrasOutputFlag() { return m_noClrasOutputFlag;}
362	Void setNoClrasOutputFlag(Bool x) { m_noClrasOutputFlag = x; }
363	Int getLayerInitializedFlag() { return m_layerInitializedFlag;}
364	Void setLayerInitializedFlag(Bool x) { m_layerInitializedFlag = x; }
365	Int getFirstPicInLayerDecodedFlag() { return m_firstPicInLayerDecodedFlag;}
366	Void setFirstPicInLayerDecodedFlag(Bool x) { m_firstPicInLayerDecodedFlag = x; }
367	#endif
368	#if Q0048_CGS_3D_ASYMLUT
369	Void initAsymLut(TComSlice *pcSlice);
370	#endif
371	#if POC_RESET_RESTRICTIONS
372	Void resetPocRestrictionCheckParameters();
373	#endif
374	};// END CLASS DEFINITION TDecTop
375
376
377
378	//! \}
379
380	#endif // __TDECTOP__
381

Note: See TracBrowser for help on using the repository browser.

JCT-VC SHVC

Context navigation

source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibDecoder/TDecTop.h @ 852

Download in other formats: