1 /***************************************************************************
2 * Copyright (C) 2008 by Jean-Baptiste Mardelle (jb@kdenlive.org) *
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. *
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. *
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 ***************************************************************************/
21 #include "documentchecker.h"
23 #include "definitions.h"
24 #include "kdenlivesettings.h"
27 #include <KGlobalSettings>
29 #include <KIO/NetAccess>
30 #include <KFileDialog>
31 #include <KApplication>
32 #include <KUrlRequesterDialog>
33 #include <KMessageBox>
35 #include <QTreeWidgetItem>
37 #include <QHeaderView>
41 #include <QCryptographicHash>
43 const int hashRole = Qt::UserRole;
44 const int sizeRole = Qt::UserRole + 1;
45 const int idRole = Qt::UserRole + 2;
46 const int statusRole = Qt::UserRole + 3;
48 const int CLIPMISSING = 0;
50 const int CLIPPLACEHOLDER = 2;
51 const int LUMAMISSING = 10;
52 const int LUMAOK = 11;
53 const int LUMAPLACEHOLDER = 12;
55 DocumentChecker::DocumentChecker(QList <QDomElement> missingClips, QDomDocument doc, QWidget * parent) :
59 setFont(KGlobalSettings::toolBarFont());
62 QStringList missingLumas;
63 QDomNodeList trans = doc.elementsByTagName("transition");
64 for (int i = 0; i < trans.count(); i++) {
65 QString luma = getProperty(trans.at(i).toElement(), "luma");
66 if (!luma.isEmpty() && !QFile::exists(luma)) {
67 if (!missingLumas.contains(luma)) {
68 missingLumas.append(luma);
69 QTreeWidgetItem *item = new QTreeWidgetItem(treeWidget, QStringList() << i18n("Luma file") << luma);
70 item->setIcon(0, KIcon("dialog-close"));
71 item->setData(0, idRole, luma);
72 item->setData(0, statusRole, LUMAMISSING);
77 buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);
78 for (int i = 0; i < missingClips.count(); i++) {
79 e = missingClips.at(i).toElement();
81 switch (e.attribute("type").toInt()) {
83 clipType = i18n("Video clip");
86 clipType = i18n("Mute video clip");
89 clipType = i18n("Audio clip");
92 clipType = i18n("Playlist clip");
95 clipType = i18n("Image clip");
98 clipType = i18n("Slideshow clip");
101 clipType = i18n("Video clip");
103 QTreeWidgetItem *item = new QTreeWidgetItem(treeWidget, QStringList() << clipType << e.attribute("resource"));
104 item->setIcon(0, KIcon("dialog-close"));
105 item->setData(0, hashRole, e.attribute("file_hash"));
106 item->setData(0, sizeRole, e.attribute("file_size"));
107 item->setData(0, idRole, e.attribute("id"));
108 item->setData(0, statusRole, CLIPMISSING);
110 connect(recursiveSearch, SIGNAL(pressed()), this, SLOT(slotSearchClips()));
111 connect(usePlaceholders, SIGNAL(pressed()), this, SLOT(slotPlaceholders()));
112 connect(removeSelected, SIGNAL(pressed()), this, SLOT(slotDeleteSelected()));
113 connect(treeWidget, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(slotEditItem(QTreeWidgetItem *, int)));
117 DocumentChecker::~DocumentChecker() {}
120 QString DocumentChecker::getProperty(QDomElement effect, const QString &name)
122 QDomNodeList params = effect.elementsByTagName("property");
123 for (int i = 0; i < params.count(); i++) {
124 QDomElement e = params.item(i).toElement();
125 if (e.attribute("name") == name) {
126 return e.firstChild().nodeValue();
132 void DocumentChecker::setProperty(QDomElement effect, const QString &name, const QString value)
134 QDomNodeList params = effect.elementsByTagName("property");
135 for (int i = 0; i < params.count(); i++) {
136 QDomElement e = params.item(i).toElement();
137 if (e.attribute("name") == name) {
138 e.firstChild().setNodeValue(value);
143 void DocumentChecker::slotSearchClips()
145 QString newpath = KFileDialog::getExistingDirectory(KUrl("kfiledialog:///clipfolder"), kapp->activeWindow(), i18n("Clips folder"));
146 if (newpath.isEmpty()) return;
148 recursiveSearch->setEnabled(false);
149 QTreeWidgetItem *child = treeWidget->topLevelItem(ix);
151 if (child->data(0, statusRole).toInt() == CLIPMISSING) {
152 QString clipPath = searchFileRecursively(QDir(newpath), child->data(0, sizeRole).toString(), child->data(0, hashRole).toString());
153 if (!clipPath.isEmpty()) {
154 child->setText(1, clipPath);
155 child->setIcon(0, KIcon("dialog-ok"));
156 child->setData(0, statusRole, CLIPOK);
159 else if (child->data(0, statusRole).toInt() == LUMAMISSING) {
160 QString fileName = searchLuma(child->data(0, idRole).toString());
161 if (!fileName.isEmpty()) {
162 child->setText(1, fileName);
163 child->setIcon(0, KIcon("dialog-ok"));
164 child->setData(0, statusRole, LUMAOK);
168 child = treeWidget->topLevelItem(ix);
170 recursiveSearch->setEnabled(true);
175 QString DocumentChecker::searchLuma(QString file) const
177 KUrl searchPath(KdenliveSettings::mltpath());
178 if (file.contains("PAL"))
179 searchPath.cd("../lumas/PAL");
181 searchPath.cd("../lumas/NTSC");
182 QString result = searchPath.path(KUrl::AddTrailingSlash) + KUrl(file).fileName();
183 if (QFile::exists(result))
189 QString DocumentChecker::searchFileRecursively(const QDir &dir, const QString &matchSize, const QString &matchHash) const
191 QString foundFileName;
194 QStringList filesAndDirs = dir.entryList(QDir::Files | QDir::Readable);
195 for (int i = 0; i < filesAndDirs.size() && foundFileName.isEmpty(); i++) {
196 QFile file(dir.absoluteFilePath(filesAndDirs.at(i)));
197 if (QString::number(file.size()) == matchSize) {
198 if (file.open(QIODevice::ReadOnly)) {
200 * 1 MB = 1 second per 450 files (or faster)
201 * 10 MB = 9 seconds per 450 files (or faster)
203 if (file.size() > 1000000*2) {
204 fileData = file.read(1000000);
205 if (file.seek(file.size() - 1000000))
206 fileData.append(file.readAll());
208 fileData = file.readAll();
210 fileHash = QCryptographicHash::hash(fileData, QCryptographicHash::Md5);
211 if (QString(fileHash.toHex()) == matchHash)
212 return file.fileName();
215 //kDebug() << filesAndDirs.at(i) << file.size() << fileHash.toHex();
217 filesAndDirs = dir.entryList(QDir::Dirs | QDir::Readable | QDir::Executable | QDir::NoDotAndDotDot);
218 for (int i = 0; i < filesAndDirs.size() && foundFileName.isEmpty(); i++) {
219 foundFileName = searchFileRecursively(dir.absoluteFilePath(filesAndDirs.at(i)), matchSize, matchHash);
220 if (!foundFileName.isEmpty())
223 return foundFileName;
226 void DocumentChecker::slotEditItem(QTreeWidgetItem *item, int)
228 KUrl url = KUrlRequesterDialog::getUrl(item->text(1), this, i18n("Enter new location for file"));
229 if (url.isEmpty()) return;
230 item->setText(1, url.path());
231 if (KIO::NetAccess::exists(url, KIO::NetAccess::SourceSide, 0)) {
232 item->setIcon(0, KIcon("dialog-ok"));
233 int id = item->data(0, statusRole).toInt();
234 if (id < 10) item->setData(0, statusRole, CLIPOK);
235 else item->setData(0, statusRole, LUMAOK);
241 void DocumentChecker::accept()
243 QDomElement e, property;
244 QDomNodeList producers = m_doc.elementsByTagName("producer");
245 QDomNodeList infoproducers = m_doc.elementsByTagName("kdenlive_producer");
246 QDomNodeList properties;
249 // prepare transitions
250 QDomNodeList trans = m_doc.elementsByTagName("transition");
252 QTreeWidgetItem *child = treeWidget->topLevelItem(ix);
254 if (child->data(0, statusRole).toInt() == CLIPOK) {
255 QString id = child->data(0, idRole).toString();
256 for (int i = 0; i < infoproducers.count(); i++) {
257 e = infoproducers.item(i).toElement();
258 if (e.attribute("id") == id) {
260 e.setAttribute("resource", child->text(1));
261 e.setAttribute("name", KUrl(child->text(1)).fileName());
265 for (int i = 0; i < producers.count(); i++) {
266 e = producers.item(i).toElement();
267 if (e.attribute("id").section('_', 0, 0) == id) {
269 properties = e.childNodes();
270 for (int j = 0; j < properties.count(); ++j) {
271 property = properties.item(j).toElement();
272 if (property.attribute("name") == "resource") {
273 property.firstChild().setNodeValue(child->text(1));
280 } else if (child->data(0, statusRole).toInt() == CLIPPLACEHOLDER) {
281 QString id = child->data(0, idRole).toString();
282 for (int i = 0; i < infoproducers.count(); i++) {
283 e = infoproducers.item(i).toElement();
284 if (e.attribute("id") == id) {
286 e.setAttribute("placeholder", '1');
290 } else if (child->data(0, statusRole).toInt() == LUMAOK) {
291 for (int i = 0; i < trans.count(); i++) {
292 QString luma = getProperty(trans.at(i).toElement(), "luma");
293 kDebug()<< "luma: "<<luma;
294 if (!luma.isEmpty() && luma == child->data(0, idRole).toString()) {
295 setProperty(trans.at(i).toElement(), "luma", child->text(1));
296 kDebug()<< "replace with; "<<child->text(1);
299 } else if (child->data(0, statusRole).toInt() == LUMAMISSING) {
300 for (int i = 0; i < trans.count(); i++) {
301 QString luma = getProperty(trans.at(i).toElement(), "luma");
302 if (!luma.isEmpty() && luma == child->data(0, idRole).toString()) {
303 setProperty(trans.at(i).toElement(), "luma", QString());
308 child = treeWidget->topLevelItem(ix);
313 void DocumentChecker::slotPlaceholders()
316 QTreeWidgetItem *child = treeWidget->topLevelItem(ix);
318 if (child->data(0, statusRole).toInt() == CLIPMISSING) {
319 child->setData(0, statusRole, CLIPPLACEHOLDER);
320 child->setIcon(0, KIcon("dialog-ok"));
322 else if (child->data(0, statusRole).toInt() == LUMAMISSING) {
323 child->setData(0, statusRole, LUMAPLACEHOLDER);
324 child->setIcon(0, KIcon("dialog-ok"));
327 child = treeWidget->topLevelItem(ix);
333 void DocumentChecker::checkStatus()
337 QTreeWidgetItem *child = treeWidget->topLevelItem(ix);
339 if (child->data(0, statusRole).toInt() == CLIPMISSING || child->data(0, statusRole).toInt() == LUMAMISSING) {
344 child = treeWidget->topLevelItem(ix);
346 buttonBox->button(QDialogButtonBox::Ok)->setEnabled(status);
350 void DocumentChecker::slotDeleteSelected()
352 if (KMessageBox::warningContinueCancel(this, i18n("This will remove the selected clips from this project"), i18n("Remove clips")) == KMessageBox::Cancel) return;
354 QStringList deletedIds;
355 QTreeWidgetItem *child = treeWidget->topLevelItem(ix);
356 QDomNodeList playlists = m_doc.elementsByTagName("playlist");
359 int id = child->data(0, statusRole).toInt();
360 if (child->isSelected() && id < 10) {
361 QString id = child->data(0, idRole).toString();
362 deletedIds.append(id);
363 for (int j = 0; j < playlists.count(); j++)
364 deletedIds.append(id + '_' + QString::number(j));
367 child = treeWidget->topLevelItem(ix);
369 kDebug() << "// Clips to delete: " << deletedIds;
371 if (!deletedIds.isEmpty()) {
373 QDomNodeList producers = m_doc.elementsByTagName("producer");
374 QDomNodeList infoproducers = m_doc.elementsByTagName("kdenlive_producer");
376 QDomElement mlt = m_doc.firstChildElement("mlt");
377 QDomElement kdenlivedoc = mlt.firstChildElement("kdenlivedoc");
379 for (int i = 0; i < infoproducers.count(); i++) {
380 e = infoproducers.item(i).toElement();
381 if (deletedIds.contains(e.attribute("id"))) {
383 kdenlivedoc.removeChild(e);
388 for (int i = 0; i < producers.count(); i++) {
389 e = producers.item(i).toElement();
390 if (deletedIds.contains(e.attribute("id"))) {
397 for (int i = 0; i < playlists.count(); i++) {
398 QDomNodeList entries = playlists.at(i).toElement().elementsByTagName("entry");
399 for (int j = 0; j < playlists.count(); j++) {
400 e = entries.item(j).toElement();
401 if (deletedIds.contains(e.attribute("producer"))) {
402 // Replace clip with blank
403 e.setTagName("blank");
404 e.removeAttribute("producer");
405 int length = e.attribute("out").toInt() - e.attribute("in").toInt();
406 e.setAttribute("length", length);
414 #include "documentchecker.moc"