Context navigation

source: SHVCSoftware/branches/HM-10.0-dev-SHM/source/App/TAppEncoder/TAppEncCfg.cpp @ 54

Visit:

Last change on this file since 54 was 54, checked in by seregin, 12 years ago
port simulcast
File size: 79.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 TAppEncCfg.cpp
35	\brief Handle encoder configuration parameters
36	*/
37
38	#include <stdlib.h>
39	#include <cassert>
40	#include <cstring>
41	#include <string>
42	#include "TLibCommon/TComRom.h"
43	#include "TAppEncCfg.h"
44
45	static istream& operator>>(istream &, Level::Name &);
46	static istream& operator>>(istream &, Level::Tier &);
47	static istream& operator>>(istream &, Profile::Name &);
48
49	#include "TAppCommon/program_options_lite.h"
50	#include "TLibEncoder/TEncRateCtrl.h"
51	#ifdef WIN32
52	#define strdup _strdup
53	#endif
54
55	using namespace std;
56	namespace po = df::program_options_lite;
57
58	//! \ingroup TAppEncoder
59	//! \{
60
61	// ====================================================================================================================
62	// Constructor / destructor / initialization / destroy
63	// ====================================================================================================================
64
65	#if SVC_EXTENSION
66	TAppEncCfg::TAppEncCfg()
67	: m_pBitstreamFile()
68	, m_pColumnWidth()
69	, m_pRowHeight()
70	, m_scalingListFile()
71	#if REF_IDX_FRAMEWORK
72	, m_elRapSliceBEnabled(0)
73	#endif
74	{
75	for(UInt layer=0; layer<MAX_LAYERS; layer++)
76	{
77	m_acLayerCfg[layer].setAppEncCfg(this);
78	}
79	}
80	#else
81	TAppEncCfg::TAppEncCfg()
82	: m_pchInputFile()
83	, m_pchBitstreamFile()
84	, m_pchReconFile()
85	, m_pchdQPFile()
86	, m_pColumnWidth()
87	, m_pRowHeight()
88	, m_scalingListFile()
89	{
90	m_aidQP = NULL;
91	}
92	#endif
93
94	TAppEncCfg::~TAppEncCfg()
95	{
96	#if !SVC_EXTENSION
97	if ( m_aidQP )
98	{
99	delete[] m_aidQP;
100	}
101	free(m_pchInputFile);
102	#endif
103	free(m_pBitstreamFile);
104	#if !SVC_EXTENSION
105	free(m_pchReconFile);
106	free(m_pchdQPFile);
107	#endif
108	free(m_pColumnWidth);
109	free(m_pRowHeight);
110	free(m_scalingListFile);
111	}
112
113	Void TAppEncCfg::create()
114	{
115	}
116
117	Void TAppEncCfg::destroy()
118	{
119	}
120
121	std::istringstream &operator>>(std::istringstream &in, GOPEntry &entry) //input
122	{
123	in>>entry.m_sliceType;
124	in>>entry.m_POC;
125	in>>entry.m_QPOffset;
126	in>>entry.m_QPFactor;
127	in>>entry.m_tcOffsetDiv2;
128	in>>entry.m_betaOffsetDiv2;
129	in>>entry.m_temporalId;
130	in>>entry.m_numRefPicsActive;
131	in>>entry.m_numRefPics;
132	for ( Int i = 0; i < entry.m_numRefPics; i++ )
133	{
134	in>>entry.m_referencePics[i];
135	}
136	in>>entry.m_interRPSPrediction;
137	#if AUTO_INTER_RPS
138	if (entry.m_interRPSPrediction==1)
139	{
140	in>>entry.m_deltaRPS;
141	in>>entry.m_numRefIdc;
142	for ( Int i = 0; i < entry.m_numRefIdc; i++ )
143	{
144	in>>entry.m_refIdc[i];
145	}
146	}
147	else if (entry.m_interRPSPrediction==2)
148	{
149	in>>entry.m_deltaRPS;
150	}
151	#else
152	if (entry.m_interRPSPrediction)
153	{
154	in>>entry.m_deltaRPS;
155	in>>entry.m_numRefIdc;
156	for ( Int i = 0; i < entry.m_numRefIdc; i++ )
157	{
158	in>>entry.m_refIdc[i];
159	}
160	}
161	#endif
162	return in;
163	}
164
165	#if SVC_EXTENSION
166	void TAppEncCfg::getDirFilename(string& filename, string& dir, const string path)
167	{
168	size_t pos = path.find_last_of("\\");
169	if(pos != std::string::npos)
170	{
171	filename.assign(path.begin() + pos + 1, path.end());
172	dir.assign(path.begin(), path.begin() + pos + 1);
173	}
174	else
175	{
176	pos = path.find_last_of("/");
177	if(pos != std::string::npos)
178	{
179	filename.assign(path.begin() + pos + 1, path.end());
180	dir.assign(path.begin(), path.begin() + pos + 1);
181	}
182	else
183	{
184	filename = path;
185	dir.assign("");
186	}
187	}
188	}
189	#endif
190
191	static const struct MapStrToProfile {
192	const Char* str;
193	Profile::Name value;
194	} strToProfile[] = {
195	{"none", Profile::NONE},
196	{"main", Profile::MAIN},
197	{"main10", Profile::MAIN10},
198	{"main-still-picture", Profile::MAINSTILLPICTURE},
199	};
200
201	static const struct MapStrToTier {
202	const Char* str;
203	Level::Tier value;
204	} strToTier[] = {
205	{"main", Level::MAIN},
206	{"high", Level::HIGH},
207	};
208
209	static const struct MapStrToLevel {
210	const Char* str;
211	Level::Name value;
212	} strToLevel[] = {
213	{"none",Level::NONE},
214	{"1", Level::LEVEL1},
215	{"2", Level::LEVEL2},
216	{"2.1", Level::LEVEL2_1},
217	{"3", Level::LEVEL3},
218	{"3.1", Level::LEVEL3_1},
219	{"4", Level::LEVEL4},
220	{"4.1", Level::LEVEL4_1},
221	{"5", Level::LEVEL5},
222	{"5.1", Level::LEVEL5_1},
223	{"5.2", Level::LEVEL5_2},
224	{"6", Level::LEVEL6},
225	{"6.1", Level::LEVEL6_1},
226	{"6.2", Level::LEVEL6_2},
227	};
228
229	template<typename T, typename P>
230	static istream& readStrToEnum(P map[], unsigned long mapLen, istream &in, T &val)
231	{
232	string str;
233	in >> str;
234
235	for (Int i = 0; i < mapLen; i++)
236	{
237	if (str == map[i].str)
238	{
239	val = map[i].value;
240	goto found;
241	}
242	}
243	/* not found */
244	in.setstate(ios::failbit);
245	found:
246	return in;
247	}
248
249	static istream& operator>>(istream &in, Profile::Name &profile)
250	{
251	return readStrToEnum(strToProfile, sizeof(strToProfile)/sizeof(*strToProfile), in, profile);
252	}
253
254	static istream& operator>>(istream &in, Level::Tier &tier)
255	{
256	return readStrToEnum(strToTier, sizeof(strToTier)/sizeof(*strToTier), in, tier);
257	}
258
259	static istream& operator>>(istream &in, Level::Name &level)
260	{
261	return readStrToEnum(strToLevel, sizeof(strToLevel)/sizeof(*strToLevel), in, level);
262	}
263
264	#if SIGNAL_BITRATE_PICRATE_IN_VPS
265	Void readBoolString(const string inpString, const Int numEntries, Bool* &memberArray, const char *elementName);
266	Void readIntString(const string inpString, const Int numEntries, Int* &memberArray, const char *elementName);
267	#endif
268	// ====================================================================================================================
269	// Public member functions
270	// ====================================================================================================================
271
272	/** \param argc number of arguments
273	\param argv array of arguments
274	\retval true when success
275	*/
276	Bool TAppEncCfg::parseCfg( Int argc, Char* argv[] )
277	{
278	Bool do_help = false;
279
280	#if SVC_EXTENSION
281	string cfg_LayerCfgFile [MAX_LAYERS];
282	string cfg_BitstreamFile;
283	string* cfg_InputFile [MAX_LAYERS];
284	string* cfg_ReconFile [MAX_LAYERS];
285	Double* cfg_fQP [MAX_LAYERS];
286
287	Int* cfg_SourceWidth [MAX_LAYERS];
288	Int* cfg_SourceHeight [MAX_LAYERS];
289	Int* cfg_FrameRate [MAX_LAYERS];
290	Int* cfg_IntraPeriod [MAX_LAYERS];
291	Int* cfg_conformanceMode [MAX_LAYERS];
292	for(UInt layer = 0; layer < MAX_LAYERS; layer++)
293	{
294	cfg_InputFile[layer] = &m_acLayerCfg[layer].m_cInputFile;
295	cfg_ReconFile[layer] = &m_acLayerCfg[layer].m_cReconFile;
296	cfg_fQP[layer] = &m_acLayerCfg[layer].m_fQP;
297	cfg_SourceWidth[layer] = &m_acLayerCfg[layer].m_iSourceWidth;
298	cfg_SourceHeight[layer] = &m_acLayerCfg[layer].m_iSourceHeight;
299	cfg_FrameRate[layer] = &m_acLayerCfg[layer].m_iFrameRate;
300	cfg_IntraPeriod[layer] = &m_acLayerCfg[layer].m_iIntraPeriod;
301	cfg_conformanceMode[layer] = &m_acLayerCfg[layer].m_conformanceMode;
302	}
303	#if AVC_SYNTAX
304	string cfg_BLSyntaxFile;
305	#endif
306	#else
307	string cfg_InputFile;
308	string cfg_BitstreamFile;
309	string cfg_ReconFile;
310	string cfg_dQPFile;
311	#endif
312	string cfg_ColumnWidth;
313	string cfg_RowHeight;
314	string cfg_ScalingListFile;
315	#if SIGNAL_BITRATE_PICRATE_IN_VPS
316	string cfg_bitRateInfoPresentFlag;
317	string cfg_picRateInfoPresentFlag;
318	string cfg_avgBitRate;
319	string cfg_maxBitRate;
320	string cfg_avgPicRate;
321	string cfg_constantPicRateIdc;
322	#endif
323	po::Options opts;
324	opts.addOptions()
325	("help", do_help, false, "this help text")
326	("c", po::parseConfigFile, "configuration file name")
327
328	// File, I/O and source parameters
329	#if SVC_EXTENSION
330	("InputFile%d,-i%d", cfg_InputFile, string(""), MAX_LAYERS, "original YUV input file name for layer %d")
331	("ReconFile%d,-o%d", cfg_ReconFile, string(""), MAX_LAYERS, "reconstruction YUV input file name for layer %d")
332	("LayerConfig%d,-lc%d", cfg_LayerCfgFile, string(""), MAX_LAYERS, "layer %d configuration file name")
333	("SourceWidth%d,-wdt%d", cfg_SourceWidth, 0, MAX_LAYERS, "Source picture width for layer %d")
334	("SourceHeight%d,-hgt%d", cfg_SourceHeight, 0, MAX_LAYERS, "Source picture height for layer %d")
335	("FrameRate%d,-fr%d", cfg_FrameRate, 0, MAX_LAYERS, "Frame rate for layer %d")
336	("LambdaModifier%d,-LM%d", m_adLambdaModifier, ( double )1.0, MAX_TLAYER, "Lambda modifier for temporal layer %d")
337	("NumLayers", m_numLayers, 1, "Number of layers to code")
338	("ConformanceMode%d", cfg_conformanceMode,0, MAX_LAYERS, "Window conformance mode (0: no cropping, 1:automatic padding, 2: padding, 3:cropping")
339
340	("BitstreamFile,b", cfg_BitstreamFile, string(""), "Bitstream output file name")
341	("InputBitDepth", m_inputBitDepthY, 8, "Bit-depth of input file")
342	("OutputBitDepth", m_outputBitDepthY, 0, "Bit-depth of output file (default:InternalBitDepth)")
343	("InternalBitDepth", m_internalBitDepthY, 0, "Bit-depth the codec operates at. (default:InputBitDepth)"
344	"If different to InputBitDepth, source data will be converted")
345	("InputBitDepthC", m_inputBitDepthC, 0, "As per InputBitDepth but for chroma component. (default:InputBitDepth)")
346	("OutputBitDepthC", m_outputBitDepthC, 0, "As per OutputBitDepth but for chroma component. (default:InternalBitDepthC)")
347	("InternalBitDepthC", m_internalBitDepthC, 0, "As per InternalBitDepth but for chroma component. (default:IntrenalBitDepth)")
348
349	#if AVC_BASE
350	("InputBLFile,-ibl", *cfg_InputFile[0], string(""), "Base layer rec YUV input file name")
351	#if AVC_SYNTAX
352	("InputBLSyntaxFile,-ibs", cfg_BLSyntaxFile, string(""), "Base layer syntax input file name")
353	#endif
354	#endif
355	#if REF_IDX_FRAMEWORK
356	("EnableElRapB,-use-rap-b", m_elRapSliceBEnabled, 0, "Set ILP over base-layer I picture to B picture (default is P picture_")
357	#endif
358	#else
359	("InputFile,i", cfg_InputFile, string(""), "Original YUV input file name")
360	("BitstreamFile,b", cfg_BitstreamFile, string(""), "Bitstream output file name")
361	("ReconFile,o", cfg_ReconFile, string(""), "Reconstructed YUV output file name")
362	("SourceWidth,-wdt", m_iSourceWidth, 0, "Source picture width")
363	("SourceHeight,-hgt", m_iSourceHeight, 0, "Source picture height")
364	("InputBitDepth", m_inputBitDepthY, 8, "Bit-depth of input file")
365	("OutputBitDepth", m_outputBitDepthY, 0, "Bit-depth of output file (default:InternalBitDepth)")
366	("InternalBitDepth", m_internalBitDepthY, 0, "Bit-depth the codec operates at. (default:InputBitDepth)"
367	"If different to InputBitDepth, source data will be converted")
368	("InputBitDepthC", m_inputBitDepthC, 0, "As per InputBitDepth but for chroma component. (default:InputBitDepth)")
369	("OutputBitDepthC", m_outputBitDepthC, 0, "As per OutputBitDepth but for chroma component. (default:InternalBitDepthC)")
370	("InternalBitDepthC", m_internalBitDepthC, 0, "As per InternalBitDepth but for chroma component. (default:IntrenalBitDepth)")
371	("ConformanceMode", m_conformanceMode, 0, "Window conformance mode (0: no window, 1:automatic padding, 2:padding, 3:conformance")
372	("HorizontalPadding,-pdx",m_aiPad[0], 0, "Horizontal source padding for conformance window mode 2")
373	("VerticalPadding,-pdy", m_aiPad[1], 0, "Vertical source padding for conformance window mode 2")
374	("ConfLeft", m_confLeft, 0, "Left offset for window conformance mode 3")
375	("ConfRight", m_confRight, 0, "Right offset for window conformance mode 3")
376	("ConfTop", m_confTop, 0, "Top offset for window conformance mode 3")
377	("ConfBottom", m_confBottom, 0, "Bottom offset for window conformance mode 3")
378	("FrameRate,-fr", m_iFrameRate, 0, "Frame rate")
379	#endif
380	("FrameSkip,-fs", m_FrameSkip, 0u, "Number of frames to skip at start of input YUV")
381	("FramesToBeEncoded,f", m_framesToBeEncoded, 0, "Number of frames to be encoded (default=all)")
382
383	// Profile and level
384	("Profile", m_profile, Profile::NONE, "Profile to be used when encoding (Incomplete)")
385	("Level", m_level, Level::NONE, "Level limit to be used, eg 5.1 (Incomplete)")
386	("Tier", m_levelTier, Level::MAIN, "Tier to use for interpretation of --Level")
387
388	#if L0046_CONSTRAINT_FLAGS
389	("ProgressiveSource", m_progressiveSourceFlag, false, "Indicate that source is progressive")
390	("InterlacedSource", m_interlacedSourceFlag, false, "Indicate that source is interlaced")
391	("NonPackedSource", m_nonPackedConstraintFlag, false, "Indicate that source does not contain frame packing")
392	("FrameOnly", m_frameOnlyConstraintFlag, false, "Indicate that the bitstream contains only frames")
393	#endif
394
395	// Unit definition parameters
396	("MaxCUWidth", m_uiMaxCUWidth, 64u)
397	("MaxCUHeight", m_uiMaxCUHeight, 64u)
398	// todo: remove defaults from MaxCUSize
399	("MaxCUSize,s", m_uiMaxCUWidth, 64u, "Maximum CU size")
400	("MaxCUSize,s", m_uiMaxCUHeight, 64u, "Maximum CU size")
401	("MaxPartitionDepth,h", m_uiMaxCUDepth, 4u, "CU depth")
402
403	("QuadtreeTULog2MaxSize", m_uiQuadtreeTULog2MaxSize, 6u, "Maximum TU size in logarithm base 2")
404	("QuadtreeTULog2MinSize", m_uiQuadtreeTULog2MinSize, 2u, "Minimum TU size in logarithm base 2")
405
406	("QuadtreeTUMaxDepthIntra", m_uiQuadtreeTUMaxDepthIntra, 1u, "Depth of TU tree for intra CUs")
407	("QuadtreeTUMaxDepthInter", m_uiQuadtreeTUMaxDepthInter, 2u, "Depth of TU tree for inter CUs")
408
409	// Coding structure paramters
410	#if SVC_EXTENSION
411	("IntraPeriod%d,-ip%d", cfg_IntraPeriod, -1, MAX_LAYERS, "intra period in frames for layer %d, (-1: only first frame)")
412	#else
413	("IntraPeriod,-ip", m_iIntraPeriod, -1, "Intra period in frames, (-1: only first frame)")
414	#endif
415	("DecodingRefreshType,-dr", m_iDecodingRefreshType, 0, "Intra refresh type (0:none 1:CRA 2:IDR)")
416	("GOPSize,g", m_iGOPSize, 1, "GOP size of temporal structure")
417	("ListCombination,-lc", m_bUseLComb, true, "Combined reference list for uni-prediction estimation in B-slices")
418	// motion options
419	("FastSearch", m_iFastSearch, 1, "0:Full search 1:Diamond 2:PMVFAST")
420	("SearchRange,-sr", m_iSearchRange, 96, "Motion search range")
421	("BipredSearchRange", m_bipredSearchRange, 4, "Motion search range for bipred refinement")
422	("HadamardME", m_bUseHADME, true, "Hadamard ME for fractional-pel")
423	("ASR", m_bUseASR, false, "Adaptive motion search range")
424
425	#if SVC_EXTENSION
426	("LambdaModifier%d,-LM%d", m_adLambdaModifier, ( double )1.0, MAX_TLAYER, "Lambda modifier for temporal layer %d")
427	#else
428	// Mode decision parameters
429	("LambdaModifier0,-LM0", m_adLambdaModifier[ 0 ], ( Double )1.0, "Lambda modifier for temporal layer 0")
430	("LambdaModifier1,-LM1", m_adLambdaModifier[ 1 ], ( Double )1.0, "Lambda modifier for temporal layer 1")
431	("LambdaModifier2,-LM2", m_adLambdaModifier[ 2 ], ( Double )1.0, "Lambda modifier for temporal layer 2")
432	("LambdaModifier3,-LM3", m_adLambdaModifier[ 3 ], ( Double )1.0, "Lambda modifier for temporal layer 3")
433	("LambdaModifier4,-LM4", m_adLambdaModifier[ 4 ], ( Double )1.0, "Lambda modifier for temporal layer 4")
434	("LambdaModifier5,-LM5", m_adLambdaModifier[ 5 ], ( Double )1.0, "Lambda modifier for temporal layer 5")
435	("LambdaModifier6,-LM6", m_adLambdaModifier[ 6 ], ( Double )1.0, "Lambda modifier for temporal layer 6")
436	("LambdaModifier7,-LM7", m_adLambdaModifier[ 7 ], ( Double )1.0, "Lambda modifier for temporal layer 7")
437	#endif
438
439	/* Quantization parameters */
440	#if SVC_EXTENSION
441	("QP%d,-q%d", cfg_fQP, 30.0, MAX_LAYERS, "Qp value for layer %d, if value is float, QP is switched once during encoding")
442	#else
443	("QP,q", m_fQP, 30.0, "Qp value, if value is float, QP is switched once during encoding")
444	#endif
445	("DeltaQpRD,-dqr",m_uiDeltaQpRD, 0u, "max dQp offset for slice")
446	("MaxDeltaQP,d", m_iMaxDeltaQP, 0, "max dQp offset for block")
447	("MaxCuDQPDepth,-dqd", m_iMaxCuDQPDepth, 0, "max depth for a minimum CuDQP")
448
449	("CbQpOffset,-cbqpofs", m_cbQpOffset, 0, "Chroma Cb QP Offset")
450	("CrQpOffset,-crqpofs", m_crQpOffset, 0, "Chroma Cr QP Offset")
451
452	#if ADAPTIVE_QP_SELECTION
453	("AdaptiveQpSelection,-aqps", m_bUseAdaptQpSelect, false, "AdaptiveQpSelection")
454	#endif
455
456	("AdaptiveQP,-aq", m_bUseAdaptiveQP, false, "QP adaptation based on a psycho-visual model")
457	("MaxQPAdaptationRange,-aqr", m_iQPAdaptationRange, 6, "QP adaptation range")
458	#if !SVC_EXTENSION
459	("dQPFile,m", cfg_dQPFile, string(""), "dQP file name")
460	#endif
461	("RDOQ", m_useRDOQ, true )
462	("RDOQTS", m_useRDOQTS, true )
463	#if L0232_RD_PENALTY
464	("RDpenalty", m_rdPenalty, 0, "RD-penalty for 32x32 TU for intra in non-intra slices. 0:disbaled 1:RD-penalty 2:maximum RD-penalty")
465	#endif
466	// Entropy coding parameters
467	("SBACRD", m_bUseSBACRD, true, "SBAC based RD estimation")
468
469	// Deblocking filter parameters
470	("LoopFilterDisable", m_bLoopFilterDisable, false )
471	("LoopFilterOffsetInPPS", m_loopFilterOffsetInPPS, false )
472	("LoopFilterBetaOffset_div2", m_loopFilterBetaOffsetDiv2, 0 )
473	("LoopFilterTcOffset_div2", m_loopFilterTcOffsetDiv2, 0 )
474	("DeblockingFilterControlPresent", m_DeblockingFilterControlPresent, false )
475
476	// Coding tools
477	("AMP", m_enableAMP, true, "Enable asymmetric motion partitions")
478	("TransformSkip", m_useTransformSkip, false, "Intra transform skipping")
479	("TransformSkipFast", m_useTransformSkipFast, false, "Fast intra transform skipping")
480	("SAO", m_bUseSAO, true, "Enable Sample Adaptive Offset")
481	("MaxNumOffsetsPerPic", m_maxNumOffsetsPerPic, 2048, "Max number of SAO offset per picture (Default: 2048)")
482	("SAOLcuBoundary", m_saoLcuBoundary, false, "0: right/bottom LCU boundary areas skipped from SAO parameter estimation, 1: non-deblocked pixels are used for those areas")
483	("SAOLcuBasedOptimization", m_saoLcuBasedOptimization, true, "0: SAO picture-based optimization, 1: SAO LCU-based optimization ")
484	("SliceMode", m_sliceMode, 0, "0: Disable all Recon slice limits, 1: Enforce max # of LCUs, 2: Enforce max # of bytes, 3:specify tiles per dependent slice")
485	("SliceArgument", m_sliceArgument, 0, "Depending on SliceMode being:"
486	"\t1: max number of CTUs per slice"
487	"\t2: max number of bytes per slice"
488	"\t3: max number of tiles per slice")
489	("SliceSegmentMode", m_sliceSegmentMode, 0, "0: Disable all slice segment limits, 1: Enforce max # of LCUs, 2: Enforce max # of bytes, 3:specify tiles per dependent slice")
490	("SliceSegmentArgument", m_sliceSegmentArgument, 0, "Depending on SliceSegmentMode being:"
491	"\t1: max number of CTUs per slice segment"
492	"\t2: max number of bytes per slice segment"
493	"\t3: max number of tiles per slice segment")
494	("LFCrossSliceBoundaryFlag", m_bLFCrossSliceBoundaryFlag, true)
495
496	("ConstrainedIntraPred", m_bUseConstrainedIntraPred, false, "Constrained Intra Prediction")
497
498	("PCMEnabledFlag", m_usePCM, false)
499	("PCMLog2MaxSize", m_pcmLog2MaxSize, 5u)
500	("PCMLog2MinSize", m_uiPCMLog2MinSize, 3u)
501	("PCMInputBitDepthFlag", m_bPCMInputBitDepthFlag, true)
502	("PCMFilterDisableFlag", m_bPCMFilterDisableFlag, false)
503
504	("LosslessCuEnabled", m_useLossless, false)
505
506	("WeightedPredP,-wpP", m_useWeightedPred, false, "Use weighted prediction in P slices")
507	("WeightedPredB,-wpB", m_useWeightedBiPred, false, "Use weighted (bidirectional) prediction in B slices")
508	("Log2ParallelMergeLevel", m_log2ParallelMergeLevel, 2u, "Parallel merge estimation region")
509	("UniformSpacingIdc", m_iUniformSpacingIdr, 0, "Indicates if the column and row boundaries are distributed uniformly")
510	("NumTileColumnsMinus1", m_iNumColumnsMinus1, 0, "Number of columns in a picture minus 1")
511	("ColumnWidthArray", cfg_ColumnWidth, string(""), "Array containing ColumnWidth values in units of LCU")
512	("NumTileRowsMinus1", m_iNumRowsMinus1, 0, "Number of rows in a picture minus 1")
513	("RowHeightArray", cfg_RowHeight, string(""), "Array containing RowHeight values in units of LCU")
514	("LFCrossTileBoundaryFlag", m_bLFCrossTileBoundaryFlag, true, "1: cross-tile-boundary loop filtering. 0:non-cross-tile-boundary loop filtering")
515	("WaveFrontSynchro", m_iWaveFrontSynchro, 0, "0: no synchro; 1 synchro with TR; 2 TRR etc")
516	("ScalingList", m_useScalingListId, 0, "0: no scaling list, 1: default scaling lists, 2: scaling lists specified in ScalingListFile")
517	("ScalingListFile", cfg_ScalingListFile, string(""), "Scaling list file name")
518	("SignHideFlag,-SBH", m_signHideFlag, 1)
519	("MaxNumMergeCand", m_maxNumMergeCand, 5u, "Maximum number of merge candidates")
520
521	/* Misc. */
522	("SEIDecodedPictureHash", m_decodedPictureHashSEIEnabled, 0, "Control generation of decode picture hash SEI messages\n"
523	"\t3: checksum\n"
524	"\t2: CRC\n"
525	"\t1: use MD5\n"
526	"\t0: disable")
527	("SEIpictureDigest", m_decodedPictureHashSEIEnabled, 0, "deprecated alias for SEIDecodedPictureHash")
528	("TMVPMode", m_TMVPModeId, 1, "TMVP mode 0: TMVP disable for all slices. 1: TMVP enable for all slices (default) 2: TMVP enable for certain slices only")
529	("FEN", m_bUseFastEnc, false, "fast encoder setting")
530	("ECU", m_bUseEarlyCU, false, "Early CU setting")
531	("FDM", m_useFastDecisionForMerge, true, "Fast decision for Merge RD Cost")
532	("CFM", m_bUseCbfFastMode, false, "Cbf fast mode setting")
533	("ESD", m_useEarlySkipDetection, false, "Early SKIP detection setting")
534	#if RATE_CONTROL_LAMBDA_DOMAIN
535	( "RateControl", m_RCEnableRateControl, false, "Rate control: enable rate control" )
536	( "TargetBitrate", m_RCTargetBitrate, 0, "Rate control: target bitrate" )
537	( "KeepHierarchicalBit", m_RCKeepHierarchicalBit, false, "Rate control: keep hierarchical bit allocation in rate control algorithm" )
538	( "LCULevelRateControl", m_RCLCULevelRC, true, "Rate control: true: LCU level RC; false: picture level RC" )
539	( "RCLCUSeparateModel", m_RCUseLCUSeparateModel, true, "Rate control: use LCU level separate R-lambda model" )
540	( "InitialQP", m_RCInitialQP, 0, "Rate control: initial QP" )
541	( "RCForceIntraQP", m_RCForceIntraQP, false, "Rate control: force intra QP to be equal to initial QP" )
542	#else
543	("RateCtrl,-rc", m_enableRateCtrl, false, "Rate control on/off")
544	("TargetBitrate,-tbr", m_targetBitrate, 0, "Input target bitrate")
545	("NumLCUInUnit,-nu", m_numLCUInUnit, 0, "Number of LCUs in an Unit")
546	#endif
547
548	("TransquantBypassEnableFlag", m_TransquantBypassEnableFlag, false, "transquant_bypass_enable_flag indicator in PPS")
549	("CUTransquantBypassFlagValue", m_CUTransquantBypassFlagValue, false, "Fixed cu_transquant_bypass_flag value, when transquant_bypass_enable_flag is enabled")
550	("RecalculateQPAccordingToLambda", m_recalculateQPAccordingToLambda, false, "Recalculate QP values according to lambda values. Do not suggest to be enabled in all intra case")
551	("StrongIntraSmoothing,-sis", m_useStrongIntraSmoothing, true, "Enable strong intra smoothing for 32x32 blocks")
552	("SEIActiveParameterSets", m_activeParameterSetsSEIEnabled, 0, "Enable generation of active parameter sets SEI messages")
553	("VuiParametersPresent,-vui", m_vuiParametersPresentFlag, false, "Enable generation of vui_parameters()")
554	("AspectRatioInfoPresent", m_aspectRatioInfoPresentFlag, false, "Signals whether aspect_ratio_idc is present")
555	("AspectRatioIdc", m_aspectRatioIdc, 0, "aspect_ratio_idc")
556	("SarWidth", m_sarWidth, 0, "horizontal size of the sample aspect ratio")
557	("SarHeight", m_sarHeight, 0, "vertical size of the sample aspect ratio")
558	("OverscanInfoPresent", m_overscanInfoPresentFlag, false, "Indicates whether conformant decoded pictures are suitable for display using overscan\n")
559	("OverscanAppropriate", m_overscanAppropriateFlag, false, "Indicates whether conformant decoded pictures are suitable for display using overscan\n")
560	("VideoSignalTypePresent", m_videoSignalTypePresentFlag, false, "Signals whether video_format, video_full_range_flag, and colour_description_present_flag are present")
561	("VideoFormat", m_videoFormat, 5, "Indicates representation of pictures")
562	("VideoFullRange", m_videoFullRangeFlag, false, "Indicates the black level and range of luma and chroma signals")
563	("ColourDescriptionPresent", m_colourDescriptionPresentFlag, false, "Signals whether colour_primaries, transfer_characteristics and matrix_coefficients are present")
564	("ColourPrimaries", m_colourPrimaries, 2, "Indicates chromaticity coordinates of the source primaries")
565	("TransferCharateristics", m_transferCharacteristics, 2, "Indicates the opto-electronic transfer characteristics of the source")
566	("MatrixCoefficients", m_matrixCoefficients, 2, "Describes the matrix coefficients used in deriving luma and chroma from RGB primaries")
567	("ChromaLocInfoPresent", m_chromaLocInfoPresentFlag, false, "Signals whether chroma_sample_loc_type_top_field and chroma_sample_loc_type_bottom_field are present")
568	("ChromaSampleLocTypeTopField", m_chromaSampleLocTypeTopField, 0, "Specifies the location of chroma samples for top field")
569	("ChromaSampleLocTypeBottomField", m_chromaSampleLocTypeBottomField, 0, "Specifies the location of chroma samples for bottom field")
570	("NeutralChromaIndication", m_neutralChromaIndicationFlag, false, "Indicates that the value of all decoded chroma samples is equal to 1<<(BitDepthCr-1)")
571	("DefaultDisplayWindowFlag", m_defaultDisplayWindowFlag, false, "Indicates the presence of the Default Window parameters")
572	("DefDispWinLeftOffset", m_defDispWinLeftOffset, 0, "Specifies the left offset of the default display window from the conformance window")
573	("DefDispWinRightOffset", m_defDispWinRightOffset, 0, "Specifies the right offset of the default display window from the conformance window")
574	("DefDispWinTopOffset", m_defDispWinTopOffset, 0, "Specifies the top offset of the default display window from the conformance window")
575	("DefDispWinBottomOffset", m_defDispWinBottomOffset, 0, "Specifies the bottom offset of the default display window from the conformance window")
576	("FrameFieldInfoPresentFlag", m_frameFieldInfoPresentFlag, false, "Indicates that pic_struct and field coding related values are present in picture timing SEI messages")
577	("PocProportionalToTimingFlag", m_pocProportionalToTimingFlag, false, "Indicates that the POC value is proportional to the output time w.r.t. first picture in CVS")
578	("NumTicksPocDiffOneMinus1", m_numTicksPocDiffOneMinus1, 0, "Number of ticks minus 1 that for a POC difference of one")
579	("BitstreamRestriction", m_bitstreamRestrictionFlag, false, "Signals whether bitstream restriction parameters are present")
580	("TilesFixedStructure", m_tilesFixedStructureFlag, false, "Indicates that each active picture parameter set has the same values of the syntax elements related to tiles")
581	("MotionVectorsOverPicBoundaries", m_motionVectorsOverPicBoundariesFlag, false, "Indicates that no samples outside the picture boundaries are used for inter prediction")
582	("MaxBytesPerPicDenom", m_maxBytesPerPicDenom, 2, "Indicates a number of bytes not exceeded by the sum of the sizes of the VCL NAL units associated with any coded picture")
583	("MaxBitsPerMinCuDenom", m_maxBitsPerMinCuDenom, 1, "Indicates an upper bound for the number of bits of coding_unit() data")
584	("Log2MaxMvLengthHorizontal", m_log2MaxMvLengthHorizontal, 15, "Indicate the maximum absolute value of a decoded horizontal MV component in quarter-pel luma units")
585	("Log2MaxMvLengthVertical", m_log2MaxMvLengthVertical, 15, "Indicate the maximum absolute value of a decoded vertical MV component in quarter-pel luma units")
586	("SEIRecoveryPoint", m_recoveryPointSEIEnabled, 0, "Control generation of recovery point SEI messages")
587	("SEIBufferingPeriod", m_bufferingPeriodSEIEnabled, 0, "Control generation of buffering period SEI messages")
588	("SEIPictureTiming", m_pictureTimingSEIEnabled, 0, "Control generation of picture timing SEI messages")
589	("SEIFramePacking", m_framePackingSEIEnabled, 0, "Control generation of frame packing SEI messages")
590	("SEIFramePackingType", m_framePackingSEIType, 0, "Define frame packing arrangement\n"
591	"\t0: checkerboard - pixels alternatively represent either frames\n"
592	"\t1: column alternation - frames are interlaced by column\n"
593	"\t2: row alternation - frames are interlaced by row\n"
594	"\t3: side by side - frames are displayed horizontally\n"
595	"\t4: top bottom - frames are displayed vertically\n"
596	"\t5: frame alternation - one frame is alternated with the other")
597	("SEIFramePackingId", m_framePackingSEIId, 0, "Id of frame packing SEI message for a given session")
598	("SEIFramePackingQuincunx", m_framePackingSEIQuincunx, 0, "Indicate the presence of a Quincunx type video frame")
599	("SEIFramePackingInterpretation", m_framePackingSEIInterpretation, 0, "Indicate the interpretation of the frame pair\n"
600	"\t0: unspecified\n"
601	"\t1: stereo pair, frame0 represents left view\n"
602	"\t2: stereo pair, frame0 represents right view")
603	("SEIDisplayOrientation", m_displayOrientationSEIAngle, 0, "Control generation of display orientation SEI messages\n"
604	"\tN: 0 < N < (2^16 - 1) enable display orientation SEI message with anticlockwise_rotation = N and display_orientation_repetition_period = 1\n"
605	"\t0: disable")
606	("SEITemporalLevel0Index", m_temporalLevel0IndexSEIEnabled, 0, "Control generation of temporal level 0 index SEI messages")
607	("SEIGradualDecodingRefreshInfo", m_gradualDecodingRefreshInfoEnabled, 0, "Control generation of gradual decoding refresh information SEI message")
608	("SEIDecodingUnitInfo", m_decodingUnitInfoSEIEnabled, 0, "Control generation of decoding unit information SEI message.")
609	#if SIGNAL_BITRATE_PICRATE_IN_VPS
610	("BitRatePicRateMaxTLayers", m_bitRatePicRateMaxTLayers, 0, "Maximum number of sub-layers signalled; can be inferred otherwise; here for easy parsing of config. file")
611	("BitRateInfoPresent", cfg_bitRateInfoPresentFlag, string(""), "Control signalling of bit rate information of avg. bit rate and max. bit rate in VPS\n"
612	"\t0: Do not sent bit rate info\n"
613	"\tN (N > 0): Send bit rate info for N sub-layers. N should equal maxTempLayers.")
614	("PicRateInfoPresent", cfg_picRateInfoPresentFlag, string(""), "Control signalling of picture rate information of avg. bit rate and max. bit rate in VPS\n"
615	"\t0: Do not sent picture rate info\n"
616	"\tN (N > 0): Send picture rate info for N sub-layers. N should equal maxTempLayers.")
617	("AvgBitRate", cfg_avgBitRate, string(""), "List of avg. bit rates for the different sub-layers; include non-negative number even if corresponding flag is 0")
618	("MaxBitRate", cfg_maxBitRate, string(""), "List of max. bit rates for the different sub-layers; include non-negative number even if corresponding flag is 0")
619	("AvgPicRate", cfg_avgPicRate, string(""), "List of avg. picture rates for the different sub-layers; include non-negative number even if corresponding flag is 0")
620	("ConstantPicRateIdc", cfg_constantPicRateIdc, string(""), "List of constant picture rate IDCs; include non-negative number even if corresponding flag is 0")
621	#endif
622	;
623
624	for(Int i=1; i<MAX_GOP+1; i++) {
625	std::ostringstream cOSS;
626	cOSS<<"Frame"<<i;
627	opts.addOptions()(cOSS.str(), m_GOPList[i-1], GOPEntry());
628	}
629	po::setDefaults(opts);
630	const list<const Char>& argv_unhandled = po::scanArgv(opts, argc, (const Char*) argv);
631
632	for (list<const Char*>::const_iterator it = argv_unhandled.begin(); it != argv_unhandled.end(); it++)
633	{
634	fprintf(stderr, "Unhandled argument ignored: `%s'\n", *it);
635	}
636
637	if (argc == 1 \|\| do_help)
638	{
639	/* argc == 1: no options have been specified */
640	po::doHelp(cout, opts);
641	return false;
642	}
643
644	/*
645	* Set any derived parameters
646	*/
647	/* convert std::string to c string for compatability */
648	#if SVC_EXTENSION
649	m_pBitstreamFile = cfg_BitstreamFile.empty() ? NULL : strdup(cfg_BitstreamFile.c_str());
650	#if AVC_SYNTAX
651	m_BLSyntaxFile = cfg_BLSyntaxFile.empty() ? NULL : strdup(cfg_BLSyntaxFile.c_str());
652	#endif
653	#else
654	m_pchInputFile = cfg_InputFile.empty() ? NULL : strdup(cfg_InputFile.c_str());
655	m_phBitstreamFile = cfg_BitstreamFile.empty() ? NULL : strdup(cfg_BitstreamFile.c_str());
656	m_pchReconFile = cfg_ReconFile.empty() ? NULL : strdup(cfg_ReconFile.c_str());
657	m_pchdQPFile = cfg_dQPFile.empty() ? NULL : strdup(cfg_dQPFile.c_str());
658	#endif
659
660	Char* pColumnWidth = cfg_ColumnWidth.empty() ? NULL: strdup(cfg_ColumnWidth.c_str());
661	Char* pRowHeight = cfg_RowHeight.empty() ? NULL : strdup(cfg_RowHeight.c_str());
662	if( m_iUniformSpacingIdr == 0 && m_iNumColumnsMinus1 > 0 )
663	{
664	char *columnWidth;
665	int i=0;
666	m_pColumnWidth = new UInt[m_iNumColumnsMinus1];
667	columnWidth = strtok(pColumnWidth, " ,-");
668	while(columnWidth!=NULL)
669	{
670	if( i>=m_iNumColumnsMinus1 )
671	{
672	printf( "The number of columns whose width are defined is larger than the allowed number of columns.\n" );
673	exit( EXIT_FAILURE );
674	}
675	*( m_pColumnWidth + i ) = atoi( columnWidth );
676	columnWidth = strtok(NULL, " ,-");
677	i++;
678	}
679	if( i<m_iNumColumnsMinus1 )
680	{
681	printf( "The width of some columns is not defined.\n" );
682	exit( EXIT_FAILURE );
683	}
684	}
685	else
686	{
687	m_pColumnWidth = NULL;
688	}
689
690	if( m_iUniformSpacingIdr == 0 && m_iNumRowsMinus1 > 0 )
691	{
692	char *rowHeight;
693	int i=0;
694	m_pRowHeight = new UInt[m_iNumRowsMinus1];
695	rowHeight = strtok(pRowHeight, " ,-");
696	while(rowHeight!=NULL)
697	{
698	if( i>=m_iNumRowsMinus1 )
699	{
700	printf( "The number of rows whose height are defined is larger than the allowed number of rows.\n" );
701	exit( EXIT_FAILURE );
702	}
703	*( m_pRowHeight + i ) = atoi( rowHeight );
704	rowHeight = strtok(NULL, " ,-");
705	i++;
706	}
707	if( i<m_iNumRowsMinus1 )
708	{
709	printf( "The height of some rows is not defined.\n" );
710	exit( EXIT_FAILURE );
711	}
712	}
713	else
714	{
715	m_pRowHeight = NULL;
716	}
717	#if SIGNAL_BITRATE_PICRATE_IN_VPS
718	readBoolString(cfg_bitRateInfoPresentFlag, m_bitRatePicRateMaxTLayers, m_bitRateInfoPresentFlag, "bit rate info. present flag" );
719	readIntString (cfg_avgBitRate, m_bitRatePicRateMaxTLayers, m_avgBitRate, "avg. bit rate" );
720	readIntString (cfg_maxBitRate, m_bitRatePicRateMaxTLayers, m_maxBitRate, "max. bit rate" );
721	readBoolString(cfg_picRateInfoPresentFlag, m_bitRatePicRateMaxTLayers, m_picRateInfoPresentFlag, "bit rate info. present flag" );
722	readIntString (cfg_avgPicRate, m_bitRatePicRateMaxTLayers, m_avgPicRate, "avg. pic rate" );
723	readIntString (cfg_constantPicRateIdc, m_bitRatePicRateMaxTLayers, m_constantPicRateIdc, "constant pic rate Idc" );
724	#endif
725	m_scalingListFile = cfg_ScalingListFile.empty() ? NULL : strdup(cfg_ScalingListFile.c_str());
726
727	/* rules for input, output and internal bitdepths as per help text */
728	if (!m_internalBitDepthY) { m_internalBitDepthY = m_inputBitDepthY; }
729	if (!m_internalBitDepthC) { m_internalBitDepthC = m_internalBitDepthY; }
730	if (!m_inputBitDepthC) { m_inputBitDepthC = m_inputBitDepthY; }
731	if (!m_outputBitDepthY) { m_outputBitDepthY = m_internalBitDepthY; }
732	if (!m_outputBitDepthC) { m_outputBitDepthC = m_internalBitDepthC; }
733
734	#if !SVC_EXTENSION
735	// TODO:ChromaFmt assumes 4:2:0 below
736	switch (m_conformanceMode)
737	{
738	case 0:
739	{
740	// no conformance or padding
741	m_confLeft = m_confRight = m_confTop = m_confBottom = 0;
742	m_aiPad[1] = m_aiPad[0] = 0;
743	break;
744	}
745	case 1:
746	{
747	// automatic padding to minimum CU size
748	Int minCuSize = m_uiMaxCUHeight >> (m_uiMaxCUDepth - 1);
749	if (m_iSourceWidth % minCuSize)
750	{
751	m_aiPad[0] = m_confRight = ((m_iSourceWidth / minCuSize) + 1) * minCuSize - m_iSourceWidth;
752	m_iSourceWidth += m_confRight;
753	}
754	if (m_iSourceHeight % minCuSize)
755	{
756	m_aiPad[1] = m_confBottom = ((m_iSourceHeight / minCuSize) + 1) * minCuSize - m_iSourceHeight;
757	m_iSourceHeight += m_confBottom;
758	}
759	if (m_aiPad[0] % TComSPS::getWinUnitX(CHROMA_420) != 0)
760	{
761	fprintf(stderr, "Error: picture width is not an integer multiple of the specified chroma subsampling\n");
762	exit(EXIT_FAILURE);
763	}
764	if (m_aiPad[1] % TComSPS::getWinUnitY(CHROMA_420) != 0)
765	{
766	fprintf(stderr, "Error: picture height is not an integer multiple of the specified chroma subsampling\n");
767	exit(EXIT_FAILURE);
768	}
769	break;
770	}
771	case 2:
772	{
773	//padding
774	m_iSourceWidth += m_aiPad[0];
775	m_iSourceHeight += m_aiPad[1];
776	m_confRight = m_aiPad[0];
777	m_confBottom = m_aiPad[1];
778	break;
779	}
780	case 3:
781	{
782	// conformance
783	if ((m_confLeft == 0) && (m_confRight == 0) && (m_confTop == 0) && (m_confBottom == 0))
784	{
785	fprintf(stderr, "Warning: Conformance window enabled, but all conformance window parameters set to zero\n");
786	}
787	if ((m_aiPad[1] != 0) \|\| (m_aiPad[0]!=0))
788	{
789	fprintf(stderr, "Warning: Conformance window enabled, padding parameters will be ignored\n");
790	}
791	m_aiPad[1] = m_aiPad[0] = 0;
792	break;
793	}
794	}
795
796	// allocate slice-based dQP values
797	m_aidQP = new Int[ m_framesToBeEncoded + m_iGOPSize + 1 ];
798	::memset( m_aidQP, 0, sizeof(Int)*( m_framesToBeEncoded + m_iGOPSize + 1 ) );
799
800	// handling of floating-point QP values
801	// if QP is not integer, sequence is split into two sections having QP and QP+1
802	m_iQP = (Int)( m_fQP );
803	if ( m_iQP < m_fQP )
804	{
805	Int iSwitchPOC = (Int)( m_framesToBeEncoded - (m_fQP - m_iQP)*m_framesToBeEncoded + 0.5 );
806
807	iSwitchPOC = (Int)( (Double)iSwitchPOC / m_iGOPSize + 0.5 )*m_iGOPSize;
808	for ( Int i=iSwitchPOC; i<m_framesToBeEncoded + m_iGOPSize + 1; i++ )
809	{
810	m_aidQP[i] = 1;
811	}
812	}
813
814	// reading external dQP description from file
815	if ( m_pchdQPFile )
816	{
817	FILE* fpt=fopen( m_pchdQPFile, "r" );
818	if ( fpt )
819	{
820	Int iValue;
821	Int iPOC = 0;
822	while ( iPOC < m_framesToBeEncoded )
823	{
824	if ( fscanf(fpt, "%d", &iValue ) == EOF ) break;
825	m_aidQP[ iPOC ] = iValue;
826	iPOC++;
827	}
828	fclose(fpt);
829	}
830	}
831	m_iWaveFrontSubstreams = m_iWaveFrontSynchro ? (m_iSourceHeight + m_uiMaxCUHeight - 1) / m_uiMaxCUHeight : 1;
832	#endif
833	// check validity of input parameters
834	xCheckParameter();
835
836	// set global varibles
837	xSetGlobal();
838
839	// print-out parameters
840	xPrintParameter();
841
842	return true;
843	}
844	#if SIGNAL_BITRATE_PICRATE_IN_VPS
845	Void readBoolString(const string inpString, const Int numEntries, Bool* &memberArray, const char *elementName)
846	{
847	Char* inpArray = inpString.empty() ? NULL : strdup(inpString.c_str());
848	Int i = 0;
849	if(numEntries)
850	{
851	Char* tempArray = strtok(inpArray, " ,-");
852	memberArray = new Bool[numEntries];
853	while( tempArray != NULL )
854	{
855	if( i >= numEntries )
856	{
857	printf( "The number of %s defined is larger than the allowed number\n", elementName );
858	exit( EXIT_FAILURE );
859	}
860	assert( (atoi(tempArray) == 0) \|\| (atoi(tempArray) == 1) );
861	*( memberArray + i ) = atoi(tempArray);
862	tempArray = strtok(NULL, " ,-");
863	i++;
864	}
865	if( i < numEntries )
866	{
867	printf( "Some %s are not defined\n", elementName );
868	exit( EXIT_FAILURE );
869	}
870	}
871	else
872	{
873	memberArray = NULL;
874	}
875	}
876
877	Void readIntString(const string inpString, const Int numEntries, Int* &memberArray, const char *elementName)
878	{
879	Char* inpArray = inpString.empty() ? NULL : strdup(inpString.c_str());
880	Int i = 0;
881	if(numEntries)
882	{
883	Char* tempArray = strtok(inpArray, " ,-");
884	memberArray = new Int[numEntries];
885	while( tempArray != NULL )
886	{
887	if( i >= numEntries )
888	{
889	printf( "The number of %s defined is larger than the allowed number\n", elementName );
890	exit( EXIT_FAILURE );
891	}
892	*( memberArray + i ) = atoi(tempArray);
893	tempArray = strtok(NULL, " ,-");
894	i++;
895	}
896	if( i < numEntries )
897	{
898	printf( "Some %s are not defined\n", elementName );
899	exit( EXIT_FAILURE );
900	}
901	}
902	else
903	{
904	memberArray = NULL;
905	}
906	}
907	#endif
908	// ====================================================================================================================
909	// Private member functions
910	// ====================================================================================================================
911
912	Bool confirmPara(Bool bflag, const Char* message);
913
914	Void TAppEncCfg::xCheckParameter()
915	{
916	if (!m_decodedPictureHashSEIEnabled)
917	{
918	fprintf(stderr, "******************************************************************\n");
919	fprintf(stderr, " WARNING: --SEIDecodedPictureHash is now disabled by default. \n");
920	fprintf(stderr, " Automatic verification of decoded pictures by a \n");
921	fprintf(stderr, " decoder requires this option to be enabled. \n");
922	fprintf(stderr, "******************************************************************\n");
923	}
924
925	Bool check_failed = false; /* abort if there is a fatal configuration problem */
926	#define xConfirmPara(a,b) check_failed \|= confirmPara(a,b)
927	// check range of parameters
928	xConfirmPara( m_inputBitDepthY < 8, "InputBitDepth must be at least 8" );
929	xConfirmPara( m_inputBitDepthC < 8, "InputBitDepthC must be at least 8" );
930	#if !SVC_EXTENSION
931	xConfirmPara( m_iFrameRate <= 0, "Frame rate must be more than 1" );
932	#endif
933	xConfirmPara( m_framesToBeEncoded <= 0, "Total Number Of Frames encoded must be more than 0" );
934	xConfirmPara( m_iGOPSize < 1 , "GOP Size must be greater or equal to 1" );
935	xConfirmPara( m_iGOPSize > 1 && m_iGOPSize % 2, "GOP Size must be a multiple of 2, if GOP Size is greater than 1" );
936	#if !SVC_EXTENSION
937	xConfirmPara( (m_iIntraPeriod > 0 && m_iIntraPeriod < m_iGOPSize) \|\| m_iIntraPeriod == 0, "Intra period must be more than GOP size, or -1 , not 0" );
938	#endif
939	xConfirmPara( m_iDecodingRefreshType < 0 \|\| m_iDecodingRefreshType > 2, "Decoding Refresh Type must be equal to 0, 1 or 2" );
940	#if !SVC_EXTENSION
941	xConfirmPara( m_iQP < -6 * (m_internalBitDepthY - 8) \|\| m_iQP > 51, "QP exceeds supported range (-QpBDOffsety to 51)" );
942	#endif
943	xConfirmPara( m_loopFilterBetaOffsetDiv2 < -13 \|\| m_loopFilterBetaOffsetDiv2 > 13, "Loop Filter Beta Offset div. 2 exceeds supported range (-13 to 13)");
944	xConfirmPara( m_loopFilterTcOffsetDiv2 < -13 \|\| m_loopFilterTcOffsetDiv2 > 13, "Loop Filter Tc Offset div. 2 exceeds supported range (-13 to 13)");
945	xConfirmPara( m_iFastSearch < 0 \|\| m_iFastSearch > 2, "Fast Search Mode is not supported value (0:Full search 1:Diamond 2:PMVFAST)" );
946	xConfirmPara( m_iSearchRange < 0 , "Search Range must be more than 0" );
947	xConfirmPara( m_bipredSearchRange < 0 , "Search Range must be more than 0" );
948	xConfirmPara( m_iMaxDeltaQP > 7, "Absolute Delta QP exceeds supported range (0 to 7)" );
949	xConfirmPara( m_iMaxCuDQPDepth > m_uiMaxCUDepth - 1, "Absolute depth for a minimum CuDQP exceeds maximum coding unit depth" );
950
951	xConfirmPara( m_cbQpOffset < -12, "Min. Chroma Cb QP Offset is -12" );
952	xConfirmPara( m_cbQpOffset > 12, "Max. Chroma Cb QP Offset is 12" );
953	xConfirmPara( m_crQpOffset < -12, "Min. Chroma Cr QP Offset is -12" );
954	xConfirmPara( m_crQpOffset > 12, "Max. Chroma Cr QP Offset is 12" );
955
956	xConfirmPara( m_iQPAdaptationRange <= 0, "QP Adaptation Range must be more than 0" );
957	#if !SVC_EXTENSION
958	if (m_iDecodingRefreshType == 2)
959	{
960	xConfirmPara( m_iIntraPeriod > 0 && m_iIntraPeriod <= m_iGOPSize , "Intra period must be larger than GOP size for periodic IDR pictures");
961	}
962	#endif
963	xConfirmPara( (m_uiMaxCUWidth >> m_uiMaxCUDepth) < 4, "Minimum partition width size should be larger than or equal to 8");
964	xConfirmPara( (m_uiMaxCUHeight >> m_uiMaxCUDepth) < 4, "Minimum partition height size should be larger than or equal to 8");
965	xConfirmPara( m_uiMaxCUWidth < 16, "Maximum partition width size should be larger than or equal to 16");
966	xConfirmPara( m_uiMaxCUHeight < 16, "Maximum partition height size should be larger than or equal to 16");
967	#if !SVC_EXTENSION
968	xConfirmPara( (m_iSourceWidth % (m_uiMaxCUWidth >> (m_uiMaxCUDepth-1)))!=0, "Resulting coded frame width must be a multiple of the minimum CU size");
969	xConfirmPara( (m_iSourceHeight % (m_uiMaxCUHeight >> (m_uiMaxCUDepth-1)))!=0, "Resulting coded frame height must be a multiple of the minimum CU size");
970	#endif
971
972	xConfirmPara( m_uiQuadtreeTULog2MinSize < 2, "QuadtreeTULog2MinSize must be 2 or greater.");
973	xConfirmPara( m_uiQuadtreeTULog2MaxSize > 5, "QuadtreeTULog2MaxSize must be 5 or smaller.");
974	xConfirmPara( (1<<m_uiQuadtreeTULog2MaxSize) > m_uiMaxCUWidth, "QuadtreeTULog2MaxSize must be log2(maxCUSize) or smaller.");
975
976	xConfirmPara( m_uiQuadtreeTULog2MaxSize < m_uiQuadtreeTULog2MinSize, "QuadtreeTULog2MaxSize must be greater than or equal to m_uiQuadtreeTULog2MinSize.");
977	xConfirmPara( (1<<m_uiQuadtreeTULog2MinSize)>(m_uiMaxCUWidth >>(m_uiMaxCUDepth-1)), "QuadtreeTULog2MinSize must not be greater than minimum CU size" ); // HS
978	xConfirmPara( (1<<m_uiQuadtreeTULog2MinSize)>(m_uiMaxCUHeight>>(m_uiMaxCUDepth-1)), "QuadtreeTULog2MinSize must not be greater than minimum CU size" ); // HS
979	xConfirmPara( ( 1 << m_uiQuadtreeTULog2MinSize ) > ( m_uiMaxCUWidth >> m_uiMaxCUDepth ), "Minimum CU width must be greater than minimum transform size." );
980	xConfirmPara( ( 1 << m_uiQuadtreeTULog2MinSize ) > ( m_uiMaxCUHeight >> m_uiMaxCUDepth ), "Minimum CU height must be greater than minimum transform size." );
981	xConfirmPara( m_uiQuadtreeTUMaxDepthInter < 1, "QuadtreeTUMaxDepthInter must be greater than or equal to 1" );
982	xConfirmPara( m_uiMaxCUWidth < ( 1 << (m_uiQuadtreeTULog2MinSize + m_uiQuadtreeTUMaxDepthInter - 1) ), "QuadtreeTUMaxDepthInter must be less than or equal to the difference between log2(maxCUSize) and QuadtreeTULog2MinSize plus 1" );
983	xConfirmPara( m_uiQuadtreeTUMaxDepthIntra < 1, "QuadtreeTUMaxDepthIntra must be greater than or equal to 1" );
984	xConfirmPara( m_uiMaxCUWidth < ( 1 << (m_uiQuadtreeTULog2MinSize + m_uiQuadtreeTUMaxDepthIntra - 1) ), "QuadtreeTUMaxDepthInter must be less than or equal to the difference between log2(maxCUSize) and QuadtreeTULog2MinSize plus 1" );
985
986	xConfirmPara( m_maxNumMergeCand < 1, "MaxNumMergeCand must be 1 or greater.");
987	xConfirmPara( m_maxNumMergeCand > 5, "MaxNumMergeCand must be 5 or smaller.");
988
989	#if !SVC_EXTENSION
990	#if ADAPTIVE_QP_SELECTION
991	xConfirmPara( m_bUseAdaptQpSelect == true && m_iQP < 0, "AdaptiveQpSelection must be disabled when QP < 0.");
992	xConfirmPara( m_bUseAdaptQpSelect == true && (m_cbQpOffset !=0 \|\| m_crQpOffset != 0 ), "AdaptiveQpSelection must be disabled when ChromaQpOffset is not equal to 0.");
993	#endif
994	#endif
995
996	if( m_usePCM)
997	{
998	xConfirmPara( m_uiPCMLog2MinSize < 3, "PCMLog2MinSize must be 3 or greater.");
999	xConfirmPara( m_uiPCMLog2MinSize > 5, "PCMLog2MinSize must be 5 or smaller.");
1000	xConfirmPara( m_pcmLog2MaxSize > 5, "PCMLog2MaxSize must be 5 or smaller.");
1001	xConfirmPara( m_pcmLog2MaxSize < m_uiPCMLog2MinSize, "PCMLog2MaxSize must be equal to or greater than m_uiPCMLog2MinSize.");
1002	}
1003
1004	xConfirmPara( m_sliceMode < 0 \|\| m_sliceMode > 3, "SliceMode exceeds supported range (0 to 3)" );
1005	if (m_sliceMode!=0)
1006	{
1007	xConfirmPara( m_sliceArgument < 1 , "SliceArgument should be larger than or equal to 1" );
1008	}
1009	xConfirmPara( m_sliceSegmentMode < 0 \|\| m_sliceSegmentMode > 3, "SliceSegmentMode exceeds supported range (0 to 3)" );
1010	if (m_sliceSegmentMode!=0)
1011	{
1012	xConfirmPara( m_sliceSegmentArgument < 1 , "SliceSegmentArgument should be larger than or equal to 1" );
1013	}
1014
1015	Bool tileFlag = (m_iNumColumnsMinus1 > 0 \|\| m_iNumRowsMinus1 > 0 );
1016	xConfirmPara( tileFlag && m_iWaveFrontSynchro, "Tile and Wavefront can not be applied together");
1017
1018	//TODO:ChromaFmt assumes 4:2:0 below
1019	#if !SVC_EXTENSION
1020	xConfirmPara( m_iSourceWidth % TComSPS::getWinUnitX(CHROMA_420) != 0, "Picture width must be an integer multiple of the specified chroma subsampling");
1021	xConfirmPara( m_iSourceHeight % TComSPS::getWinUnitY(CHROMA_420) != 0, "Picture height must be an integer multiple of the specified chroma subsampling");
1022
1023	xConfirmPara( m_aiPad[0] % TComSPS::getWinUnitX(CHROMA_420) != 0, "Horizontal padding must be an integer multiple of the specified chroma subsampling");
1024	xConfirmPara( m_aiPad[1] % TComSPS::getWinUnitY(CHROMA_420) != 0, "Vertical padding must be an integer multiple of the specified chroma subsampling");
1025
1026	xConfirmPara( m_confLeft % TComSPS::getWinUnitX(CHROMA_420) != 0, "Left conformance window offset must be an integer multiple of the specified chroma subsampling");
1027	xConfirmPara( m_confRight % TComSPS::getWinUnitX(CHROMA_420) != 0, "Right conformance window offset must be an integer multiple of the specified chroma subsampling");
1028	xConfirmPara( m_confTop % TComSPS::getWinUnitY(CHROMA_420) != 0, "Top conformance window offset must be an integer multiple of the specified chroma subsampling");
1029	xConfirmPara( m_confBottom % TComSPS::getWinUnitY(CHROMA_420) != 0, "Bottom conformance window offset must be an integer multiple of the specified chroma subsampling");
1030	#endif
1031
1032	#if REF_IDX_ME_AROUND_ZEROMV \|\| REF_IDX_ME_ZEROMV
1033	xConfirmPara( REF_IDX_ME_AROUND_ZEROMV && REF_IDX_ME_ZEROMV, "REF_IDX_ME_AROUND_ZEROMV and REF_IDX_ME_ZEROMV cannot be enabled simultaneously");
1034	#endif
1035	// max CU width and height should be power of 2
1036	UInt ui = m_uiMaxCUWidth;
1037	while(ui)
1038	{
1039	ui >>= 1;
1040	if( (ui & 1) == 1)
1041	xConfirmPara( ui != 1 , "Width should be 2^n");
1042	}
1043	ui = m_uiMaxCUHeight;
1044	while(ui)
1045	{
1046	ui >>= 1;
1047	if( (ui & 1) == 1)
1048	xConfirmPara( ui != 1 , "Height should be 2^n");
1049	}
1050
1051
1052	/* if this is an intra-only sequence, ie IntraPeriod=1, don't verify the GOP structure
1053	* This permits the ability to omit a GOP structure specification */
1054	#if SVC_EXTENSION
1055	for(UInt layer = 0; layer < MAX_LAYERS; layer++)
1056	{
1057	Int m_iIntraPeriod = m_acLayerCfg[layer].m_iIntraPeriod;
1058	#endif
1059	if (m_iIntraPeriod == 1 && m_GOPList[0].m_POC == -1) {
1060	m_GOPList[0] = GOPEntry();
1061	m_GOPList[0].m_QPFactor = 1;
1062	m_GOPList[0].m_betaOffsetDiv2 = 0;
1063	m_GOPList[0].m_tcOffsetDiv2 = 0;
1064	m_GOPList[0].m_POC = 1;
1065	m_GOPList[0].m_numRefPicsActive = 4;
1066	}
1067	#if SVC_EXTENSION
1068	}
1069	#endif
1070
1071	Bool verifiedGOP=false;
1072	Bool errorGOP=false;
1073	Int checkGOP=1;
1074	Int numRefs = 1;
1075	Int refList[MAX_NUM_REF_PICS+1];
1076	refList[0]=0;
1077	Bool isOK[MAX_GOP];
1078	for(Int i=0; i<MAX_GOP; i++)
1079	{
1080	isOK[i]=false;
1081	}
1082	Int numOK=0;
1083	#if !SVC_EXTENSION
1084	xConfirmPara( m_iIntraPeriod >=0&&(m_iIntraPeriod%m_iGOPSize!=0), "Intra period must be a multiple of GOPSize, or -1" );
1085	#endif
1086
1087	for(Int i=0; i<m_iGOPSize; i++)
1088	{
1089	if(m_GOPList[i].m_POC==m_iGOPSize)
1090	{
1091	xConfirmPara( m_GOPList[i].m_temporalId!=0 , "The last frame in each GOP must have temporal ID = 0 " );
1092	}
1093	}
1094
1095	#if SVC_EXTENSION
1096	// verify layer configuration parameters
1097	for(UInt layer=0; layer<m_numLayers; layer++)
1098	{
1099	if(m_acLayerCfg[layer].xCheckParameter())
1100	{
1101	printf("\nError: invalid configuration parameter found in layer %d \n", layer);
1102	check_failed = true;
1103	}
1104	}
1105	#endif
1106
1107	#if SVC_EXTENSION
1108	// verify layer configuration parameters
1109	for(UInt layer=0; layer<m_numLayers; layer++)
1110	{
1111	Int m_iIntraPeriod = m_acLayerCfg[layer].m_iIntraPeriod;
1112	#endif
1113	if ( (m_iIntraPeriod != 1) && !m_loopFilterOffsetInPPS && m_DeblockingFilterControlPresent && (!m_bLoopFilterDisable) )
1114	{
1115	for(Int i=0; i<m_iGOPSize; i++)
1116	{
1117	xConfirmPara( (m_GOPList[i].m_betaOffsetDiv2 + m_loopFilterBetaOffsetDiv2) < -6 \|\| (m_GOPList[i].m_betaOffsetDiv2 + m_loopFilterBetaOffsetDiv2) > 6, "Loop Filter Beta Offset div. 2 for one of the GOP entries exceeds supported range (-6 to 6)" );
1118	xConfirmPara( (m_GOPList[i].m_tcOffsetDiv2 + m_loopFilterTcOffsetDiv2) < -6 \|\| (m_GOPList[i].m_tcOffsetDiv2 + m_loopFilterTcOffsetDiv2) > 6, "Loop Filter Tc Offset div. 2 for one of the GOP entries exceeds supported range (-6 to 6)" );
1119	}
1120	}
1121	#if SVC_EXTENSION
1122	}
1123	#endif
1124
1125	m_extraRPSs=0;
1126	//start looping through frames in coding order until we can verify that the GOP structure is correct.
1127	while(!verifiedGOP&&!errorGOP)
1128	{
1129	Int curGOP = (checkGOP-1)%m_iGOPSize;
1130	Int curPOC = ((checkGOP-1)/m_iGOPSize)*m_iGOPSize + m_GOPList[curGOP].m_POC;
1131	if(m_GOPList[curGOP].m_POC<0)
1132	{
1133	printf("\nError: found fewer Reference Picture Sets than GOPSize\n");
1134	errorGOP=true;
1135	}
1136	else
1137	{
1138	//check that all reference pictures are available, or have a POC < 0 meaning they might be available in the next GOP.
1139	Bool beforeI = false;
1140	for(Int i = 0; i< m_GOPList[curGOP].m_numRefPics; i++)
1141	{
1142	Int absPOC = curPOC+m_GOPList[curGOP].m_referencePics[i];
1143	if(absPOC < 0)
1144	{
1145	beforeI=true;
1146	}
1147	else
1148	{
1149	Bool found=false;
1150	for(Int j=0; j<numRefs; j++)
1151	{
1152	if(refList[j]==absPOC)
1153	{
1154	found=true;
1155	for(Int k=0; k<m_iGOPSize; k++)
1156	{
1157	if(absPOC%m_iGOPSize == m_GOPList[k].m_POC%m_iGOPSize)
1158	{
1159	if(m_GOPList[k].m_temporalId==m_GOPList[curGOP].m_temporalId)
1160	{
1161	m_GOPList[k].m_refPic = true;
1162	}
1163	m_GOPList[curGOP].m_usedByCurrPic[i]=m_GOPList[k].m_temporalId<=m_GOPList[curGOP].m_temporalId;
1164	}
1165	}
1166	}
1167	}
1168	if(!found)
1169	{
1170	printf("\nError: ref pic %d is not available for GOP frame %d\n",m_GOPList[curGOP].m_referencePics[i],curGOP+1);
1171	errorGOP=true;
1172	}
1173	}
1174	}
1175	if(!beforeI&&!errorGOP)
1176	{
1177	//all ref frames were present
1178	if(!isOK[curGOP])
1179	{
1180	numOK++;
1181	isOK[curGOP]=true;
1182	if(numOK==m_iGOPSize)
1183	{
1184	verifiedGOP=true;
1185	}
1186	}
1187	}
1188	else
1189	{
1190	//create a new GOPEntry for this frame containing all the reference pictures that were available (POC > 0)
1191	m_GOPList[m_iGOPSize+m_extraRPSs]=m_GOPList[curGOP];
1192	Int newRefs=0;
1193	for(Int i = 0; i< m_GOPList[curGOP].m_numRefPics; i++)
1194	{
1195	Int absPOC = curPOC+m_GOPList[curGOP].m_referencePics[i];
1196	if(absPOC>=0)
1197	{
1198	m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[newRefs]=m_GOPList[curGOP].m_referencePics[i];
1199	m_GOPList[m_iGOPSize+m_extraRPSs].m_usedByCurrPic[newRefs]=m_GOPList[curGOP].m_usedByCurrPic[i];
1200	newRefs++;
1201	}
1202	}
1203	Int numPrefRefs = m_GOPList[curGOP].m_numRefPicsActive;
1204
1205	for(Int offset = -1; offset>-checkGOP; offset--)
1206	{
1207	//step backwards in coding order and include any extra available pictures we might find useful to replace the ones with POC < 0.
1208	Int offGOP = (checkGOP-1+offset)%m_iGOPSize;
1209	Int offPOC = ((checkGOP-1+offset)/m_iGOPSize)*m_iGOPSize + m_GOPList[offGOP].m_POC;
1210	if(offPOC>=0&&m_GOPList[offGOP].m_temporalId<=m_GOPList[curGOP].m_temporalId)
1211	{
1212	Bool newRef=false;
1213	for(Int i=0; i<numRefs; i++)
1214	{
1215	if(refList[i]==offPOC)
1216	{
1217	newRef=true;
1218	}
1219	}
1220	for(Int i=0; i<newRefs; i++)
1221	{
1222	if(m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[i]==offPOC-curPOC)
1223	{
1224	newRef=false;
1225	}
1226	}
1227	if(newRef)
1228	{
1229	Int insertPoint=newRefs;
1230	//this picture can be added, find appropriate place in list and insert it.
1231	if(m_GOPList[offGOP].m_temporalId==m_GOPList[curGOP].m_temporalId)
1232	{
1233	m_GOPList[offGOP].m_refPic = true;
1234	}
1235	for(Int j=0; j<newRefs; j++)
1236	{
1237	if(m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[j]<offPOC-curPOC\|\|m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[j]>0)
1238	{
1239	insertPoint = j;
1240	break;
1241	}
1242	}
1243	Int prev = offPOC-curPOC;
1244	Int prevUsed = m_GOPList[offGOP].m_temporalId<=m_GOPList[curGOP].m_temporalId;
1245	for(Int j=insertPoint; j<newRefs+1; j++)
1246	{
1247	Int newPrev = m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[j];
1248	Int newUsed = m_GOPList[m_iGOPSize+m_extraRPSs].m_usedByCurrPic[j];
1249	m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[j]=prev;
1250	m_GOPList[m_iGOPSize+m_extraRPSs].m_usedByCurrPic[j]=prevUsed;
1251	prevUsed=newUsed;
1252	prev=newPrev;
1253	}
1254	newRefs++;
1255	}
1256	}
1257	if(newRefs>=numPrefRefs)
1258	{
1259	break;
1260	}
1261	}
1262	m_GOPList[m_iGOPSize+m_extraRPSs].m_numRefPics=newRefs;
1263	m_GOPList[m_iGOPSize+m_extraRPSs].m_POC = curPOC;
1264	if (m_extraRPSs == 0)
1265	{
1266	m_GOPList[m_iGOPSize+m_extraRPSs].m_interRPSPrediction = 0;
1267	m_GOPList[m_iGOPSize+m_extraRPSs].m_numRefIdc = 0;
1268	}
1269	else
1270	{
1271	Int rIdx = m_iGOPSize + m_extraRPSs - 1;
1272	Int refPOC = m_GOPList[rIdx].m_POC;
1273	Int refPics = m_GOPList[rIdx].m_numRefPics;
1274	Int newIdc=0;
1275	for(Int i = 0; i<= refPics; i++)
1276	{
1277	Int deltaPOC = ((i != refPics)? m_GOPList[rIdx].m_referencePics[i] : 0); // check if the reference abs POC is >= 0
1278	Int absPOCref = refPOC+deltaPOC;
1279	Int refIdc = 0;
1280	for (Int j = 0; j < m_GOPList[m_iGOPSize+m_extraRPSs].m_numRefPics; j++)
1281	{
1282	if ( (absPOCref - curPOC) == m_GOPList[m_iGOPSize+m_extraRPSs].m_referencePics[j])
1283	{
1284	if (m_GOPList[m_iGOPSize+m_extraRPSs].m_usedByCurrPic[j])
1285	{
1286	refIdc = 1;
1287	}
1288	else
1289	{
1290	refIdc = 2;
1291	}
1292	}
1293	}
1294	m_GOPList[m_iGOPSize+m_extraRPSs].m_refIdc[newIdc]=refIdc;
1295	newIdc++;
1296	}
1297	m_GOPList[m_iGOPSize+m_extraRPSs].m_interRPSPrediction = 1;
1298	m_GOPList[m_iGOPSize+m_extraRPSs].m_numRefIdc = newIdc;
1299	m_GOPList[m_iGOPSize+m_extraRPSs].m_deltaRPS = refPOC - m_GOPList[m_iGOPSize+m_extraRPSs].m_POC;
1300	}
1301	curGOP=m_iGOPSize+m_extraRPSs;
1302	m_extraRPSs++;
1303	}
1304	numRefs=0;
1305	for(Int i = 0; i< m_GOPList[curGOP].m_numRefPics; i++)
1306	{
1307	Int absPOC = curPOC+m_GOPList[curGOP].m_referencePics[i];
1308	if(absPOC >= 0)
1309	{
1310	refList[numRefs]=absPOC;
1311	numRefs++;
1312	}
1313	}
1314	refList[numRefs]=curPOC;
1315	numRefs++;
1316	}
1317	checkGOP++;
1318	}
1319	xConfirmPara(errorGOP,"Invalid GOP structure given");
1320	m_maxTempLayer = 1;
1321	for(Int i=0; i<m_iGOPSize; i++)
1322	{
1323	if(m_GOPList[i].m_temporalId >= m_maxTempLayer)
1324	{
1325	m_maxTempLayer = m_GOPList[i].m_temporalId+1;
1326	}
1327	xConfirmPara(m_GOPList[i].m_sliceType!='B'&&m_GOPList[i].m_sliceType!='P', "Slice type must be equal to B or P");
1328	}
1329	for(Int i=0; i<MAX_TLAYER; i++)
1330	{
1331	m_numReorderPics[i] = 0;
1332	m_maxDecPicBuffering[i] = 0;
1333	}
1334	for(Int i=0; i<m_iGOPSize; i++)
1335	{
1336	if(m_GOPList[i].m_numRefPics > m_maxDecPicBuffering[m_GOPList[i].m_temporalId])
1337	{
1338	m_maxDecPicBuffering[m_GOPList[i].m_temporalId] = m_GOPList[i].m_numRefPics;
1339	}
1340	Int highestDecodingNumberWithLowerPOC = 0;
1341	for(Int j=0; j<m_iGOPSize; j++)
1342	{
1343	if(m_GOPList[j].m_POC <= m_GOPList[i].m_POC)
1344	{
1345	highestDecodingNumberWithLowerPOC = j;
1346	}
1347	}
1348	Int numReorder = 0;
1349	for(Int j=0; j<highestDecodingNumberWithLowerPOC; j++)
1350	{
1351	if(m_GOPList[j].m_temporalId <= m_GOPList[i].m_temporalId &&
1352	m_GOPList[j].m_POC > m_GOPList[i].m_POC)
1353	{
1354	numReorder++;
1355	}
1356	}
1357	if(numReorder > m_numReorderPics[m_GOPList[i].m_temporalId])
1358	{
1359	m_numReorderPics[m_GOPList[i].m_temporalId] = numReorder;
1360	}
1361	}
1362	for(Int i=0; i<MAX_TLAYER-1; i++)
1363	{
1364	// a lower layer can not have higher value of m_numReorderPics than a higher layer
1365	if(m_numReorderPics[i+1] < m_numReorderPics[i])
1366	{
1367	m_numReorderPics[i+1] = m_numReorderPics[i];
1368	}
1369	// the value of num_reorder_pics[ i ] shall be in the range of 0 to max_dec_pic_buffering[ i ], inclusive
1370	if(m_numReorderPics[i] > m_maxDecPicBuffering[i])
1371	{
1372	m_maxDecPicBuffering[i] = m_numReorderPics[i];
1373	}
1374	// a lower layer can not have higher value of m_uiMaxDecPicBuffering than a higher layer
1375	if(m_maxDecPicBuffering[i+1] < m_maxDecPicBuffering[i])
1376	{
1377	m_maxDecPicBuffering[i+1] = m_maxDecPicBuffering[i];
1378	}
1379	}
1380	// the value of num_reorder_pics[ i ] shall be in the range of 0 to max_dec_pic_buffering[ i ], inclusive
1381	if(m_numReorderPics[MAX_TLAYER-1] > m_maxDecPicBuffering[MAX_TLAYER-1])
1382	{
1383	m_maxDecPicBuffering[MAX_TLAYER-1] = m_numReorderPics[MAX_TLAYER-1];
1384	}
1385
1386	#if SVC_EXTENSION // ToDo: it should be checked for the case when parameters are different for the layers
1387	for(UInt layer = 0; layer < MAX_LAYERS; layer++)
1388	{
1389	Int m_iSourceWidth = m_acLayerCfg[layer].m_iSourceWidth;
1390	Int m_iSourceHeight = m_acLayerCfg[layer].m_iSourceHeight;
1391	#endif
1392	if(m_vuiParametersPresentFlag && m_bitstreamRestrictionFlag)
1393	{
1394	Int PicSizeInSamplesY = m_iSourceWidth * m_iSourceHeight;
1395	if(tileFlag)
1396	{
1397	Int maxTileWidth = 0;
1398	Int maxTileHeight = 0;
1399	Int widthInCU = (m_iSourceWidth % m_uiMaxCUWidth) ? m_iSourceWidth/m_uiMaxCUWidth + 1: m_iSourceWidth/m_uiMaxCUWidth;
1400	Int heightInCU = (m_iSourceHeight % m_uiMaxCUHeight) ? m_iSourceHeight/m_uiMaxCUHeight + 1: m_iSourceHeight/m_uiMaxCUHeight;
1401	if(m_iUniformSpacingIdr)
1402	{
1403	maxTileWidth = m_uiMaxCUWidth*((widthInCU+m_iNumColumnsMinus1)/(m_iNumColumnsMinus1+1));
1404	maxTileHeight = m_uiMaxCUHeight*((heightInCU+m_iNumRowsMinus1)/(m_iNumRowsMinus1+1));
1405	// if only the last tile-row is one treeblock higher than the others
1406	// the maxTileHeight becomes smaller if the last row of treeblocks has lower height than the others
1407	if(!((heightInCU-1)%(m_iNumRowsMinus1+1)))
1408	{
1409	maxTileHeight = maxTileHeight - m_uiMaxCUHeight + (m_iSourceHeight % m_uiMaxCUHeight);
1410	}
1411	// if only the last tile-column is one treeblock wider than the others
1412	// the maxTileWidth becomes smaller if the last column of treeblocks has lower width than the others
1413	if(!((widthInCU-1)%(m_iNumColumnsMinus1+1)))
1414	{
1415	maxTileWidth = maxTileWidth - m_uiMaxCUWidth + (m_iSourceWidth % m_uiMaxCUWidth);
1416	}
1417	}
1418	else // not uniform spacing
1419	{
1420	if(m_iNumColumnsMinus1<1)
1421	{
1422	maxTileWidth = m_iSourceWidth;
1423	}
1424	else
1425	{
1426	Int accColumnWidth = 0;
1427	for(Int col=0; col<(m_iNumColumnsMinus1); col++)
1428	{
1429	maxTileWidth = m_pColumnWidth[col]>maxTileWidth ? m_pColumnWidth[col]:maxTileWidth;
1430	accColumnWidth += m_pColumnWidth[col];
1431	}
1432	maxTileWidth = (widthInCU-accColumnWidth)>maxTileWidth ? m_uiMaxCUWidth(widthInCU-accColumnWidth):m_uiMaxCUWidthmaxTileWidth;
1433	}
1434	if(m_iNumRowsMinus1<1)
1435	{
1436	maxTileHeight = m_iSourceHeight;
1437	}
1438	else
1439	{
1440	Int accRowHeight = 0;
1441	for(Int row=0; row<(m_iNumRowsMinus1); row++)
1442	{
1443	maxTileHeight = m_pRowHeight[row]>maxTileHeight ? m_pRowHeight[row]:maxTileHeight;
1444	accRowHeight += m_pRowHeight[row];
1445	}
1446	maxTileHeight = (heightInCU-accRowHeight)>maxTileHeight ? m_uiMaxCUHeight(heightInCU-accRowHeight):m_uiMaxCUHeightmaxTileHeight;
1447	}
1448	}
1449	Int maxSizeInSamplesY = maxTileWidth*maxTileHeight;
1450	m_minSpatialSegmentationIdc = 4*PicSizeInSamplesY/maxSizeInSamplesY-4;
1451	}
1452	else if(m_iWaveFrontSynchro)
1453	{
1454	m_minSpatialSegmentationIdc = 4PicSizeInSamplesY/((2m_iSourceHeight+m_iSourceWidth)*m_uiMaxCUHeight)-4;
1455	}
1456	else if(m_sliceMode == 1)
1457	{
1458	m_minSpatialSegmentationIdc = 4PicSizeInSamplesY/(m_sliceArgumentm_uiMaxCUWidth*m_uiMaxCUHeight)-4;
1459	}
1460	else
1461	{
1462	m_minSpatialSegmentationIdc = 0;
1463	}
1464	}
1465	#if SVC_EXTENSION
1466	}
1467	#endif
1468
1469	xConfirmPara( m_bUseLComb==false && m_numReorderPics[MAX_TLAYER-1]!=0, "ListCombination can only be 0 in low delay coding (more precisely when L0 and L1 are identical)" ); // Note however this is not the full necessary condition as ref_pic_list_combination_flag can only be 0 if L0 == L1.
1470	xConfirmPara( m_iWaveFrontSynchro < 0, "WaveFrontSynchro cannot be negative" );
1471	#if !SVC_EXTENSION
1472	xConfirmPara( m_iWaveFrontSubstreams <= 0, "WaveFrontSubstreams must be positive" );
1473	xConfirmPara( m_iWaveFrontSubstreams > 1 && !m_iWaveFrontSynchro, "Must have WaveFrontSynchro > 0 in order to have WaveFrontSubstreams > 1" );
1474	#endif
1475
1476	xConfirmPara( m_decodedPictureHashSEIEnabled<0 \|\| m_decodedPictureHashSEIEnabled>3, "this hash type is not correct!\n");
1477
1478	#if RATE_CONTROL_LAMBDA_DOMAIN
1479	if ( m_RCEnableRateControl )
1480	{
1481	if ( m_RCForceIntraQP )
1482	{
1483	if ( m_RCInitialQP == 0 )
1484	{
1485	printf( "\nInitial QP for rate control is not specified. Reset not to use force intra QP!" );
1486	m_RCForceIntraQP = false;
1487	}
1488	}
1489	xConfirmPara( m_uiDeltaQpRD > 0, "Rate control cannot be used together with slice level multiple-QP optimization!\n" );
1490	}
1491	#else
1492	if(m_enableRateCtrl)
1493	{
1494	Int numLCUInWidth = (m_iSourceWidth / m_uiMaxCUWidth) + (( m_iSourceWidth % m_uiMaxCUWidth ) ? 1 : 0);
1495	Int numLCUInHeight = (m_iSourceHeight / m_uiMaxCUHeight)+ (( m_iSourceHeight % m_uiMaxCUHeight) ? 1 : 0);
1496	Int numLCUInPic = numLCUInWidth * numLCUInHeight;
1497
1498	xConfirmPara( (numLCUInPic % m_numLCUInUnit) != 0, "total number of LCUs in a frame should be completely divided by NumLCUInUnit" );
1499
1500	m_iMaxDeltaQP = MAX_DELTA_QP;
1501	m_iMaxCuDQPDepth = MAX_CUDQP_DEPTH;
1502	}
1503	#endif
1504
1505	xConfirmPara(!m_TransquantBypassEnableFlag && m_CUTransquantBypassFlagValue, "CUTransquantBypassFlagValue cannot be 1 when TransquantBypassEnableFlag is 0");
1506
1507	xConfirmPara(m_log2ParallelMergeLevel < 2, "Log2ParallelMergeLevel should be larger than or equal to 2");
1508	#if L0444_FPA_TYPE
1509	if (m_framePackingSEIEnabled)
1510	{
1511	xConfirmPara(m_framePackingSEIType < 3 \|\| m_framePackingSEIType > 5 , "SEIFramePackingType must be in rage 3 to 5");
1512	}
1513	#endif
1514
1515	#undef xConfirmPara
1516	if (check_failed)
1517	{
1518	exit(EXIT_FAILURE);
1519	}
1520	}
1521
1522	/** \todo use of global variables should be removed later
1523	*/
1524	Void TAppEncCfg::xSetGlobal()
1525	{
1526	// set max CU width & height
1527	g_uiMaxCUWidth = m_uiMaxCUWidth;
1528	g_uiMaxCUHeight = m_uiMaxCUHeight;
1529
1530	// compute actual CU depth with respect to config depth and max transform size
1531	g_uiAddCUDepth = 0;
1532	while( (m_uiMaxCUWidth>>m_uiMaxCUDepth) > ( 1 << ( m_uiQuadtreeTULog2MinSize + g_uiAddCUDepth ) ) ) g_uiAddCUDepth++;
1533
1534	m_uiMaxCUDepth += g_uiAddCUDepth;
1535	g_uiAddCUDepth++;
1536	g_uiMaxCUDepth = m_uiMaxCUDepth;
1537
1538	// set internal bit-depth and constants
1539	g_bitDepthY = m_internalBitDepthY;
1540	g_bitDepthC = m_internalBitDepthC;
1541
1542	g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_inputBitDepthY : m_internalBitDepthY;
1543	g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_inputBitDepthC : m_internalBitDepthC;
1544	}
1545
1546	Void TAppEncCfg::xPrintParameter()
1547	{
1548	printf("\n");
1549	#if SVC_EXTENSION
1550	printf("Total number of layers : %d\n", m_numLayers );
1551	for(UInt layer=0; layer<m_numLayers; layer++)
1552	{
1553	printf("=== Layer %d settings === \n", layer);
1554	m_acLayerCfg[layer].xPrintParameter();
1555	printf("\n");
1556	}
1557	printf("=== Common configuration settings === \n");
1558	printf("Bitstream File : %s\n", m_pBitstreamFile );
1559	#else
1560	printf("Input File : %s\n", m_pchInputFile );
1561	printf("Bitstream File : %s\n", m_pchBitstreamFile );
1562	printf("Reconstruction File : %s\n", m_pchReconFile );
1563	printf("Real Format : %dx%d %dHz\n", m_iSourceWidth - m_confLeft - m_confRight, m_iSourceHeight - m_confTop - m_confBottom, m_iFrameRate );
1564	printf("Internal Format : %dx%d %dHz\n", m_iSourceWidth, m_iSourceHeight, m_iFrameRate );
1565	#endif
1566	printf("Frame index : %u - %d (%d frames)\n", m_FrameSkip, m_FrameSkip+m_framesToBeEncoded-1, m_framesToBeEncoded );
1567	printf("CU size / depth : %d / %d\n", m_uiMaxCUWidth, m_uiMaxCUDepth );
1568	printf("RQT trans. size (min / max) : %d / %d\n", 1 << m_uiQuadtreeTULog2MinSize, 1 << m_uiQuadtreeTULog2MaxSize );
1569	printf("Max RQT depth inter : %d\n", m_uiQuadtreeTUMaxDepthInter);
1570	printf("Max RQT depth intra : %d\n", m_uiQuadtreeTUMaxDepthIntra);
1571	printf("Min PCM size : %d\n", 1 << m_uiPCMLog2MinSize);
1572	printf("Motion search range : %d\n", m_iSearchRange );
1573	#if !SVC_EXTENSION
1574	printf("Intra period : %d\n", m_iIntraPeriod );
1575	#endif
1576	printf("Decoding refresh type : %d\n", m_iDecodingRefreshType );
1577	#if !SVC_EXTENSION
1578	printf("QP : %5.2f\n", m_fQP );
1579	#endif
1580	printf("Max dQP signaling depth : %d\n", m_iMaxCuDQPDepth);
1581
1582	printf("Cb QP Offset : %d\n", m_cbQpOffset );
1583	printf("Cr QP Offset : %d\n", m_crQpOffset);
1584
1585	printf("QP adaptation : %d (range=%d)\n", m_bUseAdaptiveQP, (m_bUseAdaptiveQP ? m_iQPAdaptationRange : 0) );
1586	printf("GOP size : %d\n", m_iGOPSize );
1587	printf("Internal bit depth : (Y:%d, C:%d)\n", m_internalBitDepthY, m_internalBitDepthC );
1588	printf("PCM sample bit depth : (Y:%d, C:%d)\n", g_uiPCMBitDepthLuma, g_uiPCMBitDepthChroma );
1589	#if RATE_CONTROL_LAMBDA_DOMAIN
1590	printf("RateControl : %d\n", m_RCEnableRateControl );
1591	if(m_RCEnableRateControl)
1592	{
1593	printf("TargetBitrate : %d\n", m_RCTargetBitrate );
1594	printf("KeepHierarchicalBit : %d\n", m_RCKeepHierarchicalBit );
1595	printf("LCULevelRC : %d\n", m_RCLCULevelRC );
1596	printf("UseLCUSeparateModel : %d\n", m_RCUseLCUSeparateModel );
1597	printf("InitialQP : %d\n", m_RCInitialQP );
1598	printf("ForceIntraQP : %d\n", m_RCForceIntraQP );
1599	}
1600	#else
1601	printf("RateControl : %d\n", m_enableRateCtrl);
1602	if(m_enableRateCtrl)
1603	{
1604	printf("TargetBitrate : %d\n", m_targetBitrate);
1605	printf("NumLCUInUnit : %d\n", m_numLCUInUnit);
1606	}
1607	#endif
1608	printf("Max Num Merge Candidates : %d\n", m_maxNumMergeCand);
1609	printf("\n");
1610
1611	printf("TOOL CFG: ");
1612	printf("IBD:%d ", g_bitDepthY > m_inputBitDepthY \|\| g_bitDepthC > m_inputBitDepthC);
1613	printf("HAD:%d ", m_bUseHADME );
1614	printf("SRD:%d ", m_bUseSBACRD );
1615	printf("RDQ:%d ", m_useRDOQ );
1616	printf("RDQTS:%d ", m_useRDOQTS );
1617	#if L0232_RD_PENALTY
1618	printf("RDpenalty:%d ", m_rdPenalty );
1619	#endif
1620	printf("SQP:%d ", m_uiDeltaQpRD );
1621	printf("ASR:%d ", m_bUseASR );
1622	printf("LComb:%d ", m_bUseLComb );
1623	printf("FEN:%d ", m_bUseFastEnc );
1624	printf("ECU:%d ", m_bUseEarlyCU );
1625	printf("FDM:%d ", m_useFastDecisionForMerge );
1626	printf("CFM:%d ", m_bUseCbfFastMode );
1627	printf("ESD:%d ", m_useEarlySkipDetection );
1628	printf("RQT:%d ", 1 );
1629	printf("TransformSkip:%d ", m_useTransformSkip );
1630	printf("TransformSkipFast:%d ", m_useTransformSkipFast );
1631	printf("Slice: M=%d ", m_sliceMode);
1632	if (m_sliceMode!=0)
1633	{
1634	printf("A=%d ", m_sliceArgument);
1635	}
1636	printf("SliceSegment: M=%d ",m_sliceSegmentMode);
1637	if (m_sliceSegmentMode!=0)
1638	{
1639	printf("A=%d ", m_sliceSegmentArgument);
1640	}
1641	printf("CIP:%d ", m_bUseConstrainedIntraPred);
1642	printf("SAO:%d ", (m_bUseSAO)?(1):(0));
1643	printf("PCM:%d ", (m_usePCM && (1<<m_uiPCMLog2MinSize) <= m_uiMaxCUWidth)? 1 : 0);
1644	printf("SAOLcuBasedOptimization:%d ", (m_saoLcuBasedOptimization)?(1):(0));
1645
1646	printf("LosslessCuEnabled:%d ", (m_useLossless)? 1:0 );
1647	printf("WPP:%d ", (Int)m_useWeightedPred);
1648	printf("WPB:%d ", (Int)m_useWeightedBiPred);
1649	printf("PME:%d ", m_log2ParallelMergeLevel);
1650	#if !SVC_EXTENSION
1651	printf(" WaveFrontSynchro:%d WaveFrontSubstreams:%d",
1652	m_iWaveFrontSynchro, m_iWaveFrontSubstreams);
1653	#endif
1654	printf(" ScalingList:%d ", m_useScalingListId );
1655	printf("TMVPMode:%d ", m_TMVPModeId );
1656	#if ADAPTIVE_QP_SELECTION
1657	printf("AQpS:%d", m_bUseAdaptQpSelect );
1658	#endif
1659
1660	printf(" SignBitHidingFlag:%d ", m_signHideFlag);
1661	#if SVC_EXTENSION
1662	printf("RecalQP:%d ", m_recalculateQPAccordingToLambda ? 1 : 0 );
1663	printf("AVC_BASE:%d ", AVC_BASE);
1664	#if REF_IDX_FRAMEWORK
1665	printf("REF_IDX_FRAMEWORK:%d ", REF_IDX_FRAMEWORK);
1666	printf("EL_RAP_SliceType: %d ", m_elRapSliceBEnabled);
1667	printf("REF_IDX_ME_AROUND_ZEROMV:%d ", REF_IDX_ME_AROUND_ZEROMV);
1668	printf("REF_IDX_ME_ZEROMV: %d", REF_IDX_ME_ZEROMV);
1669	#elif INTRA_BL
1670	printf("INTRA_BL:%d ", INTRA_BL);
1671	#if !AVC_BASE
1672	printf("SVC_MVP:%d ", SVC_MVP );
1673	printf("SVC_BL_CAND_INTRA:%d", SVC_BL_CAND_INTRA );
1674	#endif
1675	#endif
1676	#else
1677	printf("RecalQP:%d", m_recalculateQPAccordingToLambda ? 1 : 0 );
1678	#endif
1679	printf("\n\n");
1680
1681	fflush(stdout);
1682	}
1683
1684	Bool confirmPara(Bool bflag, const Char* message)
1685	{
1686	if (!bflag)
1687	return false;
1688
1689	printf("Error: %s\n",message);
1690	return true;
1691	}
1692
1693	//! \}

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

Download in other formats: