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 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 int m_processTimeoutCounter = 24;
  83
  84         //Constants
  85         const QUuid m_jobId;
  86         const QString m_sourceFileName;
  87         const QString m_outputFileName;
  88         const OptionsModel *m_options;
  89         const QString m_binDir;
  90         const bool m_x264_x64;
  91         const bool m_avs2yuv_x64;
  92
  93         //Flags
  94         volatile bool m_abort;
  95         volatile bool m_pause;
  96
  97         //Synchronization
  98         QSemaphore m_semaphorePaused;
  99
 100         //Job handle
 101         void *m_handle_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 avs2yuv_x64, bool usePipe, unsigned int frames, const QString &indexFile, int pass = 0, const QString &passLogFile = QString());
 114         QStringList buildCommandLine(bool usePipe, unsigned int frames, const QString &indexFile, int pass = 0, const QString &passLogFile = QString());
 115         unsigned int checkVersionX264(bool x64, bool &modified);
 116         unsigned int checkVersionAvs2yuv(bool x64);
 117         bool checkProperties(bool x64, unsigned int &frames);
 118
 119         //Auxiallary Stuff
 120         void log(const QString &text) { emit messageLogged(m_jobId, text); }
 121         inline void setStatus(JobStatus newStatus);
 122         inline void setProgress(unsigned int newProgress);
 123         inline void setDetails(const QString &text);
 124         bool startProcess(QProcess &process, const QString &program, const QStringList &args, bool mergeChannels = true);
 125         QString pathToLocal(const QString &longPath, bool create = false, bool keep = true);
 126
 127         //Static functions
 128         static QString commandline2string(const QString &program, const QStringList &arguments);
 129
 130 signals:
 131         void statusChanged(const QUuid &jobId, EncodeThread::JobStatus newStatus);
 132         void progressChanged(const QUuid &jobId, unsigned int newProgress);
 133         void messageLogged(const QUuid &jobId, const QString &text);
 134         void detailsChanged(const QUuid &jobId, const QString &details);
 135
 136 public slots:
 137         void start(Priority priority = InheritPriority);
 138 };
 139