src/Filter_Downmix.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2018 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 #define IS_VALID(X) (((X) != 0U) && ((X) != UINT_MAX))
  39
  40 DownmixFilter::DownmixFilter(void)
  41 :
  42         m_binary(lamexp_tools_lookup("sox.exe"))
  43 {
  44         if(m_binary.isEmpty())
  45         {
  46                 MUTILS_THROW("Error initializing SoX filter. Tool 'sox.exe' is not registred!");
  47         }
  48 }
  49
  50 DownmixFilter::~DownmixFilter(void)
  51 {
  52 }
  53
  54 AbstractFilter::FilterResult DownmixFilter::apply(const QString &sourceFile, const QString &outputFile, AudioFileModel_TechInfo *const formatInfo, QAtomicInt &abortFlag)
  55 {
  56         unsigned int channels = formatInfo->audioChannels();
  57         emit messageLogged(QString().sprintf("--> Number of channels is: %d\n", channels));
  58
  59         if(IS_VALID(channels) && (channels <= 2))
  60         {
  61                 messageLogged("Skipping downmix!");
  62                 qDebug("Dowmmix not required/possible for Mono or Stereo input, skipping!");
  63                 return AbstractFilter::FILTER_SKIPPED;
  64         }
  65
  66         QProcess process;
  67         QStringList args;
  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 3: //3.0 (L/R/C)
  77                 args << "remix" << "1v0.66,3v0.34" << "2v0.66,3v0.34";
  78                 break;
  79         case 4: //3.1 (L/R/C/LFE)
  80                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,4v0.25";
  81                 break;
  82         case 5: //5.0 (L/R/C/BL/BR)
  83                 args << "remix" << "1v0.5,3v0.25,4v0.25" << "2v0.5,3v0.25,5v0.25";
  84                 break;
  85         case 6: //5.1 (L/R/C/LFE/BL/BR)
  86                 args << "remix" << "1v0.4,3v0.2,4v0.2,5v0.2" << "2v0.4,3v0.2,4v0.2,6v0.2";
  87                 break;
  88         case 7: //7.0 (L/R/C/BL/BR/SL/SR)
  89                 args << "remix" << "1v0.4,3v0.2,4v0.2,6v0.2" << "2v0.4,3v0.2,5v0.2,7v0.2";
  90                 break;
  91         case 8: //7.1 (L/R/C/LFE/BL/BR/SL/SR)
  92                 args << "remix" << "1v0.36,3v0.16,4v0.16,5v0.16,7v0.16" << "2v0.36,3v0.16,4v0.16,6v0.16,8v0.16";
  93                 break;
  94         case 9: //8.1 (L/R/C/LFE/BL/BR/SL/SR/BC)
  95                 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";
  96                 break;
  97         default: //Unknown
  98                 qWarning("Downmixer: Unknown channel configuration!");
  99                 args << "channels" << QString::number(2);
 100                 break;
 101         }
 102
 103         if(!startProcess(process, m_binary, args, QFileInfo(outputFile).canonicalPath()))
 104         {
 105                 return AbstractFilter::FILTER_FAILURE;
 106         }
 107
 108         int prevProgress = -1;
 109         QRegExp regExp("In:(\\d+)(\\.\\d+)*%");
 110
 111         const result_t result = awaitProcess(process, abortFlag, [this, &prevProgress, &regExp](const QString &text)
 112         {
 113                 if (regExp.lastIndexIn(text) >= 0)
 114                 {
 115                         qint32 newProgress;
 116                         if (MUtils::regexp_parse_int32(regExp, newProgress))
 117                         {
 118                                 if (newProgress > prevProgress)
 119                                 {
 120                                         emit statusUpdated(newProgress);
 121                                         prevProgress = NEXT_PROGRESS(newProgress);
 122                                 }
 123                         }
 124                         return true;
 125                 }
 126                 return false;
 127         });
 128
 129         if (result != RESULT_SUCCESS)
 130         {
 131                 return AbstractFilter::FILTER_FAILURE;
 132         }
 133
 134         formatInfo->setAudioChannels(2);
 135         return AbstractFilter::FILTER_SUCCESS;
 136 }