OSDN Git Service

Set creation/modified time of the encoded file the same value as the original file...
[lamexp/LameXP.git] / src / Encoder_MP3.cpp
index 2b983d5..cc5ea71 100644 (file)
@@ -1,11 +1,12 @@
 ///////////////////////////////////////////////////////////////////////////////
 // LameXP - Audio Encoder Front-End
-// Copyright (C) 2004-2010 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>
+#include <limits.h>
 
-#define IS_UNICODE(STR) (qstricmp(STR.toUtf8().constData(), QString::fromLocal8Bit(STR.toLocal8Bit()).toUtf8().constData()))
+static const int g_lameAgorithmQualityLUT[5] = {7, 5, 2, 0, INT_MAX};
+static const int g_mp3BitrateLUT[15] = {32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, -1};
+static const int g_lameVBRQualityLUT[11] = {9, 8, 7, 6, 5, 4, 3, 2, 1, 0, INT_MAX};
+
+///////////////////////////////////////////////////////////////////////////////
+// Encoder Info
+///////////////////////////////////////////////////////////////////////////////
+
+class MP3EncoderInfo : public AbstractEncoderInfo
+{
+       virtual bool isModeSupported(int mode) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+               case SettingsModel::ABRMode:
+               case SettingsModel::CBRMode:
+                       return true;
+                       break;
+               default:
+                       MUTILS_THROW("Bad RC mode specified!");
+               }
+       }
+
+       virtual int valueCount(int mode) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+                       return 10;
+                       break;
+               case SettingsModel::ABRMode:
+               case SettingsModel::CBRMode:
+                       return 14;
+                       break;
+               default:
+                       MUTILS_THROW("Bad RC mode specified!");
+               }
+       }
+
+       virtual int valueAt(int mode, int index) const
+       {
+               switch(mode)
+               {
+               case SettingsModel::VBRMode:
+                       return g_lameVBRQualityLUT[qBound(0, index, 9)];
+                       break;
+               case SettingsModel::ABRMode:
+               case SettingsModel::CBRMode:
+                       return g_mp3BitrateLUT[qBound(0, index, 13)];
+                       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:
+                       return TYPE_APPROX_BITRATE;
+                       break;
+               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 = "LAME MP3 Encoder";
+               return s_description;
+       }
+
+       virtual const char *extension(void) const
+       {
+               static const char* s_extension = "mp3";
+               return s_extension;
+       }
+}
+static const g_mp3EncoderInfo;
+
+///////////////////////////////////////////////////////////////////////////////
+// Encoder implementation
+///////////////////////////////////////////////////////////////////////////////
 
 MP3Encoder::MP3Encoder(void)
 :
-       m_binary(lamexp_lookup_tool("lame.exe"))
+       m_binary(lamexp_tools_lookup("lame.exe"))
 {
        if(m_binary.isEmpty())
        {
-               throw "Error initializing MP3 encoder. Tool 'lame.exe' is not registred!";
+               MUTILS_THROW("Error initializing MP3 encoder. Tool 'lame.exe' is not registred!");
        }
+       
+       m_algorithmQuality = 2;
+       m_configBitrateMaximum = 0;
+       m_configBitrateMinimum = 0;
+       m_configSamplingRate = 0;
+       m_configChannelMode = 0;
 }
 
 MP3Encoder::~MP3Encoder(void)
 {
 }
 
-bool MP3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInfo, const QString &outputFile, volatile bool *abortFlag)
+bool MP3Encoder::encode(const QString &sourceFile, const AudioFileModel_MetaInfo &metaInfo, const unsigned int duration, const QString &outputFile, volatile bool *abortFlag)
 {
        QProcess process;
        QStringList args;
 
        args << "--nohist";
-       args << "-h";
+       args << "-q" << QString::number(g_lameAgorithmQualityLUT[m_algorithmQuality]);
                
        switch(m_configRCMode)
        {
        case SettingsModel::VBRMode:
-               args << "-V" << QString::number(9 - min(9, m_configBitrate));
+               args << "-V" << QString::number(g_lameVBRQualityLUT[qBound(0, m_configBitrate, 9)]);
                break;
        case SettingsModel::ABRMode:
-               args << "--abr" << QString::number(SettingsModel::mp3Bitrates[min(13, m_configBitrate)]);
+               args << "--abr" << QString::number(g_mp3BitrateLUT[qBound(0, m_configBitrate, 13)]);
                break;
        case SettingsModel::CBRMode:
                args << "--cbr";
-               args << "-b" << QString::number(SettingsModel::mp3Bitrates[min(13, m_configBitrate)]);
+               args << "-b" << QString::number(g_mp3BitrateLUT[qBound(0, m_configBitrate, 13)]);
                break;
        default:
-               throw "Bad rate-control mode!";
+               MUTILS_THROW("Bad rate-control mode!");
                break;
        }
 
-       if(!metaInfo.fileName().isEmpty()) args << (IS_UNICODE(metaInfo.fileName()) ? "--uTitle" : "--lTitle") << metaInfo.fileName();
-       if(!metaInfo.fileArtist().isEmpty()) args << (IS_UNICODE(metaInfo.fileArtist()) ? "--uArtist" : "--lArtist") << metaInfo.fileArtist();
-       if(!metaInfo.fileAlbum().isEmpty()) args << (IS_UNICODE(metaInfo.fileAlbum()) ? "--uAlbum" : "--lAlbum") << metaInfo.fileAlbum();
-       if(!metaInfo.fileGenre().isEmpty()) args << (IS_UNICODE(metaInfo.fileGenre()) ? "--uGenre" : "--lGenre") << metaInfo.fileGenre();
-       if(!metaInfo.fileComment().isEmpty()) args << (IS_UNICODE(metaInfo.fileComment()) ? "--uComment" : "--lComment") << metaInfo.fileComment();
-       if(metaInfo.fileYear()) args << "--ty" << QString::number(metaInfo.fileYear());
-       if(metaInfo.filePosition()) args << "--tn" << QString::number(metaInfo.filePosition());
-       
-       //args << "--tv" << QString().sprintf("Encoder=LameXP v%d.%02d.%04d [%s]", lamexp_version_major(), lamexp_version_minor(), lamexp_version_build(), lamexp_version_release());
+       if((m_configBitrateMaximum > 0) && (m_configBitrateMinimum > 0) && (m_configBitrateMinimum <= m_configBitrateMaximum))
+       {
+               if(m_configRCMode != SettingsModel::CBRMode)
+               {
+                       args << "-b" << QString::number(clipBitrate(m_configBitrateMinimum));
+                       args << "-B" << QString::number(clipBitrate(m_configBitrateMaximum));
+               }
+       }
+
+       if(m_configSamplingRate > 0)
+       {
+               args << "--resample" << QString::number(m_configSamplingRate);
+       }
+
+       switch(m_configChannelMode)
+       {
+       case 1:
+               args << "-m" << "j";
+               break;
+       case 2:
+               args << "-m" << "f";
+               break;
+       case 3:
+               args << "-m" << "s";
+               break;
+       case 4:
+               args << "-m" << "d";
+               break;
+       case 5:
+               args << "-m" << "m";
+               break;
+       }
+
+       bool bUseUCS2 = false;
+
+       if(isUnicode(metaInfo.title()))   bUseUCS2 = true;
+       if(isUnicode(metaInfo.artist()))  bUseUCS2 = true;
+       if(isUnicode(metaInfo.album()))   bUseUCS2 = true;
+       if(isUnicode(metaInfo.genre()))   bUseUCS2 = true;
+       if(isUnicode(metaInfo.comment())) bUseUCS2 = true;
+
+       if(bUseUCS2) args << "--id3v2-ucs2"; //Must specify this BEFORE "--tt" and friends!
+
+       if(!metaInfo.title().isEmpty())   args << "--tt" << cleanTag(metaInfo.title());
+       if(!metaInfo.artist().isEmpty())  args << "--ta" << cleanTag(metaInfo.artist());
+       if(!metaInfo.album().isEmpty())   args << "--tl" << cleanTag( metaInfo.album());
+       if(!metaInfo.genre().isEmpty())   args << "--tg" << cleanTag(metaInfo.genre());
+       if(!metaInfo.comment().isEmpty()) args << "--tc" << cleanTag(metaInfo.comment());
+       if(metaInfo.year())               args << "--ty" << QString::number(metaInfo.year());
+       if(metaInfo.position())           args << "--tn" << QString::number(metaInfo.position());
+       if(!metaInfo.cover().isEmpty())   args << "--ti" << QDir::toNativeSeparators(metaInfo.cover());
+
+       if(!m_configCustomParams.isEmpty()) args << m_configCustomParams.split(" ", QString::SkipEmptyParts);
 
        args << QDir::toNativeSeparators(sourceFile);
        args << QDir::toNativeSeparators(outputFile);
@@ -88,6 +230,7 @@ bool MP3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
 
        bool bTimeout = false;
        bool bAborted = false;
+       int prevProgress = -1;
 
        QRegExp regExp("\\(.*(\\d+)%\\)\\|");
 
@@ -100,11 +243,12 @@ bool MP3Encoder::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();
                        qWarning("LAME process timed out <-- killing!");
+                       emit messageLogged("\nPROCESS TIMEOUT !!!");
                        bTimeout = true;
                        break;
                }
@@ -116,7 +260,11 @@ bool MP3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
                        {
                                bool ok = false;
                                int progress = regExp.cap(1).toInt(&ok);
-                               if(ok) emit statusUpdated(progress);
+                               if(ok && (progress > prevProgress))
+                               {
+                                       emit statusUpdated(progress);
+                                       prevProgress = qMin(progress + 2, 99);
+                               }
                        }
                        else if(!text.isEmpty())
                        {
@@ -132,9 +280,10 @@ bool MP3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
                process.waitForFinished(-1);
        }
        
+       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;
        }
@@ -142,11 +291,6 @@ bool MP3Encoder::encode(const QString &sourceFile, const AudioFileModel &metaInf
        return true;
 }
 
-QString MP3Encoder::extension(void)
-{
-       return "mp3";
-}
-
 bool MP3Encoder::isFormatSupported(const QString &containerType, const QString &containerProfile, const QString &formatType, const QString &formatProfile, const QString &formatVersion)
 {
        if(containerType.compare("Wave", Qt::CaseInsensitive) == 0)
@@ -172,3 +316,60 @@ bool MP3Encoder::isFormatSupported(const QString &containerType, const QString &
 
        return false;
 }
+
+const unsigned int *MP3Encoder::supportedChannelCount(void)
+{
+       static const unsigned int supportedChannels[] = {1, 2, NULL};
+       return supportedChannels;
+}
+
+void MP3Encoder::setAlgoQuality(int value)
+{
+       m_algorithmQuality = qBound(0, value, 3);
+}
+
+void MP3Encoder::setBitrateLimits(int minimumBitrate, int maximumBitrate)
+{
+       m_configBitrateMinimum = minimumBitrate;
+       m_configBitrateMaximum = maximumBitrate;
+}
+
+void MP3Encoder::setSamplingRate(int value)
+{
+       m_configSamplingRate = value;
+}
+
+void MP3Encoder::setChannelMode(int value)
+{
+       m_configChannelMode = value;
+}
+
+int MP3Encoder::clipBitrate(int bitrate)
+{
+       int targetBitrate = qMin(qMax(bitrate, 32), 320);
+       
+       int minDiff = INT_MAX;
+       int minIndx = -1;
+
+       for(int i = 0; g_mp3BitrateLUT[i] > 0; i++)
+       {
+               int currentDiff = abs(targetBitrate - g_mp3BitrateLUT[i]);
+               if(currentDiff < minDiff)
+               {
+                       minDiff = currentDiff;
+                       minIndx = i;
+               }
+       }
+
+       if(minIndx >= 0)
+       {
+               return g_mp3BitrateLUT[minIndx];
+       }
+
+       return targetBitrate;
+}
+
+const AbstractEncoderInfo *MP3Encoder::getEncoderInfo(void)
+{
+       return &g_mp3EncoderInfo;
+}