1 files changed, 324 insertions, 0 deletions
diff --git a/src/plugins/qmlprofilerextension/flamegraphmodel.cpp b/src/plugins/qmlprofilerextension/flamegraphmodel.cpp
new file mode 100644
index 0000000000..e181a66109
--- /dev/null
+++ b/src/plugins/qmlprofilerextension/flamegraphmodel.cpp
@@ -0,0 +1,324 @@
+/****************************************************************************
+**
+** Copyright (C) 2016 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of Qt Creator.
+**
+** 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.
+**
+****************************************************************************/
+
+#include "flamegraphmodel.h"
+
+#include <qmlprofiler/qmlprofilermodelmanager.h>
+#include <qmlprofiler/qmlprofilerdatamodel.h>
+
+#include <utils/algorithm.h>
+#include <utils/qtcassert.h>
+
+#include <QVector>
+#include <QString>
+#include <QStack>
+#include <QQueue>
+#include <QSet>
+
+namespace QmlProfilerExtension {
+namespace Internal {
+
+FlameGraphModel::FlameGraphModel(QmlProfiler::QmlProfilerModelManager *modelManager,
+                                 QObject *parent) : QAbstractItemModel(parent)
+{
+    m_modelManager = modelManager;
+    connect(modelManager->qmlModel(), &QmlProfiler::QmlProfilerDataModel::changed,
+            this, [this](){loadData();});
+    connect(modelManager->notesModel(), &Timeline::TimelineNotesModel::changed,
+            this, [this](int typeId, int, int){loadNotes(typeId, true);});
+    m_modelId = modelManager->registerModelProxy();
+
+    // We're iterating twice in loadData.
+    modelManager->setProxyCountWeight(m_modelId, 2);
+
+    m_acceptedTypes << QmlDebug::Compiling << QmlDebug::Creating << QmlDebug::Binding
+                    << QmlDebug::HandlingSignal << QmlDebug::Javascript;
+    modelManager->announceFeatures(m_modelId, QmlDebug::Constants::QML_JS_RANGE_FEATURES);
+}
+
+void FlameGraphModel::setEventTypeAccepted(QmlDebug::RangeType type, bool accepted)
+{
+    if (accepted && !m_acceptedTypes.contains(type))
+        m_acceptedTypes << type;
+    else if (!accepted && m_acceptedTypes.contains(type))
+        m_acceptedTypes.removeOne(type);
+}
+
+bool FlameGraphModel::eventTypeAccepted(QmlDebug::RangeType type) const
+{
+    return m_acceptedTypes.contains(type);
+}
+
+void FlameGraphModel::clear()
+{
+    m_modelManager->modelProxyCountUpdated(m_modelId, 0, 1);
+    m_stackBottom = FlameGraphData();
+    m_typeIdsWithNotes.clear();
+}
+
+void FlameGraphModel::loadNotes(int typeIndex, bool emitSignal)
+{
+    QSet<int> changedNotes;
+    Timeline::TimelineNotesModel *notes = m_modelManager->notesModel();
+    if (typeIndex == -1) {
+        changedNotes = m_typeIdsWithNotes;
+        m_typeIdsWithNotes.clear();
+        for (int i = 0; i < notes->count(); ++i)
+            m_typeIdsWithNotes.insert(notes->typeId(i));
+        changedNotes += m_typeIdsWithNotes;
+    } else {
+        changedNotes.insert(typeIndex);
+        if (notes->byTypeId(typeIndex).isEmpty())
+            m_typeIdsWithNotes.remove(typeIndex);
+        else
+            m_typeIdsWithNotes.insert(typeIndex);
+    }
+
+    if (!emitSignal)
+        return;
+
+    QQueue<QModelIndex> queue;
+    queue.append(QModelIndex());
+
+    QVector<int> roles = {Note};
+    while (!queue.isEmpty()) {
+        QModelIndex index = queue.takeFirst();
+        if (index.isValid()) {
+            FlameGraphData *data = static_cast<FlameGraphData *>(index.internalPointer());
+            if (changedNotes.contains(data->typeIndex))
+                emit dataChanged(index, index, roles);
+            for (int row = 0; row < rowCount(index); ++row)
+                queue.append(index.child(row, 0));
+        }
+
+    }
+}
+
+void FlameGraphModel::loadData(qint64 rangeStart, qint64 rangeEnd)
+{
+    const bool checkRanges = (rangeStart != -1) && (rangeEnd != -1);
+    if (m_modelManager->state() == QmlProfiler::QmlProfilerModelManager::ClearingData) {
+        beginResetModel();
+        clear();
+        endResetModel();
+        return;
+    } else if (m_modelManager->state() != QmlProfiler::QmlProfilerModelManager::ProcessingData &&
+               m_modelManager->state() != QmlProfiler::QmlProfilerModelManager::Done) {
+        return;
+    }
+
+    beginResetModel();
+    clear();
+
+    const QVector<QmlProfiler::QmlProfilerDataModel::QmlEventData> &eventList
+            = m_modelManager->qmlModel()->getEvents();
+    const QVector<QmlProfiler::QmlProfilerDataModel::QmlEventTypeData> &typesList
+            = m_modelManager->qmlModel()->getEventTypes();
+
+    // used by binding loop detection
+    QStack<const QmlProfiler::QmlProfilerDataModel::QmlEventData *> callStack;
+    callStack.append(0);
+    FlameGraphData *stackTop = &m_stackBottom;
+
+    for (int i = 0; i < eventList.size(); ++i) {
+        const QmlProfiler::QmlProfilerDataModel::QmlEventData *event = &eventList[i];
+        int typeIndex = event->typeIndex();
+        const QmlProfiler::QmlProfilerDataModel::QmlEventTypeData *type = &typesList[typeIndex];
+
+        if (!m_acceptedTypes.contains(type->rangeType))
+            continue;
+
+        if (checkRanges) {
+            if ((event->startTime() + event->duration() < rangeStart)
+                    || (event->startTime() > rangeEnd))
+                continue;
+        }
+
+        const QmlProfiler::QmlProfilerDataModel::QmlEventData *potentialParent = callStack.top();
+        while (potentialParent &&
+               potentialParent->startTime() + potentialParent->duration() <= event->startTime()) {
+            callStack.pop();
+            stackTop = stackTop->parent;
+            potentialParent = callStack.top();
+        }
+
+        callStack.push(event);
+        stackTop = pushChild(stackTop, event);
+
+        m_modelManager->modelProxyCountUpdated(m_modelId, i, eventList.count());
+    }
+
+    foreach (FlameGraphData *child, m_stackBottom.children)
+        m_stackBottom.duration += child->duration;
+
+    loadNotes(-1, false);
+
+    m_modelManager->modelProxyCountUpdated(m_modelId, 1, 1);
+    endResetModel();
+}
+
+static QString nameForType(QmlDebug::RangeType typeNumber)
+{
+    switch (typeNumber) {
+    case QmlDebug::Painting: return FlameGraphModel::tr("Paint");
+    case QmlDebug::Compiling: return FlameGraphModel::tr("Compile");
+    case QmlDebug::Creating: return FlameGraphModel::tr("Create");
+    case QmlDebug::Binding: return FlameGraphModel::tr("Binding");
+    case QmlDebug::HandlingSignal: return FlameGraphModel::tr("Signal");
+    case QmlDebug::Javascript: return FlameGraphModel::tr("JavaScript");
+    default: return QString();
+    }
+}
+
+QVariant FlameGraphModel::lookup(const FlameGraphData &stats, int role) const
+{
+    switch (role) {
+    case TypeId: return stats.typeIndex;
+    case Note: {
+        QString ret;
+        if (!m_typeIdsWithNotes.contains(stats.typeIndex))
+            return ret;
+        QmlProfiler::QmlProfilerNotesModel *notes = m_modelManager->notesModel();
+        foreach (const QVariant &item, notes->byTypeId(stats.typeIndex)) {
+            if (ret.isEmpty())
+                ret = item.toString();
+            else
+                ret += QChar::LineFeed + item.toString();
+        }
+        return ret;
+    }
+    case Duration: return stats.duration;
+    case CallCount: return stats.calls;
+    case TimePerCall: return stats.duration / stats.calls;
+    case TimeInPercent: return stats.duration * 100 / m_stackBottom.duration;
+    default: break;
+    }
+
+    if (stats.typeIndex != -1) {
+        const QVector<QmlProfiler::QmlProfilerDataModel::QmlEventTypeData> &typeList =
+                m_modelManager->qmlModel()->getEventTypes();
+        const QmlProfiler::QmlProfilerDataModel::QmlEventTypeData &type = typeList[stats.typeIndex];
+
+        switch (role) {
+        case Filename: return type.location.filename;
+        case Line: return type.location.line;
+        case Column: return type.location.column;
+        case Type: return nameForType(type.rangeType);
+        case RangeType: return type.rangeType;
+        case Details: return type.data.isEmpty() ?
+                        FlameGraphModel::tr("Source code not available") : type.data;
+        default: return QVariant();
+        }
+    } else {
+        return QVariant();
+    }
+}
+
+FlameGraphData::FlameGraphData(FlameGraphData *parent, int typeIndex, qint64 duration) :
+    duration(duration), calls(1), typeIndex(typeIndex), parent(parent) {}
+
+FlameGraphData::~FlameGraphData()
+{
+    qDeleteAll(children);
+}
+
+FlameGraphData *FlameGraphModel::pushChild(
+        FlameGraphData *parent, const QmlProfiler::QmlProfilerDataModel::QmlEventData *data)
+{
+    foreach (FlameGraphData *child, parent->children) {
+        if (child->typeIndex == data->typeIndex()) {
+            ++child->calls;
+            child->duration += data->duration();
+            return child;
+        }
+    }
+
+    FlameGraphData *child = new FlameGraphData(parent, data->typeIndex(), data->duration());
+    parent->children.append(child);
+    return child;
+}
+
+QModelIndex FlameGraphModel::index(int row, int column, const QModelIndex &parent) const
+{
+    if (parent.isValid()) {
+        FlameGraphData *parentData = static_cast<FlameGraphData *>(parent.internalPointer());
+        return createIndex(row, column, parentData->children[row]);
+    } else {
+        return createIndex(row, column, row >= 0 ? m_stackBottom.children[row] : 0);
+    }
+}
+
+QModelIndex FlameGraphModel::parent(const QModelIndex &child) const
+{
+    if (child.isValid()) {
+        FlameGraphData *childData = static_cast<FlameGraphData *>(child.internalPointer());
+        return createIndex(0, 0, childData->parent);
+    } else {
+        return QModelIndex();
+    }
+}
+
+int FlameGraphModel::rowCount(const QModelIndex &parent) const
+{
+    if (parent.isValid()) {
+        FlameGraphData *parentData = static_cast<FlameGraphData *>(parent.internalPointer());
+        return parentData->children.count();
+    } else {
+        return m_stackBottom.children.count();
+    }
+}
+
+int FlameGraphModel::columnCount(const QModelIndex &parent) const
+{
+    Q_UNUSED(parent);
+    return 1;
+}
+
+QVariant FlameGraphModel::data(const QModelIndex &index, int role) const
+{
+    FlameGraphData *data = static_cast<FlameGraphData *>(index.internalPointer());
+    return lookup(data ? *data : m_stackBottom, role);
+}
+
+QHash<int, QByteArray> FlameGraphModel::roleNames() const
+{
+    QHash<int, QByteArray> names = QAbstractItemModel::roleNames();
+    names[TypeId] = "typeId";
+    names[Type] = "type";
+    names[Duration] = "duration";
+    names[CallCount] = "callCount";
+    names[Details] = "details";
+    names[Filename] = "filename";
+    names[Line] = "line";
+    names[Column] = "column";
+    names[Note] = "note";
+    names[TimePerCall] = "timePerCall";
+    names[TimeInPercent] = "timeInPercent";
+    names[RangeType] = "rangeType";
+    return names;
+}
+
+} // namespace Internal
+} // namespace QmlProfilerExtension