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
  21 #include "dvdwizard.h"
  22 #include "kdenlivesettings.h"
  23 #include "profilesdialog.h"
  24
  25 #include <KStandardDirs>
  26 #include <KLocale>
  27 #include <KFileDialog>
  28 #include <kmimetype.h>
  29 #include <KIO/NetAccess>
  30 #include <KMessageBox>
  31
  32 #include <QFile>
  33 #include <QApplication>
  34 #include <QTimer>
  35 #include <QDomDocument>
  36
  37
  38 DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent) :
  39         QWizard(parent),
  40         m_dvdauthor(NULL),
  41         m_mkiso(NULL)
  42 {
  43     //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
  44     setAttribute(Qt::WA_DeleteOnClose);
  45     m_pageVob = new DvdWizardVob(this);
  46     m_pageVob->setTitle(i18n("Select Files For Your DVD"));
  47     addPage(m_pageVob);
  48
  49     if (!url.isEmpty()) m_pageVob->setUrl(url);
  50
  51     m_pageMenu = new DvdWizardMenu(profile, this);
  52     m_pageMenu->setTitle(i18n("Create DVD Menu"));
  53     addPage(m_pageMenu);
  54
  55
  56     QWizardPage *page3 = new QWizardPage;
  57     page3->setTitle(i18n("DVD Image"));
  58     m_iso.setupUi(page3);
  59     m_iso.tmp_folder->setPath(KdenliveSettings::currenttmpfolder());
  60     m_iso.iso_image->setPath(QDir::homePath() + "/untitled.iso");
  61     m_iso.iso_image->setFilter("*.iso");
  62     m_iso.iso_image->fileDialog()->setOperationMode(KFileDialog::Saving);
  63     addPage(page3);
  64
  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     addPage(page4);
  70
  71     connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
  72
  73 //    connect(m_standard.button_all, SIGNAL(toggled(bool)), this, SLOT(slotCheckStandard()));
  74 }
  75
  76 DvdWizard::~DvdWizard()
  77 {
  78     // m_menuFile.remove();
  79     if (m_dvdauthor) {
  80         m_dvdauthor->close();
  81         delete m_dvdauthor;
  82     }
  83     if (m_mkiso) {
  84         m_mkiso->close();
  85         delete m_mkiso;
  86     }
  87 }
  88
  89
  90 void DvdWizard::slotPageChanged(int page)
  91 {
  92     kDebug() << "// PAGE CHGD: " << page;
  93     if (page == 1) {
  94         m_pageMenu->setTargets(m_pageVob->selectedUrls());
  95     } else if (page == 2) {
  96         m_pageMenu->buttonsInfo();
  97     } else if (page == 3) {
  98         // clear job icons
  99         for (int i = 0; i < m_status.job_progress->count(); i++)
 100             m_status.job_progress->item(i)->setIcon(KIcon());
 101         QString warnMessage;
 102         if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
 103             warnMessage.append(i18n("Folder %1 already exists. Overwrite ?" + '\n', m_iso.tmp_folder->url().path() + "/DVD"));
 104         if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
 105             warnMessage.append(i18n("Image file %1 already exists. Overwrite ?", m_iso.iso_image->url().path()));
 106
 107         if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
 108             back();
 109         } else {
 110             KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 111             QTimer::singleShot(300, this, SLOT(generateDvd()));
 112         }
 113     }
 114 }
 115
 116
 117
 118 void DvdWizard::generateDvd()
 119 {
 120     KTemporaryFile temp1;
 121     temp1.setSuffix(".png");
 122     //temp1.setAutoRemove(false);
 123     temp1.open();
 124
 125     KTemporaryFile temp2;
 126     temp2.setSuffix(".png");
 127     //temp2.setAutoRemove(false);
 128     temp2.open();
 129
 130     KTemporaryFile temp3;
 131     temp3.setSuffix(".png");
 132     //temp3.setAutoRemove(false);
 133     temp3.open();
 134
 135     KTemporaryFile temp4;
 136     temp4.setSuffix(".png");
 137     //temp4.setAutoRemove(false);
 138     temp4.open();
 139
 140     KTemporaryFile temp5;
 141     temp5.setSuffix(".vob");
 142     //temp5.setAutoRemove(false);
 143     temp5.open();
 144
 145     KTemporaryFile temp6;
 146     temp6.setSuffix(".xml");
 147     //temp6.setAutoRemove(false);
 148     temp6.open();
 149
 150     m_menuFile.setSuffix(".mpg");
 151     m_menuFile.setAutoRemove(false);
 152     m_menuFile.open();
 153
 154     m_authorFile.setSuffix(".xml");
 155     m_authorFile.setAutoRemove(false);
 156     m_authorFile.open();
 157
 158     QListWidgetItem *images =  m_status.job_progress->item(0);
 159     images->setIcon(KIcon("system-run"));
 160     qApp->processEvents();
 161     QMap <int, QRect> buttons = m_pageMenu->buttonsInfo();
 162     QStringList buttonsTarget;
 163
 164     if (m_pageMenu->createMenu()) {
 165         m_pageMenu->createButtonImages(temp1.fileName(), temp2.fileName(), temp3.fileName());
 166         m_pageMenu->createBackgroundImage(temp4.fileName());
 167
 168
 169         images->setIcon(KIcon("dialog-ok"));
 170
 171         kDebug() << "/// STARTING MLT VOB CREATION";
 172         if (!m_pageMenu->menuMovie()) {
 173             // create menu vob file
 174             QListWidgetItem *vobitem =  m_status.job_progress->item(1);
 175             vobitem->setIcon(KIcon("system-run"));
 176             qApp->processEvents();
 177
 178             QStringList args;
 179             args.append("-profile");
 180             if (m_pageMenu->isPalMenu()) args.append("dv_pal");
 181             else  args.append("dv_ntsc");
 182             args.append(temp4.fileName());
 183             args.append("in=0");
 184             args.append("out=100");
 185             args << "-consumer" << "avformat:" + temp5.fileName();
 186             if (m_pageMenu->isPalMenu()) {
 187                 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";
 188             } else {
 189                 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";
 190             }
 191
 192             kDebug() << "MLT ARGS: " << args;
 193             QProcess renderbg;
 194             renderbg.start("inigo", args);
 195             if (renderbg.waitForFinished()) {
 196                 if (renderbg.exitStatus() == QProcess::CrashExit) {
 197                     kDebug() << "/// RENDERING MENU vob crashed";
 198                     QByteArray result = renderbg.readAllStandardError();
 199                     vobitem->setIcon(KIcon("dialog-close"));
 200                     m_status.error_log->setText(result);
 201                     m_status.error_box->setHidden(false);
 202                     return;
 203                 }
 204             } else {
 205                 kDebug() << "/// RENDERING MENU vob timed out";
 206                 vobitem->setIcon(KIcon("dialog-close"));
 207                 m_status.error_log->setText(i18n("Rendering job timed out"));
 208                 m_status.error_box->setHidden(false);
 209                 return;
 210             }
 211             vobitem->setIcon(KIcon("dialog-ok"));
 212         }
 213         kDebug() << "/// STARTING SPUMUX";
 214
 215         // create xml spumux file
 216         QListWidgetItem *spuitem =  m_status.job_progress->item(2);
 217         spuitem->setIcon(KIcon("system-run"));
 218         qApp->processEvents();
 219         QDomDocument doc;
 220         QDomElement sub = doc.createElement("subpictures");
 221         doc.appendChild(sub);
 222         QDomElement stream = doc.createElement("stream");
 223         sub.appendChild(stream);
 224         QDomElement spu = doc.createElement("spu");
 225         stream.appendChild(spu);
 226         spu.setAttribute("force", "yes");
 227         spu.setAttribute("start", "00:00:00.00");
 228         spu.setAttribute("image", temp1.fileName());
 229         spu.setAttribute("select", temp2.fileName());
 230         spu.setAttribute("highlight", temp3.fileName());
 231         /*spu.setAttribute("autooutline", "infer");
 232         spu.setAttribute("outlinewidth", "12");
 233         spu.setAttribute("autoorder", "rows");*/
 234
 235         int max = buttons.count() - 1;
 236         int i = 0;
 237         QMapIterator<int, QRect> it(buttons);
 238         while (it.hasNext()) {
 239             it.next();
 240             QDomElement but = doc.createElement("button");
 241             but.setAttribute("name", 'b' + QString::number(i));
 242             if (i < max) but.setAttribute("down", 'b' + QString::number(i + 1));
 243             else but.setAttribute("down", "b0");
 244             if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
 245             else but.setAttribute("up", 'b' + QString::number(max));
 246             QRect r = it.value();
 247             int target = it.key();
 248             // TODO: solve play all button
 249             if (target == 0) target = 1;
 250             buttonsTarget.append(QString::number(target));
 251             but.setAttribute("x0", QString::number(r.x()));
 252             but.setAttribute("y0", QString::number(r.y()));
 253             but.setAttribute("x1", QString::number(r.right()));
 254             but.setAttribute("y1", QString::number(r.bottom()));
 255             spu.appendChild(but);
 256             i++;
 257         }
 258
 259         QFile data(temp6.fileName());
 260         if (data.open(QFile::WriteOnly)) {
 261             data.write(doc.toString().toUtf8());
 262         }
 263         data.close();
 264
 265         kDebug() << " SPUMUX DATA: " << doc.toString();
 266
 267         QStringList args;
 268         args.append(temp6.fileName());
 269         kDebug() << "SPM ARGS: " << args << temp5.fileName() << m_menuFile.fileName();
 270
 271         QProcess spumux;
 272
 273         if (m_pageMenu->menuMovie()) spumux.setStandardInputFile(m_pageMenu->menuMoviePath());
 274         else spumux.setStandardInputFile(temp5.fileName());
 275         spumux.setStandardOutputFile(m_menuFile.fileName());
 276         spumux.start("spumux", args);
 277         spumux.setProcessChannelMode(QProcess::MergedChannels);
 278         if (spumux.waitForFinished()) {
 279             kDebug() << QString(spumux.readAll()).simplified();
 280             if (spumux.exitStatus() == QProcess::CrashExit) {
 281                 QByteArray result = spumux.readAllStandardError();
 282                 spuitem->setIcon(KIcon("dialog-close"));
 283                 m_status.error_log->setText(result);
 284                 m_status.error_box->setHidden(false);
 285                 kDebug() << "/// RENDERING SPUMUX MENU crashed";
 286                 return;
 287             }
 288         } else {
 289             kDebug() << "/// RENDERING SPUMUX MENU timed out";
 290             spuitem->setIcon(KIcon("dialog-close"));
 291             m_status.error_log->setText(i18n("Menu job timed out"));
 292             m_status.error_box->setHidden(false);
 293             return;
 294         }
 295
 296         spuitem->setIcon(KIcon("dialog-ok"));
 297         kDebug() << "/// DONE: " << m_menuFile.fileName();
 298     }
 299
 300     // create dvdauthor xml
 301     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 302     authitem->setIcon(KIcon("system-run"));
 303     qApp->processEvents();
 304     KIO::NetAccess::mkdir(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 305
 306     QDomDocument dvddoc;
 307     QDomElement auth = dvddoc.createElement("dvdauthor");
 308     auth.setAttribute("dest", m_iso.tmp_folder->url().path() + "/DVD");
 309     dvddoc.appendChild(auth);
 310     QDomElement vmgm = dvddoc.createElement("vmgm");
 311     auth.appendChild(vmgm);
 312
 313     if (m_pageMenu->createMenu() && !m_pageVob->introMovie().isEmpty()) {
 314         // intro movie
 315         QDomElement menus = dvddoc.createElement("menus");
 316         vmgm.appendChild(menus);
 317         QDomElement pgc = dvddoc.createElement("pgc");
 318         pgc.setAttribute("entry", "title");
 319         menus.appendChild(pgc);
 320         QDomElement menuvob = dvddoc.createElement("vob");
 321         menuvob.setAttribute("file", m_pageVob->introMovie());
 322         pgc.appendChild(menuvob);
 323         QDomElement post = dvddoc.createElement("post");
 324         QDomText call = dvddoc.createTextNode("jump titleset 1 menu;");
 325         post.appendChild(call);
 326         pgc.appendChild(post);
 327     }
 328     QDomElement titleset = dvddoc.createElement("titleset");
 329     auth.appendChild(titleset);
 330
 331     if (m_pageMenu->createMenu()) {
 332
 333         // DVD main menu
 334         QDomElement menus = dvddoc.createElement("menus");
 335         titleset.appendChild(menus);
 336         QDomElement pgc = dvddoc.createElement("pgc");
 337         pgc.setAttribute("entry", "root");
 338         menus.appendChild(pgc);
 339         for (int i = 0; i < buttons.count(); i++) {
 340             QDomElement button = dvddoc.createElement("button");
 341             button.setAttribute("name", 'b' + QString::number(i));
 342             QDomText nametext = dvddoc.createTextNode("jump title " + buttonsTarget.at(i) + ';');
 343             button.appendChild(nametext);
 344             pgc.appendChild(button);
 345         }
 346         QDomElement menuvob = dvddoc.createElement("vob");
 347         menuvob.setAttribute("file", m_menuFile.fileName());
 348         menuvob.setAttribute("pause", "inf");
 349         pgc.appendChild(menuvob);
 350     }
 351
 352     QDomElement titles = dvddoc.createElement("titles");
 353     titleset.appendChild(titles);
 354
 355     QStringList voburls = m_pageVob->selectedUrls();
 356
 357     QDomElement pgc2 = dvddoc.createElement("pgc");
 358     titles.appendChild(pgc2);
 359     if (m_pageMenu->createMenu()) {
 360         QDomElement post = dvddoc.createElement("post");
 361         QDomText call = dvddoc.createTextNode("call menu;");
 362         post.appendChild(call);
 363         pgc2.appendChild(post);
 364     }
 365
 366
 367     for (int i = 0; i < voburls.count(); i++) {
 368         if (!voburls.at(i).isEmpty()) {
 369             // Add vob entry
 370             QDomElement vob = dvddoc.createElement("vob");
 371             vob.setAttribute("file", voburls.at(i));
 372             pgc2.appendChild(vob);
 373         }
 374     }
 375
 376     /*
 377         // create one pgc for each video
 378         for (int i = 0; i < voburls.count(); i++) {
 379             if (!voburls.at(i).isEmpty()) {
 380                 // Add vob entry
 381
 382          QDomElement pgc2 = dvddoc.createElement("pgc");
 383          titles.appendChild(pgc2);
 384          if (m_pageMenu->createMenu()) {
 385       QDomElement post = dvddoc.createElement("post");
 386       QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
 387       post.appendChild(call);
 388       pgc2.appendChild(post);
 389          }
 390
 391                 QDomElement vob = dvddoc.createElement("vob");
 392                 vob.setAttribute("file", voburls.at(i));
 393                 pgc2.appendChild(vob);
 394             }
 395         }
 396     */
 397     QFile data2(m_authorFile.fileName());
 398     if (data2.open(QFile::WriteOnly)) {
 399         data2.write(dvddoc.toString().toUtf8());
 400     }
 401     data2.close();
 402     /*kDebug() << "------------------";
 403     kDebug() << dvddoc.toString();
 404     kDebug() << "------------------";*/
 405
 406     QStringList args;
 407     args << "-x" << m_authorFile.fileName();
 408     kDebug() << "// DVDAUTH ARGS: " << args;
 409     if (m_dvdauthor) {
 410         m_dvdauthor->close();
 411         delete m_dvdauthor;
 412         m_dvdauthor = NULL;
 413     }
 414     m_creationLog.clear();
 415     m_dvdauthor = new QProcess(this);
 416     connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
 417     m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
 418     m_dvdauthor->start("dvdauthor", args);
 419
 420 }
 421
 422 void DvdWizard::slotRenderFinished(int /*exitCode*/, QProcess::ExitStatus status)
 423 {
 424     QListWidgetItem *authitem =  m_status.job_progress->item(3);
 425     if (status == QProcess::CrashExit) {
 426         QByteArray result = m_dvdauthor->readAllStandardError();
 427         m_status.error_log->setText(result);
 428         m_status.error_box->setHidden(false);
 429         kDebug() << "DVDAuthor process crashed";
 430         authitem->setIcon(KIcon("dialog-close"));
 431         m_dvdauthor->close();
 432         delete m_dvdauthor;
 433         m_dvdauthor = NULL;
 434         cleanup();
 435         return;
 436     }
 437     m_creationLog.append(m_dvdauthor->readAllStandardError());
 438     m_dvdauthor->close();
 439     delete m_dvdauthor;
 440     m_dvdauthor = NULL;
 441
 442     // Check if DVD structure has the necessary infos
 443     if (!QFile::exists(m_iso.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
 444         m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD structure broken"));
 445         m_status.error_box->setHidden(false);
 446         kDebug() << "DVDAuthor process crashed";
 447         authitem->setIcon(KIcon("dialog-close"));
 448         cleanup();
 449         return;
 450     }
 451     authitem->setIcon(KIcon("dialog-ok"));
 452     qApp->processEvents();
 453     QStringList args;
 454     args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
 455
 456     if (m_mkiso) {
 457         m_mkiso->close();
 458         delete m_mkiso;
 459         m_mkiso = NULL;
 460     }
 461     m_mkiso = new QProcess(this);
 462     connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
 463     m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
 464     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 465     isoitem->setIcon(KIcon("system-run"));
 466     m_mkiso->start("mkisofs", args);
 467
 468 }
 469
 470 void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status)
 471 {
 472     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
 473     if (status == QProcess::CrashExit) {
 474         QByteArray result = m_mkiso->readAllStandardError();
 475         m_status.error_log->setText(result);
 476         m_status.error_box->setHidden(false);
 477         m_mkiso->close();
 478         delete m_mkiso;
 479         m_mkiso = NULL;
 480         cleanup();
 481         kDebug() << "Iso process crashed";
 482         isoitem->setIcon(KIcon("dialog-close"));
 483         return;
 484     }
 485
 486     m_creationLog.append(m_mkiso->readAllStandardError());
 487     delete m_mkiso;
 488     m_mkiso = NULL;
 489
 490     // Check if DVD iso is ok
 491     QFile iso(m_iso.iso_image->url().path());
 492     if (!iso.exists() || iso.size() == 0) {
 493         if (iso.exists()) {
 494             KIO::NetAccess::del(m_iso.iso_image->url(), this);
 495         }
 496         m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD ISO is broken"));
 497         m_status.error_box->setHidden(false);
 498         isoitem->setIcon(KIcon("dialog-close"));
 499         cleanup();
 500         return;
 501     }
 502
 503     isoitem->setIcon(KIcon("dialog-ok"));
 504     kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
 505     cleanup();
 506     kDebug() << m_creationLog;
 507     KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
 508
 509 }
 510
 511
 512 void DvdWizard::cleanup()
 513 {
 514     m_authorFile.remove();
 515     m_menuFile.remove();
 516     KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
 517 }
 518