src/source_abstract.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Simple x264 Launcher
   3 // Copyright (C) 2004-2021 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 "source_abstract.h"
  23
  24 //Internal
  25 #include "global.h"
  26 #include "model_sysinfo.h"
  27 #include "model_options.h"
  28 #include "model_preferences.h"
  29
  30 //MUtils
  31 #include <MUtils/Global.h>
  32 #include <MUtils/OSSupport.h>
  33 #include <MUtils/Exception.h>
  34
  35 //Qt
  36 #include <QProcess>
  37 #include <QTextCodec>
  38 #include <QDir>
  39 #include <QPair>
  40
  41 // ------------------------------------------------------------
  42 // Constructor & Destructor
  43 // ------------------------------------------------------------
  44
  45 AbstractSource::AbstractSource(JobObject *jobObject, const OptionsModel *options, const SysinfoModel *const sysinfo, const PreferencesModel *const preferences, JobStatus &jobStatus, volatile bool *abort, volatile bool *pause, QSemaphore *semaphorePause, const QString &sourceFile)
  46 :
  47         AbstractTool(jobObject, options, sysinfo, preferences, jobStatus, abort, pause, semaphorePause),
  48         m_sourceFile(sourceFile)
  49 {
  50         /*Nothing to do here*/
  51 }
  52
  53 AbstractSource::~AbstractSource(void)
  54 {
  55         /*Nothing to do here*/
  56 }
  57
  58 // ------------------------------------------------------------
  59 // Check Source Properties
  60 // ------------------------------------------------------------
  61
  62 bool AbstractSource::checkSourceProperties(ClipInfo &clipInfo)
  63 {
  64         QStringList cmdLine;
  65         QList<QRegExp*> patterns;
  66         QProcess process;
  67
  68         checkSourceProperties_init(patterns, cmdLine);
  69
  70         log("Creating process:");
  71         if(!startProcess(process, getBinaryPath(), cmdLine, true, &getExtraPaths(), &getExtraEnv()))
  72         {
  73                 return false;;
  74         }
  75
  76         QTextCodec *localCodec = QTextCodec::codecForName("System");
  77
  78         bool bTimeout = false;
  79         bool bAborted = false;
  80
  81         clipInfo.reset();
  82
  83         unsigned int waitCounter = 0;
  84
  85         while(process.state() != QProcess::NotRunning)
  86         {
  87                 if(*m_abort)
  88                 {
  89                         process.kill();
  90                         bAborted = true;
  91                         break;
  92                 }
  93                 if(!process.waitForReadyRead(m_processTimeoutInterval))
  94                 {
  95                         if(process.state() == QProcess::Running)
  96                         {
  97                                 if(++waitCounter > m_processTimeoutMaxCounter)
  98                                 {
  99                                         if(m_preferences->getAbortOnTimeout())
 100                                         {
 101                                                 process.kill();
 102                                                 qWarning("Source process timed out <-- killing!");
 103                                                 log("\nPROCESS TIMEOUT !!!");
 104                                                 log("\nInput process has encountered a deadlock or your script takes EXTREMELY long to initialize!");
 105                                                 bTimeout = true;
 106                                                 break;
 107                                         }
 108                                 }
 109                                 else if(waitCounter == m_processTimeoutWarning)
 110                                 {
 111                                         unsigned int timeOut = (waitCounter * m_processTimeoutInterval) / 1000U;
 112                                         log(tr("Warning: Input process did not respond for %1 seconds, potential deadlock...").arg(QString::number(timeOut)));
 113                                 }
 114                         }
 115                         continue;
 116                 }
 117
 118                 waitCounter = 0;
 119                 PROCESS_PENDING_LINES(process, checkSourceProperties_parseLine, patterns, clipInfo);
 120         }
 121
 122         if(!(bTimeout || bAborted))
 123         {
 124                 PROCESS_PENDING_LINES(process, checkSourceProperties_parseLine, patterns, clipInfo);
 125         }
 126
 127         process.waitForFinished();
 128         if(process.state() != QProcess::NotRunning)
 129         {
 130                 process.kill();
 131                 process.waitForFinished(-1);
 132         }
 133
 134         while(!patterns.isEmpty())
 135         {
 136                 QRegExp *pattern = patterns.takeFirst();
 137                 MUTILS_DELETE(pattern);
 138         }
 139
 140         if(bTimeout || bAborted || process.exitCode() != EXIT_SUCCESS)
 141         {
 142                 if(!(bTimeout || bAborted))
 143                 {
 144                         const int exitCode = process.exitCode();
 145                         log(tr("\nPROCESS EXITED WITH ERROR CODE: %1").arg(QString::number(exitCode)));
 146                         if((exitCode < 0) || (exitCode >= 32))
 147                         {
 148                                 log(tr("\nIMPORTANT: The input process terminated abnormally. This means Avisynth/VapourSynth or one of its plugins crashed!"));
 149                         }
 150                 }
 151                 return false;
 152         }
 153
 154         if(clipInfo.getFrameCount() < 1)
 155         {
 156                 log(tr("\nFAILED TO DETERMINE CLIP PROPERTIES !!!"));
 157                 return false;
 158         }
 159
 160         log("");
 161
 162         const QPair<quint32, quint32> frameSize = clipInfo.getFrameSize();
 163         if((frameSize.first > 0) && (frameSize.second > 0))
 164         {
 165                 log(tr("Resolution: %1 x %2").arg(QString::number(frameSize.first), QString::number(frameSize.second)));
 166         }
 167
 168         const QPair<quint32, quint32> frameRate = clipInfo.getFrameRate();
 169         if((frameRate.first > 0) && (frameRate.second > 0))
 170         {
 171                 log(tr("Frame Rate: %1/%2").arg(QString::number(frameRate.first), QString::number(frameRate.second)));
 172         }
 173         else if(frameRate.first > 0)
 174         {
 175                 log(tr("Frame Rate: %1").arg(QString::number(frameRate.first)));
 176         }
 177
 178         log(tr("No. Frames: %1").arg(QString::number(clipInfo.getFrameCount())));
 179         return true;
 180 }
 181
 182 // ------------------------------------------------------------
 183 // Source Processing
 184 // ------------------------------------------------------------
 185
 186 bool AbstractSource::createProcess(QProcess &processEncode, QProcess&processInput)
 187 {
 188         processInput.setStandardOutputProcess(&processEncode);
 189
 190         QStringList cmdLine_Input;
 191         buildCommandLine(cmdLine_Input);
 192
 193         log("Creating input process:");
 194         if(!startProcess(processInput, getBinaryPath(), cmdLine_Input, false, &getExtraPaths(), &getExtraEnv()))
 195         {
 196                 return false;
 197         }
 198
 199         return true;
 200 }
 201
 202 // ------------------------------------------------------------
 203 // Source Info
 204 // ------------------------------------------------------------
 205
 206 const AbstractSourceInfo& AbstractSource::getSourceInfo(void)
 207 {
 208         MUTILS_THROW("[getSourceInfo] This function must be overwritten in sub-classes!");
 209 }
 210
 211 // ------------------------------------------------------------
 212 // Auxiliary FUnctions
 213 // ------------------------------------------------------------
 214
 215 QHash<QString, QString> AbstractSource::getExtraEnv(void) const
 216 {
 217         QHash<QString, QString> extraEnv;
 218
 219         const QString profilePath = MUtils::OS::known_folder(MUtils::OS::FOLDER_PROFILE_USER);
 220         if (!profilePath.isEmpty())
 221         {
 222                 extraEnv.insert("USERPROFILE", QDir::toNativeSeparators(profilePath));
 223         }
 224
 225         const QString appDataPath = MUtils::OS::known_folder(MUtils::OS::FOLDER_APPDATA_ROAM);
 226         if (!appDataPath.isEmpty())
 227         {
 228                 extraEnv.insert("APPDATA", QDir::toNativeSeparators(appDataPath));
 229         }
 230
 231         return extraEnv;
 232 }