1 files changed, 77 insertions, 11 deletions

diff --git a/libkdepim/categoryselectdialog.cpp b/libkdepim/categoryselectdialog.cpp
index 7df9154..8b2bc4e 100644
--- a/libkdepim/categoryselectdialog.cpp
+++ b/libkdepim/categoryselectdialog.cpp
@@ -28,2 +28,3 @@
 #include <qmessagebox.h>
+#include <kmessagebox.h>
 
@@ -43,11 +44,15 @@ CategorySelectDialog::CategorySelectDialog( KPimPrefs *prefs, QWidget* parent,
 {
-  mCategories->header()->hide();
+    mColorItem = 0;
+    mColorEnabled = false;
+    mCategories->header()->hide();
 
-  setCategories();
+    setCategories();
  
-  connect(mButtonEdit,SIGNAL(clicked()),this, SLOT(editCategoriesDialog()));
-  if ( QApplication::desktop()->width() > 460 )
-      resize( 300, 360 );
-  else
-      showMaximized();
+    connect(mButtonEdit,SIGNAL(clicked()),this, SLOT(editCategoriesDialog()));
+    if ( QApplication::desktop()->width() > 460 )
+        resize( 300, 360 );
+    else
+        showMaximized();
+    connect( mSetColorCat, SIGNAL( clicked() ), this, SLOT( setColorCat() ) );
+    connect( mCategories, SIGNAL( clicked(QListViewItem *) ), this, SLOT( clicked(QListViewItem *) ) );
 }
@@ -57,3 +62,3 @@ void CategorySelectDialog::editCategoriesDialog()
     
-    ced->exec();
+    ced->exec(); 
     delete ced;
@@ -63,2 +68,3 @@ void CategorySelectDialog::setCategories()
 {
+  mColorItem = 0;
   mCategories->clear();
@@ -67,3 +73,2 @@ void CategorySelectDialog::setCategories()
   QStringList::Iterator it;
-
   for (it = mPrefs->mCustomCategories.begin();
@@ -91,2 +96,4 @@ void CategorySelectDialog::setSelected(const QStringList &selList)
         item->setOn(true);
+        if ( ! found )
+            setColorItem( item );
         found = true;
@@ -115,3 +122,56 @@ QStringList CategorySelectDialog::selectedCategories() const
 }
+void CategorySelectDialog::setColorEnabled()
+{
+    mColorEnabled = true;
+    mSetColorCat->show();
+}
+void CategorySelectDialog::setColorCat()
+{
+    QCheckListItem * newColorItem = (QCheckListItem * )mCategories->currentItem ();
+   if ( !newColorItem ) {
+       KMessageBox::error(this,i18n("There is no current item."));
+       return;
+   }
+   if ( !newColorItem->isOn() )
+       newColorItem->setOn( true );
+   setColorItem( newColorItem );
 
+}
+void CategorySelectDialog::clicked ( QListViewItem * it )
+{
+    if ( ! it )
+        return;
+    QCheckListItem *i = (QCheckListItem *) it;
+    qDebug("click %d ", i->isOn());
+    if ( !i->isOn() && i==mColorItem) {
+        setColorItem( 0);
+        QCheckListItem *item = (QCheckListItem *)mCategories->firstChild();
+        while (item) {
+            if (item->isOn()) {
+                setColorItem( item );
+                break;
+            }
+            item = (QCheckListItem *)item->nextSibling();
+        }
+    } else if ( i->isOn() && !mColorItem) {
+        setColorItem( i);
+    }
+}
+void  CategorySelectDialog::setColorItem( QCheckListItem * newColorItem )
+{
+    if ( !mColorEnabled )
+        return;
+    if ( mColorItem == newColorItem) 
+        return;
+    if ( mColorItem ) {
+        mColorItem->setPixmap ( 0, QPixmap() );
+        mColorItem = 0;
+    }
+    if ( newColorItem ) {
+        QPixmap pix (newColorItem->height()/2, newColorItem->height()/2 );
+        pix.fill(Qt::blue );
+        newColorItem->setPixmap ( 0, pix );
+        mColorItem = newColorItem;
+    }
+}
 void CategorySelectDialog::slotApply()
@@ -120,5 +180,9 @@ void CategorySelectDialog::slotApply()
   QCheckListItem *item = (QCheckListItem *)mCategories->firstChild();
+  QString colcat;
   while (item) {
     if (item->isOn()) {
-      categories.append(item->text());
+        if ( item == mColorItem)
+            colcat = item->text();
+        else
+            categories.append(item->text());
     }
@@ -126,3 +190,5 @@ void CategorySelectDialog::slotApply()
   }
-  
+  categories.sort();
+  if ( ! colcat.isEmpty() )
+      categories.prepend( colcat );
   QString categoriesStr = categories.join(",");