Context navigation

source: 3DVCSoftware/branches/0.1-poznan-univ/source/Lib/TLibEncoder/TEncCfg.h @ 47

Visit:

Last change on this file since 47 was 2, checked in by hhi, 13 years ago
inital import
Property svn:eol-style set to `native`
File size: 18.3 KB

Line
1
2	/** \file TEncCfg.h
3	\brief encoder configuration class (header)
4	*/
5
6	#ifndef __TENCCFG__
7	#define __TENCCFG__
8
9	#if _MSC_VER > 1000
10	#pragma once
11	#endif // _MSC_VER > 1000
12
13	#include "../TLibCommon/CommonDef.h"
14	#include "TEncSeqStructure.h"
15	#include <assert.h>
16
17	#include <vector>
18
19	// ====================================================================================================================
20	// Class definition
21	// ====================================================================================================================
22
23	/// encoder configuration class
24	class TEncCfg
25	{
26	protected:
27	//==== File I/O ========
28	Int m_iFrameRate;
29	Int m_FrameSkip;
30	Int m_iSourceWidth;
31	Int m_iSourceHeight;
32	Int m_iFrameToBeEncoded;
33
34	//====== Coding Structure ========
35	#if DCM_DECODING_REFRESH
36	UInt m_uiDecodingRefreshType; ///< the type of decoding refresh employed for the random access.
37	#endif
38
39	UInt m_uiCodedPictureStoreSize ;
40	Int m_iGOPSize;
41	Int m_iRateGOPSize;
42
43	// SB
44	std::string m_cInputFormatString ;
45
46	Int m_iQP; // if (AdaptiveQP == OFF)
47
48	Int m_aiTLayerQPOffset[MAX_TLAYER];
49	Int m_aiPad[2];
50
51	//======= Transform =============
52	UInt m_uiQuadtreeTULog2MaxSize;
53	UInt m_uiQuadtreeTULog2MinSize;
54	UInt m_uiQuadtreeTUMaxDepthInter;
55	UInt m_uiQuadtreeTUMaxDepthIntra;
56
57	//====== B Slice ========
58
59	//====== Entropy Coding ========
60	Int m_iSymbolMode; // (CAVLC, CABAC)
61
62	//====== Loop/Deblock Filter ========
63	Bool m_bLoopFilterDisable;
64	Int m_iLoopFilterAlphaC0Offset;
65	Int m_iLoopFilterBetaOffset;
66
67	#if MTK_SAO
68	Bool m_bUseSAO;
69	#endif
70
71	//====== Motion search ========
72	Int m_iFastSearch; // 0:Full search 1:Diamond 2:PMVFAST
73	Int m_iSearchRange; // 0:Full frame
74	Int m_bipredSearchRange;
75	Int m_iMaxDeltaQP; // Max. absolute delta QP (1:default)
76
77	//GT VSO
78	//====== View Synthesis Optimization ======
79	Bool m_bForceLambdaScale;
80	#if RDO_DIST_INT
81	Bool m_bAllowNegDist;
82	#endif
83	Double m_dLambdaScaleVSO;
84	UInt m_uiVSOMode;
85	//GT VSO end
86
87	//====== Tool list ========
88	Bool m_bUseSBACRD;
89	Bool m_bUseALF;
90	#if MQT_ALF_NPASS
91	Int m_iALFEncodePassReduction;
92	#endif
93	Bool m_bUseASR;
94	Bool m_bUseHADME;
95	#if DCM_COMB_LIST
96	Bool m_bUseLComb;
97	Bool m_bLCMod;
98	#endif
99	Bool m_bUseRDOQ;
100	#if !SB_NO_LowDelayCoding
101	Bool m_bUseLDC;
102	#endif
103	Bool m_bUsePAD;
104	Bool m_bUseFastEnc;
105
106	//GT VSO
107	Bool m_bUseVSO;
108	//GT VSO end
109
110	Bool m_bOmitUnusedBlocks;
111	Bool m_bUseMRG; // SOPH:
112	#if LM_CHROMA
113	Bool m_bUseLMChroma;
114	#endif
115
116	Int* m_aidQP;
117	UInt m_uiDeltaQpRD;
118
119	#if HHI_RMP_SWITCH
120	Bool m_bUseRMP;
121	#endif
122	#ifdef ROUNDING_CONTROL_BIPRED
123	Bool m_useRoundingControlBipred;
124	#endif
125	#if CONSTRAINED_INTRA_PRED
126	Bool m_bUseConstrainedIntraPred;
127	#endif
128	#if HHI_DMM_INTRA
129	Bool m_bUseDepthModelModes;
130	#endif
131	Bool m_bUseMVI;
132
133	//====== Slice ========
134	Int m_iSliceMode;
135	Int m_iSliceArgument;
136	//====== Entropy Slice ========
137	Int m_iEntropySliceMode;
138	Int m_iEntropySliceArgument;
139	#if MTK_NONCROSS_INLOOP_FILTER
140	Bool m_bLFCrossSliceBoundaryFlag;
141	#endif
142
143	bool m_pictureDigestEnabled; ///< enable(1)/disable(0) md5 computation and SEI signalling
144
145	#ifdef WEIGHT_PRED
146	//====== Weighted Prediction ========
147	Bool m_bUseWeightPred; // Use of Weighting Prediction (P_SLICE)
148	UInt m_uiBiPredIdc; // Use of Bi-Directional Weighting Prediction (B_SLICE)
149	#endif
150
151	TEncSeqStructure m_cSequenceStructure;
152	//std::vector<int> m_aiLayerQPOffset;
153
154	UInt m_uiViewId;
155	Int m_iViewOrderIdx;
156	Bool m_bIsDepth;
157	UInt m_uiCamParPrecision;
158	Bool m_bCamParInSliceHeader;
159	Int** m_aaiCodedScale;
160	Int** m_aaiCodedOffset;
161
162	UInt m_uiPredDepthMapGeneration;
163	UInt m_uiMultiviewMvPredMode;
164	UInt m_uiPdmPrecision;
165	Int** m_aaiPdmScaleNomDelta;
166	Int** m_aaiPdmOffset;
167	UInt m_uiMultiviewMvRegMode;
168	Double m_dMultiviewMvRegLambdaScale;
169	UInt m_uiMultiviewResPredMode;
170
171	PicOrderCnt m_iQpChangeFrame;
172	Int m_iQpChangeOffsetVideo;
173	Int m_iQpChangeOffsetDepth;
174	#if SB_INTERVIEW_SKIP
175	UInt m_uiInterViewSkip;
176	#if SB_INTERVIEW_SKIP_LAMBDA_SCALE
177	Double m_dInterViewSkipLambdaScale;
178	#endif
179	#endif
180
181	public:
182	TEncCfg() {}
183	virtual ~TEncCfg() {}
184
185	Void setFrameRate ( Int i ) { m_iFrameRate = i; }
186	Void setFrameSkip ( unsigned int i ) { m_FrameSkip = i; }
187	Void setSourceWidth ( Int i ) { m_iSourceWidth = i; }
188	Void setSourceHeight ( Int i ) { m_iSourceHeight = i; }
189	Void setFrameToBeEncoded ( Int i ) { m_iFrameToBeEncoded = i; }
190
191	//====== Coding Structure ========
192	#if DCM_DECODING_REFRESH
193	Void setDecodingRefreshType ( Int i ) { m_uiDecodingRefreshType = (UInt)i; }
194	#endif
195	Void setCPSSize ( Int i ) { m_uiCodedPictureStoreSize = (UInt) i ;}
196	Void setGOPSize ( Int i ) { m_iGOPSize = i; }
197	Void setRateGOPSize ( Int i ) { m_iRateGOPSize = i; }
198
199	Void setViewId ( UInt u ) { m_uiViewId = u; }
200	Void setViewOrderIdx ( Int i ) { m_iViewOrderIdx = i; }
201	Void setIsDepth ( Bool b ) { m_bIsDepth = b; }
202	Void setCamParPrecision ( UInt u ) { m_uiCamParPrecision = u; }
203	Void setCamParInSliceHeader ( Bool b ) { m_bCamParInSliceHeader = b; }
204	Void setCodedScale ( Int** p ) { m_aaiCodedScale = p; }
205	Void setCodedOffset ( Int** p ) { m_aaiCodedOffset = p; }
206
207	Void setPredDepthMapGeneration ( UInt u ) { m_uiPredDepthMapGeneration = u; }
208	Void setMultiviewMvPredMode ( UInt u ) { m_uiMultiviewMvPredMode = u; }
209	Void setPdmPrecision ( UInt u ) { m_uiPdmPrecision = u; }
210	Void setPdmScaleNomDelta ( Int** p ) { m_aaiPdmScaleNomDelta = p; }
211	Void setPdmOffset ( Int** p ) { m_aaiPdmOffset = p; }
212	Void setMultiviewMvRegMode ( UInt u ) { m_uiMultiviewMvRegMode = u; }
213	Void setMultiviewMvRegLambdaScale ( Double d) { m_dMultiviewMvRegLambdaScale = d; }
214	Void setMultiviewResPredMode ( UInt u ) { m_uiMultiviewResPredMode = u; }
215
216	#if SB_INTERVIEW_SKIP
217	Void setInterViewSkip ( UInt u ) { m_uiInterViewSkip = u; }
218	Bool getInterViewSkip ( ) { return bool(m_uiInterViewSkip) ;}
219	#if SB_INTERVIEW_SKIP_LAMBDA_SCALE
220	Void setInterViewSkipLambdaScale ( UInt u ) { m_dInterViewSkipLambdaScale = u; }
221	Double getInterViewSkipLambdaScale () { return m_dInterViewSkipLambdaScale; }
222	#endif
223	#endif
224
225	Bool isDepthCoder () { return m_bIsDepth; }
226
227	Void setSeqStructure( std::string s ) { m_cSequenceStructure.init( s ); }
228
229	Void setQP ( Int i ) { m_iQP = i; }
230
231	Void setTemporalLayerQPOffset ( Int* piTemporalLayerQPOffset ) { for ( Int i = 0; i < MAX_TLAYER; i++ ) m_aiTLayerQPOffset[i] = piTemporalLayerQPOffset[i]; }
232	Void setPad ( Int* iPad ) { for ( Int i = 0; i < 2; i++ ) m_aiPad[i] = iPad[i]; }
233
234	//======== Transform =============
235	Void setQuadtreeTULog2MaxSize ( UInt u ) { m_uiQuadtreeTULog2MaxSize = u; }
236	Void setQuadtreeTULog2MinSize ( UInt u ) { m_uiQuadtreeTULog2MinSize = u; }
237	Void setQuadtreeTUMaxDepthInter ( UInt u ) { m_uiQuadtreeTUMaxDepthInter = u; }
238	Void setQuadtreeTUMaxDepthIntra ( UInt u ) { m_uiQuadtreeTUMaxDepthIntra = u; }
239
240	//====== b; Slice ========
241
242	//====== Entropy Coding ========
243	Void setSymbolMode ( Int i ) { m_iSymbolMode = i; }
244
245	//====== Loop/Deblock Filter ========
246	Void setLoopFilterDisable ( Bool b ) { m_bLoopFilterDisable = b; }
247	Void setLoopFilterAlphaC0Offset ( Int i ) { m_iLoopFilterAlphaC0Offset = i; }
248	Void setLoopFilterBetaOffset ( Int i ) { m_iLoopFilterBetaOffset = i; }
249
250	//====== Motion search ========
251	Void setFastSearch ( Int i ) { m_iFastSearch = i; }
252	Void setSearchRange ( Int i ) { m_iSearchRange = i; }
253	Void setBipredSearchRange ( Int i ) { m_bipredSearchRange = i; }
254	Void setMaxDeltaQP ( Int i ) { m_iMaxDeltaQP = i; }
255
256	//GT VSO
257	//==== VSO ==========
258	Void setVSOMode ( UInt ui ) { m_uiVSOMode = ui; }
259	Void setForceLambdaScaleVSO ( Bool b ) { m_bForceLambdaScale = b; };
260	Void setLambdaScaleVSO ( Double d ) { m_dLambdaScaleVSO = d; };
261	#if RDO_DIST_INT
262	Void setAllowNegDist ( Bool b ) { m_bAllowNegDist = b; };
263	#endif
264	//GT VSO end
265
266	//====== Sequence ========
267	Int getFrameRate () { return m_iFrameRate; }
268	unsigned int getFrameSkip () { return m_FrameSkip; }
269	Int getSourceWidth () { return m_iSourceWidth; }
270	Int getSourceHeight () { return m_iSourceHeight; }
271	Int getFrameToBeEncoded () { return m_iFrameToBeEncoded; }
272
273	//==== Coding Structure ========
274	#if DCM_DECODING_REFRESH
275	UInt getDecodingRefreshType () { return m_uiDecodingRefreshType; }
276	#endif
277	UInt getCodedPictureBufferSize () { return m_uiCodedPictureStoreSize ;}
278
279	Int getGOPSize () { return m_iGOPSize; }
280	Int getRateGOPSize () { return m_iRateGOPSize; }
281
282	Int getQP () { return m_iQP; }
283
284	Int getTemporalLayerQPOffset ( Int i ) { assert (i < MAX_TLAYER ); return m_aiTLayerQPOffset[i]; }
285	Int getPad ( Int i ) { assert (i < 2 ); return m_aiPad[i]; }
286
287	//======== Transform =============
288	UInt getQuadtreeTULog2MaxSize () const { return m_uiQuadtreeTULog2MaxSize; }
289	UInt getQuadtreeTULog2MinSize () const { return m_uiQuadtreeTULog2MinSize; }
290	UInt getQuadtreeTUMaxDepthInter () const { return m_uiQuadtreeTUMaxDepthInter; }
291	UInt getQuadtreeTUMaxDepthIntra () const { return m_uiQuadtreeTUMaxDepthIntra; }
292
293	//==== b; Slice ========
294
295	//==== Entropy Coding ========
296	Int getSymbolMode () { return m_iSymbolMode; }
297
298	//==== Loop/Deblock Filter ========
299	Bool getLoopFilterDisable () { return m_bLoopFilterDisable; }
300	Int getLoopFilterAlphaC0Offget () { return m_iLoopFilterAlphaC0Offset; }
301	Int getLoopFilterBetaOffget () { return m_iLoopFilterBetaOffset; }
302
303	//==== Motion search ========
304	Int getFastSearch () { return m_iFastSearch; }
305	Int getSearchRange () { return m_iSearchRange; }
306	Int getMaxDeltaQP () { return m_iMaxDeltaQP; }
307
308	UInt getMultiviewMvRegMode () { return m_uiMultiviewMvRegMode; }
309	Double getMultiviewMvRegLambdaScale () { return m_dMultiviewMvRegLambdaScale; }
310
311	//GT VSO
312	//==== VSO ==========
313	UInt getVSOMode () { return m_uiVSOMode; }
314	Bool getForceLambdaScaleVSO () { return m_bForceLambdaScale; }
315	Double getLambdaScaleVSO () { return m_dLambdaScaleVSO; }
316	#if RDO_DIST_INT
317	Bool getAllowNegDist () { return m_bAllowNegDist; }
318	#endif
319
320	//GT VSO end
321
322	//==== Tool list ========
323	Void setUseSBACRD ( Bool b ) { m_bUseSBACRD = b; }
324	Void setUseASR ( Bool b ) { m_bUseASR = b; }
325	Void setUseHADME ( Bool b ) { m_bUseHADME = b; }
326	Void setUseALF ( Bool b ) { m_bUseALF = b; }
327	#if DCM_COMB_LIST
328	Void setUseLComb ( Bool b ) { m_bUseLComb = b; }
329	Void setLCMod ( Bool b ) { m_bLCMod = b; }
330	#endif
331	Void setUseRDOQ ( Bool b ) { m_bUseRDOQ = b; }
332	#if !SB_NO_LowDelayCoding
333	Void setUseLDC ( Bool b ) { m_bUseLDC = b; }
334	#endif
335	Void setUsePAD ( Bool b ) { m_bUsePAD = b; }
336	Void setUseFastEnc ( Bool b ) { m_bUseFastEnc = b; }
337	//GT VSO
338	Void setUseVSO ( Bool b ) { m_bUseVSO = b; }
339	//GT VSO end
340	Void setUseMRG ( Bool b ) { m_bUseMRG = b; } // SOPH:
341	#if CONSTRAINED_INTRA_PRED
342	Void setUseConstrainedIntraPred ( Bool b ) { m_bUseConstrainedIntraPred = b; }
343	#endif
344	Void setdQPs ( Int* p ) { m_aidQP = p; }
345	Void setDeltaQpRD ( UInt u ) {m_uiDeltaQpRD = u; }
346
347	Void setOmitUnusedBlocks ( Bool b ) { m_bOmitUnusedBlocks = b; }
348
349	Bool getUseSBACRD () { return m_bUseSBACRD; }
350	Bool getUseASR () { return m_bUseASR; }
351	Bool getUseHADME () { return m_bUseHADME; }
352	Bool getUseALF () { return m_bUseALF; }
353	#if MQT_ALF_NPASS
354	Void setALFEncodePassReduction (Int i) { m_iALFEncodePassReduction = i; }
355	Int getALFEncodePassReduction () { return m_iALFEncodePassReduction; }
356	#endif
357	#if DCM_COMB_LIST
358	Bool getUseLComb () { return m_bUseLComb; }
359	Bool getLCMod () { return m_bLCMod; }
360	#endif
361	Bool getUseRDOQ () { return m_bUseRDOQ; }
362	#if !SB_NO_LowDelayCoding
363	Bool getUseLDC () { return m_bUseLDC; }
364	#endif
365	Bool getUsePAD () { return m_bUsePAD; }
366	Bool getUseFastEnc () { return m_bUseFastEnc; }
367
368	//GT VSO
369	Bool getUseVSO () { return m_bUseVSO; }
370	//GT VSO end
371	Bool getUseMRG () { return m_bUseMRG; } // SOPH:
372	#if CONSTRAINED_INTRA_PRED
373	Bool getUseConstrainedIntraPred () { return m_bUseConstrainedIntraPred; }
374	#endif
375	#if HHI_DMM_INTRA
376	Void setUseDepthModelModes( Bool b) { m_bUseDepthModelModes = b; }
377	Bool getUseDepthModelModes() { return m_bUseDepthModelModes; }
378	#endif
379
380	Bool getOmitUnusedBlocks () { return m_bOmitUnusedBlocks; }
381
382	#if LM_CHROMA
383	Bool getUseLMChroma () { return m_bUseLMChroma; }
384	Void setUseLMChroma ( Bool b ) { m_bUseLMChroma = b; }
385	#endif
386
387	Int* getdQPs () { return m_aidQP; }
388	UInt getDeltaQpRD () { return m_uiDeltaQpRD; }
389	#if HHI_RMP_SWITCH
390	Void setUseRMP ( Bool b ) { m_bUseRMP = b; }
391	Bool getUseRMP () {return m_bUseRMP; }
392	#endif
393	#ifdef ROUNDING_CONTROL_BIPRED
394	Void setUseRoundingControlBipred(Bool b) { m_useRoundingControlBipred = b; }
395	Bool getUseRoundingControlBipred() { return m_useRoundingControlBipred; }
396	#endif
397	//====== Slice ========
398	Void setSliceMode ( Int i ) { m_iSliceMode = i; }
399	Void setSliceArgument ( Int i ) { m_iSliceArgument = i; }
400	Int getSliceMode () { return m_iSliceMode; }
401	Int getSliceArgument () { return m_iSliceArgument; }
402	//====== Entropy Slice ========
403	Void setEntropySliceMode ( Int i ) { m_iEntropySliceMode = i; }
404	Void setEntropySliceArgument ( Int i ) { m_iEntropySliceArgument = i; }
405	Int getEntropySliceMode () { return m_iEntropySliceMode; }
406	Int getEntropySliceArgument () { return m_iEntropySliceArgument;}
407	#if MTK_NONCROSS_INLOOP_FILTER
408	Void setLFCrossSliceBoundaryFlag ( Bool bValue ) { m_bLFCrossSliceBoundaryFlag = bValue; }
409	Bool getLFCrossSliceBoundaryFlag () { return m_bLFCrossSliceBoundaryFlag; }
410	#endif
411	#if MTK_SAO
412	Void setUseSAO (Bool bVal) {m_bUseSAO = bVal;}
413	Bool getUseSAO () {return m_bUseSAO;}
414	#endif
415	Void setUseMVI (Bool bVal) {m_bUseMVI = bVal;}
416
417	void setPictureDigestEnabled(bool b) { m_pictureDigestEnabled = b; }
418	bool getPictureDigestEnabled() { return m_pictureDigestEnabled; }
419
420	#ifdef WEIGHT_PRED
421	Void setUseWP ( Bool b ) { m_bUseWeightPred = b; }
422	Void setWPBiPredIdc ( UInt u ) { m_uiBiPredIdc = u; }
423	Bool getUseWP () { return m_bUseWeightPred; }
424	UInt getWPBiPredIdc () { return m_uiBiPredIdc; }
425	#endif
426
427	Void setQpChangeFrame( PicOrderCnt iPoc ) { m_iQpChangeFrame = iPoc; }
428	Int getQpChangeFrame() { return (Int)m_iQpChangeFrame; }
429	Void setQpChangeOffsetVideo( Int iOffset ) { m_iQpChangeOffsetVideo = iOffset; }
430	Int getQpChangeOffsetVideo() { return m_iQpChangeOffsetVideo; }
431	Void setQpChangeOffsetDepth( Int iOffset ) { m_iQpChangeOffsetDepth = iOffset; }
432	Int getQpChangeOffsetDepth() { return m_iQpChangeOffsetDepth; }
433	};
434
435	#endif // !defined(AFX_TENCCFG_H__6B99B797_F4DA_4E46_8E78_7656339A6C41__INCLUDED_)
436

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

Download in other formats: