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