Context navigation

source: 3DVCSoftware/trunk/source/Lib/TLibEncoder/TEncCfg.h @ 34

Visit:

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

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

Download in other formats: