Context navigation

source: SHVCSoftware/branches/SHM-6-dev/source/Lib/TLibEncoder/TEncTop.h @ 674

Visit:

Last change on this file since 674 was 652, checked in by seregin, 11 years ago
update to HM-14.0
Property svn:eol-style set to `native`
File size: 15.1 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
	4	* granted under this license.
	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 TEncTop.h
	35	\brief encoder class (header)
	36	*/
	37
	38	#ifndef __TENCTOP__
	39	#define __TENCTOP__
	40
	41	// Include files
	42	#include "TLibCommon/TComList.h"
	43	#include "TLibCommon/TComPrediction.h"
	44	#include "TLibCommon/TComTrQuant.h"
	45	#include "TLibCommon/AccessUnit.h"
	46
	47	#include "TLibVideoIO/TVideoIOYuv.h"
	48
	49	#include "TEncCfg.h"
	50	#include "TEncGOP.h"
	51	#include "TEncSlice.h"
	52	#include "TEncEntropy.h"
	53	#include "TEncCavlc.h"
	54	#include "TEncSbac.h"
	55	#include "TEncSearch.h"
	56	#include "TEncSampleAdaptiveOffset.h"
	57	#include "TEncPreanalyzer.h"
	58	#include "TEncRateCtrl.h"
	59	//! \ingroup TLibEncoder
	60	//! \{
	61
	62	// ====================================================================================================================
	63	// Class definition
	64	// ====================================================================================================================
	65
	66	/// encoder class
	67	class TEncTop : public TEncCfg
	68	{
	69	private:
	70	// picture
	71	Int m_iPOCLast; ///< time index (POC)
	72	Int m_iNumPicRcvd; ///< number of received pictures
	73	UInt m_uiNumAllPicCoded; ///< number of coded pictures
	74	TComList<TComPic*> m_cListPic; ///< dynamic list of pictures
[595]	75
[313]	76	// encoder search
	77	TEncSearch m_cSearch; ///< encoder search class
[540]	78	//TEncEntropy* m_pcEntropyCoder; ///< entropy encoder
[313]	79	TEncCavlc* m_pcCavlcCoder; ///< CAVLC encoder
	80	// coding tool
	81	TComTrQuant m_cTrQuant; ///< transform & quantization class
	82	TComLoopFilter m_cLoopFilter; ///< deblocking filter class
	83	TEncSampleAdaptiveOffset m_cEncSAO; ///< sample adaptive offset class
	84	TEncEntropy m_cEntropyCoder; ///< entropy encoder
	85	TEncCavlc m_cCavlcCoder; ///< CAVLC encoder
	86	TEncSbac m_cSbacCoder; ///< SBAC encoder
	87	TEncBinCABAC m_cBinCoderCABAC; ///< bin coder CABAC
	88	TEncSbac* m_pcSbacCoders; ///< SBAC encoders (to encode substreams )
	89	TEncBinCABAC* m_pcBinCoderCABACs; ///< bin coders CABAC (one per substream)
	90
	91	// processing unit
	92	TEncGOP m_cGOPEncoder; ///< GOP encoder
	93	TEncSlice m_cSliceEncoder; ///< slice encoder
	94	TEncCu m_cCuEncoder; ///< CU encoder
	95	// SPS
	96	TComSPS m_cSPS; ///< SPS
	97	TComPPS m_cPPS; ///< PPS
	98	// RD cost computation
	99	TComBitCounter m_cBitCounter; ///< bit counter for RD optimization
	100	TComRdCost m_cRdCost; ///< RD cost computation class
	101	TEncSbac*** m_pppcRDSbacCoder; ///< temporal storage for RD computation
	102	TEncSbac m_cRDGoOnSbacCoder; ///< going on SBAC model for RD stage
	103	#if FAST_BIT_EST
	104	TEncBinCABACCounter*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
	105	TEncBinCABACCounter m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
	106	#else
	107	TEncBinCABAC*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
	108	TEncBinCABAC m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
	109	#endif
	110	Int m_iNumSubstreams; ///< # of top-level elements allocated.
	111	TComBitCounter* m_pcBitCounters; ///< bit counters for RD optimization per substream
	112	TComRdCost* m_pcRdCosts; ///< RD cost computation class per substream
	113	TEncSbac**** m_ppppcRDSbacCoders; ///< temporal storage for RD computation per substream
	114	TEncSbac* m_pcRDGoOnSbacCoders; ///< going on SBAC model for RD stage per substream
	115	TEncBinCABAC**** m_ppppcBinCodersCABAC; ///< temporal CABAC state storage for RD computation per substream
	116	TEncBinCABAC* m_pcRDGoOnBinCodersCABAC; ///< going on bin coder CABAC for RD stage per substream
	117
	118	// quality control
	119	TEncPreanalyzer m_cPreanalyzer; ///< image characteristics analyzer for TM5-step3-like adaptive QP
	120
	121	TComScalingList m_scalingList; ///< quantization matrix information
	122	TEncRateCtrl m_cRateCtrl; ///< Rate control class
	123
	124	#if SVC_EXTENSION
[595]	125	static Int m_iSPSIdCnt; ///< next Id number for SPS
	126	static Int m_iPPSIdCnt; ///< next Id number for PPS
	127	#if AVC_SYNTAX
	128	fstream* m_pBLSyntaxFile;
	129	#endif
[313]	130	TEncTop** m_ppcTEncTop;
	131	TEncTop* getLayerEnc(UInt layer) { return m_ppcTEncTop[layer]; }
	132	TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture
	133	#if REF_IDX_MFM
	134	Bool m_bMFMEnabledFlag;
	135	#endif
	136	UInt m_numScaledRefLayerOffsets;
[540]	137	#if O0098_SCALED_REF_LAYER_ID
	138	UInt m_scaledRefLayerId[MAX_LAYERS];
	139	#endif
[313]	140	Window m_scaledRefLayerWindow[MAX_LAYERS];
[644]	141	#if P0312_VERT_PHASE_ADJ
	142	Bool m_vertPhasePositionEnableFlag[MAX_LAYERS];
	143	#endif
[442]	144	#if POC_RESET_FLAG
	145	Int m_pocAdjustmentValue;
	146	#endif
[540]	147	#if NO_CLRAS_OUTPUT_FLAG
	148	Bool m_noClrasOutputFlag;
	149	Bool m_layerInitializedFlag;
	150	Bool m_firstPicInLayerDecodedFlag;
	151	Bool m_noOutputOfPriorPicsFlags;
	152	#endif
[588]	153	#if O0194_WEIGHTED_PREDICTION_CGS
	154	Bool m_interLayerWeightedPredFlag;
	155	#endif
[494]	156	#endif //SVC_EXTENSION
[313]	157	protected:
	158	Void xGetNewPicBuffer ( TComPic*& rpcPic ); ///< get picture buffer which will be processed
	159	Void xInitSPS (); ///< initialize SPS from encoder options
	160	Void xInitPPS (); ///< initialize PPS from encoder options
	161
	162	Void xInitPPSforTiles ();
[442]	163	Void xInitRPS (Bool isFieldCoding); ///< initialize PPS from encoder options
	164	#if SVC_EXTENSION
[595]	165	Void xInitILRP();
[313]	166	#endif
	167	public:
	168	TEncTop();
	169	virtual ~TEncTop();
	170
	171	Void create ();
	172	Void destroy ();
[442]	173	Void init (Bool isFieldCoding);
[313]	174	Void deletePicBuffer ();
	175
	176	Void createWPPCoders(Int iNumSubstreams);
	177
	178	// -------------------------------------------------------------------------------------------------------------------
	179	// member access functions
	180	// -------------------------------------------------------------------------------------------------------------------
	181
	182	TComList<TComPic> getListPic () { return &m_cListPic; }
	183	TEncSearch* getPredSearch () { return &m_cSearch; }
	184
	185	TComTrQuant* getTrQuant () { return &m_cTrQuant; }
	186	TComLoopFilter* getLoopFilter () { return &m_cLoopFilter; }
	187	TEncSampleAdaptiveOffset* getSAO () { return &m_cEncSAO; }
	188	TEncGOP* getGOPEncoder () { return &m_cGOPEncoder; }
	189	TEncSlice* getSliceEncoder () { return &m_cSliceEncoder; }
	190	TEncCu* getCuEncoder () { return &m_cCuEncoder; }
	191	TEncEntropy* getEntropyCoder () { return &m_cEntropyCoder; }
	192	TEncCavlc* getCavlcCoder () { return &m_cCavlcCoder; }
	193	TEncSbac* getSbacCoder () { return &m_cSbacCoder; }
	194	TEncBinCABAC* getBinCABAC () { return &m_cBinCoderCABAC; }
	195	TEncSbac* getSbacCoders () { return m_pcSbacCoders; }
	196	TEncBinCABAC* getBinCABACs () { return m_pcBinCoderCABACs; }
	197
	198	TComBitCounter* getBitCounter () { return &m_cBitCounter; }
	199	TComRdCost* getRdCost () { return &m_cRdCost; }
	200	TEncSbac*** getRDSbacCoder () { return m_pppcRDSbacCoder; }
	201	TEncSbac* getRDGoOnSbacCoder () { return &m_cRDGoOnSbacCoder; }
	202	TComBitCounter* getBitCounters () { return m_pcBitCounters; }
	203	TComRdCost* getRdCosts () { return m_pcRdCosts; }
	204	TEncSbac**** getRDSbacCoders () { return m_ppppcRDSbacCoders; }
	205	TEncSbac* getRDGoOnSbacCoders () { return m_pcRDGoOnSbacCoders; }
	206	TEncRateCtrl* getRateCtrl () { return &m_cRateCtrl; }
	207	TComSPS* getSPS () { return &m_cSPS; }
	208	TComPPS* getPPS () { return &m_cPPS; }
	209	Void selectReferencePictureSet(TComSlice* slice, Int POCCurr, Int GOPid );
	210	Int getReferencePictureSetIdxForSOP(TComSlice* slice, Int POCCurr, Int GOPid );
	211	TComScalingList* getScalingList () { return &m_scalingList; }
[595]	212	// -------------------------------------------------------------------------------------------------------------------
	213	// encoder function
	214	// -------------------------------------------------------------------------------------------------------------------
	215
	216	/// encode several number of pictures until end-of-sequence
[313]	217	#if SVC_EXTENSION
[644]	218	Void setLayerEnc(TEncTop** p) { m_ppcTEncTop = p; }
	219	TEncTop** getLayerEnc() { return m_ppcTEncTop; }
	220	Int getPOCLast () { return m_iPOCLast; }
	221	Int getNumPicRcvd () { return m_iNumPicRcvd; }
	222	Void setNumPicRcvd ( Int num ) { m_iNumPicRcvd = num; }
	223	Void setNumScaledRefLayerOffsets(Int x) { m_numScaledRefLayerOffsets = x; }
	224	UInt getNumScaledRefLayerOffsets() { return m_numScaledRefLayerOffsets; }
[540]	225	#if O0098_SCALED_REF_LAYER_ID
[644]	226	Void setScaledRefLayerId(Int x, UInt id) { m_scaledRefLayerId[x] = id; }
	227	UInt getScaledRefLayerId(Int x) { return m_scaledRefLayerId[x]; }
	228	Window& getScaledRefLayerWindowForLayer(Int layerId);
[540]	229	#endif
[644]	230	Window& getScaledRefLayerWindow(Int x) { return m_scaledRefLayerWindow[x]; }
	231	#if P0312_VERT_PHASE_ADJ
	232	Void setVertPhasePositionEnableFlag(Int x, Bool b) { m_vertPhasePositionEnableFlag[x] = b; }
	233	UInt getVertPhasePositionEnableFlag(Int x) { return m_vertPhasePositionEnableFlag[x]; }
	234	#endif
	235
[313]	236	TComPic** getIlpList() { return m_cIlpPic; }
	237	#if REF_IDX_MFM
[644]	238	Void setMFMEnabledFlag (Bool flag) { m_bMFMEnabledFlag = flag; }
	239	Bool getMFMEnabledFlag() { return m_bMFMEnabledFlag; }
[313]	240	#endif
[588]	241	#if O0194_WEIGHTED_PREDICTION_CGS
	242	Void setInterLayerWeightedPredFlag(Bool flag) { m_interLayerWeightedPredFlag = flag; }
	243	Bool getInterLayerWeightedPredFlag() { return m_interLayerWeightedPredFlag; }
[313]	244	#endif
	245	#if AVC_SYNTAX
	246	Void setBLSyntaxFile( fstream* pFile ) { m_pBLSyntaxFile = pFile; }
	247	fstream* getBLSyntaxFile() { return m_pBLSyntaxFile; }
	248	#endif
[442]	249	Void encode( TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int iPicIdInGOP );
[313]	250	Void encodePrep( TComPicYuv* pcPicYuvOrg );
[442]	251	Void encode( TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int iPicIdInGOP, Bool isTff );
	252	Void encodePrep( TComPicYuv* pcPicYuvOrg, Bool isTff );
[313]	253	#if VPS_EXTN_DIRECT_REF_LAYERS
	254	TEncTop* getRefLayerEnc(UInt refLayerIdc);
	255	#endif
[442]	256	#if POC_RESET_FLAG
[644]	257	Int getPocAdjustmentValue() { return m_pocAdjustmentValue;}
	258	Void setPocAdjustmentValue(Int x) { m_pocAdjustmentValue = x; }
[442]	259	#endif
[540]	260	#if NO_CLRAS_OUTPUT_FLAG
	261	Int getNoClrasOutputFlag() { return m_noClrasOutputFlag;}
	262	Void setNoClrasOutputFlag(Bool x) { m_noClrasOutputFlag = x; }
	263	Int getLayerInitializedFlag() { return m_layerInitializedFlag;}
	264	Void setLayerInitializedFlag(Bool x) { m_layerInitializedFlag = x; }
	265	Int getFirstPicInLayerDecodedFlag() { return m_firstPicInLayerDecodedFlag;}
	266	Void setFirstPicInLayerDecodedFlag(Bool x) { m_firstPicInLayerDecodedFlag = x; }
	267	Int getNoOutputOfPriorPicsFlags() { return m_noOutputOfPriorPicsFlags;}
	268	Void setNoOutputOfPriorPicsFlags(Bool x) { m_noOutputOfPriorPicsFlags = x; }
	269	#endif
[442]	270	#else //SVC_EXTENSION
[313]	271	Void encode( Bool bEos, TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut,
[442]	272	std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded );
[313]	273
[442]	274	/// encode several number of pictures until end-of-sequence
[652]	275	Void encode( Bool bEos, TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut,
[442]	276	std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Bool isTff);
	277
	278	Void printSummary(Bool isField) { m_cGOPEncoder.printOutSummary (m_uiNumAllPicCoded, isField); }
	279	#endif //#if SVC_EXTENSION
[313]	280	};
	281
	282	//! \}
	283
	284	#endif // __TENCTOP__
[595]	285

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

Download in other formats: