1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
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, but always including the *additional*
9 // restrictions defined in the "License.txt" file.
11 // This program is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 // GNU General Public License for more details.
16 // You should have received a copy of the GNU General Public License along
17 // with this program; if not, write to the Free Software Foundation, Inc.,
18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 // http://www.gnu.org/licenses/gpl-2.0.txt
21 ///////////////////////////////////////////////////////////////////////////////
23 #include "Encoder_Opus.h"
26 #include <MUtils/Global.h>
30 #include "Model_Settings.h"
31 #include "MimeTypes.h"
38 ///////////////////////////////////////////////////////////////////////////////
40 ///////////////////////////////////////////////////////////////////////////////
42 class OpusEncoderInfo
: public AbstractEncoderInfo
44 virtual bool isModeSupported(int mode
) const
48 case SettingsModel::VBRMode
:
49 case SettingsModel::ABRMode
:
50 case SettingsModel::CBRMode
:
54 MUTILS_THROW("Bad RC mode specified!");
58 virtual int valueCount(int mode
) const
62 case SettingsModel::VBRMode
:
63 case SettingsModel::ABRMode
:
64 case SettingsModel::CBRMode
:
68 MUTILS_THROW("Bad RC mode specified!");
72 virtual int valueAt(int mode
, int index
) const
76 case SettingsModel::VBRMode
:
77 case SettingsModel::ABRMode
:
78 case SettingsModel::CBRMode
:
79 return qBound(8, (index
+ 1) * 8, 256);
82 MUTILS_THROW("Bad RC mode specified!");
86 virtual int valueType(int mode
) const
90 case SettingsModel::VBRMode
:
91 case SettingsModel::ABRMode
:
92 return TYPE_APPROX_BITRATE
;
94 case SettingsModel::CBRMode
:
98 MUTILS_THROW("Bad RC mode specified!");
102 virtual const char *description(void) const
104 static const char* s_description
= "Opus-Tools OpusEnc (libopus)";
105 return s_description
;
108 virtual const char *extension(void) const
110 static const char* s_extension
= "opus";
114 virtual bool isResamplingSupported(void) const
119 static const g_opusEncoderInfo
;
121 ///////////////////////////////////////////////////////////////////////////////
122 // Encoder implementation
123 ///////////////////////////////////////////////////////////////////////////////
125 OpusEncoder::OpusEncoder(void)
127 m_binary(lamexp_tools_lookup(L1S("opusenc.exe")))
129 if(m_binary
.isEmpty())
131 MUTILS_THROW("Error initializing Opus encoder. Tool 'opusenc.exe' is not registred!");
134 m_configOptimizeFor
= 0;
135 m_configEncodeComplexity
= 10;
136 m_configFrameSize
= 3;
139 OpusEncoder::~OpusEncoder(void)
143 bool OpusEncoder::encode(const QString
&sourceFile
, const AudioFileModel_MetaInfo
&metaInfo
, const unsigned int duration
, const unsigned int channels
, const QString
&outputFile
, QAtomicInt
&abortFlag
)
148 switch(m_configRCMode
)
150 case SettingsModel::VBRMode
:
151 args
<< L1S("--vbr");
153 case SettingsModel::ABRMode
:
154 args
<< L1S("--cvbr");
156 case SettingsModel::CBRMode
:
157 args
<< L1S("--hard-cbr");
160 MUTILS_THROW("Bad rate-control mode!");
164 args
<< "--comp" << QString::number(m_configEncodeComplexity
);
166 switch(m_configFrameSize
)
169 args
<< L1S("--framesize") << L1S("2.5");
172 args
<< L1S("--framesize") << L1S("5");
175 args
<< L1S("--framesize") << L1S("10");
178 args
<< L1S("--framesize") << L1S("20");
181 args
<< L1S("--framesize") << L1S("40");
184 args
<< L1S("--framesize") << L1S("60");
188 args
<< L1S("--bitrate") << QString::number(qBound(8, (m_configBitrate
+ 1) * 8, 256));
190 if(!metaInfo
.title().isEmpty()) args
<< L1S("--title") << cleanTag(metaInfo
.title());
191 if(!metaInfo
.artist().isEmpty()) args
<< L1S("--artist") << cleanTag(metaInfo
.artist());
192 if(!metaInfo
.album().isEmpty()) args
<< L1S("--album") << cleanTag(metaInfo
.album());
193 if(!metaInfo
.genre().isEmpty()) args
<< L1S("--genre") << cleanTag(metaInfo
.genre());
194 if(metaInfo
.year()) args
<< L1S("--date") << QString::number(metaInfo
.year());
195 if(metaInfo
.position()) args
<< L1S("--comment") << QString("tracknumber=%1").arg(QString::number(metaInfo
.position()));
196 if(!metaInfo
.comment().isEmpty()) args
<< L1S("--comment") << QString("comment=%1").arg(cleanTag(metaInfo
.comment()));
197 if(!metaInfo
.cover().isEmpty()) args
<< L1S("--picture") << makeCoverParam(metaInfo
.cover());
199 if(!m_configCustomParams
.isEmpty()) args
<< m_configCustomParams
.split(" ", QString::SkipEmptyParts
);
201 args
<< QDir::toNativeSeparators(sourceFile
);
202 args
<< QDir::toNativeSeparators(outputFile
);
204 if(!startProcess(process
, m_binary
, args
))
209 bool bTimeout
= false;
210 bool bAborted
= false;
211 int prevProgress
= -1;
213 QRegExp
regExp(L1S("\\((\\d+)%\\)"));
215 while(process
.state() != QProcess::NotRunning
)
217 if (checkFlag(abortFlag
))
221 emit
messageLogged(L1S("\nABORTED BY USER !!!"));
224 process
.waitForReadyRead(m_processTimeoutInterval
);
225 if(!process
.bytesAvailable() && process
.state() == QProcess::Running
)
228 qWarning("Opus process timed out <-- killing!");
229 emit
messageLogged(L1S("\nPROCESS TIMEOUT !!!"));
233 while(process
.bytesAvailable() > 0)
235 QByteArray line
= process
.readLine();
236 QString text
= QString::fromUtf8(line
.constData()).simplified();
237 if(regExp
.lastIndexIn(text
) >= 0)
240 int progress
= regExp
.cap(1).toInt(&ok
);
241 if(ok
&& (progress
> prevProgress
))
243 emit
statusUpdated(progress
);
244 prevProgress
= qMin(progress
+ 2, 99);
247 else if(!text
.isEmpty())
249 emit
messageLogged(text
);
254 process
.waitForFinished();
255 if(process
.state() != QProcess::NotRunning
)
258 process
.waitForFinished(-1);
261 emit
statusUpdated(100);
262 emit
messageLogged(QString().sprintf("\nExited with code: 0x%04X", process
.exitCode()));
264 if(bTimeout
|| bAborted
|| process
.exitCode() != EXIT_SUCCESS
)
272 QString
OpusEncoder::detectMimeType(const QString
&coverFile
)
274 const QString suffix
= QFileInfo(coverFile
).suffix();
275 for (size_t i
= 0; MIME_TYPES
[i
].type
; i
++)
277 for (size_t k
= 0; MIME_TYPES
[i
].ext
[k
]; k
++)
279 if (suffix
.compare(QString::fromLatin1(MIME_TYPES
[i
].ext
[k
]), Qt::CaseInsensitive
) == 0)
281 return QString::fromLatin1(MIME_TYPES
[i
].type
);
286 qWarning("Unknown MIME type for extension '%s' -> using default!", MUTILS_UTF8(coverFile
));
287 return QString::fromLatin1(MIME_TYPES
[0].type
);
290 QString
OpusEncoder::makeCoverParam(const QString
&coverFile
)
292 return QString("3|%1|||%2").arg(detectMimeType(coverFile
), QDir::toNativeSeparators(coverFile
));
295 void OpusEncoder::setOptimizeFor(int optimizeFor
)
297 m_configOptimizeFor
= qBound(0, optimizeFor
, 2);
300 void OpusEncoder::setEncodeComplexity(int complexity
)
302 m_configEncodeComplexity
= qBound(0, complexity
, 10);
305 void OpusEncoder::setFrameSize(int frameSize
)
307 m_configFrameSize
= qBound(0, frameSize
, 5);
310 bool OpusEncoder::isFormatSupported(const QString
&containerType
, const QString
&containerProfile
, const QString
&formatType
, const QString
&formatProfile
, const QString
&formatVersion
)
312 if(containerType
.compare(L1S("Wave"), Qt::CaseInsensitive
) == 0)
314 if(formatType
.compare(L1S("PCM"), Qt::CaseInsensitive
) == 0)
323 const unsigned int *OpusEncoder::supportedChannelCount(void)
328 const unsigned int *OpusEncoder::supportedBitdepths(void)
330 static const unsigned int supportedBPS
[] = {8, 16, 24, AudioFileModel::BITDEPTH_IEEE_FLOAT32
, NULL
};
334 const bool OpusEncoder::needsTimingInfo(void)
339 const AbstractEncoderInfo
*OpusEncoder::getEncoderInfo(void)
341 return &g_opusEncoderInfo
;