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