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

   1 /***************************************************************************
   2  *   Copyright (C) 2008 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 "documentchecker.h"
  22 #include "kthumb.h"
  23 #include "docclipbase.h"
  24 #include "titlewidget.h"
  25 #include "definitions.h"
  26 #include "kdenlivesettings.h"
  27
  28 #include <KDebug>
  29 #include <KGlobalSettings>
  30 #include <KFileItem>
  31 #include <KIO/NetAccess>
  32 #include <KFileDialog>
  33 #include <KApplication>
  34 #include <KUrlRequesterDialog>
  35 #include <KMessageBox>
  36 #include <KStandardDirs>
  37
  38 #include <QTreeWidgetItem>
  39 #include <QFile>
  40 #include <QHeaderView>
  41 #include <QIcon>
  42 #include <QPixmap>
  43 #include <QTimer>
  44 #include <QCryptographicHash>
  45
  46 const int hashRole = Qt::UserRole;
  47 const int sizeRole = Qt::UserRole + 1;
  48 const int idRole = Qt::UserRole + 2;
  49 const int statusRole = Qt::UserRole + 3;
  50 const int typeRole = Qt::UserRole + 4;
  51 const int typeOriginalResource = Qt::UserRole + 5;
  52 const int resetDurationRole = Qt::UserRole + 6;
  53
  54 const int CLIPMISSING = 0;
  55 const int CLIPOK = 1;
  56 const int CLIPPLACEHOLDER = 2;
  57 const int CLIPWRONGDURATION = 3;
  58 const int PROXYMISSING = 4;
  59 const int SOURCEMISSING = 5;
  60
  61 const int LUMAMISSING = 10;
  62 const int LUMAOK = 11;
  63 const int LUMAPLACEHOLDER = 12;
  64
  65 enum TITLECLIPTYPE { TITLE_IMAGE_ELEMENT = 20, TITLE_FONT_ELEMENT = 21 };
  66
  67 DocumentChecker::DocumentChecker(const QDomNodeList &infoproducers, const QDomDocument &doc):
  68     m_info(infoproducers), m_doc(doc), m_dialog(NULL)
  69 {
  70
  71 }
  72
  73
  74 bool DocumentChecker::hasErrorInClips()
  75 {
  76     int clipType;
  77     QDomElement e;
  78     QString resource;
  79     int max;
  80     QDomNodeList documentProducers = m_doc.elementsByTagName("producer");
  81     QList <QDomElement> wrongDurationClips;
  82     // List clips whose proxy is missing
  83     QList <QDomElement> missingProxies;
  84     // List clips who have a working proxy but no source clip
  85     QList <QDomElement> missingSources;
  86     m_safeImages.clear();
  87     m_safeFonts.clear();
  88     max = m_info.count();
  89     for (int i = 0; i < max; ++i) {
  90         e = m_info.item(i).toElement();
  91         clipType = e.attribute("type").toInt();
  92         if (clipType == COLOR) continue;
  93         if (clipType != TEXT && clipType != IMAGE && clipType != SLIDESHOW) {
  94             QString id = e.attribute("id");
  95             int duration = e.attribute("duration").toInt();
  96             int mltDuration = -1;
  97             QDomElement mltProd;
  98             QString prodId;
  99             // Check that the duration is in sync between Kdenlive's info and MLT's playlist
 100             int prodsCount = documentProducers.count();
 101             for (int j = 0; j < prodsCount; j++) {
 102                 mltProd = documentProducers.at(j).toElement();
 103                 prodId = mltProd.attribute("id");
 104                 // Don't check slowmotion clips for now... (TODO?)
 105                 if (prodId.startsWith("slowmotion")) continue;
 106                 if (prodId.contains('_')) prodId = prodId.section('_', 0, 0);
 107                 if (prodId != id) continue;
 108                 if (mltDuration > 0 ) {
 109                     // We have several MLT producers for the same clip (probably track producers)
 110                     int newLength = EffectsList::property(mltProd, "length").toInt();
 111                     if (newLength != mltDuration) {
 112                         // we have a different duration for the same clip, that is not safe
 113                         e.setAttribute("_resetDuration", 1);
 114                     }
 115                 }
 116                 mltDuration = EffectsList::property(mltProd, "length").toInt();
 117                 if (mltDuration != duration) {
 118                     // Duration mismatch
 119                     e.setAttribute("_mismatch", mltDuration);
 120                     if (mltDuration == 15000) {
 121                         // a length of 15000 might indicate a wrong clip length since it is a default length
 122                         e.setAttribute("_resetDuration", 1);
 123                     }
 124                     if (!wrongDurationClips.contains(e)) wrongDurationClips.append(e);
 125                 }
 126             }
 127         }
 128
 129         if (clipType == TEXT) {
 130             //TODO: Check is clip template is missing (xmltemplate) or hash changed
 131             QStringList images = TitleWidget::extractImageList(e.attribute("xmldata"));
 132             QStringList fonts = TitleWidget::extractFontList(e.attribute("xmldata"));
 133             checkMissingImagesAndFonts(images, fonts, e.attribute("id"), e.attribute("name"));
 134             continue;
 135         }
 136         resource = e.attribute("resource");
 137         if (e.hasAttribute("proxy")) {
 138             QString proxyresource = e.attribute("proxy");
 139             if (!proxyresource.isEmpty() && proxyresource != "-") {
 140                 // clip has a proxy
 141                 if (!KIO::NetAccess::exists(KUrl(proxyresource), KIO::NetAccess::SourceSide, 0)) {
 142                     // Missing clip found
 143                     missingProxies.append(e);
 144                 }
 145                 else if (!KIO::NetAccess::exists(KUrl(resource), KIO::NetAccess::SourceSide, 0)) {
 146                     // clip has proxy but original clip is missing
 147                     missingSources.append(e);
 148                     continue;
 149                 }
 150             }
 151         }
 152         if (clipType == SLIDESHOW) resource = KUrl(resource).directory();
 153         if (!KIO::NetAccess::exists(KUrl(resource), KIO::NetAccess::SourceSide, 0)) {
 154             // Missing clip found
 155             m_missingClips.append(e);
 156         } else {
 157             // Check if the clip has changed
 158             if (clipType != SLIDESHOW && e.hasAttribute("file_hash")) {
 159                 if (e.attribute("file_hash") != DocClipBase::getHash(e.attribute("resource")))
 160                     e.removeAttribute("file_hash");
 161             }
 162         }
 163     }
 164
 165     // Get list of used Luma files
 166     QStringList missingLumas;
 167     QStringList filesToCheck;
 168     QString filePath;
 169     QString root = m_doc.documentElement().attribute("root");
 170     if (!root.isEmpty()) root = KUrl(root).path(KUrl::AddTrailingSlash);
 171     QDomNodeList trans = m_doc.elementsByTagName("transition");
 172     max = trans.count();
 173     for (int i = 0; i < max; ++i) {
 174         QString luma = getProperty(trans.at(i).toElement(), "luma");
 175         if (!luma.isEmpty() && !filesToCheck.contains(luma))
 176             filesToCheck.append(luma);
 177     }
 178     // Check existence of luma files
 179     foreach (const QString &lumafile, filesToCheck) {
 180         filePath = lumafile;
 181         if (!filePath.startsWith('/')) filePath.prepend(root);
 182         if (!QFile::exists(filePath)) {
 183             missingLumas.append(lumafile);
 184         }
 185     }
 186
 187
 188
 189     if (m_missingClips.isEmpty() && missingLumas.isEmpty() && wrongDurationClips.isEmpty() && missingProxies.isEmpty() && missingSources.isEmpty())
 190         return false;
 191
 192     m_dialog = new QDialog();
 193     m_dialog->setFont(KGlobalSettings::toolBarFont());
 194     m_ui.setupUi(m_dialog);
 195
 196     foreach(const QString &l, missingLumas) {
 197         QTreeWidgetItem *item = new QTreeWidgetItem(m_ui.treeWidget, QStringList() << i18n("Luma file") << l);
 198         item->setIcon(0, KIcon("dialog-close"));
 199         item->setData(0, idRole, l);
 200         item->setData(0, statusRole, LUMAMISSING);
 201     }
 202
 203     m_ui.buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);
 204     max = m_missingClips.count();
 205     for (int i = 0; i < max; ++i) {
 206         e = m_missingClips.at(i).toElement();
 207         QString clipType;
 208         int t = e.attribute("type").toInt();
 209         switch (t) {
 210         case AV:
 211             clipType = i18n("Video clip");
 212             break;
 213         case VIDEO:
 214             clipType = i18n("Mute video clip");
 215             break;
 216         case AUDIO:
 217             clipType = i18n("Audio clip");
 218             break;
 219         case PLAYLIST:
 220             clipType = i18n("Playlist clip");
 221             break;
 222         case IMAGE:
 223             clipType = i18n("Image clip");
 224             break;
 225         case SLIDESHOW:
 226             clipType = i18n("Slideshow clip");
 227             break;
 228         case TITLE_IMAGE_ELEMENT:
 229             clipType = i18n("Title Image");
 230             break;
 231         case TITLE_FONT_ELEMENT:
 232             clipType = i18n("Title Font");
 233             break;
 234         default:
 235             clipType = i18n("Video clip");
 236         }
 237         QTreeWidgetItem *item = new QTreeWidgetItem(m_ui.treeWidget, QStringList() << clipType);
 238         if (t == TITLE_IMAGE_ELEMENT) {
 239             item->setIcon(0, KIcon("dialog-warning"));
 240             item->setToolTip(1, e.attribute("name"));
 241             item->setText(1, e.attribute("resource"));
 242             item->setData(0, statusRole, CLIPPLACEHOLDER);
 243             item->setData(0, typeOriginalResource, e.attribute("resource"));
 244         } else if (t == TITLE_FONT_ELEMENT) {
 245             item->setIcon(0, KIcon("dialog-warning"));
 246             item->setToolTip(1, e.attribute("name"));
 247             QString ft = e.attribute("resource");
 248             QString newft = QFontInfo(QFont(ft)).family();
 249             item->setText(1, i18n("%1 will be replaced by %2", ft, newft));
 250             item->setData(0, statusRole, CLIPPLACEHOLDER);
 251         } else {
 252             item->setIcon(0, KIcon("dialog-close"));
 253             item->setText(1, e.attribute("resource"));
 254             item->setData(0, hashRole, e.attribute("file_hash"));
 255             item->setData(0, sizeRole, e.attribute("file_size"));
 256             item->setData(0, statusRole, CLIPMISSING);
 257         }
 258         item->setData(0, typeRole, t);
 259         item->setData(0, idRole, e.attribute("id"));
 260         item->setToolTip(0, i18n("Missing item"));
 261     }
 262
 263     if (m_missingClips.count() > 0) {
 264         if (wrongDurationClips.count() > 0) {
 265             m_ui.infoLabel->setText(i18n("The project file contains missing clips or files and clip duration mismatch"));
 266         }
 267         else {
 268             m_ui.infoLabel->setText(i18n("The project file contains missing clips or files"));
 269         }
 270     }
 271     else if (wrongDurationClips.count() > 0) {
 272         m_ui.infoLabel->setText(i18n("The project file contains clips with duration mismatch"));
 273     }
 274     if (missingProxies.count() > 0) {
 275         if (!m_ui.infoLabel->text().isEmpty()) m_ui.infoLabel->setText(m_ui.infoLabel->text() + ". ");
 276         m_ui.infoLabel->setText(m_ui.infoLabel->text() + i18n("Missing proxies will be recreated after opening."));
 277     }
 278     if (missingSources.count() > 0) {
 279         if (!m_ui.infoLabel->text().isEmpty()) m_ui.infoLabel->setText(m_ui.infoLabel->text() + ". ");
 280         m_ui.infoLabel->setText(m_ui.infoLabel->text() + i18np("The project file contains a missing clip, you can still work with its proxy.", "The project file contains missing clips, you can still work with their proxies.", missingSources.count()));
 281     }
 282
 283     m_ui.removeSelected->setEnabled(!m_missingClips.isEmpty());
 284     m_ui.recursiveSearch->setEnabled(!m_missingClips.isEmpty() || !missingLumas.isEmpty() || !missingSources.isEmpty());
 285     m_ui.usePlaceholders->setEnabled(!m_missingClips.isEmpty());
 286     m_ui.fixDuration->setEnabled(!wrongDurationClips.isEmpty());
 287
 288     max = wrongDurationClips.count();
 289     for (int i = 0; i < max; ++i) {
 290         e = wrongDurationClips.at(i).toElement();
 291         QString clipType;
 292         int t = e.attribute("type").toInt();
 293         switch (t) {
 294         case AV:
 295             clipType = i18n("Video clip");
 296             break;
 297         case VIDEO:
 298             clipType = i18n("Mute video clip");
 299             break;
 300         case AUDIO:
 301             clipType = i18n("Audio clip");
 302             break;
 303         case PLAYLIST:
 304             clipType = i18n("Playlist clip");
 305             break;
 306         case IMAGE:
 307             clipType = i18n("Image clip");
 308             break;
 309         case SLIDESHOW:
 310             clipType = i18n("Slideshow clip");
 311             break;
 312         default:
 313             clipType = i18n("Video clip");
 314         }
 315         QTreeWidgetItem *item = new QTreeWidgetItem(m_ui.treeWidget, QStringList() << clipType);
 316         item->setIcon(0, KIcon("timeadjust"));
 317         item->setText(1, e.attribute("resource"));
 318         item->setData(0, hashRole, e.attribute("file_hash"));
 319         item->setData(0, sizeRole, e.attribute("_mismatch"));
 320         e.removeAttribute("_mismatch");
 321         item->setData(0, resetDurationRole, (int) e.hasAttribute("_resetDuration"));
 322         e.removeAttribute("_resetDuration");
 323         item->setData(0, statusRole, CLIPWRONGDURATION);
 324         item->setData(0, typeRole, t);
 325         item->setData(0, idRole, e.attribute("id"));
 326         item->setToolTip(0, i18n("Duration mismatch"));
 327     }
 328
 329     // Check missing proxies
 330     max = missingProxies.count();
 331     if (max > 0) {
 332         QTreeWidgetItem *item = new QTreeWidgetItem(m_ui.treeWidget, QStringList() << i18n("Proxy clip"));
 333         item->setIcon(0, KIcon("dialog-warning"));
 334         item->setText(1, i18n("%1 missing proxy clips, will be recreated on project opening", max));
 335         item->setData(0, hashRole, e.attribute("file_hash"));
 336         item->setData(0, statusRole, PROXYMISSING);
 337         item->setToolTip(0, i18n("Missing proxy"));
 338     }
 339
 340     for (int i = 0; i < max; ++i) {
 341         e = missingProxies.at(i).toElement();
 342         QString realPath = e.attribute("resource");
 343         QString id = e.attribute("id");
 344         // Tell Kdenlive to recreate proxy
 345         e.setAttribute("_replaceproxy", "1");
 346         // Replace proxy url with real clip in MLT producers
 347         QDomNodeList properties;
 348         QDomElement mltProd;
 349         QDomElement property;
 350         int prodsCount = documentProducers.count();
 351         for (int j = 0; j < prodsCount; j++) {
 352             mltProd = documentProducers.at(j).toElement();
 353             QString prodId = mltProd.attribute("id");
 354             bool slowmotion = false;
 355             if (prodId.startsWith("slowmotion")) {
 356                 slowmotion = true;
 357                 prodId = prodId.section(':', 1, 1);
 358             }
 359             if (prodId.contains('_')) prodId = prodId.section('_', 0, 0);
 360             if (prodId == id) {
 361                 // Hit, we must replace url
 362                 properties = mltProd.childNodes();
 363                 for (int k = 0; k < properties.count(); ++k) {
 364                     property = properties.item(k).toElement();
 365                     if (property.attribute("name") == "resource") {
 366                         QString resource = property.firstChild().nodeValue();
 367                         QString suffix;
 368                         if (slowmotion) suffix = '?' + resource.section('?', -1);
 369                         property.firstChild().setNodeValue(realPath + suffix);
 370                         break;
 371                     }
 372                 }
 373             }
 374         }
 375     }
 376
 377     if (max > 0) {
 378         // original doc was modified
 379         QDomElement infoXml = m_doc.elementsByTagName("kdenlivedoc").at(0).toElement();
 380         infoXml.setAttribute("modified", "1");
 381     }
 382
 383     // Check clips with available proxies but missing original source clips
 384     max = missingSources.count();
 385     if (max > 0) {
 386         QTreeWidgetItem *item = new QTreeWidgetItem(m_ui.treeWidget, QStringList() << i18n("Source clip"));
 387         item->setIcon(0, KIcon("dialog-warning"));
 388         item->setText(1, i18n("%1 missing source clips, you can only use the proxies", max));
 389         item->setData(0, hashRole, e.attribute("file_hash"));
 390         item->setData(0, statusRole, SOURCEMISSING);
 391         item->setToolTip(0, i18n("Missing source clip"));
 392         for (int i = 0; i < max; ++i) {
 393             e = missingSources.at(i).toElement();
 394             QString clipType;
 395             QString realPath = e.attribute("resource");
 396             QString id = e.attribute("id");
 397             // Tell Kdenlive the source is missing
 398             e.setAttribute("_missingsource", "1");
 399             QTreeWidgetItem *subitem = new QTreeWidgetItem(item, QStringList() << i18n("Source clip"));
 400             kDebug()<<"// Adding missing source clip: "<<realPath;
 401             subitem->setIcon(0, KIcon("dialog-close"));
 402             subitem->setText(1, realPath);
 403             subitem->setData(0, hashRole, e.attribute("file_hash"));
 404             subitem->setData(0, sizeRole, e.attribute("file_size"));
 405             subitem->setData(0, statusRole, CLIPMISSING);
 406             int t = e.attribute("type").toInt();
 407             subitem->setData(0, typeRole, t);
 408             subitem->setData(0, idRole, id);
 409         }
 410     }
 411
 412     if (max > 0) {
 413         // original doc was modified
 414         QDomElement infoXml = m_doc.elementsByTagName("kdenlivedoc").at(0).toElement();
 415         infoXml.setAttribute("modified", "1");
 416     }
 417
 418     connect(m_ui.recursiveSearch, SIGNAL(pressed()), this, SLOT(slotSearchClips()));
 419     connect(m_ui.usePlaceholders, SIGNAL(pressed()), this, SLOT(slotPlaceholders()));
 420     connect(m_ui.removeSelected, SIGNAL(pressed()), this, SLOT(slotDeleteSelected()));
 421     connect(m_ui.fixDuration, SIGNAL(pressed()), this, SLOT(slotFixDuration()));
 422     connect(m_ui.treeWidget, SIGNAL(itemDoubleClicked(QTreeWidgetItem*,int)), this, SLOT(slotEditItem(QTreeWidgetItem*,int)));
 423     connect(m_ui.treeWidget, SIGNAL(itemSelectionChanged()), this, SLOT(slotCheckButtons()));
 424     //adjustSize();
 425     if (m_ui.treeWidget->topLevelItem(0)) m_ui.treeWidget->setCurrentItem(m_ui.treeWidget->topLevelItem(0));
 426     checkStatus();
 427     int acceptMissing = m_dialog->exec();
 428     if (acceptMissing == QDialog::Accepted) acceptDialog();
 429     return (acceptMissing != QDialog::Accepted);
 430 }
 431
 432 DocumentChecker::~DocumentChecker()
 433 {
 434     delete m_dialog;
 435 }
 436
 437
 438 QString DocumentChecker::getProperty(QDomElement effect, const QString &name)
 439 {
 440     QDomNodeList params = effect.elementsByTagName("property");
 441     for (int i = 0; i < params.count(); ++i) {
 442         QDomElement e = params.item(i).toElement();
 443         if (e.attribute("name") == name) {
 444             return e.firstChild().nodeValue();
 445         }
 446     }
 447     return QString();
 448 }
 449
 450 void DocumentChecker::setProperty(QDomElement effect, const QString &name, const QString &value)
 451 {
 452     QDomNodeList params = effect.elementsByTagName("property");
 453     for (int i = 0; i < params.count(); ++i) {
 454         QDomElement e = params.item(i).toElement();
 455         if (e.attribute("name") == name) {
 456             e.firstChild().setNodeValue(value);
 457         }
 458     }
 459 }
 460
 461 void DocumentChecker::slotSearchClips()
 462 {
 463     QString newpath = KFileDialog::getExistingDirectory(KUrl("kfiledialog:///clipfolder"), kapp->activeWindow(), i18n("Clips folder"));
 464     if (newpath.isEmpty()) return;
 465     int ix = 0;
 466     bool fixed = false;
 467     m_ui.recursiveSearch->setChecked(true);
 468     qApp->processEvents();
 469     QTreeWidgetItem *child = m_ui.treeWidget->topLevelItem(ix);
 470     QDir searchDir(newpath);
 471     while (child) {
 472         if (child->data(0, statusRole).toInt() == SOURCEMISSING) {
 473             for (int j = 0; j < child->childCount(); j++) {
 474                 QTreeWidgetItem *subchild = child->child(j);
 475                 QString clipPath = searchFileRecursively(searchDir, subchild->data(0, sizeRole).toString(), subchild->data(0, hashRole).toString());
 476                 if (!clipPath.isEmpty()) {
 477                     fixed = true;
 478
 479                     subchild->setText(1, clipPath);
 480                     subchild->setIcon(0, KIcon("dialog-ok"));
 481                     subchild->setData(0, statusRole, CLIPOK);
 482                 }
 483             }
 484         }
 485         else if (child->data(0, statusRole).toInt() == CLIPMISSING) {
 486             QString clipPath = searchFileRecursively(searchDir, child->data(0, sizeRole).toString(), child->data(0, hashRole).toString());
 487             if (!clipPath.isEmpty()) {
 488                 fixed = true;
 489                 child->setText(1, clipPath);
 490                 child->setIcon(0, KIcon("dialog-ok"));
 491                 child->setData(0, statusRole, CLIPOK);
 492             }
 493         } else if (child->data(0, statusRole).toInt() == LUMAMISSING) {
 494             QString fileName = searchLuma(searchDir, child->data(0, idRole).toString());
 495             if (!fileName.isEmpty()) {
 496                 fixed = true;
 497                 child->setText(1, fileName);
 498                 child->setIcon(0, KIcon("dialog-ok"));
 499                 child->setData(0, statusRole, LUMAOK);
 500             }
 501         }
 502         else if (child->data(0, typeRole).toInt() == TITLE_IMAGE_ELEMENT && child->data(0, statusRole).toInt() == CLIPPLACEHOLDER) {
 503             // Search missing title images
 504             QString missingFileName = KUrl(child->text(1)).fileName();
 505             QString newPath = searchPathRecursively(searchDir, missingFileName);
 506             if (!newPath.isEmpty()) {
 507                 // File found
 508                 fixed = true;
 509                 child->setText(1, newPath);
 510                 child->setIcon(0, KIcon("dialog-ok"));
 511                 child->setData(0, statusRole, CLIPOK);
 512             }
 513         }
 514         ix++;
 515         child = m_ui.treeWidget->topLevelItem(ix);
 516     }
 517     m_ui.recursiveSearch->setChecked(false);
 518     m_ui.recursiveSearch->setEnabled(true);
 519     if (fixed) {
 520         // original doc was modified
 521         QDomElement infoXml = m_doc.elementsByTagName("kdenlivedoc").at(0).toElement();
 522         infoXml.setAttribute("modified", "1");
 523     }
 524     checkStatus();
 525 }
 526
 527
 528 QString DocumentChecker::searchLuma(const QDir &dir, const QString &file) const
 529 {
 530     KUrl searchPath(KdenliveSettings::mltpath());
 531     QString fname = KUrl(file).fileName();
 532     if (file.contains("PAL"))
 533         searchPath.cd("../lumas/PAL");
 534     else
 535         searchPath.cd("../lumas/NTSC");
 536     QString result = searchPath.path(KUrl::AddTrailingSlash) + fname;
 537     if (QFile::exists(result))
 538         return result;
 539     // try to find luma in application path
 540     searchPath.clear();
 541     searchPath = KUrl(QCoreApplication::applicationDirPath());
 542     searchPath.cd("../share/apps/kdenlive/lumas");
 543     result = searchPath.path(KUrl::AddTrailingSlash) + fname;
 544     if (QFile::exists(result))
 545         return result;
 546     // Try in Kdenlive's standard KDE path
 547     result = KStandardDirs::locate("appdata", "lumas/" + fname);
 548     if (!result.isEmpty()) return result;
 549     // Try in user's chosen folder
 550     return searchPathRecursively(dir, fname);
 551 }
 552
 553 QString DocumentChecker::searchPathRecursively(const QDir &dir, const QString &fileName) const
 554 {
 555     QString foundFileName;
 556     QStringList filters;
 557     filters << fileName;
 558     QDir searchDir(dir);
 559     searchDir.setNameFilters(filters);
 560     QStringList filesAndDirs = searchDir.entryList(QDir::Files | QDir::Readable);
 561     if (!filesAndDirs.isEmpty()) return searchDir.absoluteFilePath(filesAndDirs.at(0));
 562     searchDir.setNameFilters(QStringList());
 563     filesAndDirs = searchDir.entryList(QDir::Dirs | QDir::Readable | QDir::Executable | QDir::NoDotAndDotDot);
 564     for (int i = 0; i < filesAndDirs.size() && foundFileName.isEmpty(); ++i) {
 565         foundFileName = searchPathRecursively(searchDir.absoluteFilePath(filesAndDirs.at(i)), fileName);
 566         if (!foundFileName.isEmpty())
 567             break;
 568     }
 569     return foundFileName;
 570 }
 571
 572 QString DocumentChecker::searchFileRecursively(const QDir &dir, const QString &matchSize, const QString &matchHash) const
 573 {
 574     QString foundFileName;
 575     QByteArray fileData;
 576     QByteArray fileHash;
 577     QStringList filesAndDirs = dir.entryList(QDir::Files | QDir::Readable);
 578     for (int i = 0; i < filesAndDirs.size() && foundFileName.isEmpty(); ++i) {
 579         QFile file(dir.absoluteFilePath(filesAndDirs.at(i)));
 580         if (QString::number(file.size()) == matchSize) {
 581             if (file.open(QIODevice::ReadOnly)) {
 582                 /*
 583                 * 1 MB = 1 second per 450 files (or faster)
 584                 * 10 MB = 9 seconds per 450 files (or faster)
 585                 */
 586                 if (file.size() > 1000000 * 2) {
 587                     fileData = file.read(1000000);
 588                     if (file.seek(file.size() - 1000000))
 589                         fileData.append(file.readAll());
 590                 } else
 591                     fileData = file.readAll();
 592                 file.close();
 593                 fileHash = QCryptographicHash::hash(fileData, QCryptographicHash::Md5);
 594                 if (QString(fileHash.toHex()) == matchHash)
 595                     return file.fileName();
 596             }
 597         }
 598         //kDebug() << filesAndDirs.at(i) << file.size() << fileHash.toHex();
 599     }
 600     filesAndDirs = dir.entryList(QDir::Dirs | QDir::Readable | QDir::Executable | QDir::NoDotAndDotDot);
 601     for (int i = 0; i < filesAndDirs.size() && foundFileName.isEmpty(); ++i) {
 602         foundFileName = searchFileRecursively(dir.absoluteFilePath(filesAndDirs.at(i)), matchSize, matchHash);
 603         if (!foundFileName.isEmpty())
 604             break;
 605     }
 606     return foundFileName;
 607 }
 608
 609 void DocumentChecker::slotEditItem(QTreeWidgetItem *item, int)
 610 {
 611     int t = item->data(0, typeRole).toInt();
 612     if (t == TITLE_FONT_ELEMENT || t == UNKNOWN) return;
 613     //|| t == TITLE_IMAGE_ELEMENT) {
 614
 615     KUrl url = KUrlRequesterDialog::getUrl(item->text(1), m_dialog, i18n("Enter new location for file"));
 616     if (url.isEmpty()) return;
 617     item->setText(1, url.path());
 618     if (KIO::NetAccess::exists(url, KIO::NetAccess::SourceSide, 0)) {
 619         item->setIcon(0, KIcon("dialog-ok"));
 620         int id = item->data(0, statusRole).toInt();
 621         if (id < 10) item->setData(0, statusRole, CLIPOK);
 622         else item->setData(0, statusRole, LUMAOK);
 623         checkStatus();
 624     } else {
 625         item->setIcon(0, KIcon("dialog-close"));
 626         int id = item->data(0, statusRole).toInt();
 627         if (id < 10) item->setData(0, statusRole, CLIPMISSING);
 628         else item->setData(0, statusRole, LUMAMISSING);
 629         checkStatus();
 630     }
 631 }
 632
 633
 634 void DocumentChecker::acceptDialog()
 635 {
 636     QDomNodeList producers = m_doc.elementsByTagName("producer");
 637     QDomNodeList infoproducers = m_doc.elementsByTagName("kdenlive_producer");
 638     int ix = 0;
 639
 640     // prepare transitions
 641     QDomNodeList trans = m_doc.elementsByTagName("transition");
 642
 643     // Mark document as modified
 644     m_doc.documentElement().setAttribute("modified", 1);
 645
 646     QTreeWidgetItem *child = m_ui.treeWidget->topLevelItem(ix);
 647     while (child) {
 648         if (child->data(0, statusRole).toInt() == SOURCEMISSING) {
 649             for (int j = 0; j < child->childCount(); j++) {
 650                 fixClipItem(child->child(j), producers, infoproducers, trans);
 651             }
 652         }
 653         else fixClipItem(child, producers, infoproducers, trans);
 654         ix++;
 655         child = m_ui.treeWidget->topLevelItem(ix);
 656     }
 657     //QDialog::accept();
 658 }
 659
 660 void DocumentChecker::fixClipItem(QTreeWidgetItem *child, QDomNodeList producers, QDomNodeList infoproducers, QDomNodeList trans)
 661 {
 662     QDomElement e, property;
 663     QDomNodeList properties;
 664     int t = child->data(0, typeRole).toInt();
 665     if (child->data(0, statusRole).toInt() == CLIPOK) {
 666         QString id = child->data(0, idRole).toString();
 667         if (t == TITLE_IMAGE_ELEMENT) {
 668             // edit images embedded in titles
 669             for (int i = 0; i < infoproducers.count(); ++i) {
 670                 e = infoproducers.item(i).toElement();
 671                 if (e.attribute("id") == id) {
 672                     // Fix clip
 673                     QString xml = e.attribute("xmldata");
 674                     xml.replace(child->data(0, typeOriginalResource).toString(), child->text(1));
 675                     e.setAttribute("xmldata", xml);
 676                     break;
 677                 }
 678             }
 679             for (int i = 0; i < producers.count(); ++i) {
 680                 e = producers.item(i).toElement();
 681                 if (e.attribute("id").section('_', 0, 0) == id) {
 682                     // Fix clip
 683                     properties = e.childNodes();
 684                     for (int j = 0; j < properties.count(); ++j) {
 685                         property = properties.item(j).toElement();
 686                         if (property.attribute("name") == "xmldata") {
 687                             QString xml = property.firstChild().nodeValue();
 688                             xml.replace(child->data(0, typeOriginalResource).toString(), child->text(1));
 689                             property.firstChild().setNodeValue(xml);
 690                             break;
 691                         }
 692                     }
 693                 }
 694             }
 695         } else {
 696             // edit clip url
 697             for (int i = 0; i < infoproducers.count(); ++i) {
 698                 e = infoproducers.item(i).toElement();
 699                 if (e.attribute("id") == id) {
 700                     // Fix clip
 701                     e.setAttribute("resource", child->text(1));
 702                     e.setAttribute("name", KUrl(child->text(1)).fileName());
 703                     e.removeAttribute("_missingsource");
 704                     break;
 705                 }
 706             }
 707             for (int i = 0; i < producers.count(); ++i) {
 708                 e = producers.item(i).toElement();
 709                 if (e.attribute("id").section('_', 0, 0) == id || e.attribute("id").section(':', 1, 1) == id) {
 710                     // Fix clip
 711                     properties = e.childNodes();
 712                     for (int j = 0; j < properties.count(); ++j) {
 713                         property = properties.item(j).toElement();
 714                         if (property.attribute("name") == "resource") {
 715                             QString resource = property.firstChild().nodeValue();
 716                             if (resource.contains(QRegExp("\\?[0-9]+\\.[0-9]+(&amp;strobe=[0-9]+)?$")))
 717                                 property.firstChild().setNodeValue(child->text(1) + '?' + resource.section('?', -1));
 718                             else
 719                                 property.firstChild().setNodeValue(child->text(1));
 720                             break;
 721                         }
 722                     }
 723                 }
 724             }
 725         }
 726     } else if (child->data(0, statusRole).toInt() == CLIPPLACEHOLDER && t != TITLE_FONT_ELEMENT && t != TITLE_IMAGE_ELEMENT) {
 727         QString id = child->data(0, idRole).toString();
 728         for (int i = 0; i < infoproducers.count(); ++i) {
 729             e = infoproducers.item(i).toElement();
 730             if (e.attribute("id") == id) {
 731                 // Fix clip
 732                 e.setAttribute("placeholder", '1');
 733                 break;
 734             }
 735         }
 736     } else if (child->data(0, statusRole).toInt() == LUMAOK) {
 737         for (int i = 0; i < trans.count(); ++i) {
 738             QString luma = getProperty(trans.at(i).toElement(), "luma");
 739             if (!luma.isEmpty() && luma == child->data(0, idRole).toString()) {
 740                 setProperty(trans.at(i).toElement(), "luma", child->text(1));
 741                 kDebug() << "replace with; " << child->text(1);
 742             }
 743         }
 744     } else if (child->data(0, statusRole).toInt() == LUMAMISSING) {
 745         for (int i = 0; i < trans.count(); ++i) {
 746             QString luma = getProperty(trans.at(i).toElement(), "luma");
 747             if (!luma.isEmpty() && luma == child->data(0, idRole).toString()) {
 748                 setProperty(trans.at(i).toElement(), "luma", QString());
 749             }
 750         }
 751     }
 752 }
 753
 754 void DocumentChecker::slotPlaceholders()
 755 {
 756     int ix = 0;
 757     QTreeWidgetItem *child = m_ui.treeWidget->topLevelItem(ix);
 758     while (child) {
 759         if (child->data(0, statusRole).toInt() == CLIPMISSING) {
 760             child->setData(0, statusRole, CLIPPLACEHOLDER);
 761             child->setIcon(0, KIcon("dialog-ok"));
 762         } else if (child->data(0, statusRole).toInt() == LUMAMISSING) {
 763             child->setData(0, statusRole, LUMAPLACEHOLDER);
 764             child->setIcon(0, KIcon("dialog-ok"));
 765         }
 766         ix++;
 767         child = m_ui.treeWidget->topLevelItem(ix);
 768     }
 769     checkStatus();
 770 }
 771
 772 void DocumentChecker::slotFixDuration()
 773 {
 774     int ix = 0;
 775     QTreeWidgetItem *child = m_ui.treeWidget->topLevelItem(ix);
 776     QDomNodeList documentProducers = m_doc.elementsByTagName("producer");
 777     while (child) {
 778         if (child->data(0, statusRole).toInt() == CLIPWRONGDURATION) {
 779             QString id = child->data(0, idRole).toString();
 780             bool resetDuration = child->data(0, resetDurationRole).toInt();
 781
 782             for (int i = 0; i < m_info.count(); ++i) {
 783                 QDomElement e = m_info.at(i).toElement();
 784                 if (e.attribute("id") == id) {
 785                     if (m_missingClips.contains(e)) {
 786                         // we cannot fix duration of missing clips
 787                         resetDuration = false;
 788                     }
 789                     else {
 790                         if (resetDuration) e.removeAttribute("duration");
 791                         else e.setAttribute("duration", child->data(0, sizeRole).toString());
 792                         child->setData(0, statusRole, CLIPOK);
 793                         child->setIcon(0, KIcon("dialog-ok"));
 794                     }
 795                     break;
 796                 }
 797             }
 798             if (resetDuration) {
 799                 // something is wrong in clip durations, so remove them so mlt fetches them again
 800                 for (int j = 0; j < documentProducers.count(); j++) {
 801                     QDomElement mltProd = documentProducers.at(j).toElement();
 802                     QString prodId = mltProd.attribute("id");
 803                     if (prodId == id || prodId.startsWith(id + '_')) {
 804                         EffectsList::removeProperty(mltProd, "length");
 805                     }
 806                 }
 807             }
 808         }
 809         ix++;
 810         child = m_ui.treeWidget->topLevelItem(ix);
 811     }
 812     QDomElement infoXml = m_doc.elementsByTagName("kdenlivedoc").at(0).toElement();
 813     infoXml.setAttribute("modified", "1");
 814     m_ui.fixDuration->setEnabled(false);
 815     checkStatus();
 816 }
 817
 818
 819 void DocumentChecker::checkStatus()
 820 {
 821     bool status = true;
 822     int ix = 0;
 823     QTreeWidgetItem *child = m_ui.treeWidget->topLevelItem(ix);
 824     while (child) {
 825         int status = child->data(0, statusRole).toInt();
 826         if (status == CLIPMISSING || status == LUMAMISSING || status == CLIPWRONGDURATION) {
 827             status = false;
 828             break;
 829         }
 830         ix++;
 831         child = m_ui.treeWidget->topLevelItem(ix);
 832     }
 833     m_ui.buttonBox->button(QDialogButtonBox::Ok)->setEnabled(status);
 834 }
 835
 836
 837 void DocumentChecker::slotDeleteSelected()
 838 {
 839     if (KMessageBox::warningContinueCancel(m_dialog, i18np("This will remove the selected clip from this project", "This will remove the selected clips from this project", m_ui.treeWidget->selectedItems().count()), i18n("Remove clips")) == KMessageBox::Cancel)
 840         return;
 841     QStringList deletedIds;
 842     QStringList deletedLumas;
 843     QDomNodeList playlists = m_doc.elementsByTagName("playlist");
 844
 845     foreach(QTreeWidgetItem *child, m_ui.treeWidget->selectedItems()) {
 846         int id = child->data(0, statusRole).toInt();
 847         if (id == CLIPMISSING) {
 848             deletedIds.append(child->data(0, idRole).toString());
 849             delete child;
 850         }
 851         else if (id == LUMAMISSING) {
 852             deletedLumas.append(child->data(0, idRole).toString());
 853             delete child;
 854         }
 855     }
 856
 857     if (!deletedLumas.isEmpty()) {
 858         QDomElement e;
 859         QDomNodeList transitions = m_doc.elementsByTagName("transition");
 860         foreach (const QString &lumaPath, deletedLumas) {
 861             for (int i = 0; i < transitions.count(); ++i) {
 862                 e = transitions.item(i).toElement();
 863                 QString resource = EffectsList::property(e, "luma");
 864                 if (resource == lumaPath) EffectsList::removeProperty(e, "luma");
 865             }
 866         }
 867     }
 868
 869     if (!deletedIds.isEmpty()) {
 870         QDomElement e;
 871         QDomNodeList producers = m_doc.elementsByTagName("producer");
 872         QDomNodeList infoproducers = m_doc.elementsByTagName("kdenlive_producer");
 873
 874         QDomNode mlt = m_doc.elementsByTagName("mlt").at(0);
 875         QDomNode kdenlivedoc = m_doc.elementsByTagName("kdenlivedoc").at(0);
 876
 877         for (int i = 0, j = 0; i < infoproducers.count() && j < deletedIds.count(); ++i) {
 878             e = infoproducers.item(i).toElement();
 879             if (deletedIds.contains(e.attribute("id"))) {
 880                 // Remove clip
 881                 kdenlivedoc.removeChild(e);
 882                 --i;
 883                 j++;
 884             }
 885         }
 886
 887         for (int i = 0; i < producers.count(); ++i) {
 888             e = producers.item(i).toElement();
 889             if (deletedIds.contains(e.attribute("id").section('_', 0, 0)) || deletedIds.contains(e.attribute("id").section(':', 1, 1).section('_', 0, 0))) {
 890                 // Remove clip
 891                 mlt.removeChild(e);
 892                 --i;
 893             }
 894         }
 895
 896         for (int i = 0; i < playlists.count(); ++i) {
 897             QDomNodeList entries = playlists.at(i).toElement().elementsByTagName("entry");
 898             for (int j = 0; j < entries.count(); j++) {
 899                 e = entries.item(j).toElement();
 900                 if (deletedIds.contains(e.attribute("producer").section('_', 0, 0)) || deletedIds.contains(e.attribute("producer").section(':', 1, 1).section('_', 0, 0))) {
 901                     // Replace clip with blank
 902                     while (e.childNodes().count() > 0)
 903                         e.removeChild(e.firstChild());
 904                     e.setTagName("blank");
 905                     e.removeAttribute("producer");
 906                     int length = e.attribute("out").toInt() - e.attribute("in").toInt();
 907                     e.setAttribute("length", length);
 908                     j--;
 909                 }
 910             }
 911         }
 912         QDomElement infoXml = m_doc.elementsByTagName("kdenlivedoc").at(0).toElement();
 913         infoXml.setAttribute("modified", "1");
 914         checkStatus();
 915     }
 916 }
 917
 918 void DocumentChecker::checkMissingImagesAndFonts(const QStringList &images, const QStringList &fonts, const QString &id, const QString &baseClip)
 919 {
 920     QDomDocument doc;
 921     foreach(const QString &img, images) {
 922         if (m_safeImages.contains(img)) continue;
 923         if (!KIO::NetAccess::exists(KUrl(img), KIO::NetAccess::SourceSide, 0)) {
 924             QDomElement e = doc.createElement("missingclip");
 925             e.setAttribute("type", TITLE_IMAGE_ELEMENT);
 926             e.setAttribute("resource", img);
 927             e.setAttribute("id", id);
 928             e.setAttribute("name", baseClip);
 929             m_missingClips.append(e);
 930         }
 931         else m_safeImages.append(img);
 932     }
 933     foreach(const QString &fontelement, fonts) {
 934         if (m_safeFonts.contains(fontelement)) continue;
 935         QFont f(fontelement);
 936         //kDebug() << "/ / / CHK FONTS: " << fontelement << " = " << QFontInfo(f).family();
 937         if (fontelement != QFontInfo(f).family()) {
 938             QDomElement e = doc.createElement("missingclip");
 939             e.setAttribute("type", TITLE_FONT_ELEMENT);
 940             e.setAttribute("resource", fontelement);
 941             e.setAttribute("id", id);
 942             e.setAttribute("name", baseClip);
 943             m_missingClips.append(e);
 944         }
 945         else m_safeFonts.append(fontelement);
 946     }
 947 }
 948
 949
 950 void DocumentChecker::slotCheckButtons()
 951 {
 952     if (m_ui.treeWidget->currentItem()) {
 953         QTreeWidgetItem *item = m_ui.treeWidget->currentItem();
 954         int t = item->data(0, typeRole).toInt();
 955         int s = item->data(0, statusRole).toInt();
 956         if (t == TITLE_FONT_ELEMENT || t == TITLE_IMAGE_ELEMENT || s == PROXYMISSING) {
 957             m_ui.removeSelected->setEnabled(false);
 958         } else m_ui.removeSelected->setEnabled(true);
 959     }
 960
 961 }
 962
 963 #include "documentchecker.moc"
 964
 965