src/thread_encode.h

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Simple x264 Launcher
   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 #pragma once
  23
  24 #include <QThread>
  25 #include <QUuid>
  26 #include <QMutex>
  27 #include <QStringList>
  28 #include <QSemaphore>
  29
  30 class OptionsModel;
  31 class QProcess;
  32
  33 class EncodeThread : public QThread
  34 {
  35         Q_OBJECT
  36
  37 public:
  38         enum JobStatus
  39         {
  40                 JobStatus_Enqueued = 0,
  41                 JobStatus_Starting = 1,
  42                 JobStatus_Indexing = 2,
  43                 JobStatus_Running = 3,
  44                 JobStatus_Running_Pass1 = 4,
  45                 JobStatus_Running_Pass2 = 5,
  46                 JobStatus_Completed = 6,
  47                 JobStatus_Failed = 7,
  48                 JobStatus_Pausing = 8,
  49                 JobStatus_Paused = 9,
  50                 JobStatus_Resuming = 10,
  51                 JobStatus_Aborting = 11,
  52                 JobStatus_Aborted = 12,
  53                 JobStatus_Undefined = 666
  54         };
  55
  56         EncodeThread(const QString &sourceFileName, const QString &outputFileName, const OptionsModel *options, const QString &binDir, bool x264_x64, bool x264_10bit, bool avs2yuv_x64);
  57         ~EncodeThread(void);
  58
  59         QUuid getId(void) { return this->m_jobId; };
  60         const QString &sourceFileName(void) { return this->m_sourceFileName; }
  61         const QString &outputFileName(void) { return this->m_outputFileName; }
  62         const OptionsModel *options(void) { return m_options; }
  63
  64         void pauseJob(void)
  65         {
  66                 m_pause = true;
  67         }
  68         void resumeJob(void)
  69         {
  70                 m_pause = false;
  71                 m_semaphorePaused.release();
  72         }
  73         void abortJob(void)
  74         {
  75                 m_abort = true;
  76                 m_pause = false;
  77                 m_semaphorePaused.release();
  78         }
  79
  80 protected:
  81         static QMutex m_mutex_startProcess;
  82         static const unsigned int m_processTimeoutInterval = 2500;
  83         static const unsigned int m_processTimeoutMaxCounter = 120;
  84         static const unsigned int m_processTimeoutWarning = 24;
  85
  86         //Constants
  87         const QUuid m_jobId;
  88         const QString m_sourceFileName;
  89         const QString m_outputFileName;
  90         const OptionsModel *m_options;
  91         const QString m_binDir;
  92         const bool m_x264_x64;
  93         const bool m_x264_10bit;
  94         const bool m_avs2yuv_x64;
  95
  96         //Flags
  97         volatile bool m_abort;
  98         volatile bool m_pause;
  99
 100         //Synchronization
 101         QSemaphore m_semaphorePaused;
 102
 103         //Job handle
 104         void *m_handle_jobObject;
 105
 106         //Internal status values
 107         JobStatus m_status;
 108         unsigned int m_progress;
 109
 110         //Entry point
 111         virtual void run(void);
 112         virtual void checkedRun(void);
 113
 114         //Encode functions
 115         void encode(void);
 116         bool runEncodingPass(bool x264_x64, bool x264_10bit, bool avs2yuv_x64, bool usePipe, unsigned int frames, const QString &indexFile, int pass = 0, const QString &passLogFile = QString());
 117         QStringList buildCommandLine(bool usePipe, bool use10Bit, unsigned int frames, const QString &indexFile, int pass = 0, const QString &passLogFile = QString());
 118         unsigned int checkVersionX264(bool use_x64, bool use_10bit, bool &modified);
 119         unsigned int checkVersionAvs2yuv(bool x64);
 120         bool checkProperties(bool x64, unsigned int &frames);
 121
 122         //Auxiallary Stuff
 123         void log(const QString &text) { emit messageLogged(m_jobId, text); }
 124         inline void setStatus(JobStatus newStatus);
 125         inline void setProgress(unsigned int newProgress);
 126         inline void setDetails(const QString &text);
 127         bool startProcess(QProcess &process, const QString &program, const QStringList &args, bool mergeChannels = true);
 128         QString pathToLocal(const QString &longPath, bool create = false, bool keep = true);
 129         QStringList splitParams(const QString &params);
 130         qint64 estimateSize(int progress);
 131
 132         //Static functions
 133         static QString commandline2string(const QString &program, const QStringList &arguments);
 134         static QString sizeToString(qint64 size);
 135
 136 signals:
 137         void statusChanged(const QUuid &jobId, EncodeThread::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