AronTreeWidgetContextMenu.cpp

Go to the documentation of this file.

#include "AronTreeWidgetContextMenu.h"
 
#include <QMenu>
#include <QPoint>
#include <QTreeWidget>
#include <QTreeWidgetItem>
 
#include "../AronTreeWidgetItem.h"
#include "../ListDictHelper.h"
#include "AronTreeWidgetCreator.h"
 
namespace armarx::skills::gui
{
    AronTreeWidgetContextMenuVisitor::AronTreeWidgetContextMenuVisitor(
        AronTreeWidgetItem* i,
        const QPoint& pos,
        QTreeWidget* contextMenuParent,
        int x) :
        parentItem(i), contextMenuParent(contextMenuParent), pos(pos), index(x)
    {
    }
 
    void
    AronTreeWidgetContextMenuVisitor::addDeleteAction()
    {
        auto* castedParent = AronTreeWidgetItem::DynamicCast(parentItem->QTreeWidgetItem::parent());
        if (!castedParent)
        {
            // must be top level element
            return;
        }
        auto aronType = castedParent->aronType->getDescriptor();
        if (aron::type::Descriptor::DICT == aronType || aron::type::Descriptor::LIST == aronType)
        {
            QMenu contextMenu("Context menu", contextMenuParent);
            actions.emplace_back("remove element", contextMenuParent);
            action_callbacks.push_back([this]() mutable { this->executeDelete(); });
        }
    }
 
    void
    AronTreeWidgetContextMenuVisitor::executeDelete()
    {
        auto* containerPtr = parentItem->QTreeWidgetItem::parent();
        containerPtr->removeChild(parentItem);
        auto* castedContainer = AronTreeWidgetItem::DynamicCast(containerPtr);
 
        // if the parent item is a List, we need to redo the numbers
        if (castedContainer &&
            castedContainer->aronType->getDescriptor() == aron::type::Descriptor::LIST)
        {
            // start renumbering from the removed child onwards
            for (int i = index; i < castedContainer->childCount(); ++i)
            {
                std::string numberString = std::to_string(i);
                castedContainer->child(i)->setText(0, numberString.c_str());
            }
            // This displays the number of children also when the list is collapsed
            QString numElemsText = misc::generateNumElementsText(castedContainer->childCount());
            containerPtr->setText(1, numElemsText);
            // set italic
            auto currFont = castedContainer->font(1);
            currFont.setItalic(true);
            castedContainer->setFont(1, currFont);
        }
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::ObjectPtr&)
    {
        addDeleteAction();
    }
 
    // lol
    void
    AronTreeWidgetContextMenuVisitor::addAddAction()
    {
        actions.emplace_back("Add element", contextMenuParent);
        action_callbacks.push_back([this]() mutable { this->executeAddElement(); });
    }
 
    void
    AronTreeWidgetContextMenuVisitor::executeAddElement()
    {
        AronTreeWidgetCreatorVisitor creator(parentItem);
        aron::type::visit(creator, parentItem->aronType->getChildren()[0]);
 
        if (!creator.createdQWidgetItem)
        {
            throw std::runtime_error("Creation of TreeElementChild failed unexpectedly");
        }
        // if it is a list, we update the number of children at the top
        auto* castedContainer = AronTreeWidgetItem::DynamicCast(parentItem);
 
        // if the parent item is a List, we need to redo the numbers
        if (castedContainer &&
            castedContainer->aronType->getDescriptor() == aron::type::Descriptor::LIST)
        {
            // This displays the number of children also when the list is collapsed
            auto numElemsText = misc::generateNumElementsText(castedContainer->childCount());
            castedContainer->setText(1, numElemsText);
            // set italic
            auto currFont = castedContainer->font(1);
            currFont.setItalic(true);
            castedContainer->setFont(1, currFont);
        }
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::DictPtr&)
    {
        addAddAction();
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::PairPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::TuplePtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::ListPtr&)
    {
        addAddAction();
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::NDArrayPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::MatrixPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::QuaternionPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::ImagePtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::PointCloudPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::IntEnumPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::IntPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::LongPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::FloatPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::DoublePtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::BoolPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitAronVariant(const aron::type::StringPtr&)
    {
        addDeleteAction();
    }
 
    void
    AronTreeWidgetContextMenuVisitor::visitUnknown(Input&)
    {
        ARMARX_WARNING << "Tried to open Context menu on unknown aron type";
    }
 
    void
    AronTreeWidgetContextMenuVisitor::showMenuAndExecute()
    {
        QMenu menu("Context Menu", contextMenuParent);
        for (auto& el : actions)
        {
            menu.addAction(&el);
        }
        auto* chosenAction = menu.exec(contextMenuParent->mapToGlobal(pos));
 
        if (!chosenAction)
        {
            return;
        }
 
        // not elegant, but is a small loop anyway
        auto it = actions.begin();
        size_t count = 0;
        while (it != actions.end())
        {
            if (chosenAction == &*it)
            {
                action_callbacks[count]();
                break;
            }
            ++it;
            ++count;
        }
    }
} // namespace armarx::skills::gui