OSDN Git Service

Various improvements to ShellIntegration code.
[lamexp/LameXP.git] / src / Encoder_AC3.cpp
index 39c625e..59e08da 100644 (file)
@@ -1,11 +1,12 @@
 ///////////////////////////////////////////////////////////////////////////////
 // LameXP - Audio Encoder Front-End
-// Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
+// Copyright (C) 2004-2015 LoRd_MuldeR <MuldeR2@GMX.de>
 //
 // This program is free software; you can redistribute it and/or modify
 // it under the terms of the GNU General Public License as published by
 // the Free Software Foundation; either version 2 of the License, or
-// (at your option) any later version.
+// (at your option) any later version, but always including the *additional*
+// restrictions defined in the "License.txt" file.
 //
 // This program is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
 #include <QProcess>
 #include <QDir>
 
-#define max(a,b) (((a) > (b)) ? (a) : (b))
-#define min(a,b) (((a) < (b)) ? (a) : (b))
+static const int g_ac3BitratesLUT[20] = {32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384, 448, 512, 576, 640, -1};
+
+///////////////////////////////////////////////////////////////////////////////
+// Encoder Info
+///////////////////////////////////////////////////////////////////////////////
+
+class AC3EncoderInfo : public AbstractEncoderInfo
+{
+       virtual bool isModeSupported(int mode) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+               case SettingsModel::CBRMode:
+                       return true;
+                       break;
+               case SettingsModel::ABRMode:
+                       return false;
+                       break;
+               default:
+                       MUTILS_THROW("Bad RC mode specified!");
+               }
+       }
+
+       virtual int valueCount(int mode) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+                       return 65;
+                       break;
+               case SettingsModel::ABRMode:
+                       return 0;
+                       break;
+               case SettingsModel::CBRMode:
+                       return 19;
+                       break;
+               default:
+                       MUTILS_THROW("Bad RC mode specified!");
+               }
+       }
+
+       virtual int valueAt(int mode, int index) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+                       return qBound(0, index * 16, 1023);
+                       break;
+               case SettingsModel::ABRMode:
+               case SettingsModel::CBRMode:
+                       return g_ac3BitratesLUT[qBound(0, index, 18)];
+                       break;
+               default:
+                       MUTILS_THROW("Bad RC mode specified!");
+               }
+       }
+
+       virtual int valueType(int mode) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+                       return TYPE_QUALITY_LEVEL_INT;
+                       break;
+               case SettingsModel::ABRMode:
+               case SettingsModel::CBRMode:
+                       return TYPE_BITRATE;
+                       break;
+               default:
+                       MUTILS_THROW("Bad RC mode specified!");
+               }
+       }
+
+       virtual const char *description(void) const
+       {
+               static const char* s_description = "Aften: A/52 Audio Encoder";
+               return s_description;
+       }
+}
+static const g_aftenEncoderInfo;
+
+///////////////////////////////////////////////////////////////////////////////
+// Encoder implementation
+///////////////////////////////////////////////////////////////////////////////
 
 AC3Encoder::AC3Encoder(void)
 :
-       m_binary(lamexp_lookup_tool("aften.exe"))
+       m_binary(lamexp_tools_lookup("aften.exe"))
 {
        if(m_binary.isEmpty())
        {
-               throw "Error initializing FLAC encoder. Tool 'aften.exe' is not registred!";
+               MUTILS_THROW("Error initializing FLAC encoder. Tool 'aften.exe' is not registred!");
        }
 
        m_configAudioCodingMode = 0;
@@ -49,7 +133,7 @@ AC3Encoder::~AC3Encoder(void)
 {
 }
 
-bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInfo, const QString &outputFile, volatile bool *abortFlag)
+bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel_MetaInfo &metaInfo, const unsigned int duration, const QString &outputFile, volatile bool *abortFlag)
 {
        QProcess process;
        QStringList args;
@@ -57,13 +141,13 @@ bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
        switch(m_configRCMode)
        {
        case SettingsModel::VBRMode:
-               args << "-q" << QString::number(max(0, min(1023, m_configBitrate * 64)));
+               args << "-q" << QString::number(qBound(0, m_configBitrate * 16, 1023));
                break;
        case SettingsModel::CBRMode:
-               args << "-b" << QString::number(SettingsModel::ac3Bitrates[max(0, min(18, m_configBitrate))]);
+               args << "-b" << QString::number(g_ac3BitratesLUT[qBound(0, m_configBitrate, 18)]);
                break;
        default:
-               throw "Bad rate-control mode!";
+               MUTILS_THROW("Bad rate-control mode!");
                break;
        }
 
@@ -96,6 +180,7 @@ bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
 
        bool bTimeout = false;
        bool bAborted = false;
+       int prevProgress = -1;
 
        QRegExp regExp("progress:(\\s+)(\\d+)%(\\s+)\\|");
 
@@ -108,7 +193,7 @@ bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
                        emit messageLogged("\nABORTED BY USER !!!");
                        break;
                }
-               process.waitForReadyRead();
+               process.waitForReadyRead(m_processTimeoutInterval);
                if(!process.bytesAvailable() && process.state() == QProcess::Running)
                {
                        process.kill();
@@ -125,7 +210,11 @@ bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
                        {
                                bool ok = false;
                                int progress = regExp.cap(2).toInt(&ok);
-                               if(ok) emit statusUpdated(progress);
+                               if(ok && (progress > prevProgress))
+                               {
+                                       emit statusUpdated(progress);
+                                       prevProgress = qMin(progress + 2, 99);
+                               }
                        }
                        else if(!text.isEmpty())
                        {
@@ -144,7 +233,7 @@ bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
        emit statusUpdated(100);
        emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", process.exitCode()));
 
-       if(bTimeout || bAborted || process.exitStatus() != QProcess::NormalExit)
+       if(bTimeout || bAborted || process.exitCode() != EXIT_SUCCESS)
        {
                return false;
        }
@@ -154,17 +243,17 @@ bool AC3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
 
 void AC3Encoder::setAudioCodingMode(int value)
 {
-       m_configAudioCodingMode = min(8, max(0, value));
+       m_configAudioCodingMode = qMin(8, qMax(0, value));
 }
 
 void AC3Encoder::setDynamicRangeCompression(int value)
 {
-       m_configDynamicRangeCompression = min(5, max(0, value));
+       m_configDynamicRangeCompression = qMin(5, qMax(0, value));
 }
 
 void AC3Encoder::setExponentSearchSize(int value)
 {
-       m_configExponentSearchSize = min(32, max(1, value));
+       m_configExponentSearchSize = qMin(32, qMax(1, value));
 }
 
 void AC3Encoder::setFastBitAllocation(bool value)
@@ -177,7 +266,13 @@ QString AC3Encoder::extension(void)
        return "ac3";
 }
 
-const unsigned int *AC3Encoder::requiresDownsample(void)
+const unsigned int *AC3Encoder::supportedChannelCount(void)
+{
+       static const unsigned int supportedChannels[] = {1, 2, 3, 4, 5, 6, NULL};
+       return supportedChannels;
+}
+
+const unsigned int *AC3Encoder::supportedSamplerates(void)
 {
        static const unsigned int supportedRates[] = {48000, 44100, 32000, NULL};
        return supportedRates;
@@ -195,3 +290,8 @@ bool AC3Encoder::isFormatSupported(const QString &containerType, const QString &
 
        return false;
 }
+
+const AbstractEncoderInfo *AC3Encoder::getEncoderInfo(void)
+{
+       return &g_aftenEncoderInfo;
+}