src/Encoder_Vorbis.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 "Encoder_Vorbis.h"
  23
  24 #include "Global.h"
  25 #include "Model_Settings.h"
  26
  27 #include <QProcess>
  28 #include <QDir>
  29
  30 VorbisEncoder::VorbisEncoder(void)
  31 :
  32         m_binary(lamexp_lookup_tool("oggenc2.exe"))
  33 {
  34         if(m_binary.isEmpty())
  35         {
  36                 throw "Error initializing Vorbis encoder. Tool 'oggenc2.exe' is not registred!";
  37         }
  38
  39         m_configBitrateMaximum = 0;
  40         m_configBitrateMinimum = 0;
  41         m_configSamplingRate = 0;
  42 }
  43
  44 VorbisEncoder::~VorbisEncoder(void)
  45 {
  46 }
  47
  48 bool VorbisEncoder::encode(const QString &sourceFile, const AudioFileModel &metaInfo, const QString &outputFile, volatile bool *abortFlag)
  49 {
  50         QProcess process;
  51         QStringList args;
  52         const QString baseName = QFileInfo(outputFile).fileName();
  53
  54         switch(m_configRCMode)
  55         {
  56         case SettingsModel::VBRMode:
  57                 args << "-q" << QString::number(qMax(-2, qMin(10, m_configBitrate)));
  58                 break;
  59         case SettingsModel::ABRMode:
  60                 args << "-b" << QString::number(qMax(32, qMin(500, (m_configBitrate * 8))));
  61                 break;
  62         default:
  63                 throw "Bad rate-control mode!";
  64                 break;
  65         }
  66
  67         if((m_configBitrateMaximum > 0) && (m_configBitrateMinimum > 0) && (m_configBitrateMinimum <= m_configBitrateMaximum))
  68         {
  69                 args << "--min-bitrate" << QString::number(qMin(qMax(m_configBitrateMinimum, 32), 500));
  70                 args << "--max-bitrate" << QString::number(qMin(qMax(m_configBitrateMaximum, 32), 500));
  71         }
  72
  73         if(m_configSamplingRate > 0)
  74         {
  75                 args << "--resample" << QString::number(m_configSamplingRate) << "--converter" << QString::number(0);
  76         }
  77
  78         if(!metaInfo.fileName().isEmpty()) args << "-t" << metaInfo.fileName();
  79         if(!metaInfo.fileArtist().isEmpty()) args << "-a" << metaInfo.fileArtist();
  80         if(!metaInfo.fileAlbum().isEmpty()) args << "-l" << metaInfo.fileAlbum();
  81         if(!metaInfo.fileGenre().isEmpty()) args << "-G" << metaInfo.fileGenre();
  82         if(!metaInfo.fileComment().isEmpty()) args << "-c" << QString("comment=%1").arg(metaInfo.fileComment());
  83         if(metaInfo.fileYear()) args << "-d" << QString::number(metaInfo.fileYear());
  84         if(metaInfo.filePosition()) args << "-N" << QString::number(metaInfo.filePosition());
  85
  86         //args << "--tv" << QString().sprintf("Encoder=LameXP v%d.%02d.%04d [%s]", lamexp_version_major(), lamexp_version_minor(), lamexp_version_build(), lamexp_version_release());
  87
  88         if(!m_configCustomParams.isEmpty()) args << m_configCustomParams.split(" ", QString::SkipEmptyParts);
  89
  90         args << "-o" << QDir::toNativeSeparators(outputFile);
  91         args << QDir::toNativeSeparators(sourceFile);
  92
  93         if(!startProcess(process, m_binary, args))
  94         {
  95                 return false;
  96         }
  97
  98         bool bTimeout = false;
  99         bool bAborted = false;
 100         int prevProgress = -1;
 101
 102         QRegExp regExp("\\[.*(\\d+)[.,](\\d+)%\\]");
 103
 104         while(process.state() != QProcess::NotRunning)
 105         {
 106                 if(*abortFlag)
 107                 {
 108                         process.kill();
 109                         bAborted = true;
 110                         emit messageLogged("\nABORTED BY USER !!!");
 111                         break;
 112                 }
 113                 process.waitForReadyRead(m_processTimeoutInterval);
 114                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
 115                 {
 116                         process.kill();
 117                         qWarning("OggEnc process timed out <-- killing!");
 118                         emit messageLogged("\nPROCESS TIMEOUT !!!");
 119                         bTimeout = true;
 120                         break;
 121                 }
 122                 while(process.bytesAvailable() > 0)
 123                 {
 124                         QByteArray line = process.readLine();
 125                         QString text = QString::fromUtf8(line.constData()).simplified();
 126                         if(regExp.lastIndexIn(text) >= 0)
 127                         {
 128                                 bool ok = false;
 129                                 int progress = regExp.cap(1).toInt(&ok);
 130                                 if(ok && (progress > prevProgress))
 131                                 {
 132                                         emit statusUpdated(progress);
 133                                         prevProgress = qMin(progress + 2, 99);
 134                                 }
 135                         }
 136                         else if(!text.isEmpty())
 137                         {
 138                                 emit messageLogged(text);
 139                         }
 140                 }
 141         }
 142
 143         process.waitForFinished();
 144         if(process.state() != QProcess::NotRunning)
 145         {
 146                 process.kill();
 147                 process.waitForFinished(-1);
 148         }
 149
 150         emit statusUpdated(100);
 151         emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", process.exitCode()));
 152
 153         if(bTimeout || bAborted || process.exitCode() != EXIT_SUCCESS)
 154         {
 155                 return false;
 156         }
 157
 158         return true;
 159 }
 160
 161 QString VorbisEncoder::extension(void)
 162 {
 163         return "ogg";
 164 }
 165
 166 bool VorbisEncoder::isFormatSupported(const QString &containerType, const QString &containerProfile, const QString &formatType, const QString &formatProfile, const QString &formatVersion)
 167 {
 168         if(containerType.compare("Wave", Qt::CaseInsensitive) == 0)
 169         {
 170                 if(formatType.compare("PCM", Qt::CaseInsensitive) == 0)
 171                 {
 172                         return true;
 173                 }
 174         }
 175         else if(containerType.compare("FLAC", Qt::CaseInsensitive) == 0)
 176         {
 177                 if(formatType.compare("FLAC", Qt::CaseInsensitive) == 0)
 178                 {
 179                         return true;
 180                 }
 181         }
 182
 183         return false;
 184 }
 185
 186 void VorbisEncoder::setBitrateLimits(int minimumBitrate, int maximumBitrate)
 187 {
 188         m_configBitrateMinimum = minimumBitrate;
 189         m_configBitrateMaximum = maximumBitrate;
 190 }
 191
 192 void VorbisEncoder::setSamplingRate(int value)
 193 {
 194         m_configSamplingRate = value;
 195 }