]> git.sesse.net Git - kdenlive/blob - src/clipmanager.cpp
Fix audio thumbs not displayed on project opening and abortion if audio thumbs creation
[kdenlive] / src / clipmanager.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 "clipmanager.h"
22 #include "addclipcommand.h"
23 #include "kdenlivesettings.h"
24 #include "docclipbase.h"
25 #include "kdenlivedoc.h"
26 #include "abstractclipitem.h"
27 #include "abstractgroupitem.h"
28
29 #include <mlt++/Mlt.h>
30
31 #include <KDebug>
32 #include <KMessageBox>
33 #include <KFileDialog>
34 #include <KApplication>
35 #include <kio/netaccess.h>
36
37 #include <QGraphicsItemGroup>
38
39 ClipManager::ClipManager(KdenliveDoc *doc) :
40         QObject(),
41         m_audioThumbsQueue(),
42         m_doc(doc),
43         m_generatingAudioId()
44 {
45     m_clipIdCounter = 1;
46     m_folderIdCounter = 1;
47     connect(&m_fileWatcher, SIGNAL(dirty(const QString &)), this, SLOT(slotClipModified(const QString &)));
48     connect(&m_fileWatcher, SIGNAL(deleted(const QString &)), this, SLOT(slotClipMissing(const QString &)));
49     connect(&m_fileWatcher, SIGNAL(created(const QString &)), this, SLOT(slotClipAvailable(const QString &)));
50 }
51
52 ClipManager::~ClipManager()
53 {
54     qDeleteAll(m_clipList);
55     m_clipList.clear();
56     m_audioThumbsQueue.clear();
57     m_generatingAudioId.clear();
58 }
59
60 void ClipManager::clear()
61 {
62     qDeleteAll(m_clipList);
63     m_clipList.clear();
64     m_clipIdCounter = 1;
65     m_folderIdCounter = 1;
66     m_folderList.clear();
67     m_audioThumbsQueue.clear();
68 }
69
70 void ClipManager::checkAudioThumbs()
71 {
72     if (!KdenliveSettings::audiothumbnails()) {
73       if (!m_generatingAudioId.isEmpty()) {
74             DocClipBase *clip = getClipById(m_generatingAudioId);
75             if (clip) clip->slotClearAudioCache();
76         }
77         m_audioThumbsQueue.clear();
78         m_generatingAudioId.clear();
79         return;
80     }
81     
82     for (int i = 0; i < m_clipList.count(); i++) {
83         m_audioThumbsQueue.append(m_clipList.at(i)->getId());
84     }
85     if (m_generatingAudioId.isEmpty()) startAudioThumbsGeneration();
86 }
87
88 void ClipManager::askForAudioThumb(const QString &id)
89 {
90     DocClipBase *clip = getClipById(id);
91     if (clip && KdenliveSettings::audiothumbnails()) {
92         m_audioThumbsQueue.append(id);
93         if (m_generatingAudioId.isEmpty()) startAudioThumbsGeneration();
94     }
95 }
96
97 void ClipManager::startAudioThumbsGeneration()
98 {
99     if (!KdenliveSettings::audiothumbnails()) {
100         m_audioThumbsQueue.clear();
101         m_generatingAudioId.clear();
102         return;
103     }
104     if (!m_audioThumbsQueue.isEmpty()) {
105         m_generatingAudioId = m_audioThumbsQueue.takeFirst();
106         DocClipBase *clip = getClipById(m_generatingAudioId);
107         if (!clip || !clip->slotGetAudioThumbs())
108             endAudioThumbsGeneration(m_generatingAudioId);
109     } else {
110         m_generatingAudioId.clear();
111     }
112 }
113
114 void ClipManager::endAudioThumbsGeneration(const QString &requestedId)
115 {
116     if (!KdenliveSettings::audiothumbnails()) {
117         m_audioThumbsQueue.clear();
118         m_generatingAudioId.clear();
119         return;
120     }
121     if (!m_audioThumbsQueue.isEmpty()) {
122         if (m_generatingAudioId == requestedId) {
123             startAudioThumbsGeneration();
124         }
125     } else {
126         m_generatingAudioId.clear();
127     }
128 }
129
130 void ClipManager::setThumbsProgress(const QString &message, int progress)
131 {
132     m_doc->setThumbsProgress(message, progress);
133 }
134
135 QList <DocClipBase*> ClipManager::documentClipList() const
136 {
137     return m_clipList;
138 }
139
140 QMap <QString, QString> ClipManager::documentFolderList() const
141 {
142     return m_folderList;
143 }
144
145 void ClipManager::addClip(DocClipBase *clip)
146 {
147     m_clipList.append(clip);
148     if (clip->clipType() != COLOR && clip->clipType() != SLIDESHOW  && !clip->fileURL().isEmpty()) {
149         // listen for file change
150         //kDebug() << "// LISTEN FOR: " << clip->fileURL().path();
151         m_fileWatcher.addFile(clip->fileURL().path());
152     }
153     const QString id = clip->getId();
154     if (id.toInt() >= m_clipIdCounter) m_clipIdCounter = id.toInt() + 1;
155     const QString gid = clip->getProperty("groupid");
156     if (!gid.isEmpty() && gid.toInt() >= m_folderIdCounter) m_folderIdCounter = gid.toInt() + 1;
157 }
158
159 void ClipManager::slotDeleteClips(QStringList ids)
160 {
161     QUndoCommand *delClips = new QUndoCommand();
162     delClips->setText(i18np("Delete clip", "Delete clips", ids.size()));
163
164     for (int i = 0; i < ids.size(); i++) {
165         DocClipBase *clip = getClipById(ids.at(i));
166         if (clip) {
167             new AddClipCommand(m_doc, clip->toXML(), ids.at(i), false, delClips);
168         }
169     }
170     m_doc->commandStack()->push(delClips);
171 }
172
173 void ClipManager::deleteClip(const QString &clipId)
174 {
175     for (int i = 0; i < m_clipList.count(); i++) {
176         if (m_clipList.at(i)->getId() == clipId) {
177             if (m_clipList.at(i)->clipType() != COLOR && m_clipList.at(i)->clipType() != SLIDESHOW  && !m_clipList.at(i)->fileURL().isEmpty()) {
178                 //if (m_clipList.at(i)->clipType() == IMAGE || m_clipList.at(i)->clipType() == AUDIO || (m_clipList.at(i)->clipType() == TEXT && !m_clipList.at(i)->fileURL().isEmpty())) {
179                 // listen for file change
180                 m_fileWatcher.removeFile(m_clipList.at(i)->fileURL().path());
181             }
182             DocClipBase *clip = m_clipList.takeAt(i);
183             delete clip;
184             clip = NULL;
185             break;
186         }
187     }
188 }
189
190 DocClipBase *ClipManager::getClipAt(int pos)
191 {
192     return m_clipList.at(pos);
193 }
194
195 DocClipBase *ClipManager::getClipById(QString clipId)
196 {
197     //kDebug() << "++++  CLIP MAN, LOOKING FOR CLIP ID: " << clipId;
198     clipId = clipId.section('_', 0, 0);
199     for (int i = 0; i < m_clipList.count(); i++) {
200         if (m_clipList.at(i)->getId() == clipId) {
201             //kDebug() << "++++  CLIP MAN, FOUND FOR CLIP ID: " << clipId;
202             return m_clipList.at(i);
203         }
204     }
205     return NULL;
206 }
207
208 const QList <DocClipBase *> ClipManager::getClipByResource(QString resource)
209 {
210     QList <DocClipBase *> list;
211     QString clipResource;
212     for (int i = 0; i < m_clipList.count(); i++) {
213         clipResource = m_clipList.at(i)->getProperty("resource");
214         if (clipResource.isEmpty()) clipResource = m_clipList.at(i)->getProperty("colour");
215         if (clipResource == resource) {
216             list.append(m_clipList.at(i));
217         }
218     }
219     return list;
220 }
221
222 void ClipManager::updatePreviewSettings()
223 {
224     for (int i = 0; i < m_clipList.count(); i++) {
225         if (m_clipList.at(i)->clipType() == AV || m_clipList.at(i)->clipType() == VIDEO) {
226             if (m_clipList.at(i)->producerProperty("meta.media.0.codec.name") && strcmp(m_clipList.at(i)->producerProperty("meta.media.0.codec.name"), "h264") == 0) {
227                 if (KdenliveSettings::dropbframes()) {
228                     m_clipList[i]->setProducerProperty("skip_loop_filter", "all");
229                     m_clipList[i]->setProducerProperty("skip_frame", "bidir");
230                 } else {
231                     m_clipList[i]->setProducerProperty("skip_loop_filter", "");
232                     m_clipList[i]->setProducerProperty("skip_frame", "");
233                 }
234             }
235         }
236     }
237 }
238
239 void ClipManager::clearUnusedProducers()
240 {
241     for (int i = 0; i < m_clipList.count(); i++) {
242         if (m_clipList.at(i)->numReferences() == 0) m_clipList.at(i)->deleteProducers();
243     }
244 }
245
246 void ClipManager::resetProducersList(const QList <Mlt::Producer *> prods)
247 {
248     for (int i = 0; i < m_clipList.count(); i++) {
249         if (m_clipList.at(i)->numReferences() > 0) {
250             m_clipList.at(i)->clearProducers();
251         }
252     }
253     QString id;
254     for (int i = 0; i < prods.count(); i++) {
255         id = prods.at(i)->get("id");
256         kDebug() << "// // // REPLACE CLIP: " << id;
257         if (id.contains('_')) id = id.section('_', 0, 0);
258         DocClipBase *clip = getClipById(id);
259         if (clip) {
260             clip->setProducer(prods.at(i));
261         }
262     }
263     emit checkAllClips();
264 }
265
266 void ClipManager::slotAddClipList(const KUrl::List urls, const QString group, const QString &groupId)
267 {
268     QUndoCommand *addClips = new QUndoCommand();
269
270     foreach(const KUrl &file, urls) {
271         if (KIO::NetAccess::exists(file, KIO::NetAccess::SourceSide, NULL)) {
272             if (!getClipByResource(file.path()).empty()) {
273                 if (KMessageBox::warningContinueCancel(kapp->activeWindow(), i18n("Clip <b>%1</b><br />already exists in project, what do you want to do?", file.path()), i18n("Clip already exists")) == KMessageBox::Cancel)
274                     continue;
275             }
276             kDebug() << "Adding clip: " << file.path();
277             QDomDocument doc;
278             QDomElement prod = doc.createElement("producer");
279             doc.appendChild(prod);
280             prod.setAttribute("resource", file.path());
281             uint id = m_clipIdCounter++;
282             prod.setAttribute("id", QString::number(id));
283             if (!group.isEmpty()) {
284                 prod.setAttribute("groupname", group);
285                 prod.setAttribute("groupid", groupId);
286             }
287             KMimeType::Ptr type = KMimeType::findByUrl(file);
288             if (type->name().startsWith("image/")) {
289                 prod.setAttribute("type", (int) IMAGE);
290                 prod.setAttribute("in", 0);
291                 prod.setAttribute("out", m_doc->getFramePos(KdenliveSettings::image_duration()) - 1);
292             } else if (type->is("application/x-kdenlivetitle")) {
293                 // opening a title file
294                 QDomDocument txtdoc("titledocument");
295                 QFile txtfile(file.path());
296                 if (txtfile.open(QIODevice::ReadOnly) && txtdoc.setContent(&txtfile)) {
297                     txtfile.close();
298                     prod.setAttribute("type", (int) TEXT);
299                     prod.setAttribute("xmldata", txtdoc.toString());
300                     prod.setAttribute("transparency", 1);
301                     prod.setAttribute("in", 0);
302                     int out = txtdoc.documentElement().attribute("out").toInt();
303                     if (out > 0)
304                         prod.setAttribute("out", out);
305                     else
306                         prod.setAttribute("out", m_doc->getFramePos(KdenliveSettings::image_duration()) - 1);
307                 } else
308                     txtfile.close();
309             }
310             new AddClipCommand(m_doc, doc.documentElement(), QString::number(id), true, addClips);
311         }
312     }
313     if (addClips->childCount() > 0) {
314         addClips->setText(i18np("Add clip", "Add clips", addClips->childCount()));
315         m_doc->commandStack()->push(addClips);
316     }
317 }
318
319 void ClipManager::slotAddClipFile(const KUrl url, const QString group, const QString &groupId)
320 {
321     slotAddClipList(KUrl::List(url), group, groupId);
322 }
323
324
325 void ClipManager::slotAddColorClipFile(const QString name, const QString color, QString duration, const QString group, const QString &groupId)
326 {
327     QDomDocument doc;
328     QDomElement prod = doc.createElement("producer");
329     doc.appendChild(prod);
330     prod.setAttribute("mlt_service", "colour");
331     prod.setAttribute("colour", color);
332     prod.setAttribute("type", (int) COLOR);
333     uint id = m_clipIdCounter++;
334     prod.setAttribute("id", QString::number(id));
335     prod.setAttribute("in", "0");
336     prod.setAttribute("out", m_doc->getFramePos(duration) - 1);
337     prod.setAttribute("name", name);
338     if (!group.isEmpty()) {
339         prod.setAttribute("groupname", group);
340         prod.setAttribute("groupid", groupId);
341     }
342     AddClipCommand *command = new AddClipCommand(m_doc, doc.documentElement(), QString::number(id), true);
343     m_doc->commandStack()->push(command);
344 }
345
346 void ClipManager::slotAddSlideshowClipFile(const QString name, const QString path, int count, const QString duration, const bool loop, const bool fade, const QString &luma_duration, const QString &luma_file, const int softness, QString group, const QString &groupId)
347 {
348     QDomDocument doc;
349     QDomElement prod = doc.createElement("producer");
350     doc.appendChild(prod);
351     prod.setAttribute("resource", path);
352     prod.setAttribute("type", (int) SLIDESHOW);
353     uint id = m_clipIdCounter++;
354     prod.setAttribute("id", QString::number(id));
355     prod.setAttribute("in", "0");
356     prod.setAttribute("out", m_doc->getFramePos(duration) * count - 1);
357     prod.setAttribute("ttl", m_doc->getFramePos(duration));
358     prod.setAttribute("luma_duration", m_doc->getFramePos(luma_duration));
359     prod.setAttribute("name", name);
360     prod.setAttribute("loop", loop);
361     prod.setAttribute("fade", fade);
362     prod.setAttribute("softness", QString::number(softness));
363     prod.setAttribute("luma_file", luma_file);
364     if (!group.isEmpty()) {
365         prod.setAttribute("groupname", group);
366         prod.setAttribute("groupid", groupId);
367     }
368     AddClipCommand *command = new AddClipCommand(m_doc, doc.documentElement(), QString::number(id), true);
369     m_doc->commandStack()->push(command);
370 }
371
372
373
374 void ClipManager::slotAddTextClipFile(const QString titleName, int out, const QString xml, const QString group, const QString &groupId)
375 {
376     QDomDocument doc;
377     QDomElement prod = doc.createElement("producer");
378     doc.appendChild(prod);
379     //prod.setAttribute("resource", imagePath);
380     prod.setAttribute("name", titleName);
381     prod.setAttribute("xmldata", xml);
382     uint id = m_clipIdCounter++;
383     prod.setAttribute("id", QString::number(id));
384     if (!group.isEmpty()) {
385         prod.setAttribute("groupname", group);
386         prod.setAttribute("groupid", groupId);
387     }
388     prod.setAttribute("type", (int) TEXT);
389     prod.setAttribute("transparency", "1");
390     prod.setAttribute("in", "0");
391     prod.setAttribute("out", out);
392     AddClipCommand *command = new AddClipCommand(m_doc, doc.documentElement(), QString::number(id), true);
393     m_doc->commandStack()->push(command);
394 }
395
396 void ClipManager::slotAddTextTemplateClip(QString titleName, const KUrl path, const QString group, const QString &groupId)
397 {
398     QDomDocument doc;
399     QDomElement prod = doc.createElement("producer");
400     doc.appendChild(prod);
401     prod.setAttribute("name", titleName);
402     prod.setAttribute("resource", path.path());
403     uint id = m_clipIdCounter++;
404     prod.setAttribute("id", QString::number(id));
405     if (!group.isEmpty()) {
406         prod.setAttribute("groupname", group);
407         prod.setAttribute("groupid", groupId);
408     }
409     prod.setAttribute("type", (int) TEXT);
410     prod.setAttribute("transparency", "1");
411     prod.setAttribute("in", "0");
412
413     int out = 0;
414     QDomDocument titledoc;
415     QFile txtfile(path.path());
416     if (txtfile.open(QIODevice::ReadOnly) && titledoc.setContent(&txtfile)) {
417         txtfile.close();
418         out = titledoc.documentElement().attribute("out").toInt();
419     } else txtfile.close();
420
421     if (out == 0) out = m_doc->getFramePos(KdenliveSettings::image_duration()) - 1;
422     prod.setAttribute("out", out);
423
424     AddClipCommand *command = new AddClipCommand(m_doc, doc.documentElement(), QString::number(id), true);
425     m_doc->commandStack()->push(command);
426 }
427
428 int ClipManager::getFreeClipId()
429 {
430     return m_clipIdCounter++;
431 }
432
433 int ClipManager::getFreeFolderId()
434 {
435     return m_folderIdCounter++;
436 }
437
438 int ClipManager::lastClipId() const
439 {
440     return m_clipIdCounter - 1;
441 }
442
443 QString ClipManager::projectFolder() const
444 {
445     return m_doc->projectFolder().path();
446 }
447
448 void ClipManager::addFolder(const QString &id, const QString &name)
449 {
450     m_folderList.insert(id, name);
451 }
452
453 void ClipManager::deleteFolder(const QString &id)
454 {
455     m_folderList.remove(id);
456 }
457
458 AbstractGroupItem *ClipManager::createGroup()
459 {
460     AbstractGroupItem *group = new AbstractGroupItem(m_doc->fps());
461     m_groupsList.append(group);
462     return group;
463 }
464
465 void ClipManager::removeGroup(AbstractGroupItem *group)
466 {
467     m_groupsList.removeAll(group);
468 }
469
470 QDomElement ClipManager::groupsXml() const
471 {
472     QDomDocument doc;
473     QDomElement groups = doc.createElement("groups");
474     doc.appendChild(groups);
475     for (int i = 0; i < m_groupsList.count(); i++) {
476         QDomElement group = doc.createElement("group");
477         groups.appendChild(group);
478         QList <QGraphicsItem *> children = m_groupsList.at(i)->childItems();
479         for (int j = 0; j < children.count(); j++) {
480             if (children.at(j)->type() == AVWIDGET || children.at(j)->type() == TRANSITIONWIDGET) {
481                 AbstractClipItem *item = static_cast <AbstractClipItem *>(children.at(j));
482                 ItemInfo info = item->info();
483                 if (item->type() == AVWIDGET) {
484                     QDomElement clip = doc.createElement("clipitem");
485                     clip.setAttribute("track", info.track);
486                     clip.setAttribute("position", info.startPos.frames(m_doc->fps()));
487                     group.appendChild(clip);
488                 } else if (item->type() == TRANSITIONWIDGET) {
489                     QDomElement clip = doc.createElement("transitionitem");
490                     clip.setAttribute("track", info.track);
491                     clip.setAttribute("position", info.startPos.frames(m_doc->fps()));
492                     group.appendChild(clip);
493                 }
494             }
495         }
496     }
497     return doc.documentElement();
498 }
499
500
501 void ClipManager::slotClipModified(const QString &path)
502 {
503     // kDebug() << "// CLIP: " << path << " WAS MODIFIED";
504     const QList <DocClipBase *> list = getClipByResource(path);
505     for (int i = 0; i < list.count(); i++) {
506         DocClipBase *clip = list.at(i);
507         if (clip != NULL) emit reloadClip(clip->getId());
508     }
509 }
510
511 void ClipManager::slotClipMissing(const QString &path)
512 {
513     // kDebug() << "// CLIP: " << path << " WAS MISSING";
514     const QList <DocClipBase *> list = getClipByResource(path);
515     for (int i = 0; i < list.count(); i++) {
516         DocClipBase *clip = list.at(i);
517         if (clip != NULL) emit missingClip(clip->getId());
518     }
519 }
520
521 void ClipManager::slotClipAvailable(const QString &path)
522 {
523     // kDebug() << "// CLIP: " << path << " WAS ADDED";
524     const QList <DocClipBase *> list = getClipByResource(path);
525     for (int i = 0; i < list.count(); i++) {
526         DocClipBase *clip = list.at(i);
527         if (clip != NULL) emit availableClip(clip->getId());
528     }
529 }
530
531 int ClipManager::clipsCount() const
532 {
533     return m_clipList.count();
534 }
535
536