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