[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

1 files changed, 36 insertions, 0 deletions

diff --git a/pokemodr/models/MapTrainerModel.cpp b/pokemodr/models/MapTrainerModel.cpp
index f5db2f24..8698bfe4 100644
--- a/pokemodr/models/MapTrainerModel.cpp
+++ b/pokemodr/models/MapTrainerModel.cpp
@@ -27,10 +27,14 @@
 // Pokemod includes
 #include "../../pokemod/Map.h"
 #include "../../pokemod/MapTrainer.h"
+#include "../../pokemod/MapTrainerTeamMember.h"
 
 // Qt includes
 #include <QFile>
 
+// KDE includes
+#include <KMenu>
+
 MapTrainerModel::MapTrainerModel(BaseModel* parent, MapTrainer* trainer) :
         GroupObjectModel(parent, trainer)
 {
@@ -57,9 +61,41 @@ QVariant MapTrainerModel::data(int role) const
         QWidget* widget = new MapTrainerUI(static_cast<MapTrainer*>(m_object), NULL);
         return QVariant::fromValue(widget);
     }
+    else if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Add Team Member", this, SLOT(addObject()));
+        menu->addSeparator();
+        menu->addAction("Delete Trainer", this, SLOT(deleteSelf()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
     return GroupObjectModel::data(role);
 }
 
+void MapTrainerModel::addObject(Object* object)
+{
+    if (!object)
+        object = static_cast<MapTrainer*>(m_object)->newTeamMember();
+    if (object->className() == "MapTrainerTeamMember")
+        m_objects.append(new MapTrainerTeamMemberModel(this, static_cast<MapTrainerTeamMember*>(object)));
+}
+
+void MapTrainerModel::deleteObject(BaseModel* model)
+{
+    const int index = m_objects.indexOf(model);
+    if (0 <= index)
+    {
+        static_cast<MapTrainer*>(m_object)->deleteTeamMember(index);
+        m_objects[index]->deleteLater();
+        m_objects.removeAt(index);
+    }
+}
+
+void MapTrainerModel::deleteSelf()
+{
+//     static_cast<GroupModel*>(m_parent)->deleteObject(this);
+}
+
 bool MapTrainerModel::setData(const QVariant& value, int role)
 {
     if (role == BaseModel::XmlRole)