src/Tool_Abstract.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2014 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 "Tool_Abstract.h"
  24
  25 //Internal
  26 #include "Global.h"
  27 #include "JobObject.h"
  28
  29 //MUtils
  30 #include <MUtils/Global.h>
  31 #include <MUtils/OSSupport.h>
  32
  33 //Qt
  34 #include <QProcess>
  35 #include <QMutex>
  36 #include <QMutexLocker>
  37 #include <QLibrary>
  38 #include <QProcessEnvironment>
  39 #include <QDir>
  40 #include <QElapsedTimer>
  41
  42 /*
  43  * Static Objects
  44  */
  45 QScopedPointer<JobObject>     AbstractTool::s_jobObjectInstance;
  46 QScopedPointer<QElapsedTimer> AbstractTool::s_startProcessTimer;
  47
  48 /*
  49  * Synchronization
  50  */
  51 QMutex AbstractTool::s_startProcessMutex;
  52 QMutex AbstractTool::s_createObjectMutex;
  53
  54 /*
  55  * Ref Counter
  56  */
  57 quint64 AbstractTool::s_referenceCounter = 0ui64;
  58
  59 /*
  60  * Const
  61  */
  62 static const qint64 START_DELAY = 64i64;        //in milliseconds
  63
  64 /*
  65  * Constructor
  66  */
  67 AbstractTool::AbstractTool(void)
  68 :
  69         m_firstLaunch(true)
  70
  71 {
  72         QMutexLocker lock(&s_createObjectMutex);
  73
  74         if(s_referenceCounter++ == 0)
  75         {
  76                 s_jobObjectInstance.reset(new JobObject());
  77                 s_startProcessTimer.reset(new QElapsedTimer());
  78         }
  79 }
  80
  81 /*
  82  * Destructor
  83  */
  84 AbstractTool::~AbstractTool(void)
  85 {
  86         QMutexLocker lock(&s_createObjectMutex);
  87
  88         if(--s_referenceCounter == 0)
  89         {
  90                 s_jobObjectInstance.reset(NULL);
  91                 s_startProcessTimer.reset(NULL);
  92         }
  93 }
  94
  95 /*
  96  * Initialize and launch process object
  97  */
  98 bool AbstractTool::startProcess(QProcess &process, const QString &program, const QStringList &args)
  99 {
 100         QMutexLocker lock(&s_startProcessMutex);
 101
 102         if((!s_startProcessTimer.isNull()) && s_startProcessTimer->isValid())
 103         {
 104                 qint64 elapsed = s_startProcessTimer->elapsed();
 105                 while(elapsed < START_DELAY)
 106                 {
 107                         lock.unlock();
 108                         MUtils::OS::sleep_ms((size_t)(START_DELAY - elapsed));
 109                         lock.relock();
 110                         elapsed = s_startProcessTimer->elapsed();
 111                 }
 112         }
 113
 114         emit messageLogged(commandline2string(program, args) + "\n");
 115         MUtils::init_process(process, QFileInfo(program).absolutePath());
 116
 117         process.start(program, args);
 118
 119         if(process.waitForStarted())
 120         {
 121                 if(!s_jobObjectInstance.isNull())
 122                 {
 123                         if(!s_jobObjectInstance->addProcessToJob(&process))
 124                         {
 125                                 qWarning("Failed to assign process to job object!");
 126                         }
 127                 }
 128
 129                 MUtils::OS::change_process_priority(&process, -1);
 130
 131                 if(m_firstLaunch)
 132                 {
 133                         emit statusUpdated(0);
 134                         m_firstLaunch = false;
 135                 }
 136
 137                 s_startProcessTimer->start();
 138                 return true;
 139         }
 140
 141         emit messageLogged("Process creation has failed :-(");
 142         QString errorMsg= process.errorString().trimmed();
 143         if(!errorMsg.isEmpty()) emit messageLogged(errorMsg);
 144
 145         process.kill();
 146         process.waitForFinished(-1);
 147
 148         s_startProcessTimer->start();
 149         return false;
 150 }
 151
 152 /*
 153  * Convert program arguments to single string
 154  */
 155 QString AbstractTool::commandline2string(const QString &program, const QStringList &arguments)
 156 {
 157         QString commandline = (program.contains(' ') ? QString("\"%1\"").arg(program) : program);
 158
 159         for(int i = 0; i < arguments.count(); i++)
 160         {
 161                 commandline += (arguments.at(i).contains(' ') ? QString(" \"%1\"").arg(arguments.at(i)) : QString(" %1").arg(arguments.at(i)));
 162         }
 163
 164         return commandline;
 165 }
 166
 167