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 #include "timecode.h"
  25 #include "monitormanager.h"
  26
  27 #include <KStandardDirs>
  28 #include <KLocale>
  29 #include <KFileDialog>
  30 #include <kmimetype.h>
  31 #include <KIO/NetAccess>
  32 #include <KMessageBox>
  33
  34 #include <QFile>
  35 #include <QApplication>
  36 #include <QTimer>
  37 #include <QDomDocument>
  38 #include <QMenu>
  39 #include <QGridLayout>
  40
  41
  42 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent) :
  43         QWizard(parent),
  44         m_dvdauthor(NULL),
  45         m_mkiso(NULL),
  46         m_burnMenu(new QMenu(this))
  47 {
  48     setWindowTitle(i18n("DVD Wizard"));
  49     //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
  50     m_pageVob = new DvdWizardVob(profile, this);
  51     m_pageVob->setTitle(i18n("Select Files For Your DVD"));
  52     addPage(m_pageVob);
  53     if (!url.isEmpty()) m_pageVob->setUrl(url);
  54
  55
  56     m_pageChapters = new DvdWizardChapters(m_pageVob->isPal(), this);
  57     m_pageChapters->setTitle(i18n("DVD Chapters"));
  58     addPage(m_pageChapters);
  59
  60
  61
  62     m_pageMenu = new DvdWizardMenu(profile, this);
  63     m_pageMenu->setTitle(i18n("Create DVD Menu"));
  64     addPage(m_pageMenu);
  65
  66     QWizardPage *page4 = new QWizardPage;
  67     page4->setTitle(i18n("Creating DVD Image"));
  68     m_status.setupUi(page4);
  69     m_status.error_box->setHidden(true);
  70     m_status.error_box->setTabBarHidden(true);
  71     m_status.tmp_folder->setUrl(KUrl(KdenliveSettings::currenttmpfolder()));
  72     m_status.tmp_folder->setMode(KFile::Directory | KFile::ExistingOnly);
  73     m_status.iso_image->setUrl(KUrl(QDir::homePath() + "/untitled.iso"));
  74     m_status.iso_image->setFilter("*.iso");
  75     m_status.iso_image->setMode(KFile::File);
  76     m_status.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
  77
  78 #if KDE_IS_VERSION(4,7,0)
  79     m_isoMessage = new KMessageWidget;
  80     QGridLayout *s =  static_cast <QGridLayout*> (page4->layout());
  81     s->addWidget(m_isoMessage, 5, 0, 1, -1);
  82     m_isoMessage->hide();
  83 #endif
  84
  85     addPage(page4);
  86
  87     connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
  88     connect(m_status.button_start, SIGNAL(clicked()), this, SLOT(slotGenerate()));
  89     connect(m_status.button_abort, SIGNAL(clicked()), this, SLOT(slotAbort()));
  90     connect(m_status.button_preview, SIGNAL(clicked()), this, SLOT(slotPreview()));
  91
  92     QString programName("k3b");
  93     QString exec = KStandardDirs::findExe(programName);
  94     if (!exec.isEmpty()) {
  95         //Add K3b action
  96         QAction *k3b = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
  97         k3b->setData(exec);
  98     }
  99     programName = "brasero";
 100     exec = KStandardDirs::findExe(programName);
 101     if (!exec.isEmpty()) {
 102         //Add Brasero action
 103         QAction *brasero = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
 104         brasero->setData(exec);
 105     }
 106     if (m_burnMenu->isEmpty()) m_burnMenu->addAction(i18n("No burning program found (K3b, Brasero)"));
 107     m_status.button_burn->setMenu(m_burnMenu);
 108     m_status.button_burn->setIcon(KIcon("tools-media-optical-burn"));
 109     m_status.button_burn->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
 110     m_status.button_preview->setIcon(KIcon("media-playback-start"));
 111
 112     setButtonText(QWizard::CustomButton1, i18n("Load"));
 113     setButtonText(QWizard::CustomButton2, i18n("Save"));
 114     button(QWizard::CustomButton1)->setIcon(KIcon("document-open"));
 115     button(QWizard::CustomButton2)->setIcon(KIcon("document-save"));
 116     connect(button(QWizard::CustomButton1), SIGNAL(clicked()), this, SLOT(slotLoad()));
 117     connect(button(QWizard::CustomButton2), SIGNAL(clicked()), this, SLOT(slotSave()));
 118     setOption(QWizard::HaveCustomButton1, true);
 119     setOption(QWizard::HaveCustomButton2, true);
 120     QList<QWizard::WizardButton> layout;
 121     layout << QWizard::CustomButton1 << QWizard::CustomButton2 << QWizard::Stretch << QWizard::BackButton << QWizard::NextButton << QWizard::CancelButton << QWizard::FinishButton;
 122     setButtonLayout(layout);
 123 }
 124
 125 DvdWizard::~DvdWizard()
 126 {
 127     m_authorFile.remove();
 128     m_menuFile.remove();
 129     blockSignals(true);
 130     delete m_burnMenu;
 131     if (m_dvdauthor) {
 132         m_dvdauthor->blockSignals(true);
 133         m_dvdauthor->close();
 134         delete m_dvdauthor;
 135     }
 136     if (m_mkiso) {
 137         m_mkiso->blockSignals(true);
 138         m_mkiso->close();
 139         delete m_mkiso;
 140     }
 141 }
 142
 143
 144 void DvdWizard::slotPageChanged(int page)
 145 {
 146     //kDebug() << "// PAGE CHGD: " << page << ", ID: " << visitedPages();
 147     if (page == 0) {
 148         // Update chapters that were modified in page 1
 149         m_pageChapters->stopMonitor();
 150         m_pageVob->updateChapters(m_pageChapters->chaptersData());
 151     } else if (page == 1) {
 152         m_pageChapters->setVobFiles(m_pageVob->isPal(), m_pageVob->isWide(), m_pageVob->selectedUrls(), m_pageVob->durations(), m_pageVob->chapters());
 153     } else if (page == 2) {
 154         m_pageChapters->stopMonitor();
 155         m_pageVob->updateChapters(m_pageChapters->chaptersData());
 156         m_pageMenu->setTargets(m_pageChapters->selectedTitles(), m_pageChapters->selectedTargets());
 157         m_pageMenu->changeProfile(m_pageVob->isPal());
 158     }
 159 }
 160
 161
 162
 163 void DvdWizard::generateDvd()
 164 {
 165 #if KDE_IS_VERSION(4,7,0)
 166     m_isoMessage->animatedHide();
 167 #endif
 168     m_status.error_box->setHidden(true);
 169     m_status.error_box->setCurrentIndex(0);
 170     m_status.error_box->setTabBarHidden(true);
 171     m_status.menu_file->clear();
 172     m_status.dvd_file->clear();
 173     KTemporaryFile temp1;
 174     temp1.setSuffix(".png");
 175     //temp1.setAutoRemove(false);
 176     temp1.open();
 177
 178     KTemporaryFile temp2;
 179     temp2.setSuffix(".png");
 180     //temp2.setAutoRemove(false);
 181     temp2.open();
 182
 183     KTemporaryFile temp3;
 184     temp3.setSuffix(".png");
 185     //temp3.setAutoRemove(false);
 186     temp3.open();
 187
 188     KTemporaryFile temp4;
 189     temp4.setSuffix(".png");
 190     //temp4.setAutoRemove(false);
 191     temp4.open();
 192
 193     KTemporaryFile temp5;
 194     temp5.setSuffix(".vob");
 195     //temp5.setAutoRemove(false);
 196     temp5.open();
 197
 198     KTemporaryFile temp6;
 199     temp6.setSuffix(".xml");
 200     //temp6.setAutoRemove(false);
 201     temp6.open();
 202
 203     m_menuFile.close();
 204     m_menuFile.setSuffix(".mpg");
 205     m_menuFile.setAutoRemove(false);
 206     m_menuFile.open();
 207
 208     m_authorFile.close();
 209     m_authorFile.setSuffix(".xml");
 210     m_authorFile.setAutoRemove(false);
 211     m_authorFile.open();
 212
 213     QListWidgetItem *images =  m_status.job_progress->item(0);
 214     m_status.job_progress->setCurrentRow(0);
 215     images->setIcon(KIcon("system-run"));
 216     qApp->processEvents();
 217     QMap <QString, QRect> buttons = m_pageMenu->buttonsInfo();
 218     QStringList buttonsTarget;
 219     m_status.error_log->clear();
 220     // initialize html content
 221     m_status.error_log->setText("<html></html>");
 222
 223     if (m_pageMenu->createMenu()) {
 224         m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
 225         m_pageMenu->createBackgroundImage(temp4.fileName());
 226
 227
 228         images->setIcon(KIcon("dialog-ok"));
 229
 230         kDebug() << "/// STARTING MLT VOB CREATION";
 231         if (!m_pageMenu->menuMovie()) {
 232             // create menu vob file
 233             QListWidgetItem *vobitem =  m_status.job_progress->item(1);
 234             m_status.job_progress->setCurrentRow(1);
 235             vobitem->setIcon(KIcon("system-run"));
 236             qApp->processEvents();
 237
 238             QStringList args;
 239             args.append("-profile");
 240             if (m_pageMenu->isPalMenu()) args.append("dv_pal");
 241             else args.append("dv_ntsc");
 242             args.append(temp4.fileName());
 243             args.append("in=0");
 244             args.append("out=100");
 245             args << "-consumer" << "avformat:" + temp5.fileName();
 246             if (m_pageMenu->isPalMenu()) {
 247                 args << "f=dvd" << "acodec=ac3" << "ab=192k" << "ar=48000" << "vcodec=mpeg2video" << "vb=5000k" << "maxrate=8000k" << "minrate=0" << "bufsize=1835008" << "mux_packet_s=2048" << "mux_rate=10080000" << "s=720x576" << "g=15" << "me_range=63" << "trellis=1";
 248             } else {
 249                 args << "f=dvd" << "acodec=ac3" << "ab=192k" << "ar=48000" << "vcodec=mpeg2video" << "vb=6000k" << "maxrate=9000k" << "minrate=0" << "bufsize=1835008" << "mux_packet_s=2048" << "mux_rate=10080000" << "s=720x480" << "g=18" << "me_range=63" << "trellis=1";
 250             }
 251
 252             kDebug() << "MLT ARGS: " << args;
 253             QProcess renderbg;
 254             renderbg.start(KdenliveSettings::rendererpath(), args);
 255             if (renderbg.waitForFinished()) {
 256                 if (renderbg.exitStatus() == QProcess::CrashExit) {
 257                     //TODO: inform user via the m_isoMessage widget after string freeze
 258                     kDebug() << "/// RENDERING MENU vob crashed";
 259                     QByteArray result = renderbg.readAllStandardError();
 260                     vobitem->setIcon(KIcon("dialog-close"));
 261                     m_status.error_log->append(result);
 262                     m_status.error_box->setHidden(false);
 263                     m_status.button_start->setEnabled(true);
 264                     m_status.button_abort->setEnabled(false);
 265                     return;
 266                 }
 267             } else {
 268                 kDebug() << "/// RENDERING MENU vob timed out";
 269 #if KDE_IS_VERSION(4,7,0)
 270                 m_isoMessage->setText(i18n("Rendering job timed out"));
 271                 m_isoMessage->setMessageType(KMessageWidget::Error);
 272                 m_isoMessage->animatedShow();
 273 #endif
 274                 vobitem->setIcon(KIcon("dialog-close"));
 275                 m_status.error_log->append("<a name=\"result\" /><br /><strong>" + i18n("Rendering job timed out"));
 276                 m_status.error_log->scrollToAnchor("result");
 277                 m_status.error_box->setHidden(false);
 278                 m_status.button_start->setEnabled(true);
 279                 m_status.button_abort->setEnabled(false);
 280                 return;
 281             }
 282             vobitem->setIcon(KIcon("dialog-ok"));
 283         }
 284         kDebug() << "/// STARTING SPUMUX";
 285
 286         // create xml spumux file
 287         QListWidgetItem *spuitem =  m_status.job_progress->item(2);
 288         m_status.job_progress->setCurrentRow(2);
 289         spuitem->setIcon(KIcon("system-run"));
 290         qApp->processEvents();
 291         QDomDocument doc;
 292         QDomElement sub = doc.createElement("subpictures");
 293         doc.appendChild(sub);
 294         QDomElement stream = doc.createElement("stream");
 295         sub.appendChild(stream);
 296         QDomElement spu = doc.createElement("spu");
 297         stream.appendChild(spu);
 298         spu.setAttribute("force", "yes");
 299         spu.setAttribute("start", "00:00:00.00");
 300         spu.setAttribute("image", temp1.fileName());
 301         spu.setAttribute("select", temp2.fileName());
 302         spu.setAttribute("highlight", temp3.fileName());
 303         /*spu.setAttribute("autooutline", "infer");
 304         spu.setAttribute("outlinewidth", "12");
 305         spu.setAttribute("autoorder", "rows");*/
 306
 307         int max = buttons.count() - 1;
 308         int i = 0;
 309         QMapIterator<QString, QRect> it(buttons);
 310         while (it.hasNext()) {
 311             it.next();
 312             QDomElement but = doc.createElement("button");
 313             but.setAttribute("name", 'b' + QString::number(i));
 314             if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
 315             else but.setAttribute("down", "b0");
 316             if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
 317             else but.setAttribute("up", 'b' + QString::number(max));
 318             QRect r = it.value();
 319             //int target = it.key();
 320             // TODO: solve play all button
 321             //if (target == 0) target = 1;
 322
 323             // We need to make sure that the y coordinate is a multiple of 2, otherwise button may not be displayed
 324             buttonsTarget.append(it.key());
 325             but.setAttribute("x0", QString::number(r.x()));
 326             but.setAttribute("y0", QString::number((int) 2 * (r.y() / 2)));
 327             but.setAttribute("x1", QString::number(r.right()));
 328             but.setAttribute("y1", QString::number((int) 2 * (r.bottom() / 2)));
 329             spu.appendChild(but);
 330             i++;
 331         }
 332
 333         QFile data(temp6.fileName());
 334         if (data.open(QFile::WriteOnly)) {
 335             data.write(doc.toString().toUtf8());
 336         }
 337         data.close();
 338
 339         kDebug() << " SPUMUX DATA: " << doc.toString();
 340
 341         QStringList args;
 342         args.append(temp6.fileName());
 343         kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
 344
 345         QProcess spumux;
 346
 347 #if QT_VERSION >= 0x040600
 348         QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
 349         env.insert("VIDEO_FORMAT", m_pageVob->isPal() ? "PAL" : "NTSC");
 350         spumux.setProcessEnvironment(env);
 351 #else
 352         QStringList env = QProcess::systemEnvironment();
 353         env << QString("VIDEO_FORMAT=") + QString(m_pageVob->isPal() ? "PAL" : "NTSC");
 354         spumux.setEnvironment(env);
 355 #endif
 356
 357         if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
 358         else spumux.setStandardInputFile(temp5.fileName());
 359         spumux.setStandardOutputFile(m_menuFile.fileName());
 360         spumux.start("spumux", args);
 361         if (spumux.waitForFinished()) {
 362             m_status.error_log->append(spumux.readAllStandardError());
 363             if (spumux.exitStatus() == QProcess::CrashExit) {
 364                 //TODO: inform user via messagewidget after string freeze
 365                 QByteArray result = spumux.readAllStandardError();
 366                 spuitem->setIcon(KIcon("dialog-close"));
 367                 m_status.error_log->append(result);
 368                 m_status.error_box->setHidden(false);
 369                 m_status.error_box->setTabBarHidden(false);
 370                 m_status.menu_file->setPlainText(m_menuFile.readAll());
 371                 m_status.dvd_file->setPlainText(m_authorFile.readAll());
 372                 m_status.button_start->setEnabled(true);
 373                 kDebug() << "/// RENDERING SPUMUX MENU crashed";
 374                 return;
 375             }
 376         } else {
 377             kDebug() << "/// RENDERING SPUMUX MENU timed out";
 378 #if KDE_IS_VERSION(4,7,0)
 379             m_isoMessage->setText(i18n("Rendering job timed out"));
 380             m_isoMessage->setMessageType(KMessageWidget::Error);
 381             m_isoMessage->animatedShow();
 382 #endif
 383             spuitem->setIcon(KIcon("dialog-close"));
 384             m_status.error_log->append("<a name=\"result\" /><br /><strong>" + i18n("Menu job timed out"));
 385             m_status.error_log->scrollToAnchor("result");
 386             m_status.error_box->setHidden(false);
 387             m_status.error_box->setTabBarHidden(false);
 388             m_status.menu_file->setPlainText(m_menuFile.readAll());
 389             m_status.dvd_file->setPlainText(m_authorFile.readAll());
 390             m_status.button_start->setEnabled(true);
 391             return;
 392         }
 393
 394         spuitem->setIcon(KIcon("dialog-ok"));
 395         kDebug() << "/// DONE: " << m_menuFile.fileName();
 396     }
 397
 398     // create dvdauthor xml
 399     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 400     m_status.job_progress->setCurrentRow(3);
 401     authitem->setIcon(KIcon("system-run"));
 402     qApp->processEvents();
 403     KIO::NetAccess::mkdir(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), this);
 404
 405     QDomDocument dvddoc;
 406     QDomElement auth = dvddoc.createElement("dvdauthor");
 407     auth.setAttribute("dest", m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD");
 408     dvddoc.appendChild(auth);
 409     QDomElement vmgm = dvddoc.createElement("vmgm");
 410     auth.appendChild(vmgm);
 411
 412     if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
 413         // intro movie
 414         QDomElement menus = dvddoc.createElement("menus");
 415         vmgm.appendChild(menus);
 416         QDomElement pgc = dvddoc.createElement("pgc");
 417         pgc.setAttribute("entry", "title");
 418         menus.appendChild(pgc);
 419         QDomElement menuvob = dvddoc.createElement("vob");
 420         menuvob.setAttribute("file", m_pageVob->introMovie());
 421         pgc.appendChild(menuvob);
 422         QDomElement post = dvddoc.createElement("post");
 423         QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
 424         post.appendChild(call);
 425         pgc.appendChild(post);
 426     }
 427     QDomElement titleset = dvddoc.createElement("titleset");
 428     auth.appendChild(titleset);
 429
 430     if (m_pageMenu->createMenu()) {
 431
 432         // DVD main menu
 433         QDomElement menus = dvddoc.createElement("menus");
 434         titleset.appendChild(menus);
 435         QDomElement pgc = dvddoc.createElement("pgc");
 436         pgc.setAttribute("entry", "root");
 437         menus.appendChild(pgc);
 438         QDomElement pre = dvddoc.createElement("pre");
 439         pgc.appendChild(pre);
 440         QDomText nametext = dvddoc.createTextNode("{g1 = 0;}");
 441         pre.appendChild(nametext);
 442         for (int i = 0; i < buttons.count(); i++) {
 443             QDomElement button = dvddoc.createElement("button");
 444             button.setAttribute("name", 'b' + QString::number(i));
 445             nametext = dvddoc.createTextNode('{' + buttonsTarget.at(i) + ";}");
 446             button.appendChild(nametext);
 447             pgc.appendChild(button);
 448         }
 449         QDomElement menuvob = dvddoc.createElement("vob");
 450         menuvob.setAttribute("file", m_menuFile.fileName());
 451         pgc.appendChild(menuvob);
 452
 453         if (m_pageMenu->loopMovie()) {
 454             QDomElement menuloop = dvddoc.createElement("post");
 455             nametext = dvddoc.createTextNode("jump titleset 1 menu;");
 456             menuloop.appendChild(nametext);
 457             pgc.appendChild(menuloop);
 458         } else menuvob.setAttribute("pause", "inf");
 459
 460     }
 461
 462     QDomElement titles = dvddoc.createElement("titles");
 463     titleset.appendChild(titles);
 464     QDomElement video = dvddoc.createElement("video");
 465     titles.appendChild(video);
 466     if (m_pageVob->isPal()) video.setAttribute("format", "pal");
 467     else video.setAttribute("format", "ntsc");
 468
 469     if (m_pageVob->isWide()) video.setAttribute("aspect", "16:9");
 470     else video.setAttribute("aspect", "4:3");
 471
 472     QStringList voburls = m_pageVob->selectedUrls();
 473
 474     QDomElement pgc2;
 475
 476
 477     for (int i = 0; i < voburls.count(); i++) {
 478         if (!voburls.at(i).isEmpty()) {
 479             // Add vob entry
 480             pgc2 = dvddoc.createElement("pgc");
 481             pgc2.setAttribute("pause", 0);
 482             titles.appendChild(pgc2);
 483             QDomElement vob = dvddoc.createElement("vob");
 484             vob.setAttribute("file", voburls.at(i));
 485             // Add chapters
 486             QStringList chaptersList = m_pageChapters->chapters(i);
 487             if (!chaptersList.isEmpty()) vob.setAttribute("chapters", chaptersList.join(","));
 488
 489             pgc2.appendChild(vob);
 490             if (m_pageMenu->createMenu()) {
 491                 QDomElement post = dvddoc.createElement("post");
 492                 QDomText call;
 493                 if (i == voburls.count() - 1) call = dvddoc.createTextNode("{g1 = 0; call menu;}");
 494                 else {
 495                     call = dvddoc.createTextNode("{if ( g1 eq 999 ) { call menu; } jump title " + QString::number(i + 2).rightJustified(2, '0') + ";}");
 496                 }
 497                 post.appendChild(call);
 498                 pgc2.appendChild(post);
 499             }
 500         }
 501     }
 502
 503
 504     QFile data2(m_authorFile.fileName());
 505     if (data2.open(QFile::WriteOnly)) {
 506         data2.write(dvddoc.toString().toUtf8());
 507     }
 508     data2.close();
 509     /*kDebug() << "------------------";
 510     kDebug() << dvddoc.toString();
 511     kDebug() << "------------------";*/
 512
 513     QStringList args;
 514     args << "-x" << m_authorFile.fileName();
 515     kDebug() << "// DVDAUTH ARGS: " << args;
 516     if (m_dvdauthor) {
 517         m_dvdauthor->blockSignals(true);
 518         m_dvdauthor->close();
 519         delete m_dvdauthor;
 520         m_dvdauthor = NULL;
 521     }
 522     m_creationLog.clear();
 523     m_dvdauthor = new QProcess(this);
 524     // Set VIDEO_FORMAT variable (required by dvdauthor 0.7)
 525 #if QT_VERSION >= 0x040600
 526     QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
 527     env.insert("VIDEO_FORMAT", m_pageVob->isPal() ? "PAL" : "NTSC");
 528     m_dvdauthor->setProcessEnvironment(env);
 529 #else
 530     QStringList env = QProcess::systemEnvironment();
 531     env << QString("VIDEO_FORMAT=") + QString(m_pageVob->isPal() ? "PAL" : "NTSC");
 532     m_dvdauthor->setEnvironment(env);
 533 #endif
 534     connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
 535     connect(m_dvdauthor, SIGNAL(readyReadStandardOutput()), this, SLOT(slotShowRenderInfo()));
 536     m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
 537     m_dvdauthor->start("dvdauthor", args);
 538     m_status.button_abort->setEnabled(true);
 539     button(QWizard::FinishButton)->setEnabled(false);
 540 }
 541
 542 void DvdWizard::slotShowRenderInfo()
 543 {
 544     QString log = QString(m_dvdauthor->readAll());
 545     m_status.error_log->append(log);
 546     m_status.error_box->setHidden(false);
 547 }
 548
 549 void DvdWizard::slotRenderFinished(int exitCode, QProcess::ExitStatus status)
 550 {
 551     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 552     if (status == QProcess::CrashExit || exitCode != 0) {
 553 #if KDE_IS_VERSION(4,7,0)
 554         //TODO: Clean up message string after string freeze
 555         QString message = i18n("DVDAuthor process crashed.</strong><br />");
 556         message.remove(QRegExp("<[^>]*>"));
 557         m_isoMessage->setText(message);
 558         m_isoMessage->setMessageType(KMessageWidget::Error);
 559         m_isoMessage->animatedShow();
 560 #endif
 561         QString result(m_dvdauthor->readAllStandardError());
 562         result.append("<a name=\"result\" /><br /><strong>");
 563         result.append(i18n("DVDAuthor process crashed.</strong><br />"));
 564         m_status.error_log->append(result);
 565         m_status.error_log->scrollToAnchor("result");
 566         m_status.error_box->setHidden(false);
 567         m_status.error_box->setTabBarHidden(false);
 568         m_status.menu_file->setPlainText(m_menuFile.readAll());
 569         m_status.dvd_file->setPlainText(m_authorFile.readAll());
 570         kDebug() << "DVDAuthor process crashed";
 571         authitem->setIcon(KIcon("dialog-close"));
 572         m_dvdauthor->close();
 573         delete m_dvdauthor;
 574         m_dvdauthor = NULL;
 575         m_status.button_start->setEnabled(true);
 576         m_status.button_abort->setEnabled(false);
 577         cleanup();
 578         button(QWizard::FinishButton)->setEnabled(true);
 579         return;
 580     }
 581     m_creationLog.append(m_dvdauthor->readAllStandardError());
 582     m_dvdauthor->close();
 583     delete m_dvdauthor;
 584     m_dvdauthor = NULL;
 585
 586     // Check if DVD structure has the necessary infos
 587     if (!QFile::exists(m_status.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
 588 #if KDE_IS_VERSION(4,7,0)
 589         m_isoMessage->setText(i18n("DVD structure broken"));
 590         m_isoMessage->setMessageType(KMessageWidget::Error);
 591         m_isoMessage->animatedShow();
 592 #endif
 593         m_status.error_log->append(m_creationLog + "<a name=\"result\" /><br /><strong>" + i18n("DVD structure broken"));
 594         m_status.error_log->scrollToAnchor("result");
 595         m_status.error_box->setHidden(false);
 596         m_status.error_box->setTabBarHidden(false);
 597         m_status.menu_file->setPlainText(m_menuFile.readAll());
 598         m_status.dvd_file->setPlainText(m_authorFile.readAll());
 599         kDebug() << "DVDAuthor process crashed";
 600         authitem->setIcon(KIcon("dialog-close"));
 601         m_status.button_start->setEnabled(true);
 602         m_status.button_abort->setEnabled(false);
 603         cleanup();
 604         button(QWizard::FinishButton)->setEnabled(true);
 605         return;
 606     }
 607     authitem->setIcon(KIcon("dialog-ok"));
 608     qApp->processEvents();
 609     QStringList args;
 610     args << "-dvd-video" << "-v" << "-o" << m_status.iso_image->url().path() << m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD";
 611
 612     if (m_mkiso) {
 613         m_mkiso->blockSignals(true);
 614         m_mkiso->close();
 615         delete m_mkiso;
 616         m_mkiso = NULL;
 617     }
 618     m_mkiso = new QProcess(this);
 619     connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
 620     connect(m_mkiso, SIGNAL(readyReadStandardOutput()), this, SLOT(slotShowIsoInfo()));
 621     m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
 622     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 623     m_status.job_progress->setCurrentRow(4);
 624     isoitem->setIcon(KIcon("system-run"));
 625     if (!KStandardDirs::findExe("genisoimage").isEmpty()) m_mkiso->start("genisoimage", args);
 626     else m_mkiso->start("mkisofs", args);
 627
 628 }
 629
 630 void DvdWizard::slotShowIsoInfo()
 631 {
 632     QString log = QString(m_mkiso->readAll());
 633     m_status.error_log->append(log);
 634     m_status.error_box->setHidden(false);
 635 }
 636
 637 void DvdWizard::slotIsoFinished(int exitCode, QProcess::ExitStatus status)
 638 {
 639     button(QWizard::FinishButton)->setEnabled(true);
 640     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 641     if (status == QProcess::CrashExit || exitCode != 0) {
 642 #if KDE_IS_VERSION(4,7,0)
 643         m_isoMessage->setText(i18n("ISO creation process crashed."));
 644         m_isoMessage->setMessageType(KMessageWidget::Error);
 645         m_isoMessage->animatedShow();
 646 #endif
 647         QString result(m_mkiso->readAllStandardError());
 648         result.append("<a name=\"result\" /><br /><strong>");
 649         result.append(i18n("ISO creation process crashed."));
 650         m_status.error_log->append(result);
 651         m_status.error_log->scrollToAnchor("result");
 652         m_status.error_box->setHidden(false);
 653         m_status.error_box->setTabBarHidden(false);
 654         m_status.menu_file->setPlainText(m_menuFile.readAll());
 655         m_status.dvd_file->setPlainText(m_authorFile.readAll());
 656         m_mkiso->close();
 657         delete m_mkiso;
 658         m_mkiso = NULL;
 659         cleanup();
 660         kDebug() << "Iso process crashed";
 661         isoitem->setIcon(KIcon("dialog-close"));
 662         m_status.button_start->setEnabled(true);
 663         m_status.button_abort->setEnabled(false);
 664         return;
 665     }
 666
 667     m_creationLog.append(m_mkiso->readAllStandardError());
 668     delete m_mkiso;
 669     m_mkiso = NULL;
 670     m_status.button_start->setEnabled(true);
 671     m_status.button_abort->setEnabled(false);
 672
 673     // Check if DVD iso is ok
 674     QFile iso(m_status.iso_image->url().path());
 675     if (!iso.exists() || iso.size() == 0) {
 676         if (iso.exists()) {
 677             KIO::NetAccess::del(m_status.iso_image->url(), this);
 678         }
 679 #if KDE_IS_VERSION(4,7,0)
 680         m_isoMessage->setText(i18n("DVD ISO is broken"));
 681         m_isoMessage->setMessageType(KMessageWidget::Error);
 682         m_isoMessage->animatedShow();
 683 #endif
 684         m_status.error_log->append(m_creationLog + "<br /><a name=\"result\" /><strong>" + i18n("DVD ISO is broken") + "</strong>");
 685         m_status.error_log->scrollToAnchor("result");
 686         m_status.error_box->setHidden(false);
 687         m_status.error_box->setTabBarHidden(false);
 688         m_status.menu_file->setPlainText(m_menuFile.readAll());
 689         m_status.dvd_file->setPlainText(m_authorFile.readAll());
 690         isoitem->setIcon(KIcon("dialog-close"));
 691         cleanup();
 692         return;
 693     }
 694
 695     isoitem->setIcon(KIcon("dialog-ok"));
 696     kDebug() << "ISO IMAGE " << m_status.iso_image->url().path() << " Successfully created";
 697     cleanup();
 698     kDebug() << m_creationLog;
 699 #if KDE_IS_VERSION(4,7,0)
 700         m_isoMessage->setText(i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()));
 701         m_isoMessage->setMessageType(KMessageWidget::Positive);
 702         m_isoMessage->animatedShow();
 703 #endif
 704
 705     m_status.error_log->append("<a name=\"result\" /><strong>" + i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()) + "</strong>");
 706     m_status.error_log->scrollToAnchor("result");
 707     m_status.button_preview->setEnabled(true);
 708     m_status.button_burn->setEnabled(true);
 709     m_status.error_box->setHidden(false);
 710     //KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()));
 711
 712 }
 713
 714
 715 void DvdWizard::cleanup()
 716 {
 717     KIO::NetAccess::del(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), this);
 718 }
 719
 720
 721 void DvdWizard::slotPreview()
 722 {
 723     QString programName("xine");
 724     QString exec = KStandardDirs::findExe(programName);
 725     if (exec.isEmpty()) KMessageBox::sorry(this, i18n("You need program <b>%1</b> to perform this action", programName));
 726     else QProcess::startDetached(exec, QStringList() << "dvd://" + m_status.iso_image->url().path());
 727 }
 728
 729 void DvdWizard::slotBurn()
 730 {
 731     QAction *action = qobject_cast<QAction *>(sender());
 732     QString exec = action->data().toString();
 733     QStringList args;
 734     if (exec.endsWith("k3b")) args << "--image" << m_status.iso_image->url().path();
 735     else args << "--image=" + m_status.iso_image->url().path();
 736     QProcess::startDetached(exec, args);
 737 }
 738
 739
 740 void DvdWizard::slotGenerate()
 741 {
 742     // clear job icons
 743     if ((m_dvdauthor && m_dvdauthor->state() != QProcess::NotRunning) || (m_mkiso && m_mkiso->state() != QProcess::NotRunning)) return;
 744     for (int i = 0; i < m_status.job_progress->count(); i++)
 745         m_status.job_progress->item(i)->setIcon(KIcon());
 746     QString warnMessage;
 747     if (KIO::NetAccess::exists(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), KIO::NetAccess::SourceSide, this))
 748         warnMessage.append(i18n("Folder %1 already exists. Overwrite?\n", m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"));
 749     if (KIO::NetAccess::exists(KUrl(m_status.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
 750         warnMessage.append(i18n("Image file %1 already exists. Overwrite?", m_status.iso_image->url().path()));
 751
 752     if (warnMessage.isEmpty() || KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::Yes) {
 753         KIO::NetAccess::del(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), this);
 754         QTimer::singleShot(300, this, SLOT(generateDvd()));
 755         m_status.button_preview->setEnabled(false);
 756         m_status.button_burn->setEnabled(false);
 757         m_status.job_progress->setEnabled(true);
 758         m_status.button_start->setEnabled(false);
 759     }
 760 }
 761
 762 void DvdWizard::slotAbort()
 763 {
 764     // clear job icons
 765     if (m_dvdauthor && m_dvdauthor->state() != QProcess::NotRunning) m_dvdauthor->terminate();
 766     else if (m_mkiso && m_mkiso->state() != QProcess::NotRunning) m_mkiso->terminate();
 767 }
 768
 769 void DvdWizard::slotSave()
 770 {
 771     KUrl url = KFileDialog::getSaveUrl(KUrl("kfiledialog:///projectfolder"), "*.kdvd", this, i18n("Save DVD Project"));
 772     if (url.isEmpty()) return;
 773
 774     if (currentId() == 0) m_pageChapters->setVobFiles(m_pageVob->isPal(), m_pageVob->isWide(), m_pageVob->selectedUrls(), m_pageVob->durations(), m_pageVob->chapters());
 775
 776     QDomDocument doc;
 777     QDomElement dvdproject = doc.createElement("dvdproject");
 778     QString profile;
 779     if (m_pageVob->isPal()) profile = "dv_pal";
 780     else profile = "dv_ntsc";
 781     if (m_pageVob->isWide()) profile.append("_wide");
 782     dvdproject.setAttribute("profile", profile);
 783
 784     dvdproject.setAttribute("tmp_folder", m_status.tmp_folder->url().path());
 785     dvdproject.setAttribute("iso_image", m_status.iso_image->url().path());
 786
 787     dvdproject.setAttribute("intro_movie", m_pageVob->introMovie());
 788
 789     doc.appendChild(dvdproject);
 790     QDomElement menu = m_pageMenu->toXml();
 791     if (!menu.isNull()) dvdproject.appendChild(doc.importNode(menu, true));
 792     QDomElement chaps = m_pageChapters->toXml();
 793     if (!chaps.isNull()) dvdproject.appendChild(doc.importNode(chaps, true));
 794
 795     QFile file(url.path());
 796     if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
 797         kWarning() << "//////  ERROR writing to file: " << url.path();
 798         KMessageBox::error(this, i18n("Cannot write to file %1", url.path()));
 799         return;
 800     }
 801
 802     file.write(doc.toString().toUtf8());
 803     if (file.error() != QFile::NoError) {
 804         KMessageBox::error(this, i18n("Cannot write to file %1", url.path()));
 805     }
 806     file.close();
 807 }
 808
 809
 810 void DvdWizard::slotLoad()
 811 {
 812     KUrl url = KFileDialog::getOpenUrl(KUrl("kfiledialog:///projectfolder"), "*.kdvd");
 813     if (url.isEmpty()) return;
 814     QDomDocument doc;
 815     QFile file(url.path());
 816     doc.setContent(&file, false);
 817     file.close();
 818     QDomElement dvdproject = doc.documentElement();
 819     if (dvdproject.tagName() != "dvdproject") {
 820         KMessageBox::error(this, i18n("File %1 is not a Kdenlive project file.", url.path()));
 821         return;
 822     }
 823
 824     QString profile = dvdproject.attribute("profile");
 825     m_pageVob->setProfile(profile);
 826
 827     m_status.tmp_folder->setUrl(KUrl(dvdproject.attribute("tmp_folder")));
 828     m_status.iso_image->setUrl(KUrl(dvdproject.attribute("iso_image")));
 829     m_pageVob->setIntroMovie(dvdproject.attribute("intro_movie"));
 830
 831     QDomNodeList vobs = doc.elementsByTagName("vob");
 832     m_pageVob->clear();
 833     for (int i = 0; i < vobs.count(); i++) {
 834         QDomElement e = vobs.at(i).toElement();
 835         m_pageVob->slotAddVobFile(KUrl(e.attribute("file")), e.attribute("chapters"));
 836     }
 837     m_pageMenu->loadXml(dvdproject.firstChildElement("menu"));
 838 }