src/Filter_Downmix.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2012 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 #include "Tool_WaveProperties.h"
  26 #include "Model_AudioFile.h"
  27
  28 #include <QDir>
  29 #include <QProcess>
  30 #include <QRegExp>
  31
  32 DownmixFilter::DownmixFilter(void)
  33 :
  34         m_binary(lamexp_lookup_tool("sox.exe"))
  35 {
  36         if(m_binary.isEmpty())
  37         {
  38                 throw "Error initializing SoX filter. Tool 'sox.exe' is not registred!";
  39         }
  40 }
  41
  42 DownmixFilter::~DownmixFilter(void)
  43 {
  44 }
  45
  46 bool DownmixFilter::apply(const QString &sourceFile, const QString &outputFile, AudioFileModel *formatInfo, volatile bool *abortFlag)
  47 {
  48         unsigned int channels = formatInfo->formatAudioChannels(); //detectChannels(sourceFile, abortFlag);
  49         emit messageLogged(QString().sprintf("--> Number of channels is: %d\n", channels));
  50
  51         if(channels == 2)
  52         {
  53                 messageLogged("Skipping downmix!");
  54                 qDebug("Dowmmix not required for Stereo input, skipping!");
  55                 return true;
  56         }
  57
  58         QProcess process;
  59         QStringList args;
  60
  61         process.setWorkingDirectory(QFileInfo(outputFile).canonicalPath());
  62
  63         args << "-V3" << "-S";
  64         args << "--guard" << "--temp" << ".";
  65         args << QDir::toNativeSeparators(sourceFile);
  66         args << QDir::toNativeSeparators(outputFile);
  67
  68         switch(channels)
  69         {
  70         case 2: //Unknown
  71                 qWarning("Downmixer: Nothing to do!");
  72         case 3: //3.0 (L/R/C)
  73                 args << "remix" << "1v0.66,3v0.34" << "2v0.66,3v0.34";
  74                 break;
  75         case 4: //3.1 (L/R/C/LFE)
  76                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,4v0.25";
  77                 break;
  78         case 5: //5.0 (L/R/C/BL/BR)
  79                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,5v0.25";
  80                 break;
  81         case 6: //5.1 (L/R/C/LFE/BL/BR)
  82                 args << "remix" << "1v0.4,3v0.2,4v0.2,5v0.2" << "2v0.4,3v0.2,4v0.2,6v0.2";
  83                 break;
  84         case 7: //7.0 (L/R/C/BL/BR/SL/SR)
  85                 args << "remix" << "1v0.4,3v0.2,4v0.2,6v0.2" << "2v0.4,3v0.2,5v0.2,7v0.2";
  86                 break;
  87         case 8: //7.1 (L/R/C/LFE/BL/BR/SL/SR)
  88                 args << "remix" << "1v0.36,3v0.16,4v0.16,5v0.16,7v0.16" << "2v0.36,3v0.16,4v0.16,6v0.16,8v0.16";
  89                 break;
  90         case 9: //8.1 (L/R/C/LFE/BL/BR/SL/SR/BC)
  91                 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";
  92                 break;
  93         default: //Unknown
  94                 qWarning("Downmixer: Unknown channel configuration!");
  95                 args << "channels" << "2";
  96                 break;
  97         }
  98
  99         if(!startProcess(process, m_binary, args))
 100         {
 101                 return false;
 102         }
 103
 104         bool bTimeout = false;
 105         bool bAborted = false;
 106
 107         QRegExp regExp("In:(\\d+)(\\.\\d+)*%");
 108
 109         while(process.state() != QProcess::NotRunning)
 110         {
 111                 if(*abortFlag)
 112                 {
 113                         process.kill();
 114                         bAborted = true;
 115                         emit messageLogged("\nABORTED BY USER !!!");
 116                         break;
 117                 }
 118                 process.waitForReadyRead(m_processTimeoutInterval);
 119                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
 120                 {
 121                         process.kill();
 122                         qWarning("SoX process timed out <-- killing!");
 123                         emit messageLogged("\nPROCESS TIMEOUT !!!");
 124                         bTimeout = true;
 125                         break;
 126                 }
 127                 while(process.bytesAvailable() > 0)
 128                 {
 129                         QByteArray line = process.readLine();
 130                         QString text = QString::fromUtf8(line.constData()).simplified();
 131                         if(regExp.lastIndexIn(text) >= 0)
 132                         {
 133                                 bool ok = false;
 134                                 int progress = regExp.cap(1).toInt(&ok);
 135                                 if(ok) emit statusUpdated(progress);
 136                         }
 137                         else if(!text.isEmpty())
 138                         {
 139                                 emit messageLogged(text);
 140                         }
 141                 }
 142         }
 143
 144         process.waitForFinished();
 145         if(process.state() != QProcess::NotRunning)
 146         {
 147                 process.kill();
 148                 process.waitForFinished(-1);
 149         }
 150
 151         emit statusUpdated(100);
 152         emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", process.exitCode()));
 153
 154         if(bTimeout || bAborted || process.exitCode() != EXIT_SUCCESS || QFileInfo(outputFile).size() == 0)
 155         {
 156                 return false;
 157         }
 158
 159         formatInfo->setFormatAudioChannels(2);
 160         return true;
 161 }