From ff8c440c3a2aeb0e1abc6f93e9dd7775533382ff Mon Sep 17 00:00:00 2001
From: Ben Boeckel <MathStuf@gmail.com>
Date: Mon, 26 May 2008 01:32:07 +0000
Subject: [FIX] Context menus added [FIX] No more copy/cut/paste for objects;
 only available from context menus [FIX] Can now add objects during runtime

git-svn-id: https://pokegen.svn.sourceforge.net/svnroot/pokegen/trunk@174 6ecfd1a5-f3ed-3746-8530-beee90d26b22
---
 pokemodr/models/MapWildListModel.cpp | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

(limited to 'pokemodr/models/MapWildListModel.cpp')

diff --git a/pokemodr/models/MapWildListModel.cpp b/pokemodr/models/MapWildListModel.cpp
index 181d0ccd..a2154def 100644
--- a/pokemodr/models/MapWildListModel.cpp
+++ b/pokemodr/models/MapWildListModel.cpp
@@ -27,10 +27,14 @@
 // Pokemod includes
 #include "../../pokemod/Map.h"
 #include "../../pokemod/MapWildList.h"
+#include "../../pokemod/MapWildListEncounter.h"
 
 // Qt includes
 #include <QFile>
 
+// KDE includes
+#include <KMenu>
+
 MapWildListModel::MapWildListModel(BaseModel* parent, MapWildList* wildList) :
         GroupObjectModel(parent, wildList)
 {
@@ -61,9 +65,41 @@ QVariant MapWildListModel::data(int role) const
         QWidget* widget = new MapWildListUI(static_cast<MapWildList*>(m_object), NULL);
         return QVariant::fromValue(widget);
     }
+    else if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Add Encounter", this, SLOT(addObject()));
+        menu->addSeparator();
+        menu->addAction("Delete Wild List", this, SLOT(deleteSelf()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
     return GroupObjectModel::data(role);
 }
 
+void MapWildListModel::addObject(Object* object)
+{
+    if (!object)
+        object = static_cast<MapWildList*>(m_object)->newEncounter();
+    if (object->className() == "MapWildListEncounter")
+        m_objects.append(new MapWildListEncounterModel(this, static_cast<MapWildListEncounter*>(object)));
+}
+
+void MapWildListModel::deleteObject(BaseModel* model)
+{
+    const int index = m_objects.indexOf(model);
+    if (0 <= index)
+    {
+        static_cast<MapWildList*>(m_object)->deleteEncounter(index);
+        m_objects[index]->deleteLater();
+        m_objects.removeAt(index);
+    }
+}
+
+void MapWildListModel::deleteSelf()
+{
+//     static_cast<GroupModel*>(m_parent)->deleteObject(this);
+}
+
 bool MapWildListModel::setData(const QVariant& value, int role)
 {
     if (role == BaseModel::XmlRole)
-- 
cgit