git.sesse.net Git - kdenlive/blob - renderer/renderjob.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Jean-Baptiste Mardelle (jb@kdenlive.org)        *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA          *
  18  ***************************************************************************/
  19
  20 #include <QtDBus>
  21 #include <QFile>
  22 #include <QThread>
  23 #include "renderjob.h"
  24
  25 // Can't believe I need to do this to sleep.
  26 class SleepThread : QThread {
  27 public:
  28     virtual void run() {};
  29     static void msleep(unsigned long msecs) {
  30         QThread::msleep(msecs);
  31     }
  32 };
  33
  34 static QDBusConnection connection(QLatin1String(""));
  35
  36 RenderJob::RenderJob(bool erase, bool usekuiserver, const QString &renderer, const QString &profile, const QString &rendermodule, const QString &player, const QString &scenelist, const QString &dest, const QStringList &preargs, const QStringList &args, int in, int out) : QObject(), m_usekuiserver(usekuiserver), m_jobUiserver(NULL), m_kdenliveinterface(NULL) {
  37     m_scenelist = scenelist;
  38     m_dest = dest;
  39     m_player = player;
  40     m_progress = 0;
  41     m_erase = erase;
  42     m_renderProcess = new QProcess;
  43     m_prog = renderer;
  44     m_args << scenelist;
  45     if (in != -1) m_args << "in=" + QString::number(in);
  46     if (out != -1) m_args << "out=" + QString::number(out);
  47     m_args << preargs;
  48     //qDebug()<<"PRE ARGS: "<<preargs;
  49     if (scenelist.startsWith("consumer:")) {
  50         // Use MLT's producer_consumer, needs a different syntax for profile:
  51         m_args << "profile=" + profile;
  52     } else m_args << "-profile" << profile;
  53     m_args << "-consumer" << rendermodule + ":" + m_dest << "progress=1" << args;
  54     connect(m_renderProcess, SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(slotIsOver(int, QProcess::ExitStatus)));
  55     m_renderProcess->setReadChannel(QProcess::StandardError);
  56     // Create a log of every render process.
  57     m_logfile.setAutoRemove(false);
  58     m_logfile.setFileTemplate(QDir::tempPath() + "/kdenlive_render.log.XXXXXXXX");
  59     if (m_logfile.open()) {
  60         qDebug() << "Writing render log to " << m_logfile.fileName();
  61
  62     } else {
  63         qDebug() << "Unable to log to " << m_logfile.fileName();
  64     }
  65     m_logstream.setDevice(&m_logfile);
  66     m_logstream << "Log starting. Dumping contents of " << scenelist << endl;
  67     QFile file(scenelist);
  68     if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
  69         m_logstream << "Unable to read contents of " << scenelist << endl;
  70     } else {
  71         m_logstream.flush();
  72         QTextStream in(&file);
  73         m_logstream << in.readAll() << endl;
  74     }
  75 }
  76
  77
  78 RenderJob::~RenderJob() {
  79     if (m_renderProcess) delete m_renderProcess;
  80     m_logfile.close();
  81 }
  82
  83 void RenderJob::slotAbort(const QString& url) {
  84     if (m_dest == url) slotAbort();
  85 }
  86
  87 void RenderJob::slotAbort() {
  88     qDebug() << "Kdenlive-render: JOB ABORTED BY USER...";
  89     m_renderProcess->kill();
  90
  91     if (m_kdenliveinterface) {
  92         m_dbusargs[1] = -3;
  93         m_dbusargs.append(QString());
  94         m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingFinished", m_dbusargs);
  95     }
  96     if (m_jobUiserver) m_jobUiserver->call("terminate", QString());
  97     if (m_erase) {
  98         QFile f(m_scenelist);
  99         f.remove();
 100     }
 101     QFile f(m_dest);
 102     f.remove();
 103     m_logstream << "Job aborted by user" << endl;
 104     m_logstream.flush();
 105     m_logfile.close();
 106     qApp->quit();
 107 }
 108
 109 void RenderJob::receivedStderr() {
 110     QString result = QString(m_renderProcess->readAllStandardError()).simplified();
 111     if (!result.startsWith("Current Frame")) m_errorMessage.append(result + "<br>");
 112     m_logstream << "ReceivedStderr from inigo: " << result << endl;
 113     result = result.section(" ", -1);
 114     int pro = result.toInt();
 115     if (m_kdenliveinterface && pro > m_progress) {
 116         m_dbusargs[1] = pro;
 117         m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingProgress", m_dbusargs);
 118     }
 119     if (m_jobUiserver && pro > m_progress) {
 120         m_progress = pro;
 121         m_jobUiserver->call("setPercent", (uint) m_progress);
 122         int seconds = m_startTime.secsTo(QTime::currentTime());
 123         seconds = seconds * (100 - m_progress) / m_progress;
 124         m_jobUiserver->call("setDescriptionField", (uint) 1, tr("Remaining time"), QTime(0, 0, seconds).toString("hh:mm:ss"));
 125     }
 126
 127 }
 128
 129 void RenderJob::start() {
 130     QDBusConnectionInterface* interface = QDBusConnection::sessionBus().interface();
 131     if (interface && m_usekuiserver) {
 132         if (!interface->isServiceRegistered("org.kde.JobViewServer")) {
 133             qDebug() << "No org.kde.JobViewServer registered, trying to start kuiserver";
 134             m_logstream << "No org.kde.JobViewServer registered, trying to start kuiserver";
 135             if (QProcess::startDetached("kuiserver")) {
 136                 qDebug() << "Started kuiserver";
 137                 m_logstream << "Started kuiserver";
 138                 // Give it a couple of seconds to start
 139                 QTime t;
 140                 t.start();
 141                 while (!interface->isServiceRegistered("org.kde.JobViewServer") && t.elapsed() < 3000) {
 142                     SleepThread::msleep(100);   //Sleep 100 ms
 143                 }
 144             } else {
 145                 qDebug() << "Failed to start kuiserver";
 146                 m_logstream << "Failed to start kuiserver";
 147             }
 148         }
 149
 150         if (interface->isServiceRegistered("org.kde.JobViewServer")) {
 151             QDBusInterface kuiserver("org.kde.JobViewServer", "/JobViewServer", "org.kde.JobViewServer");
 152             QDBusReply<QDBusObjectPath> objectPath = kuiserver.call("requestView", "kdenlive", "kdenlive", 1);
 153             QString reply = ((QDBusObjectPath) objectPath).path();
 154             m_jobUiserver = new QDBusInterface("org.kde.JobViewServer", reply, "org.kde.JobView");
 155             if (m_jobUiserver) {
 156                 m_startTime = QTime::currentTime();
 157                 m_jobUiserver->call("setPercent", (uint) 0);
 158                 QDBusReply<QString> reply = m_jobUiserver->call("setInfoMessage", tr("Rendering %1").arg(QFileInfo(m_dest).fileName()));
 159                 m_jobUiserver->call("setDescriptionField", (uint) 0, tr("Rendering to"), m_dest);
 160                 QDBusConnection::sessionBus().connect("org.kde.JobViewServer", reply, "org.kde.JobView", "cancelRequested", this, SLOT(slotAbort()));
 161             }
 162         }
 163     }
 164
 165     QString kdenliveId;
 166     QDBusConnection connection = QDBusConnection::sessionBus();
 167     QDBusConnectionInterface *ibus = connection.interface();
 168     const QStringList services = ibus->registeredServiceNames();
 169     foreach(const QString &service, services) {
 170         if (!service.startsWith("org.kde.kdenlive"))
 171             continue;
 172         kdenliveId = service;
 173         break;
 174     }
 175
 176     QDBusConnection bus = QDBusConnection::sessionBus();
 177     m_kdenliveinterface = new QDBusInterface(kdenliveId,
 178             "/MainWindow",
 179             "org.kdenlive.MainWindow",
 180             bus,
 181             this);
 182
 183     if (m_kdenliveinterface) {
 184         m_dbusargs.append(m_dest);
 185         m_dbusargs.append((int) 0);
 186         m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingProgress", m_dbusargs);
 187         connect(m_kdenliveinterface, SIGNAL(abortRenderJob(const QString &)),
 188                 this, SLOT(slotAbort(const QString&)));
 189     }
 190
 191     // Because of the logging, we connect to stderr in all cases.
 192     connect(m_renderProcess, SIGNAL(readyReadStandardError()), this, SLOT(receivedStderr()));
 193     m_renderProcess->start(m_prog, m_args);
 194     m_logstream << "Started render process: " << m_prog << " " << m_args.join(" ") << endl;
 195     qDebug() << "Started render process: " << m_prog << " " << m_args.join(" ");
 196 }
 197
 198
 199 void RenderJob::slotIsOver(int exitcode, QProcess::ExitStatus status) {
 200     if (m_jobUiserver) m_jobUiserver->call("terminate", QString());
 201     if (m_erase) {
 202         QFile f(m_scenelist);
 203         f.remove();
 204     }
 205     if (status == QProcess::CrashExit) {
 206         // rendering crashed
 207         if (m_kdenliveinterface) {
 208             m_dbusargs[1] = (int) - 2;
 209             m_dbusargs.append(m_errorMessage);
 210             m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingFinished", m_dbusargs);
 211         }
 212         QStringList args;
 213         args << "--error" << tr("Rendering of %1 aborted, resulting video will probably be corrupted.").arg(m_dest);
 214         m_logstream << "Rendering of " << m_dest << " aborted, resulting video will probably be corrupted." << endl;
 215         qDebug() << "Rendering of " << m_dest << " aborted, resulting video will probably be corrupted.";
 216         QProcess::startDetached("kdialog", args);
 217     } else {
 218         if (m_kdenliveinterface) {
 219             m_dbusargs[1] = (int) - 1;
 220             m_dbusargs.append(QString());
 221             m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingFinished", m_dbusargs);
 222         }
 223         QDBusConnectionInterface* interface = QDBusConnection::sessionBus().interface();
 224         if (interface && interface->isServiceRegistered("org.kde.knotify")) {
 225             QDBusMessage m = QDBusMessage::createMethodCall("org.kde.knotify",
 226                              "/Notify",
 227                              "org.kde.KNotify",
 228                              "event");
 229             int seconds = m_startTime.secsTo(QTime::currentTime());
 230             QList<QVariant> args;
 231             args.append(QString("RenderFinished"));   // action name
 232             args.append(QString("kdenlive"));   // app name
 233             args.append(QVariantList());   // contexts
 234             args.append(tr("Rendering of %1 finished in %2").arg(m_dest, QTime(0, 0, seconds).toString("hh:mm:ss")));   // body
 235             args.append(QByteArray());   // app icon
 236             QStringList actionList;
 237             args.append(actionList);   // actions
 238             qlonglong wid;
 239             args.append(wid);   // win id
 240
 241             m.setArguments(args);
 242             QDBusMessage replyMsg = QDBusConnection::sessionBus().call(m);
 243         }
 244         m_logstream << "Rendering of " << m_dest << " finished" << endl;
 245         qDebug() << "Rendering of " << m_dest << " finished";
 246         if (m_player != "-") {
 247             m_logstream << "Starting player" << endl;
 248             QStringList args;
 249             args << m_dest;
 250             QProcess::startDetached(m_player, args);
 251         }
 252     }
 253     qApp->quit();
 254 }
 255
 256 #include "renderjob.moc"