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>
38 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent): QWizard(parent), m_profile(profile), m_dvdauthor(NULL), m_mkiso(NULL) {
39 //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
40 setAttribute(Qt::WA_DeleteOnClose);
41 m_pageVob = new DvdWizardVob(this);
42 m_pageVob->setTitle(i18n("Select Files For Your DVD"));
45 if (!url.isEmpty()) m_pageVob->setUrl(url);
47 m_pageMenu = new DvdWizardMenu(m_profile, this);
48 m_pageMenu->setTitle(i18n("Create DVD Menu"));
52 QWizardPage *page3 = new QWizardPage;
53 page3->setTitle(i18n("DVD Image"));
55 m_iso.tmp_folder->setPath(KdenliveSettings::currenttmpfolder());
56 m_iso.iso_image->setPath(QDir::homePath() + "/untitled.iso");
57 m_iso.iso_image->setFilter("*.iso");
58 m_iso.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
61 QWizardPage *page4 = new QWizardPage;
62 page4->setTitle(i18n("Creating DVD Image"));
63 m_status.setupUi(page4);
64 m_status.error_box->setHidden(true);
67 connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
69 // connect(m_standard.button_all, SIGNAL(toggled(bool)), this, SLOT(slotCheckStandard()));
72 DvdWizard::~DvdWizard() {
73 // m_menuFile.remove();
85 void DvdWizard::slotPageChanged(int page) {
86 kDebug() << "// PAGE CHGD: " << page;
88 m_pageMenu->setTargets(m_pageVob->selectedUrls());
89 } else if (page == 2) {
90 m_pageMenu->buttonsInfo();
91 } else if (page == 3) {
93 for (int i = 0; i < m_status.job_progress->count(); i++)
94 m_status.job_progress->item(i)->setIcon(KIcon());
96 if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
97 warnMessage.append(i18n("Folder %1 already exists. Overwrite ?" + '\n', m_iso.tmp_folder->url().path() + "/DVD"));
98 if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
99 warnMessage.append(i18n("Image file %1 already exists. Overwrite ?", m_iso.iso_image->url().path()));
101 if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
104 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
105 QTimer::singleShot(300, this, SLOT(generateDvd()));
112 void DvdWizard::generateDvd() {
113 KTemporaryFile temp1;
114 temp1.setSuffix(".png");
115 //temp1.setAutoRemove(false);
118 KTemporaryFile temp2;
119 temp2.setSuffix(".png");
120 //temp2.setAutoRemove(false);
123 KTemporaryFile temp3;
124 temp3.setSuffix(".png");
125 //temp3.setAutoRemove(false);
128 KTemporaryFile temp4;
129 temp4.setSuffix(".png");
130 //temp4.setAutoRemove(false);
133 KTemporaryFile temp5;
134 temp5.setSuffix(".vob");
135 //temp5.setAutoRemove(false);
138 KTemporaryFile temp6;
139 temp6.setSuffix(".xml");
140 //temp6.setAutoRemove(false);
143 m_menuFile.setSuffix(".mpg");
144 m_menuFile.setAutoRemove(false);
147 m_authorFile.setSuffix(".xml");
148 m_authorFile.setAutoRemove(false);
151 QListWidgetItem *images = m_status.job_progress->item(0);
152 images->setIcon(KIcon("system-run"));
153 qApp->processEvents();
154 QMap <int, QRect> buttons = m_pageMenu->buttonsInfo();
155 QStringList buttonsTarget;
157 if (m_pageMenu->createMenu()) {
158 m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
159 m_pageMenu->createBackgroundImage(temp4.fileName());
162 images->setIcon(KIcon("dialog-ok"));
164 kDebug() << "/// STARTING MLT VOB CREATION";
165 if (!m_pageMenu->menuMovie()) {
166 // create menu vob file
167 QListWidgetItem *vobitem = m_status.job_progress->item(1);
168 vobitem->setIcon(KIcon("system-run"));
169 qApp->processEvents();
172 args.append("-profile");
173 if (m_pageMenu->isPalMenu()) args.append("dv_pal");
174 else args.append("dv_ntsc");
175 args.append(temp4.fileName());
177 args.append("out=100");
178 args << "-consumer" << "avformat:" + temp5.fileName();
179 if (m_pageMenu->isPalMenu()) {
180 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";
182 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";
185 kDebug() << "MLT ARGS: " << args;
187 renderbg.start("inigo", args);
188 if (renderbg.waitForFinished()) {
189 if (renderbg.exitStatus() == QProcess::CrashExit) {
190 kDebug() << "/// RENDERING MENU vob crashed";
191 QByteArray result = renderbg.readAllStandardError();
192 vobitem->setIcon(KIcon("dialog-close"));
193 m_status.error_log->setText(result);
194 m_status.error_box->setHidden(false);
198 kDebug() << "/// RENDERING MENU vob timed out";
199 vobitem->setIcon(KIcon("dialog-close"));
200 m_status.error_log->setText(i18n("Rendering job timed out"));
201 m_status.error_box->setHidden(false);
204 vobitem->setIcon(KIcon("dialog-ok"));
206 kDebug() << "/// STARTING SPUMUX";
208 // create xml spumux file
209 QListWidgetItem *spuitem = m_status.job_progress->item(2);
210 spuitem->setIcon(KIcon("system-run"));
211 qApp->processEvents();
213 QDomElement sub = doc.createElement("subpictures");
214 doc.appendChild(sub);
215 QDomElement stream = doc.createElement("stream");
216 sub.appendChild(stream);
217 QDomElement spu = doc.createElement("spu");
218 stream.appendChild(spu);
219 spu.setAttribute("force", "yes");
220 spu.setAttribute("start", "00:00:00.00");
221 spu.setAttribute("image", temp1.fileName());
222 spu.setAttribute("select", temp2.fileName());
223 spu.setAttribute("highlight", temp3.fileName());
224 /*spu.setAttribute("autooutline", "infer");
225 spu.setAttribute("outlinewidth", "12");
226 spu.setAttribute("autoorder", "rows");*/
228 int max = buttons.count() - 1;
230 QMapIterator<int, QRect> it(buttons);
231 while (it.hasNext()) {
233 QDomElement but = doc.createElement("button");
234 but.setAttribute("name", 'b' + QString::number(i));
235 if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
236 else but.setAttribute("down", "b0");
237 if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
238 else but.setAttribute("up", 'b' + QString::number(max));
239 QRect r = it.value();
240 int target = it.key();
241 // TODO: solve play all button
242 if (target == 0) target = 1;
243 buttonsTarget.append(QString::number(target));
244 but.setAttribute("x0", QString::number(r.x()));
245 but.setAttribute("y0", QString::number(r.y()));
246 but.setAttribute("x1", QString::number(r.right()));
247 but.setAttribute("y1", QString::number(r.bottom()));
248 spu.appendChild(but);
252 kDebug() << "///// SPU: ";
253 kDebug() << doc.toString();
255 QFile data(temp6.fileName());
256 if (data.open(QFile::WriteOnly)) {
257 QTextStream out(&data);
258 out << doc.toString();
262 kDebug() << " SPUMUX DATA: " << doc.toString();
265 args.append(temp6.fileName());
266 kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
270 if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
271 else spumux.setStandardInputFile(temp5.fileName());
272 spumux.setStandardOutputFile(m_menuFile.fileName());
273 spumux.start("spumux", args);
274 spumux.setProcessChannelMode(QProcess::MergedChannels);
275 if (spumux.waitForFinished()) {
276 kDebug() << QString(spumux.readAll()).simplified();
277 if (spumux.exitStatus() == QProcess::CrashExit) {
278 QByteArray result = spumux.readAllStandardError();
279 spuitem->setIcon(KIcon("dialog-close"));
280 m_status.error_log->setText(result);
281 m_status.error_box->setHidden(false);
282 kDebug() << "/// RENDERING SPUMUX MENU crashed";
286 kDebug() << "/// RENDERING SPUMUX MENU timed out";
287 spuitem->setIcon(KIcon("dialog-close"));
288 m_status.error_log->setText(i18n("Menu job timed out"));
289 m_status.error_box->setHidden(false);
293 spuitem->setIcon(KIcon("dialog-ok"));
294 kDebug() << "/// DONE: " << m_menuFile.fileName();
297 // create dvdauthor xml
298 QListWidgetItem *authitem = m_status.job_progress->item(3);
299 authitem->setIcon(KIcon("system-run"));
300 qApp->processEvents();
301 KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
304 QDomElement auth = dvddoc.createElement("dvdauthor");
305 auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
306 dvddoc.appendChild(auth);
307 QDomElement vmgm = dvddoc.createElement("vmgm");
308 auth.appendChild(vmgm);
310 if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
312 QDomElement menus = dvddoc.createElement("menus");
313 vmgm.appendChild(menus);
314 QDomElement pgc = dvddoc.createElement("pgc");
315 pgc.setAttribute("entry", "title");
316 menus.appendChild(pgc);
317 QDomElement menuvob = dvddoc.createElement("vob");
318 menuvob.setAttribute("file", m_pageVob->introMovie());
319 pgc.appendChild(menuvob);
320 QDomElement post = dvddoc.createElement("post");
321 QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
322 post.appendChild(call);
323 pgc.appendChild(post);
325 QDomElement titleset = dvddoc.createElement("titleset");
326 auth.appendChild(titleset);
328 if (m_pageMenu->createMenu()) {
331 QDomElement menus = dvddoc.createElement("menus");
332 titleset.appendChild(menus);
333 QDomElement pgc = dvddoc.createElement("pgc");
334 pgc.setAttribute("entry", "root");
335 menus.appendChild(pgc);
336 for (int i = 0; i < buttons.count(); i++) {
337 QDomElement button = dvddoc.createElement("button");
338 button.setAttribute("name", 'b' + QString::number(i));
339 QDomText nametext = dvddoc.createTextNode("jump title " + buttonsTarget.at(i) + ';');
340 button.appendChild(nametext);
341 pgc.appendChild(button);
343 QDomElement menuvob = dvddoc.createElement("vob");
344 menuvob.setAttribute("file", m_menuFile.fileName());
345 menuvob.setAttribute("pause", "inf");
346 pgc.appendChild(menuvob);
349 QDomElement titles = dvddoc.createElement("titles");
350 titleset.appendChild(titles);
352 QStringList voburls = m_pageVob->selectedUrls();
354 QDomElement pgc2 = dvddoc.createElement("pgc");
355 titles.appendChild(pgc2);
356 if (m_pageMenu->createMenu()) {
357 QDomElement post = dvddoc.createElement("post");
358 QDomText call = dvddoc.createTextNode("call menu;");
359 post.appendChild(call);
360 pgc2.appendChild(post);
364 for (int i = 0; i < voburls.count(); i++) {
365 if (!voburls.at(i).isEmpty()) {
367 QDomElement vob = dvddoc.createElement("vob");
368 vob.setAttribute("file", voburls.at(i));
369 pgc2.appendChild(vob);
374 // create one pgc for each video
375 for (int i = 0; i < voburls.count(); i++) {
376 if (!voburls.at(i).isEmpty()) {
379 QDomElement pgc2 = dvddoc.createElement("pgc");
380 titles.appendChild(pgc2);
381 if (m_pageMenu->createMenu()) {
382 QDomElement post = dvddoc.createElement("post");
383 QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
384 post.appendChild(call);
385 pgc2.appendChild(post);
388 QDomElement vob = dvddoc.createElement("vob");
389 vob.setAttribute("file", voburls.at(i));
390 pgc2.appendChild(vob);
394 QFile data2(m_authorFile.fileName());
395 if (data2.open(QFile::WriteOnly)) {
396 QTextStream out(&data2);
397 out << dvddoc.toString();
400 /*kDebug() << "------------------";
401 kDebug() << dvddoc.toString();
402 kDebug() << "------------------";*/
405 args << "-x" << m_authorFile.fileName();
406 kDebug() << "// DVDAUTH ARGS: " << args;
408 m_dvdauthor->close();
412 m_creationLog.clear();
413 m_dvdauthor = new QProcess(this);
414 connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
415 m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
416 m_dvdauthor->start("dvdauthor", args);
420 void DvdWizard::slotRenderFinished(int /*exitCode*/, QProcess::ExitStatus status) {
421 QListWidgetItem *authitem = m_status.job_progress->item(3);
422 if (status == QProcess::CrashExit) {
423 QByteArray result = m_dvdauthor->readAllStandardError();
424 m_status.error_log->setText(result);
425 m_status.error_box->setHidden(false);
426 kDebug() << "DVDAuthor process crashed";
427 authitem->setIcon(KIcon("dialog-close"));
428 m_dvdauthor->close();
434 m_creationLog.append(m_dvdauthor->readAllStandardError());
435 m_dvdauthor->close();
439 // Check if DVD structure has the necessary infos
440 if (!QFile::exists(m_iso.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
441 m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD structure broken"));
442 m_status.error_box->setHidden(false);
443 kDebug() << "DVDAuthor process crashed";
444 authitem->setIcon(KIcon("dialog-close"));
448 authitem->setIcon(KIcon("dialog-ok"));
449 qApp->processEvents();
451 args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
458 m_mkiso = new QProcess(this);
459 connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
460 m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
461 QListWidgetItem *isoitem = m_status.job_progress->item(4);
462 isoitem->setIcon(KIcon("system-run"));
463 m_mkiso->start("mkisofs", args);
467 void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status) {
468 QListWidgetItem *isoitem = m_status.job_progress->item(4);
469 if (status == QProcess::CrashExit) {
470 QByteArray result = m_mkiso->readAllStandardError();
471 m_status.error_log->setText(result);
472 m_status.error_box->setHidden(false);
477 kDebug() << "Iso process crashed";
478 isoitem->setIcon(KIcon("dialog-close"));
482 m_creationLog.append(m_mkiso->readAllStandardError());
486 // Check if DVD iso is ok
487 QFile iso(m_iso.iso_image->url().path());
488 if (!iso.exists() || iso.size() == 0) {
490 KIO::NetAccess::del(m_iso.iso_image->url(), this);
492 m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD ISO is broken"));
493 m_status.error_box->setHidden(false);
494 isoitem->setIcon(KIcon("dialog-close"));
499 isoitem->setIcon(KIcon("dialog-ok"));
500 kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
502 kDebug() << m_creationLog;
503 KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
508 void DvdWizard::cleanup() {
509 m_authorFile.remove();
511 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);