Context navigation

source: 3DVCSoftware/trunk/source/Lib/TLibEncoder/TEncTop.h @ 168

Visit:

Last change on this file since 168 was 56, checked in by hschwarz, 13 years ago
updated trunk (move to HM6.1)
Property svn:eol-style set to `native`
File size: 12.5 KB

Rev	Line
[5]	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
[56]	4	* granted under this license.
[5]	5	*
[56]	6	* Copyright (c) 2010-2012, ITU/ISO/IEC
[5]	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.
[56]	17	* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
[5]	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	*/
[2]	33
	34	/** \file TEncTop.h
	35	\brief encoder class (header)
	36	*/
	37
	38	#ifndef __TENCTOP__
	39	#define __TENCTOP__
	40
	41	// Include files
[56]	42	#include "TLibCommon/TComList.h"
	43	#include "TLibCommon/TComPrediction.h"
	44	#include "TLibCommon/TComTrQuant.h"
	45	#include "TLibCommon/AccessUnit.h"
	46	#include "TLibCommon/TComDepthMapGenerator.h"
[2]	47	#include "../TLibCommon/TComResidualGenerator.h"
	48
[56]	49	#include "TLibVideoIO/TVideoIOYuv.h"
[2]	50
	51	#include "TEncCfg.h"
	52	#include "TEncGOP.h"
	53	#include "TEncSlice.h"
	54	#include "TEncEntropy.h"
	55	#include "TEncCavlc.h"
	56	#include "TEncSbac.h"
	57	#include "TEncSearch.h"
	58	#include "TEncAdaptiveLoopFilter.h"
[56]	59	#include "TEncSampleAdaptiveOffset.h"
	60	#include "TEncPreanalyzer.h"
[2]	61
[56]	62	//! \ingroup TLibEncoder
	63	//! \{
[2]	64
	65	// ====================================================================================================================
	66	// Class definition
	67	// ====================================================================================================================
	68
[56]	69	class TAppEncTop;
[2]	70
	71	/// encoder class
	72	class TEncTop : public TEncCfg
	73	{
	74	private:
	75	// picture
	76	Int m_iPOCLast; ///< time index (POC)
	77	Int m_iNumPicRcvd; ///< number of received pictures
	78	UInt m_uiNumAllPicCoded; ///< number of coded pictures
	79	TComList<TComPic*> m_cListPic; ///< dynamic list of pictures
[56]	80
[2]	81	// encoder search
	82	TEncSearch m_cSearch; ///< encoder search class
[56]	83	TEncEntropy* m_pcEntropyCoder; ///< entropy encoder
	84	TEncCavlc* m_pcCavlcCoder; ///< CAVLC encoder
[2]	85	// coding tool
	86	TComTrQuant m_cTrQuant; ///< transform & quantization class
	87	TComLoopFilter m_cLoopFilter; ///< deblocking filter class
[56]	88	TEncSampleAdaptiveOffset m_cEncSAO; ///< sample adaptive offset class
[2]	89	TEncAdaptiveLoopFilter m_cAdaptiveLoopFilter; ///< adaptive loop filter class
	90	TEncEntropy m_cEntropyCoder; ///< entropy encoder
	91	TEncCavlc m_cCavlcCoder; ///< CAVLC encoder
	92	TEncSbac m_cSbacCoder; ///< SBAC encoder
	93	TEncBinCABAC m_cBinCoderCABAC; ///< bin coder CABAC
[56]	94	TEncSbac* m_pcSbacCoders; ///< SBAC encoders (to encode substreams )
	95	TEncBinCABAC* m_pcBinCoderCABACs; ///< bin coders CABAC (one per substream)
	96
[2]	97	// processing unit
[56]	98	TEncGOP m_cGOPEncoder; ///< GOP encoder
[2]	99	TEncSlice m_cSliceEncoder; ///< slice encoder
	100	TEncCu m_cCuEncoder; ///< CU encoder
[56]	101
[5]	102	#if DEPTH_MAP_GENERATION
[2]	103	TComDepthMapGenerator m_cDepthMapGenerator; ///< depth map generator
[5]	104	#endif
	105	#if HHI_INTER_VIEW_RESIDUAL_PRED
[2]	106	TComResidualGenerator m_cResidualGenerator; ///< generator for residual pictures
[5]	107	#endif
[2]	108	// SPS
	109	TComSPS m_cSPS; ///< SPS
	110	TComPPS m_cPPS; ///< PPS
[56]	111	std::vector<TComAPS> m_vAPS; //!< APS container
	112	TComRPSList m_RPSList; ///< RPS
	113
[2]	114	// RD cost computation
	115	TComBitCounter m_cBitCounter; ///< bit counter for RD optimization
	116	TComRdCost m_cRdCost; ///< RD cost computation class
	117	TEncSbac*** m_pppcRDSbacCoder; ///< temporal storage for RD computation
	118	TEncSbac m_cRDGoOnSbacCoder; ///< going on SBAC model for RD stage
[56]	119	#if FAST_BIT_EST
	120	TEncBinCABACCounter*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
	121	TEncBinCABACCounter m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
	122	#else
[2]	123	TEncBinCABAC*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
	124	TEncBinCABAC m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
[56]	125	#endif
	126	Int m_iNumSubstreams; ///< # of top-level elements allocated.
	127	TComBitCounter* m_pcBitCounters; ///< bit counters for RD optimization per substream
	128	TComRdCost* m_pcRdCosts; ///< RD cost computation class per substream
	129	TEncSbac**** m_ppppcRDSbacCoders; ///< temporal storage for RD computation per substream
	130	TEncSbac* m_pcRDGoOnSbacCoders; ///< going on SBAC model for RD stage per substream
	131	TEncBinCABAC**** m_ppppcBinCodersCABAC; ///< temporal CABAC state storage for RD computation per substream
	132	TEncBinCABAC* m_pcRDGoOnBinCodersCABAC; ///< going on bin coder CABAC for RD stage per substream
[2]	133
[56]	134	// quality control
	135	TEncPreanalyzer m_cPreanalyzer; ///< image characteristics analyzer for TM5-step3-like adaptive QP
	136
	137	TComScalingList m_scalingList; ///< quantization matrix information
	138
[2]	139	std::vector<TEncTop> m_pacTEncTopList;
[56]	140	TAppEncTop* m_pcTAppEncTop;
[2]	141
[56]	142	TEncAnalyze m_cAnalyzeAll;
	143	TEncAnalyze m_cAnalyzeI;
	144	TEncAnalyze m_cAnalyzeP;
	145	TEncAnalyze m_cAnalyzeB;
[2]	146
	147	protected:
	148	Void xGetNewPicBuffer ( TComPic*& rpcPic ); ///< get picture buffer which will be processed
	149	Void xInitSPS (); ///< initialize SPS from encoder options
	150	Void xInitPPS (); ///< initialize PPS from encoder options
[56]	151
	152	Void xInitPPSforTiles ();
	153	Void xInitRPS (); ///< initialize PPS from encoder options
	154	Void xInitSPSforInterViewRefs();
[2]	155
	156	public:
	157	TEncTop();
	158	virtual ~TEncTop();
[56]	159
[2]	160	Void create ();
	161	Void destroy ();
	162	Void init ( TAppEncTop* pcTAppEncTop );
	163	Void deletePicBuffer ();
	164
[56]	165	UInt getFrameId (Int iGOPid) {
	166	if(m_iPOCLast == 0)
	167	{
	168	return(0 );
	169	}
	170	else
	171	{
	172	return m_iPOCLast -m_iNumPicRcvd+ getGOPEntry(iGOPid).m_POC ;
	173	}
	174	}
	175	#if HHI_INTERVIEW_SKIP \|\| HHI_INTER_VIEW_MOTION_PRED \|\| HHI_INTER_VIEW_RESIDUAL_PRED
[2]	176	Void deleteExtraPicBuffers ( Int iPoc );
	177	#endif
	178
[56]	179	Void compressMotion ( Int iPoc );
[2]	180
[56]	181	Void initNewPic(TComPicYuv* pcPicYuvOrg, TComPicYuv* pcOrgPdmDepth = 0);
[2]	182
[56]	183	Void createWPPCoders(Int iNumSubstreams);
	184
[2]	185	// -------------------------------------------------------------------------------------------------------------------
	186	// member access functions
	187	// -------------------------------------------------------------------------------------------------------------------
[56]	188
[2]	189	TComList<TComPic> getListPic () { return &m_cListPic; }
	190	TEncSearch* getPredSearch () { return &m_cSearch; }
[56]	191
[2]	192	TComTrQuant* getTrQuant () { return &m_cTrQuant; }
	193	TComLoopFilter* getLoopFilter () { return &m_cLoopFilter; }
	194	TEncAdaptiveLoopFilter* getAdaptiveLoopFilter () { return &m_cAdaptiveLoopFilter; }
[56]	195	TEncSampleAdaptiveOffset* getSAO () { return &m_cEncSAO; }
	196	TEncGOP* getGOPEncoder () { return &m_cGOPEncoder; }
[2]	197	TEncSlice* getSliceEncoder () { return &m_cSliceEncoder; }
	198	TEncCu* getCuEncoder () { return &m_cCuEncoder; }
	199	TEncEntropy* getEntropyCoder () { return &m_cEntropyCoder; }
	200	TEncCavlc* getCavlcCoder () { return &m_cCavlcCoder; }
	201	TEncSbac* getSbacCoder () { return &m_cSbacCoder; }
	202	TEncBinCABAC* getBinCABAC () { return &m_cBinCoderCABAC; }
[56]	203	TEncSbac* getSbacCoders () { return m_pcSbacCoders; }
	204	TEncBinCABAC* getBinCABACs () { return m_pcBinCoderCABACs; }
	205
[2]	206	TComBitCounter* getBitCounter () { return &m_cBitCounter; }
	207	TComRdCost* getRdCost () { return &m_cRdCost; }
	208	TEncSbac*** getRDSbacCoder () { return m_pppcRDSbacCoder; }
	209	TEncSbac* getRDGoOnSbacCoder () { return &m_cRDGoOnSbacCoder; }
[56]	210	TComBitCounter* getBitCounters () { return m_pcBitCounters; }
	211	TComRdCost* getRdCosts () { return m_pcRdCosts; }
	212	TEncSbac**** getRDSbacCoders () { return m_ppppcRDSbacCoders; }
	213	TEncSbac* getRDGoOnSbacCoders () { return m_pcRDGoOnSbacCoders; }
	214
	215	Void setTEncTopList ( std::vector<TEncTop> pacTEncTopList );
	216	TAppEncTop* getEncTop () { return m_pcTAppEncTop; }
	217	Int getNumAllPicCoded () { return m_uiNumAllPicCoded; }
	218	Void printOutSummary ( UInt uiNumAllPicCoded );
	219
	220	TComSPS* getSPS () { return &m_cSPS; }
	221	TComPPS* getPPS () { return &m_cPPS; }
	222	std::vector<TComAPS>& getAPS () { return m_vAPS; }
	223	TComRPSList* getRPSList () { return &m_RPSList; }
	224
	225	Void selectReferencePictureSet(TComSlice* slice, Int POCCurr, Int GOPid,TComList<TComPic*>& listPic );
	226	TComScalingList* getScalingList () { return &m_scalingList; }
	227
	228	TEncAnalyze* getAnalyzeAll () { return &m_cAnalyzeAll; }
	229	TEncAnalyze* getAnalyzeI () { return &m_cAnalyzeI; }
	230	TEncAnalyze* getAnalyzeP () { return &m_cAnalyzeP; }
	231	TEncAnalyze* getAnalyzeB () { return &m_cAnalyzeB; }
	232
[5]	233	#if DEPTH_MAP_GENERATION
[2]	234	TComDepthMapGenerator* getDepthMapGenerator () { return &m_cDepthMapGenerator; }
[5]	235	#endif
	236	#if HHI_INTER_VIEW_RESIDUAL_PRED
[2]	237	TComResidualGenerator* getResidualGenerator () { return &m_cResidualGenerator; }
[5]	238	#endif
[2]	239
	240	// -------------------------------------------------------------------------------------------------------------------
	241	// encoder function
	242	// -------------------------------------------------------------------------------------------------------------------
	243
	244	/// encode several number of pictures until end-of-sequence
[56]	245	Void encode( bool bEos, TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut,
	246	std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Int gopId );
	247	};
[2]	248
[56]	249	//! \}
[2]	250
	251	#endif // __TENCTOP__

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

Download in other formats: