File indexing completed on 2025-04-27 04:08:16

 /*
  * This file is part of the KDE project
  * SPDX-FileCopyrightText: 2014 Dan Leinir Turthra Jensen <admin@leinir.dk>
  *
  * SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
  *
  */
 
 #include "TemplatesModel.h"
 #include <KisTemplateTree.h>
 #include <KisTemplateGroup.h>
 #include <KisTemplate.h>
 #include <KisPart.h>
 #include <kis_icon.h>
 
 
 struct TemplatesModel::ItemData {
 public:
     ItemData()
         : favourite(false)
         , groupFolded(false)
     {};
     QString name;
     QString description;
     QString file;
     QString icon;
     QString groupName;
     bool favourite;
     bool groupFolded;
 };
 
 class TemplatesModel::Private {
 public:
     Private() {}
     ~Private() { qDeleteAll(items);}
     QList<ItemData*> items;
 };
 
 TemplatesModel::TemplatesModel(QObject* parent)
     : QAbstractListModel(parent)
     , d(new Private)
 {
     // Prefill a couple of
     ItemData* customItem = new ItemData();
     customItem->name = "Custom Image";
     customItem->file = "custom";
     customItem->icon = "filenew-black";
     d->items << customItem;
     ItemData* clipItem = new ItemData();
     clipItem->name = "From Clipboard";
     clipItem->file = "clip";
     clipItem->icon = "fileclip-black";
     d->items << clipItem;
     ItemData* screenItem = new ItemData();
     screenItem->name = "Blank Image (Screen Size)";
     screenItem->file = "screen";
     screenItem->icon = "filenew-black";
     d->items << screenItem;
     ItemData* a4pItem = new ItemData();
     a4pItem->name = "Blank Image (A4 Portrait)";
     a4pItem->file = "a4p";
     a4pItem->icon = "A4portrait-black";
     d->items << a4pItem;
     ItemData* a4lItem = new ItemData();
     a4lItem->name = "Blank Image (A4 Landscape)";
     a4lItem->file = "a4l";
     a4lItem->icon = "A4landscape-black";
     d->items << a4lItem;
 
     populate();
 }
 
 TemplatesModel::~TemplatesModel()
 {
     delete d;
 }
 
 QHash<int, QByteArray> TemplatesModel::roleNames() const
 {
     QHash<int, QByteArray> roleNames;
     roleNames[NameRole] = "name";
     roleNames[DescriptionRole] = "description";
     roleNames[FileRole] = "file";
     roleNames[IconRole] = "icon";
     roleNames[GroupName] = "groupName";
     roleNames[GroupFolded] = "groupFolded";
     return roleNames;
 }
 
 QVariant TemplatesModel::data(const QModelIndex& index, int role) const
 {
     QVariant data;
     if(index.isValid() && index.row() > -1 && index.row() < d->items.count())
     {
         ItemData* item = d->items[index.row()];
         switch(role) {
             case NameRole:
                 data = item->name;
                 break;
             case DescriptionRole:
                 data = item->description;
                 break;
             case FileRole:
                 data = item->file;
                 break;
             case IconRole:
                 data = item->icon;
                 break;
             case GroupName:
                 data = item->groupName;
                 break;
             case GroupFolded:
                 data = item->groupFolded;
                 break;
             default:
                 break;
         }
     }
     return data;
 }
 
 int TemplatesModel::rowCount(const QModelIndex& parent) const
 {
     if(parent.isValid())
         return 0;
     return d->items.count();
 }
 
 QString TemplatesModel::groupNameOf(int index) const
 {
     if(index > 0 && index < d->items.count())
         return d->items[index]->groupName;
     return QString();
 }
 
 void TemplatesModel::toggleGroup(const QString& name)
 {
     Q_FOREACH (ItemData* item, d->items) {
         if(item->groupName == name)
             item->groupFolded = !item->groupFolded;
     }
     dataChanged(index(0), index(d->items.count() - 1));
 }
 
 void TemplatesModel::populate()
 {
     KisTemplateTree templateTree( QStringLiteral("templates/"), true);
     Q_FOREACH (KisTemplateGroup *group, templateTree.groups()) {
         if (group->isHidden()) {
             continue;
         }
         Q_FOREACH (KisTemplate* t, group->templates()) {
             if (t->isHidden())
                 continue;
 
             ItemData* item = new ItemData();
             item->name = t->name();
             item->description = t->description();
             item->file = QString("template://").append(t->file());
             // QT5TODO: support custom pictures and icon symbol ids again
             // item->icon = KIconLoader::global()->iconPath(t->picture(), KIconLoader::Desktop);
             item->icon = "filenew-black";
             item->groupName = group->name();
             item->groupFolded = true; // default hide groups
             d->items << item;
         }
     }
 }