blob: 3b894d915752ac2473a5454b65ec512f0f7772b6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
#include "effectmakernodesmodel.h"
#include <utils/hostosinfo.h>
#include <QCoreApplication>
namespace EffectMaker {
EffectMakerNodesModel::EffectMakerNodesModel(QObject *parent)
: QAbstractListModel{parent}
{
}
QHash<int, QByteArray> EffectMakerNodesModel::roleNames() const
{
QHash<int, QByteArray> roles;
roles[CategoryNameRole] = "categoryName";
roles[CategoryNodesRole] = "categoryNodes";
return roles;
}
int EffectMakerNodesModel::rowCount(const QModelIndex &parent) const
{
Q_UNUSED(parent)
return m_categories.count();
}
QVariant EffectMakerNodesModel::data(const QModelIndex &index, int role) const
{
QTC_ASSERT(index.isValid() && index.row() < m_categories.size(), return {});
QTC_ASSERT(roleNames().contains(role), return {});
return m_categories.at(index.row())->property(roleNames().value(role));
}
void EffectMakerNodesModel::findNodesPath()
{
if (m_nodesPath.exists() || m_probeNodesDir)
return;
QDir nodesDir;
if (!qEnvironmentVariable("EFFECT_MAKER_NODES_PATH").isEmpty())
nodesDir.setPath(qEnvironmentVariable("EFFECT_MAKER_NODES_PATH"));
else if (Utils::HostOsInfo::isMacHost())
nodesDir.setPath(QCoreApplication::applicationDirPath() + "/../Resources/effect_maker_nodes");
// search for nodesDir from exec dir and up
if (nodesDir.dirName() == ".") {
m_probeNodesDir = true; // probe only once
nodesDir.setPath(QCoreApplication::applicationDirPath());
while (!nodesDir.cd("effect_maker_nodes") && nodesDir.cdUp())
; // do nothing
if (nodesDir.dirName() != "effect_maker_nodes") // bundlePathDir not found
return;
}
m_nodesPath = Utils::FilePath::fromString(nodesDir.path());
}
void EffectMakerNodesModel::loadModel()
{
findNodesPath();
if (!m_nodesPath.exists()) {
qWarning() << __FUNCTION__ << "Effects not found.";
return;
}
QDirIterator itCategories(m_nodesPath.toString(), QDir::Dirs | QDir::NoDotAndDotDot);
while (itCategories.hasNext()) {
itCategories.next();
if (itCategories.fileName() == "images" || itCategories.fileName() == "common")
continue;
QString catName = itCategories.fileName();
QList<EffectNode *> effects = {};
Utils::FilePath categoryPath = m_nodesPath.resolvePath(itCategories.fileName());
QDirIterator itEffects(categoryPath.toString(), {"*.qen"}, QDir::Files);
while (itEffects.hasNext()) {
itEffects.next();
effects.push_back(new EffectNode(itEffects.filePath()));
}
catName[0] = catName[0].toUpper(); // capitalize first letter
EffectNodesCategory *category = new EffectNodesCategory(catName, effects);
m_categories.push_back(category);
}
resetModel();
}
void EffectMakerNodesModel::resetModel()
{
beginResetModel();
endResetModel();
}
} // namespace EffectMaker
|