File indexing completed on 2024-05-05 04:52:53

 /*
     SPDX-FileCopyrightText: 2016 Jean-Baptiste Mardelle <jb@kdenlive.org>
 
 SPDX-License-Identifier: GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
 */
 
 #include "generators.h"
 #include "assets/abstractassetsrepository.hpp"
 #include "doc/kthumb.h"
 #include "effects/effectsrepository.hpp"
 #include "kdenlivesettings.h"
 #include "widgets/timecodedisplay.h"
 
 #include <QDialogButtonBox>
 #include <QDir>
 #include <QDomDocument>
 #include <QFileDialog>
 #include <QLabel>
 #include <QStandardPaths>
 #include <QVBoxLayout>
 
 #include "core.h"
 #include "klocalizedstring.h"
 #include "profiles/profilemodel.hpp"
 #include <KMessageBox>
 #include <KRecentDirs>
 #include <memory>
 #include <mlt++/MltConsumer.h>
 #include <mlt++/MltProducer.h>
 #include <mlt++/MltProfile.h>
 #include <mlt++/MltTractor.h>
 
 Generators::Generators(const QString &path, QWidget *parent)
     : QDialog(parent)
     , m_producer(nullptr)
     , m_timePos(nullptr)
     , m_container(nullptr)
     , m_preview(nullptr)
 {
     QDomDocument doc;
     if (!Xml::docContentFromFile(doc, path, false)) {
         return;
     }
 
     QDomElement base = doc.documentElement();
     if (base.tagName() == QLatin1String("generator")) {
         QString generatorTag = base.attribute(QStringLiteral("tag"));
         setWindowTitle(i18n(base.firstChildElement(QStringLiteral("name")).text().toUtf8().data()));
         auto *lay = new QVBoxLayout(this);
         m_preview = new QLabel;
         m_preview->setMinimumSize(1, 1);
         lay->addWidget(m_preview);
         m_producer = new Mlt::Producer(pCore->getProjectProfile(), generatorTag.toUtf8().constData());
         m_pixmap = QPixmap::fromImage(KThumb::getFrame(m_producer, 0, pCore->getCurrentProfile()->width(), pCore->getCurrentProfile()->height()));
         m_preview->setPixmap(m_pixmap.scaledToWidth(m_preview->width()));
         auto *hlay = new QHBoxLayout;
         hlay->addWidget(new QLabel(i18n("Duration:")));
         m_timePos = new TimecodeDisplay(this);
         if (base.hasAttribute(QStringLiteral("updateonduration"))) {
             connect(m_timePos, &TimecodeDisplay::timeCodeEditingFinished, this, &Generators::updateDuration);
         }
         hlay->addWidget(m_timePos);
         lay->addLayout(hlay);
         QWidget *frameWidget = new QWidget;
         lay->addWidget(frameWidget);
 
         m_view = new AssetParameterView(frameWidget);
         lay->addWidget(m_view);
         QString tag = base.attribute(QStringLiteral("tag"), QString());
 
         auto prop = std::make_unique<Mlt::Properties>(m_producer->get_properties());
         m_assetModel.reset(new AssetParameterModel(std::move(prop), base, tag, ObjectId())); // NOLINT
         m_view->setModel(m_assetModel, QSize(1920, 1080), false);
         connect(m_assetModel.get(), &AssetParameterModel::modelChanged, this, [this]() { updateProducer(); });
 
         lay->addStretch(10);
         QDialogButtonBox *buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
         connect(buttonBox, &QDialogButtonBox::accepted, this, &QDialog::accept);
         connect(buttonBox, &QDialogButtonBox::rejected, this, &QDialog::reject);
         lay->addWidget(buttonBox);
         m_timePos->setValue(KdenliveSettings::color_duration());
     }
 }
 
 void Generators::updateProducer()
 {
     int w = m_pixmap.width();
     int h = m_pixmap.height();
     m_pixmap = QPixmap::fromImage(KThumb::getFrame(m_producer, 0, w, h));
     m_preview->setPixmap(m_pixmap.scaledToWidth(m_preview->width()));
 }
 
 void Generators::resizeEvent(QResizeEvent *event)
 {
     QDialog::resizeEvent(event);
     m_preview->setPixmap(m_pixmap.scaledToWidth(m_preview->width()));
 }
 
 Generators::~Generators()
 {
     delete m_timePos;
 }
 
 // static
 void Generators::getGenerators(const QStringList &producers, QMenu *menu)
 {
     const QStringList generatorFolders =
         QStandardPaths::locateAll(QStandardPaths::AppDataLocation, QStringLiteral("generators"), QStandardPaths::LocateDirectory);
     const QStringList filters = QStringList() << QStringLiteral("*.xml");
     QStringList parsedGenerators;
     for (const QString &folder : generatorFolders) {
         QDir directory(folder);
         const QStringList filesnames = directory.entryList(filters, QDir::Files);
         for (const QString &fname : filesnames) {
             if (parsedGenerators.contains(fname)) {
                 continue;
             }
             parsedGenerators << fname;
             QPair<QString, QString> result = parseGenerator(directory.absoluteFilePath(fname), producers);
             if (!result.first.isEmpty()) {
                 QAction *action = menu->addAction(i18n(result.first.toUtf8().data()));
                 action->setData(result.second);
             }
         }
     }
 }
 
 // static
 QPair<QString, QString> Generators::parseGenerator(const QString &path, const QStringList &producers)
 {
     QPair<QString, QString> result;
     QDomDocument doc;
     if (!Xml::docContentFromFile(doc, path, false)) {
         return result;
     }
 
     QDomElement base = doc.documentElement();
     if (base.tagName() == QLatin1String("generator")) {
         QString generatorTag = base.attribute(QStringLiteral("tag"));
         if (producers.contains(generatorTag)) {
             result.first = base.firstChildElement(QStringLiteral("name")).text();
             result.second = path;
         }
     }
     return result;
 }
 
 void Generators::updateDuration(int duration)
 {
     m_producer->set("length", duration);
     m_producer->set_in_and_out(0, duration - 1);
     updateProducer();
 }
 
 QUrl Generators::getSavedClip(QString clipFolder)
 {
     if (clipFolder.isEmpty()) {
         clipFolder = KRecentDirs::dir(QStringLiteral(":KdenliveClipFolder"));
     }
     if (clipFolder.isEmpty()) {
         clipFolder = QDir::homePath();
     }
     QFileDialog fd(this);
     fd.setDirectory(clipFolder);
     fd.setNameFilter(i18n("MLT Playlist (*.mlt)"));
     fd.setAcceptMode(QFileDialog::AcceptSave);
     fd.setFileMode(QFileDialog::AnyFile);
     fd.setDefaultSuffix(QStringLiteral("mlt"));
     if (fd.exec() != QDialog::Accepted || fd.selectedUrls().isEmpty()) {
         return QUrl();
     }
     QUrl url = fd.selectedUrls().constFirst();
 
     if (url.isValid()) {
         Mlt::Tractor trac(pCore->getProjectProfile());
         m_producer->set("length", m_timePos->getValue());
         m_producer->set_in_and_out(0, m_timePos->getValue() - 1);
         trac.set_track(*m_producer, 0);
         Mlt::Consumer c(pCore->getProjectProfile(), "xml", url.toLocalFile().toUtf8().constData());
         c.connect(trac);
         c.run();
         return url;
     }
     return QUrl();
 }