[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, 33 insertions, 5 deletions

diff --git a/pokemodr/models/GlobalScriptGroupModel.cpp b/pokemodr/models/GlobalScriptGroupModel.cpp
index e951327a..c2ad6258 100644
--- a/pokemodr/models/GlobalScriptGroupModel.cpp
+++ b/pokemodr/models/GlobalScriptGroupModel.cpp
@@ -22,21 +22,49 @@
 #include "GlobalScriptModel.h"
 
 // Pokemod includes
+#include "../../pokemod/GlobalScript.h"
 #include "../../pokemod/Pokemod.h"
 
+// KDE includes
+#include <KMenu>
+
 GlobalScriptGroupModel::GlobalScriptGroupModel(BaseModel* parent, Pokemod* pokemod) :
         GroupModel(parent, pokemod, "Global Scripts")
 {
-    setupData();
+    for (int i = 0; i < pokemod->globalScriptCount(); ++i)
+        addObject(pokemod->globalScript(i));
 }
 
 GlobalScriptGroupModel::~GlobalScriptGroupModel()
 {
 }
 
-void GlobalScriptGroupModel::setupData()
+QVariant GlobalScriptGroupModel::data(const int role) const
 {
-    Pokemod* pokemod = static_cast<Pokemod*>(m_object);
-    for (int i = 0; i < pokemod->globalScriptCount(); ++i)
-        m_objects.append(new GlobalScriptModel(this, pokemod->globalScript(i)));
+    if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Add GlobalScript", this, SLOT(addObject()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
+    return GroupModel::data(role);
+}
+
+void GlobalScriptGroupModel::addObject(Object* object)
+{
+    if (!object)
+        object = static_cast<Pokemod*>(m_object)->newGlobalScript();
+    if (object->className() == "GlobalScript")
+        m_objects.append(new GlobalScriptModel(this, static_cast<GlobalScript*>(object)));
+}
+
+void GlobalScriptGroupModel::deleteObject(BaseObjectModel* model)
+{
+    const int index = m_objects.indexOf(model);
+    if (0 <= index)
+    {
+        static_cast<Pokemod*>(m_object)->deleteGlobalScript(index);
+        m_objects[index]->deleteLater();
+        m_objects.removeAt(index);
+    }
 }