diff options
Diffstat (limited to 'pokemodr/models/MapWildListModel.cpp')
| -rw-r--r-- | pokemodr/models/MapWildListModel.cpp | 36 |
1 files changed, 36 insertions, 0 deletions
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) |