]> git.sesse.net Git - kdenlive/blobdiff - src/dvdwizard.cpp
Save chapters now works in DVD Wizard
[kdenlive] / src / dvdwizard.cpp
index ad608371bd2438c6d65b4a79404633fa6832b73e..b2e23c87281b8c79d4967883270b2c851aad0ef9 100644 (file)
@@ -21,6 +21,8 @@
 #include "dvdwizard.h"
 #include "kdenlivesettings.h"
 #include "profilesdialog.h"
+#include "timecode.h"
+#include "monitormanager.h"
 
 #include <KStandardDirs>
 #include <KLocale>
 #include <QApplication>
 #include <QTimer>
 #include <QDomDocument>
+#include <QMenu>
 
 
-DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent): QWizard(parent), m_profile(profile) {
+DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent) :
+        QWizard(parent),
+        m_dvdauthor(NULL),
+        m_mkiso(NULL),
+        m_burnMenu(new QMenu(this))
+{
     //setPixmap(QWizard::WatermarkPixmap, QPixmap(KStandardDirs::locate("appdata", "banner.png")));
     setAttribute(Qt::WA_DeleteOnClose);
-    m_pageVob = new DvdWizardVob(this);
+    m_pageVob = new DvdWizardVob(profile, this);
     m_pageVob->setTitle(i18n("Select Files For Your DVD"));
     addPage(m_pageVob);
-
     if (!url.isEmpty()) m_pageVob->setUrl(url);
 
-    m_pageMenu = new DvdWizardMenu(m_profile, this);
+
+    m_pageChapters = new DvdWizardChapters(m_pageVob->isPal(), this);
+    m_pageChapters->setTitle(i18n("DVD Chapters"));
+    addPage(m_pageChapters);
+
+
+
+    m_pageMenu = new DvdWizardMenu(profile, this);
     m_pageMenu->setTitle(i18n("Create DVD Menu"));
     addPage(m_pageMenu);
 
@@ -61,33 +75,69 @@ DvdWizard::DvdWizard(const QString &url, const QString &profile, QWidget *parent
     QWizardPage *page4 = new QWizardPage;
     page4->setTitle(i18n("Creating DVD Image"));
     m_status.setupUi(page4);
+    m_status.error_box->setHidden(true);
     addPage(page4);
 
     connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(slotPageChanged(int)));
 
-//    connect(m_standard.button_all, SIGNAL(toggled(bool)), this, SLOT(slotCheckStandard()));
+    connect(m_status.button_preview, SIGNAL(clicked()), this, SLOT(slotPreview()));
+
+    QString programName("k3b");
+    QString exec = KStandardDirs::findExe(programName);
+    if (!exec.isEmpty()) {
+        //Add K3b action
+        QAction *k3b = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
+        k3b->setData(exec);
+    }
+    programName = "brasero";
+    exec = KStandardDirs::findExe(programName);
+    if (!exec.isEmpty()) {
+        //Add Brasero action
+        QAction *brasero = m_burnMenu->addAction(KIcon(programName), i18n("Burn with %1", programName), this, SLOT(slotBurn()));
+        brasero->setData(exec);
+    }
+    if (m_burnMenu->isEmpty()) m_burnMenu->addAction(i18n("No burning program found (K3b, Brasero)"));
+    m_status.button_burn->setMenu(m_burnMenu);
+    m_status.button_burn->setIcon(KIcon("tools-media-optical-burn"));
+    m_status.button_burn->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
+    m_status.button_preview->setIcon(KIcon("media-playback-start"));
+
 }
 
-DvdWizard::~DvdWizard() {
+DvdWizard::~DvdWizard()
+{
     // m_menuFile.remove();
+    delete m_burnMenu;
+    if (m_dvdauthor) {
+        m_dvdauthor->close();
+        delete m_dvdauthor;
+    }
+    if (m_mkiso) {
+        m_mkiso->close();
+        delete m_mkiso;
+    }
 }
 
 
-void DvdWizard::slotPageChanged(int page) {
-    kDebug() << "// PAGE CHGD: " << page;
+void DvdWizard::slotPageChanged(int page)
+{
+    //kDebug() << "// PAGE CHGD: " << page << ", ID: " << visitedPages();
     if (page == 1) {
-        m_pageMenu->setTargets(m_pageVob->selectedUrls());
+        m_pageChapters->setVobFiles(m_pageVob->isPal(), m_pageVob->selectedUrls(), m_pageVob->durations());
     } else if (page == 2) {
-        m_pageMenu->buttonsInfo();
+        m_pageMenu->setTargets(m_pageChapters->selectedTitles(), m_pageChapters->selectedTargets());
+        m_pageMenu->changeProfile(m_pageVob->isPal());
     } else if (page == 3) {
+        //m_pageMenu->buttonsInfo();
+    } else if (page == 4) {
         // clear job icons
         for (int i = 0; i < m_status.job_progress->count(); i++)
             m_status.job_progress->item(i)->setIcon(KIcon());
         QString warnMessage;
         if (KIO::NetAccess::exists(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), KIO::NetAccess::SourceSide, this))
-            warnMessage.append(i18n("Folder %1 already exists. Overwrite ?<br />", m_iso.tmp_folder->url().path() + "/DVD"));
+            warnMessage.append(i18n("Folder %1 already exists. Overwrite?" + '\n', m_iso.tmp_folder->url().path() + "/DVD"));
         if (KIO::NetAccess::exists(KUrl(m_iso.iso_image->url().path()), KIO::NetAccess::SourceSide, this))
-            warnMessage.append(i18n("Image file %1 already exists. Overwrite ?", m_iso.iso_image->url().path()));
+            warnMessage.append(i18n("Image file %1 already exists. Overwrite?", m_iso.iso_image->url().path()));
 
         if (!warnMessage.isEmpty() && KMessageBox::questionYesNo(this, warnMessage) == KMessageBox::No) {
             back();
@@ -95,12 +145,15 @@ void DvdWizard::slotPageChanged(int page) {
             KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
             QTimer::singleShot(300, this, SLOT(generateDvd()));
         }
+        m_status.button_preview->setEnabled(false);
+        m_status.button_burn->setEnabled(false);
     }
 }
 
 
 
-void DvdWizard::generateDvd() {
+void DvdWizard::generateDvd()
+{
     KTemporaryFile temp1;
     temp1.setSuffix(".png");
     //temp1.setAutoRemove(false);
@@ -142,7 +195,7 @@ void DvdWizard::generateDvd() {
     QListWidgetItem *images =  m_status.job_progress->item(0);
     images->setIcon(KIcon("system-run"));
     qApp->processEvents();
-    QMap <int, QRect> buttons = m_pageMenu->buttonsInfo();
+    QMap <QString, QRect> buttons = m_pageMenu->buttonsInfo();
     QStringList buttonsTarget;
 
     if (m_pageMenu->createMenu()) {
@@ -175,16 +228,21 @@ void DvdWizard::generateDvd() {
 
             kDebug() << "MLT ARGS: " << args;
             QProcess renderbg;
-            renderbg.start("inigo", args);
+            renderbg.start(KdenliveSettings::rendererpath(), args);
             if (renderbg.waitForFinished()) {
                 if (renderbg.exitStatus() == QProcess::CrashExit) {
-                    kDebug() << "/// RENDERING MENU vob crashed";
+                    kDebug() << "/// RENDERING MENU vob crashed";
+                    QByteArray result = renderbg.readAllStandardError();
                     vobitem->setIcon(KIcon("dialog-close"));
+                    m_status.error_log->setText(result);
+                    m_status.error_box->setHidden(false);
                     return;
                 }
             } else {
-                kDebug() << "/// RENDERING MENU vob timed out";
+                kDebug() << "/// RENDERING MENU vob timed out";
                 vobitem->setIcon(KIcon("dialog-close"));
+                m_status.error_log->setText(i18n("Rendering job timed out"));
+                m_status.error_box->setHidden(false);
                 return;
             }
             vobitem->setIcon(KIcon("dialog-ok"));
@@ -213,7 +271,7 @@ void DvdWizard::generateDvd() {
 
         int max = buttons.count() - 1;
         int i = 0;
-        QMapIterator<int, QRect> it(buttons);
+        QMapIterator<QString, QRect> it(buttons);
         while (it.hasNext()) {
             it.next();
             QDomElement but = doc.createElement("button");
@@ -223,10 +281,10 @@ void DvdWizard::generateDvd() {
             if (i > 0) but.setAttribute("up", 'b' + QString::number(i - 1));
             else but.setAttribute("up", 'b' + QString::number(max));
             QRect r = it.value();
-            int target = it.key();
-            // TODO: solve play all button
-            if (target == 0) target = 1;
-            buttonsTarget.append(QString::number(target));
+            //int target = it.key();
+            // TODO: solve play all button
+            //if (target == 0) target = 1;
+            buttonsTarget.append(it.key());
             but.setAttribute("x0", QString::number(r.x()));
             but.setAttribute("y0", QString::number(r.y()));
             but.setAttribute("x1", QString::number(r.right()));
@@ -235,13 +293,9 @@ void DvdWizard::generateDvd() {
             i++;
         }
 
-        kDebug() << "///// SPU: ";
-        kDebug() << doc.toString();
-
         QFile data(temp6.fileName());
         if (data.open(QFile::WriteOnly)) {
-            QTextStream out(&data);
-            out << doc.toString();
+            data.write(doc.toString().toUtf8());
         }
         data.close();
 
@@ -261,13 +315,18 @@ void DvdWizard::generateDvd() {
         if (spumux.waitForFinished()) {
             kDebug() << QString(spumux.readAll()).simplified();
             if (spumux.exitStatus() == QProcess::CrashExit) {
-                kDebug() << "/// RENDERING SPUMUX MENU crashed";
+                QByteArray result = spumux.readAllStandardError();
                 spuitem->setIcon(KIcon("dialog-close"));
+                m_status.error_log->setText(result);
+                m_status.error_box->setHidden(false);
+                kDebug() << "/// RENDERING SPUMUX MENU crashed";
                 return;
             }
         } else {
             kDebug() << "/// RENDERING SPUMUX MENU timed out";
             spuitem->setIcon(KIcon("dialog-close"));
+            m_status.error_log->setText(i18n("Menu job timed out"));
+            m_status.error_box->setHidden(false);
             return;
         }
 
@@ -314,10 +373,14 @@ void DvdWizard::generateDvd() {
         QDomElement pgc = dvddoc.createElement("pgc");
         pgc.setAttribute("entry", "root");
         menus.appendChild(pgc);
+        QDomElement pre = dvddoc.createElement("pre");
+        pgc.appendChild(pre);
+        QDomText nametext = dvddoc.createTextNode("{g1 = 0;}");
+        pre.appendChild(nametext);
         for (int i = 0; i < buttons.count(); i++) {
             QDomElement button = dvddoc.createElement("button");
             button.setAttribute("name", 'b' + QString::number(i));
-            QDomText nametext = dvddoc.createTextNode("jump title " + buttonsTarget.at(i) + ';');
+            nametext = dvddoc.createTextNode('{' + buttonsTarget.at(i) + ";}");
             button.appendChild(nametext);
             pgc.appendChild(button);
         }
@@ -332,101 +395,187 @@ void DvdWizard::generateDvd() {
 
     QStringList voburls = m_pageVob->selectedUrls();
 
-    QDomElement pgc2 = dvddoc.createElement("pgc");
-    titles.appendChild(pgc2);
-    if (m_pageMenu->createMenu()) {
-        QDomElement post = dvddoc.createElement("post");
-        QDomText call = dvddoc.createTextNode("call menu;");
-        post.appendChild(call);
-        pgc2.appendChild(post);
-    }
+    QDomElement pgc2;
 
 
     for (int i = 0; i < voburls.count(); i++) {
         if (!voburls.at(i).isEmpty()) {
             // Add vob entry
+            pgc2 = dvddoc.createElement("pgc");
+            pgc2.setAttribute("pause", 0);
+            titles.appendChild(pgc2);
             QDomElement vob = dvddoc.createElement("vob");
             vob.setAttribute("file", voburls.at(i));
+            // Add chapters
+            QStringList chaptersList = m_pageChapters->chapters(i);
+            if (!chaptersList.isEmpty()) vob.setAttribute("chapters", chaptersList.join(","));
+
             pgc2.appendChild(vob);
+            if (m_pageMenu->createMenu()) {
+                QDomElement post = dvddoc.createElement("post");
+                QDomText call;
+                if (i == voburls.count() - 1) call = dvddoc.createTextNode("{g1 = 0; call menu;}");
+                else {
+                    call = dvddoc.createTextNode("{if ( g1 eq 999 ) { call menu; } jump title " + QString::number(i + 2) + ";}");
+                }
+                post.appendChild(call);
+                pgc2.appendChild(post);
+            }
         }
     }
 
-    /*
-        // create one pgc for each video
-        for (int i = 0; i < voburls.count(); i++) {
-            if (!voburls.at(i).isEmpty()) {
-                // Add vob entry
-
-         QDomElement pgc2 = dvddoc.createElement("pgc");
-         titles.appendChild(pgc2);
-         if (m_pageMenu->createMenu()) {
-      QDomElement post = dvddoc.createElement("post");
-      QDomText call = dvddoc.createTextNode("call vmgm menu 1;");
-      post.appendChild(call);
-      pgc2.appendChild(post);
-         }
-
-                QDomElement vob = dvddoc.createElement("vob");
-                vob.setAttribute("file", voburls.at(i));
-                pgc2.appendChild(vob);
-            }
-        }
-    */
+
     QFile data2(m_authorFile.fileName());
     if (data2.open(QFile::WriteOnly)) {
-        QTextStream out(&data2);
-        out << dvddoc.toString();
+        data2.write(dvddoc.toString().toUtf8());
     }
     data2.close();
-
+    /*kDebug() << "------------------";
+    kDebug() << dvddoc.toString();
+    kDebug() << "------------------";*/
 
     QStringList args;
     args << "-x" << m_authorFile.fileName();
     kDebug() << "// DVDAUTH ARGS: " << args;
-    QProcess *dvdauth = new QProcess(this);
-    connect(dvdauth, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
-    dvdauth->setProcessChannelMode(QProcess::MergedChannels);
-    dvdauth->start("dvdauthor", args);
+    if (m_dvdauthor) {
+        m_dvdauthor->close();
+        delete m_dvdauthor;
+        m_dvdauthor = NULL;
+    }
+    m_creationLog.clear();
+    m_dvdauthor = new QProcess(this);
+    connect(m_dvdauthor, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotRenderFinished(int, QProcess::ExitStatus)));
+    m_dvdauthor->setProcessChannelMode(QProcess::MergedChannels);
+    m_dvdauthor->start("dvdauthor", args);
 
 }
 
-void DvdWizard::slotRenderFinished(int /*exitCode*/, QProcess::ExitStatus status) {
+void DvdWizard::slotRenderFinished(int /*exitCode*/, QProcess::ExitStatus status)
+{
     QListWidgetItem *authitem =  m_status.job_progress->item(3);
     if (status == QProcess::CrashExit) {
+        QByteArray result = m_dvdauthor->readAllStandardError();
+        m_status.error_log->setText(result);
+        m_status.error_box->setHidden(false);
+        kDebug() << "DVDAuthor process crashed";
+        authitem->setIcon(KIcon("dialog-close"));
+        m_dvdauthor->close();
+        delete m_dvdauthor;
+        m_dvdauthor = NULL;
+        cleanup();
+        return;
+    }
+    m_creationLog.append(m_dvdauthor->readAllStandardError());
+    m_dvdauthor->close();
+    delete m_dvdauthor;
+    m_dvdauthor = NULL;
+
+    // Check if DVD structure has the necessary infos
+    if (!QFile::exists(m_iso.tmp_folder->url().path() + "/DVD/VIDEO_TS/VIDEO_TS.IFO")) {
+        m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD structure broken"));
+        m_status.error_box->setHidden(false);
         kDebug() << "DVDAuthor process crashed";
         authitem->setIcon(KIcon("dialog-close"));
+        cleanup();
         return;
     }
     authitem->setIcon(KIcon("dialog-ok"));
     qApp->processEvents();
     QStringList args;
     args << "-dvd-video" << "-v" << "-o" << m_iso.iso_image->url().path() << m_iso.tmp_folder->url().path() + "/DVD";
-    QProcess *mkiso = new QProcess(this);
-    connect(mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
-    mkiso->setProcessChannelMode(QProcess::MergedChannels);
+
+    if (m_mkiso) {
+        m_mkiso->close();
+        delete m_mkiso;
+        m_mkiso = NULL;
+    }
+    m_mkiso = new QProcess(this);
+    connect(m_mkiso, SIGNAL(finished(int , QProcess::ExitStatus)), this, SLOT(slotIsoFinished(int, QProcess::ExitStatus)));
+    m_mkiso->setProcessChannelMode(QProcess::MergedChannels);
     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
     isoitem->setIcon(KIcon("system-run"));
-    mkiso->start("mkisofs", args);
+    m_mkiso->start("mkisofs", args);
 
 }
 
-void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status) {
+void DvdWizard::slotIsoFinished(int /*exitCode*/, QProcess::ExitStatus status)
+{
     QListWidgetItem *isoitem =  m_status.job_progress->item(4);
     if (status == QProcess::CrashExit) {
-        m_authorFile.remove();
-        m_menuFile.remove();
-        KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
+        QByteArray result = m_mkiso->readAllStandardError();
+        m_status.error_log->setText(result);
+        m_status.error_box->setHidden(false);
+        m_mkiso->close();
+        delete m_mkiso;
+        m_mkiso = NULL;
+        cleanup();
         kDebug() << "Iso process crashed";
         isoitem->setIcon(KIcon("dialog-close"));
         return;
     }
+
+    m_creationLog.append(m_mkiso->readAllStandardError());
+    delete m_mkiso;
+    m_mkiso = NULL;
+
+    // Check if DVD iso is ok
+    QFile iso(m_iso.iso_image->url().path());
+    if (!iso.exists() || iso.size() == 0) {
+        if (iso.exists()) {
+            KIO::NetAccess::del(m_iso.iso_image->url(), this);
+        }
+        m_status.error_log->setText(m_creationLog + '\n' + i18n("DVD ISO is broken"));
+        m_status.error_box->setHidden(false);
+        isoitem->setIcon(KIcon("dialog-close"));
+        cleanup();
+        return;
+    }
+
     isoitem->setIcon(KIcon("dialog-ok"));
     kDebug() << "ISO IMAGE " << m_iso.iso_image->url().path() << " Successfully created";
+    cleanup();
+    kDebug() << m_creationLog;
+
+    m_status.error_log->setText(i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
+    m_status.button_preview->setEnabled(true);
+    m_status.button_burn->setEnabled(true);
+    m_status.error_box->setHidden(false);
+    //KMessageBox::information(this, i18n("DVD ISO image %1 successfully created.", m_iso.iso_image->url().path()));
+
+}
+
+
+void DvdWizard::cleanup()
+{
     m_authorFile.remove();
     m_menuFile.remove();
     KIO::NetAccess::del(KUrl(m_iso.tmp_folder->url().path() + "/DVD"), this);
-    KMessageBox::information(this, i18n("DVD iso image %1 successfully created.", m_iso.iso_image->url().path()));
+}
+
 
+void DvdWizard::slotPreview()
+{
+    QString programName("xine");
+    QString exec = KStandardDirs::findExe(programName);
+    if (exec.isEmpty()) KMessageBox::sorry(this, i18n("You need program <b>%1</b> to perform this action", programName));
+    else QProcess::startDetached(exec, QStringList() << "dvd://" + m_iso.iso_image->url().path());
+}
+
+void DvdWizard::slotBurn()
+{
+    QAction *action = qobject_cast<QAction *>(sender());
+    QString exec = action->data().toString();
+    QStringList args;
+    if (exec.endsWith("k3b")) args << "--image" << m_iso.iso_image->url().path();
+    else args << "--image=" + m_iso.iso_image->url().path();
+    QProcess::startDetached(exec, args);
 }
 
 
+
+
+
+
+
+
+