src/Filter_Downmix.cpp

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