git.sesse.net Git - kdenlive/blob - src/dvdwizardvob.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2009 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 "dvdwizardvob.h"
  21 #include "kthumb.h"
  22 #include "timecode.h"
  23
  24 #include <mlt++/Mlt.h>
  25
  26 #include <KUrlRequester>
  27 #include <KDebug>
  28 #include <KStandardDirs>
  29 #include <KFileItem>
  30 #include <KFileDialog>
  31
  32 #include <QHBoxLayout>
  33 #include <QDomDocument>
  34 #include <QTreeWidgetItem>
  35 #include <QHeaderView>
  36
  37 DvdWizardVob::DvdWizardVob(const QString &profile, QWidget *parent) :
  38         QWizardPage(parent)
  39 {
  40     m_view.setupUi(this);
  41     m_view.intro_vob->setEnabled(false);
  42     m_view.intro_vob->setFilter("video/mpeg");
  43     m_view.button_add->setIcon(KIcon("document-new"));
  44     m_view.button_delete->setIcon(KIcon("edit-delete"));
  45     m_view.button_up->setIcon(KIcon("go-up"));
  46     m_view.button_down->setIcon(KIcon("go-down"));
  47     connect(m_view.use_intro, SIGNAL(toggled(bool)), m_view.intro_vob, SLOT(setEnabled(bool)));
  48     connect(m_view.button_add, SIGNAL(clicked()), this, SLOT(slotAddVobFile()));
  49     connect(m_view.button_delete, SIGNAL(clicked()), this, SLOT(slotDeleteVobFile()));
  50     connect(m_view.button_up, SIGNAL(clicked()), this, SLOT(slotItemUp()));
  51     connect(m_view.button_down, SIGNAL(clicked()), this, SLOT(slotItemDown()));
  52     connect(m_view.vobs_list, SIGNAL(itemSelectionChanged()), this, SLOT(slotCheckVobList()));
  53     m_view.vobs_list->setIconSize(QSize(60, 45));
  54
  55     if (KStandardDirs::findExe("dvdauthor").isEmpty()) m_errorMessage.append(i18n("<strong>Program %1 is required for the DVD wizard.</strong>", i18n("dvdauthor")));
  56     if (KStandardDirs::findExe("mkisofs").isEmpty()) m_errorMessage.append(i18n("<strong>Program %1 is required for the DVD wizard.</strong>", i18n("mkisofs")));
  57     if (m_errorMessage.isEmpty()) m_view.error_message->setVisible(false);
  58     else m_view.error_message->setText(m_errorMessage);
  59
  60     m_view.dvd_profile->addItems(QStringList() << i18n("PAL 4:3") << i18n("PAL 16:9") << i18n("NTSC 4:3") << i18n("NTSC 16:9"));
  61     if (profile == "dv_pal_wide") m_view.dvd_profile->setCurrentIndex(1);
  62     else if (profile == "dv_ntsc") m_view.dvd_profile->setCurrentIndex(2);
  63     else if (profile == "dv_ntsc_wide") m_view.dvd_profile->setCurrentIndex(3);
  64
  65     connect(m_view.dvd_profile, SIGNAL(activated(int)), this, SLOT(changeFormat()));
  66     m_view.vobs_list->header()->setStretchLastSection(false);
  67     m_view.vobs_list->header()->setResizeMode(0, QHeaderView::Stretch);
  68     m_view.vobs_list->header()->setResizeMode(1, QHeaderView::Custom);
  69     m_view.vobs_list->header()->setResizeMode(2, QHeaderView::Custom);
  70
  71 #if KDE_IS_VERSION(4,2,0)
  72     m_capacityBar = new KCapacityBar(KCapacityBar::DrawTextInline, this);
  73     QHBoxLayout *layout = new QHBoxLayout;
  74     layout->addWidget(m_capacityBar);
  75     m_view.size_box->setLayout(layout);
  76 #else
  77     m_view.size_box->setHidden(true);
  78 #endif
  79     slotCheckVobList();
  80 }
  81
  82 DvdWizardVob::~DvdWizardVob()
  83 {
  84 #if KDE_IS_VERSION(4,2,0)
  85     delete m_capacityBar;
  86 #endif
  87 }
  88
  89
  90 void DvdWizardVob::slotAddVobFile(KUrl url, const QString &chapters)
  91 {
  92     if (url.isEmpty()) url = KFileDialog::getOpenUrl(KUrl("kfiledialog:///projectfolder"), "video/mpeg", this, i18n("Add new video file"));
  93     if (url.isEmpty()) return;
  94     QFile f(url.path());
  95     qint64 fileSize = f.size();
  96     QString profilename;
  97     switch (m_view.dvd_profile->currentIndex()) {
  98     case 1:
  99         profilename = "dv_pal_wide";
 100         break;
 101     case 2:
 102         profilename = "dv_ntsc";
 103         break;
 104     case 3:
 105         profilename = "dv_ntsc_wide";
 106         break;
 107     default:
 108         profilename = "dv_pal";
 109         break;
 110     }
 111
 112     char *tmp = (char*) qstrdup(profilename.toUtf8().data());
 113     Mlt::Profile profile(tmp);
 114     delete[] tmp;
 115     QTreeWidgetItem *item = new QTreeWidgetItem(m_view.vobs_list, QStringList() << url.path() << QString() << KIO::convertSize(fileSize));
 116     item->setData(0, Qt::UserRole, fileSize);
 117     item->setIcon(0, KIcon("video-x-generic"));
 118
 119     QPixmap pix(60, 45);
 120
 121     tmp = (char *) qstrdup(url.path().toUtf8().data());
 122     Mlt::Producer *producer = new Mlt::Producer(profile, tmp);
 123     delete[] tmp;
 124
 125     if (producer->is_blank() == false) {
 126         int width = 45.0 * profile.dar();
 127         if (width % 2 == 1) width++;
 128         pix = KThumb::getFrame(producer, 0, width, 45);
 129         item->setIcon(0, pix);
 130         int playTime = producer->get_playtime();
 131         item->setText(1, Timecode::getStringTimecode(playTime, profile.fps()));
 132         item->setData(1, Qt::UserRole, playTime);
 133     }
 134     delete producer;
 135
 136     if (chapters.isEmpty() == false)
 137         item->setData(1, Qt::UserRole + 1, chapters);
 138     else if (QFile::exists(url.path() + ".dvdchapter")) {
 139         // insert chapters as children
 140         QFile file(url.path() + ".dvdchapter");
 141         if (file.open(QIODevice::ReadOnly)) {
 142             QDomDocument doc;
 143             doc.setContent(&file);
 144             file.close();
 145             QDomNodeList chapters = doc.elementsByTagName("chapter");
 146             QStringList chaptersList;
 147             for (int j = 0; j < chapters.count(); j++) {
 148                 chaptersList.append(QString::number(chapters.at(j).toElement().attribute("time").toInt()));
 149             }
 150             item->setData(1, Qt::UserRole + 1, chaptersList.join(";"));
 151         }
 152     } else // Explicitly add a chapter at 00:00:00:00
 153         item->setData(1, Qt::UserRole + 1, "0");
 154
 155     slotCheckVobList();
 156 }
 157
 158 void DvdWizardVob::changeFormat()
 159 {
 160     int max = m_view.vobs_list->topLevelItemCount();
 161     QString profilename;
 162     switch (m_view.dvd_profile->currentIndex()) {
 163     case 1:
 164         profilename = "dv_pal_wide";
 165         break;
 166     case 2:
 167         profilename = "dv_ntsc";
 168         break;
 169     case 3:
 170         profilename = "dv_ntsc_wide";
 171         break;
 172     default:
 173         profilename = "dv_pal";
 174         break;
 175     }
 176
 177     char *tmp = (char*) qstrdup(profilename.toUtf8().data());
 178     Mlt::Profile profile(tmp);
 179     delete[] tmp;
 180     QPixmap pix(180, 135);
 181
 182     for (int i = 0; i < max; i++) {
 183         QTreeWidgetItem *item = m_view.vobs_list->topLevelItem(i);
 184         tmp = (char *) qstrdup(item->text(0).toUtf8().data());
 185         Mlt::Producer *producer = new Mlt::Producer(profile, tmp);
 186         delete[] tmp;
 187
 188         if (producer->is_blank() == false) {
 189             //pix = KThumb::getFrame(producer, 0, 135 * profile.dar(), 135);
 190             //item->setIcon(0, pix);
 191             item->setText(1, Timecode::getStringTimecode(producer->get_playtime(), profile.fps()));
 192         }
 193         delete producer;
 194         int submax = item->childCount();
 195         for (int j = 0; j < submax; j++) {
 196             QTreeWidgetItem *subitem = item->child(j);
 197             subitem->setText(1, Timecode::getStringTimecode(subitem->data(1, Qt::UserRole).toInt(), profile.fps()));
 198         }
 199     }
 200     slotCheckVobList();
 201 }
 202
 203 void DvdWizardVob::slotDeleteVobFile()
 204 {
 205     QTreeWidgetItem *item = m_view.vobs_list->currentItem();
 206     if (item == NULL) return;
 207     delete item;
 208     slotCheckVobList();
 209 }
 210
 211
 212 // virtual
 213 bool DvdWizardVob::isComplete() const
 214 {
 215     if (!m_view.error_message->text().isEmpty()) return false;
 216     if (m_view.vobs_list->topLevelItemCount() == 0) return false;
 217     return true;
 218 }
 219
 220 void DvdWizardVob::setUrl(const QString &url)
 221 {
 222     slotAddVobFile(KUrl(url));
 223 }
 224
 225 QStringList DvdWizardVob::selectedUrls() const
 226 {
 227     QStringList result;
 228     QString path;
 229     int max = m_view.vobs_list->topLevelItemCount();
 230     for (int i = 0; i < max; i++) {
 231         QTreeWidgetItem *item = m_view.vobs_list->topLevelItem(i);
 232         if (item) result.append(item->text(0));
 233     }
 234     return result;
 235 }
 236
 237
 238 QStringList DvdWizardVob::durations() const
 239 {
 240     QStringList result;
 241     QString path;
 242     int max = m_view.vobs_list->topLevelItemCount();
 243     for (int i = 0; i < max; i++) {
 244         QTreeWidgetItem *item = m_view.vobs_list->topLevelItem(i);
 245         if (item) result.append(QString::number(item->data(1, Qt::UserRole).toInt()));
 246     }
 247     return result;
 248 }
 249
 250 QStringList DvdWizardVob::chapters() const
 251 {
 252     QStringList result;
 253     QString path;
 254     int max = m_view.vobs_list->topLevelItemCount();
 255     for (int i = 0; i < max; i++) {
 256         QTreeWidgetItem *item = m_view.vobs_list->topLevelItem(i);
 257         if (item) result.append(item->data(1, Qt::UserRole + 1).toString());
 258     }
 259     return result;
 260 }
 261
 262 void DvdWizardVob::updateChapters(QMap <QString, QString> chaptersdata)
 263 {
 264     int max = m_view.vobs_list->topLevelItemCount();
 265     for (int i = 0; i < max; i++) {
 266         QTreeWidgetItem *item = m_view.vobs_list->topLevelItem(i);
 267         item->setData(1, Qt::UserRole + 1, chaptersdata.value(item->text(0)));
 268     }
 269 }
 270
 271 int DvdWizardVob::duration(int ix) const
 272 {
 273     int result = -1;
 274     QTreeWidgetItem *item = m_view.vobs_list->topLevelItem(ix);
 275     if (item) {
 276         result = item->data(1, Qt::UserRole).toInt();
 277     }
 278     return result;
 279 }
 280
 281
 282 QString DvdWizardVob::introMovie() const
 283 {
 284     if (!m_view.use_intro->isChecked()) return QString();
 285     return m_view.intro_vob->url().path();
 286 }
 287
 288 void DvdWizardVob::setIntroMovie(const QString path)
 289 {
 290     m_view.intro_vob->setUrl(KUrl(path));
 291     m_view.use_intro->setChecked(path.isEmpty() == false);
 292 }
 293
 294
 295 void DvdWizardVob::slotCheckVobList()
 296 {
 297     emit completeChanged();
 298     int max = m_view.vobs_list->topLevelItemCount();
 299     QTreeWidgetItem *item = m_view.vobs_list->currentItem();
 300     bool hasItem = true;
 301     if (item == NULL) hasItem = false;
 302     m_view.button_delete->setEnabled(hasItem);
 303     if (hasItem && m_view.vobs_list->indexOfTopLevelItem(item) == 0) m_view.button_up->setEnabled(false);
 304     else m_view.button_up->setEnabled(hasItem);
 305     if (hasItem && m_view.vobs_list->indexOfTopLevelItem(item) == max - 1) m_view.button_down->setEnabled(false);
 306     else m_view.button_down->setEnabled(hasItem);
 307
 308 #if KDE_IS_VERSION(4,2,0)
 309     qint64 totalSize = 0;
 310     for (int i = 0; i < max; i++) {
 311         item = m_view.vobs_list->topLevelItem(i);
 312         if (item) totalSize += (qint64) item->data(0, Qt::UserRole).toInt();
 313     }
 314
 315     qint64 maxSize = (qint64) 47000 * 100000;
 316     m_capacityBar->setValue(100 * totalSize / maxSize);
 317     m_capacityBar->setText(KIO::convertSize(totalSize));
 318 #endif
 319 }
 320
 321 void DvdWizardVob::slotItemUp()
 322 {
 323     QTreeWidgetItem *item = m_view.vobs_list->currentItem();
 324     if (item == NULL) return;
 325     int index = m_view.vobs_list->indexOfTopLevelItem(item);
 326     if (index == 0) return;
 327     m_view.vobs_list->insertTopLevelItem(index - 1, m_view.vobs_list->takeTopLevelItem(index));
 328 }
 329
 330 void DvdWizardVob::slotItemDown()
 331 {
 332     int max = m_view.vobs_list->topLevelItemCount();
 333     QTreeWidgetItem *item = m_view.vobs_list->currentItem();
 334     if (item == NULL) return;
 335     int index = m_view.vobs_list->indexOfTopLevelItem(item);
 336     if (index == max - 1) return;
 337     m_view.vobs_list->insertTopLevelItem(index + 1, m_view.vobs_list->takeTopLevelItem(index));
 338 }
 339
 340 bool DvdWizardVob::isPal() const
 341 {
 342     return m_view.dvd_profile->currentIndex() < 2;
 343 }
 344
 345 bool DvdWizardVob::isWide() const
 346 {
 347     return (m_view.dvd_profile->currentIndex() == 1 || m_view.dvd_profile->currentIndex() == 3);
 348 }
 349
 350 void DvdWizardVob::setProfile(const QString profile)
 351 {
 352     if (profile == "dv_pal") m_view.dvd_profile->setCurrentIndex(0);
 353     else if (profile == "dv_pal_wide") m_view.dvd_profile->setCurrentIndex(1);
 354     else if (profile == "dv_ntsc") m_view.dvd_profile->setCurrentIndex(2);
 355     else if (profile == "dv_ntsc_wide") m_view.dvd_profile->setCurrentIndex(3);
 356 }
 357
 358 void DvdWizardVob::clear()
 359 {
 360     m_view.vobs_list->clear();
 361 }