git.sesse.net Git - kdenlive/blob - 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 #include <QLabel>
  21 #include <QFile>
  22 #include <QXmlStreamWriter>
  23 #include <QApplication>
  24 #include <QTimer>
  25 #include <QGraphicsView>
  26 #include <QDomDocument>
  27
  28 #include <KStandardDirs>
  29 #include <KLocale>
  30 #include <KFileDialog>
  31 #include <kmimetype.h>
  32 #include <KIO/NetAccess>
  33 #include <KMessageBox>
  34
  35 #include "kdenlivesettings.h"
  36 #include "profilesdialog.h"
  37 #include "dvdwizard.h"
  38
  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"));
  44     addPage(m_pageVob);
  45
  46     if (!url.isEmpty()) m_pageVob->setUrl(url);
  47
  48     m_pageMenu = new DvdWizardMenu(m_profile, this);
  49     m_pageMenu->setTitle(i18n("Create DVD Menu"));
  50     addPage(m_pageMenu);
  51
  52
  53     QWizardPage *page3 = new QWizardPage;
  54     page3->setTitle(i18n("DVD Image"));
  55     m_iso.setupUi(page3);
  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);
  60     addPage(page3);
  61
  62     QWizardPage *page4 = new QWizardPage;
  63     page4->setTitle(i18n("Creating DVD Image"));
  64     m_status.setupUi(page4);
  65     addPage(page4);
  66
  67     connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
  68
  69 //    connect(m_standard.button_all, SIGNAL(toggled(bool)), this, SLOT(slotCheckStandard()));
  70 }
  71
  72 DvdWizard::~DvdWizard() {
  73     // m_menuFile.remove();
  74 }
  75
  76
  77 void DvdWizard::slotPageChanged(int page) {
  78     kDebug() << "// PAGE CHGD: " << page;
  79     if (page == 1) {
  80         m_pageMenu->setTargets(m_pageVob->selectedUrls());
  81     } else if (page == 2) {
  82         m_pageMenu->buttonsInfo();
  83     } else if (page == 3) {
  84         KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
  85         QTimer::singleShot(300, this, SLOT(generateDvd()));
  86     }
  87 }
  88
  89
  90
  91 void DvdWizard::generateDvd() {
  92     KTemporaryFile temp1;
  93     temp1.setSuffix(".png");
  94     //temp1.setAutoRemove(false);
  95     temp1.open();
  96
  97     KTemporaryFile temp2;
  98     temp2.setSuffix(".png");
  99     //temp2.setAutoRemove(false);
 100     temp2.open();
 101
 102     KTemporaryFile temp3;
 103     temp3.setSuffix(".png");
 104     //temp3.setAutoRemove(false);
 105     temp3.open();
 106
 107     KTemporaryFile temp4;
 108     temp4.setSuffix(".png");
 109     //temp4.setAutoRemove(false);
 110     temp4.open();
 111
 112     KTemporaryFile temp5;
 113     temp5.setSuffix(".vob");
 114     //temp5.setAutoRemove(false);
 115     temp5.open();
 116
 117     KTemporaryFile temp6;
 118     temp6.setSuffix(".xml");
 119     //temp6.setAutoRemove(false);
 120     temp6.open();
 121
 122     m_menuFile.setSuffix(".mpg");
 123     m_menuFile.setAutoRemove(false);
 124     m_menuFile.open();
 125
 126     m_authorFile.setSuffix(".xml");
 127     m_authorFile.setAutoRemove(false);
 128     m_authorFile.open();
 129
 130     QListWidgetItem *images =  m_status.job_progress->item(0);
 131     images->setIcon(KIcon("system-run"));
 132     qApp->processEvents();
 133     QMap <int, QRect> buttons = m_pageMenu->buttonsInfo();
 134     QStringList buttonsTarget;
 135
 136     if (m_pageMenu->createMenu()) {
 137         m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
 138         m_pageMenu->createBackgroundImage(temp4.fileName());
 139
 140
 141         images->setIcon(KIcon("dialog-ok"));
 142
 143         kDebug() << "/// STARTING MLT VOB CREATION";
 144         if (!m_pageMenu->menuMovie()) {
 145             // create menu vob file
 146             QListWidgetItem *vobitem =  m_status.job_progress->item(1);
 147             vobitem->setIcon(KIcon("system-run"));
 148             qApp->processEvents();
 149
 150             QStringList args;
 151             args.append("-profile");
 152             if (m_pageMenu->isPalMenu()) args.append("dv_pal");
 153             else  args.append("dv_ntsc");
 154             args.append(temp4.fileName());
 155             args.append("in=0");
 156             args.append("out=100");
 157             args << "-consumer" << "avformat:" + temp5.fileName();
 158             if (m_pageMenu->isPalMenu()) {
 159                 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";
 160             } else {
 161                 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";
 162             }
 163
 164             kDebug() << "MLT ARGS: " << args;
 165             QProcess renderbg;
 166             renderbg.start("inigo", args);
 167             if (renderbg.waitForFinished()) {
 168                 if (renderbg.exitStatus() == QProcess::CrashExit) {
 169                     kDebug() << "/// RENDERING MENU vob crashed";
 170                     vobitem->setIcon(KIcon("dialog-close"));
 171                     return;
 172                 }
 173             } else {
 174                 kDebug() << "/// RENDERING MENU vob timed out";
 175                 vobitem->setIcon(KIcon("dialog-close"));
 176                 return;
 177             }
 178             vobitem->setIcon(KIcon("dialog-ok"));
 179         }
 180         kDebug() << "/// STARTING SPUMUX";
 181
 182         // create xml spumux file
 183         QListWidgetItem *spuitem =  m_status.job_progress->item(2);
 184         spuitem->setIcon(KIcon("system-run"));
 185         qApp->processEvents();
 186         QDomDocument doc;
 187         QDomElement sub = doc.createElement("subpictures");
 188         doc.appendChild(sub);
 189         QDomElement stream = doc.createElement("stream");
 190         sub.appendChild(stream);
 191         QDomElement spu = doc.createElement("spu");
 192         stream.appendChild(spu);
 193         spu.setAttribute("force", "yes");
 194         spu.setAttribute("start", "00:00:00.00");
 195         spu.setAttribute("image", temp1.fileName());
 196         spu.setAttribute("select", temp2.fileName());
 197         spu.setAttribute("highlight", temp3.fileName());
 198         /*spu.setAttribute("autooutline", "infer");
 199         spu.setAttribute("outlinewidth", "12");
 200         spu.setAttribute("autoorder", "rows");*/
 201
 202         int max = buttons.count() - 1;
 203         int i = 0;
 204         QMapIterator<int, QRect> it(buttons);
 205         while (it.hasNext()) {
 206             it.next();
 207             QDomElement but = doc.createElement("button");
 208             but.setAttribute("name", "b" + QString::number(i));
 209             if (i < max) but.setAttribute("down", "b" + QString::number(i + 1));
 210             else but.setAttribute("down", "b0");
 211             if (i > 0) but.setAttribute("up", "b" + QString::number(i - 1));
 212             else but.setAttribute("up", "b" + QString::number(max));
 213             QRect r = it.value();
 214             int target = it.key();
 215             // TODO: solve play all button
 216             if (target == 0) target = 1;
 217             buttonsTarget.append(QString::number(target));
 218             but.setAttribute("x0", QString::number(r.x()));
 219             but.setAttribute("y0", QString::number(r.y()));
 220             but.setAttribute("x1", QString::number(r.right()));
 221             but.setAttribute("y1", QString::number(r.bottom()));
 222             spu.appendChild(but);
 223             i++;
 224         }
 225
 226         kDebug() << "///// SPU: ";
 227         kDebug() << doc.toString();
 228
 229         QFile data(temp6.fileName());
 230         if (data.open(QFile::WriteOnly)) {
 231             QTextStream out(&data);
 232             out << doc.toString();
 233         }
 234         data.close();
 235
 236         kDebug() << " SPUMUX DATA: " << doc.toString();
 237
 238         QStringList args;
 239         args.append(temp6.fileName());
 240         kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
 241
 242         QProcess spumux;
 243
 244         if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
 245         else spumux.setStandardInputFile(temp5.fileName());
 246         spumux.setStandardOutputFile(m_menuFile.fileName());
 247         spumux.start("spumux", args);
 248         spumux.setProcessChannelMode(QProcess::MergedChannels);
 249         if (spumux.waitForFinished()) {
 250             kDebug() << QString(spumux.readAll()).simplified();
 251             if (spumux.exitStatus() == QProcess::CrashExit) {
 252                 kDebug() << "/// RENDERING SPUMUX MENU crashed";
 253                 spuitem->setIcon(KIcon("dialog-close"));
 254                 return;
 255             }
 256         } else {
 257             kDebug() << "/// RENDERING SPUMUX MENU timed out";
 258             spuitem->setIcon(KIcon("dialog-close"));
 259             return;
 260         }
 261
 262         spuitem->setIcon(KIcon("dialog-ok"));
 263         kDebug() << "/// DONE: " << m_menuFile.fileName();
 264     }
 265
 266     // create dvdauthor xml
 267     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 268     authitem->setIcon(KIcon("system-run"));
 269     qApp->processEvents();
 270     KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 271
 272     QDomDocument dvddoc;
 273     QDomElement auth = dvddoc.createElement("dvdauthor");
 274     auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
 275     dvddoc.appendChild(auth);
 276     QDomElement vmgm = dvddoc.createElement("vmgm");
 277     auth.appendChild(vmgm);
 278
 279     if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
 280         // intro movie
 281         QDomElement menus = dvddoc.createElement("menus");
 282         vmgm.appendChild(menus);
 283         QDomElement pgc = dvddoc.createElement("pgc");
 284         pgc.setAttribute("entry", "title");
 285         menus.appendChild(pgc);
 286         QDomElement menuvob = dvddoc.createElement("vob");
 287         menuvob.setAttribute("file", m_pageVob->introMovie());
 288         pgc.appendChild(menuvob);
 289         QDomElement post = dvddoc.createElement("post");
 290         QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
 291         post.appendChild(call);
 292         pgc.appendChild(post);
 293     }
 294     QDomElement titleset = dvddoc.createElement("titleset");
 295     auth.appendChild(titleset);
 296
 297     if (m_pageMenu->createMenu()) {
 298
 299         // DVD main menu
 300         QDomElement menus = dvddoc.createElement("menus");
 301         titleset.appendChild(menus);
 302         QDomElement pgc = dvddoc.createElement("pgc");
 303         pgc.setAttribute("entry", "root");
 304         menus.appendChild(pgc);
 305         for (int i = 0; i < buttons.count(); i++) {
 306             QDomElement button = dvddoc.createElement("button");
 307             button.setAttribute("name", "b" + QString::number(i));
 308             QDomText nametext = dvddoc.createTextNode("jump title " + buttonsTarget.at(i) + ";");
 309             button.appendChild(nametext);
 310             pgc.appendChild(button);
 311         }
 312         QDomElement menuvob = dvddoc.createElement("vob");
 313         menuvob.setAttribute("file", m_menuFile.fileName());
 314         menuvob.setAttribute("pause", "inf");
 315         pgc.appendChild(menuvob);
 316     }
 317
 318     QDomElement titles = dvddoc.createElement("titles");
 319     titleset.appendChild(titles);
 320
 321     QStringList voburls = m_pageVob->selectedUrls();
 322
 323     QDomElement pgc2 = dvddoc.createElement("pgc");
 324     titles.appendChild(pgc2);
 325     if (m_pageMenu->createMenu()) {
 326         QDomElement post = dvddoc.createElement("post");
 327         QDomText call = dvddoc.createTextNode("call menu;");
 328         post.appendChild(call);
 329         pgc2.appendChild(post);
 330     }
 331
 332
 333     for (int i = 0; i < voburls.count(); i++) {
 334         if (!voburls.at(i).isEmpty()) {
 335             // Add vob entry
 336             QDomElement vob = dvddoc.createElement("vob");
 337             vob.setAttribute("file", voburls.at(i));
 338             pgc2.appendChild(vob);
 339         }
 340     }
 341
 342     /*
 343         // create one pgc for each video
 344         for (int i = 0; i < voburls.count(); i++) {
 345             if (!voburls.at(i).isEmpty()) {
 346                 // Add vob entry
 347
 348          QDomElement pgc2 = dvddoc.createElement("pgc");
 349          titles.appendChild(pgc2);
 350          if (m_pageMenu->createMenu()) {
 351       QDomElement post = dvddoc.createElement("post");
 352       QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
 353       post.appendChild(call);
 354       pgc2.appendChild(post);
 355          }
 356
 357                 QDomElement vob = dvddoc.createElement("vob");
 358                 vob.setAttribute("file", voburls.at(i));
 359                 pgc2.appendChild(vob);
 360             }
 361         }
 362     */
 363     QFile data2(m_authorFile.fileName());
 364     if (data2.open(QFile::WriteOnly)) {
 365         QTextStream out(&data2);
 366         out << dvddoc.toString();
 367     }
 368     data2.close();
 369
 370
 371     QStringList args;
 372     args << "-x" << m_authorFile.fileName();
 373     kDebug() << "// DVDAUTH ARGS: " << args;
 374     QProcess *dvdauth = new QProcess(this);
 375     connect(dvdauth, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
 376     dvdauth->setProcessChannelMode(QProcess::MergedChannels);
 377     dvdauth->start("dvdauthor", args);
 378
 379 }
 380
 381 void DvdWizard::slotRenderFinished(int exitCode, QProcess::ExitStatus status) {
 382     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 383     if (status == QProcess::CrashExit) {
 384         kDebug() << "DVDAuthor process crashed";
 385         authitem->setIcon(KIcon("dialog-close"));
 386         return;
 387     }
 388     authitem->setIcon(KIcon("dialog-ok"));
 389     qApp->processEvents();
 390     QStringList args;
 391     args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
 392     QProcess *mkiso = new QProcess(this);
 393     connect(mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
 394     mkiso->setProcessChannelMode(QProcess::MergedChannels);
 395     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 396     isoitem->setIcon(KIcon("system-run"));
 397     mkiso->start("mkisofs", args);
 398
 399 }
 400
 401 void DvdWizard::slotIsoFinished(int exitCode, QProcess::ExitStatus status) {
 402     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 403     if (status == QProcess::CrashExit) {
 404         //m_authorFile.remove();
 405         //m_menuFile.remove();
 406         KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 407         kDebug() << "Iso process crashed";
 408         isoitem->setIcon(KIcon("dialog-close"));
 409         return;
 410     }
 411     isoitem->setIcon(KIcon("dialog-ok"));
 412     kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
 413     //m_authorFile.remove();
 414     //m_menuFile.remove();
 415     KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 416     KMessageBox::information(this, i18n("DVD iso image %1 successfully created.", m_iso.iso_image->url().path()));
 417
 418 }
 419
 420