1 ///////////////////////////////////////////////////////////////////////////////
2 // Simple x264 Launcher
3 // Copyright (C) 2004-2017 LoRd_MuldeR <MuldeR2@GMX.de>
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
8 // (at your option) any later version.
10 // This program is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 // GNU General Public License for more details.
15 // You should have received a copy of the GNU General Public License along
16 // with this program; if not, write to the Free Software Foundation, Inc.,
17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 // http://www.gnu.org/licenses/gpl-2.0.txt
20 ///////////////////////////////////////////////////////////////////////////////
22 #include "encoder_nvencc.h"
26 #include "model_options.h"
27 #include "model_status.h"
28 #include "mediainfo.h"
29 #include "model_sysinfo.h"
30 #include "model_clipInfo.h"
33 #include <MUtils/Global.h>
34 #include <MUtils/Exception.h>
37 #include <QStringList>
43 static const unsigned int VERSION_NVENCC_MINIMUM_VER
= 307;
44 static const unsigned int VERSION_NVENCC_MINIMUM_API
= 71;
46 // ------------------------------------------------------------
48 // ------------------------------------------------------------
50 #define NVENCC_UPDATE_PROGRESS(X) do \
53 unsigned int progressFrames = (X)->cap(1).toUInt(&ok); \
54 double progress = 0.0; \
55 setStatus(JobStatus_Running); \
56 if(ok && (clipInfo.getFrameCount() > 0)) \
58 progress = (double(progressFrames) / double(clipInfo.getFrameCount())); \
59 if(!qFuzzyCompare(progress, last_progress)) \
61 setProgress(floor(progress * 100.0)); \
62 size_estimate = qFuzzyIsNull(size_estimate) ? estimateSize(m_outputFile, progress) : ((0.667 * size_estimate) + (0.333 * estimateSize(m_outputFile, progress))); \
63 last_progress = progress; \
66 setDetails(tr("[%1] %2, est. file size %3").arg(QString().sprintf("%.1f%%", 100.0 * progress), line.mid(offset).trimmed(), sizeToString(qRound64(size_estimate)))); \
70 #define REMOVE_CUSTOM_ARG(LIST, ITER, FLAG, PARAM) do \
72 if(ITER != LIST.end()) \
74 if((*ITER).compare(PARAM, Qt::CaseInsensitive) == 0) \
76 log(tr("WARNING: Custom parameter \"" PARAM "\" will be ignored in Pipe'd mode!\n")); \
77 ITER = LIST.erase(ITER); \
78 if(ITER != LIST.end()) \
80 if(!((*ITER).startsWith("--", Qt::CaseInsensitive))) ITER = LIST.erase(ITER); \
88 // ------------------------------------------------------------
90 // ------------------------------------------------------------
92 class NVEncEncoderInfo
: public AbstractEncoderInfo
95 virtual QString
getName(void) const
100 virtual QList
<ArchId
> getArchitectures(void) const
102 return QList
<ArchId
>()
103 << qMakePair(QString("32-Bit (x86)"), ARCH_TYPE_X86
)
104 << qMakePair(QString("64-Bit (x64)"), ARCH_TYPE_X64
);
107 virtual QStringList
getVariants(void) const
109 return QStringList() << "AVC" << "HEVC";
112 virtual QList
<RCMode
> getRCModes(void) const
114 return QList
<RCMode
>()
115 << qMakePair(QString("CQP"), RC_TYPE_QUANTIZER
)
116 << qMakePair(QString("VBR"), RC_TYPE_RATE_KBPS
)
117 << qMakePair(QString("VBR2"), RC_TYPE_RATE_KBPS
)
118 << qMakePair(QString("CBR"), RC_TYPE_RATE_KBPS
);
121 virtual QStringList
getTunings(void) const
123 return QStringList();
126 virtual QStringList
getPresets(void) const
128 return QStringList();
131 virtual QStringList
getProfiles(const quint32
&variant
) const
133 QStringList profiles
;
136 case 0: profiles
<< "baseline" << "main" << "high"; break;
137 case 1: profiles
<< "main"; break;
138 default: MUTILS_THROW("Unknown encoder variant!");
143 virtual QStringList
supportedOutputFormats(void) const
146 extLst
<< "mp4" << "264" << "hevc";
150 virtual bool isInputTypeSupported(const int format
) const
154 case MediaInfo::FILETYPE_YUV4MPEG2
:
161 virtual QString
getBinaryPath(const SysinfoModel
*sysinfo
, const quint32
&encArch
, const quint32
&encVariant
) const
166 case 0: arch
= "x86"; break;
167 case 1: arch
= "x64"; break;
168 default: MUTILS_THROW("Unknown encoder arch!");
174 default: MUTILS_THROW("Unknown encoder variant!");
176 return QString("%1/toolset/%2/nvencc_%2.exe").arg(sysinfo
->getAppPath(), arch
);
179 virtual QStringList
getDependencies(const SysinfoModel
*sysinfo
, const quint32
&encArch
, const quint32
&encVariant
) const
184 case 0: arch
= "x86"; break;
185 case 1: arch
= "x64"; break;
186 default: MUTILS_THROW("Unknown encoder arch!");
192 default: MUTILS_THROW("Unknown encoder variant!");
196 << QString("%1/toolset/%2/avcodec-57.dll" ).arg(sysinfo
->getAppPath(), arch
)
197 << QString("%1/toolset/%2/avfilter-6.dll" ).arg(sysinfo
->getAppPath(), arch
)
198 << QString("%1/toolset/%2/avformat-57.dll" ).arg(sysinfo
->getAppPath(), arch
)
199 << QString("%1/toolset/%2/avutil-55.dll" ).arg(sysinfo
->getAppPath(), arch
)
200 << QString("%1/toolset/%2/swresample-2.dll").arg(sysinfo
->getAppPath(), arch
);
203 virtual QString
getHelpCommand(void) const
209 static const NVEncEncoderInfo s_nvencEncoderInfo
;
211 const AbstractEncoderInfo
& NVEncEncoder::encoderInfo(void)
213 return s_nvencEncoderInfo
;
216 const AbstractEncoderInfo
&NVEncEncoder::getEncoderInfo(void) const
218 return encoderInfo();
221 // ------------------------------------------------------------
222 // Constructor & Destructor
223 // ------------------------------------------------------------
225 NVEncEncoder::NVEncEncoder(JobObject
*jobObject
, const OptionsModel
*options
, const SysinfoModel
*const sysinfo
, const PreferencesModel
*const preferences
, JobStatus
&jobStatus
, volatile bool *abort
, volatile bool *pause
, QSemaphore
*semaphorePause
, const QString
&sourceFile
, const QString
&outputFile
)
227 AbstractEncoder(jobObject
, options
, sysinfo
, preferences
, jobStatus
, abort
, pause
, semaphorePause
, sourceFile
, outputFile
)
229 if(options
->encType() != OptionsModel::EncType_NVEnc
)
231 MUTILS_THROW("Invalid encoder type!");
235 NVEncEncoder::~NVEncEncoder(void)
237 /*Nothing to do here*/
240 QString
NVEncEncoder::getName(void) const
242 return s_nvencEncoderInfo
.getFullName(m_options
->encArch(), m_options
->encVariant());
245 // ------------------------------------------------------------
247 // ------------------------------------------------------------
249 void NVEncEncoder::checkVersion_init(QList
<QRegExp
*> &patterns
, QStringList
&cmdLine
)
251 cmdLine
<< "--version";
252 patterns
<< new QRegExp("\\bNVEncC\\s+\\(x\\d+\\)\\s+(\\d)\\.(\\d+).*\\[NVENC\\s+API\\s+v(\\d+)\\.(\\d+)[^\\d]+", Qt::CaseInsensitive
);
255 void NVEncEncoder::checkVersion_parseLine(const QString
&line
, const QList
<QRegExp
*> &patterns
, unsigned int &core
, unsigned int &build
, bool &modified
)
257 if(patterns
[0]->lastIndexIn(line
) >= 0)
259 unsigned int temp
[4];
260 if(MUtils::regexp_parse_uint32(*patterns
[0], temp
, 4))
262 core
= (100 * temp
[0]) + temp
[1];
263 build
= (10 * temp
[2]) + temp
[3];
273 bool NVEncEncoder::checkVersion_succeeded(const int &exitCode
)
275 return (exitCode
== 0) || (exitCode
== 1);
278 QString
NVEncEncoder::printVersion(const unsigned int &revision
, const bool &modified
)
280 unsigned int core
, build
;
281 splitRevision(revision
, core
, build
);
283 return tr("NVEncC version: %1.%2 [API: %3.%4]").arg(QString::number(core
/ 100), QString::number(core
% 100).leftJustified(2, QLatin1Char('0')), QString::number(build
/ 10), QString::number(build
% 10));
286 bool NVEncEncoder::isVersionSupported(const unsigned int &revision
, const bool &modified
)
288 unsigned int core
, build
;
289 splitRevision(revision
, core
, build
);
291 if(core
< VERSION_NVENCC_MINIMUM_VER
)
293 log(tr("\nERROR: Your version of NVEncC is too old! (Minimum required version is %1.%2)").arg(QString::number(VERSION_NVENCC_MINIMUM_VER
/ 100), QString::number(VERSION_NVENCC_MINIMUM_VER
% 100)));
296 else if(core
> VERSION_NVENCC_MINIMUM_VER
)
298 log(tr("\nWARNING: Your version of NVEncC is newer than the latest tested version, take care!"));
299 log(tr("This application works best with NVEncC version %1.%2. Newer versions may work or not.").arg(QString::number(VERSION_NVENCC_MINIMUM_VER
/ 100), QString::number(VERSION_NVENCC_MINIMUM_VER
% 100)));
302 if (build
< VERSION_NVENCC_MINIMUM_API
)
304 log(tr("\nERROR: Your version of NVENC API is too old! (Minimum required version is %1.%2)").arg(QString::number(VERSION_NVENCC_MINIMUM_API
/ 10), QString::number(VERSION_NVENCC_MINIMUM_API
% 10)));
311 // ------------------------------------------------------------
312 // Encoding Functions
313 // ------------------------------------------------------------
315 void NVEncEncoder::buildCommandLine(QStringList
&cmdLine
, const bool &usePipe
, const ClipInfo
&clipInfo
, const QString
&indexFile
, const int &pass
, const QString
&passLogFile
)
317 switch (m_options
->encVariant())
320 cmdLine
<< "--codec" << "avc";
323 cmdLine
<< "--codec" << "hevc";
326 MUTILS_THROW("Bad encoder variant !!!");
329 switch(m_options
->rcMode())
332 cmdLine
<< "--cqp" << QString::number(qRound(m_options
->quantizer()));
335 cmdLine
<< "--vbr" << QString::number(m_options
->bitrate());
338 cmdLine
<< "--vbr2" << QString::number(m_options
->bitrate());
341 cmdLine
<< "--cbr" << QString::number(m_options
->bitrate());
344 MUTILS_THROW("Bad rate-control mode !!!");
347 const QString profile
= m_options
->profile().simplified().toLower();
348 if(!profile
.isEmpty())
350 if(profile
.compare(QString::fromLatin1(OptionsModel::PROFILE_UNRESTRICTED
), Qt::CaseInsensitive
) != 0)
352 cmdLine
<< "--profile" << profile
;
356 if(!m_options
->customEncParams().isEmpty())
358 QStringList customArgs
= splitParams(m_options
->customEncParams(), m_sourceFile
, m_outputFile
);
361 QStringList::iterator i
= customArgs
.begin();
362 while(i
!= customArgs
.end())
364 bool bModified
= false;
365 REMOVE_CUSTOM_ARG(customArgs
, i
, bModified
, "--fps");
366 REMOVE_CUSTOM_ARG(customArgs
, i
, bModified
, "--frames");
370 cmdLine
.append(customArgs
);
373 cmdLine
<< "--output" << QDir::toNativeSeparators(m_outputFile
);
377 cmdLine
<< "--y4m" << "--input" << "-";
381 cmdLine
<< "--input" << QDir::toNativeSeparators(m_sourceFile
);
385 void NVEncEncoder::runEncodingPass_init(QList
<QRegExp
*> &patterns
)
387 patterns
<< new QRegExp("^(\\d+) frames:");
388 patterns
<< new QRegExp("Selected\\s+codec\\s+is\\s+not\\s+supported", Qt::CaseInsensitive
);
389 patterns
<< new QRegExp("nvEncodeAPI.dll\\s+does\\s+not\\s+exists\\s+in\\s+your\\s+system", Qt::CaseInsensitive
);
392 void NVEncEncoder::runEncodingPass_parseLine(const QString
&line
, const QList
<QRegExp
*> &patterns
, const ClipInfo
&clipInfo
, const int &pass
, double &last_progress
, double &size_estimate
)
395 if((offset
= patterns
[0]->lastIndexIn(line
)) >= 0)
397 NVENCC_UPDATE_PROGRESS(patterns
[0]);
399 else if ((offset
= patterns
[1]->lastIndexIn(line
)) >= 0)
401 log(QString("ERROR: YOUR HARDWARE DOES *NOT* SUPPORT THE '%1' CODEC !!!\n").arg(s_nvencEncoderInfo
.variantToString(m_options
->encVariant())));
403 else if ((offset
= patterns
[2]->lastIndexIn(line
)) >= 0)
405 log("ERROR: NVIDIA ENCODER API (NVENCODEAPI.DLL) IS *NOT* AVAILABLE !!!\n");
407 else if(!line
.isEmpty())