aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/effectcomposer/effectcomposernodesmodel.h
blob: f1de958f38665fd138450922dd62e42d946157b4 (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
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0

#pragma once

#include "effectnodescategory.h"

#include <QStandardItemModel>

namespace EffectComposer {

class EffectNode;

class EffectComposerNodesModel : public QAbstractListModel
{
    Q_OBJECT

    enum Roles {
        CategoryNameRole = Qt::UserRole + 1,
        CategoryNodesRole
    };

public:
    EffectComposerNodesModel(QObject *parent = nullptr);

    QHash<int, QByteArray> roleNames() const override;
    int rowCount(const QModelIndex & parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

    void loadModel();
    void loadCustomNodes();
    void resetModel();

    bool nodeExists(const QString &name);
    bool isBuiltIn(const QString &name);

    QList<EffectNodesCategory *> categories() const { return  m_categories; }

    void updateCanBeAdded(const QStringList &uniforms, const QStringList &nodeNames);

    QHash<QString, QString> defaultImagesForNode(const QString &name) const;

    void removeEffectNode(const QString &name);

private:
    QString nodesSourcesPath() const;

    QList<EffectNodesCategory *> m_categories;
    bool m_probeNodesDir = false;
    bool m_modelLoaded = false;
    QHash<QString, QHash<QString, QString>> m_defaultImagesHash;
    QStringList m_builtInNodeNames;
    QStringList m_customNodeNames;
    EffectNode *m_builtinCustomNode = nullptr;
    EffectNodesCategory *m_customCategory = nullptr;
};

} // namespace EffectComposer