git.sesse.net Git - kdenlive/blob - src/dvdwizard.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 "dvdwizard.h"
  22 #include "kdenlivesettings.h"
  23 #include "profilesdialog.h"
  24
  25 #include <KStandardDirs>
  26 #include <KLocale>
  27 #include <KFileDialog>
  28 #include <kmimetype.h>
  29 #include <KIO/NetAccess>
  30 #include <KMessageBox>
  31
  32 #include <QFile>
  33 #include <QApplication>
  34 #include <QTimer>
  35 #include <QDomDocument>
  36
  37
  38 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent): QWizard(parent), m_profile(profile), m_dvdauthor(NULL), m_mkiso(NULL) {
  39     //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
  40     setAttribute(Qt::WA_DeleteOnClose);
  41     m_pageVob = new DvdWizardVob(this);
  42     m_pageVob->setTitle(i18n("Select Files For Your DVD"));
  43     addPage(m_pageVob);
  44
  45     if (!url.isEmpty()) m_pageVob->setUrl(url);
  46
  47     m_pageMenu = new DvdWizardMenu(m_profile, this);
  48     m_pageMenu->setTitle(i18n("Create DVD Menu"));
  49     addPage(m_pageMenu);
  50
  51
  52     QWizardPage *page3 = new QWizardPage;
  53     page3->setTitle(i18n("DVD Image"));
  54     m_iso.setupUi(page3);
  55     m_iso.tmp_folder->setPath(KdenliveSettings::currenttmpfolder());
  56     m_iso.iso_image->setPath(QDir::homePath() + "/untitled.iso");
  57     m_iso.iso_image->setFilter("*.iso");
  58     m_iso.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
  59     addPage(page3);
  60
  61     QWizardPage *page4 = new QWizardPage;
  62     page4->setTitle(i18n("Creating DVD Image"));
  63     m_status.setupUi(page4);
  64     m_status.error_box->setHidden(true);
  65     addPage(page4);
  66
  67     connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
  68
  69 //    connect(m_standard.button_all, SIGNAL(toggled(bool)), this, SLOT(slotCheckStandard()));
  70 }
  71
  72 DvdWizard::~DvdWizard() {
  73     // m_menuFile.remove();
  74     if (m_dvdauthor) {
  75         m_dvdauthor->close();
  76         delete m_dvdauthor;
  77     }
  78     if (m_mkiso) {
  79         m_mkiso->close();
  80         delete m_mkiso;
  81     }
  82 }
  83
  84
  85 void DvdWizard::slotPageChanged(int page) {
  86     kDebug() << "// PAGE CHGD: " << page;
  87     if (page == 1) {
  88         m_pageMenu->setTargets(m_pageVob->selectedUrls());
  89     } else if (page == 2) {
  90         m_pageMenu->buttonsInfo();
  91     } else if (page == 3) {
  92         // clear job icons
  93         for (int i = 0; i < m_status.job_progress->count(); i++)
  94             m_status.job_progress->item(i)->setIcon(KIcon());
  95         QString warnMessage;
  96         if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
  97             warnMessage.append(i18n("Folder %1 already exists. Overwrite ?" + '\n', m_iso.tmp_folder->url().path() + "/DVD"));
  98         if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
  99             warnMessage.append(i18n("Image file %1 already exists. Overwrite ?", m_iso.iso_image->url().path()));
 100
 101         if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
 102             back();
 103         } else {
 104             KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 105             QTimer::singleShot(300, this, SLOT(generateDvd()));
 106         }
 107     }
 108 }
 109
 110
 111
 112 void DvdWizard::generateDvd() {
 113     KTemporaryFile temp1;
 114     temp1.setSuffix(".png");
 115     //temp1.setAutoRemove(false);
 116     temp1.open();
 117
 118     KTemporaryFile temp2;
 119     temp2.setSuffix(".png");
 120     //temp2.setAutoRemove(false);
 121     temp2.open();
 122
 123     KTemporaryFile temp3;
 124     temp3.setSuffix(".png");
 125     //temp3.setAutoRemove(false);
 126     temp3.open();
 127
 128     KTemporaryFile temp4;
 129     temp4.setSuffix(".png");
 130     //temp4.setAutoRemove(false);
 131     temp4.open();
 132
 133     KTemporaryFile temp5;
 134     temp5.setSuffix(".vob");
 135     //temp5.setAutoRemove(false);
 136     temp5.open();
 137
 138     KTemporaryFile temp6;
 139     temp6.setSuffix(".xml");
 140     //temp6.setAutoRemove(false);
 141     temp6.open();
 142
 143     m_menuFile.setSuffix(".mpg");
 144     m_menuFile.setAutoRemove(false);
 145     m_menuFile.open();
 146
 147     m_authorFile.setSuffix(".xml");
 148     m_authorFile.setAutoRemove(false);
 149     m_authorFile.open();
 150
 151     QListWidgetItem *images =  m_status.job_progress->item(0);
 152     images->setIcon(KIcon("system-run"));
 153     qApp->processEvents();
 154     QMap <int, QRect> buttons = m_pageMenu->buttonsInfo();
 155     QStringList buttonsTarget;
 156
 157     if (m_pageMenu->createMenu()) {
 158         m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
 159         m_pageMenu->createBackgroundImage(temp4.fileName());
 160
 161
 162         images->setIcon(KIcon("dialog-ok"));
 163
 164         kDebug() << "/// STARTING MLT VOB CREATION";
 165         if (!m_pageMenu->menuMovie()) {
 166             // create menu vob file
 167             QListWidgetItem *vobitem =  m_status.job_progress->item(1);
 168             vobitem->setIcon(KIcon("system-run"));
 169             qApp->processEvents();
 170
 171             QStringList args;
 172             args.append("-profile");
 173             if (m_pageMenu->isPalMenu()) args.append("dv_pal");
 174             else  args.append("dv_ntsc");
 175             args.append(temp4.fileName());
 176             args.append("in=0");
 177             args.append("out=100");
 178             args << "-consumer" << "avformat:" + temp5.fileName();
 179             if (m_pageMenu->isPalMenu()) {
 180                 args << "f=dvd" << "vcodec=mpeg2video" << "acodec=ac3" << "b=5000k" << "maxrate=8000k" << "minrate=0" << "bufsize=1835008" << "mux_packet_s=2048" << "mux_rate=10080000" << "ab=192k" << "ar=48000" << "s=720x576" << "g=15" << "me_range=63" << "trellis=1" << "profile=dv_pal";
 181             } else {
 182                 args << "f=dvd" << "vcodec=mpeg2video" << "acodec=ac3" << "b=6000k" << "maxrate=9000k" << "minrate=0" << "bufsize=1835008" << "mux_packet_s=2048" << "mux_rate=10080000" << "ab=192k" << "ar=48000" << "s=720x480" << "g=18" << "me_range=63" << "trellis=1" << "profile=dv_ntsc";
 183             }
 184
 185             kDebug() << "MLT ARGS: " << args;
 186             QProcess renderbg;
 187             renderbg.start("inigo", args);
 188             if (renderbg.waitForFinished()) {
 189                 if (renderbg.exitStatus() == QProcess::CrashExit) {
 190                     kDebug() << "/// RENDERING MENU vob crashed";
 191                     QByteArray result = renderbg.readAllStandardError();
 192                     vobitem->setIcon(KIcon("dialog-close"));
 193                     m_status.error_log->setText(result);
 194                     m_status.error_box->setHidden(false);
 195                     return;
 196                 }
 197             } else {
 198                 kDebug() << "/// RENDERING MENU vob timed out";
 199                 vobitem->setIcon(KIcon("dialog-close"));
 200                 m_status.error_log->setText(i18n("Rendering job timed out"));
 201                 m_status.error_box->setHidden(false);
 202                 return;
 203             }
 204             vobitem->setIcon(KIcon("dialog-ok"));
 205         }
 206         kDebug() << "/// STARTING SPUMUX";
 207
 208         // create xml spumux file
 209         QListWidgetItem *spuitem =  m_status.job_progress->item(2);
 210         spuitem->setIcon(KIcon("system-run"));
 211         qApp->processEvents();
 212         QDomDocument doc;
 213         QDomElement sub = doc.createElement("subpictures");
 214         doc.appendChild(sub);
 215         QDomElement stream = doc.createElement("stream");
 216         sub.appendChild(stream);
 217         QDomElement spu = doc.createElement("spu");
 218         stream.appendChild(spu);
 219         spu.setAttribute("force", "yes");
 220         spu.setAttribute("start", "00:00:00.00");
 221         spu.setAttribute("image", temp1.fileName());
 222         spu.setAttribute("select", temp2.fileName());
 223         spu.setAttribute("highlight", temp3.fileName());
 224         /*spu.setAttribute("autooutline", "infer");
 225         spu.setAttribute("outlinewidth", "12");
 226         spu.setAttribute("autoorder", "rows");*/
 227
 228         int max = buttons.count() - 1;
 229         int i = 0;
 230         QMapIterator<int, QRect> it(buttons);
 231         while (it.hasNext()) {
 232             it.next();
 233             QDomElement but = doc.createElement("button");
 234             but.setAttribute("name", 'b' + QString::number(i));
 235             if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
 236             else but.setAttribute("down", "b0");
 237             if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
 238             else but.setAttribute("up", 'b' + QString::number(max));
 239             QRect r = it.value();
 240             int target = it.key();
 241             // TODO: solve play all button
 242             if (target == 0) target = 1;
 243             buttonsTarget.append(QString::number(target));
 244             but.setAttribute("x0", QString::number(r.x()));
 245             but.setAttribute("y0", QString::number(r.y()));
 246             but.setAttribute("x1", QString::number(r.right()));
 247             but.setAttribute("y1", QString::number(r.bottom()));
 248             spu.appendChild(but);
 249             i++;
 250         }
 251
 252         kDebug() << "///// SPU: ";
 253         kDebug() << doc.toString();
 254
 255         QFile data(temp6.fileName());
 256         if (data.open(QFile::WriteOnly)) {
 257             QTextStream out(&data);
 258             out << doc.toString();
 259         }
 260         data.close();
 261
 262         kDebug() << " SPUMUX DATA: " << doc.toString();
 263
 264         QStringList args;
 265         args.append(temp6.fileName());
 266         kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
 267
 268         QProcess spumux;
 269
 270         if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
 271         else spumux.setStandardInputFile(temp5.fileName());
 272         spumux.setStandardOutputFile(m_menuFile.fileName());
 273         spumux.start("spumux", args);
 274         spumux.setProcessChannelMode(QProcess::MergedChannels);
 275         if (spumux.waitForFinished()) {
 276             kDebug() << QString(spumux.readAll()).simplified();
 277             if (spumux.exitStatus() == QProcess::CrashExit) {
 278                 QByteArray result = spumux.readAllStandardError();
 279                 spuitem->setIcon(KIcon("dialog-close"));
 280                 m_status.error_log->setText(result);
 281                 m_status.error_box->setHidden(false);
 282                 kDebug() << "/// RENDERING SPUMUX MENU crashed";
 283                 return;
 284             }
 285         } else {
 286             kDebug() << "/// RENDERING SPUMUX MENU timed out";
 287             spuitem->setIcon(KIcon("dialog-close"));
 288             m_status.error_log->setText(i18n("Menu job timed out"));
 289             m_status.error_box->setHidden(false);
 290             return;
 291         }
 292
 293         spuitem->setIcon(KIcon("dialog-ok"));
 294         kDebug() << "/// DONE: " << m_menuFile.fileName();
 295     }
 296
 297     // create dvdauthor xml
 298     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 299     authitem->setIcon(KIcon("system-run"));
 300     qApp->processEvents();
 301     KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 302
 303     QDomDocument dvddoc;
 304     QDomElement auth = dvddoc.createElement("dvdauthor");
 305     auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
 306     dvddoc.appendChild(auth);
 307     QDomElement vmgm = dvddoc.createElement("vmgm");
 308     auth.appendChild(vmgm);
 309
 310     if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
 311         // intro movie
 312         QDomElement menus = dvddoc.createElement("menus");
 313         vmgm.appendChild(menus);
 314         QDomElement pgc = dvddoc.createElement("pgc");
 315         pgc.setAttribute("entry", "title");
 316         menus.appendChild(pgc);
 317         QDomElement menuvob = dvddoc.createElement("vob");
 318         menuvob.setAttribute("file", m_pageVob->introMovie());
 319         pgc.appendChild(menuvob);
 320         QDomElement post = dvddoc.createElement("post");
 321         QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
 322         post.appendChild(call);
 323         pgc.appendChild(post);
 324     }
 325     QDomElement titleset = dvddoc.createElement("titleset");
 326     auth.appendChild(titleset);
 327
 328     if (m_pageMenu->createMenu()) {
 329
 330         // DVD main menu
 331         QDomElement menus = dvddoc.createElement("menus");
 332         titleset.appendChild(menus);
 333         QDomElement pgc = dvddoc.createElement("pgc");
 334         pgc.setAttribute("entry", "root");
 335         menus.appendChild(pgc);
 336         for (int i = 0; i < buttons.count(); i++) {
 337             QDomElement button = dvddoc.createElement("button");
 338             button.setAttribute("name", 'b' + QString::number(i));
 339             QDomText nametext = dvddoc.createTextNode("jump title " + buttonsTarget.at(i) + ';');
 340             button.appendChild(nametext);
 341             pgc.appendChild(button);
 342         }
 343         QDomElement menuvob = dvddoc.createElement("vob");
 344         menuvob.setAttribute("file", m_menuFile.fileName());
 345         menuvob.setAttribute("pause", "inf");
 346         pgc.appendChild(menuvob);
 347     }
 348
 349     QDomElement titles = dvddoc.createElement("titles");
 350     titleset.appendChild(titles);
 351
 352     QStringList voburls = m_pageVob->selectedUrls();
 353
 354     QDomElement pgc2 = dvddoc.createElement("pgc");
 355     titles.appendChild(pgc2);
 356     if (m_pageMenu->createMenu()) {
 357         QDomElement post = dvddoc.createElement("post");
 358         QDomText call = dvddoc.createTextNode("call menu;");
 359         post.appendChild(call);
 360         pgc2.appendChild(post);
 361     }
 362
 363
 364     for (int i = 0; i < voburls.count(); i++) {
 365         if (!voburls.at(i).isEmpty()) {
 366             // Add vob entry
 367             QDomElement vob = dvddoc.createElement("vob");
 368             vob.setAttribute("file", voburls.at(i));
 369             pgc2.appendChild(vob);
 370         }
 371     }
 372
 373     /*
 374         // create one pgc for each video
 375         for (int i = 0; i < voburls.count(); i++) {
 376             if (!voburls.at(i).isEmpty()) {
 377                 // Add vob entry
 378
 379          QDomElement pgc2 = dvddoc.createElement("pgc");
 380          titles.appendChild(pgc2);
 381          if (m_pageMenu->createMenu()) {
 382       QDomElement post = dvddoc.createElement("post");
 383       QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
 384       post.appendChild(call);
 385       pgc2.appendChild(post);
 386          }
 387
 388                 QDomElement vob = dvddoc.createElement("vob");
 389                 vob.setAttribute("file", voburls.at(i));
 390                 pgc2.appendChild(vob);
 391             }
 392         }
 393     */
 394     QFile data2(m_authorFile.fileName());
 395     if (data2.open(QFile::WriteOnly)) {
 396         QTextStream out(&data2);
 397         out << dvddoc.toString();
 398     }
 399     data2.close();
 400     /*kDebug() << "------------------";
 401     kDebug() << dvddoc.toString();
 402     kDebug() << "------------------";*/
 403
 404     QStringList args;
 405     args << "-x" << m_authorFile.fileName();
 406     kDebug() << "// DVDAUTH ARGS: " << args;
 407     if (m_dvdauthor) {
 408         m_dvdauthor->close();
 409         delete m_dvdauthor;
 410         m_dvdauthor = NULL;
 411     }
 412     m_creationLog.clear();
 413     m_dvdauthor = new QProcess(this);
 414     connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
 415     m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
 416     m_dvdauthor->start("dvdauthor", args);
 417
 418 }
 419
 420 void DvdWizard::slotRenderFinished(int exitCode, QProcess::ExitStatus status) {
 421     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 422     if (status == QProcess::CrashExit) {
 423         QByteArray result = m_dvdauthor->readAllStandardError();
 424         m_status.error_log->setText(result);
 425         m_status.error_box->setHidden(false);
 426         kDebug() << "DVDAuthor process crashed";
 427         authitem->setIcon(KIcon("dialog-close"));
 428         m_dvdauthor->close();
 429         delete m_dvdauthor;
 430         m_dvdauthor = NULL;
 431         cleanup();
 432         return;
 433     }
 434     m_creationLog.append(m_dvdauthor->readAllStandardError());
 435     m_dvdauthor->close();
 436     delete m_dvdauthor;
 437     m_dvdauthor = NULL;
 438
 439     // Check if DVD structure has the necessary infos
 440     if (!QFile::exists(m_iso.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
 441         m_status.error_log->setText(m_creationLog + '\n' + i18n("Dvd structure broken"));
 442         m_status.error_box->setHidden(false);
 443         kDebug() << "DVDAuthor process crashed";
 444         authitem->setIcon(KIcon("dialog-close"));
 445         cleanup();
 446         return;
 447     }
 448     authitem->setIcon(KIcon("dialog-ok"));
 449     qApp->processEvents();
 450     QStringList args;
 451     args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
 452
 453     if (m_mkiso) {
 454         m_mkiso->close();
 455         delete m_mkiso;
 456         m_mkiso = NULL;
 457     }
 458     m_mkiso = new QProcess(this);
 459     connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
 460     m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
 461     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 462     isoitem->setIcon(KIcon("system-run"));
 463     m_mkiso->start("mkisofs", args);
 464
 465 }
 466
 467 void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status) {
 468     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 469     if (status == QProcess::CrashExit) {
 470         QByteArray result = m_mkiso->readAllStandardError();
 471         m_status.error_log->setText(result);
 472         m_status.error_box->setHidden(false);
 473         m_mkiso->close();
 474         delete m_mkiso;
 475         m_mkiso = NULL;
 476         cleanup();
 477         kDebug() << "Iso process crashed";
 478         isoitem->setIcon(KIcon("dialog-close"));
 479         return;
 480     }
 481
 482     m_creationLog.append(m_mkiso->readAllStandardError());
 483     delete m_mkiso;
 484     m_mkiso = NULL;
 485
 486     // Check if DVD iso is ok
 487     QFile iso(m_iso.iso_image->url().path());
 488     if (!iso.exists() || iso.size() == 0) {
 489         if (iso.exists()) {
 490             KIO::NetAccess::del(m_iso.iso_image->url(), this);
 491         }
 492         m_status.error_log->setText(m_creationLog + '\n' + i18n("Dvd iso is broken"));
 493         m_status.error_box->setHidden(false);
 494         isoitem->setIcon(KIcon("dialog-close"));
 495         cleanup();
 496         return;
 497     }
 498
 499     isoitem->setIcon(KIcon("dialog-ok"));
 500     kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
 501     cleanup();
 502     kDebug() << m_creationLog;
 503     KMessageBox::information(this, i18n("DVD iso image %1 successfully created.", m_iso.iso_image->url().path()));
 504
 505 }
 506
 507
 508 void DvdWizard::cleanup() {
 509     m_authorFile.remove();
 510     m_menuFile.remove();
 511     KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 512 }
 513