1 files changed, 27 insertions, 5 deletions
diff --git a/pokemodr/models/MapModel.cpp b/pokemodr/models/MapModel.cpp
index 5a0e76a0..609c9a90 100644
--- a/pokemodr/models/MapModel.cpp
+++ b/pokemodr/models/MapModel.cpp
@@ -33,10 +33,12 @@
 // Qt includes
 #include <QFile>
 
+// KDE includes
+#include <KMenu>
+
 MapModel::MapModel(BaseModel* parent, Map* map) :
         GroupObjectModel(parent, map)
 {
-    setupData();
 }
 
 MapModel::~MapModel()
@@ -58,6 +60,12 @@ QVariant MapModel::data(int role) const
         QWidget* widget = new MapUI(static_cast<Map*>(m_object), NULL);
         return QVariant::fromValue(widget);
     }
+    else if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Delete Map", this, SLOT(deleteSelf()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
     return GroupObjectModel::data(role);
 }
 
@@ -85,10 +93,24 @@ bool MapModel::setData(const QVariant& value, int role)
     return false;
 }
 
+void MapModel::addObject(Object* /*object*/)
+{
+}
+
+void MapModel::deleteObject(BaseModel* /*model*/)
+{
+}
+
+void MapModel::deleteSelf()
+{
+//     static_cast<GroupModel*>(m_parent)->deleteObject(this);
+}
+
 void MapModel::setupData()
 {
-    m_objects.append(new MapEffectGroupModel(this, static_cast<Map*>(m_object)));
-    m_objects.append(new MapTrainerGroupModel(this, static_cast<Map*>(m_object)));
-    m_objects.append(new MapWarpGroupModel(this, static_cast<Map*>(m_object)));
-    m_objects.append(new MapWildListGroupModel(this, static_cast<Map*>(m_object)));
+    Map* map = static_cast<Map*>(m_object);
+    m_objects.append(new MapEffectGroupModel(this, map));
+    m_objects.append(new MapTrainerGroupModel(this, map));
+    m_objects.append(new MapWarpGroupModel(this, map));
+    m_objects.append(new MapWildListGroupModel(this, map));
 }