1 files changed, 41 insertions, 3 deletions

diff --git a/korganizer/koeventpopupmenu.cpp b/korganizer/koeventpopupmenu.cpp
index 17ef81e..90caaf4 100644
--- a/korganizer/koeventpopupmenu.cpp
+++ b/korganizer/koeventpopupmenu.cpp
@@ -27,14 +27,15 @@
 #include <kdebug.h>
 #include <kiconloader.h>
 
 #include <libkcal/event.h>
 
 #include "koeventpopupmenu.h"
+#include "koprefs.h"
 
-KOEventPopupMenu::KOEventPopupMenu()
+KOEventPopupMenu::KOEventPopupMenu(): QPopupMenu()
 {
   mCurrentIncidence = 0;
   mHasAdditionalItems = false;
 
   
   mSingleOnlyItems.append( insertItem (i18n("&Show"),this,SLOT(popupShow())));
@@ -49,27 +50,66 @@ KOEventPopupMenu::KOEventPopupMenu()
   mEditOnlyItems.append(insertItem (i18n("&Beam..."),
       this,SLOT(popupBeam())));
 #endif
   mEditOnlyItems.append(insertItem (i18n("&Toggle Cancel"),
       this,SLOT(popupCancel())));
    isDisabled = false;  
+   mCatPopup = new QPopupMenu ( this );
+   mCatPopup->setCheckable (true);
+   connect(mCatPopup,SIGNAL( aboutToShow ()), this ,SLOT( fillCatPopup()));
+   connect(mCatPopup,SIGNAL(  activated ( int ) ), this ,SLOT( computeCatPopup( int )));
+   mEditOnlyItems.append(insertItem (i18n("Categories"),mCatPopup ));
    QValueList<int>::Iterator it;
    for( it = mEditOnlyItems.begin(); it != mEditOnlyItems.end(); ++it ) {
        mSingleOnlyItems.append(*it);
    }
+
 }
 void KOEventPopupMenu::enableDefault( bool enable )
 {
     isDisabled = !enable;
     QValueList<int>::Iterator it;
     for( it = mSingleOnlyItems.begin(); it != mSingleOnlyItems.end(); ++it ) {
         setItemEnabled(*it,enable);
     }
     
 }
 
+void KOEventPopupMenu::fillCatPopup()
+{
+    mCatPopup->clear(); 
+    QStringList checkedCategories = mCurrentIncidence->categories();
+    int index = 0;
+    for (QStringList::Iterator it = KOPrefs::instance()->mCustomCategories.begin ();
+         it != KOPrefs::instance()->mCustomCategories.end ();
+         ++it) {
+        mCatPopup->insertItem (*it, index );
+        if (checkedCategories.find (*it) != checkedCategories.end ()) {
+            mCatPopup->setItemChecked (index, true);
+        }
+        ++index;
+    }
+}
+void KOEventPopupMenu::computeCatPopup( int index )
+{
+    QStringList categories =  mCurrentIncidence->categories();
+    QString colcat = categories.first();
+    if (categories.find (KOPrefs::instance()->mCustomCategories[index]) != categories.end ())
+      categories.remove (KOPrefs::instance()->mCustomCategories[index]);
+    else
+        categories.insert (categories.end(), KOPrefs::instance()->mCustomCategories[index]);
+    categories.sort ();
+    if ( !colcat.isEmpty() ) {
+        if ( categories.find ( colcat ) != categories.end () ) {
+            categories.remove( colcat );
+            categories.prepend( colcat );
+        }
+    }
+     mCurrentIncidence->setCategories( categories );
+    emit categoryChanged( mCurrentIncidence );
+}
 void KOEventPopupMenu::showIncidencePopup(Incidence *incidence)
 {
   mCurrentIncidence = incidence;
   
   if (mCurrentIncidence) {
     // Enable/Disabled menu items only valid for editable events.
@@ -77,14 +117,12 @@ void KOEventPopupMenu::showIncidencePopup(Incidence *incidence)
           QValueList<int>::Iterator it;
           for( it = mEditOnlyItems.begin(); it != mEditOnlyItems.end(); ++it ) {
               setItemEnabled(*it,!mCurrentIncidence->isReadOnly());
           }
       }
     popup(QCursor::pos());
-  } else {
-    kdDebug() << "KOEventPopupMenu::showEventPopup(): No event selected" << endl;
   }
 }
 
 void KOEventPopupMenu::addAdditionalItem(const QIconSet &icon,const QString &text,
                                     const QObject *receiver, const char *member,
                                     bool editOnly)