source: 3DVCSoftware/branches/HTM-DEV-0.1-dev/source/App/TAppEncoder/TAppEncCfg.h @ 367

Last change on this file since 367 was 367, checked in by tech, 12 years ago

Further minor cleanups.

  • Property svn:eol-style set to native
File size: 22.9 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     TAppEncCfg.h
35    \brief    Handle encoder configuration parameters (header)
36*/
37
38#ifndef __TAPPENCCFG__
39#define __TAPPENCCFG__
40
41#include "TLibCommon/CommonDef.h"
42
43#include "TLibEncoder/TEncCfg.h"
44#include <sstream>
45//! \ingroup TAppEncoder
46//! \{
47
48// ====================================================================================================================
49// Class definition
50// ====================================================================================================================
51
52/// encoder configuration class
53class TAppEncCfg
54{
55protected:
56  // file I/O
57  Char*     m_pchInputFile;                                   ///< source file name
58  Char*     m_pchBitstreamFile;                               ///< output bitstream file
59  Char*     m_pchReconFile;                                   ///< output reconstruction file
60  // VPS specification
61  Double    m_adLambdaModifier[ MAX_TLAYER ];                 ///< Lambda modifier array for each temporal layer
62  // source specification
63  Int       m_iFrameRate;                                     ///< source frame-rates (Hz)
64  UInt      m_FrameSkip;                                      ///< number of skipped frames from the beginning
65  Int       m_iSourceWidth;                                   ///< source width in pixel
66  Int       m_iSourceHeight;                                  ///< source height in pixel
67  Int       m_conformanceMode;
68  Int       m_confLeft;
69  Int       m_confRight;
70  Int       m_confTop;
71  Int       m_confBottom;
72  Int       m_framesToBeEncoded;                              ///< number of encoded frames
73  Int       m_aiPad[2];                                       ///< number of padded pixels for width and height
74 
75  // profile/level
76  Profile::Name m_profile;
77  Level::Tier   m_levelTier;
78  Level::Name   m_level;
79#if L0046_CONSTRAINT_FLAGS
80  Bool m_progressiveSourceFlag;
81  Bool m_interlacedSourceFlag;
82  Bool m_nonPackedConstraintFlag;
83  Bool m_frameOnlyConstraintFlag;
84#endif
85 
86  // coding structure
87  Int       m_iIntraPeriod;                                   ///< period of I-slice (random access period)
88  Int       m_iDecodingRefreshType;                           ///< random access type
89  Int       m_iGOPSize;                                       ///< GOP size of hierarchical structure
90  Int       m_extraRPSs;                                      ///< extra RPSs added to handle CRA
91  GOPEntry  m_GOPList[MAX_GOP];                               ///< the coding structure entries from the config file
92  Int       m_numReorderPics[MAX_TLAYER];                     ///< total number of reorder pictures
93#if L0323_DPB
94  Int       m_maxDecPicBuffering[MAX_TLAYER];                 ///< total number of pictures in the decoded picture buffer
95#else
96  Int       m_maxDecPicBuffering[MAX_TLAYER];                 ///< total number of reference pictures needed for decoding
97#endif
98#if !L0034_COMBINED_LIST_CLEANUP
99  Bool      m_bUseLComb;                                      ///< flag for using combined reference list for uni-prediction in B-slices (JCTVC-D421)
100#endif
101  Bool      m_useTransformSkip;                               ///< flag for enabling intra transform skipping
102  Bool      m_useTransformSkipFast;                           ///< flag for enabling fast intra transform skipping
103  Bool      m_enableAMP;
104  // coding quality
105  Double    m_fQP;                                            ///< QP value of key-picture (floating point)
106  Int       m_iQP;                                            ///< QP value of key-picture (integer)
107  Char*     m_pchdQPFile;                                     ///< QP offset for each slice (initialized from external file)
108  Int*      m_aidQP;                                          ///< array of slice QP values
109  Int       m_iMaxDeltaQP;                                    ///< max. |delta QP|
110  UInt      m_uiDeltaQpRD;                                    ///< dQP range for multi-pass slice QP optimization
111  Int       m_iMaxCuDQPDepth;                                 ///< Max. depth for a minimum CuDQPSize (0:default)
112
113  Int       m_cbQpOffset;                                     ///< Chroma Cb QP Offset (0:default)
114  Int       m_crQpOffset;                                     ///< Chroma Cr QP Offset (0:default)
115
116#if ADAPTIVE_QP_SELECTION
117  Bool      m_bUseAdaptQpSelect;
118#endif
119
120  Bool      m_bUseAdaptiveQP;                                 ///< Flag for enabling QP adaptation based on a psycho-visual model
121  Int       m_iQPAdaptationRange;                             ///< dQP range by QP adaptation
122 
123  Int       m_maxTempLayer;                                  ///< Max temporal layer
124
125  // coding unit (CU) definition
126  UInt      m_uiMaxCUWidth;                                   ///< max. CU width in pixel
127  UInt      m_uiMaxCUHeight;                                  ///< max. CU height in pixel
128  UInt      m_uiMaxCUDepth;                                   ///< max. CU depth
129 
130  // transfom unit (TU) definition
131  UInt      m_uiQuadtreeTULog2MaxSize;
132  UInt      m_uiQuadtreeTULog2MinSize;
133 
134  UInt      m_uiQuadtreeTUMaxDepthInter;
135  UInt      m_uiQuadtreeTUMaxDepthIntra;
136 
137  // coding tools (bit-depth)
138  Int       m_inputBitDepthY;                               ///< bit-depth of input file (luma component)
139  Int       m_inputBitDepthC;                               ///< bit-depth of input file (chroma component)
140  Int       m_outputBitDepthY;                              ///< bit-depth of output file (luma component)
141  Int       m_outputBitDepthC;                              ///< bit-depth of output file (chroma component)
142  Int       m_internalBitDepthY;                            ///< bit-depth codec operates at in luma (input/output files will be converted)
143  Int       m_internalBitDepthC;                            ///< bit-depth codec operates at in chroma (input/output files will be converted)
144
145  // coding tools (PCM bit-depth)
146  Bool      m_bPCMInputBitDepthFlag;                          ///< 0: PCM bit-depth is internal bit-depth. 1: PCM bit-depth is input bit-depth.
147
148  // coding tool (lossless)
149  Bool      m_useLossless;                                    ///< flag for using lossless coding
150  Bool      m_bUseSAO; 
151  Int       m_maxNumOffsetsPerPic;                            ///< SAO maximun number of offset per picture
152  Bool      m_saoLcuBoundary;                                 ///< SAO parameter estimation using non-deblocked pixels for LCU bottom and right boundary areas
153  Bool      m_saoLcuBasedOptimization;                        ///< SAO LCU-based optimization
154  // coding tools (loop filter)
155  Bool      m_bLoopFilterDisable;                             ///< flag for using deblocking filter
156  Bool      m_loopFilterOffsetInPPS;                         ///< offset for deblocking filter in 0 = slice header, 1 = PPS
157  Int       m_loopFilterBetaOffsetDiv2;                     ///< beta offset for deblocking filter
158  Int       m_loopFilterTcOffsetDiv2;                       ///< tc offset for deblocking filter
159  Bool      m_DeblockingFilterControlPresent;                 ///< deblocking filter control present flag in PPS
160#if L0386_DB_METRIC
161  Bool      m_DeblockingFilterMetric;                         ///< blockiness metric in encoder
162#endif
163 
164  // coding tools (PCM)
165  Bool      m_usePCM;                                         ///< flag for using IPCM
166  UInt      m_pcmLog2MaxSize;                                 ///< log2 of maximum PCM block size
167  UInt      m_uiPCMLog2MinSize;                               ///< log2 of minimum PCM block size
168  Bool      m_bPCMFilterDisableFlag;                          ///< PCM filter disable flag
169
170  // coding tools (encoder-only parameters)
171  Bool      m_bUseSBACRD;                                     ///< flag for using RD optimization based on SBAC
172  Bool      m_bUseASR;                                        ///< flag for using adaptive motion search range
173  Bool      m_bUseHADME;                                      ///< flag for using HAD in sub-pel ME
174  Bool      m_useRDOQ;                                       ///< flag for using RD optimized quantization
175  Bool      m_useRDOQTS;                                     ///< flag for using RD optimized quantization for transform skip
176#if L0232_RD_PENALTY
177  Int      m_rdPenalty;                                      ///< RD-penalty for 32x32 TU for intra in non-intra slices (0: no RD-penalty, 1: RD-penalty, 2: maximum RD-penalty)
178#endif
179  Int       m_iFastSearch;                                    ///< ME mode, 0 = full, 1 = diamond, 2 = PMVFAST
180  Int       m_iSearchRange;                                   ///< ME search range
181  Int       m_bipredSearchRange;                              ///< ME search range for bipred refinement
182  Bool      m_bUseFastEnc;                                    ///< flag for using fast encoder setting
183  Bool      m_bUseEarlyCU;                                    ///< flag for using Early CU setting
184  Bool      m_useFastDecisionForMerge;                        ///< flag for using Fast Decision Merge RD-Cost
185  Bool      m_bUseCbfFastMode;                              ///< flag for using Cbf Fast PU Mode Decision
186  Bool      m_useEarlySkipDetection;                         ///< flag for using Early SKIP Detection
187  Int       m_sliceMode;                                     ///< 0: no slice limits, 1 : max number of CTBs per slice, 2: max number of bytes per slice,
188                                                             ///< 3: max number of tiles per slice
189  Int       m_sliceArgument;                                 ///< argument according to selected slice mode
190  Int       m_sliceSegmentMode;                              ///< 0: no slice segment limits, 1 : max number of CTBs per slice segment, 2: max number of bytes per slice segment,
191                                                             ///< 3: max number of tiles per slice segment
192  Int       m_sliceSegmentArgument;                          ///< argument according to selected slice segment mode
193
194  Bool      m_bLFCrossSliceBoundaryFlag;  ///< 1: filter across slice boundaries 0: do not filter across slice boundaries
195  Bool      m_bLFCrossTileBoundaryFlag;   ///< 1: filter across tile boundaries  0: do not filter across tile boundaries
196  Int       m_iUniformSpacingIdr;
197  Int       m_iNumColumnsMinus1;
198  Char*     m_pchColumnWidth;
199  Int       m_iNumRowsMinus1;
200  Char*     m_pchRowHeight;
201  UInt*     m_pColumnWidth;
202  UInt*     m_pRowHeight;
203  Int       m_iWaveFrontSynchro; //< 0: no WPP. >= 1: WPP is enabled, the "Top right" from which inheritance occurs is this LCU offset in the line above the current.
204  Int       m_iWaveFrontSubstreams; //< If iWaveFrontSynchro, this is the number of substreams per frame (dependent tiles) or per tile (independent tiles).
205
206  Bool      m_bUseConstrainedIntraPred;                       ///< flag for using constrained intra prediction
207 
208  Int       m_decodedPictureHashSEIEnabled;                    ///< Checksum(3)/CRC(2)/MD5(1)/disable(0) acting on decoded picture hash SEI message
209  Int       m_recoveryPointSEIEnabled;
210  Int       m_bufferingPeriodSEIEnabled;
211  Int       m_pictureTimingSEIEnabled;
212#if J0149_TONE_MAPPING_SEI
213  Bool      m_toneMappingInfoSEIEnabled;
214  Int       m_toneMapId;
215  Bool      m_toneMapCancelFlag;
216  Bool      m_toneMapPersistenceFlag;
217  Int       m_toneMapCodedDataBitDepth;
218  Int       m_toneMapTargetBitDepth;
219  Int       m_toneMapModelId; 
220  Int       m_toneMapMinValue;
221  Int       m_toneMapMaxValue;
222  Int       m_sigmoidMidpoint;
223  Int       m_sigmoidWidth;
224  Int       m_numPivots;
225  Int       m_cameraIsoSpeedIdc;
226  Int       m_cameraIsoSpeedValue;
227  Int       m_exposureCompensationValueSignFlag;
228  Int       m_exposureCompensationValueNumerator;
229  Int       m_exposureCompensationValueDenomIdc;
230  Int       m_refScreenLuminanceWhite;
231  Int       m_extendedRangeWhiteLevel;
232  Int       m_nominalBlackLevelLumaCodeValue;
233  Int       m_nominalWhiteLevelLumaCodeValue;
234  Int       m_extendedWhiteLevelLumaCodeValue;
235  Int*      m_startOfCodedInterval;
236  Int*      m_codedPivotValue;
237  Int*      m_targetPivotValue;
238#endif
239  Int       m_framePackingSEIEnabled;
240  Int       m_framePackingSEIType;
241  Int       m_framePackingSEIId;
242  Int       m_framePackingSEIQuincunx;
243  Int       m_framePackingSEIInterpretation;
244  Int       m_displayOrientationSEIAngle;
245  Int       m_temporalLevel0IndexSEIEnabled;
246  Int       m_gradualDecodingRefreshInfoEnabled;
247  Int       m_decodingUnitInfoSEIEnabled;
248#if L0208_SOP_DESCRIPTION_SEI
249  Int       m_SOPDescriptionSEIEnabled;
250#endif
251#if K0180_SCALABLE_NESTING_SEI
252  Int       m_scalableNestingSEIEnabled;
253#endif
254  // weighted prediction
255  Bool      m_useWeightedPred;                    ///< Use of weighted prediction in P slices
256  Bool      m_useWeightedBiPred;                  ///< Use of bi-directional weighted prediction in B slices
257 
258  UInt      m_log2ParallelMergeLevel;                         ///< Parallel merge estimation region
259  UInt      m_maxNumMergeCand;                                ///< Max number of merge candidates
260
261  Int       m_TMVPModeId;
262  Int       m_signHideFlag;
263#if RATE_CONTROL_LAMBDA_DOMAIN
264  Bool      m_RCEnableRateControl;                ///< enable rate control or not
265  Int       m_RCTargetBitrate;                    ///< target bitrate when rate control is enabled
266  Bool      m_RCKeepHierarchicalBit;              ///< whether keeping hierarchical bit allocation structure or not
267  Bool      m_RCLCULevelRC;                       ///< true: LCU level rate control; false: picture level rate control
268  Bool      m_RCUseLCUSeparateModel;              ///< use separate R-lambda model at LCU level
269  Int       m_RCInitialQP;                        ///< inital QP for rate control
270  Bool      m_RCForceIntraQP;                     ///< force all intra picture to use initial QP or not
271#else
272  Bool      m_enableRateCtrl;                                   ///< Flag for using rate control algorithm
273  Int       m_targetBitrate;                                 ///< target bitrate
274  Int       m_numLCUInUnit;                                  ///< Total number of LCUs in a frame should be completely divided by the NumLCUInUnit
275#endif
276  Int       m_useScalingListId;                               ///< using quantization matrix
277  Char*     m_scalingListFile;                                ///< quantization matrix file name
278
279  Bool      m_TransquantBypassEnableFlag;                     ///< transquant_bypass_enable_flag setting in PPS.
280  Bool      m_CUTransquantBypassFlagValue;                    ///< if transquant_bypass_enable_flag, the fixed value to use for the per-CU cu_transquant_bypass_flag.
281
282  Bool      m_recalculateQPAccordingToLambda;                 ///< recalculate QP value according to the lambda value
283  Bool      m_useStrongIntraSmoothing;                        ///< enable strong intra smoothing for 32x32 blocks where the reference samples are flat
284  Int       m_activeParameterSetsSEIEnabled;
285
286  Bool      m_vuiParametersPresentFlag;                       ///< enable generation of VUI parameters
287  Bool      m_aspectRatioInfoPresentFlag;                     ///< Signals whether aspect_ratio_idc is present
288  Int       m_aspectRatioIdc;                                 ///< aspect_ratio_idc
289  Int       m_sarWidth;                                       ///< horizontal size of the sample aspect ratio
290  Int       m_sarHeight;                                      ///< vertical size of the sample aspect ratio
291  Bool      m_overscanInfoPresentFlag;                        ///< Signals whether overscan_appropriate_flag is present
292  Bool      m_overscanAppropriateFlag;                        ///< Indicates whether conformant decoded pictures are suitable for display using overscan
293  Bool      m_videoSignalTypePresentFlag;                     ///< Signals whether video_format, video_full_range_flag, and colour_description_present_flag are present
294  Int       m_videoFormat;                                    ///< Indicates representation of pictures
295  Bool      m_videoFullRangeFlag;                             ///< Indicates the black level and range of luma and chroma signals
296  Bool      m_colourDescriptionPresentFlag;                   ///< Signals whether colour_primaries, transfer_characteristics and matrix_coefficients are present
297  Int       m_colourPrimaries;                                ///< Indicates chromaticity coordinates of the source primaries
298  Int       m_transferCharacteristics;                        ///< Indicates the opto-electronic transfer characteristics of the source
299  Int       m_matrixCoefficients;                             ///< Describes the matrix coefficients used in deriving luma and chroma from RGB primaries
300  Bool      m_chromaLocInfoPresentFlag;                       ///< Signals whether chroma_sample_loc_type_top_field and chroma_sample_loc_type_bottom_field are present
301  Int       m_chromaSampleLocTypeTopField;                    ///< Specifies the location of chroma samples for top field
302  Int       m_chromaSampleLocTypeBottomField;                 ///< Specifies the location of chroma samples for bottom field
303  Bool      m_neutralChromaIndicationFlag;                    ///< Indicates that the value of all decoded chroma samples is equal to 1<<(BitDepthCr-1)
304  Bool      m_defaultDisplayWindowFlag;                       ///< Indicates the presence of the default window parameters
305  Int       m_defDispWinLeftOffset;                           ///< Specifies the left offset from the conformance window of the default window
306  Int       m_defDispWinRightOffset;                          ///< Specifies the right offset from the conformance window of the default window
307  Int       m_defDispWinTopOffset;                            ///< Specifies the top offset from the conformance window of the default window
308  Int       m_defDispWinBottomOffset;                         ///< Specifies the bottom offset from the conformance window of the default window
309  Bool      m_frameFieldInfoPresentFlag;                      ///< Indicates that pic_struct values are present in picture timing SEI messages
310  Bool      m_pocProportionalToTimingFlag;                    ///< Indicates that the POC value is proportional to the output time w.r.t. first picture in CVS
311  Int       m_numTicksPocDiffOneMinus1;                       ///< Number of ticks minus 1 that for a POC difference of one
312  Bool      m_bitstreamRestrictionFlag;                       ///< Signals whether bitstream restriction parameters are present
313  Bool      m_tilesFixedStructureFlag;                        ///< Indicates that each active picture parameter set has the same values of the syntax elements related to tiles
314  Bool      m_motionVectorsOverPicBoundariesFlag;             ///< Indicates that no samples outside the picture boundaries are used for inter prediction
315  Int       m_minSpatialSegmentationIdc;                      ///< Indicates the maximum size of the spatial segments in the pictures in the coded video sequence
316  Int       m_maxBytesPerPicDenom;                            ///< Indicates a number of bytes not exceeded by the sum of the sizes of the VCL NAL units associated with any coded picture
317  Int       m_maxBitsPerMinCuDenom;                           ///< Indicates an upper bound for the number of bits of coding_unit() data
318  Int       m_log2MaxMvLengthHorizontal;                      ///< Indicate the maximum absolute value of a decoded horizontal MV component in quarter-pel luma units
319  Int       m_log2MaxMvLengthVertical;                        ///< Indicate the maximum absolute value of a decoded vertical MV component in quarter-pel luma units
320
321  // internal member functions
322  Void  xSetGlobal      ();                                   ///< set global variables
323  Void  xCheckParameter ();                                   ///< check validity of configuration values
324  Void  xPrintParameter ();                                   ///< print configuration values
325  Void  xPrintUsage     ();                                   ///< print usage
326#if SIGNAL_BITRATE_PICRATE_IN_VPS
327  Int       m_bitRatePicRateMaxTLayers;                       ///< Indicates max. number of sub-layers for which bit rate is signalled.
328  Bool*     m_bitRateInfoPresentFlag;                         ///< Indicates whether bit rate information is signalled
329  Bool*     m_picRateInfoPresentFlag;                         ///< Indicates whether pic rate information is signalled
330  Int*      m_avgBitRate;                                     ///< Indicates avg. bit rate information for various sub-layers
331  Int*      m_maxBitRate;                                     ///< Indicates max. bit rate information for various sub-layers
332  Int*      m_avgPicRate;                                     ///< Indicates avg. picture rate information for various sub-layers
333  Int*      m_constantPicRateIdc;                                ///< Indicates constant picture rate idc for various sub-layers
334#endif
335public:
336  TAppEncCfg();
337  virtual ~TAppEncCfg();
338 
339public:
340  Void  create    ();                                         ///< create option handling class
341  Void  destroy   ();                                         ///< destroy option handling class
342  Bool  parseCfg  ( Int argc, Char* argv[] );                 ///< parse configuration file to fill member variables
343 
344};// END CLASS DEFINITION TAppEncCfg
345
346//! \}
347
348#endif // __TAPPENCCFG__
349
Note: See TracBrowser for help on using the repository browser.