ladybird/Ladybird/ModelTranslator.cpp

88 lines
2.3 KiB
C++

/*
* 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());
}
}