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