1 files changed, 17 insertions, 0 deletions
diff --git a/pokemodr/models/MapEffectModel.cpp b/pokemodr/models/MapEffectModel.cpp
index 208a5c62..a43d2e53 100644
--- a/pokemodr/models/MapEffectModel.cpp
+++ b/pokemodr/models/MapEffectModel.cpp
@@ -18,6 +18,9 @@
 // Header include
 #include "MapEffectModel.h"
 
+// Model includes
+#include "GroupModel.h"
+
 // PokeModr includes
 #include "../MapEffectUI.h"
 
@@ -28,6 +31,9 @@
 // Qt includes
 #include <QFile>
 
+// KDE includes
+#include <KMenu>
+
 MapEffectModel::MapEffectModel(BaseModel* parent, MapEffect* effect) :
         ObjectModel(parent, effect)
 {
@@ -52,6 +58,12 @@ QVariant MapEffectModel::data(int role) const
         QWidget* widget = new MapEffectUI(static_cast<MapEffect*>(m_object), NULL);
         return QVariant::fromValue(widget);
     }
+    else if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Delete Effect", this, SLOT(deleteSelf()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
     return ObjectModel::data(role);
 }
 
@@ -76,3 +88,8 @@ bool MapEffectModel::setData(const QVariant& value, int role)
     }
     return false;
 }
+
+void MapEffectModel::deleteSelf()
+{
+//     static_cast<GroupModel*>(m_parent)->deleteObject(this);
+}