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 ***************************************************************************/
22 #include <QXmlStreamWriter>
23 #include <QApplication>
25 #include <QGraphicsView>
26 #include <QDomDocument>
28 #include <KStandardDirs>
30 #include <KFileDialog>
31 #include <kmimetype.h>
32 #include <KIO/NetAccess>
33 #include <KMessageBox>
35 #include "kdenlivesettings.h"
36 #include "profilesdialog.h"
37 #include "dvdwizard.h"
39 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent): QWizard(parent), m_profile(profile) {
40 //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
41 setAttribute(Qt::WA_DeleteOnClose);
42 m_pageVob = new DvdWizardVob(this);
43 m_pageVob->setTitle(i18n("Select Files For Your DVD"));
46 if (!url.isEmpty()) m_pageVob->setUrl(url);
48 m_pageMenu = new DvdWizardMenu(m_profile, this);
49 m_pageMenu->setTitle(i18n("Create DVD Menu"));
53 QWizardPage *page3 = new QWizardPage;
54 page3->setTitle(i18n("DVD Image"));
56 m_iso.tmp_folder->setPath(KdenliveSettings::currenttmpfolder());
57 m_iso.iso_image->setPath(QDir::homePath() + "/untitled.iso");
58 m_iso.iso_image->setFilter("*.iso");
59 m_iso.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
62 QWizardPage *page4 = new QWizardPage;
63 page4->setTitle(i18n("Creating DVD Image"));
64 m_status.setupUi(page4);
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();
77 void DvdWizard::slotPageChanged(int page) {
78 kDebug() << "// PAGE CHGD: " << page;
80 m_pageMenu->setTargets(m_pageVob->selectedUrls());
81 } else if (page == 2) {
82 m_pageMenu->buttonsInfo();
83 } else if (page == 3) {
85 for (int i = 0; i < m_status.job_progress->count(); i++)
86 m_status.job_progress->item(i)->setIcon(KIcon());
88 if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
89 warnMessage.append(i18n("Folder %1 already exists. Overwrite ?<br />", m_iso.tmp_folder->url().path() + "/DVD"));
90 if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
91 warnMessage.append(i18n("Image file %1 already exists. Overwrite ?", m_iso.iso_image->url().path()));
93 if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
96 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
97 QTimer::singleShot(300, this, SLOT(generateDvd()));
104 void DvdWizard::generateDvd() {
105 KTemporaryFile temp1;
106 temp1.setSuffix(".png");
107 //temp1.setAutoRemove(false);
110 KTemporaryFile temp2;
111 temp2.setSuffix(".png");
112 //temp2.setAutoRemove(false);
115 KTemporaryFile temp3;
116 temp3.setSuffix(".png");
117 //temp3.setAutoRemove(false);
120 KTemporaryFile temp4;
121 temp4.setSuffix(".png");
122 //temp4.setAutoRemove(false);
125 KTemporaryFile temp5;
126 temp5.setSuffix(".vob");
127 //temp5.setAutoRemove(false);
130 KTemporaryFile temp6;
131 temp6.setSuffix(".xml");
132 //temp6.setAutoRemove(false);
135 m_menuFile.setSuffix(".mpg");
136 m_menuFile.setAutoRemove(false);
139 m_authorFile.setSuffix(".xml");
140 m_authorFile.setAutoRemove(false);
143 QListWidgetItem *images = m_status.job_progress->item(0);
144 images->setIcon(KIcon("system-run"));
145 qApp->processEvents();
146 QMap <int, QRect> buttons = m_pageMenu->buttonsInfo();
147 QStringList buttonsTarget;
149 if (m_pageMenu->createMenu()) {
150 m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
151 m_pageMenu->createBackgroundImage(temp4.fileName());
154 images->setIcon(KIcon("dialog-ok"));
156 kDebug() << "/// STARTING MLT VOB CREATION";
157 if (!m_pageMenu->menuMovie()) {
158 // create menu vob file
159 QListWidgetItem *vobitem = m_status.job_progress->item(1);
160 vobitem->setIcon(KIcon("system-run"));
161 qApp->processEvents();
164 args.append("-profile");
165 if (m_pageMenu->isPalMenu()) args.append("dv_pal");
166 else args.append("dv_ntsc");
167 args.append(temp4.fileName());
169 args.append("out=100");
170 args << "-consumer" << "avformat:" + temp5.fileName();
171 if (m_pageMenu->isPalMenu()) {
172 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";
174 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";
177 kDebug() << "MLT ARGS: " << args;
179 renderbg.start("inigo", args);
180 if (renderbg.waitForFinished()) {
181 if (renderbg.exitStatus() == QProcess::CrashExit) {
182 kDebug() << "/// RENDERING MENU vob crashed";
183 vobitem->setIcon(KIcon("dialog-close"));
187 kDebug() << "/// RENDERING MENU vob timed out";
188 vobitem->setIcon(KIcon("dialog-close"));
191 vobitem->setIcon(KIcon("dialog-ok"));
193 kDebug() << "/// STARTING SPUMUX";
195 // create xml spumux file
196 QListWidgetItem *spuitem = m_status.job_progress->item(2);
197 spuitem->setIcon(KIcon("system-run"));
198 qApp->processEvents();
200 QDomElement sub = doc.createElement("subpictures");
201 doc.appendChild(sub);
202 QDomElement stream = doc.createElement("stream");
203 sub.appendChild(stream);
204 QDomElement spu = doc.createElement("spu");
205 stream.appendChild(spu);
206 spu.setAttribute("force", "yes");
207 spu.setAttribute("start", "00:00:00.00");
208 spu.setAttribute("image", temp1.fileName());
209 spu.setAttribute("select", temp2.fileName());
210 spu.setAttribute("highlight", temp3.fileName());
211 /*spu.setAttribute("autooutline", "infer");
212 spu.setAttribute("outlinewidth", "12");
213 spu.setAttribute("autoorder", "rows");*/
215 int max = buttons.count() - 1;
217 QMapIterator<int, QRect> it(buttons);
218 while (it.hasNext()) {
220 QDomElement but = doc.createElement("button");
221 but.setAttribute("name", "b" + QString::number(i));
222 if (i < max) but.setAttribute("down", "b" + QString::number(i + 1));
223 else but.setAttribute("down", "b0");
224 if (i > 0) but.setAttribute("up", "b" + QString::number(i - 1));
225 else but.setAttribute("up", "b" + QString::number(max));
226 QRect r = it.value();
227 int target = it.key();
228 // TODO: solve play all button
229 if (target == 0) target = 1;
230 buttonsTarget.append(QString::number(target));
231 but.setAttribute("x0", QString::number(r.x()));
232 but.setAttribute("y0", QString::number(r.y()));
233 but.setAttribute("x1", QString::number(r.right()));
234 but.setAttribute("y1", QString::number(r.bottom()));
235 spu.appendChild(but);
239 kDebug() << "///// SPU: ";
240 kDebug() << doc.toString();
242 QFile data(temp6.fileName());
243 if (data.open(QFile::WriteOnly)) {
244 QTextStream out(&data);
245 out << doc.toString();
249 kDebug() << " SPUMUX DATA: " << doc.toString();
252 args.append(temp6.fileName());
253 kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
257 if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
258 else spumux.setStandardInputFile(temp5.fileName());
259 spumux.setStandardOutputFile(m_menuFile.fileName());
260 spumux.start("spumux", args);
261 spumux.setProcessChannelMode(QProcess::MergedChannels);
262 if (spumux.waitForFinished()) {
263 kDebug() << QString(spumux.readAll()).simplified();
264 if (spumux.exitStatus() == QProcess::CrashExit) {
265 kDebug() << "/// RENDERING SPUMUX MENU crashed";
266 spuitem->setIcon(KIcon("dialog-close"));
270 kDebug() << "/// RENDERING SPUMUX MENU timed out";
271 spuitem->setIcon(KIcon("dialog-close"));
275 spuitem->setIcon(KIcon("dialog-ok"));
276 kDebug() << "/// DONE: " << m_menuFile.fileName();
279 // create dvdauthor xml
280 QListWidgetItem *authitem = m_status.job_progress->item(3);
281 authitem->setIcon(KIcon("system-run"));
282 qApp->processEvents();
283 KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
286 QDomElement auth = dvddoc.createElement("dvdauthor");
287 auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
288 dvddoc.appendChild(auth);
289 QDomElement vmgm = dvddoc.createElement("vmgm");
290 auth.appendChild(vmgm);
292 if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
294 QDomElement menus = dvddoc.createElement("menus");
295 vmgm.appendChild(menus);
296 QDomElement pgc = dvddoc.createElement("pgc");
297 pgc.setAttribute("entry", "title");
298 menus.appendChild(pgc);
299 QDomElement menuvob = dvddoc.createElement("vob");
300 menuvob.setAttribute("file", m_pageVob->introMovie());
301 pgc.appendChild(menuvob);
302 QDomElement post = dvddoc.createElement("post");
303 QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
304 post.appendChild(call);
305 pgc.appendChild(post);
307 QDomElement titleset = dvddoc.createElement("titleset");
308 auth.appendChild(titleset);
310 if (m_pageMenu->createMenu()) {
313 QDomElement menus = dvddoc.createElement("menus");
314 titleset.appendChild(menus);
315 QDomElement pgc = dvddoc.createElement("pgc");
316 pgc.setAttribute("entry", "root");
317 menus.appendChild(pgc);
318 for (int i = 0; i < buttons.count(); i++) {
319 QDomElement button = dvddoc.createElement("button");
320 button.setAttribute("name", "b" + QString::number(i));
321 QDomText nametext = dvddoc.createTextNode("jump title " + buttonsTarget.at(i) + ";");
322 button.appendChild(nametext);
323 pgc.appendChild(button);
325 QDomElement menuvob = dvddoc.createElement("vob");
326 menuvob.setAttribute("file", m_menuFile.fileName());
327 menuvob.setAttribute("pause", "inf");
328 pgc.appendChild(menuvob);
331 QDomElement titles = dvddoc.createElement("titles");
332 titleset.appendChild(titles);
334 QStringList voburls = m_pageVob->selectedUrls();
336 QDomElement pgc2 = dvddoc.createElement("pgc");
337 titles.appendChild(pgc2);
338 if (m_pageMenu->createMenu()) {
339 QDomElement post = dvddoc.createElement("post");
340 QDomText call = dvddoc.createTextNode("call menu;");
341 post.appendChild(call);
342 pgc2.appendChild(post);
346 for (int i = 0; i < voburls.count(); i++) {
347 if (!voburls.at(i).isEmpty()) {
349 QDomElement vob = dvddoc.createElement("vob");
350 vob.setAttribute("file", voburls.at(i));
351 pgc2.appendChild(vob);
356 // create one pgc for each video
357 for (int i = 0; i < voburls.count(); i++) {
358 if (!voburls.at(i).isEmpty()) {
361 QDomElement pgc2 = dvddoc.createElement("pgc");
362 titles.appendChild(pgc2);
363 if (m_pageMenu->createMenu()) {
364 QDomElement post = dvddoc.createElement("post");
365 QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
366 post.appendChild(call);
367 pgc2.appendChild(post);
370 QDomElement vob = dvddoc.createElement("vob");
371 vob.setAttribute("file", voburls.at(i));
372 pgc2.appendChild(vob);
376 QFile data2(m_authorFile.fileName());
377 if (data2.open(QFile::WriteOnly)) {
378 QTextStream out(&data2);
379 out << dvddoc.toString();
385 args << "-x" << m_authorFile.fileName();
386 kDebug() << "// DVDAUTH ARGS: " << args;
387 QProcess *dvdauth = new QProcess(this);
388 connect(dvdauth, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
389 dvdauth->setProcessChannelMode(QProcess::MergedChannels);
390 dvdauth->start("dvdauthor", args);
394 void DvdWizard::slotRenderFinished(int exitCode, QProcess::ExitStatus status) {
395 QListWidgetItem *authitem = m_status.job_progress->item(3);
396 if (status == QProcess::CrashExit) {
397 kDebug() << "DVDAuthor process crashed";
398 authitem->setIcon(KIcon("dialog-close"));
401 authitem->setIcon(KIcon("dialog-ok"));
402 qApp->processEvents();
404 args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
405 QProcess *mkiso = new QProcess(this);
406 connect(mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
407 mkiso->setProcessChannelMode(QProcess::MergedChannels);
408 QListWidgetItem *isoitem = m_status.job_progress->item(4);
409 isoitem->setIcon(KIcon("system-run"));
410 mkiso->start("mkisofs", args);
414 void DvdWizard::slotIsoFinished(int exitCode, QProcess::ExitStatus status) {
415 QListWidgetItem *isoitem = m_status.job_progress->item(4);
416 if (status == QProcess::CrashExit) {
417 m_authorFile.remove();
419 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
420 kDebug() << "Iso process crashed";
421 isoitem->setIcon(KIcon("dialog-close"));
424 isoitem->setIcon(KIcon("dialog-ok"));
425 kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
426 m_authorFile.remove();
428 KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
429 KMessageBox::information(this, i18n("DVD iso image %1 successfully created.", m_iso.iso_image->url().path()));