DataTreeBuilderBase.cpp

Go to the documentation of this file.

#include "DataTreeBuilderBase.h"
 
#include <QTreeWidgetItem>
 
#include <ArmarXCore/core/exceptions/local/ExpressionException.h>
// #include <ArmarXCore/core/logging/Logging.h>
 
#include <RobotAPI/libraries/armem_gui/TreeWidgetBuilder.h>
#include <RobotAPI/libraries/armem_gui/instance/display_visitors/DataDisplayVisitor.h>
#include <RobotAPI/libraries/armem_gui/instance/sanitize_typename.h>
 
namespace armarx::armem::gui::instance
{
 
    const int keyIndexRole = Qt::UserRole + 10;
 
    DataTreeBuilderBase::DataTreeBuilderBase()
    {
    }
 
    DataTreeBuilderBase::~DataTreeBuilderBase()
    {
    }
 
    void
    DataTreeBuilderBase::setColumns(int key, int value, int type)
    {
        this->columnKey = key;
        this->columnType = type;
        this->columnValue = value;
    }
 
    QTreeWidgetItem*
    DataTreeBuilderBase::makeItem(const std::string& key) const
    {
        return new QTreeWidgetItem(QStringList{QString::fromStdString(key)});
    }
 
    QTreeWidgetItem*
    DataTreeBuilderBase::makeItem(size_t key) const
    {
        QTreeWidgetItem* item = new QTreeWidgetItem();
        item->setData(0, keyIndexRole, static_cast<int>(key));
        return item;
    }
 
    void
    DataTreeBuilderBase::setRowTexts(QTreeWidgetItem* item,
                                     const std::string& key,
                                     const std::string& value,
                                     const std::string& typeName) const
    {
        item->setText(columnKey, QString::fromStdString(key));
        item->setText(columnValue, QString::fromStdString(value));
        item->setText(columnType, QString::fromStdString(typeName));
    }
 
    void
    DataTreeBuilderBase::setRowTexts(QTreeWidgetItem* item,
                                     const std::string& key,
                                     const aron::data::VariantPtr& data)
    {
        if (!data)
        {
            return;
        }
        const std::string value = armarx::aron::DataDisplayVisitor::getValue(data);
        setRowTexts(item, key, value, sanitizeTypeName(data->getFullName()));
    }
 
    DataTreeBuilderBase::DictBuilder
    DataTreeBuilderBase::getDictBuilder() const
    {
        DictBuilder builder;
        builder.setCompareFn([](const std::string& key, QTreeWidgetItem* item)
                             { return armarx::detail::compare(key, item->text(0).toStdString()); });
        builder.setMakeItemFn([this](const std::string& key) { return this->makeItem(key); });
        return builder;
    }
 
    DataTreeBuilderBase::ListBuilder
    DataTreeBuilderBase::getListBuilder() const
    {
        ListBuilder builder;
        builder.setCompareFn(
            [](size_t key, QTreeWidgetItem* item)
            {
                QVariant itemKey = item->data(0, keyIndexRole);
                ARMARX_CHECK_EQUAL(itemKey.type(), QVariant::Type::Int);
                // << VAROUT(key) << " | " << VAROUT(item->text(0).toStdString()) << itemKey.typeName();
 
                return armarx::detail::compare(static_cast<int>(key), itemKey.toInt());
            });
        builder.setMakeItemFn([this](size_t key) { return this->makeItem(key); });
        return builder;
    }
 
    std::vector<size_t>
    DataTreeBuilderBase::getIndex(size_t size) const
    {
        std::vector<size_t> index;
        index.reserve(size);
        for (size_t i = 0; i < size; ++i)
        {
            index.push_back(i);
        }
        return index;
    }
 
} // namespace armarx::armem::gui::instance