summaryrefslogtreecommitdiffstats
path: root/src/generators/objectinstances_p.h
blob: 142d84e78f3e9b4905fac411289c5f494455db52 (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
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only

#ifndef OBJECTINSTANCES_H
#define OBJECTINSTANCES_H

#include <QFile>
#include <QQmlEngine>
#include <QQuickWindow>

QT_BEGIN_NAMESPACE

class ObjectInstances : public QObject
{
    Q_OBJECT
    Q_PROPERTY(QString outputPrefix READ outputPrefix WRITE setOutputPrefix NOTIFY outputPrefixChanged)
    Q_PROPERTY(OutputFormat outputFormat READ outputFormat WRITE setOutputFormat NOTIFY outputFormatChanged)
    Q_PROPERTY(QObject* root READ root WRITE setRoot NOTIFY rootChanged)
    Q_PROPERTY(ColorSource nodeColorSource READ nodeColorSource WRITE setNodeColorSource NOTIFY nodeColorSourceChanged)
    QML_ELEMENT

public:
    enum class OutputFormat {
//        QML,
//        PlantUML,
        Graphviz
    };
    Q_ENUM(OutputFormat)

    enum class ColorSource {
        Default,
        ItemColor
    };
    Q_ENUM(ColorSource)

    ObjectInstances(QObject *parent = nullptr);

    QString outputPrefix() const { return m_outputPrefix; }
    void setOutputPrefix(QString outputPrefix);

    OutputFormat outputFormat() const;
    void setOutputFormat(OutputFormat fmt);

    QObject *root() const;
    void setRoot(QObject *newRoot);

    Q_INVOKABLE void generate();
    Q_INVOKABLE QQuickWindow *findQuickWindow();

    ColorSource nodeColorSource() const;
    void setNodeColorSource(ColorSource newNodeColorSource);

signals:
    void outputPrefixChanged();
    void outputFormatChanged();
    void rootChanged();
    void nodeColorSourceChanged();

private:
    void parseClassAndMethod(const QString &classAndMethod, QString &className, QString &methodName);
    static QString objectId(const void *obj);
    void writeDot(const QString &plainFilePath, const QObject *o);
    void writeDotRecur(QFile &f, const QObject *o);

private:
    QString m_outputPrefix;
    OutputFormat m_outputFormat = OutputFormat::Graphviz;
    ColorSource m_nodeColorSource = ColorSource::Default;
    QObject *m_root = nullptr;
};

QT_END_NAMESPACE

#endif // OBJECTINSTANCES_H