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