src/Thread_FileAnalyzer.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 "Thread_FileAnalyzer.h"
  23
  24 #include "Global.h"
  25 #include "LockedFile.h"
  26 #include "Model_AudioFile.h"
  27 #include "Thread_FileAnalyzer_Task.h"
  28
  29 #include <QDir>
  30 #include <QFileInfo>
  31 #include <QProcess>
  32 #include <QDate>
  33 #include <QTime>
  34 #include <QDebug>
  35 #include <QImage>
  36 #include <QThreadPool>
  37
  38 ////////////////////////////////////////////////////////////
  39 // Constructor
  40 ////////////////////////////////////////////////////////////
  41
  42 FileAnalyzer::FileAnalyzer(const QStringList &inputFiles)
  43 :
  44         m_abortFlag(false),
  45         m_inputFiles(inputFiles),
  46         m_templateFile(NULL)
  47 {
  48         m_bSuccess = false;
  49         m_bAborted = false;
  50 }
  51
  52 FileAnalyzer::~FileAnalyzer(void)
  53 {
  54         if(m_templateFile)
  55         {
  56                 QString templatePath = m_templateFile->filePath();
  57                 LAMEXP_DELETE(m_templateFile);
  58                 if(QFile::exists(templatePath)) QFile::remove(templatePath);
  59         }
  60
  61         AnalyzeTask::reset();
  62 }
  63
  64 ////////////////////////////////////////////////////////////
  65 // Static data
  66 ////////////////////////////////////////////////////////////
  67
  68 const char *FileAnalyzer::g_tags_gen[] =
  69 {
  70         "ID",
  71         "Format",
  72         "Format_Profile",
  73         "Format_Version",
  74         "Duration",
  75         "Title", "Track",
  76         "Track/Position",
  77         "Artist", "Performer",
  78         "Album",
  79         "Genre",
  80         "Released_Date", "Recorded_Date",
  81         "Comment",
  82         "Cover",
  83         "Cover_Type",
  84         "Cover_Mime",
  85         "Cover_Data",
  86         NULL
  87 };
  88
  89 const char *FileAnalyzer::g_tags_aud[] =
  90 {
  91         "ID",
  92         "Source",
  93         "Format",
  94         "Format_Profile",
  95         "Format_Version",
  96         "Channel(s)",
  97         "SamplingRate",
  98         "BitDepth",
  99         "BitRate",
 100         "BitRate_Mode",
 101         NULL
 102 };
 103
 104 ////////////////////////////////////////////////////////////
 105 // Thread Main
 106 ////////////////////////////////////////////////////////////
 107
 108 void FileAnalyzer::run()
 109 {
 110         m_bSuccess = false;
 111         m_bAborted = false;
 112
 113         m_inputFiles.sort();
 114         m_abortFlag = false;
 115
 116         if(!m_templateFile)
 117         {
 118                 if(!createTemplate())
 119                 {
 120                         qWarning("Failed to create template file!");
 121                         return;
 122                 }
 123         }
 124
 125         AnalyzeTask::reset();
 126         QThreadPool *pool = new QThreadPool();
 127
 128         while(!(m_inputFiles.isEmpty() || m_bAborted))
 129         {
 130                 while(!(m_inputFiles.isEmpty() || m_bAborted))
 131                 {
 132                         const QString currentFile = QDir::fromNativeSeparators(m_inputFiles.takeFirst());
 133
 134                         if(m_inputFiles.isEmpty())
 135                         {
 136                                 pool->waitForDone();
 137                         }
 138
 139                         AnalyzeTask *task = new AnalyzeTask(currentFile, m_templateFile->filePath(), &m_abortFlag);
 140                         connect(task, SIGNAL(fileSelected(QString)), this, SIGNAL(fileSelected(QString)), Qt::DirectConnection);
 141                         connect(task, SIGNAL(fileAnalyzed(AudioFileModel)), this, SIGNAL(fileAnalyzed(AudioFileModel)), Qt::DirectConnection);
 142
 143                         while(!pool->tryStart(task))
 144                         {
 145                                 QThread::msleep(125); //No more free threads, wait for active threads!
 146                                 if(m_abortFlag) { LAMEXP_DELETE(task); break; }
 147                         }
 148
 149                         if(m_abortFlag)
 150                         {
 151                                 MessageBeep(MB_ICONERROR);
 152                                 m_bAborted = true;
 153                         }
 154
 155                         QThread::yieldCurrentThread();
 156                 }
 157
 158                 if(!m_bAborted)
 159                 {
 160                         pool->waitForDone();
 161                         AnalyzeTask::getAdditionalFiles(m_inputFiles);
 162                 }
 163         }
 164
 165         pool->waitForDone();
 166         LAMEXP_DELETE(pool);
 167
 168
 169         if(m_bAborted)
 170         {
 171                 qWarning("Operation cancelled by user!");
 172                 return;
 173         }
 174
 175         qDebug("All files added.\n");
 176         m_bSuccess = true;
 177 }
 178
 179 ////////////////////////////////////////////////////////////
 180 // Privtae Functions
 181 ////////////////////////////////////////////////////////////
 182
 183 bool FileAnalyzer::createTemplate(void)
 184 {
 185         if(m_templateFile)
 186         {
 187                 qWarning("Template file already exists!");
 188                 return true;
 189         }
 190
 191         QString templatePath = QString("%1/%2.txt").arg(lamexp_temp_folder2(), lamexp_rand_str());
 192
 193         QFile templateFile(templatePath);
 194         if(!templateFile.open(QIODevice::WriteOnly))
 195         {
 196                 return false;
 197         }
 198
 199         templateFile.write("General;");
 200         for(size_t i = 0; g_tags_gen[i]; i++)
 201         {
 202                 templateFile.write(QString("Gen_%1=%%1%\\n").arg(g_tags_gen[i]).toLatin1().constData());
 203         }
 204         templateFile.write("\\n\r\n");
 205
 206         templateFile.write("Audio;");
 207         for(size_t i = 0; g_tags_aud[i]; i++)
 208         {
 209                 templateFile.write(QString("Aud_%1=%%1%\\n").arg(g_tags_aud[i]).toLatin1().constData());
 210         }
 211         templateFile.write("\\n\r\n");
 212
 213         bool success = (templateFile.error() == QFile::NoError);
 214         templateFile.close();
 215
 216         if(!success)
 217         {
 218                 QFile::remove(templatePath);
 219                 return false;
 220         }
 221
 222         try
 223         {
 224                 m_templateFile = new LockedFile(templatePath);
 225         }
 226         catch(...)
 227         {
 228                 qWarning("Failed to lock template file!");
 229                 return false;
 230         }
 231
 232         return true;
 233 }
 234
 235 ////////////////////////////////////////////////////////////
 236 // Public Functions
 237 ////////////////////////////////////////////////////////////
 238
 239 unsigned int FileAnalyzer::filesAccepted(void)
 240 {
 241         return AnalyzeTask::filesAccepted();
 242 }
 243
 244 unsigned int FileAnalyzer::filesRejected(void)
 245 {
 246         return AnalyzeTask::filesRejected();
 247 }
 248
 249 unsigned int FileAnalyzer::filesDenied(void)
 250 {
 251         return AnalyzeTask::filesDenied();
 252 }
 253
 254 unsigned int FileAnalyzer::filesDummyCDDA(void)
 255 {
 256         return AnalyzeTask::filesDummyCDDA();
 257 }
 258
 259 unsigned int FileAnalyzer::filesCueSheet(void)
 260 {
 261         return AnalyzeTask::filesCueSheet();
 262 }
 263
 264 ////////////////////////////////////////////////////////////
 265 // EVENTS
 266 ////////////////////////////////////////////////////////////
 267
 268 /*NONE*/