src/thread_encode.h

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Simple x264 Launcher
   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.
   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 #pragma once
  23
  24 #include "model_status.h"
  25
  26 #include <QThread>
  27 #include <QUuid>
  28 #include <QMutex>
  29 #include <QStringList>
  30 #include <QSemaphore>
  31
  32 class OptionsModel;
  33 class QProcess;
  34 class JobObject;
  35
  36 class EncodeThread : public QThread
  37 {
  38         Q_OBJECT
  39
  40 public:
  41         EncodeThread(const QString &sourceFileName, const QString &outputFileName, const OptionsModel *options, const QString &binDir, const QString &vpsDir, const bool &x264_x64, const bool &x264_10bit, const bool &avs2yuv_x64, const bool &skipVersionTest, const int &processPriroity, const bool &abortOnTimeout);
  42         ~EncodeThread(void);
  43
  44         QUuid getId(void) { return this->m_jobId; };
  45         const QString &sourceFileName(void) { return this->m_sourceFileName; }
  46         const QString &outputFileName(void) { return this->m_outputFileName; }
  47         const OptionsModel *options(void) { return m_options; }
  48
  49         void pauseJob(void)
  50         {
  51                 m_pause = true;
  52         }
  53         void resumeJob(void)
  54         {
  55                 m_pause = false;
  56                 m_semaphorePaused.release();
  57         }
  58         void abortJob(void)
  59         {
  60                 m_abort = true;
  61                 m_pause = false;
  62                 m_semaphorePaused.release();
  63         }
  64
  65 protected:
  66         static QMutex m_mutex_startProcess;
  67         static const unsigned int m_processTimeoutInterval = 2500;
  68         static const unsigned int m_processTimeoutMaxCounter = 120;
  69         static const unsigned int m_processTimeoutWarning = 24;
  70
  71         //Constants
  72         const QUuid m_jobId;
  73         const QString m_sourceFileName;
  74         const QString m_outputFileName;
  75         const OptionsModel *m_options;
  76         const QString m_binDir;
  77         const QString m_vpsDir;
  78         const bool m_x264_x64;
  79         const bool m_x264_10bit;
  80         const bool m_avs2yuv_x64;
  81         const bool m_skipVersionTest;
  82         const int m_processPriority;
  83         const bool m_abortOnTimeout;
  84
  85         //Types
  86         enum inputType_t
  87         {
  88                 INPUT_NATIVE = 0,
  89                 INPUT_AVISYN = 1,
  90                 INPUT_VAPOUR = 2
  91         };
  92
  93         //Flags
  94         volatile bool m_abort;
  95         volatile bool m_pause;
  96
  97         //Synchronization
  98         QSemaphore m_semaphorePaused;
  99
 100         //Job Object
 101         JobObject *m_jobObject;
 102
 103         //Internal status values
 104         JobStatus m_status;
 105         unsigned int m_progress;
 106
 107         //Entry point
 108         virtual void run(void);
 109         virtual void checkedRun(void);
 110
 111         //Encode functions
 112         void encode(void);
 113         bool runEncodingPass(bool x264_x64, bool x264_10bit, bool avs2yuv_x64, int inputType, unsigned int frames, const QString &indexFile, int pass = 0, const QString &passLogFile = QString());
 114         QStringList buildCommandLine(bool usePipe, bool use10Bit, unsigned int frames, const QString &indexFile, int pass = 0, const QString &passLogFile = QString());
 115         unsigned int checkVersionX264(bool use_x64, bool use_10bit, bool &modified);
 116         unsigned int checkVersionAvs2yuv(bool x64);
 117         bool checkVersionVapoursynth(void);
 118         bool checkPropertiesAvisynth(bool x64, unsigned int &frames);
 119         bool checkPropertiesVapoursynth(unsigned int &frames);
 120
 121         //Auxiallary Stuff
 122         void log(const QString &text) { emit messageLogged(m_jobId, text); }
 123         inline void setStatus(JobStatus newStatus);
 124         inline void setProgress(unsigned int newProgress);
 125         inline void setDetails(const QString &text);
 126         bool startProcess(QProcess &process, const QString &program, const QStringList &args, bool mergeChannels = true);
 127         QStringList splitParams(const QString &params);
 128         qint64 estimateSize(int progress);
 129
 130         //Static functions
 131         static QString commandline2string(const QString &program, const QStringList &arguments);
 132         static QString sizeToString(qint64 size);
 133         static int getInputType(const QString &fileExt);
 134         static QString stringToHash(const QString &string);
 135
 136 signals:
 137         void statusChanged(const QUuid &jobId, JobStatus newStatus);
 138         void progressChanged(const QUuid &jobId, unsigned int newProgress);
 139         void messageLogged(const QUuid &jobId, const QString &text);
 140         void detailsChanged(const QUuid &jobId, const QString &details);
 141
 142 public slots:
 143         void start(Priority priority = InheritPriority);
 144 };
 145