OSDN Git Service

Updated Monkey's Audio binary to v4.11 (2013-01-20), including STDERR flush fix.
[lamexp/LameXP.git] / src / Model_CueSheet.cpp
index b05c23f..c2e18d0 100644 (file)
@@ -1,6 +1,6 @@
 ///////////////////////////////////////////////////////////////////////////////
 // LameXP - Audio Encoder Front-End
-// Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
+// Copyright (C) 2004-2013 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
 #include <QDir>
 #include <QFileInfo>
 #include <QFont>
+#include <QTime>
+#include <QTextCodec>
+#include <QTextStream>
 
 #include <float.h>
 #include <limits>
 
+#define UNQUOTE(STR) STR.split("\"",  QString::SkipEmptyParts).first().trimmed()
+
 ////////////////////////////////////////////////////////////
 // Helper Classes
 ////////////////////////////////////////////////////////////
@@ -52,17 +57,22 @@ public:
        {
                m_startIndex = std::numeric_limits<double>::quiet_NaN();
                m_duration = std::numeric_limits<double>::infinity();
+               m_year = 0;
        }
        int trackNo(void) { return m_trackNo; }
        double startIndex(void) { return m_startIndex; }
        double duration(void) { return m_duration; }
        QString title(void) { return m_title; }
        QString performer(void) { return m_performer; }
+       QString genre(void) { return m_genre; }
+       unsigned int year(void) { return m_year; }
        CueSheetFile *parent(void) { return m_parent; }
        void setStartIndex(double startIndex) { m_startIndex = startIndex; }
        void setDuration(double duration) { m_duration = duration; }
        void setTitle(const QString &title, bool update = false) { if(!update || (m_title.isEmpty() && !title.isEmpty())) m_title = title; }
        void setPerformer(const QString &performer, bool update = false) { if(!update || (m_performer.isEmpty() && !performer.isEmpty())) m_performer = performer; }
+       void setGenre(const QString &genre, bool update = false) { if(!update || (m_genre.isEmpty() && !m_genre.isEmpty())) m_genre = genre; }
+       void setYear(const unsigned int year, bool update = false) { if(!update || (year == 0)) m_year = year; }
        virtual bool isValid(void) { return !(_isnan(m_startIndex) || (m_trackNo < 0)); }
        virtual const char* type(void) { return "CueSheetTrack"; }
 private:
@@ -71,6 +81,8 @@ private:
        double m_duration;
        QString m_title;
        QString m_performer;
+       QString m_genre;
+       unsigned int m_year;
        CueSheetFile *m_parent;
 };
 
@@ -95,12 +107,15 @@ private:
 // Constructor & Destructor
 ////////////////////////////////////////////////////////////
 
+QMutex CueSheetModel::m_mutex(QMutex::Recursive);
+
 CueSheetModel::CueSheetModel()
 :
        m_fileIcon(":/icons/music.png"),
        m_trackIcon(":/icons/control_play_blue.png")
 {
        int trackNo = 0;
+       m_albumYear = 0;
        
        for(int i = 0; i < 5; i++)
        {
@@ -127,15 +142,17 @@ CueSheetModel::~CueSheetModel(void)
 
 QModelIndex CueSheetModel::index(int row, int column, const QModelIndex &parent) const
 {
+       QMutexLocker lock(&m_mutex);
+       
        if(!parent.isValid())
        {
-               return createIndex(row, column, m_files.at(row));
+               return (row < m_files.count()) ? createIndex(row, column, m_files.at(row)) : QModelIndex();
        }
 
        CueSheetItem *parentItem = static_cast<CueSheetItem*>(parent.internalPointer());
        if(CueSheetFile *filePtr = dynamic_cast<CueSheetFile*>(parentItem))
        {
-               return createIndex(row, column, filePtr->track(row));
+               return (row < filePtr->trackCount()) ? createIndex(row, column, filePtr->track(row)) : QModelIndex();
        }
 
        return QModelIndex();
@@ -143,11 +160,14 @@ QModelIndex CueSheetModel::index(int row, int column, const QModelIndex &parent)
 
 int CueSheetModel::columnCount(const QModelIndex &parent) const
 {
+       QMutexLocker lock(&m_mutex);
        return 4;
 }
 
 int CueSheetModel::rowCount(const QModelIndex &parent) const
 {
+       QMutexLocker lock(&m_mutex);
+
        if(!parent.isValid())
        {
                return m_files.count();
@@ -164,6 +184,8 @@ int CueSheetModel::rowCount(const QModelIndex &parent) const
 
 QModelIndex CueSheetModel::parent(const QModelIndex &child) const
 {
+       QMutexLocker lock(&m_mutex);
+       
        if(child.isValid())
        {
                CueSheetItem *childItem = static_cast<CueSheetItem*>(child.internalPointer());
@@ -178,6 +200,8 @@ QModelIndex CueSheetModel::parent(const QModelIndex &child) const
 
 QVariant CueSheetModel::headerData (int section, Qt::Orientation orientation, int role) const
 {
+       QMutexLocker lock(&m_mutex);
+       
        if(role == Qt::DisplayRole)
        {
                switch(section)
@@ -207,6 +231,8 @@ QVariant CueSheetModel::headerData (int section, Qt::Orientation orientation, in
 
 QVariant CueSheetModel::data(const QModelIndex &index, int role) const
 {
+       QMutexLocker lock(&m_mutex);
+
        if(role == Qt::DisplayRole)
        {
                CueSheetItem *item = reinterpret_cast<CueSheetItem*>(index.internalPointer());
@@ -331,6 +357,8 @@ QVariant CueSheetModel::data(const QModelIndex &index, int role) const
 
 void CueSheetModel::clearData(void)
 {
+       QMutexLocker lock(&m_mutex);
+       
        beginResetModel();
        while(!m_files.isEmpty()) delete m_files.takeLast();
        endResetModel();
@@ -342,11 +370,14 @@ void CueSheetModel::clearData(void)
 
 int CueSheetModel::getFileCount(void)
 {
+       QMutexLocker lock(&m_mutex);
        return m_files.count();
 }
 
 QString CueSheetModel::getFileName(int fileIndex)
 {
+       QMutexLocker lock(&m_mutex);
+       
        if(fileIndex < 0 || fileIndex >= m_files.count())
        {
                return QString();
@@ -357,6 +388,8 @@ QString CueSheetModel::getFileName(int fileIndex)
 
 int CueSheetModel::getTrackCount(int fileIndex)
 {
+       QMutexLocker lock(&m_mutex);
+
        if(fileIndex < 0 || fileIndex >= m_files.count())
        {
                return -1;
@@ -367,6 +400,8 @@ int CueSheetModel::getTrackCount(int fileIndex)
 
 int CueSheetModel::getTrackNo(int fileIndex, int trackIndex)
 {
+       QMutexLocker lock(&m_mutex);
+       
        if(fileIndex >= 0 && fileIndex < m_files.count())
        {
                CueSheetFile *currentFile = m_files.at(fileIndex);
@@ -381,6 +416,8 @@ int CueSheetModel::getTrackNo(int fileIndex, int trackIndex)
 
 void CueSheetModel::getTrackIndex(int fileIndex, int trackIndex, double *startIndex, double *duration)
 {
+       QMutexLocker lock(&m_mutex);
+       
        *startIndex = std::numeric_limits<double>::quiet_NaN();
        *duration = std::numeric_limits<double>::quiet_NaN();
 
@@ -398,6 +435,8 @@ void CueSheetModel::getTrackIndex(int fileIndex, int trackIndex, double *startIn
 
 QString CueSheetModel::getTrackPerformer(int fileIndex, int trackIndex)
 {      
+       QMutexLocker lock(&m_mutex);
+       
        if(fileIndex >= 0 && fileIndex < m_files.count())
        {
                CueSheetFile *currentFile = m_files.at(fileIndex);
@@ -407,11 +446,14 @@ QString CueSheetModel::getTrackPerformer(int fileIndex, int trackIndex)
                        return currentTrack->performer();
                }
        }
+       
        return QString();
 }
 
 QString CueSheetModel::getTrackTitle(int fileIndex, int trackIndex)
 {
+       QMutexLocker lock(&m_mutex);
+       
        if(fileIndex >= 0 && fileIndex < m_files.count())
        {
                CueSheetFile *currentFile = m_files.at(fileIndex);
@@ -421,15 +463,76 @@ QString CueSheetModel::getTrackTitle(int fileIndex, int trackIndex)
                        return currentTrack->title();
                }
        }
+       
+       return QString();
+}
+
+QString CueSheetModel::getTrackGenre(int fileIndex, int trackIndex)
+{
+       QMutexLocker lock(&m_mutex);
+       
+       if(fileIndex >= 0 && fileIndex < m_files.count())
+       {
+               CueSheetFile *currentFile = m_files.at(fileIndex);
+               if(trackIndex >= 0 && trackIndex < currentFile->trackCount())
+               {
+                       CueSheetTrack *currentTrack = currentFile->track(trackIndex);
+                       return currentTrack->genre();
+               }
+       }
+       
        return QString();
 }
 
+unsigned int CueSheetModel::getTrackYear(int fileIndex, int trackIndex)
+{
+       QMutexLocker lock(&m_mutex);
+       
+       if(fileIndex >= 0 && fileIndex < m_files.count())
+       {
+               CueSheetFile *currentFile = m_files.at(fileIndex);
+               if(trackIndex >= 0 && trackIndex < currentFile->trackCount())
+               {
+                       CueSheetTrack *currentTrack = currentFile->track(trackIndex);
+                       return currentTrack->year();
+               }
+       }
+       
+       return 0;
+}
+
+QString CueSheetModel::getAlbumPerformer(void)
+{
+       QMutexLocker lock(&m_mutex);
+       return m_albumPerformer;
+}
+
+QString CueSheetModel::getAlbumTitle(void)
+{
+       QMutexLocker lock(&m_mutex);
+       return m_albumTitle;
+}
+
+QString CueSheetModel::getAlbumGenre(void)
+{
+       QMutexLocker lock(&m_mutex);
+       return m_albumGenre;
+}
+
+unsigned int CueSheetModel::getAlbumYear(void)
+{
+       QMutexLocker lock(&m_mutex);
+       return m_albumYear;
+}
 ////////////////////////////////////////////////////////////
 // Cue Sheet Parser
 ////////////////////////////////////////////////////////////
 
-int CueSheetModel::loadCueSheet(const QString &cueFileName, QCoreApplication *application)
+int CueSheetModel::loadCueSheet(const QString &cueFileName, QCoreApplication *application, QTextCodec *forceCodec)
 {
+       QMutexLocker lock(&m_mutex);
+       const QTextCodec *codec = (forceCodec != NULL) ? forceCodec : QTextCodec::codecForName("System");
+       
        QFile cueFile(cueFileName);
        if(!cueFile.open(QIODevice::ReadOnly))
        {
@@ -439,16 +542,17 @@ int CueSheetModel::loadCueSheet(const QString &cueFileName, QCoreApplication *ap
        clearData();
 
        beginResetModel();
-       int iResult = parseCueFile(cueFile, QDir(QFileInfo(cueFile).canonicalPath()), application);
+       int iResult = parseCueFile(cueFile, QDir(QFileInfo(cueFile).canonicalPath()), application, codec);
        endResetModel();
 
        return iResult;
 }
 
-int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplication *application)
+int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplication *application, const QTextCodec *codec)
 {
-       cueFile.seek(0);
+       cueFile.reset();
        qDebug("\n[Cue Sheet Import]");
+       bool bForceLatin1 = false;
 
        //Reject very large files, as parsing might take until forever
        if(cueFile.size() >= 10485760i64)
@@ -457,17 +561,31 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                return 2;
        }
 
-       //Check for UTF-8 BOM in order to guess encoding
-       QByteArray bomCheck = cueFile.peek(128);
-       bool bUTF8 = bomCheck.contains("\xef\xbb\xbf");
-       qDebug("Encoding is %s.", (bUTF8 ? "UTF-8" : "Local 8-Bit"));
-       bomCheck.clear();
+       //Test selected Codepage for decoding errors
+       qDebug("Character encoding is: %s.", codec->name().constData());
+       const QString replacementSymbol = QString(QChar(QChar::ReplacementCharacter));
+       QByteArray testData = cueFile.peek(1048576);
+       if((!testData.isEmpty()) && codec->toUnicode(testData.constData(), testData.size()).contains(replacementSymbol))
+       {
+               qWarning("Decoding error using selected codepage (%s). Enforcing Latin-1.", codec->name().constData());
+               bForceLatin1 = true;
+       }
+       testData.clear();
+
+       //Init text stream
+       QTextStream cueStream(&cueFile);
+       cueStream.setAutoDetectUnicode(false);
+       cueStream.setCodec(bForceLatin1 ? "latin1" : codec->name());
+       cueStream.seek(0i64);
 
-       QRegExp rxFile("^FILE\\s+\"([^\"]+)\"\\s+(\\w+)$", Qt::CaseInsensitive);
+       //Create regular expressions
+       QRegExp rxFile("^FILE\\s+(\"[^\"]+\"|\\S+)\\s+(\\w+)$", Qt::CaseInsensitive);
        QRegExp rxTrack("^TRACK\\s+(\\d+)\\s(\\w+)$", Qt::CaseInsensitive);
        QRegExp rxIndex("^INDEX\\s+(\\d+)\\s+([0-9:]+)$", Qt::CaseInsensitive);
-       QRegExp rxTitle("^TITLE\\s+\"([^\"]+)\"$", Qt::CaseInsensitive);
-       QRegExp rxPerformer("^PERFORMER\\s+\"([^\"]+)\"$", Qt::CaseInsensitive);
+       QRegExp rxTitle("^TITLE\\s+(\"[^\"]+\"|\\S+)$", Qt::CaseInsensitive);
+       QRegExp rxPerformer("^PERFORMER\\s+(\"[^\"]+\"|\\S+)$", Qt::CaseInsensitive);
+       QRegExp rxGenre("^REM\\s+GENRE\\s+(\"[^\"]+\"|\\S+)$", Qt::CaseInsensitive);
+       QRegExp rxYear("^REM\\s+DATE\\s+(\\d+)$", Qt::CaseInsensitive);
        
        bool bPreamble = true;
        bool bUnsupportedTrack = false;
@@ -477,6 +595,8 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
 
        m_albumTitle.clear();
        m_albumPerformer.clear();
+       m_albumGenre.clear();
+       m_albumYear = 0;
 
        //Loop over the Cue Sheet until all lines were processed
        for(int lines = 0; lines < INT_MAX; lines++)
@@ -487,14 +607,13 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                        if(lines < 128) Sleep(10);
                }
                
-               QByteArray lineData = cueFile.readLine();
-               if(lineData.size() <= 0)
+               if(cueStream.atEnd())
                {
                        qDebug("End of Cue Sheet file.");
                        break;
                }
 
-               QString line = bUTF8 ? QString::fromUtf8(lineData.constData(), lineData.size()).trimmed() : QString::fromLocal8Bit(lineData.constData(), lineData.size()).trimmed();
+               QString line = cueStream.readLine().trimmed();
                
                /* --- FILE --- */
                if(rxFile.indexIn(line) >= 0)
@@ -530,7 +649,7 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                        }
                        if(!rxFile.cap(2).compare("WAVE", Qt::CaseInsensitive) || !rxFile.cap(2).compare("MP3", Qt::CaseInsensitive) || !rxFile.cap(2).compare("AIFF", Qt::CaseInsensitive))
                        {
-                               currentFile = new CueSheetFile(baseDir.absoluteFilePath(rxFile.cap(1)));
+                               currentFile = new CueSheetFile(baseDir.absoluteFilePath(UNQUOTE(rxFile.cap(1))));
                                qDebug("%03d File path: <%s>", lines, currentFile->fileName().toUtf8().constData());
                        }
                        else
@@ -600,12 +719,12 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                {
                        if(bPreamble)
                        {
-                               m_albumTitle = rxTitle.cap(1).simplified();
+                               m_albumTitle = UNQUOTE(rxTitle.cap(1)).simplified();
                        }
                        else if(currentFile && currentTrack)
                        {
                                qDebug("%03d     Title: <%s>", lines, rxTitle.cap(1).toUtf8().constData());
-                               currentTrack->setTitle(rxTitle.cap(1));
+                               currentTrack->setTitle(UNQUOTE(rxTitle.cap(1)).simplified());
                        }
                        continue;
                }
@@ -615,12 +734,62 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                {
                        if(bPreamble)
                        {
-                               m_albumPerformer = rxPerformer.cap(1).simplified();
+                               m_albumPerformer = UNQUOTE(rxPerformer.cap(1)).simplified();
                        }
                        else if(currentFile && currentTrack)
                        {
                                qDebug("%03d     Title: <%s>", lines, rxPerformer.cap(1).toUtf8().constData());
-                               currentTrack->setPerformer(rxPerformer.cap(1));
+                               currentTrack->setPerformer(UNQUOTE(rxPerformer.cap(1)).simplified());
+                       }
+                       continue;
+               }
+
+               /* --- GENRE --- */
+               if(rxGenre.indexIn(line) >= 0)
+               {
+                       if(bPreamble)
+                       {
+                               QString temp = UNQUOTE(rxGenre.cap(1)).simplified();
+                               for(int i = 0; g_lamexp_generes[i]; i++)
+                               {
+                                       if(temp.compare(g_lamexp_generes[i], Qt::CaseInsensitive) == 0)
+                                       {
+                                               m_albumGenre = QString(g_lamexp_generes[i]);
+                                               break;
+                                       }
+                               }
+                       }
+                       else if(currentFile && currentTrack)
+                       {
+                               qDebug("%03d     Genre: <%s>", lines, rxGenre.cap(1).toUtf8().constData());
+                               QString temp = UNQUOTE(rxGenre.cap(1).simplified());
+                               for(int i = 0; g_lamexp_generes[i]; i++)
+                               {
+                                       if(temp.compare(g_lamexp_generes[i], Qt::CaseInsensitive) == 0)
+                                       {
+                                               currentTrack->setGenre(QString(g_lamexp_generes[i]));
+                                               break;
+                                       }
+                               }
+                       }
+                       continue;
+               }
+
+               /* --- YEAR --- */
+               if(rxYear.indexIn(line) >= 0)
+               {
+                       if(bPreamble)
+                       {
+                               bool ok = false;
+                               unsigned int temp = rxYear.cap(1).toUInt(&ok);
+                               if(ok) m_albumYear =  temp;
+                       }
+                       else if(currentFile && currentTrack)
+                       {
+                               qDebug("%03d     Year: <%s>", lines, rxPerformer.cap(1).toUtf8().constData());
+                               bool ok = false;
+                               unsigned int temp = rxYear.cap(1).toUInt(&ok);
+                               if(ok) currentTrack->setYear(temp);
                        }
                        continue;
                }
@@ -671,7 +840,7 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                                CueSheetTrack *currentTrack = currentFile->track(j);
                                CueSheetTrack *previousTrack = currentFile->track(j-1);
                                double duration = currentTrack->startIndex() - previousTrack->startIndex();
-                               previousTrack->setDuration(max(0.0, duration));
+                               previousTrack->setDuration(qMax(0.0, duration));
                        }
                }
        }
@@ -679,12 +848,14 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
        //Sanity check of track numbers
        if(nFiles > 0)
        {
+               bool hasTracks = false;
                int previousTrackNo = -1;
                bool trackNo[100];
                for(int i = 0; i < 100; i++)
                {
                        trackNo[i] = false;
                }
+
                for(int i = 0; i < nFiles; i++)
                {
                        if(application)
@@ -706,7 +877,7 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                                        }
                                        if(currentTrackNo <= previousTrackNo)
                                        {
-                                               qWarning("Non-increasing track numbers, Cue Sheet is inconsistent!", currentTrackNo);
+                                               qWarning("Non-increasing track numbers (%02d -> %02d), Cue Sheet is inconsistent!", previousTrackNo, currentTrackNo);
                                                return ErrorInconsistent;
                                        }
                                        if(trackNo[currentTrackNo])
@@ -716,13 +887,22 @@ int CueSheetModel::parseCueFile(QFile &cueFile, const QDir &baseDir, QCoreApplic
                                        }
                                        trackNo[currentTrackNo] = true;
                                        previousTrackNo = currentTrackNo;
+                                       hasTracks = true;
                                }
                        }
                }
+               
+               if(!hasTracks)
+               {
+                       qWarning("Could not find at least one valid track in the Cue Sheet!");
+                       return ErrorInconsistent;
+               }
+
                return ErrorSuccess;
        }
        else
        {
+               qWarning("Could not find at least one valid input file in the Cue Sheet!");
                return bUnsupportedTrack ? ErrorUnsupported : ErrorBadFile;
        }
 }
@@ -742,7 +922,7 @@ double CueSheetModel::parseTimeIndex(const QString &index)
 
                if(minOK && secOK && frmOK)
                {
-                       return static_cast<double>(60 * min) + static_cast<double>(sec) + ((1.0/75.0) * static_cast<double>(frm));
+                       return static_cast<double>(60 * min) + static_cast<double>(sec) + (static_cast<double>(frm) / 75.0);
                }
        }
 
@@ -752,21 +932,19 @@ double CueSheetModel::parseTimeIndex(const QString &index)
 
 QString CueSheetModel::indexToString(const double index) const
 {
-       if(index == std::numeric_limits<double>::quiet_NaN())
+       if(!_finite(index) || (index < 0.0) || (index > 86400.0))
        {
-               return QString("<-NaN!->");
+               return QString("??:??.???");
        }
-       else if(index == std::numeric_limits<double>::infinity() || index < 0.0)
+       
+       QTime time = QTime().addMSecs(static_cast<int>(floor(0.5 + (index * 1000.0))));
+
+       if(time.minute() < 100)
        {
-               return QString("??:??.??");
+               return time.toString("mm:ss.zzz");
        }
        else
        {
-               int temp = static_cast<int>(floor(0.5 + (index * 100.0)));
-
-               int msec = temp % 100;
-               int secs = temp / 100;
-
-               return QString().sprintf("%02d:%02d.%02d", min(99, secs / 60), min(99, secs % 60), min(99, msec));
+               return QString("99:99.999");
        }
 }