summaryrefslogtreecommitdiff
path: root/Ladybird/ModelTranslator.cpp
blob: 31d61954debcc645c5e21832a78a51fa21bb994f (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
/*
 * 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() = default;

int ModelTranslator::columnCount(QModelIndex const& parent) const
{
    if (!m_model)
        return 0;
    return m_model->column_count(to_gui(parent));
}

int ModelTranslator::rowCount(QModelIndex const& parent) const
{
    if (!m_model)
        return 0;
    return m_model->row_count(to_gui(parent));
}

static QVariant convert_variant(GUI::Variant const& value)
{
    if (value.is_string())
        return qstring_from_ak_deprecated_string(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
{
    VERIFY(m_model);
    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
{
    VERIFY(m_model);
    return to_qt(m_model->index(row, column, to_gui(parent)));
}

QModelIndex ModelTranslator::parent(QModelIndex const& index) const
{
    VERIFY(m_model);
    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
{
    VERIFY(m_model);
    if (!index.isValid())
        return {};
    return m_model->unsafe_create_index(index.row(), index.column(), index.internalPointer());
}

}