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 "monitormanager.h"
27 #include <KStandardDirs>
29 #include <KFileDialog>
30 #include <kmimetype.h>
31 #include <KIO/NetAccess>
32 #include <KMessageBox>
35 #include <QApplication>
37 #include <QDomDocument>
41 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent) :
45 m_burnMenu(new QMenu(this))
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"));
52 if (!url.isEmpty()) m_pageVob->setUrl(url);
55 m_pageChapters = new DvdWizardChapters(m_pageVob->isPal(), this);
56 m_pageChapters->setTitle(i18n("DVD Chapters"));
57 addPage(m_pageChapters);
61 m_pageMenu = new DvdWizardMenu(profile, this);
62 m_pageMenu->setTitle(i18n("Create DVD Menu"));
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.iso_image->setPath(QDir::homePath() + "/untitled.iso");
71 m_status.iso_image->setFilter("*.iso");
72 m_status.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
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_save, SIGNAL(clicked()), this, SLOT(slotSave()));
82 connect(m_status.button_preview, SIGNAL(clicked()), this, SLOT(slotPreview()));
84 QString programName("k3b");
85 QString exec = KStandardDirs::findExe(programName);
86 if (!exec.isEmpty()) {
88 QAction *k3b = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
91 programName = "brasero";
92 exec = KStandardDirs::findExe(programName);
93 if (!exec.isEmpty()) {
95 QAction *brasero = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
96 brasero->setData(exec);
98 if (m_burnMenu->isEmpty()) m_burnMenu->addAction(i18n("No burning program found (K3b, Brasero)"));
99 m_status.button_burn->setMenu(m_burnMenu);
100 m_status.button_burn->setIcon(KIcon("tools-media-optical-burn"));
101 m_status.button_burn->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
102 m_status.button_preview->setIcon(KIcon("media-playback-start"));
106 DvdWizard::~DvdWizard()
108 // m_menuFile.remove();
111 m_dvdauthor->blockSignals(true);
112 m_dvdauthor->close();
116 m_mkiso->blockSignals(true);
123 void DvdWizard::slotPageChanged(int page)
125 //kDebug() << "// PAGE CHGD: " << page << ", ID: " << visitedPages();
127 m_pageChapters->setVobFiles(m_pageVob->isPal(), m_pageVob->selectedUrls(), m_pageVob->durations());
128 } else if (page == 2) {
129 m_pageMenu->setTargets(m_pageChapters->selectedTitles(), m_pageChapters->selectedTargets());
130 m_pageMenu->changeProfile(m_pageVob->isPal());
136 void DvdWizard::generateDvd()
138 KTemporaryFile temp1;
139 temp1.setSuffix(".png");
140 //temp1.setAutoRemove(false);
143 KTemporaryFile temp2;
144 temp2.setSuffix(".png");
145 //temp2.setAutoRemove(false);
148 KTemporaryFile temp3;
149 temp3.setSuffix(".png");
150 //temp3.setAutoRemove(false);
153 KTemporaryFile temp4;
154 temp4.setSuffix(".png");
155 //temp4.setAutoRemove(false);
158 KTemporaryFile temp5;
159 temp5.setSuffix(".vob");
160 //temp5.setAutoRemove(false);
163 KTemporaryFile temp6;
164 temp6.setSuffix(".xml");
165 //temp6.setAutoRemove(false);
168 m_menuFile.setSuffix(".mpg");
169 m_menuFile.setAutoRemove(false);
172 m_authorFile.setSuffix(".xml");
173 m_authorFile.setAutoRemove(false);
176 QListWidgetItem *images = m_status.job_progress->item(0);
177 images->setIcon(KIcon("system-run"));
178 qApp->processEvents();
179 QMap <QString, QRect> buttons = m_pageMenu->buttonsInfo();
180 QStringList buttonsTarget;
182 if (m_pageMenu->createMenu()) {
183 m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
184 m_pageMenu->createBackgroundImage(temp4.fileName());
187 images->setIcon(KIcon("dialog-ok"));
189 kDebug() << "/// STARTING MLT VOB CREATION";
190 if (!m_pageMenu->menuMovie()) {
191 // create menu vob file
192 QListWidgetItem *vobitem = m_status.job_progress->item(1);
193 vobitem->setIcon(KIcon("system-run"));
194 qApp->processEvents();
197 args.append("-profile");
198 if (m_pageMenu->isPalMenu()) args.append("dv_pal");
199 else args.append("dv_ntsc");
200 args.append(temp4.fileName());
202 args.append("out=100");
203 args << "-consumer" << "avformat:" + temp5.fileName();
204 if (m_pageMenu->isPalMenu()) {
205 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";
207 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";
210 kDebug() << "MLT ARGS: " << args;
212 renderbg.start(KdenliveSettings::rendererpath(), args);
213 if (renderbg.waitForFinished()) {
214 if (renderbg.exitStatus() == QProcess::CrashExit) {
215 kDebug() << "/// RENDERING MENU vob crashed";
216 QByteArray result = renderbg.readAllStandardError();
217 vobitem->setIcon(KIcon("dialog-close"));
218 m_status.error_log->setText(result);
219 m_status.error_box->setHidden(false);
223 kDebug() << "/// RENDERING MENU vob timed out";
224 vobitem->setIcon(KIcon("dialog-close"));
225 m_status.error_log->setText(i18n("Rendering job timed out"));
226 m_status.error_box->setHidden(false);
229 vobitem->setIcon(KIcon("dialog-ok"));
231 kDebug() << "/// STARTING SPUMUX";
233 // create xml spumux file
234 QListWidgetItem *spuitem = m_status.job_progress->item(2);
235 spuitem->setIcon(KIcon("system-run"));
236 qApp->processEvents();
238 QDomElement sub = doc.createElement("subpictures");
239 doc.appendChild(sub);
240 QDomElement stream = doc.createElement("stream");
241 sub.appendChild(stream);
242 QDomElement spu = doc.createElement("spu");
243 stream.appendChild(spu);
244 spu.setAttribute("force", "yes");
245 spu.setAttribute("start", "00:00:00.00");
246 spu.setAttribute("image", temp1.fileName());
247 spu.setAttribute("select", temp2.fileName());
248 spu.setAttribute("highlight", temp3.fileName());
249 /*spu.setAttribute("autooutline", "infer");
250 spu.setAttribute("outlinewidth", "12");
251 spu.setAttribute("autoorder", "rows");*/
253 int max = buttons.count() - 1;
255 QMapIterator<QString, QRect> it(buttons);
256 while (it.hasNext()) {
258 QDomElement but = doc.createElement("button");
259 but.setAttribute("name", 'b' + QString::number(i));
260 if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
261 else but.setAttribute("down", "b0");
262 if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
263 else but.setAttribute("up", 'b' + QString::number(max));
264 QRect r = it.value();
265 //int target = it.key();
266 // TODO: solve play all button
267 //if (target == 0) target = 1;
268 buttonsTarget.append(it.key());
269 but.setAttribute("x0", QString::number(r.x()));
270 but.setAttribute("y0", QString::number(r.y()));
271 but.setAttribute("x1", QString::number(r.right()));
272 but.setAttribute("y1", QString::number(r.bottom()));
273 spu.appendChild(but);
277 QFile data(temp6.fileName());
278 if (data.open(QFile::WriteOnly)) {
279 data.write(doc.toString().toUtf8());
283 kDebug() << " SPUMUX DATA: " << doc.toString();
286 args.append(temp6.fileName());
287 kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
291 if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
292 else spumux.setStandardInputFile(temp5.fileName());
293 spumux.setStandardOutputFile(m_menuFile.fileName());
294 spumux.start("spumux", args);
295 spumux.setProcessChannelMode(QProcess::MergedChannels);
296 if (spumux.waitForFinished()) {
297 kDebug() << QString(spumux.readAll()).simplified();
298 if (spumux.exitStatus() == QProcess::CrashExit) {
299 QByteArray result = spumux.readAllStandardError();
300 spuitem->setIcon(KIcon("dialog-close"));
301 m_status.error_log->setText(result);
302 m_status.error_box->setHidden(false);
303 kDebug() << "/// RENDERING SPUMUX MENU crashed";
307 kDebug() << "/// RENDERING SPUMUX MENU timed out";
308 spuitem->setIcon(KIcon("dialog-close"));
309 m_status.error_log->setText(i18n("Menu job timed out"));
310 m_status.error_box->setHidden(false);
314 spuitem->setIcon(KIcon("dialog-ok"));
315 kDebug() << "/// DONE: " << m_menuFile.fileName();
318 // create dvdauthor xml
319 QListWidgetItem *authitem = m_status.job_progress->item(3);
320 authitem->setIcon(KIcon("system-run"));
321 qApp->processEvents();
322 KIO::NetAccess::mkdir(KUrl(m_status.tmp_folder->url().path() + "/DVD"), this);
325 QDomElement auth = dvddoc.createElement("dvdauthor");
326 auth.setAttribute("dest", m_status.tmp_folder->url().path() + "/DVD");
327 dvddoc.appendChild(auth);
328 QDomElement vmgm = dvddoc.createElement("vmgm");
329 auth.appendChild(vmgm);
331 if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
333 QDomElement menus = dvddoc.createElement("menus");
334 vmgm.appendChild(menus);
335 QDomElement pgc = dvddoc.createElement("pgc");
336 pgc.setAttribute("entry", "title");
337 menus.appendChild(pgc);
338 QDomElement menuvob = dvddoc.createElement("vob");
339 menuvob.setAttribute("file", m_pageVob->introMovie());
340 pgc.appendChild(menuvob);
341 QDomElement post = dvddoc.createElement("post");
342 QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
343 post.appendChild(call);
344 pgc.appendChild(post);
346 QDomElement titleset = dvddoc.createElement("titleset");
347 auth.appendChild(titleset);
349 if (m_pageMenu->createMenu()) {
352 QDomElement menus = dvddoc.createElement("menus");
353 titleset.appendChild(menus);
354 QDomElement pgc = dvddoc.createElement("pgc");
355 pgc.setAttribute("entry", "root");
356 menus.appendChild(pgc);
357 QDomElement pre = dvddoc.createElement("pre");
358 pgc.appendChild(pre);
359 QDomText nametext = dvddoc.createTextNode("{g1 = 0;}");
360 pre.appendChild(nametext);
361 for (int i = 0; i < buttons.count(); i++) {
362 QDomElement button = dvddoc.createElement("button");
363 button.setAttribute("name", 'b' + QString::number(i));
364 nametext = dvddoc.createTextNode('{' + buttonsTarget.at(i) + ";}");
365 button.appendChild(nametext);
366 pgc.appendChild(button);
368 QDomElement menuvob = dvddoc.createElement("vob");
369 menuvob.setAttribute("file", m_menuFile.fileName());
370 menuvob.setAttribute("pause", "inf");
371 pgc.appendChild(menuvob);
374 QDomElement titles = dvddoc.createElement("titles");
375 titleset.appendChild(titles);
376 QDomElement video = dvddoc.createElement("video");
377 titles.appendChild(video);
378 if (m_pageVob->isPal()) video.setAttribute("format", "pal");
379 else video.setAttribute("format", "ntsc");
381 if (m_pageVob->isWide()) video.setAttribute("aspect", "16:9");
382 else video.setAttribute("aspect", "4:3");
384 QStringList voburls = m_pageVob->selectedUrls();
389 for (int i = 0; i < voburls.count(); i++) {
390 if (!voburls.at(i).isEmpty()) {
392 pgc2 = dvddoc.createElement("pgc");
393 pgc2.setAttribute("pause", 0);
394 titles.appendChild(pgc2);
395 QDomElement vob = dvddoc.createElement("vob");
396 vob.setAttribute("file", voburls.at(i));
398 QStringList chaptersList = m_pageChapters->chapters(i);
399 if (!chaptersList.isEmpty()) vob.setAttribute("chapters", chaptersList.join(","));
401 pgc2.appendChild(vob);
402 if (m_pageMenu->createMenu()) {
403 QDomElement post = dvddoc.createElement("post");
405 if (i == voburls.count() - 1) call = dvddoc.createTextNode("{g1 = 0; call menu;}");
407 call = dvddoc.createTextNode("{if ( g1 eq 999 ) { call menu; } jump title " + QString::number(i + 2) + ";}");
409 post.appendChild(call);
410 pgc2.appendChild(post);
416 QFile data2(m_authorFile.fileName());
417 if (data2.open(QFile::WriteOnly)) {
418 data2.write(dvddoc.toString().toUtf8());
421 /*kDebug() << "------------------";
422 kDebug() << dvddoc.toString();
423 kDebug() << "------------------";*/
426 args << "-x" << m_authorFile.fileName();
427 kDebug() << "// DVDAUTH ARGS: " << args;
429 m_dvdauthor->blockSignals(true);
430 m_dvdauthor->close();
434 m_creationLog.clear();
435 m_dvdauthor = new QProcess(this);
436 connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
437 m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
438 m_dvdauthor->start("dvdauthor", args);
439 m_status.button_abort->setEnabled(true);
442 void DvdWizard::slotRenderFinished(int exitCode, QProcess::ExitStatus status)
444 QListWidgetItem *authitem = m_status.job_progress->item(3);
445 if (status == QProcess::CrashExit || exitCode != 0) {
446 QByteArray result = m_dvdauthor->readAllStandardError();
447 m_status.error_log->setText(result);
448 m_status.error_box->setHidden(false);
449 kDebug() << "DVDAuthor process crashed";
450 authitem->setIcon(KIcon("dialog-close"));
451 m_dvdauthor->close();
454 m_status.button_start->setEnabled(true);
455 m_status.button_abort->setEnabled(false);
459 m_creationLog.append(m_dvdauthor->readAllStandardError());
460 m_dvdauthor->close();
464 // Check if DVD structure has the necessary infos
465 if (!QFile::exists(m_status.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 m_status.button_start->setEnabled(true);
471 m_status.button_abort->setEnabled(false);
475 authitem->setIcon(KIcon("dialog-ok"));
476 qApp->processEvents();
478 args << "-dvd-video" << "-v" << "-o" << m_status.iso_image->url().path() << m_status.tmp_folder->url().path() + "/DVD";
481 m_mkiso->blockSignals(true);
486 m_mkiso = new QProcess(this);
487 connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
488 m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
489 QListWidgetItem *isoitem = m_status.job_progress->item(4);
490 isoitem->setIcon(KIcon("system-run"));
491 m_mkiso->start("mkisofs", args);
495 void DvdWizard::slotIsoFinished(int exitCode, QProcess::ExitStatus status)
497 QListWidgetItem *isoitem = m_status.job_progress->item(4);
498 if (status == QProcess::CrashExit || exitCode != 0) {
499 QByteArray result = m_mkiso->readAllStandardError();
500 m_status.error_log->setText(result);
501 m_status.error_box->setHidden(false);
506 kDebug() << "Iso process crashed";
507 isoitem->setIcon(KIcon("dialog-close"));
508 m_status.button_start->setEnabled(true);
509 m_status.button_abort->setEnabled(false);
513 m_creationLog.append(m_mkiso->readAllStandardError());
516 m_status.button_start->setEnabled(true);
517 m_status.button_abort->setEnabled(false);
519 // Check if DVD iso is ok
520 QFile iso(m_status.iso_image->url().path());
521 if (!iso.exists() || iso.size() == 0) {
523 KIO::NetAccess::del(m_status.iso_image->url(), this);
525 m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD ISO is broken"));
526 m_status.error_box->setHidden(false);
527 isoitem->setIcon(KIcon("dialog-close"));
532 isoitem->setIcon(KIcon("dialog-ok"));
533 kDebug() << "ISO IMAGE " << m_status.iso_image->url().path() << " Successfully created";
535 kDebug() << m_creationLog;
537 m_status.error_log->setText(i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()));
538 m_status.button_preview->setEnabled(true);
539 m_status.button_burn->setEnabled(true);
540 m_status.error_box->setHidden(false);
541 //KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_status.iso_image->url().path()));
546 void DvdWizard::cleanup()
548 m_authorFile.remove();
550 KIO::NetAccess::del(KUrl(m_status.tmp_folder->url().path() + "/DVD"), this);
554 void DvdWizard::slotPreview()
556 QString programName("xine");
557 QString exec = KStandardDirs::findExe(programName);
558 if (exec.isEmpty()) KMessageBox::sorry(this, i18n("You need program <b>%1</b> to perform this action", programName));
559 else QProcess::startDetached(exec, QStringList() << "dvd://" + m_status.iso_image->url().path());
562 void DvdWizard::slotBurn()
564 QAction *action = qobject_cast<QAction *>(sender());
565 QString exec = action->data().toString();
567 if (exec.endsWith("k3b")) args << "--image" << m_status.iso_image->url().path();
568 else args << "--image=" + m_status.iso_image->url().path();
569 QProcess::startDetached(exec, args);
573 void DvdWizard::slotGenerate()
576 if ((m_dvdauthor && m_dvdauthor->state() != QProcess::NotRunning) || (m_mkiso && m_mkiso->state() != QProcess::NotRunning)) return;
577 for (int i = 0; i < m_status.job_progress->count(); i++)
578 m_status.job_progress->item(i)->setIcon(KIcon());
580 if (KIO::NetAccess::exists(KUrl(m_status.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
581 warnMessage.append(i18n("Folder %1 already exists. Overwrite?" + '\n', m_status.tmp_folder->url().path() + "/DVD"));
582 if (KIO::NetAccess::exists(KUrl(m_status.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
583 warnMessage.append(i18n("Image file %1 already exists. Overwrite?", m_status.iso_image->url().path()));
585 if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::Yes) {
586 KIO::NetAccess::del(KUrl(m_status.tmp_folder->url().path() + "/DVD"), this);
587 QTimer::singleShot(300, this, SLOT(generateDvd()));
588 m_status.button_preview->setEnabled(false);
589 m_status.button_burn->setEnabled(false);
590 m_status.job_progress->setEnabled(true);
591 m_status.button_start->setEnabled(false);
595 void DvdWizard::slotAbort()
598 if (m_dvdauthor && m_dvdauthor->state() != QProcess::NotRunning) m_dvdauthor->terminate();
599 else if (m_mkiso && m_mkiso->state() != QProcess::NotRunning) m_mkiso->terminate();
602 void DvdWizard::slotSave()
604 KMessageBox::sorry(this, "Not implemented yet");
606 KUrl url = KFileDialog::getSaveUrl(KUrl(), "*.kdvd", this, i18n("Save DVD Project"));
607 if (url.isEmpty()) return;
609 QDomElement dvdproject = doc.createElement("dvdproject");
611 if (m_pageVob->isPal()) profile = "dv_pal";
612 else profile = "dv_ntsc";
613 if (m_pageVob->isWide()) profile.append("_wide");
614 dvdproject.setAttribute("profile", profile);
615 doc.appendChild(dvdproject);
616 //for (int i = 0; i <