git.sesse.net Git - kdenlive/blob - src/dvdwizard.cpp

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