1 files changed, 137 insertions, 0 deletions
diff --git a/sigmodr/widgets/MapTrainerUI.cpp b/sigmodr/widgets/MapTrainerUI.cpp
new file mode 100644
index 00000000..70043dc3
--- /dev/null
+++ b/sigmodr/widgets/MapTrainerUI.cpp
@@ -0,0 +1,137 @@
+/*
+ * Copyright 2008-2009 Ben Boeckel <MathStuf@gmail.com>
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License along
+ * with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+// Header include
+#include "MapTrainerUI.h"
+
+// Sigmod includes
+#include "../sigmod/MapTrainer.h"
+#include "../sigmod/MapTrainerTeamMember.h"
+#include "../sigmod/Rules.h"
+#include "../sigmod/Sigmod.h"
+#include "../sigmod/Species.h"
+#include "../sigmod/Trainer.h"
+
+// Qt includes
+#include <QtGui/QListWidget>
+#include <QtGui/QListWidgetItem>
+
+Sigmodr::MapTrainerUI::MapTrainerUI(Sigmod::MapTrainer* trainer, QWidget* parent) :
+        ObjectUI(parent)
+{
+    setupUi(this);
+    setObjects(trainer, new Sigmod::MapTrainer(*trainer));
+}
+
+Sigmodr::MapTrainerUI::~MapTrainerUI()
+{
+}
+
+void Sigmodr::MapTrainerUI::refreshGui()
+{
+    const bool blockedTrainerClass = varTrainerClass->blockSignals(true);
+    varTrainerClass->clear();
+    for (int i = 0; i < sigmod()->trainerCount(); ++i)
+    {
+        const Sigmod::Trainer* trainer = sigmod()->trainer(i);
+        varTrainerClass->addItem(trainer->name(), trainer->id());
+    }
+    varTrainerClass->blockSignals(blockedTrainerClass);
+    varNumberFight->setMaximum(sigmod()->rules()->maxFight());
+    const bool blockedLeadTeamMember = varLeadTeamMember->blockSignals(true);
+    varLeadTeamMember->availableListWidget()->clear();
+    varLeadTeamMember->selectedListWidget()->clear();
+    for (int i = 0; i < qobject_cast<Sigmod::MapTrainer*>(original())->teamMemberCount(); ++i)
+    {
+        QString speciesName;
+        const Sigmod::MapTrainerTeamMember* teamMember = qobject_cast<Sigmod::MapTrainer*>(original())->teamMember(i);
+        const Sigmod::Species* species = sigmod()->species(teamMember->species());
+        speciesName = species ? species->name() : "(Invalid)";
+        QListWidgetItem* widgetItem = new QListWidgetItem(QString("%1 level %2 (%3)").arg(speciesName, teamMember->level(), i), varLeadTeamMember->availableListWidget());
+        widgetItem->setData(Qt::UserRole, teamMember->id());
+    }
+    varLeadTeamMember->blockSignals(blockedLeadTeamMember);
+    varLeadTeamMember->setButtonsEnabled();
+}
+
+void Sigmodr::MapTrainerUI::setGui()
+{
+    varName->setText(qobject_cast<Sigmod::MapTrainer*>(modified())->name());
+    varTrainerClass->setCurrentIndex(varTrainerClass->findData(qobject_cast<Sigmod::MapTrainer*>(modified())->trainerClass()));
+    varNumberFight->setValue(qobject_cast<Sigmod::MapTrainer*>(modified())->numberFight());
+    varScript->setValue(qobject_cast<Sigmod::MapTrainer*>(modified())->script());
+    for (int i = 0; i < varLeadTeamMember->availableListWidget()->count(); ++i)
+    {
+        QListWidgetItem* widgetItem = varLeadTeamMember->availableListWidget()->item(i);
+        if (qobject_cast<Sigmod::MapTrainer*>(modified())->leadTeamMember(widgetItem->data(Qt::UserRole).toInt()))
+            varLeadTeamMember->selectedListWidget()->addItem(varLeadTeamMember->availableListWidget()->takeItem(i--));
+    }
+    for (int i = 0; i < varLeadTeamMember->selectedListWidget()->count(); ++i)
+    {
+        QListWidgetItem* widgetItem = varLeadTeamMember->selectedListWidget()->item(i);
+        if (!qobject_cast<Sigmod::MapTrainer*>(modified())->leadTeamMember(widgetItem->data(Qt::UserRole).toInt()))
+            varLeadTeamMember->availableListWidget()->addItem(varLeadTeamMember->selectedListWidget()->takeItem(i--));
+    }
+    varLeadTeamMember->setButtonsEnabled();
+}
+
+void Sigmodr::MapTrainerUI::apply()
+{
+    *qobject_cast<Sigmod::MapTrainer*>(original()) = *qobject_cast<Sigmod::MapTrainer*>(modified());
+    emit(changed(false));
+}
+
+void Sigmodr::MapTrainerUI::discard()
+{
+    *qobject_cast<Sigmod::MapTrainer*>(modified()) = *qobject_cast<Sigmod::MapTrainer*>(original());
+    setGui();
+    emit(changed(false));
+}
+
+void Sigmodr::MapTrainerUI::on_varName_textChanged(const QString& name)
+{
+    const int cursor = varName->cursorPosition();
+    qobject_cast<Sigmod::MapTrainer*>(modified())->setName(name);
+    varName->setCursorPosition(cursor);
+}
+
+void Sigmodr::MapTrainerUI::on_varTrainerClass_activated(const int trainerClass)
+{
+    qobject_cast<Sigmod::MapTrainer*>(modified())->setTrainerClass(varTrainerClass->itemData(trainerClass).toInt());
+}
+
+void Sigmodr::MapTrainerUI::on_varNumberFight_valueChanged(const int numberFight)
+{
+    qobject_cast<Sigmod::MapTrainer*>(modified())->setNumberFight(numberFight);
+}
+
+void Sigmodr::MapTrainerUI::on_varScript_valueChanged(const Sigcore::Script& script)
+{
+    qobject_cast<Sigmod::MapTrainer*>(modified())->setScript(script);
+}
+
+void Sigmodr::MapTrainerUI::on_varLeadTeamMember_added(QListWidgetItem* item)
+{
+    qobject_cast<Sigmod::MapTrainer*>(modified())->setLeadTeamMember(item->data(Qt::UserRole).toInt(), true);
+    setGui();
+}
+
+void Sigmodr::MapTrainerUI::on_varLeadTeamMember_removed(QListWidgetItem* item)
+{
+    qobject_cast<Sigmod::MapTrainer*>(modified())->setLeadTeamMember(item->data(Qt::UserRole).toInt(), false);
+    setGui();
+}