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