1 /***************************************************************************
2 * Copyright (C) 2007 by Jean-Baptiste Mardelle (jb@kdenlive.org) *
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. *
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. *
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 ***************************************************************************/
21 #include "dvdwizard.h"
22 #include "kdenlivesettings.h"
23 #include "profilesdialog.h"
25 #include <KStandardDirs>
27 #include <KFileDialog>
28 #include <kmimetype.h>
29 #include <KIO/NetAccess>
30 #include <KMessageBox>
33 #include <QApplication>
35 #include <QDomDocument>
39 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent) :
43 m_burnMenu(new QMenu(this))
45 //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
46 setAttribute(Qt::WA_DeleteOnClose);
47 m_pageVob = new DvdWizardVob(this);
48 m_pageVob->setTitle(i18n("Select Files For Your DVD"));
51 if (!url.isEmpty()) m_pageVob->setUrl(url);
53 m_pageMenu = new DvdWizardMenu(profile, this);
54 m_pageMenu->setTitle(i18n("Create DVD Menu"));
58 QWizardPage *page3 = new QWizardPage;
59 page3->setTitle(i18n("DVD Image"));
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);
67 QWizardPage *page4 = new QWizardPage;
68 page4->setTitle(i18n("Creating DVD Image"));
69 m_status.setupUi(page4);
70 m_status.error_box->setHidden(true);
73 connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
75 connect(m_status.button_preview, SIGNAL(clicked()), this, SLOT(slotPreview()));
77 QString programName("k3b");
78 QString exec = KStandardDirs::findExe(programName);
79 if (!exec.isEmpty()) {
81 QAction *k3b = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
84 programName = "brasero";
85 exec = KStandardDirs::findExe(programName);
86 if (!exec.isEmpty()) {
88 QAction *brasero = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
89 brasero->setData(exec);
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"));
99 DvdWizard::~DvdWizard()
101 // m_menuFile.remove();
104 m_dvdauthor->close();
114 void DvdWizard::slotPageChanged(int page)
116 kDebug() << "// PAGE CHGD: " << page;
118 m_pageMenu->setTargets(m_pageVob->selectedTitles(), m_pageVob->selectedTargets());
119 } else if (page == 2) {
120 //m_pageMenu->buttonsInfo();
121 } else if (page == 3) {
123 for (int i = 0; i < m_status.job_progress->count(); i++)
124 m_status.job_progress->item(i)->setIcon(KIcon());
126 if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
127 warnMessage.append(i18n("Folder %1 already exists. Overwrite?" + '\n', m_iso.tmp_folder->url().path() + "/DVD"));
128 if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
129 warnMessage.append(i18n("Image file %1 already exists. Overwrite?", m_iso.iso_image->url().path()));
131 if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
134 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
135 QTimer::singleShot(300, this, SLOT(generateDvd()));
137 m_status.button_preview->setEnabled(false);
138 m_status.button_burn->setEnabled(false);
144 void DvdWizard::generateDvd()
146 KTemporaryFile temp1;
147 temp1.setSuffix(".png");
148 //temp1.setAutoRemove(false);
151 KTemporaryFile temp2;
152 temp2.setSuffix(".png");
153 //temp2.setAutoRemove(false);
156 KTemporaryFile temp3;
157 temp3.setSuffix(".png");
158 //temp3.setAutoRemove(false);
161 KTemporaryFile temp4;
162 temp4.setSuffix(".png");
163 //temp4.setAutoRemove(false);
166 KTemporaryFile temp5;
167 temp5.setSuffix(".vob");
168 //temp5.setAutoRemove(false);
171 KTemporaryFile temp6;
172 temp6.setSuffix(".xml");
173 //temp6.setAutoRemove(false);
176 m_menuFile.setSuffix(".mpg");
177 m_menuFile.setAutoRemove(false);
180 m_authorFile.setSuffix(".xml");
181 m_authorFile.setAutoRemove(false);
184 QListWidgetItem *images = m_status.job_progress->item(0);
185 images->setIcon(KIcon("system-run"));
186 qApp->processEvents();
187 QMap <QString, QRect> buttons = m_pageMenu->buttonsInfo();
188 QStringList buttonsTarget;
190 if (m_pageMenu->createMenu()) {
191 m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
192 m_pageMenu->createBackgroundImage(temp4.fileName());
195 images->setIcon(KIcon("dialog-ok"));
197 kDebug() << "/// STARTING MLT VOB CREATION";
198 if (!m_pageMenu->menuMovie()) {
199 // create menu vob file
200 QListWidgetItem *vobitem = m_status.job_progress->item(1);
201 vobitem->setIcon(KIcon("system-run"));
202 qApp->processEvents();
205 args.append("-profile");
206 if (m_pageMenu->isPalMenu()) args.append("dv_pal");
207 else args.append("dv_ntsc");
208 args.append(temp4.fileName());
210 args.append("out=100");
211 args << "-consumer" << "avformat:" + temp5.fileName();
212 if (m_pageMenu->isPalMenu()) {
213 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 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";
218 kDebug() << "MLT ARGS: " << args;
220 renderbg.start(KdenliveSettings::rendererpath(), args);
221 if (renderbg.waitForFinished()) {
222 if (renderbg.exitStatus() == QProcess::CrashExit) {
223 kDebug() << "/// RENDERING MENUÂ vob crashed";
224 QByteArray result = renderbg.readAllStandardError();
225 vobitem->setIcon(KIcon("dialog-close"));
226 m_status.error_log->setText(result);
227 m_status.error_box->setHidden(false);
231 kDebug() << "/// RENDERING MENUÂ vob timed out";
232 vobitem->setIcon(KIcon("dialog-close"));
233 m_status.error_log->setText(i18n("Rendering job timed out"));
234 m_status.error_box->setHidden(false);
237 vobitem->setIcon(KIcon("dialog-ok"));
239 kDebug() << "/// STARTING SPUMUX";
241 // create xml spumux file
242 QListWidgetItem *spuitem = m_status.job_progress->item(2);
243 spuitem->setIcon(KIcon("system-run"));
244 qApp->processEvents();
246 QDomElement sub = doc.createElement("subpictures");
247 doc.appendChild(sub);
248 QDomElement stream = doc.createElement("stream");
249 sub.appendChild(stream);
250 QDomElement spu = doc.createElement("spu");
251 stream.appendChild(spu);
252 spu.setAttribute("force", "yes");
253 spu.setAttribute("start", "00:00:00.00");
254 spu.setAttribute("image", temp1.fileName());
255 spu.setAttribute("select", temp2.fileName());
256 spu.setAttribute("highlight", temp3.fileName());
257 /*spu.setAttribute("autooutline", "infer");
258 spu.setAttribute("outlinewidth", "12");
259 spu.setAttribute("autoorder", "rows");*/
261 int max = buttons.count() - 1;
263 QMapIterator<QString, QRect> it(buttons);
264 while (it.hasNext()) {
266 QDomElement but = doc.createElement("button");
267 but.setAttribute("name", 'b' + QString::number(i));
268 if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
269 else but.setAttribute("down", "b0");
270 if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
271 else but.setAttribute("up", 'b' + QString::number(max));
272 QRect r = it.value();
273 //int target = it.key();
274 // TODO:Â solve play all button
275 //if (target == 0) target = 1;
276 buttonsTarget.append(it.key());
277 but.setAttribute("x0", QString::number(r.x()));
278 but.setAttribute("y0", QString::number(r.y()));
279 but.setAttribute("x1", QString::number(r.right()));
280 but.setAttribute("y1", QString::number(r.bottom()));
281 spu.appendChild(but);
285 QFile data(temp6.fileName());
286 if (data.open(QFile::WriteOnly)) {
287 data.write(doc.toString().toUtf8());
291 kDebug() << " SPUMUX DATA: " << doc.toString();
294 args.append(temp6.fileName());
295 kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
299 if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
300 else spumux.setStandardInputFile(temp5.fileName());
301 spumux.setStandardOutputFile(m_menuFile.fileName());
302 spumux.start("spumux", args);
303 spumux.setProcessChannelMode(QProcess::MergedChannels);
304 if (spumux.waitForFinished()) {
305 kDebug() << QString(spumux.readAll()).simplified();
306 if (spumux.exitStatus() == QProcess::CrashExit) {
307 QByteArray result = spumux.readAllStandardError();
308 spuitem->setIcon(KIcon("dialog-close"));
309 m_status.error_log->setText(result);
310 m_status.error_box->setHidden(false);
311 kDebug() << "/// RENDERING SPUMUX MENU crashed";
315 kDebug() << "/// RENDERING SPUMUX MENU timed out";
316 spuitem->setIcon(KIcon("dialog-close"));
317 m_status.error_log->setText(i18n("Menu job timed out"));
318 m_status.error_box->setHidden(false);
322 spuitem->setIcon(KIcon("dialog-ok"));
323 kDebug() << "/// DONE: " << m_menuFile.fileName();
326 // create dvdauthor xml
327 QListWidgetItem *authitem = m_status.job_progress->item(3);
328 authitem->setIcon(KIcon("system-run"));
329 qApp->processEvents();
330 KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
333 QDomElement auth = dvddoc.createElement("dvdauthor");
334 auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
335 dvddoc.appendChild(auth);
336 QDomElement vmgm = dvddoc.createElement("vmgm");
337 auth.appendChild(vmgm);
339 if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
341 QDomElement menus = dvddoc.createElement("menus");
342 vmgm.appendChild(menus);
343 QDomElement pgc = dvddoc.createElement("pgc");
344 pgc.setAttribute("entry", "title");
345 menus.appendChild(pgc);
346 QDomElement menuvob = dvddoc.createElement("vob");
347 menuvob.setAttribute("file", m_pageVob->introMovie());
348 pgc.appendChild(menuvob);
349 QDomElement post = dvddoc.createElement("post");
350 QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
351 post.appendChild(call);
352 pgc.appendChild(post);
354 QDomElement titleset = dvddoc.createElement("titleset");
355 auth.appendChild(titleset);
357 if (m_pageMenu->createMenu()) {
360 QDomElement menus = dvddoc.createElement("menus");
361 titleset.appendChild(menus);
362 QDomElement pgc = dvddoc.createElement("pgc");
363 pgc.setAttribute("entry", "root");
364 menus.appendChild(pgc);
365 for (int i = 0; i < buttons.count(); i++) {
366 QDomElement button = dvddoc.createElement("button");
367 button.setAttribute("name", 'b' + QString::number(i));
368 QDomText nametext = dvddoc.createTextNode("jump " + buttonsTarget.at(i) + ';');
369 button.appendChild(nametext);
370 pgc.appendChild(button);
372 QDomElement menuvob = dvddoc.createElement("vob");
373 menuvob.setAttribute("file", m_menuFile.fileName());
374 menuvob.setAttribute("pause", "inf");
375 pgc.appendChild(menuvob);
378 QDomElement titles = dvddoc.createElement("titles");
379 titleset.appendChild(titles);
381 QStringList voburls = m_pageVob->selectedUrls();
383 QDomElement pgc2 = dvddoc.createElement("pgc");
384 titles.appendChild(pgc2);
385 if (m_pageMenu->createMenu()) {
386 QDomElement post = dvddoc.createElement("post");
387 QDomText call = dvddoc.createTextNode("call menu;");
388 post.appendChild(call);
389 pgc2.appendChild(post);
393 for (int i = 0; i < voburls.count(); i++) {
394 if (!voburls.at(i).isEmpty()) {
396 QDomElement vob = dvddoc.createElement("vob");
397 vob.setAttribute("file", voburls.at(i));
398 if (m_pageVob->useChapters()) {
400 if (QFile::exists(voburls.at(i) + ".dvdchapter")) {
401 QFile file(voburls.at(i) + ".dvdchapter");
402 if (file.open(QIODevice::ReadOnly)) {
404 doc.setContent(&file);
406 QDomNodeList chapters = doc.elementsByTagName("chapter");
407 QStringList chaptersList;
408 for (int j = 0; j < chapters.count(); j++) {
409 chaptersList.append(chapters.at(j).toElement().attribute("time"));
411 if (!chaptersList.isEmpty()) vob.setAttribute("chapters", chaptersList.join(","));
415 pgc2.appendChild(vob);
420 // create one pgc for each video
421 for (int i = 0; i < voburls.count(); i++) {
422 if (!voburls.at(i).isEmpty()) {
425 QDomElement pgc2 = dvddoc.createElement("pgc");
426 titles.appendChild(pgc2);
427 if (m_pageMenu->createMenu()) {
428 QDomElement post = dvddoc.createElement("post");
429 QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
430 post.appendChild(call);
431 pgc2.appendChild(post);
434 QDomElement vob = dvddoc.createElement("vob");
435 vob.setAttribute("file", voburls.at(i));
436 pgc2.appendChild(vob);
440 QFile data2(m_authorFile.fileName());
441 if (data2.open(QFile::WriteOnly)) {
442 data2.write(dvddoc.toString().toUtf8());
445 /*kDebug() << "------------------";
446 kDebug() << dvddoc.toString();
447 kDebug() << "------------------";*/
450 args << "-x" << m_authorFile.fileName();
451 kDebug() << "// DVDAUTH ARGS: " << args;
453 m_dvdauthor->close();
457 m_creationLog.clear();
458 m_dvdauthor = new QProcess(this);
459 connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
460 m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
461 m_dvdauthor->start("dvdauthor", args);
465 void DvdWizard::slotRenderFinished(int /*exitCode*/, QProcess::ExitStatus status)
467 QListWidgetItem *authitem = m_status.job_progress->item(3);
468 if (status == QProcess::CrashExit) {
469 QByteArray result = m_dvdauthor->readAllStandardError();
470 m_status.error_log->setText(result);
471 m_status.error_box->setHidden(false);
472 kDebug() << "DVDAuthor process crashed";
473 authitem->setIcon(KIcon("dialog-close"));
474 m_dvdauthor->close();
480 m_creationLog.append(m_dvdauthor->readAllStandardError());
481 m_dvdauthor->close();
485 // Check if DVDÂ structure has the necessary infos
486 if (!QFile::exists(m_iso.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
487 m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD structure broken"));
488 m_status.error_box->setHidden(false);
489 kDebug() << "DVDAuthor process crashed";
490 authitem->setIcon(KIcon("dialog-close"));
494 authitem->setIcon(KIcon("dialog-ok"));
495 qApp->processEvents();
497 args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
504 m_mkiso = new QProcess(this);
505 connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
506 m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
507 QListWidgetItem *isoitem = m_status.job_progress->item(4);
508 isoitem->setIcon(KIcon("system-run"));
509 m_mkiso->start("mkisofs", args);
513 void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status)
515 QListWidgetItem *isoitem = m_status.job_progress->item(4);
516 if (status == QProcess::CrashExit) {
517 QByteArray result = m_mkiso->readAllStandardError();
518 m_status.error_log->setText(result);
519 m_status.error_box->setHidden(false);
524 kDebug() << "Iso process crashed";
525 isoitem->setIcon(KIcon("dialog-close"));
529 m_creationLog.append(m_mkiso->readAllStandardError());
533 // Check if DVDÂ iso is ok
534 QFile iso(m_iso.iso_image->url().path());
535 if (!iso.exists() || iso.size() == 0) {
537 KIO::NetAccess::del(m_iso.iso_image->url(), this);
539 m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD ISO is broken"));
540 m_status.error_box->setHidden(false);
541 isoitem->setIcon(KIcon("dialog-close"));
546 isoitem->setIcon(KIcon("dialog-ok"));
547 kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
549 kDebug() << m_creationLog;
551 m_status.error_log->setText(i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
552 m_status.button_preview->setEnabled(true);
553 m_status.button_burn->setEnabled(true);
554 m_status.error_box->setHidden(false);
555 //KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
560 void DvdWizard::cleanup()
562 m_authorFile.remove();
564 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
568 void DvdWizard::slotPreview()
570 QString programName("xine");
571 QString exec = KStandardDirs::findExe(programName);
572 if (exec.isEmpty()) KMessageBox::sorry(this, i18n("You need program <b>%1</b> to perform this action", programName));
573 else QProcess::startDetached(exec, QStringList() << "dvd://" + m_iso.iso_image->url().path());
576 void DvdWizard::slotBurn()
578 QAction *action = qobject_cast<QAction *>(sender());
579 QString exec = action->data().toString();
581 if (exec.endsWith("k3b")) args << "--image" << m_iso.iso_image->url().path();
582 else args << "--image=" + m_iso.iso_image->url().path();
583 QProcess::startDetached(exec, args);