Context navigation

source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibEncoder/TEncGOP.h @ 1171

Visit:

Last change on this file since 1171 was 1131, checked in by seregin, 9 years ago
macro cleanup: R0071_IRAP_EOS_CROSS_LAYER_IMPACTS
Property svn:eol-style set to `native`
File size: 12.9 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	*
[595]	6	* Copyright (c) 2010-2014, 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 TEncGOP.h
	35	\brief GOP encoder class (header)
	36	*/
	37
	38	#ifndef __TENCGOP__
	39	#define __TENCGOP__
	40
	41	#include <list>
	42
	43	#include <stdlib.h>
	44
	45	#include "TLibCommon/TComList.h"
	46	#include "TLibCommon/TComPic.h"
	47	#include "TLibCommon/TComBitCounter.h"
	48	#include "TLibCommon/TComLoopFilter.h"
	49	#include "TLibCommon/AccessUnit.h"
	50	#include "TEncSampleAdaptiveOffset.h"
	51	#include "TEncSlice.h"
	52	#include "TEncEntropy.h"
	53	#include "TEncCavlc.h"
	54	#include "TEncSbac.h"
	55	#include "SEIwrite.h"
[713]	56	#if Q0048_CGS_3D_ASYMLUT
	57	#include "TEnc3DAsymLUT.h"
	58	#endif
[313]	59
	60	#include "TEncAnalyze.h"
	61	#include "TEncRateCtrl.h"
	62	#include <vector>
	63
	64	//! \ingroup TLibEncoder
	65	//! \{
	66
	67	class TEncTop;
	68
	69	// ====================================================================================================================
	70	// Class definition
	71	// ====================================================================================================================
	72
	73	class TEncGOP
	74	{
	75	private:
	76	// Data
	77	Bool m_bLongtermTestPictureHasBeenCoded;
	78	Bool m_bLongtermTestPictureHasBeenCoded2;
[1029]	79	UInt m_numLongTermRefPicSPS;
	80	UInt m_ltRefPicPocLsbSps[MAX_NUM_LONG_TERM_REF_PICS];
	81	Bool m_ltRefPicUsedByCurrPicFlag[MAX_NUM_LONG_TERM_REF_PICS];
[313]	82	Int m_iLastIDR;
	83	Int m_iGopSize;
	84	Int m_iNumPicCoded;
	85	Bool m_bFirst;
[713]	86	#if ALLOW_RECOVERY_POINT_AS_RAP
	87	Int m_iLastRecoveryPicPOC;
	88	#endif
[1029]	89
[1089]	90	#if Q0074_COLOUR_REMAPPING_SEI
	91	string m_colourRemapSEIFile;
	92	Int m_colourRemapSEIId;
	93	Bool m_colourRemapSEICancelFlag;
	94	Bool m_colourRemapSEIPersistenceFlag;
	95	Bool m_colourRemapSEIVideoSignalInfoPresentFlag;
	96	Bool m_colourRemapSEIFullRangeFlag;
	97	Int m_colourRemapSEIPrimaries;
	98	Int m_colourRemapSEITransferFunction;
	99	Int m_colourRemapSEIMatrixCoefficients;
	100	Int m_colourRemapSEIInputBitDepth;
	101	Int m_colourRemapSEIBitDepth;
	102	Int m_colourRemapSEIPreLutNumValMinus1[3];
	103	Int* m_colourRemapSEIPreLutCodedValue[3];
	104	Int* m_colourRemapSEIPreLutTargetValue[3];
	105	Bool m_colourRemapSEIMatrixPresentFlag;
	106	Int m_colourRemapSEILog2MatrixDenom;
	107	Int m_colourRemapSEICoeffs[3][3];
	108	Int m_colourRemapSEIPostLutNumValMinus1[3];
	109	Int* m_colourRemapSEIPostLutCodedValue[3];
	110	Int* m_colourRemapSEIPostLutTargetValue[3];
	111	#endif
[313]	112	// Access channel
	113	TEncTop* m_pcEncTop;
	114	TEncCfg* m_pcCfg;
	115	TEncSlice* m_pcSliceEncoder;
	116	TComList<TComPic> m_pcListPic;
[1029]	117
[313]	118	TEncEntropy* m_pcEntropyCoder;
	119	TEncCavlc* m_pcCavlcCoder;
	120	TEncSbac* m_pcSbacCoder;
	121	TEncBinCABAC* m_pcBinCABAC;
	122	TComLoopFilter* m_pcLoopFilter;
	123
	124	SEIWriter m_seiWriter;
[1029]	125
[313]	126	//--Adaptive Loop filter
	127	TEncSampleAdaptiveOffset* m_pcSAO;
	128	TEncRateCtrl* m_pcRateCtrl;
	129	// indicate sequence first
	130	Bool m_bSeqFirst;
[1029]	131
[313]	132	// clean decoding refresh
	133	Bool m_bRefreshPending;
	134	Int m_pocCRA;
[595]	135	NalUnitType m_associatedIRAPType;
	136	Int m_associatedIRAPPOC;
[313]	137
	138	std::vector<Int> m_vRVM_RP;
	139	UInt m_lastBPSEI;
	140	UInt m_totalCoded;
	141	UInt m_cpbRemovalDelay;
	142	UInt m_tl0Idx;
	143	UInt m_rapIdx;
	144	Bool m_activeParameterSetSEIPresentInAU;
	145	Bool m_bufferingPeriodSEIPresentInAU;
	146	Bool m_pictureTimingSEIPresentInAU;
	147	Bool m_nestedBufferingPeriodSEIPresentInAU;
	148	Bool m_nestedPictureTimingSEIPresentInAU;
[595]	149
[1029]	150	#if POC_RESET_IDC_ENCODER
	151	Int m_pocCraWithoutReset;
	152	Int m_associatedIrapPocBeforeReset;
	153	#endif
[595]	154	#if SVC_EXTENSION
	155	UInt m_layerId;
	156	TEncTop** m_ppcTEncTop;
	157	TEncSearch* m_pcPredSearch; ///< encoder search class
[713]	158	#if Q0048_CGS_3D_ASYMLUT
	159	TEnc3DAsymLUT m_Enc3DAsymLUTPicUpdate;
	160	TEnc3DAsymLUT m_Enc3DAsymLUTPPS;
	161	TComPicYuv* m_pColorMappedPic;
	162
	163	Int m_iTap;
	164	const Int (*m_phase_filter)[13];
	165	const Int (*m_phase_filter_luma)[13];
	166	const Int (*m_phase_filter_chroma)[13];
	167	Int m_iM, m_iN;
	168	static const Int m_phase_filter_0_t0[4][13];
	169	static const Int m_phase_filter_0_t1[4][13];
	170	static const Int m_phase_filter_0_t1_chroma[4][13];
	171	static const Int m_phase_filter_1[8][13];
	172	Int **m_temp;
[595]	173	#endif
[815]	174	#if POC_RESET_IDC_ENCODER
	175	Int m_lastPocPeriodId;
[713]	176	#endif
[978]	177	Bool m_noRaslOutputFlag;
	178	Bool m_prevPicHasEos;
[815]	179	#endif
[595]	180
[313]	181	public:
	182	TEncGOP();
	183	virtual ~TEncGOP();
	184
	185	#if SVC_EXTENSION
	186	Void create ( UInt layerId );
	187	#else
	188	Void create ();
	189	#endif
	190	Void destroy ();
[1029]	191
[313]	192	Void init ( TEncTop* pcTEncTop );
[1029]	193	#if SVC_EXTENSION
	194	Void compressGOP ( Int iPicIdInGOP, Int iPOCLast, Int iNumPicRcvd, TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRec,
	195	std::list<AccessUnit>& accessUnitsInGOP, Bool isField, Bool isTff, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE );
[313]	196	#else
[1029]	197	Void compressGOP ( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRec,
	198	std::list<AccessUnit>& accessUnitsInGOP, Bool isField, Bool isTff, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE );
[313]	199	#endif
[1029]	200	Void xAttachSliceDataToNalUnit (OutputNALUnit& rNalu, TComOutputBitstream* pcBitstreamRedirect);
[313]	201
[1029]	202
[313]	203	Int getGOPSize() { return m_iGopSize; }
[1029]	204
[313]	205	TComList<TComPic> getListPic() { return m_pcListPic; }
	206
[442]	207	#if !SVC_EXTENSION
[1029]	208	Void printOutSummary ( UInt uiNumAllPicCoded, Bool isField, const Bool printMSEBasedSNR, const Bool printSequenceMSE );
[442]	209	#endif
[1029]	210	Void preLoopFilterPicAll ( TComPic* pcPic, UInt64& ruiDist );
	211
[313]	212	TEncSlice* getSliceEncoder() { return m_pcSliceEncoder; }
[595]	213	NalUnitType getNalUnitType( Int pocCurr, Int lastIdr, Bool isField );
[313]	214	Void arrangeLongtermPicturesInRPS(TComSlice , TComList<TComPic>& );
[1029]	215
	216	#if POC_RESET_IDC_ENCODER
	217	Void determinePocResetIdc( Int const pocCurr, TComSlice *const slice);
	218	Int getIntraRefreshInterval() { return m_pcCfg->getIntraPeriod(); }
	219	Int getIntraRefreshType() { return m_pcCfg->getDecodingRefreshType(); }
	220	Int getLastPocPeriodId() { return m_lastPocPeriodId; }
	221	Void setLastPocPeriodId(Int x) { m_lastPocPeriodId = x; }
	222	Void updatePocValuesOfPics( Int const pocCurr, TComSlice *const slice);
	223	#endif
	224
[313]	225	protected:
	226	TEncRateCtrl* getRateCtrl() { return m_pcRateCtrl; }
	227
	228	protected:
[1029]	229
	230	Void xInitGOP ( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRecOut, Bool isField );
	231	Void xGetBuffer ( TComList<TComPic>& rcListPic, TComList<TComPicYuv>& rcListPicYuvRecOut, Int iNumPicRcvd, Int iTimeOffset, TComPic& rpcPic, TComPicYuv& rpcPicYuvRecOut, Int pocCurr, Bool isField );
	232
	233	Void xCalculateAddPSNR ( TComPic* pcPic, TComPicYuv* pcPicD, const AccessUnit&, Double dEncTime, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE );
	234	Void xCalculateInterlacedAddPSNR( TComPic* pcPicOrgFirstField, TComPic* pcPicOrgSecondField,
	235	TComPicYuv* pcPicRecFirstField, TComPicYuv* pcPicRecSecondField,
	236	const AccessUnit& accessUnit, Double dEncTime, const InputColourSpaceConversion snr_conversion, const Bool printFrameMSE );
	237
[313]	238	UInt64 xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1);
	239
	240	Double xCalculateRVM();
	241
[1029]	242	SEIActiveParameterSets* xCreateSEIActiveParameterSets (TComSPS *sps);
	243	SEIFramePacking* xCreateSEIFramePacking();
	244	SEISegmentedRectFramePacking* xCreateSEISegmentedRectFramePacking();
	245	SEIDisplayOrientation* xCreateSEIDisplayOrientation();
	246	SEIToneMappingInfo* xCreateSEIToneMappingInfo();
	247	SEITempMotionConstrainedTileSets* xCreateSEITempMotionConstrainedTileSets ();
	248	SEIKneeFunctionInfo* xCreateSEIKneeFunctionInfo();
	249	SEIChromaSamplingFilterHint* xCreateSEIChromaSamplingFilterHint(Bool bChromaLocInfoPresent, Int iHorFilterIndex, Int iVerFilterIdc);
[313]	250
	251	Void xCreateLeadingSEIMessages (/SEIMessages seiMessages,/ AccessUnit &accessUnit, TComSPS *sps);
	252	Int xGetFirstSeiLocation (AccessUnit &accessUnit);
	253	Void xResetNonNestedSEIPresentFlags()
	254	{
	255	m_activeParameterSetSEIPresentInAU = false;
	256	m_bufferingPeriodSEIPresentInAU = false;
	257	m_pictureTimingSEIPresentInAU = false;
	258	}
	259	Void xResetNestedSEIPresentFlags()
	260	{
	261	m_nestedBufferingPeriodSEIPresentInAU = false;
	262	m_nestedPictureTimingSEIPresentInAU = false;
	263	}
[595]	264	Void dblMetric( TComPic* pcPic, UInt uiNumSlices );
	265
[1029]	266	#if Q0074_COLOUR_REMAPPING_SEI
[1089]	267	Void setCRISEIFile( Char* pch ) { m_colourRemapSEIFile = pch; }
	268
	269	Void freeColourCRI();
	270	Int readingCRIparameters();
	271	Void xCheckParameter();
	272
[1029]	273	SEIColourRemappingInfo* xCreateSEIColourRemappingInfo();
	274	#endif
[595]	275	#if SVC_EXTENSION
	276	#if LAYERS_NOT_PRESENT_SEI
	277	SEILayersNotPresent* xCreateSEILayersNotPresent ();
	278	#endif
[442]	279	#if N0383_IL_CONSTRAINED_TILE_SETS_SEI
	280	Void xBuildTileSetsMap(TComPicSym* picSym);
[595]	281	SEIInterLayerConstrainedTileSets* xCreateSEIInterLayerConstrainedTileSets();
[442]	282	#endif
[644]	283	#if O0164_MULTI_LAYER_HRD
[894]	284	#if VPS_VUI_BSP_HRD_PARAMS
[896]	285	SEIScalableNesting* xCreateBspNestingSEI(TComSlice *pcSlice, Int olsIdx, Int partitioningSchemeIdx, Int bspIdx);
[894]	286	#else
[644]	287	SEIScalableNesting* xCreateBspNestingSEI(TComSlice *pcSlice);
	288	#endif
[894]	289	#endif
[1037]	290	#if P0123_ALPHA_CHANNEL_SEI
	291	SEIAlphaChannelInfo* xCreateSEIAlphaChannelInfo();
	292	#endif
[912]	293	#if Q0096_OVERLAY_SEI
	294	SEIOverlayInfo* xCreateSEIOverlayInfo();
	295	#endif
[713]	296	#if Q0048_CGS_3D_ASYMLUT
	297	Void xDetermin3DAsymLUT( TComSlice * pSlice , TComPic * pCurPic , UInt refLayerIdc , TEncCfg * pCfg , Bool bSignalPPS );
	298	Void downScalePic( TComPicYuv* pcYuvSrc, TComPicYuv* pcYuvDest);
	299	Void downScaleComponent2x2( const Pel* pSrc, Pel* pDest, const Int iSrcStride, const Int iDestStride, const Int iSrcWidth, const Int iSrcHeight, const Int inputBitDepth, const Int outputBitDepth );
	300	inline Short xClip( Short x , Int bitdepth );
	301	Void initDs(Int iWidth, Int iHeight, Int iType);
	302	Void filterImg(
	303	Pel *src,
	304	Int iSrcStride,
	305	Pel *dst,
	306	Int iDstStride,
	307	Int height1,
	308	Int width1,
	309	Int shift,
	310	Int plane);
	311
	312	Int get_mem2DintWithPad(Int ***array2D, Int dim0, Int dim1, Int iPadY, Int iPadX);
	313	Void free_mem2DintWithPad(Int **array2D, Int iPadY, Int iPadX);
	314	#endif
[978]	315	Void xCheckLayerReset(TComSlice *slice);
	316	Void xSetNoRaslOutputFlag(TComSlice *slice);
	317	Void xSetLayerInitializedFlag(TComSlice *slice);
[595]	318	#endif //SVC_EXTENSION
[313]	319	};// END CLASS DEFINITION TEncGOP
	320
	321	//! \}
	322
	323	#endif // __TENCGOP__
	324

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

Download in other formats: