Ladybird: Add a ModelTranslator to translate a GUI::Model to a Qt model

This will be used to expose the DOMTreeModel from LibWebView :^)
Author: https://github.com/awesomekling Commit: https://github.com/SerenityOS/serenity/commit/98b6aea234 Pull-request: https://github.com/SerenityOS/serenity/pull/16583 Reviewed-by: https://github.com/ADKaster Reviewed-by: https://github.com/linusg
2024-09-29 16:21:29 +00:00 · 2022-09-25 12:08:42 +02:00 · 2022-09-25 12:08:42 +02:00 · 98b6aea234 · 2024-07-17 16:23:06 +09:00
parent 2a021084e5
commit 98b6aea234
2 changed files with 117 additions and 0 deletions
--- a/Ladybird/ModelTranslator.cpp
+++ b/Ladybird/ModelTranslator.cpp
@ -0,0 +1,84 @@
+/*
+ * Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#define AK_DONT_REPLACE_STD
+
+#include "ModelTranslator.h"
+#include "Utilities.h"
+#include <QIcon>
+
+namespace Ladybird {
+
+ModelTranslator::ModelTranslator(NonnullRefPtr<GUI::Model> model)
+    : m_model(move(model))
+{
+}
+
+ModelTranslator::~ModelTranslator() = default;
+
+int ModelTranslator::columnCount(QModelIndex const& parent) const
+{
+    return m_model->column_count(to_gui(parent));
+}
+
+int ModelTranslator::rowCount(QModelIndex const& parent) const
+{
+    return m_model->row_count(to_gui(parent));
+}
+
+static QVariant convert_variant(GUI::Variant const& value)
+{
+    if (value.is_string())
+        return qstring_from_akstring(value.as_string());
+    if (value.is_icon()) {
+        auto const& gui_icon = value.as_icon();
+        auto bitmap = gui_icon.bitmap_for_size(16);
+        VERIFY(bitmap);
+        auto qt_image = QImage(bitmap->scanline_u8(0), 16, 16, QImage::Format_ARGB32);
+        QIcon qt_icon;
+        qt_icon.addPixmap(QPixmap::fromImage(qt_image.convertToFormat(QImage::Format::Format_ARGB32_Premultiplied)));
+        return qt_icon;
+    }
+    return {};
+}
+
+QVariant ModelTranslator::data(QModelIndex const& index, int role) const
+{
+    switch (role) {
+    case Qt::DisplayRole:
+        return convert_variant(m_model->data(to_gui(index), GUI::ModelRole::Display));
+    case Qt::DecorationRole:
+        return convert_variant(m_model->data(to_gui(index), GUI::ModelRole::Icon));
+    default:
+        return {};
+    }
+}
+
+QModelIndex ModelTranslator::index(int row, int column, QModelIndex const& parent) const
+{
+    return to_qt(m_model->index(row, column, to_gui(parent)));
+}
+
+QModelIndex ModelTranslator::parent(QModelIndex const& index) const
+{
+    return to_qt(m_model->parent_index(to_gui(index)));
+}
+
+QModelIndex ModelTranslator::to_qt(GUI::ModelIndex const& index) const
+{
+    if (!index.is_valid())
+        return {};
+    return createIndex(index.row(), index.column(), index.internal_data());
+}
+
+GUI::ModelIndex ModelTranslator::to_gui(QModelIndex const& index) const
+{
+    if (!index.isValid())
+        return {};
+    return m_model->create_index_unsafe(index.row(), index.column(), index.internalPointer());
+}
+
+}
--- a/Ladybird/ModelTranslator.h
+++ b/Ladybird/ModelTranslator.h
@ -0,0 +1,33 @@
+/*
+ * Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#pragma once
+
+#include <LibGUI/Model.h>
+#include <QAbstractItemModel>
+
+namespace Ladybird {
+
+class ModelTranslator final : public QAbstractItemModel {
+    Q_OBJECT
+public:
+    explicit ModelTranslator(NonnullRefPtr<GUI::Model>);
+    virtual ~ModelTranslator() override;
+
+    virtual int columnCount(QModelIndex const& parent) const override;
+    virtual int rowCount(QModelIndex const& parent) const override;
+    virtual QVariant data(QModelIndex const&, int role) const override;
+    virtual QModelIndex index(int row, int column, QModelIndex const& parent) const override;
+    virtual QModelIndex parent(QModelIndex const& index) const override;
+
+    QModelIndex to_qt(GUI::ModelIndex const&) const;
+    GUI::ModelIndex to_gui(QModelIndex const&) const;
+
+private:
+    RefPtr<GUI::Model> m_model;
+};
+
+}