]> git.sesse.net Git - kdenlive/blob - src/dvdwizard.cpp
Updated 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
25 #include <KStandardDirs>
26 #include <KLocale>
27 #include <KFileDialog>
28 #include <kmimetype.h>
29 #include <KIO/NetAccess>
30 #include <KMessageBox>
31
32 #include <QFile>
33 #include <QApplication>
34 #include <QTimer>
35 #include <QDomDocument>
36 #include <QMenu>
37
38
39 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent) :
40         QWizard(parent),
41         m_dvdauthor(NULL),
42         m_mkiso(NULL),
43         m_burnMenu(new QMenu(this))
44 {
45     //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
46     setAttribute(Qt::WA_DeleteOnClose);
47     m_pageVob = new DvdWizardVob(profile, this);
48     m_pageVob->setTitle(i18n("Select Files For Your DVD"));
49     addPage(m_pageVob);
50
51     if (!url.isEmpty()) m_pageVob->setUrl(url);
52
53     m_pageMenu = new DvdWizardMenu(profile, this);
54     m_pageMenu->setTitle(i18n("Create DVD Menu"));
55     addPage(m_pageMenu);
56
57
58     QWizardPage *page3 = new QWizardPage;
59     page3->setTitle(i18n("DVD Image"));
60     m_iso.setupUi(page3);
61     m_iso.tmp_folder->setPath(KdenliveSettings::currenttmpfolder());
62     m_iso.iso_image->setPath(QDir::homePath() + "/untitled.iso");
63     m_iso.iso_image->setFilter("*.iso");
64     m_iso.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
65     addPage(page3);
66
67     QWizardPage *page4 = new QWizardPage;
68     page4->setTitle(i18n("Creating DVD Image"));
69     m_status.setupUi(page4);
70     m_status.error_box->setHidden(true);
71     addPage(page4);
72
73     connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
74
75     connect(m_status.button_preview, SIGNAL(clicked()), this, SLOT(slotPreview()));
76
77     QString programName("k3b");
78     QString exec = KStandardDirs::findExe(programName);
79     if (!exec.isEmpty()) {
80         //Add K3b action
81         QAction *k3b = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
82         k3b->setData(exec);
83     }
84     programName = "brasero";
85     exec = KStandardDirs::findExe(programName);
86     if (!exec.isEmpty()) {
87         //Add Brasero action
88         QAction *brasero = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
89         brasero->setData(exec);
90     }
91     if (m_burnMenu->isEmpty()) m_burnMenu->addAction(i18n("No burning program found (K3b, Brasero)"));
92     m_status.button_burn->setMenu(m_burnMenu);
93     m_status.button_burn->setIcon(KIcon("tools-media-optical-burn"));
94     m_status.button_burn->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
95     m_status.button_preview->setIcon(KIcon("media-playback-start"));
96
97 }
98
99 DvdWizard::~DvdWizard()
100 {
101     // m_menuFile.remove();
102     delete m_burnMenu;
103     if (m_dvdauthor) {
104         m_dvdauthor->close();
105         delete m_dvdauthor;
106     }
107     if (m_mkiso) {
108         m_mkiso->close();
109         delete m_mkiso;
110     }
111 }
112
113
114 void DvdWizard::slotPageChanged(int page)
115 {
116     kDebug() << "// PAGE CHGD: " << page;
117     if (page == 1) {
118         m_pageMenu->setTargets(m_pageVob->selectedTitles(), m_pageVob->selectedTargets());
119         m_pageMenu->changeProfile(m_pageVob->isPal());
120     } else if (page == 2) {
121         //m_pageMenu->buttonsInfo();
122     } else if (page == 3) {
123         // clear job icons
124         for (int i = 0; i < m_status.job_progress->count(); i++)
125             m_status.job_progress->item(i)->setIcon(KIcon());
126         QString warnMessage;
127         if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
128             warnMessage.append(i18n("Folder %1 already exists. Overwrite?" + '\n', m_iso.tmp_folder->url().path() + "/DVD"));
129         if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
130             warnMessage.append(i18n("Image file %1 already exists. Overwrite?", m_iso.iso_image->url().path()));
131
132         if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
133             back();
134         } else {
135             KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
136             QTimer::singleShot(300, this, SLOT(generateDvd()));
137         }
138         m_status.button_preview->setEnabled(false);
139         m_status.button_burn->setEnabled(false);
140     }
141 }
142
143
144
145 void DvdWizard::generateDvd()
146 {
147     KTemporaryFile temp1;
148     temp1.setSuffix(".png");
149     //temp1.setAutoRemove(false);
150     temp1.open();
151
152     KTemporaryFile temp2;
153     temp2.setSuffix(".png");
154     //temp2.setAutoRemove(false);
155     temp2.open();
156
157     KTemporaryFile temp3;
158     temp3.setSuffix(".png");
159     //temp3.setAutoRemove(false);
160     temp3.open();
161
162     KTemporaryFile temp4;
163     temp4.setSuffix(".png");
164     //temp4.setAutoRemove(false);
165     temp4.open();
166
167     KTemporaryFile temp5;
168     temp5.setSuffix(".vob");
169     //temp5.setAutoRemove(false);
170     temp5.open();
171
172     KTemporaryFile temp6;
173     temp6.setSuffix(".xml");
174     //temp6.setAutoRemove(false);
175     temp6.open();
176
177     m_menuFile.setSuffix(".mpg");
178     m_menuFile.setAutoRemove(false);
179     m_menuFile.open();
180
181     m_authorFile.setSuffix(".xml");
182     m_authorFile.setAutoRemove(false);
183     m_authorFile.open();
184
185     QListWidgetItem *images =  m_status.job_progress->item(0);
186     images->setIcon(KIcon("system-run"));
187     qApp->processEvents();
188     QMap <QString, QRect> buttons = m_pageMenu->buttonsInfo();
189     QStringList buttonsTarget;
190
191     if (m_pageMenu->createMenu()) {
192         m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
193         m_pageMenu->createBackgroundImage(temp4.fileName());
194
195
196         images->setIcon(KIcon("dialog-ok"));
197
198         kDebug() << "/// STARTING MLT VOB CREATION";
199         if (!m_pageMenu->menuMovie()) {
200             // create menu vob file
201             QListWidgetItem *vobitem =  m_status.job_progress->item(1);
202             vobitem->setIcon(KIcon("system-run"));
203             qApp->processEvents();
204
205             QStringList args;
206             args.append("-profile");
207             if (m_pageMenu->isPalMenu()) args.append("dv_pal");
208             else  args.append("dv_ntsc");
209             args.append(temp4.fileName());
210             args.append("in=0");
211             args.append("out=100");
212             args << "-consumer" << "avformat:" + temp5.fileName();
213             if (m_pageMenu->isPalMenu()) {
214                 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";
215             } else {
216                 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";
217             }
218
219             kDebug() << "MLT ARGS: " << args;
220             QProcess renderbg;
221             renderbg.start(KdenliveSettings::rendererpath(), args);
222             if (renderbg.waitForFinished()) {
223                 if (renderbg.exitStatus() == QProcess::CrashExit) {
224                     kDebug() << "/// RENDERING MENU vob crashed";
225                     QByteArray result = renderbg.readAllStandardError();
226                     vobitem->setIcon(KIcon("dialog-close"));
227                     m_status.error_log->setText(result);
228                     m_status.error_box->setHidden(false);
229                     return;
230                 }
231             } else {
232                 kDebug() << "/// RENDERING MENU vob timed out";
233                 vobitem->setIcon(KIcon("dialog-close"));
234                 m_status.error_log->setText(i18n("Rendering job timed out"));
235                 m_status.error_box->setHidden(false);
236                 return;
237             }
238             vobitem->setIcon(KIcon("dialog-ok"));
239         }
240         kDebug() << "/// STARTING SPUMUX";
241
242         // create xml spumux file
243         QListWidgetItem *spuitem =  m_status.job_progress->item(2);
244         spuitem->setIcon(KIcon("system-run"));
245         qApp->processEvents();
246         QDomDocument doc;
247         QDomElement sub = doc.createElement("subpictures");
248         doc.appendChild(sub);
249         QDomElement stream = doc.createElement("stream");
250         sub.appendChild(stream);
251         QDomElement spu = doc.createElement("spu");
252         stream.appendChild(spu);
253         spu.setAttribute("force", "yes");
254         spu.setAttribute("start", "00:00:00.00");
255         spu.setAttribute("image", temp1.fileName());
256         spu.setAttribute("select", temp2.fileName());
257         spu.setAttribute("highlight", temp3.fileName());
258         /*spu.setAttribute("autooutline", "infer");
259         spu.setAttribute("outlinewidth", "12");
260         spu.setAttribute("autoorder", "rows");*/
261
262         int max = buttons.count() - 1;
263         int i = 0;
264         QMapIterator<QString, QRect> it(buttons);
265         while (it.hasNext()) {
266             it.next();
267             QDomElement but = doc.createElement("button");
268             but.setAttribute("name", 'b' + QString::number(i));
269             if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
270             else but.setAttribute("down", "b0");
271             if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
272             else but.setAttribute("up", 'b' + QString::number(max));
273             QRect r = it.value();
274             //int target = it.key();
275             // TODO: solve play all button
276             //if (target == 0) target = 1;
277             buttonsTarget.append(it.key());
278             but.setAttribute("x0", QString::number(r.x()));
279             but.setAttribute("y0", QString::number(r.y()));
280             but.setAttribute("x1", QString::number(r.right()));
281             but.setAttribute("y1", QString::number(r.bottom()));
282             spu.appendChild(but);
283             i++;
284         }
285
286         QFile data(temp6.fileName());
287         if (data.open(QFile::WriteOnly)) {
288             data.write(doc.toString().toUtf8());
289         }
290         data.close();
291
292         kDebug() << " SPUMUX DATA: " << doc.toString();
293
294         QStringList args;
295         args.append(temp6.fileName());
296         kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
297
298         QProcess spumux;
299
300         if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
301         else spumux.setStandardInputFile(temp5.fileName());
302         spumux.setStandardOutputFile(m_menuFile.fileName());
303         spumux.start("spumux", args);
304         spumux.setProcessChannelMode(QProcess::MergedChannels);
305         if (spumux.waitForFinished()) {
306             kDebug() << QString(spumux.readAll()).simplified();
307             if (spumux.exitStatus() == QProcess::CrashExit) {
308                 QByteArray result = spumux.readAllStandardError();
309                 spuitem->setIcon(KIcon("dialog-close"));
310                 m_status.error_log->setText(result);
311                 m_status.error_box->setHidden(false);
312                 kDebug() << "/// RENDERING SPUMUX MENU crashed";
313                 return;
314             }
315         } else {
316             kDebug() << "/// RENDERING SPUMUX MENU timed out";
317             spuitem->setIcon(KIcon("dialog-close"));
318             m_status.error_log->setText(i18n("Menu job timed out"));
319             m_status.error_box->setHidden(false);
320             return;
321         }
322
323         spuitem->setIcon(KIcon("dialog-ok"));
324         kDebug() << "/// DONE: " << m_menuFile.fileName();
325     }
326
327     // create dvdauthor xml
328     QListWidgetItem *authitem =  m_status.job_progress->item(3);
329     authitem->setIcon(KIcon("system-run"));
330     qApp->processEvents();
331     KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
332
333     QDomDocument dvddoc;
334     QDomElement auth = dvddoc.createElement("dvdauthor");
335     auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
336     dvddoc.appendChild(auth);
337     QDomElement vmgm = dvddoc.createElement("vmgm");
338     auth.appendChild(vmgm);
339
340     if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
341         // intro movie
342         QDomElement menus = dvddoc.createElement("menus");
343         vmgm.appendChild(menus);
344         QDomElement pgc = dvddoc.createElement("pgc");
345         pgc.setAttribute("entry", "title");
346         menus.appendChild(pgc);
347         QDomElement menuvob = dvddoc.createElement("vob");
348         menuvob.setAttribute("file", m_pageVob->introMovie());
349         pgc.appendChild(menuvob);
350         QDomElement post = dvddoc.createElement("post");
351         QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
352         post.appendChild(call);
353         pgc.appendChild(post);
354     }
355     QDomElement titleset = dvddoc.createElement("titleset");
356     auth.appendChild(titleset);
357
358     if (m_pageMenu->createMenu()) {
359
360         // DVD main menu
361         QDomElement menus = dvddoc.createElement("menus");
362         titleset.appendChild(menus);
363         QDomElement pgc = dvddoc.createElement("pgc");
364         pgc.setAttribute("entry", "root");
365         menus.appendChild(pgc);
366         QDomElement pre = dvddoc.createElement("pre");
367         pgc.appendChild(pre);
368         QDomText nametext = dvddoc.createTextNode("{g1 = 0;}");
369         pre.appendChild(nametext);
370         for (int i = 0; i < buttons.count(); i++) {
371             QDomElement button = dvddoc.createElement("button");
372             button.setAttribute("name", 'b' + QString::number(i));
373             nametext = dvddoc.createTextNode('{' + buttonsTarget.at(i) + ";}");
374             button.appendChild(nametext);
375             pgc.appendChild(button);
376         }
377         QDomElement menuvob = dvddoc.createElement("vob");
378         menuvob.setAttribute("file", m_menuFile.fileName());
379         menuvob.setAttribute("pause", "inf");
380         pgc.appendChild(menuvob);
381     }
382
383     QDomElement titles = dvddoc.createElement("titles");
384     titleset.appendChild(titles);
385
386     QStringList voburls = m_pageVob->selectedUrls();
387
388     QDomElement pgc2;
389
390
391     for (int i = 0; i < voburls.count(); i++) {
392         if (!voburls.at(i).isEmpty()) {
393             // Add vob entry
394             pgc2 = dvddoc.createElement("pgc");
395             pgc2.setAttribute("pause", 0);
396             titles.appendChild(pgc2);
397             QDomElement vob = dvddoc.createElement("vob");
398             vob.setAttribute("file", voburls.at(i));
399             if (m_pageVob->useChapters()) {
400                 // Add chapters
401                 QStringList chaptersList = m_pageVob->chapter(i);
402                 if (!chaptersList.isEmpty()) vob.setAttribute("chapters", chaptersList.join(","));
403             }
404             pgc2.appendChild(vob);
405             if (m_pageMenu->createMenu()) {
406                 QDomElement post = dvddoc.createElement("post");
407                 QDomText call;
408                 if (i == voburls.count() - 1) call = dvddoc.createTextNode("{g1 = 0; call menu;}");
409                 else {
410                     call = dvddoc.createTextNode("{if ( g1 eq 999 ) { call menu; } jump title " + QString::number(i + 2) + ";}");
411                 }
412                 post.appendChild(call);
413                 pgc2.appendChild(post);
414             }
415         }
416     }
417
418
419     QFile data2(m_authorFile.fileName());
420     if (data2.open(QFile::WriteOnly)) {
421         data2.write(dvddoc.toString().toUtf8());
422     }
423     data2.close();
424     /*kDebug() << "------------------";
425     kDebug() << dvddoc.toString();
426     kDebug() << "------------------";*/
427
428     QStringList args;
429     args << "-x" << m_authorFile.fileName();
430     kDebug() << "// DVDAUTH ARGS: " << args;
431     if (m_dvdauthor) {
432         m_dvdauthor->close();
433         delete m_dvdauthor;
434         m_dvdauthor = NULL;
435     }
436     m_creationLog.clear();
437     m_dvdauthor = new QProcess(this);
438     connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
439     m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
440     m_dvdauthor->start("dvdauthor", args);
441
442 }
443
444 void DvdWizard::slotRenderFinished(int /*exitCode*/, QProcess::ExitStatus status)
445 {
446     QListWidgetItem *authitem =  m_status.job_progress->item(3);
447     if (status == QProcess::CrashExit) {
448         QByteArray result = m_dvdauthor->readAllStandardError();
449         m_status.error_log->setText(result);
450         m_status.error_box->setHidden(false);
451         kDebug() << "DVDAuthor process crashed";
452         authitem->setIcon(KIcon("dialog-close"));
453         m_dvdauthor->close();
454         delete m_dvdauthor;
455         m_dvdauthor = NULL;
456         cleanup();
457         return;
458     }
459     m_creationLog.append(m_dvdauthor->readAllStandardError());
460     m_dvdauthor->close();
461     delete m_dvdauthor;
462     m_dvdauthor = NULL;
463
464     // Check if DVD structure has the necessary infos
465     if (!QFile::exists(m_iso.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
466         m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD structure broken"));
467         m_status.error_box->setHidden(false);
468         kDebug() << "DVDAuthor process crashed";
469         authitem->setIcon(KIcon("dialog-close"));
470         cleanup();
471         return;
472     }
473     authitem->setIcon(KIcon("dialog-ok"));
474     qApp->processEvents();
475     QStringList args;
476     args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
477
478     if (m_mkiso) {
479         m_mkiso->close();
480         delete m_mkiso;
481         m_mkiso = NULL;
482     }
483     m_mkiso = new QProcess(this);
484     connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
485     m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
486     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
487     isoitem->setIcon(KIcon("system-run"));
488     m_mkiso->start("mkisofs", args);
489
490 }
491
492 void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status)
493 {
494     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
495     if (status == QProcess::CrashExit) {
496         QByteArray result = m_mkiso->readAllStandardError();
497         m_status.error_log->setText(result);
498         m_status.error_box->setHidden(false);
499         m_mkiso->close();
500         delete m_mkiso;
501         m_mkiso = NULL;
502         cleanup();
503         kDebug() << "Iso process crashed";
504         isoitem->setIcon(KIcon("dialog-close"));
505         return;
506     }
507
508     m_creationLog.append(m_mkiso->readAllStandardError());
509     delete m_mkiso;
510     m_mkiso = NULL;
511
512     // Check if DVD iso is ok
513     QFile iso(m_iso.iso_image->url().path());
514     if (!iso.exists() || iso.size() == 0) {
515         if (iso.exists()) {
516             KIO::NetAccess::del(m_iso.iso_image->url(), this);
517         }
518         m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD ISO is broken"));
519         m_status.error_box->setHidden(false);
520         isoitem->setIcon(KIcon("dialog-close"));
521         cleanup();
522         return;
523     }
524
525     isoitem->setIcon(KIcon("dialog-ok"));
526     kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
527     cleanup();
528     kDebug() << m_creationLog;
529
530     m_status.error_log->setText(i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
531     m_status.button_preview->setEnabled(true);
532     m_status.button_burn->setEnabled(true);
533     m_status.error_box->setHidden(false);
534     //KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
535
536 }
537
538
539 void DvdWizard::cleanup()
540 {
541     m_authorFile.remove();
542     m_menuFile.remove();
543     KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
544 }
545
546
547 void DvdWizard::slotPreview()
548 {
549     QString programName("xine");
550     QString exec = KStandardDirs::findExe(programName);
551     if (exec.isEmpty()) KMessageBox::sorry(this, i18n("You need program <b>%1</b> to perform this action", programName));
552     else QProcess::startDetached(exec, QStringList() << "dvd://" + m_iso.iso_image->url().path());
553 }
554
555 void DvdWizard::slotBurn()
556 {
557     QAction *action = qobject_cast<QAction *>(sender());
558     QString exec = action->data().toString();
559     QStringList args;
560     if (exec.endsWith("k3b")) args << "--image" << m_iso.iso_image->url().path();
561     else args << "--image=" + m_iso.iso_image->url().path();
562     QProcess::startDetached(exec, args);
563 }
564
565
566