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