Context navigation

source: SHVCSoftware/branches/SHM-5.0-dev/source/Lib/TLibEncoder/TEncTop.h @ 560

Visit:

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

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

Download in other formats: