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