Context navigation

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

Visit:

Last change on this file since 1348 was 1319, checked in by seregin, 9 years ago
port rev 4394
Property svn:eol-style set to `native`
File size: 14.2 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-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
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/TComPrediction.h"
47	#include "TLibCommon/SEI.h"
48	#if CGS_3D_ASYMLUT
49	#include "TLibCommon/TCom3DAsymLUT.h"
50	#endif
51
52	#include "TDecGop.h"
53	#include "TDecEntropy.h"
54	#include "TDecSbac.h"
55	#include "TDecCAVLC.h"
56	#include "SEIread.h"
57
58	class InputNALUnit;
59
60	//! \ingroup TLibDecoder
61	//! \{
62
63	// ====================================================================================================================
64	// Class definition
65	// ====================================================================================================================
66
67	/// decoder class
68	class TDecTop
69	{
70	private:
71	Int m_iMaxRefPicNum;
72
73	NalUnitType m_associatedIRAPType; ///< NAL unit type of the associated IRAP picture
74	Int m_pocCRA; ///< POC number of the latest CRA picture
75	Int m_pocRandomAccess; ///< POC number of the random access point (the first IDR or CRA picture)
76
77	TComList<TComPic*> m_cListPic; // Dynamic buffer
78	ParameterSetManager m_parameterSetManager; // storage for parameter sets
79	TComSlice* m_apcSlicePilot;
80
81	SEIMessages m_SEIs; ///< List of SEI messages that have been received before the first slice and between slices, excluding prefix SEIs...
82
83	// functional classes
84	TComPrediction m_cPrediction;
85	#if CGS_3D_ASYMLUT
86	TCom3DAsymLUT m_c3DAsymLUTPPS;
87	TComPicYuv* m_pColorMappedPic;
88	#endif
89	TComTrQuant m_cTrQuant;
90	TDecGop m_cGopDecoder;
91	TDecSlice m_cSliceDecoder;
92	TDecCu m_cCuDecoder;
93	TDecEntropy m_cEntropyDecoder;
94	TDecCavlc m_cCavlcDecoder;
95	TDecSbac m_cSbacDecoder;
96	TDecBinCABAC m_cBinCABAC;
97	SEIReader m_seiReader;
98	TComLoopFilter m_cLoopFilter;
99	TComSampleAdaptiveOffset m_cSAO;
100
101	Bool isSkipPictureForBLA(Int& iPOCLastDisplay);
102	Bool isRandomAccessSkipPicture(Int& iSkipFrame, Int& iPOCLastDisplay);
103	TComPic* m_pcPic;
104	UInt m_uiSliceIdx;
105	#if !SVC_EXTENSION
106	Int m_prevPOC;
107	#endif
108	Int m_prevTid0POC;
109	Bool m_bFirstSliceInPicture;
110	#if !SVC_EXTENSION
111	Bool m_bFirstSliceInSequence;
112	#endif
113	Bool m_prevSliceSkipped;
114	Int m_skippedPOC;
115	Bool m_bFirstSliceInBitstream;
116	Int m_lastPOCNoOutputPriorPics;
117	Bool m_isNoOutputPriorPics;
118	Bool m_craNoRaslOutputFlag; //value of variable NoRaslOutputFlag of the last CRA pic
119	#if O0043_BEST_EFFORT_DECODING
120	UInt m_forceDecodeBitDepth;
121	#endif
122	std::ostream *m_pDecodedSEIOutputStream;
123
124	Bool m_warningMessageSkipPicture;
125
126	std::list<InputNALUnit*> m_prefixSEINALUs; /// Buffered up prefix SEI NAL Units.
127
128	#if SVC_EXTENSION
129	Bool m_isLastNALWasEos;
130	Bool m_lastPicHasEos;
131	static UInt m_prevPOC; // POC of the previous slice
132	static UInt m_uiPrevLayerId; // LayerId of the previous slice
133	static Bool m_bFirstSliceInSequence;
134	UInt m_layerId;
135	UInt m_numLayer;
136	TDecTop** m_ppcTDecTop;
137	UInt m_smallestLayerId;
138	Bool m_pocResettingFlag;
139	Bool m_pocDecrementedInDPBFlag;
140	#if AVC_BASE
141	fstream* m_pBLReconFile;
142	#endif
143
144	Int m_numDirectRefLayers;
145	Int m_refLayerId[MAX_VPS_LAYER_IDX_PLUS1];
146	Int m_numSamplePredRefLayers;
147	Int m_numMotionPredRefLayers;
148
149	TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture
150	CommonDecoderParams* m_commonDecoderParams;
151	#if NO_CLRAS_OUTPUT_FLAG
152	Bool m_noClrasOutputFlag;
153	Bool m_layerInitializedFlag;
154	Bool m_firstPicInLayerDecodedFlag;
155	#endif
156	Int m_parseIdc;
157	Int m_lastPocPeriodId;
158	Int m_prevPicOrderCnt;
159	#if CONFORMANCE_BITSTREAM_MODE
160	Bool m_confModeFlag;
161	std::vector<TComPic> m_confListPic; // Dynamic buffer for storing pictures for conformance purposes
162	#endif
163	Bool m_isOutputLayerFlag;
164	#endif //SVC_EXTENSION
165
166	public:
167	#if SVC_EXTENSION
168	static Bool m_checkPocRestrictionsForCurrAu;
169	static Int m_pocResetIdcOrCurrAu;
170	static Bool m_baseLayerIdrFlag;
171	static Bool m_baseLayerPicPresentFlag;
172	static Bool m_baseLayerIrapFlag;
173	static Bool m_nonBaseIdrPresentFlag;
174	static Int m_nonBaseIdrType;
175	static Bool m_picNonIdrWithRadlPresentFlag;
176	static Bool m_picNonIdrNoLpPresentFlag;
177	static Int m_crossLayerPocResetPeriodId;
178	static Int m_crossLayerPocResetIdc;
179	#endif //SVC_EXTENSION
180
181	TDecTop();
182	virtual ~TDecTop();
183
184	Void create ();
185	Void destroy ();
186
187	Void setDecodedPictureHashSEIEnabled(Int enabled) { m_cGopDecoder.setDecodedPictureHashSEIEnabled(enabled); }
188
189	Void init();
190	#if SVC_EXTENSION
191	Bool decode(InputNALUnit& nalu, Int& iSkipFrame, Int& iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC);
192	#else
193	Bool decode(InputNALUnit& nalu, Int& iSkipFrame, Int& iPOCLastDisplay);
194	#endif
195
196	Void deletePicBuffer();
197
198
199	Void executeLoopFilters(Int& poc, TComList<TComPic>& rpcListPic);
200	Void checkNoOutputPriorPics (TComList<TComPic> rpcListPic);
201
202	Bool getNoOutputPriorPicsFlag () { return m_isNoOutputPriorPics; }
203	Void setNoOutputPriorPicsFlag (Bool val) { m_isNoOutputPriorPics = val; }
204	Void setFirstSliceInPicture (bool val) { m_bFirstSliceInPicture = val; }
205	Bool getFirstSliceInSequence () { return m_bFirstSliceInSequence; }
206	Void setFirstSliceInSequence (bool val) { m_bFirstSliceInSequence = val; }
207	#if O0043_BEST_EFFORT_DECODING
208	Void setForceDecodeBitDepth(UInt bitDepth) { m_forceDecodeBitDepth = bitDepth; }
209	#endif
210	Void setDecodedSEIMessageOutputStream(std::ostream *pOpStream) { m_pDecodedSEIOutputStream = pOpStream; }
211	UInt getNumberOfChecksumErrorsDetected() const { return m_cGopDecoder.getNumberOfChecksumErrorsDetected(); }
212
213	#if SVC_EXTENSION
214	Int getParseIdc () { return m_parseIdc; }
215	Void setParseIdc (Int x) { m_parseIdc = x; }
216	Void markAllPicsAsNoCurrAu ( const TComVPS *vps );
217
218	Int getLastPocPeriodId () { return m_lastPocPeriodId; }
219	Void setLastPocPeriodId (Int x) { m_lastPocPeriodId = x; }
220
221	Int getPrevPicOrderCnt () { return m_prevPicOrderCnt; }
222	Void setPrevPicOrderCnt (Int const x) { m_prevPicOrderCnt = x; }
223
224	UInt getLayerId () { return m_layerId; }
225	Void setLayerId (UInt layer) { m_layerId = layer; }
226	UInt getNumLayer () { return m_numLayer; }
227	Void setNumLayer (UInt uiNum) { m_numLayer = uiNum; }
228	TComList<TComPic> getListPic () { return &m_cListPic; }
229	Void setLayerDec (TDecTop **p) { m_ppcTDecTop = p; }
230	TDecTop* getLayerDec (UInt layerId) { return m_ppcTDecTop[layerId]; }
231	Void xDeriveSmallestLayerId(TComVPS* vps);
232
233	TDecTop* getRefLayerDec (UInt refLayerIdx);
234	Int getNumDirectRefLayers () { return m_numDirectRefLayers; }
235	Void setNumDirectRefLayers (Int num) { m_numDirectRefLayers = num; }
236
237	Int getRefLayerId (Int i) { return m_refLayerId[i]; }
238	Void setRefLayerId (Int i, Int refLayerId) { m_refLayerId[i] = refLayerId; }
239
240	Int getNumSamplePredRefLayers () { return m_numSamplePredRefLayers; }
241	Void setNumSamplePredRefLayers (Int num) { m_numSamplePredRefLayers = num; }
242
243	Int getNumMotionPredRefLayers () { return m_numMotionPredRefLayers; }
244	Void setNumMotionPredRefLayers (Int num) { m_numMotionPredRefLayers = num; }
245
246	Void setRefLayerParams( const TComVPS* vps );
247
248	#if AVC_BASE
249	Void setBLReconFile( fstream* pFile ) { m_pBLReconFile = pFile; }
250	fstream* getBLReconFile() { return m_pBLReconFile; }
251	#endif
252	Void xInitILRP(TComSlice *slice);
253	CommonDecoderParams* getCommonDecoderParams() { return m_commonDecoderParams; }
254	Void setCommonDecoderParams(CommonDecoderParams* x) { m_commonDecoderParams = x; }
255	Void checkValueOfTargetOutputLayerSetIdx(TComVPS *vps);
256
257	ParameterSetManager* getParameterSetManager() { return &m_parameterSetManager; }
258
259	#if CONFORMANCE_BITSTREAM_MODE
260	std::vector<TComPic>* getConfListPic() { return &m_confListPic; }
261	Bool getConfModeFlag() const { return m_confModeFlag; }
262	Void setConfModeFlag(Bool x) { m_confModeFlag = x; }
263	#endif
264	#endif //SVC_EXTENSION
265
266	protected:
267	#if SVC_EXTENSION
268	Void xGetNewPicBuffer ( const TComVPS &vps, const TComSPS &sps, const TComPPS &pps, TComPic*& rpcPic, const UInt temporalLayer);
269	#else
270	Void xGetNewPicBuffer (const TComSPS &sps, const TComPPS &pps, TComPic*& rpcPic, const UInt temporalLayer);
271	#endif
272	Void xCreateLostPicture (Int iLostPOC);
273
274	Void xActivateParameterSets();
275	#if SVC_EXTENSION
276	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC);
277	Void xSetSpatialEnhLayerFlag( const TComVPS &vps, const TComSPS &sps, const TComPPS &pps, TComPic* pic );
278	#else
279	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay);
280	#endif
281	Void xDecodeVPS(const std::vector<UChar> &naluData);
282	Void xDecodeSPS(const std::vector<UChar> &naluData);
283	#if CGS_3D_ASYMLUT
284	Void xDecodePPS(const std::vector<UChar> &naluData, TCom3DAsymLUT *pc3DAsymLUT);
285	#else
286	Void xDecodePPS(const std::vector<UChar> &naluData);
287	#endif
288	Void xDecodeSEI( TComInputBitstream* bs, const NalUnitType nalUnitType );
289	Void xUpdatePreviousTid0POC( TComSlice *pSlice ) { if ((pSlice->getTLayer()==0) && (pSlice->isReferenceNalu() && (pSlice->getNalUnitType()!=NAL_UNIT_CODED_SLICE_RASL_R)&& (pSlice->getNalUnitType()!=NAL_UNIT_CODED_SLICE_RADL_R))) { m_prevTid0POC=pSlice->getPOC(); } }
290	Void xParsePrefixSEImessages();
291	Void xParsePrefixSEIsForUnknownVCLNal();
292
293
294	#if SVC_EXTENSION
295	#if NO_CLRAS_OUTPUT_FLAG
296	Int getNoClrasOutputFlag() { return m_noClrasOutputFlag;}
297	Void setNoClrasOutputFlag(Bool x) { m_noClrasOutputFlag = x; }
298	Int getLayerInitializedFlag() { return m_layerInitializedFlag;}
299	Void setLayerInitializedFlag(Bool x) { m_layerInitializedFlag = x; }
300	Int getFirstPicInLayerDecodedFlag() { return m_firstPicInLayerDecodedFlag;}
301	Void setFirstPicInLayerDecodedFlag(Bool x) { m_firstPicInLayerDecodedFlag = x; }
302	#endif
303	#if CGS_3D_ASYMLUT
304	Void initAsymLut(TComSlice *pcSlice);
305	#endif
306	Void resetPocRestrictionCheckParameters();
307	Void xCheckLayerReset();
308	Void xSetNoRaslOutputFlag();
309	Void xSetLayerInitializedFlag();
310	#endif
311	};// END CLASS DEFINITION TDecTop
312
313
314	//! \}
315
316	#endif // __TDECTOP__
317

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

Download in other formats: