]> git.sesse.net Git - kdenlive/blob - src/dvdwizard.cpp
DVD wizard: don't lose chapters data when switching to menu page
[kdenlive] / 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_pageVob->updateChapters(m_pageChapters->chaptersData());
148         m_pageMenu->setTargets(m_pageChapters->selectedTitles(), m_pageChapters->selectedTargets());
149         m_pageMenu->changeProfile(m_pageVob->isPal());
150     }
151 }
152
153
154
155 void DvdWizard::generateDvd()
156 {
157     m_status.error_box->setHidden(true);
158     m_status.error_box->setCurrentIndex(0);
159     m_status.error_box->setTabBarHidden(true);
160     m_status.menu_file->clear();
161     m_status.dvd_file->clear();
162     KTemporaryFile temp1;
163     temp1.setSuffix(".png");
164     //temp1.setAutoRemove(false);
165     temp1.open();
166
167     KTemporaryFile temp2;
168     temp2.setSuffix(".png");
169     //temp2.setAutoRemove(false);
170     temp2.open();
171
172     KTemporaryFile temp3;
173     temp3.setSuffix(".png");
174     //temp3.setAutoRemove(false);
175     temp3.open();
176
177     KTemporaryFile temp4;
178     temp4.setSuffix(".png");
179     //temp4.setAutoRemove(false);
180     temp4.open();
181
182     KTemporaryFile temp5;
183     temp5.setSuffix(".vob");
184     //temp5.setAutoRemove(false);
185     temp5.open();
186
187     KTemporaryFile temp6;
188     temp6.setSuffix(".xml");
189     //temp6.setAutoRemove(false);
190     temp6.open();
191
192     m_menuFile.close();
193     m_menuFile.setSuffix(".mpg");
194     m_menuFile.setAutoRemove(false);
195     m_menuFile.open();
196
197     m_authorFile.close();
198     m_authorFile.setSuffix(".xml");
199     m_authorFile.setAutoRemove(false);
200     m_authorFile.open();
201
202     QListWidgetItem *images =  m_status.job_progress->item(0);
203     m_status.job_progress->setCurrentRow(0);
204     images->setIcon(KIcon("system-run"));
205     qApp->processEvents();
206     QMap <QString, QRect> buttons = m_pageMenu->buttonsInfo();
207     QStringList buttonsTarget;
208     m_status.error_log->clear();
209     // initialize html content
210     m_status.error_log->setText("<html></html>");
211
212     if (m_pageMenu->createMenu()) {
213         m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
214         m_pageMenu->createBackgroundImage(temp4.fileName());
215
216
217         images->setIcon(KIcon("dialog-ok"));
218
219         kDebug() << "/// STARTING MLT VOB CREATION";
220         if (!m_pageMenu->menuMovie()) {
221             // create menu vob file
222             QListWidgetItem *vobitem =  m_status.job_progress->item(1);
223             m_status.job_progress->setCurrentRow(1);
224             vobitem->setIcon(KIcon("system-run"));
225             qApp->processEvents();
226
227             QStringList args;
228             args.append("-profile");
229             if (m_pageMenu->isPalMenu()) args.append("dv_pal");
230             else args.append("dv_ntsc");
231             args.append(temp4.fileName());
232             args.append("in=0");
233             args.append("out=100");
234             args << "-consumer" << "avformat:" + temp5.fileName();
235             if (m_pageMenu->isPalMenu()) {
236                 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";
237             } else {
238                 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";
239             }
240
241             kDebug() << "MLT ARGS: " << args;
242             QProcess renderbg;
243             renderbg.start(KdenliveSettings::rendererpath(), args);
244             if (renderbg.waitForFinished()) {
245                 if (renderbg.exitStatus() == QProcess::CrashExit) {
246                     kDebug() << "/// RENDERING MENU vob crashed";
247                     QByteArray result = renderbg.readAllStandardError();
248                     vobitem->setIcon(KIcon("dialog-close"));
249                     m_status.error_log->append(result);
250                     m_status.error_box->setHidden(false);
251                     m_status.button_start->setEnabled(true);
252                     m_status.button_abort->setEnabled(false);
253                     return;
254                 }
255             } else {
256                 kDebug() << "/// RENDERING MENU vob timed out";
257                 vobitem->setIcon(KIcon("dialog-close"));
258                 m_status.error_log->append("<a name=\"result\" /><br /><strong>" + i18n("Rendering job timed out"));
259                 m_status.error_log->scrollToAnchor("result");
260                 m_status.error_box->setHidden(false);
261                 m_status.button_start->setEnabled(true);
262                 m_status.button_abort->setEnabled(false);
263                 return;
264             }
265             vobitem->setIcon(KIcon("dialog-ok"));
266         }
267         kDebug() << "/// STARTING SPUMUX";
268
269         // create xml spumux file
270         QListWidgetItem *spuitem =  m_status.job_progress->item(2);
271         m_status.job_progress->setCurrentRow(2);
272         spuitem->setIcon(KIcon("system-run"));
273         qApp->processEvents();
274         QDomDocument doc;
275         QDomElement sub = doc.createElement("subpictures");
276         doc.appendChild(sub);
277         QDomElement stream = doc.createElement("stream");
278         sub.appendChild(stream);
279         QDomElement spu = doc.createElement("spu");
280         stream.appendChild(spu);
281         spu.setAttribute("force", "yes");
282         spu.setAttribute("start", "00:00:00.00");
283         spu.setAttribute("image", temp1.fileName());
284         spu.setAttribute("select", temp2.fileName());
285         spu.setAttribute("highlight", temp3.fileName());
286         /*spu.setAttribute("autooutline", "infer");
287         spu.setAttribute("outlinewidth", "12");
288         spu.setAttribute("autoorder", "rows");*/
289
290         int max = buttons.count() - 1;
291         int i = 0;
292         QMapIterator<QString, QRect> it(buttons);
293         while (it.hasNext()) {
294             it.next();
295             QDomElement but = doc.createElement("button");
296             but.setAttribute("name", 'b' + QString::number(i));
297             if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
298             else but.setAttribute("down", "b0");
299             if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
300             else but.setAttribute("up", 'b' + QString::number(max));
301             QRect r = it.value();
302             //int target = it.key();
303             // TODO: solve play all button
304             //if (target == 0) target = 1;
305
306             // We need to make sure that the y coordinate is a multiple of 2, otherwise button may not be displayed
307             buttonsTarget.append(it.key());
308             but.setAttribute("x0", QString::number(r.x()));
309             but.setAttribute("y0", QString::number((int) 2 * (r.y() / 2)));
310             but.setAttribute("x1", QString::number(r.right()));
311             but.setAttribute("y1", QString::number((int) 2 * (r.bottom() / 2)));
312             spu.appendChild(but);
313             i++;
314         }
315
316         QFile data(temp6.fileName());
317         if (data.open(QFile::WriteOnly)) {
318             data.write(doc.toString().toUtf8());
319         }
320         data.close();
321
322         kDebug() << " SPUMUX DATA: " << doc.toString();
323
324         QStringList args;
325         args.append(temp6.fileName());
326         kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
327
328         QProcess spumux;
329
330 #if QT_VERSION >= 0x040600
331         QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
332         env.insert("VIDEO_FORMAT", m_pageVob->isPal() ? "PAL" : "NTSC");
333         spumux.setProcessEnvironment(env);
334 #else
335         QStringList env = QProcess::systemEnvironment();
336         env << QString("VIDEO_FORMAT=") + QString(m_pageVob->isPal() ? "PAL" : "NTSC");
337         spumux.setEnvironment(env);
338 #endif
339     
340         if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
341         else spumux.setStandardInputFile(temp5.fileName());
342         spumux.setStandardOutputFile(m_menuFile.fileName());
343         spumux.start("spumux", args);
344         if (spumux.waitForFinished()) {
345             m_status.error_log->append(spumux.readAllStandardError());
346             if (spumux.exitStatus() == QProcess::CrashExit) {
347                 QByteArray result = spumux.readAllStandardError();
348                 spuitem->setIcon(KIcon("dialog-close"));
349                 m_status.error_log->append(result);
350                 m_status.error_box->setHidden(false);
351                 m_status.error_box->setTabBarHidden(false);
352                 m_status.menu_file->setPlainText(m_menuFile.readAll());
353                 m_status.dvd_file->setPlainText(m_authorFile.readAll());
354                 m_status.button_start->setEnabled(true);
355                 kDebug() << "/// RENDERING SPUMUX MENU crashed";
356                 return;
357             }
358         } else {
359             kDebug() << "/// RENDERING SPUMUX MENU timed out";
360             spuitem->setIcon(KIcon("dialog-close"));
361             m_status.error_log->append("<a name=\"result\" /><br /><strong>" + i18n("Menu job timed out"));
362             m_status.error_log->scrollToAnchor("result");
363             m_status.error_box->setHidden(false);
364             m_status.error_box->setTabBarHidden(false);
365             m_status.menu_file->setPlainText(m_menuFile.readAll());
366             m_status.dvd_file->setPlainText(m_authorFile.readAll());
367             m_status.button_start->setEnabled(true);
368             return;
369         }
370
371         spuitem->setIcon(KIcon("dialog-ok"));
372         kDebug() << "/// DONE: " << m_menuFile.fileName();
373     }
374
375     // create dvdauthor xml
376     QListWidgetItem *authitem =  m_status.job_progress->item(3);
377     m_status.job_progress->setCurrentRow(3);
378     authitem->setIcon(KIcon("system-run"));
379     qApp->processEvents();
380     KIO::NetAccess::mkdir(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), this);
381
382     QDomDocument dvddoc;
383     QDomElement auth = dvddoc.createElement("dvdauthor");
384     auth.setAttribute("dest", m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD");
385     dvddoc.appendChild(auth);
386     QDomElement vmgm = dvddoc.createElement("vmgm");
387     auth.appendChild(vmgm);
388
389     if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
390         // intro movie
391         QDomElement menus = dvddoc.createElement("menus");
392         vmgm.appendChild(menus);
393         QDomElement pgc = dvddoc.createElement("pgc");
394         pgc.setAttribute("entry", "title");
395         menus.appendChild(pgc);
396         QDomElement menuvob = dvddoc.createElement("vob");
397         menuvob.setAttribute("file", m_pageVob->introMovie());
398         pgc.appendChild(menuvob);
399         QDomElement post = dvddoc.createElement("post");
400         QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
401         post.appendChild(call);
402         pgc.appendChild(post);
403     }
404     QDomElement titleset = dvddoc.createElement("titleset");
405     auth.appendChild(titleset);
406
407     if (m_pageMenu->createMenu()) {
408
409         // DVD main menu
410         QDomElement menus = dvddoc.createElement("menus");
411         titleset.appendChild(menus);
412         QDomElement pgc = dvddoc.createElement("pgc");
413         pgc.setAttribute("entry", "root");
414         menus.appendChild(pgc);
415         QDomElement pre = dvddoc.createElement("pre");
416         pgc.appendChild(pre);
417         QDomText nametext = dvddoc.createTextNode("{g1 = 0;}");
418         pre.appendChild(nametext);
419         for (int i = 0; i < buttons.count(); i++) {
420             QDomElement button = dvddoc.createElement("button");
421             button.setAttribute("name", 'b' + QString::number(i));
422             nametext = dvddoc.createTextNode('{' + buttonsTarget.at(i) + ";}");
423             button.appendChild(nametext);
424             pgc.appendChild(button);
425         }
426         QDomElement menuvob = dvddoc.createElement("vob");
427         menuvob.setAttribute("file", m_menuFile.fileName());
428         pgc.appendChild(menuvob);
429
430         if (m_pageMenu->loopMovie()) {
431             QDomElement menuloop = dvddoc.createElement("post");
432             nametext = dvddoc.createTextNode("jump titleset 1 menu;");
433             menuloop.appendChild(nametext);
434             pgc.appendChild(menuloop);
435         } else menuvob.setAttribute("pause", "inf");
436
437     }
438
439     QDomElement titles = dvddoc.createElement("titles");
440     titleset.appendChild(titles);
441     QDomElement video = dvddoc.createElement("video");
442     titles.appendChild(video);
443     if (m_pageVob->isPal()) video.setAttribute("format", "pal");
444     else video.setAttribute("format", "ntsc");
445
446     if (m_pageVob->isWide()) video.setAttribute("aspect", "16:9");
447     else video.setAttribute("aspect", "4:3");
448
449     QStringList voburls = m_pageVob->selectedUrls();
450
451     QDomElement pgc2;
452
453
454     for (int i = 0; i < voburls.count(); i++) {
455         if (!voburls.at(i).isEmpty()) {
456             // Add vob entry
457             pgc2 = dvddoc.createElement("pgc");
458             pgc2.setAttribute("pause", 0);
459             titles.appendChild(pgc2);
460             QDomElement vob = dvddoc.createElement("vob");
461             vob.setAttribute("file", voburls.at(i));
462             // Add chapters
463             QStringList chaptersList = m_pageChapters->chapters(i);
464             if (!chaptersList.isEmpty()) vob.setAttribute("chapters", chaptersList.join(","));
465
466             pgc2.appendChild(vob);
467             if (m_pageMenu->createMenu()) {
468                 QDomElement post = dvddoc.createElement("post");
469                 QDomText call;
470                 if (i == voburls.count() - 1) call = dvddoc.createTextNode("{g1 = 0; call menu;}");
471                 else {
472                     call = dvddoc.createTextNode("{if ( g1 eq 999 ) { call menu; } jump title " + QString::number(i + 2).rightJustified(2, '0') + ";}");
473                 }
474                 post.appendChild(call);
475                 pgc2.appendChild(post);
476             }
477         }
478     }
479
480
481     QFile data2(m_authorFile.fileName());
482     if (data2.open(QFile::WriteOnly)) {
483         data2.write(dvddoc.toString().toUtf8());
484     }
485     data2.close();
486     /*kDebug() << "------------------";
487     kDebug() << dvddoc.toString();
488     kDebug() << "------------------";*/
489
490     QStringList args;
491     args << "-x" << m_authorFile.fileName();
492     kDebug() << "// DVDAUTH ARGS: " << args;
493     if (m_dvdauthor) {
494         m_dvdauthor->blockSignals(true);
495         m_dvdauthor->close();
496         delete m_dvdauthor;
497         m_dvdauthor = NULL;
498     }
499     m_creationLog.clear();
500     m_dvdauthor = new QProcess(this);
501     // Set VIDEO_FORMAT variable (required by dvdauthor 0.7)
502 #if QT_VERSION >= 0x040600
503     QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
504     env.insert("VIDEO_FORMAT", m_pageVob->isPal() ? "PAL" : "NTSC"); 
505     m_dvdauthor->setProcessEnvironment(env);
506 #else
507     QStringList env = QProcess::systemEnvironment();
508     env << QString("VIDEO_FORMAT=") + QString(m_pageVob->isPal() ? "PAL" : "NTSC");
509     m_dvdauthor->setEnvironment(env);
510 #endif
511     connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
512     connect(m_dvdauthor, SIGNAL(readyReadStandardOutput()), this, SLOT(slotShowRenderInfo()));
513     m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
514     m_dvdauthor->start("dvdauthor", args);
515     m_status.button_abort->setEnabled(true);
516     button(QWizard::FinishButton)->setEnabled(false);
517 }
518
519 void DvdWizard::slotShowRenderInfo()
520 {
521     QString log = QString(m_dvdauthor->readAll());
522     m_status.error_log->append(log);
523     m_status.error_box->setHidden(false);
524 }
525
526 void DvdWizard::slotRenderFinished(int exitCode, QProcess::ExitStatus status)
527 {
528     QListWidgetItem *authitem =  m_status.job_progress->item(3);
529     if (status == QProcess::CrashExit || exitCode != 0) {
530         QString result(m_dvdauthor->readAllStandardError());
531         result.append("<a name=\"result\" /><br /><strong>");
532         result.append(i18n("DVDAuthor process crashed.</strong><br />"));
533         m_status.error_log->append(result);
534         m_status.error_log->scrollToAnchor("result");
535         m_status.error_box->setHidden(false);
536         m_status.error_box->setTabBarHidden(false);
537         m_status.menu_file->setPlainText(m_menuFile.readAll());
538         m_status.dvd_file->setPlainText(m_authorFile.readAll());
539         kDebug() << "DVDAuthor process crashed";
540         authitem->setIcon(KIcon("dialog-close"));
541         m_dvdauthor->close();
542         delete m_dvdauthor;
543         m_dvdauthor = NULL;
544         m_status.button_start->setEnabled(true);
545         m_status.button_abort->setEnabled(false);
546         cleanup();
547         button(QWizard::FinishButton)->setEnabled(true);
548         return;
549     }
550     m_creationLog.append(m_dvdauthor->readAllStandardError());
551     m_dvdauthor->close();
552     delete m_dvdauthor;
553     m_dvdauthor = NULL;
554
555     // Check if DVD structure has the necessary infos
556     if (!QFile::exists(m_status.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
557         m_status.error_log->append(m_creationLog + "<a name=\"result\" /><br /><strong>" + i18n("DVD structure broken"));
558         m_status.error_log->scrollToAnchor("result");
559         m_status.error_box->setHidden(false);
560         m_status.error_box->setTabBarHidden(false);
561         m_status.menu_file->setPlainText(m_menuFile.readAll());
562         m_status.dvd_file->setPlainText(m_authorFile.readAll());
563         kDebug() << "DVDAuthor process crashed";
564         authitem->setIcon(KIcon("dialog-close"));
565         m_status.button_start->setEnabled(true);
566         m_status.button_abort->setEnabled(false);
567         cleanup();
568         button(QWizard::FinishButton)->setEnabled(true);
569         return;
570     }
571     authitem->setIcon(KIcon("dialog-ok"));
572     qApp->processEvents();
573     QStringList args;
574     args << "-dvd-video" << "-v" << "-o" << m_status.iso_image->url().path() << m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD";
575
576     if (m_mkiso) {
577         m_mkiso->blockSignals(true);
578         m_mkiso->close();
579         delete m_mkiso;
580         m_mkiso = NULL;
581     }
582     m_mkiso = new QProcess(this);
583     connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
584     connect(m_mkiso, SIGNAL(readyReadStandardOutput()), this, SLOT(slotShowIsoInfo()));
585     m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
586     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
587     m_status.job_progress->setCurrentRow(4);
588     isoitem->setIcon(KIcon("system-run"));
589     if (!KStandardDirs::findExe("genisoimage").isEmpty()) m_mkiso->start("genisoimage", args);
590     else m_mkiso->start("mkisofs", args);
591
592 }
593
594 void DvdWizard::slotShowIsoInfo()
595 {
596     QString log = QString(m_mkiso->readAll());
597     m_status.error_log->append(log);
598     m_status.error_box->setHidden(false);
599 }
600
601 void DvdWizard::slotIsoFinished(int exitCode, QProcess::ExitStatus status)
602 {
603     button(QWizard::FinishButton)->setEnabled(true);
604     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
605     if (status == QProcess::CrashExit || exitCode != 0) {
606         QString result(m_mkiso->readAllStandardError());
607         result.append("<a name=\"result\" /><br /><strong>");
608         result.append(i18n("ISO creation process crashed."));
609         m_status.error_log->append(result);
610         m_status.error_log->scrollToAnchor("result");
611         m_status.error_box->setHidden(false);
612         m_status.error_box->setTabBarHidden(false);
613         m_status.menu_file->setPlainText(m_menuFile.readAll());
614         m_status.dvd_file->setPlainText(m_authorFile.readAll());
615         m_mkiso->close();
616         delete m_mkiso;
617         m_mkiso = NULL;
618         cleanup();
619         kDebug() << "Iso process crashed";
620         isoitem->setIcon(KIcon("dialog-close"));
621         m_status.button_start->setEnabled(true);
622         m_status.button_abort->setEnabled(false);
623         return;
624     }
625
626     m_creationLog.append(m_mkiso->readAllStandardError());
627     delete m_mkiso;
628     m_mkiso = NULL;
629     m_status.button_start->setEnabled(true);
630     m_status.button_abort->setEnabled(false);
631
632     // Check if DVD iso is ok
633     QFile iso(m_status.iso_image->url().path());
634     if (!iso.exists() || iso.size() == 0) {
635         if (iso.exists()) {
636             KIO::NetAccess::del(m_status.iso_image->url(), this);
637         }
638         m_status.error_log->append(m_creationLog + "<br /><a name=\"result\" /><strong>" + i18n("DVD ISO is broken") + "</strong>");
639         m_status.error_log->scrollToAnchor("result");
640         m_status.error_box->setHidden(false);
641         m_status.error_box->setTabBarHidden(false);
642         m_status.menu_file->setPlainText(m_menuFile.readAll());
643         m_status.dvd_file->setPlainText(m_authorFile.readAll());
644         isoitem->setIcon(KIcon("dialog-close"));
645         cleanup();
646         return;
647     }
648
649     isoitem->setIcon(KIcon("dialog-ok"));
650     kDebug() << "ISO IMAGE " << m_status.iso_image->url().path() << " Successfully created";
651     cleanup();
652     kDebug() << m_creationLog;
653
654     m_status.error_log->append("<a name=\"result\" /><strong>" + i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()) + "</strong>");
655     m_status.error_log->scrollToAnchor("result");
656     m_status.button_preview->setEnabled(true);
657     m_status.button_burn->setEnabled(true);
658     m_status.error_box->setHidden(false);
659     //KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()));
660
661 }
662
663
664 void DvdWizard::cleanup()
665 {
666     KIO::NetAccess::del(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), this);
667 }
668
669
670 void DvdWizard::slotPreview()
671 {
672     QString programName("xine");
673     QString exec = KStandardDirs::findExe(programName);
674     if (exec.isEmpty()) KMessageBox::sorry(this, i18n("You need program <b>%1</b> to perform this action", programName));
675     else QProcess::startDetached(exec, QStringList() << "dvd://" + m_status.iso_image->url().path());
676 }
677
678 void DvdWizard::slotBurn()
679 {
680     QAction *action = qobject_cast<QAction *>(sender());
681     QString exec = action->data().toString();
682     QStringList args;
683     if (exec.endsWith("k3b")) args << "--image" << m_status.iso_image->url().path();
684     else args << "--image=" + m_status.iso_image->url().path();
685     QProcess::startDetached(exec, args);
686 }
687
688
689 void DvdWizard::slotGenerate()
690 {
691     // clear job icons
692     if ((m_dvdauthor && m_dvdauthor->state() != QProcess::NotRunning) || (m_mkiso && m_mkiso->state() != QProcess::NotRunning)) return;
693     for (int i = 0; i < m_status.job_progress->count(); i++)
694         m_status.job_progress->item(i)->setIcon(KIcon());
695     QString warnMessage;
696     if (KIO::NetAccess::exists(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), KIO::NetAccess::SourceSide, this))
697         warnMessage.append(i18n("Folder %1 already exists. Overwrite?\n", m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"));
698     if (KIO::NetAccess::exists(KUrl(m_status.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
699         warnMessage.append(i18n("Image file %1 already exists. Overwrite?", m_status.iso_image->url().path()));
700
701     if (warnMessage.isEmpty() || KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::Yes) {
702         KIO::NetAccess::del(KUrl(m_status.tmp_folder->url().path(KUrl::AddTrailingSlash) + "DVD"), this);
703         QTimer::singleShot(300, this, SLOT(generateDvd()));
704         m_status.button_preview->setEnabled(false);
705         m_status.button_burn->setEnabled(false);
706         m_status.job_progress->setEnabled(true);
707         m_status.button_start->setEnabled(false);
708     }
709 }
710
711 void DvdWizard::slotAbort()
712 {
713     // clear job icons
714     if (m_dvdauthor && m_dvdauthor->state() != QProcess::NotRunning) m_dvdauthor->terminate();
715     else if (m_mkiso && m_mkiso->state() != QProcess::NotRunning) m_mkiso->terminate();
716 }
717
718 void DvdWizard::slotSave()
719 {
720     KUrl url = KFileDialog::getSaveUrl(KUrl("kfiledialog:///projectfolder"), "*.kdvd", this, i18n("Save DVD Project"));
721     if (url.isEmpty()) return;
722
723     if (currentId() == 0) m_pageChapters->setVobFiles(m_pageVob->isPal(), m_pageVob->isWide(), m_pageVob->selectedUrls(), m_pageVob->durations(), m_pageVob->chapters());
724
725     QDomDocument doc;
726     QDomElement dvdproject = doc.createElement("dvdproject");
727     QString profile;
728     if (m_pageVob->isPal()) profile = "dv_pal";
729     else profile = "dv_ntsc";
730     if (m_pageVob->isWide()) profile.append("_wide");
731     dvdproject.setAttribute("profile", profile);
732
733     dvdproject.setAttribute("tmp_folder", m_status.tmp_folder->url().path());
734     dvdproject.setAttribute("iso_image", m_status.iso_image->url().path());
735
736     dvdproject.setAttribute("intro_movie", m_pageVob->introMovie());
737
738     doc.appendChild(dvdproject);
739     QDomElement menu = m_pageMenu->toXml();
740     if (!menu.isNull()) dvdproject.appendChild(doc.importNode(menu, true));
741     QDomElement chaps = m_pageChapters->toXml();
742     if (!chaps.isNull()) dvdproject.appendChild(doc.importNode(chaps, true));
743
744     QFile file(url.path());
745     if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
746         kWarning() << "//////  ERROR writing to file: " << url.path();
747         KMessageBox::error(this, i18n("Cannot write to file %1", url.path()));
748         return;
749     }
750
751     file.write(doc.toString().toUtf8());
752     if (file.error() != QFile::NoError) {
753         KMessageBox::error(this, i18n("Cannot write to file %1", url.path()));
754     }
755     file.close();
756 }
757
758
759 void DvdWizard::slotLoad()
760 {
761     KUrl url = KFileDialog::getOpenUrl(KUrl("kfiledialog:///projectfolder"), "*.kdvd");
762     if (url.isEmpty()) return;
763     QDomDocument doc;
764     QFile file(url.path());
765     doc.setContent(&file, false);
766     file.close();
767     QDomElement dvdproject = doc.documentElement();
768     if (dvdproject.tagName() != "dvdproject") {
769         KMessageBox::error(this, i18n("File %1 is not a Kdenlive project file.", url.path()));
770         return;
771     }
772
773     QString profile = dvdproject.attribute("profile");
774     m_pageVob->setProfile(profile);
775
776     m_status.tmp_folder->setUrl(KUrl(dvdproject.attribute("tmp_folder")));
777     m_status.iso_image->setUrl(KUrl(dvdproject.attribute("iso_image")));
778     m_pageVob->setIntroMovie(dvdproject.attribute("intro_movie"));
779
780     QDomNodeList vobs = doc.elementsByTagName("vob");
781     m_pageVob->clear();
782     for (int i = 0; i < vobs.count(); i++) {
783         QDomElement e = vobs.at(i).toElement();
784         m_pageVob->slotAddVobFile(KUrl(e.attribute("file")), e.attribute("chapters"));
785     }
786     m_pageMenu->loadXml(dvdproject.firstChildElement("menu"));
787 }