/*
 * 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 "MapTrainerTeamMemberUI.h"

// Sigmod includes
#include "../sigmod/Ability.h"
#include "../sigmod/Item.h"
#include "../sigmod/MapTrainerTeamMember.h"
#include "../sigmod/Move.h"
#include "../sigmod/Nature.h"
#include "../sigmod/Rules.h"
#include "../sigmod/Sigmod.h"
#include "../sigmod/Species.h"

// Qt includes
#include <QtGui/QListWidgetItem>

Sigmodr::MapTrainerTeamMemberUI::MapTrainerTeamMemberUI(Sigmod::MapTrainerTeamMember* teamMember, QWidget* parent) :
        ObjectUI(parent)
{
    setupUi(this);
    setObjects(teamMember, new Sigmod::MapTrainerTeamMember(*teamMember));
}

Sigmodr::MapTrainerTeamMemberUI::~MapTrainerTeamMemberUI()
{
}

void Sigmodr::MapTrainerTeamMemberUI::initGui()
{
    connect(varItems->selectedListWidget(), SIGNAL(currentItemChanged(QListWidgetItem*, QListWidgetItem*)), this, SLOT(selectedItemChanged(QListWidgetItem*)));
}

void Sigmodr::MapTrainerTeamMemberUI::refreshGui()
{
    const bool blockedSpecies = varSpecies->blockSignals(true);
    varSpecies->clear();
    for (int i = 0; i < sigmod()->speciesCount(); ++i)
    {
        const Sigmod::Species* species = sigmod()->species(i);
        varSpecies->addItem(species->name(), species->id());
    }
    varSpecies->blockSignals(blockedSpecies);
    varLevel->setMaximum(sigmod()->rules()->maxLevel());
    const bool blockedAbilities = varAbilities->blockSignals(true);
    varAbilities->availableListWidget()->clear();
    varAbilities->selectedListWidget()->clear();
    for (int i = 0; i < sigmod()->abilityCount(); ++i)
    {
        const Sigmod::Ability* ability = sigmod()->ability(i);
        QListWidgetItem* widgetItem = new QListWidgetItem(ability->name(), varAbilities->availableListWidget());
        widgetItem->setData(Qt::UserRole, ability->id());
    }
    varAbilities->blockSignals(blockedAbilities);
    varAbilities->setButtonsEnabled();
    boxAbilities->setEnabled(sigmod()->rules()->maxAbilities());
    const bool blockedItems = varItems->blockSignals(true);
    varItems->availableListWidget()->clear();
    varItems->selectedListWidget()->clear();
    for (int i = 0; i < sigmod()->itemCount(); ++i)
    {
        const Sigmod::Item* item = sigmod()->item(i);
        QListWidgetItem* widgetItem = new QListWidgetItem(item->name(), varItems->availableListWidget());
        widgetItem->setData(Qt::UserRole, item->id());
    }
    varItems->blockSignals(blockedItems);
    varItems->setButtonsEnabled();
    boxItems->setEnabled(sigmod()->rules()->maxHeldItems());
    const bool blockedMoves = varMoves->blockSignals(true);
    varMoves->availableListWidget()->clear();
    varMoves->selectedListWidget()->clear();
    for (int i = 0; i < sigmod()->moveCount(); ++i)
    {
        const Sigmod::Move* move = sigmod()->move(i);
        QListWidgetItem* widgetItem = new QListWidgetItem(move->name(), varMoves->availableListWidget());
        widgetItem->setData(Qt::UserRole, move->id());
    }
    varMoves->blockSignals(blockedMoves);
    varMoves->setButtonsEnabled();
    const bool blockedNatures = varNatures->blockSignals(true);
    varNatures->availableListWidget()->clear();
    varNatures->selectedListWidget()->clear();
    for (int i = 0; i < sigmod()->natureCount(); ++i)
    {
        const Sigmod::Nature* nature = sigmod()->nature(i);
        QListWidgetItem* widgetItem = new QListWidgetItem(nature->name(), varNatures->availableListWidget());
        widgetItem->setData(Qt::UserRole, nature->id());
    }
    varNatures->blockSignals(blockedNatures);
    varNatures->setButtonsEnabled();
    boxNatures->setEnabled(sigmod()->rules()->maxNatures());
}

void Sigmodr::MapTrainerTeamMemberUI::setGui()
{
    varSpecies->setCurrentIndex(varSpecies->findData(qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->species()));
    varLevel->setValue(qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->level());
    for (int i = 0; i < varAbilities->availableListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varAbilities->availableListWidget()->item(i);
        if (qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->ability(widgetItem->data(Qt::UserRole).toInt()))
            varAbilities->selectedListWidget()->addItem(varAbilities->availableListWidget()->takeItem(i--));
    }
    for (int i = 0; i < varAbilities->selectedListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varAbilities->selectedListWidget()->item(i);
        if (!qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->ability(widgetItem->data(Qt::UserRole).toInt()))
            varAbilities->availableListWidget()->addItem(varAbilities->selectedListWidget()->takeItem(i--));
    }
    varAbilities->setButtonsEnabled();
    for (int i = 0; i < varItems->availableListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varItems->availableListWidget()->item(i);
        if (qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->item(widgetItem->data(Qt::UserRole).toInt()))
            varItems->selectedListWidget()->addItem(varItems->availableListWidget()->takeItem(i--));
    }
    for (int i = 0; i < varItems->selectedListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varItems->selectedListWidget()->item(i);
        if (!qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->item(widgetItem->data(Qt::UserRole).toInt()))
            varItems->availableListWidget()->addItem(varItems->selectedListWidget()->takeItem(i--));
    }
    varItems->setButtonsEnabled();
    QListWidgetItem* selectedItem = varItems->selectedListWidget()->currentItem();
    if (selectedItem)
        varItemCount->setValue(qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->item(selectedItem->data(Qt::UserRole).toInt()));
    for (int i = 0; i < varMoves->availableListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varMoves->availableListWidget()->item(i);
        if (qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->move(widgetItem->data(Qt::UserRole).toInt()))
            varMoves->selectedListWidget()->addItem(varMoves->availableListWidget()->takeItem(i--));
    }
    for (int i = 0; i < varMoves->selectedListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varMoves->selectedListWidget()->item(i);
        if (!qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->move(widgetItem->data(Qt::UserRole).toInt()))
            varMoves->availableListWidget()->addItem(varMoves->selectedListWidget()->takeItem(i--));
    }
    varMoves->setButtonsEnabled();
    for (int i = 0; i < varNatures->availableListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varNatures->availableListWidget()->item(i);
        if (qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->nature(widgetItem->data(Qt::UserRole).toInt()))
            varNatures->selectedListWidget()->addItem(varNatures->availableListWidget()->takeItem(i--));
    }
    for (int i = 0; i < varNatures->selectedListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = varNatures->selectedListWidget()->item(i);
        if (!qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->nature(widgetItem->data(Qt::UserRole).toInt()))
            varNatures->availableListWidget()->addItem(varNatures->selectedListWidget()->takeItem(i--));
    }
    varNatures->setButtonsEnabled();
}

void Sigmodr::MapTrainerTeamMemberUI::apply()
{
    *qobject_cast<Sigmod::MapTrainerTeamMember*>(original()) = *qobject_cast<Sigmod::MapTrainerTeamMember*>(modified());
    emit(changed(false));
}

void Sigmodr::MapTrainerTeamMemberUI::discard()
{
    *qobject_cast<Sigmod::MapTrainerTeamMember*>(modified()) = *qobject_cast<Sigmod::MapTrainerTeamMember*>(original());
    setGui();
    emit(changed(false));
}

void Sigmodr::MapTrainerTeamMemberUI::on_varSpecies_activated(const int species)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setSpecies(varSpecies->itemData(species).toInt());
}

void Sigmodr::MapTrainerTeamMemberUI::on_varLevel_valueChanged(const int level)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setLevel(level);
}

void Sigmodr::MapTrainerTeamMemberUI::on_varAbilities_added(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setAbility(item->data(Qt::UserRole).toInt(), true);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varAbilities_removed(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setAbility(item->data(Qt::UserRole).toInt(), false);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varItems_added(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setItem(item->data(Qt::UserRole).toInt(), 1);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varItems_removed(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setItem(item->data(Qt::UserRole).toInt(), 0);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::selectedItemChanged(QListWidgetItem* item)
{
    if (item)
        varItemCount->setValue(qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->item(item->data(Qt::UserRole).toInt()));
    else
        varItemCount->setEnabled(false);
}

void Sigmodr::MapTrainerTeamMemberUI::on_varItemCount_valueChanged(const int itemCount)
{
    QListWidgetItem* item = varItems->selectedListWidget()->currentItem();
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setItem(item->data(Qt::UserRole).toInt(), itemCount);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varMoves_added(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setMove(item->data(Qt::UserRole).toInt(), true);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varMoves_removed(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setMove(item->data(Qt::UserRole).toInt(), false);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varNatures_added(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setNature(item->data(Qt::UserRole).toInt(), true);
    setGui();
}

void Sigmodr::MapTrainerTeamMemberUI::on_varNatures_removed(QListWidgetItem* item)
{
    qobject_cast<Sigmod::MapTrainerTeamMember*>(modified())->setNature(item->data(Qt::UserRole).toInt(), false);
    setGui();
}