src/Thread_Process.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2016 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, but always including the *additional*
   9 // restrictions defined in the "License.txt" file.
  10 //
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License along
  17 // with this program; if not, write to the Free Software Foundation, Inc.,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 // http://www.gnu.org/licenses/gpl-2.0.txt
  21 ///////////////////////////////////////////////////////////////////////////////
  22
  23 #include "Thread_Process.h"
  24
  25 //Internal
  26 #include "Global.h"
  27 #include "Model_AudioFile.h"
  28 #include "Model_Progress.h"
  29 #include "Encoder_Abstract.h"
  30 #include "Decoder_Abstract.h"
  31 #include "Filter_Abstract.h"
  32 #include "Filter_Downmix.h"
  33 #include "Filter_Resample.h"
  34 #include "Tool_WaveProperties.h"
  35 #include "Registry_Decoder.h"
  36 #include "Model_Settings.h"
  37
  38 //MUtils
  39 #include <MUtils/Global.h>
  40 #include <MUtils/OSSupport.h>
  41 #include <MUtils/Version.h>
  42
  43 //Qt
  44 #include <QUuid>
  45 #include <QFileInfo>
  46 #include <QDir>
  47 #include <QMutex>
  48 #include <QMutexLocker>
  49 #include <QDate>
  50 #include <QThreadPool>
  51
  52 //CRT
  53 #include <limits.h>
  54 #include <time.h>
  55 #include <stdlib.h>
  56
  57 #define DIFF(X,Y) ((X > Y) ? (X-Y) : (Y-X))
  58 #define IS_WAVE(X) ((X.containerType().compare("Wave", Qt::CaseInsensitive) == 0) && (X.audioType().compare("PCM", Qt::CaseInsensitive) == 0))
  59 #define STRDEF(STR,DEF) ((!STR.isEmpty()) ? STR : DEF)
  60
  61 ////////////////////////////////////////////////////////////
  62 // Constructor
  63 ////////////////////////////////////////////////////////////
  64
  65 ProcessThread::ProcessThread(const AudioFileModel &audioFile, const QString &outputDirectory, const QString &tempDirectory, AbstractEncoder *encoder, const bool prependRelativeSourcePath)
  66 :
  67         m_audioFile(audioFile),
  68         m_outputDirectory(outputDirectory),
  69         m_tempDirectory(tempDirectory),
  70         m_encoder(encoder),
  71         m_jobId(QUuid::createUuid()),
  72         m_prependRelativeSourcePath(prependRelativeSourcePath),
  73         m_renamePattern("<BaseName>"),
  74         m_overwriteMode(OverwriteMode_KeepBoth),
  75         m_keepDateTime(false),
  76         m_initialized(-1),
  77         m_aborted(false),
  78         m_propDetect(new WaveProperties())
  79 {
  80         connect(m_encoder, SIGNAL(statusUpdated(int)), this, SLOT(handleUpdate(int)), Qt::DirectConnection);
  81         connect(m_encoder, SIGNAL(messageLogged(QString)), this, SLOT(handleMessage(QString)), Qt::DirectConnection);
  82
  83         connect(m_propDetect, SIGNAL(statusUpdated(int)), this, SLOT(handleUpdate(int)), Qt::DirectConnection);
  84         connect(m_propDetect, SIGNAL(messageLogged(QString)), this, SLOT(handleMessage(QString)), Qt::DirectConnection);
  85
  86         m_currentStep = UnknownStep;
  87 }
  88
  89 ProcessThread::~ProcessThread(void)
  90 {
  91         while(!m_tempFiles.isEmpty())
  92         {
  93                 MUtils::remove_file(m_tempFiles.takeFirst());
  94         }
  95
  96         while(!m_filters.isEmpty())
  97         {
  98                 delete m_filters.takeFirst();
  99         }
 100
 101         MUTILS_DELETE(m_encoder);
 102         MUTILS_DELETE(m_propDetect);
 103
 104         emit processFinished();
 105 }
 106
 107 ////////////////////////////////////////////////////////////
 108 // Init Function
 109 ////////////////////////////////////////////////////////////
 110
 111 bool ProcessThread::init(void)
 112 {
 113         if(m_initialized < 0)
 114         {
 115                 m_initialized = 0;
 116
 117                 //Initialize job status
 118                 qDebug("Process thread %s has started.", m_jobId.toString().toLatin1().constData());
 119                 emit processStateInitialized(m_jobId, QFileInfo(m_audioFile.filePath()).fileName(), tr("Starting..."), ProgressModel::JobRunning);
 120
 121                 //Initialize log
 122                 handleMessage(QString().sprintf("LameXP v%u.%02u (Build #%u), compiled on %s at %s", lamexp_version_major(), lamexp_version_minor(), lamexp_version_build(), MUTILS_UTF8(MUtils::Version::app_build_date().toString(Qt::ISODate)), MUTILS_UTF8(MUtils::Version::app_build_time().toString(Qt::ISODate))));
 123                 handleMessage("\n-------------------------------\n");
 124
 125                 return true;
 126         }
 127
 128         qWarning("[ProcessThread::init] Job %s already initialialized, skipping!", m_jobId.toString().toLatin1().constData());
 129         return false;
 130 }
 131
 132 bool ProcessThread::start(QThreadPool *const pool)
 133 {
 134         //Make sure object was initialized correctly
 135         if(m_initialized < 0)
 136         {
 137                 MUTILS_THROW("Object not initialized yet!");
 138         }
 139
 140         if(m_initialized < 1)
 141         {
 142                 m_initialized = 1;
 143
 144                 m_outFileName.clear();
 145                 bool bSuccess = false;
 146
 147                 //Generate output file name
 148                 switch(generateOutFileName(m_outFileName))
 149                 {
 150                 case 1:
 151                         //File name generated successfully :-)
 152                         bSuccess = true;
 153                         pool->start(this);
 154                         break;
 155                 case -1:
 156                         //File name already exists -> skipping!
 157                         emit processStateChanged(m_jobId, tr("Skipped."), ProgressModel::JobSkipped);
 158                         emit processStateFinished(m_jobId, m_outFileName, -1);
 159                         break;
 160                 default:
 161                         //File name could not be generated
 162                         emit processStateChanged(m_jobId, tr("Not found!"), ProgressModel::JobFailed);
 163                         emit processStateFinished(m_jobId, m_outFileName, 0);
 164                         break;
 165                 }
 166
 167                 if(!bSuccess)
 168                 {
 169                         emit processFinished();
 170                 }
 171
 172                 return bSuccess;
 173         }
 174
 175         qWarning("[ProcessThread::start] Job %s already started, skipping!", m_jobId.toString().toLatin1().constData());
 176         return false;
 177 }
 178
 179 ////////////////////////////////////////////////////////////
 180 // Thread Entry Point
 181 ////////////////////////////////////////////////////////////
 182
 183 void ProcessThread::run()
 184 {
 185         try
 186         {
 187                 processFile();
 188         }
 189         catch(const std::exception &error)
 190         {
 191                 MUTILS_PRINT_ERROR("\nGURU MEDITATION !!!\n\nException error:\n%s\n", error.what());
 192                 MUtils::OS::fatal_exit(L"Unhandeled C++ exception error, application will exit!");
 193         }
 194         catch(...)
 195         {
 196                 MUTILS_PRINT_ERROR("\nGURU MEDITATION !!!\n\nUnknown exception error!\n");
 197                 MUtils::OS::fatal_exit(L"Unhandeled C++ exception error, application will exit!");
 198         }
 199 }
 200
 201 void ProcessThread::processFile()
 202 {
 203         m_aborted = false;
 204         bool bSuccess = true;
 205
 206         //Make sure object was initialized correctly
 207         if(m_initialized < 1)
 208         {
 209                 MUTILS_THROW("Object not initialized yet!");
 210         }
 211
 212         QString sourceFile = m_audioFile.filePath();
 213
 214         //-----------------------------------------------------
 215         // Decode source file
 216         //-----------------------------------------------------
 217
 218         const AudioFileModel_TechInfo &formatInfo = m_audioFile.techInfo();
 219         if(!m_filters.isEmpty() || !m_encoder->isFormatSupported(formatInfo.containerType(), formatInfo.containerProfile(), formatInfo.audioType(), formatInfo.audioProfile(), formatInfo.audioVersion()))
 220         {
 221                 m_currentStep = DecodingStep;
 222                 AbstractDecoder *decoder = DecoderRegistry::lookup(formatInfo.containerType(), formatInfo.containerProfile(), formatInfo.audioType(), formatInfo.audioProfile(), formatInfo.audioVersion());
 223
 224                 if(decoder)
 225                 {
 226                         QString tempFile = generateTempFileName();
 227
 228                         connect(decoder, SIGNAL(statusUpdated(int)), this, SLOT(handleUpdate(int)), Qt::DirectConnection);
 229                         connect(decoder, SIGNAL(messageLogged(QString)), this, SLOT(handleMessage(QString)), Qt::DirectConnection);
 230
 231                         bSuccess = decoder->decode(sourceFile, tempFile, &m_aborted);
 232                         MUTILS_DELETE(decoder);
 233
 234                         if(bSuccess)
 235                         {
 236                                 sourceFile = tempFile;
 237                                 m_audioFile.techInfo().setContainerType(QString::fromLatin1("Wave"));
 238                                 m_audioFile.techInfo().setAudioType(QString::fromLatin1("PCM"));
 239
 240                                 if(QFileInfo(sourceFile).size() >= 4294967296i64)
 241                                 {
 242                                         handleMessage(tr("WARNING: Decoded file size exceeds 4 GB, problems might occur!\n"));
 243                                 }
 244
 245                                 handleMessage("\n-------------------------------\n");
 246                         }
 247                 }
 248                 else
 249                 {
 250                         if(QFileInfo(m_outFileName).exists() && (QFileInfo(m_outFileName).size() < 512)) QFile::remove(m_outFileName);
 251                         handleMessage(QString("%1\n%2\n\n%3\t%4\n%5\t%6").arg(tr("The format of this file is NOT supported:"), m_audioFile.filePath(), tr("Container Format:"), m_audioFile.containerInfo(), tr("Audio Format:"), m_audioFile.audioCompressInfo()));
 252                         emit processStateChanged(m_jobId, tr("Unsupported!"), ProgressModel::JobFailed);
 253                         emit processStateFinished(m_jobId, m_outFileName, 0);
 254                         return;
 255                 }
 256         }
 257
 258         //-----------------------------------------------------
 259         // Update audio properties after decode
 260         //-----------------------------------------------------
 261
 262         if(bSuccess && !m_aborted && IS_WAVE(m_audioFile.techInfo()))
 263         {
 264                 if(m_encoder->supportedSamplerates() || m_encoder->supportedBitdepths() || m_encoder->supportedChannelCount() || m_encoder->needsTimingInfo() || !m_filters.isEmpty())
 265                 {
 266                         m_currentStep = AnalyzeStep;
 267                         bSuccess = m_propDetect->detect(sourceFile, &m_audioFile.techInfo(), &m_aborted);
 268
 269                         if(bSuccess)
 270                         {
 271                                 handleMessage("\n-------------------------------\n");
 272
 273                                 //Do we need to take care if Stereo downmix?
 274                                 const unsigned int *const supportedChannelCount = m_encoder->supportedChannelCount();
 275                                 if(supportedChannelCount && supportedChannelCount[0])
 276                                 {
 277                                         insertDownmixFilter(supportedChannelCount);
 278                                 }
 279
 280                                 //Do we need to take care of downsampling the input?
 281                                 const unsigned int *const supportedSamplerates = m_encoder->supportedSamplerates();
 282                                 const unsigned int *const supportedBitdepths = m_encoder->supportedBitdepths();
 283                                 if((supportedSamplerates && supportedSamplerates[0]) || (supportedBitdepths && supportedBitdepths[0]))
 284                                 {
 285                                         insertDownsampleFilter(supportedSamplerates, supportedBitdepths);
 286                                 }
 287                         }
 288                 }
 289         }
 290
 291         //-----------------------------------------------------
 292         // Apply all audio filters
 293         //-----------------------------------------------------
 294
 295         if(bSuccess)
 296         {
 297                 while(!m_filters.isEmpty() && !m_aborted)
 298                 {
 299                         QString tempFile = generateTempFileName();
 300                         AbstractFilter *poFilter = m_filters.takeFirst();
 301                         m_currentStep = FilteringStep;
 302
 303                         connect(poFilter, SIGNAL(statusUpdated(int)), this, SLOT(handleUpdate(int)), Qt::DirectConnection);
 304                         connect(poFilter, SIGNAL(messageLogged(QString)), this, SLOT(handleMessage(QString)), Qt::DirectConnection);
 305
 306                         if(poFilter->apply(sourceFile, tempFile, &m_audioFile.techInfo(), &m_aborted))
 307                         {
 308                                 sourceFile = tempFile;
 309                         }
 310
 311                         handleMessage("\n-------------------------------\n");
 312                         delete poFilter;
 313                 }
 314         }
 315
 316         //-----------------------------------------------------
 317         // Encode audio file
 318         //-----------------------------------------------------
 319
 320         if(bSuccess && !m_aborted)
 321         {
 322                 m_currentStep = EncodingStep;
 323                 bSuccess = m_encoder->encode(sourceFile, m_audioFile.metaInfo(), m_audioFile.techInfo().duration(), m_audioFile.techInfo().audioChannels(), m_outFileName, &m_aborted);
 324         }
 325
 326         //Clean-up
 327         if((!bSuccess) || m_aborted)
 328         {
 329                 QFileInfo fileInfo(m_outFileName);
 330                 if(fileInfo.exists() && (fileInfo.size() < 1024))
 331                 {
 332                         QFile::remove(m_outFileName);
 333                 }
 334         }
 335
 336         //Make sure output file exists
 337         if(bSuccess && (!m_aborted))
 338         {
 339                 const QFileInfo fileInfo(m_outFileName);
 340                 bSuccess = fileInfo.exists() && fileInfo.isFile() && (fileInfo.size() >= 1024);
 341         }
 342
 343         //-----------------------------------------------------
 344         // Finalize
 345         //-----------------------------------------------------
 346
 347         if (bSuccess && (!m_aborted) && m_keepDateTime)
 348         {
 349                 updateFileTime(m_audioFile.filePath(), m_outFileName);
 350         }
 351
 352         MUtils::OS::sleep_ms(12);
 353
 354         //Report result
 355         emit processStateChanged(m_jobId, (m_aborted ? tr("Aborted!") : (bSuccess ? tr("Done.") : tr("Failed!"))), ((bSuccess && !m_aborted) ? ProgressModel::JobComplete : ProgressModel::JobFailed));
 356         emit processStateFinished(m_jobId, m_outFileName, (bSuccess ? 1 : 0));
 357
 358         qDebug("Process thread is done.");
 359 }
 360
 361 ////////////////////////////////////////////////////////////
 362 // SLOTS
 363 ////////////////////////////////////////////////////////////
 364
 365 void ProcessThread::handleUpdate(int progress)
 366 {
 367         //qDebug("Progress: %d\n", progress);
 368
 369         switch(m_currentStep)
 370         {
 371         case EncodingStep:
 372                 emit processStateChanged(m_jobId, QString("%1 (%2%)").arg(tr("Encoding"), QString::number(progress)), ProgressModel::JobRunning);
 373                 break;
 374         case AnalyzeStep:
 375                 emit processStateChanged(m_jobId, QString("%1 (%2%)").arg(tr("Analyzing"), QString::number(progress)), ProgressModel::JobRunning);
 376                 break;
 377         case FilteringStep:
 378                 emit processStateChanged(m_jobId, QString("%1 (%2%)").arg(tr("Filtering"), QString::number(progress)), ProgressModel::JobRunning);
 379                 break;
 380         case DecodingStep:
 381                 emit processStateChanged(m_jobId, QString("%1 (%2%)").arg(tr("Decoding"), QString::number(progress)), ProgressModel::JobRunning);
 382                 break;
 383         }
 384 }
 385
 386 void ProcessThread::handleMessage(const QString &line)
 387 {
 388         emit processMessageLogged(m_jobId, line);
 389 }
 390
 391 ////////////////////////////////////////////////////////////
 392 // PRIVAE FUNCTIONS
 393 ////////////////////////////////////////////////////////////
 394
 395 int ProcessThread::generateOutFileName(QString &outFileName)
 396 {
 397         outFileName.clear();
 398
 399         //Make sure the source file exists
 400         const QFileInfo sourceFile(m_audioFile.filePath());
 401         if(!(sourceFile.exists() && sourceFile.isFile()))
 402         {
 403                 handleMessage(QString("%1\n%2").arg(tr("The source audio file could not be found:"), sourceFile.absoluteFilePath()));
 404                 return 0;
 405         }
 406
 407         //Make sure the source file readable
 408         QFile readTest(sourceFile.canonicalFilePath());
 409         if(!readTest.open(QIODevice::ReadOnly))
 410         {
 411                 handleMessage(QString("%1\n%2").arg(tr("The source audio file could not be opened for reading:"), QDir::toNativeSeparators(readTest.fileName())));
 412                 return 0;
 413         }
 414         else
 415         {
 416                 readTest.close();
 417         }
 418
 419         const QString baseName = sourceFile.completeBaseName();
 420         QDir targetDir(m_outputDirectory.isEmpty() ? sourceFile.canonicalPath() : m_outputDirectory);
 421
 422         //Prepend relative source file path?
 423         if(m_prependRelativeSourcePath && !m_outputDirectory.isEmpty())
 424         {
 425                 QDir rootDir = sourceFile.dir();
 426                 while(!rootDir.isRoot())
 427                 {
 428                         if(!rootDir.cdUp()) break;
 429                 }
 430                 targetDir.setPath(QString("%1/%2").arg(targetDir.absolutePath(), QFileInfo(rootDir.relativeFilePath(sourceFile.canonicalFilePath())).path()));
 431         }
 432
 433         //Make sure output directory does exist
 434         if(!targetDir.exists())
 435         {
 436                 targetDir.mkpath(".");
 437                 if(!targetDir.exists())
 438                 {
 439                         handleMessage(QString("%1\n%2").arg(tr("The target output directory doesn't exist and could NOT be created:"), QDir::toNativeSeparators(targetDir.absolutePath())));
 440                         return 0;
 441                 }
 442         }
 443
 444         //Make sure that the output dir is writable
 445         QFile writeTest(QString("%1/.%2").arg(targetDir.canonicalPath(), MUtils::rand_str()));
 446         if(!writeTest.open(QIODevice::ReadWrite))
 447         {
 448                 handleMessage(QString("%1\n%2").arg(tr("The target output directory is NOT writable:"), QDir::toNativeSeparators(targetDir.absolutePath())));
 449                 return 0;
 450         }
 451         else
 452         {
 453                 writeTest.remove();
 454         }
 455
 456         //Apply rename pattern
 457         const QString fileName = applyRegularExpression(applyRenamePattern(baseName, m_audioFile.metaInfo()));
 458
 459         //Generate full output path
 460         const QString fileExt = m_renameFileExt.isEmpty() ?  QString::fromUtf8(m_encoder->toEncoderInfo()->extension()) : m_renameFileExt;
 461         outFileName = MUtils::clean_file_path(QString("%1/%2.%3").arg(targetDir.canonicalPath(), fileName, fileExt));
 462
 463         //Skip file, if target file exists (optional!)
 464         if((m_overwriteMode == OverwriteMode_SkipExisting) && QFileInfo(outFileName).exists())
 465         {
 466                 handleMessage(QString("%1\n%2\n").arg(tr("Target output file already exists, going to skip this file:"), QDir::toNativeSeparators(outFileName)));
 467                 handleMessage(tr("If you don't want existing files to be skipped, please change the overwrite mode!"));
 468                 return -1;
 469         }
 470
 471         //Delete file, if target file exists (optional!)
 472         if((m_overwriteMode == OverwriteMode_Overwrite) && QFileInfo(outFileName).exists() && QFileInfo(outFileName).isFile())
 473         {
 474                 handleMessage(QString("%1\n%2\n").arg(tr("Target output file already exists, going to delete existing file:"), QDir::toNativeSeparators(outFileName)));
 475                 if(sourceFile.canonicalFilePath().compare(QFileInfo(outFileName).absoluteFilePath(), Qt::CaseInsensitive) != 0)
 476                 {
 477                         for(int i = 0; i < 16; i++)
 478                         {
 479                                 if(QFile::remove(outFileName))
 480                                 {
 481                                         break;
 482                                 }
 483                                 MUtils::OS::sleep_ms(1);
 484                         }
 485                 }
 486                 if(QFileInfo(outFileName).exists())
 487                 {
 488                         handleMessage(QString("%1\n").arg(tr("Failed to delete existing target file, will save to another file name!")));
 489                 }
 490         }
 491
 492         int n = 1;
 493
 494         //Generate final name
 495         while(QFileInfo(outFileName).exists() && (n < (INT_MAX/2)))
 496         {
 497                 outFileName = MUtils::clean_file_path(QString("%1/%2 (%3).%4").arg(targetDir.canonicalPath(), fileName, QString::number(++n), fileExt));
 498         }
 499
 500         //Create placeholder
 501         QFile placeholder(outFileName);
 502         if(placeholder.open(QIODevice::WriteOnly))
 503         {
 504                 placeholder.close();
 505         }
 506
 507         return 1;
 508 }
 509
 510 QString ProcessThread::applyRenamePattern(const QString &baseName, const AudioFileModel_MetaInfo &metaInfo)
 511 {
 512         QString fileName = m_renamePattern;
 513
 514         fileName.replace("<BaseName>", STRDEF(baseName, tr("Unknown File Name")),         Qt::CaseInsensitive);
 515         fileName.replace("<TrackNo>",  QString().sprintf("%02d", metaInfo.position()),    Qt::CaseInsensitive);
 516         fileName.replace("<Title>",    STRDEF(metaInfo.title(), tr("Unknown Title")) ,    Qt::CaseInsensitive);
 517         fileName.replace("<Artist>",   STRDEF(metaInfo.artist(), tr("Unknown Artist")),   Qt::CaseInsensitive);
 518         fileName.replace("<Album>",    STRDEF(metaInfo.album(), tr("Unknown Album")),     Qt::CaseInsensitive);
 519         fileName.replace("<Year>",     QString().sprintf("%04d", metaInfo.year()),        Qt::CaseInsensitive);
 520         fileName.replace("<Comment>",  STRDEF(metaInfo.comment(), tr("Unknown Comment")), Qt::CaseInsensitive);
 521
 522         return fileName;
 523 }
 524
 525 QString ProcessThread::applyRegularExpression(const QString &fileName)
 526 {
 527         if(m_renameRegExp_Search.isEmpty() || m_renameRegExp_Replace.isEmpty())
 528         {
 529                 return fileName;
 530         }
 531
 532         QRegExp regExp(m_renameRegExp_Search);
 533         if(!regExp.isValid())
 534         {
 535                 qWarning("Invalid regular expression detected -> cannot rename!");
 536                 return fileName;
 537         }
 538
 539         return (QString(fileName).replace(regExp, m_renameRegExp_Replace));
 540 }
 541
 542 QString ProcessThread::generateTempFileName(void)
 543 {
 544         const QString tempFileName = MUtils::make_temp_file(m_tempDirectory, "wav", true);
 545         if(tempFileName.isEmpty())
 546         {
 547                 return QString("%1/~whoops%2.wav").arg(m_tempDirectory, QString::number(MUtils::next_rand32()));
 548         }
 549
 550         m_tempFiles << tempFileName;
 551         return tempFileName;
 552 }
 553
 554 bool ProcessThread::insertDownsampleFilter(const unsigned int *const supportedSamplerates, const unsigned int *const supportedBitdepths)
 555 {
 556         int targetSampleRate = 0, targetBitDepth = 0;
 557
 558         /* Adjust sample rate */
 559         if(supportedSamplerates && m_audioFile.techInfo().audioSamplerate())
 560         {
 561                 const unsigned int inputRate = m_audioFile.techInfo().audioSamplerate();
 562                 unsigned int currentDiff = UINT_MAX, minimumDiff = UINT_MAX, bestRate = UINT_MAX;
 563
 564                 //Find the most suitable supported sampling rate
 565                 for(int i = 0; supportedSamplerates[i]; i++)
 566                 {
 567                         currentDiff = DIFF(inputRate, supportedSamplerates[i]);
 568                         if((currentDiff < minimumDiff) || ((currentDiff == minimumDiff) && (bestRate < supportedSamplerates[i])))
 569                         {
 570                                 bestRate = supportedSamplerates[i];
 571                                 minimumDiff = currentDiff;
 572                                 if(!(minimumDiff > 0)) break;
 573                         }
 574                 }
 575
 576                 if(bestRate != inputRate)
 577                 {
 578                         targetSampleRate = (bestRate != UINT_MAX) ? bestRate : supportedSamplerates[0];
 579                 }
 580         }
 581
 582         /* Adjust bit depth (word size) */
 583         if(supportedBitdepths && m_audioFile.techInfo().audioBitdepth())
 584         {
 585                 const unsigned int inputBPS = m_audioFile.techInfo().audioBitdepth();
 586                 bool bAdjustBitdepth = true;
 587
 588                 //Is the input bit depth supported exactly? (including IEEE Float)
 589                 for(int i = 0; supportedBitdepths[i]; i++)
 590                 {
 591                         if(supportedBitdepths[i] == inputBPS) bAdjustBitdepth = false;
 592                 }
 593
 594                 if(bAdjustBitdepth)
 595                 {
 596                         unsigned int currentDiff = UINT_MAX, minimumDiff = UINT_MAX, bestBPS = UINT_MAX;
 597                         const unsigned int originalBPS = (inputBPS == AudioFileModel::BITDEPTH_IEEE_FLOAT32) ? 32 : inputBPS;
 598
 599                         //Find the most suitable supported bit depth
 600                         for(int i = 0; supportedBitdepths[i]; i++)
 601                         {
 602                                 if(supportedBitdepths[i] == AudioFileModel::BITDEPTH_IEEE_FLOAT32) continue;
 603
 604                                 currentDiff = DIFF(originalBPS, supportedBitdepths[i]);
 605                                 if((currentDiff < minimumDiff) || ((currentDiff == minimumDiff) && (bestBPS < supportedBitdepths[i])))
 606                                 {
 607                                         bestBPS = supportedBitdepths[i];
 608                                         minimumDiff = currentDiff;
 609                                         if(!(minimumDiff > 0)) break;
 610                                 }
 611                         }
 612
 613                         if(bestBPS != originalBPS)
 614                         {
 615                                 targetBitDepth = (bestBPS != UINT_MAX) ? bestBPS : supportedBitdepths[0];
 616                         }
 617                 }
 618         }
 619
 620         //Check if downsampling filter is already in the chain
 621         if (targetSampleRate || targetBitDepth)
 622         {
 623                 for (int i = 0; i < m_filters.count(); i++)
 624                 {
 625                         if (dynamic_cast<ResampleFilter*>(m_filters.at(i)))
 626                         {
 627                                 qWarning("Encoder requires downsampling, but user has already set resamling filter!");
 628                                 handleMessage("WARNING: Encoder may need resampling, but already using resample filter. Encoding *may* fail!\n");
 629                                 targetSampleRate = targetBitDepth = 0;
 630                         }
 631                 }
 632         }
 633
 634         /* Insert the filter */
 635         if(targetSampleRate || targetBitDepth)
 636         {
 637                 m_filters.append(new ResampleFilter(targetSampleRate, targetBitDepth));
 638                 return true;
 639         }
 640
 641         return false; /*did not insert the resample filter */
 642 }
 643
 644 bool ProcessThread::insertDownmixFilter(const unsigned int *const supportedChannels)
 645 {
 646         //Determine number of channels in source
 647         const unsigned int channels = m_audioFile.techInfo().audioChannels();
 648         bool requiresDownmix = (channels > 0);
 649
 650         //Check whether encoder requires downmixing
 651         if(requiresDownmix)
 652         {
 653                 for (int i = 0; supportedChannels[i]; i++)
 654                 {
 655                         if (supportedChannels[i] == channels)
 656                         {
 657                                 requiresDownmix = false;
 658                                 break;
 659                         }
 660                 }
 661         }
 662
 663         //Check if downmixing filter is already in the chain
 664         if (requiresDownmix)
 665         {
 666                 for (int i = 0; i < m_filters.count(); i++)
 667                 {
 668                         if (dynamic_cast<DownmixFilter*>(m_filters.at(i)))
 669                         {
 670                                 qWarning("Encoder requires Stereo downmix, but user has already forced downmix!");
 671                                 handleMessage("WARNING: Encoder may need downmixning, but already using downmixning filter. Encoding *may* fail!\n");
 672                                 requiresDownmix = false;
 673                                 break;
 674                         }
 675                 }
 676         }
 677
 678         //Now add the downmixing filter, if needed
 679         if(requiresDownmix)
 680         {
 681                 m_filters.append(new DownmixFilter());
 682                 return true;
 683         }
 684
 685         return false; /*did not insert the downmix filter*/
 686 }
 687
 688 bool ProcessThread::updateFileTime(const QString &originalFile, const QString &modifiedFile)
 689 {
 690         bool success = false;
 691
 692         QFileInfo originalFileInfo(originalFile);
 693         const QDateTime timeCreated = originalFileInfo.created(), timeLastMod = originalFileInfo.lastModified();
 694         if (timeCreated.isValid() && timeLastMod.isValid())
 695         {
 696                 if (!MUtils::OS::set_file_time(modifiedFile, timeCreated, timeLastMod))
 697                 {
 698                         qWarning("Failed to update creation/modified time of output file: \"%s\"", MUTILS_UTF8(modifiedFile));
 699                 }
 700         }
 701         else
 702         {
 703                 qWarning("Failed to read creation/modified time of source file: \"%s\"", MUTILS_UTF8(originalFile));
 704         }
 705
 706         return success;
 707 }
 708
 709 ////////////////////////////////////////////////////////////
 710 // PUBLIC FUNCTIONS
 711 ////////////////////////////////////////////////////////////
 712
 713 void ProcessThread::addFilter(AbstractFilter *filter)
 714 {
 715         m_filters.append(filter);
 716 }
 717
 718 void ProcessThread::setRenamePattern(const QString &pattern)
 719 {
 720         const QString newPattern = pattern.simplified();
 721         if(!newPattern.isEmpty()) m_renamePattern = newPattern;
 722 }
 723
 724 void ProcessThread::setRenameRegExp(const QString &search, const QString &replace)
 725 {
 726         const QString newSearch = search.trimmed(), newReplace = replace.simplified();
 727         if((!newSearch.isEmpty()) && (!newReplace.isEmpty()))
 728         {
 729                 m_renameRegExp_Search  = newSearch;
 730                 m_renameRegExp_Replace = newReplace;
 731         }
 732 }
 733
 734 void ProcessThread::setRenameFileExt(const QString &fileExtension)
 735 {
 736         m_renameFileExt = MUtils::clean_file_name(fileExtension).simplified();
 737         while(m_renameFileExt.startsWith('.'))
 738         {
 739                 m_renameFileExt = m_renameFileExt.mid(1).trimmed();
 740         }
 741 }
 742
 743 void ProcessThread::setOverwriteMode(const bool &bSkipExistingFile, const bool &bReplacesExisting)
 744 {
 745         if(bSkipExistingFile && bReplacesExisting)
 746         {
 747                 qWarning("Inconsistent overwrite flags -> reverting to default!");
 748                 m_overwriteMode = OverwriteMode_KeepBoth;
 749         }
 750         else
 751         {
 752                 m_overwriteMode = OverwriteMode_KeepBoth;
 753                 if(bSkipExistingFile) m_overwriteMode = OverwriteMode_SkipExisting;
 754                 if(bReplacesExisting) m_overwriteMode = OverwriteMode_Overwrite;
 755         }
 756 }
 757
 758 void ProcessThread::setKeepDateTime(const bool &keepDateTime)
 759 {
 760         m_keepDateTime = keepDateTime;
 761 }
 762
 763 ////////////////////////////////////////////////////////////
 764 // EVENTS
 765 ////////////////////////////////////////////////////////////
 766
 767 /*NONE*/