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

Rev	Line
[313]	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
[1029]	4	* granted under this license.
[313]	5	*
[1259]	6	* Copyright (c) 2010-2015, ITU/ISO/IEC
[313]	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"
[1263]	46	#include "TLibCommon/TComPrediction.h"
[313]	47	#include "TLibCommon/SEI.h"
[1212]	48	#if CGS_3D_ASYMLUT
[713]	49	#include "TLibCommon/TCom3DAsymLUT.h"
	50	#endif
[313]	51
	52	#include "TDecGop.h"
	53	#include "TDecEntropy.h"
	54	#include "TDecSbac.h"
	55	#include "TDecCAVLC.h"
	56	#include "SEIread.h"
	57
[1319]	58	class InputNALUnit;
[313]	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;
[1029]	72
[442]	73	NalUnitType m_associatedIRAPType; ///< NAL unit type of the associated IRAP picture
[313]	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
[1235]	78	ParameterSetManager m_parameterSetManager; // storage for parameter sets
[313]	79	TComSlice* m_apcSlicePilot;
[442]	80
[1319]	81	SEIMessages m_SEIs; ///< List of SEI messages that have been received before the first slice and between slices, excluding prefix SEIs...
[313]	82
	83	// functional classes
	84	TComPrediction m_cPrediction;
[1212]	85	#if CGS_3D_ASYMLUT
[713]	86	TCom3DAsymLUT m_c3DAsymLUTPPS;
	87	TComPicYuv* m_pColorMappedPic;
	88	#endif
[313]	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
[1295]	108	Int m_prevTid0POC;
[313]	109	Bool m_bFirstSliceInPicture;
	110	#if !SVC_EXTENSION
	111	Bool m_bFirstSliceInSequence;
	112	#endif
[442]	113	Bool m_prevSliceSkipped;
	114	Int m_skippedPOC;
[713]	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
[1029]	119	#if O0043_BEST_EFFORT_DECODING
	120	UInt m_forceDecodeBitDepth;
[713]	121	#endif
[1029]	122	std::ostream *m_pDecodedSEIOutputStream;
	123
[1292]	124	Bool m_warningMessageSkipPicture;
	125
[1319]	126	std::list<InputNALUnit*> m_prefixSEINALUs; /// Buffered up prefix SEI NAL Units.
	127
[1029]	128	#if SVC_EXTENSION
[815]	129	Bool m_isLastNALWasEos;
[978]	130	Bool m_lastPicHasEos;
[313]	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;
[1005]	137	UInt m_smallestLayerId;
[903]	138	Bool m_pocResettingFlag;
	139	Bool m_pocDecrementedInDPBFlag;
[313]	140	#if AVC_BASE
	141	fstream* m_pBLReconFile;
	142	#endif
[1148]	143
[313]	144	Int m_numDirectRefLayers;
[1047]	145	Int m_refLayerId[MAX_VPS_LAYER_IDX_PLUS1];
[313]	146	Int m_numSamplePredRefLayers;
	147	Int m_numMotionPredRefLayers;
[1148]	148
[442]	149	TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture
[540]	150	CommonDecoderParams* m_commonDecoderParams;
[713]	151	#if NO_CLRAS_OUTPUT_FLAG
[540]	152	Bool m_noClrasOutputFlag;
	153	Bool m_layerInitializedFlag;
	154	Bool m_firstPicInLayerDecodedFlag;
	155	#endif
[815]	156	Int m_parseIdc;
	157	Int m_lastPocPeriodId;
	158	Int m_prevPicOrderCnt;
[924]	159	#if CONFORMANCE_BITSTREAM_MODE
[1235]	160	Bool m_confModeFlag;
	161	std::vector<TComPic> m_confListPic; // Dynamic buffer for storing pictures for conformance purposes
[924]	162	#endif
[1235]	163	Bool m_isOutputLayerFlag;
[1111]	164	#endif //SVC_EXTENSION
	165
[313]	166	public:
[1111]	167	#if SVC_EXTENSION
[1235]	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;
[1111]	179	#endif //SVC_EXTENSION
[851]	180
[313]	181	TDecTop();
	182	virtual ~TDecTop();
[1029]	183
[313]	184	Void create ();
	185	Void destroy ();
	186
[1029]	187	Void setDecodedPictureHashSEIEnabled(Int enabled) { m_cGopDecoder.setDecodedPictureHashSEIEnabled(enabled); }
[313]	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
[1292]	198
[1029]	199	Void executeLoopFilters(Int& poc, TComList<TComPic>& rpcListPic);
	200	Void checkNoOutputPriorPics (TComList<TComPic> rpcListPic);
	201
	202	Bool getNoOutputPriorPicsFlag () { return m_isNoOutputPriorPics; }
[713]	203	Void setNoOutputPriorPicsFlag (Bool val) { m_isNoOutputPriorPics = val; }
[1029]	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; }
[713]	209	#endif
[1029]	210	Void setDecodedSEIMessageOutputStream(std::ostream *pOpStream) { m_pDecodedSEIOutputStream = pOpStream; }
[1292]	211	UInt getNumberOfChecksumErrorsDetected() const { return m_cGopDecoder.getNumberOfChecksumErrorsDetected(); }
[713]	212
[313]	213	#if SVC_EXTENSION
[1043]	214	Int getParseIdc () { return m_parseIdc; }
	215	Void setParseIdc (Int x) { m_parseIdc = x; }
[1235]	216	Void markAllPicsAsNoCurrAu ( const TComVPS *vps );
[815]	217
[1043]	218	Int getLastPocPeriodId () { return m_lastPocPeriodId; }
	219	Void setLastPocPeriodId (Int x) { m_lastPocPeriodId = x; }
[815]	220
[1043]	221	Int getPrevPicOrderCnt () { return m_prevPicOrderCnt; }
	222	Void setPrevPicOrderCnt (Int const x) { m_prevPicOrderCnt = x; }
[1210]	223
[1043]	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; }
[1049]	230	TDecTop* getLayerDec (UInt layerId) { return m_ppcTDecTop[layerId]; }
[1005]	231	Void xDeriveSmallestLayerId(TComVPS* vps);
[1148]	232
[1043]	233	TDecTop* getRefLayerDec (UInt refLayerIdx);
[313]	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
[1235]	246	Void setRefLayerParams( const TComVPS* vps );
[1148]	247
[313]	248	#if AVC_BASE
[1177]	249	Void setBLReconFile( fstream* pFile ) { m_pBLReconFile = pFile; }
	250	fstream* getBLReconFile() { return m_pBLReconFile; }
[313]	251	#endif
[442]	252	Void xInitILRP(TComSlice *slice);
[1177]	253	CommonDecoderParams* getCommonDecoderParams() { return m_commonDecoderParams; }
	254	Void setCommonDecoderParams(CommonDecoderParams* x) { m_commonDecoderParams = x; }
[595]	255	Void checkValueOfTargetOutputLayerSetIdx(TComVPS *vps);
[1223]	256
[1235]	257	ParameterSetManager* getParameterSetManager() { return &m_parameterSetManager; }
[1223]	258
[924]	259	#if CONFORMANCE_BITSTREAM_MODE
[1282]	260	std::vector<TComPic>* getConfListPic() { return &m_confListPic; }
	261	Bool getConfModeFlag() const { return m_confModeFlag; }
	262	Void setConfModeFlag(Bool x) { m_confModeFlag = x; }
[924]	263	#endif
[595]	264	#endif //SVC_EXTENSION
[820]	265
[313]	266	protected:
[1235]	267	#if SVC_EXTENSION
[1292]	268	Void xGetNewPicBuffer ( const TComVPS &vps, const TComSPS &sps, const TComPPS &pps, TComPic*& rpcPic, const UInt temporalLayer);
[1235]	269	#else
[1292]	270	Void xGetNewPicBuffer (const TComSPS &sps, const TComPPS &pps, TComPic*& rpcPic, const UInt temporalLayer);
[1235]	271	#endif
[1292]	272	Void xCreateLostPicture (Int iLostPOC);
[313]	273
	274	Void xActivateParameterSets();
	275	#if SVC_EXTENSION
	276	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC);
[1235]	277	Void xSetSpatialEnhLayerFlag( const TComVPS &vps, const TComSPS &sps, const TComPPS &pps, TComPic* pic );
[313]	278	#else
	279	Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay);
	280	#endif
[1319]	281	Void xDecodeVPS(const std::vector<UChar> &naluData);
	282	Void xDecodeSPS(const std::vector<UChar> &naluData);
[1212]	283	#if CGS_3D_ASYMLUT
[1319]	284	Void xDecodePPS(const std::vector<UChar> &naluData, TCom3DAsymLUT *pc3DAsymLUT);
[1145]	285	#else
[1319]	286	Void xDecodePPS(const std::vector<UChar> &naluData);
[713]	287	#endif
[313]	288	Void xDecodeSEI( TComInputBitstream* bs, const NalUnitType nalUnitType );
[1295]	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(); } }
[1319]	290	Void xParsePrefixSEImessages();
	291	Void xParsePrefixSEIsForUnknownVCLNal();
[313]	292
[1295]	293
[1131]	294	#if SVC_EXTENSION
[540]	295	#if NO_CLRAS_OUTPUT_FLAG
[1235]	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; }
[540]	302	#endif
[1212]	303	#if CGS_3D_ASYMLUT
[1235]	304	Void initAsymLut(TComSlice *pcSlice);
[644]	305	#endif
[1235]	306	Void resetPocRestrictionCheckParameters();
	307	Void xCheckLayerReset();
	308	Void xSetNoRaslOutputFlag();
	309	Void xSetLayerInitializedFlag();
[978]	310	#endif
[313]	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: