1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2014 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_MP3.h"
26 #include "Model_Settings.h"
32 static const int g_lameAgorithmQualityLUT
[5] = {7, 5, 2, 0, INT_MAX
};
33 static const int g_mp3BitrateLUT
[15] = {32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, -1};
34 static const int g_lameVBRQualityLUT
[11] = {9, 8, 7, 6, 5, 4, 3, 2, 1, 0, INT_MAX
};
36 ///////////////////////////////////////////////////////////////////////////////
38 ///////////////////////////////////////////////////////////////////////////////
40 class MP3EncoderInfo
: public AbstractEncoderInfo
42 virtual bool isModeSupported(int mode
) const
46 case SettingsModel::VBRMode
:
47 case SettingsModel::ABRMode
:
48 case SettingsModel::CBRMode
:
52 THROW("Bad RC mode specified!");
56 virtual int valueCount(int mode
) const
60 case SettingsModel::VBRMode
:
63 case SettingsModel::ABRMode
:
64 case SettingsModel::CBRMode
:
68 THROW("Bad RC mode specified!");
72 virtual int valueAt(int mode
, int index
) const
76 case SettingsModel::VBRMode
:
77 return g_lameVBRQualityLUT
[qBound(0, index
, 9)];
79 case SettingsModel::ABRMode
:
80 case SettingsModel::CBRMode
:
81 return g_mp3BitrateLUT
[qBound(0, index
, 13)];
84 THROW("Bad RC mode specified!");
88 virtual int valueType(int mode
) const
92 case SettingsModel::VBRMode
:
93 return TYPE_QUALITY_LEVEL_INT
;
95 case SettingsModel::ABRMode
:
96 return TYPE_APPROX_BITRATE
;
98 case SettingsModel::CBRMode
:
102 THROW("Bad RC mode specified!");
106 virtual const char *description(void) const
108 static const char* s_description
= "LAME MP3 Encoder";
109 return s_description
;
112 static const g_mp3EncoderInfo
;
114 ///////////////////////////////////////////////////////////////////////////////
115 // Encoder implementation
116 ///////////////////////////////////////////////////////////////////////////////
118 MP3Encoder::MP3Encoder(void)
120 m_binary(lamexp_lookup_tool("lame.exe"))
122 if(m_binary
.isEmpty())
124 THROW("Error initializing MP3 encoder. Tool 'lame.exe' is not registred!");
127 m_algorithmQuality
= 2;
128 m_configBitrateMaximum
= 0;
129 m_configBitrateMinimum
= 0;
130 m_configSamplingRate
= 0;
131 m_configChannelMode
= 0;
134 MP3Encoder::~MP3Encoder(void)
138 bool MP3Encoder::encode(const QString
&sourceFile
, const AudioFileModel_MetaInfo
&metaInfo
, const unsigned int duration
, const QString
&outputFile
, volatile bool *abortFlag
)
144 args
<< "-q" << QString::number(g_lameAgorithmQualityLUT
[m_algorithmQuality
]);
146 switch(m_configRCMode
)
148 case SettingsModel::VBRMode
:
149 args
<< "-V" << QString::number(g_lameVBRQualityLUT
[qBound(0, m_configBitrate
, 9)]);
151 case SettingsModel::ABRMode
:
152 args
<< "--abr" << QString::number(g_mp3BitrateLUT
[qBound(0, m_configBitrate
, 13)]);
154 case SettingsModel::CBRMode
:
156 args
<< "-b" << QString::number(g_mp3BitrateLUT
[qBound(0, m_configBitrate
, 13)]);
159 THROW("Bad rate-control mode!");
163 if((m_configBitrateMaximum
> 0) && (m_configBitrateMinimum
> 0) && (m_configBitrateMinimum
<= m_configBitrateMaximum
))
165 if(m_configRCMode
!= SettingsModel::CBRMode
)
167 args
<< "-b" << QString::number(clipBitrate(m_configBitrateMinimum
));
168 args
<< "-B" << QString::number(clipBitrate(m_configBitrateMaximum
));
172 if(m_configSamplingRate
> 0)
174 args
<< "--resample" << QString::number(m_configSamplingRate
);
177 switch(m_configChannelMode
)
196 bool bUseUCS2
= false;
198 if(!metaInfo
.title().isEmpty() && isUnicode(metaInfo
.title())) bUseUCS2
= true;
199 if(!metaInfo
.artist().isEmpty() && isUnicode(metaInfo
.artist())) bUseUCS2
= true;
200 if(!metaInfo
.album().isEmpty() && isUnicode(metaInfo
.album())) bUseUCS2
= true;
201 if(!metaInfo
.genre().isEmpty() && isUnicode(metaInfo
.genre())) bUseUCS2
= true;
202 if(!metaInfo
.comment().isEmpty() && isUnicode(metaInfo
.comment())) bUseUCS2
= true;
204 if(bUseUCS2
) args
<< "--id3v2-ucs2"; //Must specify this BEFORE "--tt" and friends!
206 if(!metaInfo
.title().isEmpty()) args
<< "--tt" << cleanTag(metaInfo
.title());
207 if(!metaInfo
.artist().isEmpty()) args
<< "--ta" << cleanTag(metaInfo
.artist());
208 if(!metaInfo
.album().isEmpty()) args
<< "--tl" <<cleanTag( metaInfo
.album());
209 if(!metaInfo
.genre().isEmpty()) args
<< "--tg" << cleanTag(metaInfo
.genre());
210 if(!metaInfo
.comment().isEmpty()) args
<< "--tc" << cleanTag(metaInfo
.comment());
211 if(metaInfo
.year()) args
<< "--ty" << QString::number(metaInfo
.year());
212 if(metaInfo
.position()) args
<< "--tn" << QString::number(metaInfo
.position());
213 if(!metaInfo
.cover().isEmpty()) args
<< "--ti" << QDir::toNativeSeparators(metaInfo
.cover());
215 if(!m_configCustomParams
.isEmpty()) args
<< m_configCustomParams
.split(" ", QString::SkipEmptyParts
);
217 args
<< QDir::toNativeSeparators(sourceFile
);
218 args
<< QDir::toNativeSeparators(outputFile
);
220 if(!startProcess(process
, m_binary
, args
))
225 bool bTimeout
= false;
226 bool bAborted
= false;
227 int prevProgress
= -1;
229 QRegExp
regExp("\\(.*(\\d+)%\\)\\|");
231 while(process
.state() != QProcess::NotRunning
)
237 emit
messageLogged("\nABORTED BY USER !!!");
240 process
.waitForReadyRead(m_processTimeoutInterval
);
241 if(!process
.bytesAvailable() && process
.state() == QProcess::Running
)
244 qWarning("LAME process timed out <-- killing!");
245 emit
messageLogged("\nPROCESS TIMEOUT !!!");
249 while(process
.bytesAvailable() > 0)
251 QByteArray line
= process
.readLine();
252 QString text
= QString::fromUtf8(line
.constData()).simplified();
253 if(regExp
.lastIndexIn(text
) >= 0)
256 int progress
= regExp
.cap(1).toInt(&ok
);
257 if(ok
&& (progress
> prevProgress
))
259 emit
statusUpdated(progress
);
260 prevProgress
= qMin(progress
+ 2, 99);
263 else if(!text
.isEmpty())
265 emit
messageLogged(text
);
270 process
.waitForFinished();
271 if(process
.state() != QProcess::NotRunning
)
274 process
.waitForFinished(-1);
277 emit
statusUpdated(100);
278 emit
messageLogged(QString().sprintf("\nExited with code: 0x%04X", process
.exitCode()));
280 if(bTimeout
|| bAborted
|| process
.exitCode() != EXIT_SUCCESS
)
288 QString
MP3Encoder::extension(void)
293 bool MP3Encoder::isFormatSupported(const QString
&containerType
, const QString
&containerProfile
, const QString
&formatType
, const QString
&formatProfile
, const QString
&formatVersion
)
295 if(containerType
.compare("Wave", Qt::CaseInsensitive
) == 0)
297 if(formatType
.compare("PCM", Qt::CaseInsensitive
) == 0)
302 else if(containerType
.compare("MPEG Audio", Qt::CaseInsensitive
) == 0)
304 if(formatType
.compare("MPEG Audio", Qt::CaseInsensitive
) == 0)
306 if(formatProfile
.compare("Layer 3", Qt::CaseInsensitive
) == 0 || formatProfile
.compare("Layer 2", Qt::CaseInsensitive
) == 0)
308 if(formatVersion
.compare("Version 1", Qt::CaseInsensitive
) == 0 || formatVersion
.compare("Version 2", Qt::CaseInsensitive
) == 0)
319 const unsigned int *MP3Encoder::supportedChannelCount(void)
321 static const unsigned int supportedChannels
[] = {1, 2, NULL
};
322 return supportedChannels
;
325 void MP3Encoder::setAlgoQuality(int value
)
327 m_algorithmQuality
= qBound(0, value
, 3);
330 void MP3Encoder::setBitrateLimits(int minimumBitrate
, int maximumBitrate
)
332 m_configBitrateMinimum
= minimumBitrate
;
333 m_configBitrateMaximum
= maximumBitrate
;
336 void MP3Encoder::setSamplingRate(int value
)
338 m_configSamplingRate
= value
;
341 void MP3Encoder::setChannelMode(int value
)
343 m_configChannelMode
= value
;
346 int MP3Encoder::clipBitrate(int bitrate
)
348 int targetBitrate
= qMin(qMax(bitrate
, 32), 320);
350 int minDiff
= INT_MAX
;
353 for(int i
= 0; g_mp3BitrateLUT
[i
] > 0; i
++)
355 int currentDiff
= abs(targetBitrate
- g_mp3BitrateLUT
[i
]);
356 if(currentDiff
< minDiff
)
358 minDiff
= currentDiff
;
365 return g_mp3BitrateLUT
[minIndx
];
368 return targetBitrate
;
371 const AbstractEncoderInfo
*MP3Encoder::getEncoderInfo(void)
373 return &g_mp3EncoderInfo
;