src/Filter_Downmix.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2016 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 *const 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 != 0) && (channels <= 2))
  58         {
  59                 messageLogged("Skipping downmix!");
  60                 qDebug("Dowmmix not required/possible for Mono or Stereo input, skipping!");
  61                 return true;
  62         }
  63
  64         QProcess process;
  65         QStringList args;
  66
  67         args << "-V3" << "-S";
  68         args << "--guard" << "--temp" << ".";
  69         args << QDir::toNativeSeparators(sourceFile);
  70         args << QDir::toNativeSeparators(outputFile);
  71
  72         switch(channels)
  73         {
  74         case 3: //3.0 (L/R/C)
  75                 args << "remix" << "1v0.66,3v0.34" << "2v0.66,3v0.34";
  76                 break;
  77         case 4: //3.1 (L/R/C/LFE)
  78                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,4v0.25";
  79                 break;
  80         case 5: //5.0 (L/R/C/BL/BR)
  81                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,5v0.25";
  82                 break;
  83         case 6: //5.1 (L/R/C/LFE/BL/BR)
  84                 args << "remix" << "1v0.4,3v0.2,4v0.2,5v0.2" << "2v0.4,3v0.2,4v0.2,6v0.2";
  85                 break;
  86         case 7: //7.0 (L/R/C/BL/BR/SL/SR)
  87                 args << "remix" << "1v0.4,3v0.2,4v0.2,6v0.2" << "2v0.4,3v0.2,5v0.2,7v0.2";
  88                 break;
  89         case 8: //7.1 (L/R/C/LFE/BL/BR/SL/SR)
  90                 args << "remix" << "1v0.36,3v0.16,4v0.16,5v0.16,7v0.16" << "2v0.36,3v0.16,4v0.16,6v0.16,8v0.16";
  91                 break;
  92         case 9: //8.1 (L/R/C/LFE/BL/BR/SL/SR/BC)
  93                 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";
  94                 break;
  95         default: //Unknown
  96                 qWarning("Downmixer: Unknown channel configuration!");
  97                 args << "channels" << QString::number(2);
  98                 break;
  99         }
 100
 101         if(!startProcess(process, m_binary, args, QFileInfo(outputFile).canonicalPath()))
 102         {
 103                 return false;
 104         }
 105
 106         bool bTimeout = false;
 107         bool bAborted = false;
 108
 109         QRegExp regExp("In:(\\d+)(\\.\\d+)*%");
 110
 111         while(process.state() != QProcess::NotRunning)
 112         {
 113                 if(*abortFlag)
 114                 {
 115                         process.kill();
 116                         bAborted = true;
 117                         emit messageLogged("\nABORTED BY USER !!!");
 118                         break;
 119                 }
 120                 process.waitForReadyRead(m_processTimeoutInterval);
 121                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
 122                 {
 123                         process.kill();
 124                         qWarning("SoX process timed out <-- killing!");
 125                         emit messageLogged("\nPROCESS TIMEOUT !!!");
 126                         bTimeout = true;
 127                         break;
 128                 }
 129                 while(process.bytesAvailable() > 0)
 130                 {
 131                         QByteArray line = process.readLine();
 132                         QString text = QString::fromUtf8(line.constData()).simplified();
 133                         if(regExp.lastIndexIn(text) >= 0)
 134                         {
 135                                 bool ok = false;
 136                                 int progress = regExp.cap(1).toInt(&ok);
 137                                 if(ok) emit statusUpdated(progress);
 138                         }
 139                         else if(!text.isEmpty())
 140                         {
 141                                 emit messageLogged(text);
 142                         }
 143                 }
 144         }
 145
 146         process.waitForFinished();
 147         if(process.state() != QProcess::NotRunning)
 148         {
 149                 process.kill();
 150                 process.waitForFinished(-1);
 151         }
 152
 153         emit statusUpdated(100);
 154         emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", process.exitCode()));
 155
 156         if(bTimeout || bAborted || process.exitCode() != EXIT_SUCCESS || QFileInfo(outputFile).size() == 0)
 157         {
 158                 return false;
 159         }
 160
 161         formatInfo->setAudioChannels(2);
 162         return true;
 163 }