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