workspace/libs/taskmanager/strategies/programgroupingstrategy.cpp

   1 /*****************************************************************
   2
   3 Copyright 2008 Christian Mollekopf <chrigi_1@hotmail.com>
   4
   5 Permission is hereby granted, free of charge, to any person obtaining a copy
   6 of this software and associated documentation files (the "Software"), to deal
   7 in the Software without restriction, including without limitation the rights
   8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9 copies of the Software, and to permit persons to whom the Software is
  10 furnished to do so, subject to the following conditions:
  11
  12 The above copyright notice and this permission notice shall be included in
  13 all copies or substantial portions of the Software.
  14
  15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
  18 AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
  19 AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  20 CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  21
  22 ******************************************************************/
  23
  24 #include "programgroupingstrategy.h"
  25
  26 #include <QAction>
  27
  28 #include <KDebug>
  29 #include <KLocale>
  30
  31 #include "abstractgroupingstrategy.h"
  32 #include "groupmanager.h"
  33
  34 namespace TaskManager
  35 {
  36
  37 class ProgramGroupingStrategy::Private
  38 {
  39 public:
  40     Private()
  41         :editableGroupProperties(AbstractGroupingStrategy::None)
  42     {
  43     }
  44     GroupManager *groupManager;
  45     AbstractGroupingStrategy::EditableGroupProperties editableGroupProperties;
  46     AbstractGroupableItem *tempItem;
  47     QStringList blackList; //Programs in this list should not be grouped
  48 };
  49
  50
  51 ProgramGroupingStrategy::ProgramGroupingStrategy(GroupManager *groupManager)
  52     :AbstractGroupingStrategy(groupManager),
  53      d(new Private)
  54 {
  55     d->groupManager = groupManager;
  56     setType(GroupManager::ProgramGrouping);
  57 }
  58
  59 ProgramGroupingStrategy::~ProgramGroupingStrategy()
  60 {
  61     delete d;
  62 }
  63
  64 QList<QAction*> ProgramGroupingStrategy::strategyActions(QObject *parent, AbstractGroupableItem *item)
  65 {
  66     QAction *a = new QAction(parent);
  67     QString name = className(item);
  68     if (d->blackList.contains(name)) {
  69         a->setText(i18n("Allow This Program to Be Grouped"));
  70     } else {
  71         a->setText(i18n("Do Not Allow This Program to Be Grouped"));
  72     }
  73     connect(a, SIGNAL(triggered()), this, SLOT(toggleGrouping()));
  74
  75     QList<QAction*> actionList;
  76     actionList.append(a);
  77     d->tempItem = item;
  78     return actionList;
  79 }
  80
  81 QString ProgramGroupingStrategy::className(AbstractGroupableItem *item)
  82 {
  83     QString name;
  84     if (item->isGroupItem()) { //maybe add the condition that the subgroup was created by programGrouping
  85         TaskGroup *group = qobject_cast<TaskGroup*>(item);
  86         TaskItem *task = qobject_cast<TaskItem*>(group->members().first()); //There are only TaskItems in programGrouping groups
  87         return task->task()->classClass();
  88     }
  89
  90     return (qobject_cast<TaskItem*>(item))->task()->classClass();
  91 }
  92
  93 void ProgramGroupingStrategy::toggleGrouping()
  94 {
  95     QString name = className(d->tempItem);
  96
  97     if (d->blackList.contains(name)) {
  98         d->blackList.removeAll(name);
  99         if (d->tempItem->isGroupItem()) {
 100             foreach (AbstractGroupableItem *item, (qobject_cast<TaskGroup*>(d->tempItem))->members()) {
 101                 handleItem(item);
 102             }
 103         } else {
 104             handleItem(d->tempItem);
 105         }
 106     } else {
 107         d->blackList.append(name);
 108         if (d->tempItem->isGroupItem()) {
 109             closeGroup(qobject_cast<TaskGroup*>(d->tempItem));
 110         } else {
 111             d->groupManager->rootGroup()->add(d->tempItem);
 112         }
 113     }
 114     d->tempItem = 0;
 115 }
 116
 117 void ProgramGroupingStrategy::handleItem(AbstractItemPtr item)
 118 {
 119     if (item->isGroupItem()) {
 120         d->groupManager->rootGroup()->add(item);
 121         return;
 122     } else if (d->blackList.contains((qobject_cast<TaskItem*>(item))->task()->classClass())) {
 123         d->groupManager->rootGroup()->add(item);
 124         return;
 125     }
 126
 127     TaskItem *task = dynamic_cast<TaskItem*>(item);
 128     if (task && !programGrouping(task, d->groupManager->rootGroup())) {
 129         //kDebug() << "joined rootGroup ";
 130         d->groupManager->rootGroup()->add(item);
 131     }
 132 }
 133
 134 bool ProgramGroupingStrategy::programGrouping(TaskItem* taskItem, TaskGroup* groupItem)
 135 {
 136     //kDebug();
 137     QHash <QString,AbstractItemPtr> itemMap;
 138
 139     foreach (AbstractItemPtr item, groupItem->members()) { //search for an existing group
 140         if (item->isGroupItem()) { //maybe add the condition that the subgroup was created by programGrouping
 141             if (programGrouping(taskItem, static_cast<TaskGroup*>(item))) {
 142                 //kDebug() << "joined subGroup";
 143                 return true;
 144             }
 145         } else {
 146             TaskItem *task = static_cast<TaskItem*>(item);
 147             if (task->task()) { //omit startup tasks
 148                 QString name = task->task()->classClass();
 149                 itemMap.insertMulti(name,item);
 150             }
 151         }
 152     }
 153
 154     if (!itemMap.values().contains(taskItem)) {
 155         itemMap.insertMulti(taskItem->task()->classClass(), taskItem);
 156     }
 157
 158     QString name = taskItem->task()->classClass();
 159     if (itemMap.count(name) >= groupItem->members().count()) { //join this group
 160         //kDebug() << "joined this Group";
 161         groupItem->add(taskItem);
 162         return true;
 163     } else if (itemMap.count(name) >= 2) { //create new subgroup with at least 2 other task
 164         //kDebug() << "create Group";
 165         QIcon icon = taskItem->task()->icon();
 166         QList <AbstractItemPtr> list(itemMap.values(name));
 167         TaskGroup* group = createGroup(list);
 168         group->setName(name);
 169         group->setColor(Qt::red);
 170         group->setIcon(icon);
 171         return true;
 172     }
 173     return false;
 174 }
 175
 176 void ProgramGroupingStrategy::checkGroup()
 177 {
 178     TaskGroup *group = qobject_cast<TaskGroup*>(sender());
 179     if (!group) {
 180         return;
 181     }
 182     if (group->members().size() <= 1) {
 183         closeGroup(group);
 184     }
 185 }
 186
 187 }//namespace
 188
 189 #include "programgroupingstrategy.moc"
 190