src/Filter_Downmix.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 "Filter_Downmix.h"
  23
  24 #include "Global.h"
  25
  26 #include <QDir>
  27 #include <QProcess>
  28 #include <QRegExp>
  29
  30 DownmixFilter::DownmixFilter(void)
  31 :
  32         m_binary(lamexp_lookup_tool("sox.exe"))
  33 {
  34         if(m_binary.isEmpty())
  35         {
  36                 throw "Error initializing SoX filter. Tool 'sox.exe' is not registred!";
  37         }
  38 }
  39
  40 DownmixFilter::~DownmixFilter(void)
  41 {
  42 }
  43
  44 bool DownmixFilter::apply(const QString &sourceFile, const QString &outputFile, volatile bool *abortFlag)
  45 {
  46         unsigned int channels = detectChannels(sourceFile, abortFlag);
  47         emit messageLogged(QString().sprintf("--> Number of channels is: %d\n", channels));
  48
  49         if((channels == 1) || (channels == 2))
  50         {
  51                 messageLogged("Skipping downmix!");
  52                 qDebug("Dowmmix not required for Mono or Stereo input, skipping!");
  53                 return false;
  54         }
  55
  56         QProcess process;
  57         QStringList args;
  58
  59         process.setWorkingDirectory(QFileInfo(outputFile).canonicalPath());
  60
  61         args << "-V3" << "-S";
  62         args << "--guard" << "--temp" << ".";
  63         args << QDir::toNativeSeparators(sourceFile);
  64         args << QDir::toNativeSeparators(outputFile);
  65
  66         switch(channels)
  67         {
  68         case 3: //3.0 (L/R/C)
  69                 args << "remix" << "1v0.66,3v0.34" << "2v0.66,3v0.34";
  70                 break;
  71         case 4: //3.1 (L/R/C/LFE)
  72                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,4v0.25";
  73                 break;
  74         case 5: //5.0 (L/R/C/BL/BR)
  75                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,5v0.25";
  76                 break;
  77         case 6: //5.1 (L/R/C/LFE/BL/BR)
  78                 args << "remix" << "1v0.4,3v0.2,4v0.2,5v0.2" << "2v0.4,3v0.2,4v0.2,6v0.2";
  79                 break;
  80         case 7: //7.0 (L/R/C/BL/BR/SL/SR)
  81                 args << "remix" << "1v0.4,3v0.2,4v0.2,6v0.2" << "2v0.4,3v0.2,5v0.2,7v0.2";
  82                 break;
  83         case 8: //7.1 (L/R/C/LFE/BL/BR/SL/SR)
  84                 args << "remix" << "1v0.36,3v0.16,4v0.16,5v0.16,7v0.16" << "2v0.36,3v0.16,4v0.16,6v0.16,8v0.16";
  85                 break;
  86         case 9: //8.1 (L/R/C/LFE/BL/BR/SL/SR/BC)
  87                 args << "remix" << "1v0.308,3v0.154,4v0.154,5v0.154,7v0.154,9v0.076" << "2v0.308,3v0.154,4v0.154,6v0.154,8v0.154,9v0.076";
  88                 break;
  89         default: //Unknown
  90                 qWarning("Downmixer: Unknown channel configuration!");
  91                 args << "channels" << "2";
  92                 break;
  93         }
  94
  95         if(!startProcess(process, m_binary, args))
  96         {
  97                 return false;
  98         }
  99
 100         bool bTimeout = false;
 101         bool bAborted = false;
 102
 103         QRegExp regExp("In:(\\d+)(\\.\\d+)*%");
 104
 105         while(process.state() != QProcess::NotRunning)
 106         {
 107                 if(*abortFlag)
 108                 {
 109                         process.kill();
 110                         bAborted = true;
 111                         emit messageLogged("\nABORTED BY USER !!!");
 112                         break;
 113                 }
 114                 process.waitForReadyRead(m_processTimeoutInterval);
 115                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
 116                 {
 117                         process.kill();
 118                         qWarning("SoX process timed out <-- killing!");
 119                         emit messageLogged("\nPROCESS TIMEOUT !!!");
 120                         bTimeout = true;
 121                         break;
 122                 }
 123                 while(process.bytesAvailable() > 0)
 124                 {
 125                         QByteArray line = process.readLine();
 126                         QString text = QString::fromUtf8(line.constData()).simplified();
 127                         if(regExp.lastIndexIn(text) >= 0)
 128                         {
 129                                 bool ok = false;
 130                                 int progress = regExp.cap(1).toInt(&ok);
 131                                 if(ok) emit statusUpdated(progress);
 132                         }
 133                         else if(!text.isEmpty())
 134                         {
 135                                 emit messageLogged(text);
 136                         }
 137                 }
 138         }
 139
 140         process.waitForFinished();
 141         if(process.state() != QProcess::NotRunning)
 142         {
 143                 process.kill();
 144                 process.waitForFinished(-1);
 145         }
 146
 147         emit statusUpdated(100);
 148         emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", process.exitCode()));
 149
 150         if(bTimeout || bAborted || process.exitStatus() != QProcess::NormalExit || QFileInfo(outputFile).size() == 0)
 151         {
 152                 return false;
 153         }
 154
 155         return true;
 156 }
 157
 158 unsigned int DownmixFilter::detectChannels(const QString &sourceFile, volatile bool *abortFlag)
 159 {
 160         unsigned int channels = 0;
 161
 162         QProcess process;
 163         QStringList args;
 164
 165         args << "--i" << sourceFile;
 166
 167         if(!startProcess(process, m_binary, args))
 168         {
 169                 return channels;
 170         }
 171
 172         bool bTimeout = false;
 173         bool bAborted = false;
 174
 175         QRegExp regExp("Channels\\s*:\\s*(\\d+)", Qt::CaseInsensitive);
 176
 177         while(process.state() != QProcess::NotRunning)
 178         {
 179                 if(*abortFlag)
 180                 {
 181                         process.kill();
 182                         bAborted = true;
 183                         emit messageLogged("\nABORTED BY USER !!!");
 184                         break;
 185                 }
 186                 process.waitForReadyRead(m_processTimeoutInterval);
 187                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
 188                 {
 189                         process.kill();
 190                         qWarning("SoX process timed out <-- killing!");
 191                         emit messageLogged("\nPROCESS TIMEOUT !!!");
 192                         bTimeout = true;
 193                         break;
 194                 }
 195                 while(process.bytesAvailable() > 0)
 196                 {
 197                         QByteArray line = process.readLine();
 198                         QString text = QString::fromUtf8(line.constData()).simplified();
 199                         if(regExp.lastIndexIn(text) >= 0)
 200                         {
 201                                 bool ok = false;
 202                                 unsigned int temp = regExp.cap(1).toUInt(&ok);
 203                                 if(ok) channels = temp;
 204                         }
 205                         if(!text.isEmpty())
 206                         {
 207                                 emit messageLogged(text);
 208                         }
 209                 }
 210         }
 211
 212         process.waitForFinished();
 213         if(process.state() != QProcess::NotRunning)
 214         {
 215                 process.kill();
 216                 process.waitForFinished(-1);
 217         }
 218
 219         return channels;
 220 }