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
  21 #include "renderjob.h"
  22
  23 #include <QtDBus>
  24 #include <QFile>
  25 #include <QThread>
  26
  27 // Can't believe I need to do this to sleep.
  28 class SleepThread : QThread
  29 {
  30 public:
  31     virtual void run() {};
  32     static void msleep(unsigned long msecs) {
  33         QThread::msleep(msecs);
  34     }
  35 };
  36
  37 static QDBusConnection connection(QLatin1String(""));
  38
  39 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)
  40 {
  41     m_scenelist = scenelist;
  42     m_dest = dest;
  43     m_player = player;
  44     m_progress = 0;
  45     m_erase = erase;
  46     m_renderProcess = new QProcess;
  47     m_prog = renderer;
  48     m_args << scenelist;
  49     if (in != -1) m_args << "in=" + QString::number(in);
  50     if (out != -1) m_args << "out=" + QString::number(out);
  51
  52     m_args << preargs;
  53     //qDebug()<<"PRE ARGS: "<<preargs;
  54     if (scenelist.startsWith("consumer:")) {
  55         // Use MLT's producer_consumer, needs a different syntax for profile:
  56         m_args << "profile=" + profile;
  57     } else m_args << "-profile" << profile;
  58     m_args << "-consumer" << rendermodule + ':' + m_dest << "progress=1" << args;
  59
  60     m_dualpass = false;
  61     if (args.contains("pass=1")) m_dualpass = true;
  62
  63     connect(m_renderProcess, SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(slotIsOver(int, QProcess::ExitStatus)));
  64     m_renderProcess->setReadChannel(QProcess::StandardError);
  65
  66     /*
  67     // Create a log of every render process.
  68     m_logfile.setAutoRemove(false);
  69     m_logfile.setFileTemplate(QDir::tempPath() + "/kdenlive_render.log.XXXXXXXX");
  70     if (m_logfile.open()) {
  71         qDebug() << "Writing render log to " << m_logfile.fileName();
  72
  73     } else {
  74         qDebug() << "Unable to log to " << m_logfile.fileName();
  75     }
  76     m_logstream.setDevice(&m_logfile);
  77     m_logstream << "Log starting. Dumping contents of " << scenelist << endl;
  78     QFile file(scenelist);
  79     if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
  80         m_logstream << "Unable to read contents of " << scenelist << endl;
  81     } else {
  82         m_logstream.flush();
  83         QTextStream in(&file);
  84         m_logstream << in.readAll() << endl;
  85     }
  86     */
  87 }
  88
  89
  90 RenderJob::~RenderJob()
  91 {
  92     if (m_renderProcess) delete m_renderProcess;
  93     // m_logfile.close();
  94 }
  95
  96 void RenderJob::slotAbort(const QString& url)
  97 {
  98     if (m_dest == url) slotAbort();
  99 }
 100
 101 void RenderJob::slotAbort()
 102 {
 103     qDebug() << "Kdenlive-render: JOB ABORTED BY USER...";
 104     m_renderProcess->kill();
 105
 106     if (m_kdenliveinterface) {
 107         m_dbusargs[1] = -3;
 108         m_dbusargs.append(QString());
 109         m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingFinished", m_dbusargs);
 110     }
 111     if (m_jobUiserver) m_jobUiserver->call("terminate", QString());
 112     if (m_erase) {
 113         QFile f(m_scenelist);
 114         f.remove();
 115     }
 116     QFile f(m_dest);
 117     f.remove();
 118     /*
 119     m_logstream << "Job aborted by user" << endl;
 120     m_logstream.flush();
 121     m_logfile.close();
 122     */
 123     qApp->quit();
 124 }
 125
 126 void RenderJob::receivedStderr()
 127 {
 128     QString result = QString(m_renderProcess->readAllStandardError()).simplified();
 129     if (!result.startsWith("Current Frame")) m_errorMessage.append(result + "<br>");
 130     else {
 131         // m_logstream << "ReceivedStderr from inigo: " << result << endl;
 132         result = result.section(' ', -1);
 133         int pro = result.toInt();
 134         if (pro < 0 || pro > 100) return;
 135         if (pro > m_progress) {
 136             m_progress = pro;
 137             if (m_kdenliveinterface) {
 138                 if (!m_kdenliveinterface->isValid()) {
 139                     delete m_kdenliveinterface;
 140                     m_kdenliveinterface = NULL;
 141                     // qDebug() << "BROKEN COMMUNICATION WITH KDENLIVE";
 142                 } else {
 143                     m_dbusargs[1] = pro;
 144                     m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingProgress", m_dbusargs);
 145                 }
 146             } else if (pro % 5 == 0) {
 147                 // Try to restart communication with Kdenlive every 5 percents
 148                 // qDebug() << "TRYING TO RESTART COMMUNICATION WITH KDENLIVE";
 149                 initKdenliveDbusInterface();
 150             }
 151
 152             if (m_jobUiserver) {
 153                 m_jobUiserver->call("setPercent", (uint) m_progress);
 154                 /*int seconds = m_startTime.secsTo(QTime::currentTime());
 155                 seconds = seconds * (100 - m_progress) / m_progress;
 156                 m_jobUiserver->call("setDescriptionField", (uint) 1, tr("Remaining time"), QTime().addSecs(seconds).toString("hh:mm:ss"));*/
 157             }
 158         }
 159     }
 160 }
 161
 162 void RenderJob::start()
 163 {
 164     QDBusConnectionInterface* interface = QDBusConnection::sessionBus().interface();
 165     if (interface && m_usekuiserver) {
 166         if (!interface->isServiceRegistered("org.kde.JobViewServer")) {
 167             qDebug() << "No org.kde.JobViewServer registered, trying to start kuiserver";
 168             // m_logstream << "No org.kde.JobViewServer registered, trying to start kuiserver";
 169             if (QProcess::startDetached("kuiserver")) {
 170                 qDebug() << "Started kuiserver";
 171                 // m_logstream << "Started kuiserver";
 172                 // Give it a couple of seconds to start
 173                 QTime t;
 174                 t.start();
 175                 while (!interface->isServiceRegistered("org.kde.JobViewServer") && t.elapsed() < 3000) {
 176                     SleepThread::msleep(100);   //Sleep 100 ms
 177                 }
 178             } else {
 179                 qDebug() << "Failed to start kuiserver";
 180                 // m_logstream << "Failed to start kuiserver";
 181             }
 182         }
 183
 184         if (interface->isServiceRegistered("org.kde.JobViewServer")) {
 185             QDBusInterface kuiserver("org.kde.JobViewServer", "/JobViewServer", "org.kde.JobViewServer");
 186             QDBusReply<QDBusObjectPath> objectPath = kuiserver.call("requestView", "kdenlive", "kdenlive", 0x0001);
 187             QString reply = ((QDBusObjectPath) objectPath).path();
 188             m_jobUiserver = new QDBusInterface("org.kde.JobViewServer", reply, "org.kde.JobView");
 189             if (m_jobUiserver) {
 190                 m_startTime = QTime::currentTime();
 191                 m_jobUiserver->call("setPercent", (uint) 0);
 192                 m_jobUiserver->call("setInfoMessage", tr("Rendering %1").arg(QFileInfo(m_dest).fileName()));
 193                 //m_jobUiserver->call("setDescriptionField", (uint) 0, tr("Rendering to"), m_dest);
 194                 QDBusConnection::sessionBus().connect("org.kde.JobViewServer", reply, "org.kde.JobView", "cancelRequested", this, SLOT(slotAbort()));
 195             }
 196         }
 197     }
 198
 199     initKdenliveDbusInterface();
 200
 201     // Because of the logging, we connect to stderr in all cases.
 202     connect(m_renderProcess, SIGNAL(readyReadStandardError()), this, SLOT(receivedStderr()));
 203     m_renderProcess->start(m_prog, m_args);
 204     // m_logstream << "Started render process: " << m_prog << " " << m_args.join(" ") << endl;
 205     qDebug() << "Started render process: " << m_prog << " " << m_args.join(" ");
 206 }
 207
 208
 209 void RenderJob::initKdenliveDbusInterface()
 210 {
 211     QString kdenliveId;
 212     QDBusConnection connection = QDBusConnection::sessionBus();
 213     QDBusConnectionInterface *ibus = connection.interface();
 214     const QStringList services = ibus->registeredServiceNames();
 215     foreach(const QString &service, services) {
 216         if (!service.startsWith("org.kde.kdenlive"))
 217             continue;
 218         kdenliveId = service;
 219         break;
 220     }
 221     m_dbusargs.clear();
 222     if (kdenliveId.isEmpty()) return;
 223     m_kdenliveinterface = new QDBusInterface(kdenliveId,
 224             "/MainWindow",
 225             "org.kdenlive.MainWindow",
 226             connection,
 227             this);
 228
 229     if (m_kdenliveinterface) {
 230         m_dbusargs.append(m_dest);
 231         m_dbusargs.append((int) 0);
 232         m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingProgress", m_dbusargs);
 233         connect(m_kdenliveinterface, SIGNAL(abortRenderJob(const QString &)),
 234                 this, SLOT(slotAbort(const QString&)));
 235     }
 236 }
 237
 238
 239 void RenderJob::slotIsOver(int /*exitcode*/, QProcess::ExitStatus status)
 240 {
 241     if (m_jobUiserver) m_jobUiserver->call("terminate", QString());
 242     if (m_erase) {
 243         QFile f(m_scenelist);
 244         f.remove();
 245     }
 246     if (status == QProcess::CrashExit) {
 247         // rendering crashed
 248         if (m_kdenliveinterface) {
 249             m_dbusargs[1] = (int) - 2;
 250             m_dbusargs.append(m_errorMessage);
 251             m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingFinished", m_dbusargs);
 252         }
 253         QStringList args;
 254         args << "--error" << tr("Rendering of %1 aborted, resulting video will probably be corrupted.").arg(m_dest);
 255         // m_logstream << "Rendering of " << m_dest << " aborted, resulting video will probably be corrupted." << endl;
 256         qDebug() << "Rendering of " << m_dest << " aborted, resulting video will probably be corrupted.";
 257         QProcess::startDetached("kdialog", args);
 258         qApp->quit();
 259     } else {
 260         if (!m_dualpass && m_kdenliveinterface) {
 261             m_dbusargs[1] = (int) - 1;
 262             m_dbusargs.append(QString());
 263             m_kdenliveinterface->callWithArgumentList(QDBus::NoBlock, "setRenderingFinished", m_dbusargs);
 264         }
 265         QDBusConnectionInterface* interface = QDBusConnection::sessionBus().interface();
 266         if (!m_dualpass && interface && interface->isServiceRegistered("org.kde.knotify")) {
 267             QDBusMessage m = QDBusMessage::createMethodCall("org.kde.knotify",
 268                              "/Notify",
 269                              "org.kde.KNotify",
 270                              "event");
 271             int seconds = m_startTime.secsTo(QTime::currentTime());
 272             QList<QVariant> args;
 273             args.append(QString("RenderFinished"));   // action name
 274             args.append(QString("kdenlive"));   // app name
 275             args.append(QVariantList());   // contexts
 276             args.append(tr("Rendering of %1 finished in %2").arg(m_dest, QTime(0, 0, seconds).toString("hh:mm:ss")));   // body
 277             args.append(QByteArray());   // app icon
 278             QStringList actionList;
 279             args.append(actionList);   // actions
 280             qlonglong wid = 0;
 281             args.append(wid);   // win id
 282
 283             m.setArguments(args);
 284             QDBusMessage replyMsg = QDBusConnection::sessionBus().call(m);
 285         }
 286         // m_logstream << "Rendering of " << m_dest << " finished" << endl;
 287         qDebug() << "Rendering of " << m_dest << " finished";
 288         if (!m_dualpass && m_player != "-") {
 289             // m_logstream << "Starting player" << endl;
 290             QStringList args;
 291             args << m_dest;
 292             QProcess::startDetached(m_player, args);
 293         }
 294         if (m_dualpass) {
 295             emit renderingFinished();
 296             deleteLater();
 297         } else qApp->quit();
 298     }
 299 }
 300
 301 #include "renderjob.moc"