ClangCodeModel: Move memory usage widget to its own set of files

Change-Id: Iaeb449a12a11e46e8ce03556c699fff4ffceffc5
Reviewed-by: <github-actions-qt-creator@cristianadam.eu>
Reviewed-by: Qt CI Bot <qt_ci_bot@qt-project.org>
Reviewed-by: David Schulz <david.schulz@qt.io>

src/plugins/clangcodemodel/clangdmemoryusagewidget.cpp

@@ -0,0 +1,200 @@
+/****************************************************************************
+**
+** Copyright (C) 2022 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 "clangdmemoryusagewidget.h"
+
+#include "clangdclient.h"
+
+#include <languageserverprotocol/jsonobject.h>
+#include <utils/itemviews.h>
+#include <utils/treemodel.h>
+
+#include <QHeaderView>
+#include <QMenu>
+#include <QPointer>
+#include <QVBoxLayout>
+
+using namespace LanguageServerProtocol;
+using namespace Utils;
+
+namespace ClangCodeModel::Internal {
+
+class MemoryTree : public JsonObject
+{
+public:
+    using JsonObject::JsonObject;
+
+    // number of bytes used, including child components
+    qint64 total() const { return qint64(typedValue<double>(totalKey())); }
+
+    // number of bytes used, excluding child components
+    qint64 self() const { return qint64(typedValue<double>(selfKey())); }
+
+    // named child components
+    using NamedComponent = std::pair<MemoryTree, QString>;
+    QList<NamedComponent> children() const
+    {
+        QList<NamedComponent> components;
+        const auto obj = operator const QJsonObject &();
+        for (auto it = obj.begin(); it != obj.end(); ++it) {
+            if (it.key() == totalKey() || it.key() == selfKey())
+                continue;
+            components << std::make_pair(MemoryTree(it.value()), it.key());
+        }
+        return components;
+    }
+
+private:
+    static QString totalKey() { return QLatin1String("_total"); }
+    static QString selfKey() { return QLatin1String("_self"); }
+};
+
+
+class MemoryTreeItem : public TreeItem
+{
+    Q_DECLARE_TR_FUNCTIONS(ClangCodeModel)
+public:
+    MemoryTreeItem(const QString &displayName, const MemoryTree &tree)
+        : m_displayName(displayName), m_bytesUsed(tree.total())
+    {
+        for (const MemoryTree::NamedComponent &component : tree.children())
+            appendChild(new MemoryTreeItem(component.second, component.first));
+    }
+
+private:
+    QVariant data(int column, int role) const override
+    {
+        switch (role) {
+        case Qt::DisplayRole:
+            if (column == 0)
+                return m_displayName;
+            return memString();
+        case Qt::TextAlignmentRole:
+            if (column == 1)
+                return Qt::AlignRight;
+            break;
+        default:
+            break;
+        }
+        return {};
+    }
+
+    QString memString() const
+    {
+        static const QList<std::pair<int, QString>> factors{
+            std::make_pair(1000000000, QString("GB")),
+            std::make_pair(1000000, QString("MB")),
+            std::make_pair(1000, QString("KB")),
+        };
+        for (const auto &factor : factors) {
+            if (m_bytesUsed > factor.first)
+                return QString::number(qint64(std::round(double(m_bytesUsed) / factor.first)))
+                        + ' ' + factor.second;
+        }
+        return QString::number(m_bytesUsed) + "  B";
+    }
+
+    const QString m_displayName;
+    const qint64 m_bytesUsed;
+};
+
+
+class MemoryTreeModel : public BaseTreeModel
+{
+public:
+    MemoryTreeModel()
+    {
+        setHeader({ClangdMemoryUsageWidget::tr("Component"),
+                   ClangdMemoryUsageWidget::tr("Total Memory")});
+    }
+
+    void update(const MemoryTree &tree)
+    {
+        setRootItem(new MemoryTreeItem({}, tree));
+    }
+};
+
+
+class ClangdMemoryUsageWidget::Private
+{
+public:
+    Private(ClangdMemoryUsageWidget *q, ClangdClient *client) : q(q), client(client)
+    {
+        setupUi();
+        getMemoryTree();
+    }
+
+    void setupUi();
+    void getMemoryTree();
+
+    ClangdMemoryUsageWidget * const q;
+    const QPointer<ClangdClient> client;
+    MemoryTreeModel model;
+    TreeView view;
+    Utils::optional<MessageId> currentRequest;
+};
+
+ClangdMemoryUsageWidget::ClangdMemoryUsageWidget(ClangdClient *client)
+    : d(new Private(this, client))
+{
+}
+
+ClangdMemoryUsageWidget::~ClangdMemoryUsageWidget()
+{
+    if (d->client && d->currentRequest.has_value())
+        d->client->cancelRequest(d->currentRequest.value());
+    delete d;
+}
+
+void ClangdMemoryUsageWidget::Private::setupUi()
+{
+    const auto layout = new QVBoxLayout(q);
+    view.setContextMenuPolicy(Qt::CustomContextMenu);
+    view.header()->setSectionResizeMode(QHeaderView::ResizeToContents);
+    view.header()->setStretchLastSection(false);
+    view.setModel(&model);
+    layout->addWidget(&view);
+    QObject::connect(&view, &QWidget::customContextMenuRequested, q, [this](const QPoint &pos) {
+        QMenu menu;
+        menu.addAction(tr("Update"), [this] { getMemoryTree(); });
+        menu.exec(view.mapToGlobal(pos));
+    });
+}
+
+void ClangdMemoryUsageWidget::Private::getMemoryTree()
+{
+    Request<MemoryTree, std::nullptr_t, JsonObject> request("$/memoryUsage", {});
+    request.setResponseCallback([this](decltype(request)::Response response) {
+        currentRequest.reset();
+        qCDebug(clangdLog) << "received memory usage response";
+        if (const auto result = response.result())
+            model.update(*result);
+    });
+    qCDebug(clangdLog) << "sending memory usage request";
+    currentRequest = request.id();
+    client->sendMessage(request, ClangdClient::SendDocUpdates::Ignore);
+}
+
+} // namespace ClangCodeModel::Internal