KisResourceItemChooserContextMenu.cpp

Go to the documentation of this file.

 
/* This file is part of the KDE project
 *    SPDX-FileCopyrightText: 2013 Sascha Suelzer <s.suelzer@gmail.com>
 *    SPDX-FileCopyrightText: 2020 Agata Cacko <cacko.azh@gmail.com>
 *
 *    SPDX-License-Identifier: LGPL-2.0-or-later
 * */
 
#include "KisResourceItemChooserContextMenu.h"
 
#include <QDebug>
#include <QGridLayout>
 
#include <KoIcon.h>
#include <klocalizedstring.h>
#include <KoResource.h>
#include <KisResourceModel.h>
#include <KisTagResourceModel.h>
#include <KisTag.h>
 
 
#include "kis_debug.h"
 
bool isSpecialTag(KisTagSP tag) {
    return !tag.isNull() && tag->id() < 0;
}
 
 
KisResourceItemChooserContextMenu::KisResourceItemChooserContextMenu(KoResourceSP resource,
                                                                     const KisTagSP currentlySelectedTag,
                                                                     KisTagChooserWidget *tagChooser)
    : m_tagChooserWidget(tagChooser)
{
    QImage image = resource->image();
    QIcon icon(QPixmap::fromImage(image));
    QAction *label = new QAction(resource->name(), this);
    label->setIcon(icon);
 
    addAction(label);
 
    QMenu *removableTagsMenu;
    QMenu *assignableTagsMenu;
 
    m_tagModel = new KisTagModel(resource->resourceType().first);
    m_tagModel->sort(KisAllTagsModel::Name);
    KisResourceModel resourceModel(resource->resourceType().first);
    KisTagResourceModel tagResourceModel(resource->resourceType().first);
 
    tagResourceModel.setResourcesFilter(QVector<KoResourceSP>() << resource);
    tagResourceModel.sort(KisAllTagResourceModel::TagName);
 
    QList<KisTagSP> removableTags;
    for (int i = 0; i < tagResourceModel.rowCount(); ++i) {
        KisTagSP tag = tagResourceModel.data(tagResourceModel.index(i, 0), Qt::UserRole + KisAllTagResourceModel::Tag).value<KisTagSP>();
        if (!tag.isNull() && tag->valid() && !isSpecialTag(tag)) {
            removableTags << tag;
        }
    }
 
    QList<KisTagSP> assignableTags;
    for (int i = 0; i < m_tagModel->rowCount(); i++) {
        QModelIndex index = m_tagModel->index(i, 0);
        KisTagSP tag = m_tagModel->tagForIndex(index);
         if (!tag.isNull() && tag->valid() && !isSpecialTag(tag)) {
             assignableTags << tag;
         }
     }
 
    CompareWithOtherTagFunctor comparer(currentlySelectedTag);
 
    bool currentTagInRemovables = !currentlySelectedTag.isNull();
    currentTagInRemovables = currentTagInRemovables
            && (std::find_if(removableTags.begin(), removableTags.end(), comparer) != removableTags.end());
 
    assignableTagsMenu = addMenu(koIcon("list-add"),i18n("Assign to tag"));
 
    if (!removableTags.isEmpty()) {
        addSeparator();
        KisTagSP currentTag = currentlySelectedTag;
 
        if (!currentTag.isNull() && currentTagInRemovables) {
            // remove the current tag from both "Remove from tag: " and "Assign to tag: " lists
            removableTags.erase(std::remove_if(removableTags.begin(),
                                               removableTags.end(),
                                               comparer),
                                removableTags.end());
            assignableTags.erase(std::remove_if(assignableTags.begin(),
                                                assignableTags.end(),
                                                comparer),
                                 assignableTags.end());
 
            SimpleExistingTagAction *removeTagAction = new SimpleExistingTagAction(resource, currentTag, this);
            removeTagAction->setText(i18n("Remove from this tag"));
            removeTagAction->setIcon(koIcon("list-remove"));
 
            connect(removeTagAction, SIGNAL(triggered(const KisTagSP, KoResourceSP)),
                    this, SLOT(removeResourceExistingTag(const KisTagSP, KoResourceSP)));
            addAction(removeTagAction);
        }
 
        if (!removableTags.isEmpty()) {
            removableTagsMenu = addMenu(koIcon("list-remove"),i18n("Remove from other tag"));
 
            KisTagSP empty;
            CompareWithOtherTagFunctor compareWithRemovable(empty);
            foreach (const KisTagSP tag, removableTags) {
 
                if (tag.isNull()) {
                    continue;
                }
 
                compareWithRemovable.setReferenceTag(tag);
                assignableTags.erase(std::remove_if(assignableTags.begin(),
                                                    assignableTags.end(),
                                                    compareWithRemovable),
                                     assignableTags.end());
 
                SimpleExistingTagAction *removeTagAction = new SimpleExistingTagAction(resource, tag, this);
 
                connect(removeTagAction, SIGNAL(triggered(const KisTagSP, KoResourceSP)),
                        this, SLOT(removeResourceExistingTag(const KisTagSP, KoResourceSP)));
                removableTagsMenu->addAction(removeTagAction);
            }
        }
 
    }
 
    foreach (const KisTagSP &tag, assignableTags) {
        if (tag.isNull()) {
            continue;
        }
 
        SimpleExistingTagAction * addTagAction = new SimpleExistingTagAction(resource, tag, this);
 
        connect(addTagAction, SIGNAL(triggered(const KisTagSP, KoResourceSP)),
                this, SLOT(addResourceExistingTag(const KisTagSP, KoResourceSP)));
 
        assignableTagsMenu->addAction(addTagAction);
    }
 
    assignableTagsMenu->addSeparator();
 
    NewTagResourceAction *addNewTagAction = new NewTagResourceAction(resource, this);
    connect(addNewTagAction, SIGNAL(triggered(QString, KoResourceSP)), m_tagChooserWidget, SLOT(addTag(QString, KoResourceSP)));
 
    assignableTagsMenu->addAction(addNewTagAction);
 
}
 
KisResourceItemChooserContextMenu::~KisResourceItemChooserContextMenu()
{
    delete m_tagModel;
}
 
void KisResourceItemChooserContextMenu::removeResourceExistingTag(const KisTagSP tag, KoResourceSP resource)
{
    KisTagResourceModel tagResourceModel(resource->resourceType().first);
    tagResourceModel.untagResources(tag, QVector<int>() << resource->resourceId());
}
 
void KisResourceItemChooserContextMenu::addResourceExistingTag(const KisTagSP tag, KoResourceSP resource)
{
    KisTagResourceModel tagResourceModel(resource->resourceType().first);
    tagResourceModel.tagResources(tag, QVector<int>() << resource->resourceId());
}