summaryrefslogtreecommitdiffstats
path: root/src/Authoring/Studio/Palettes/Project/ProjectFileSystemModel.h
blob: df104876972b5d9d4ebfacd59788203a307f6a76 (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
109
110
111
112
113
114
115
116
117
118
119
120
/****************************************************************************
**
** Copyright (C) 2017 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of Qt 3D Studio.
**
** $QT_BEGIN_LICENSE:GPL-EXCEPT$
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 as published by the Free Software
** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
** included in the packaging of this file. Please review the following
** information to ensure the GNU General Public License requirements will
** be met: https://www.gnu.org/licenses/gpl-3.0.html.
**
** $QT_END_LICENSE$
**
****************************************************************************/
#ifndef TREEVIEWADAPTOR_H
#define TREEVIEWADAPTOR_H

#include "StudioObjectTypes.h"
#include "DocumentEditorEnumerations.h"

#include <QtWidgets/qfilesystemmodel.h>
#include <QtCore/qabstractitemmodel.h>
#include <QtCore/qlist.h>
#include <QtWidgets/qmessagebox.h>
#include <QtCore/qurl.h>

QT_FORWARD_DECLARE_CLASS(QFileSystemModel)

class ProjectFileSystemModel : public QAbstractListModel
{
    Q_OBJECT

public:
    explicit ProjectFileSystemModel(QObject *parent = nullptr);

    enum {
        IsExpandableRole = QFileSystemModel::FilePermissions + 1,
        IsDraggableRole,
        IsReferencedRole,
        DepthRole,
        ExpandedRole,
    };

    void setRootPath(const QString &path);

    QHash<int, QByteArray> roleNames() const override;
    int rowCount(const QModelIndex &parent = {}) const override;
    QVariant data(const QModelIndex &index, int role) const override;
    QMimeData *mimeData(const QModelIndexList &indexes) const override;

    QString filePath(int row) const;
    bool isRefreshable(int row) const;

    void updateReferences();
    Q3DStudio::DocumentEditorFileType::Enum assetTypeForRow(int row);
    int rowForPath(const QString &path) const;

    Q_INVOKABLE void expand(int row);
    Q_INVOKABLE void collapse(int row);

    Q_INVOKABLE void importUrls(const QList<QUrl> &urls, int row, bool autoSort = true);
    Q_INVOKABLE bool hasValidUrlsForDropping(const QList<QUrl> &urls) const;

Q_SIGNALS:
    void modelChanged(QAbstractItemModel *model);

private:
    void setRootIndex(const QModelIndex &rootIndex);
    void clearModelData();
    void showModelTopLevelItems();
    void showModelChildItems(const QModelIndex &parentItem, int start, int end);
    int modelIndexRow(const QModelIndex &modelIndex) const;
    bool isExpanded(const QModelIndex &modelIndex) const;
    QString getIconName(const QString &path) const;
    EStudioObjectType getIconType(const QString &path) const;
    bool isVisible(const QModelIndex& modelIndex) const;
    bool hasVisibleChildren(const QModelIndex &modelIndex) const;
    void importUrl(QDir &targetDir, const QUrl &url);
    void importPresentationAssets(const QFileInfo &uipSrc, const QFileInfo &uipTarget,
                                  const int overrideChoice = QMessageBox::NoButton) const;

    void modelRowsInserted(const QModelIndex &parent, int start, int end);
    void modelRowsRemoved(const QModelIndex &parent, int start, int end);
    void modelRowsMoved(const QModelIndex &parent, int start, int end);
    void modelLayoutChanged();
    void importQmlAssets(const QObject *qmlNode, const QDir &srcDir, const QDir &targetDir);

    void updateDefaultDirMap();
    void addPathsToReferences(const QString &projectPath, const QString &origPath);

    struct TreeItem {
        QPersistentModelIndex index;
        int depth;
        bool expanded;
        TreeItem *parent;
        int childCount;
    };

    QFileSystemModel *m_model = nullptr;
    QPersistentModelIndex m_rootIndex;
    QList<TreeItem> m_items;
    QSet<QString> m_references;
    QHash<QString, QString> m_defaultDirToAbsPathMap;
    int m_importQmlOverrideChoice = QMessageBox::NoButton;
};

#endif // TREEVIEWADAPTOR_H