src/source_vapoursynth.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Simple x264 Launcher
   3 // Copyright (C) 2004-2018 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 "source_vapoursynth.h"
  25
  26 #include "global.h"
  27
  28 #include <QDir>
  29 #include <QProcess>
  30 #include <QPair>
  31
  32 static const unsigned int VER_X264_VSPIPE_API =  3;
  33 static const unsigned int VER_X264_VSPIPE_VER = 24;
  34
  35 // ------------------------------------------------------------
  36 // Encoder Info
  37 // ------------------------------------------------------------
  38
  39 class VapoursyntSourceInfo : public AbstractSourceInfo
  40 {
  41 public:
  42         virtual QString getBinaryPath(const SysinfoModel *const sysinfo, const bool& x64) const
  43         {
  44                 return QString("%1/core%2/vspipe.exe").arg(sysinfo->getVPSPath(), (x64 ? "64" : "32"));
  45         }
  46 };
  47
  48 static const VapoursyntSourceInfo s_vapoursynthEncoderInfo;
  49
  50 const AbstractSourceInfo &VapoursynthSource::getSourceInfo(void)
  51 {
  52         return s_vapoursynthEncoderInfo;
  53 }
  54
  55 // ------------------------------------------------------------
  56 // Constructor & Destructor
  57 // ------------------------------------------------------------
  58
  59 VapoursynthSource::VapoursynthSource(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)
  60 :
  61         AbstractSource(jobObject, options, sysinfo, preferences, jobStatus, abort, pause, semaphorePause, sourceFile)
  62 {
  63         /*Nothing to do here*/
  64 }
  65
  66 VapoursynthSource::~VapoursynthSource(void)
  67 {
  68         /*Nothing to do here*/
  69 }
  70
  71 QString VapoursynthSource::getName(void) const
  72 {
  73         return tr("VapourSynth (vpy)");
  74 }
  75
  76 // ------------------------------------------------------------
  77 // Check Version
  78 // ------------------------------------------------------------
  79
  80 bool VapoursynthSource::isSourceAvailable()
  81 {
  82         if(!(m_sysinfo->hasVapourSynth() && (!m_sysinfo->getVPSPath().isEmpty()) && QFileInfo(getBinaryPath()).isFile()))
  83         {
  84                 log(tr("\nVPY INPUT REQUIRES VAPOURSYNTH, BUT IT IS *NOT* AVAILABLE !!!"));
  85                 return false;
  86         }
  87         return true;
  88 }
  89
  90 void VapoursynthSource::checkVersion_init(QList<QRegExp*> &patterns, QStringList &cmdLine)
  91 {
  92         cmdLine << "--version";
  93         patterns << new QRegExp("\\bVapourSynth\\b", Qt::CaseInsensitive);
  94         patterns << new QRegExp("\\bCore\\s+r(\\d+)\\b", Qt::CaseInsensitive);
  95         patterns << new QRegExp("\\bAPI\\s+r(\\d+)\\b", Qt::CaseInsensitive);
  96 }
  97
  98 void VapoursynthSource::checkVersion_parseLine(const QString &line, const QList<QRegExp*> &patterns, unsigned int &core, unsigned int &build, bool &modified)
  99 {
 100         int offset = -1;
 101
 102         if((offset = patterns[1]->lastIndexIn(line)) >= 0)
 103         {
 104                 bool ok = false;
 105                 unsigned int temp = patterns[1]->cap(1).toUInt(&ok);
 106                 if(ok) build = temp;
 107         }
 108         else if((offset = patterns[2]->lastIndexIn(line)) >= 0)
 109         {
 110                 bool ok = false;
 111                 unsigned int temp = patterns[2]->cap(1).toUInt(&ok);
 112                 if(ok) core = temp;
 113         }
 114
 115         if(!line.isEmpty())
 116         {
 117                 log(line);
 118         }
 119 }
 120
 121 QString VapoursynthSource::printVersion(const unsigned int &revision, const bool &modified)
 122 {
 123         unsigned int core, build;
 124         splitRevision(revision, core, build);
 125
 126         return tr("\nVapourSynth version: r%1 (API r%2)").arg(QString::number(build), QString::number(core));
 127 }
 128
 129 bool VapoursynthSource::isVersionSupported(const unsigned int &revision, const bool &modified)
 130 {
 131         unsigned int core, build;
 132         splitRevision(revision, core, build);
 133
 134         if((build < VER_X264_VSPIPE_VER) || (core < VER_X264_VSPIPE_API))
 135         {
 136
 137                 if(core < VER_X264_VSPIPE_API)
 138                 {
 139                         log(tr("\nERROR: Your version of VapourSynth is unsupported! (requires API r%1 or newer)").arg(QString::number(VER_X264_VSPIPE_API)));
 140                 }
 141                 if(build < VER_X264_VSPIPE_VER)
 142                 {
 143                         log(tr("\nERROR: Your version of VapourSynth is unsupported! (requires version r%1 or newer)").arg(QString::number(VER_X264_VSPIPE_VER)));
 144                 }
 145                 log(tr("You can find the latest VapourSynth version at: http://www.vapoursynth.com/"));
 146                 return false;
 147         }
 148
 149         if(core != VER_X264_VSPIPE_API)
 150         {
 151                 log(tr("\nWARNING: Running with an unknown VapourSynth API version, problem may appear! (this application works best with API r%1)").arg(QString::number(VER_X264_VSPIPE_API)));
 152         }
 153
 154         return true;
 155 }
 156
 157 // ------------------------------------------------------------
 158 // Check Source Properties
 159 // ------------------------------------------------------------
 160
 161 void VapoursynthSource::checkSourceProperties_init(QList<QRegExp*> &patterns, QStringList &cmdLine)
 162 {
 163         cmdLine << "--info";
 164         cmdLine << QDir::toNativeSeparators(x264_path2ansi(m_sourceFile, true));
 165         cmdLine << "-";
 166
 167         patterns << new QRegExp("\\bFrames:\\s+(\\d+)\\b");
 168         patterns << new QRegExp("\\bWidth:\\s+(\\d+)\\b");
 169         patterns << new QRegExp("\\bHeight:\\s+(\\d+)\\b");
 170         patterns << new QRegExp("\\bFPS:\\s+(\\d+)\\b");
 171         patterns << new QRegExp("\\bFPS:\\s+(\\d+)/(\\d+)\\b");
 172 }
 173
 174 void VapoursynthSource::checkSourceProperties_parseLine(const QString &line, const QList<QRegExp*> &patterns, ClipInfo &clipInfo)
 175 {
 176         int offset = -1;
 177
 178         if((offset = patterns[0]->lastIndexIn(line)) >= 0)
 179         {
 180                 bool ok = false;
 181                 unsigned int temp = patterns[0]->cap(1).toUInt(&ok);
 182                 if(ok) clipInfo.setFrameCount(temp);
 183         }
 184         if((offset = patterns[1]->lastIndexIn(line)) >= 0)
 185         {
 186                 bool ok = false;
 187                 unsigned int temp =patterns[1]->cap(1).toUInt(&ok);
 188                 if(ok) clipInfo.setFrameSize(temp, clipInfo.getFrameSize().second);
 189         }
 190         if((offset = patterns[2]->lastIndexIn(line)) >= 0)
 191         {
 192                 bool ok = false;
 193                 unsigned int temp = patterns[2]->cap(1).toUInt(&ok);
 194                 if(ok) clipInfo.setFrameSize(clipInfo.getFrameSize().first, temp);
 195         }
 196         if((offset = patterns[3]->lastIndexIn(line)) >= 0)
 197         {
 198                 bool ok = false;
 199                 unsigned int temp = patterns[3]->cap(1).toUInt(&ok);
 200                 if(ok) clipInfo.setFrameRate(temp, 0);
 201         }
 202         if((offset = patterns[4]->lastIndexIn(line)) >= 0)
 203         {
 204                 bool ok1 = false, ok2 = false;
 205                 unsigned int temp1 = patterns[4]->cap(1).toUInt(&ok1);
 206                 unsigned int temp2 = patterns[4]->cap(2).toUInt(&ok2);
 207                 if(ok1 && ok2)
 208                 {
 209                         clipInfo.setFrameRate(temp1, temp2);
 210                 }
 211         }
 212
 213         if(!line.isEmpty())
 214         {
 215                 log(line);
 216         }
 217 }
 218
 219 // ------------------------------------------------------------
 220 // Check Source Properties
 221 // ------------------------------------------------------------
 222
 223 void VapoursynthSource::buildCommandLine(QStringList &cmdLine)
 224 {
 225         cmdLine << "--y4m";
 226         cmdLine << QDir::toNativeSeparators(x264_path2ansi(m_sourceFile, true));
 227         cmdLine << "-";
 228 }
 229
 230 void VapoursynthSource::flushProcess(QProcess &processInput)
 231 {
 232         while(processInput.bytesAvailable() > 0)
 233         {
 234                 log(tr("vpyp [info]: %1").arg(QString::fromUtf8(processInput.readLine()).simplified()));
 235         }
 236
 237         if(processInput.exitCode() != EXIT_SUCCESS)
 238         {
 239                 const int exitCode = processInput.exitCode();
 240                 log(tr("\nWARNING: Input process exited with error (code: %1), your encode might be *incomplete* !!!").arg(QString::number(exitCode)));
 241                 if((exitCode < 0) || (exitCode >= 32))
 242                 {
 243                         log(tr("\nIMPORTANT: The Vapoursynth process terminated abnormally. This means Vapoursynth or one of your Vapoursynth-Plugin's just crashed."));
 244                 }
 245         }
 246 }