diff options
Diffstat (limited to 'pokemodr/models/MapTrainerModel.cpp')
| -rw-r--r-- | pokemodr/models/MapTrainerModel.cpp | 36 |
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) |