File indexing completed on 2024-05-12 15:59:57

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