src/Encoder_AC3.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
   4 //
   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.
   9 //
  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.
  14 //
  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.
  18 //
  19 // http://www.gnu.org/licenses/gpl-2.0.txt
  20 ///////////////////////////////////////////////////////////////////////////////
  21
  22 #include "Encoder_AC3.h"
  23
  24 #include "Global.h"
  25 #include "Model_Settings.h"
  26
  27 #include <QProcess>
  28 #include <QDir>
  29
  30 #define max(a,b) (((a) > (b)) ? (a) : (b))
  31 #define min(a,b) (((a) < (b)) ? (a) : (b))
  32
  33 AC3Encoder::AC3Encoder(void)
  34 :
  35         m_binary(lamexp_lookup_tool("aften.exe"))
  36 {
  37         if(m_binary.isEmpty())
  38         {
  39                 throw "Error initializing FLAC encoder. Tool 'aften.exe' is not registred!";
  40         }
  41
  42         m_configAudioCodingMode = 0;
  43         m_configDynamicRangeCompression = 5;
  44         m_configExponentSearchSize = 8;
  45         m_configFastBitAllocation = false;
  46 }
  47
  48 AC3Encoder::~AC3Encoder(void)
  49 {
  50 }
  51
  52 bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInfo, const QString &outputFile, volatile bool *abortFlag)
  53 {
  54         QProcess process;
  55         QStringList args;
  56
  57         switch(m_configRCMode)
  58         {
  59         case SettingsModel::VBRMode:
  60                 args << "-q" << QString::number(max(0, min(1023, m_configBitrate * 64)));
  61                 break;
  62         case SettingsModel::CBRMode:
  63                 args << "-b" << QString::number(SettingsModel::ac3Bitrates[max(0, min(18, m_configBitrate))]);
  64                 break;
  65         default:
  66                 throw "Bad rate-control mode!";
  67                 break;
  68         }
  69
  70         if(m_configAudioCodingMode >= 1)
  71         {
  72                 args << "-acmod" << QString::number(m_configAudioCodingMode - 1);
  73         }
  74         if(m_configDynamicRangeCompression != 5)
  75         {
  76                 args << "-dynrng" << QString::number(m_configDynamicRangeCompression);
  77         }
  78         if(m_configExponentSearchSize != 8)
  79         {
  80                 args << "-exps" << QString::number(m_configExponentSearchSize);
  81         }
  82         if(m_configFastBitAllocation)
  83         {
  84                 args << "-fba" << QString::number(1);
  85         }
  86
  87         if(!m_configCustomParams.isEmpty()) args << m_configCustomParams.split(" ", QString::SkipEmptyParts);
  88
  89         args << QDir::toNativeSeparators(sourceFile);
  90         args << QDir::toNativeSeparators(outputFile);
  91
  92         if(!startProcess(process, m_binary, args))
  93         {
  94                 return false;
  95         }
  96
  97         bool bTimeout = false;
  98         bool bAborted = false;
  99
 100         QRegExp regExp("progress:(\\s+)(\\d+)%(\\s+)\\|");
 101
 102         while(process.state() != QProcess::NotRunning)
 103         {
 104                 if(*abortFlag)
 105                 {
 106                         process.kill();
 107                         bAborted = true;
 108                         emit messageLogged("\nABORTED BY USER !!!");
 109                         break;
 110                 }
 111                 process.waitForReadyRead(m_processTimeoutInterval);
 112                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
 113                 {
 114                         process.kill();
 115                         qWarning("Aften process timed out <-- killing!");
 116                         emit messageLogged("\nPROCESS TIMEOUT !!!");
 117                         bTimeout = true;
 118                         break;
 119                 }
 120                 while(process.bytesAvailable() > 0)
 121                 {
 122                         QByteArray line = process.readLine();
 123                         QString text = QString::fromUtf8(line.constData()).simplified();
 124                         if(regExp.lastIndexIn(text) >= 0)
 125                         {
 126                                 bool ok = false;
 127                                 int progress = regExp.cap(2).toInt(&ok);
 128                                 if(ok) emit statusUpdated(progress);
 129                         }
 130                         else if(!text.isEmpty())
 131                         {
 132                                 emit messageLogged(text);
 133                         }
 134                 }
 135         }
 136
 137         process.waitForFinished();
 138         if(process.state() != QProcess::NotRunning)
 139         {
 140                 process.kill();
 141                 process.waitForFinished(-1);
 142         }
 143
 144         emit statusUpdated(100);
 145         emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", process.exitCode()));
 146
 147         if(bTimeout || bAborted || process.exitStatus() != QProcess::NormalExit)
 148         {
 149                 return false;
 150         }
 151
 152         return true;
 153 }
 154
 155 void AC3Encoder::setAudioCodingMode(int value)
 156 {
 157         m_configAudioCodingMode = min(8, max(0, value));
 158 }
 159
 160 void AC3Encoder::setDynamicRangeCompression(int value)
 161 {
 162         m_configDynamicRangeCompression = min(5, max(0, value));
 163 }
 164
 165 void AC3Encoder::setExponentSearchSize(int value)
 166 {
 167         m_configExponentSearchSize = min(32, max(1, value));
 168 }
 169
 170 void AC3Encoder::setFastBitAllocation(bool value)
 171 {
 172         m_configFastBitAllocation = value;
 173 }
 174
 175 QString AC3Encoder::extension(void)
 176 {
 177         return "ac3";
 178 }
 179
 180 const unsigned int *AC3Encoder::requiresDownsample(void)
 181 {
 182         static const unsigned int supportedRates[] = {48000, 44100, 32000, NULL};
 183         return supportedRates;
 184 }
 185
 186 bool AC3Encoder::isFormatSupported(const QString &containerType, const QString &containerProfile, const QString &formatType, const QString &formatProfile, const QString &formatVersion)
 187 {
 188         if(containerType.compare("Wave", Qt::CaseInsensitive) == 0)
 189         {
 190                 if(formatType.compare("PCM", Qt::CaseInsensitive) == 0)
 191                 {
 192                         return true;
 193                 }
 194         }
 195
 196         return false;
 197 }