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

// Sigmodr core widget includes
#include <sigmodr/corewidgets/FractionWidget.h>
#include <sigmodr/corewidgets/ScriptWidget.h>

// Sigmod includes
#include <sigmod/Badge.h>
#include <sigmod/Game.h>
#include <sigmod/Rules.h>
#include <sigmod/Sprite.h>

// KDE includes
#include <KComboBox>
#include <KIntNumInput>
#include <KLineEdit>

// Qt includes
#include <QtCore/QFile>
#include <QtGui/QHeaderView>
#include <QtGui/QTableWidget>
#include <QtGui/QTableWidgetItem>
#include <QtGui/QVBoxLayout>
#include <QtUiTools/QUiLoader>

using namespace Sigcore;
using namespace Sigmod;
using namespace Sigmodr::CoreWidgets;
using namespace Sigmodr::Widgets;

BadgeUI::BadgeUI(Badge* badge, QWidget* parent) :
        ObjectUI(parent)
{
    setObjects(badge, new Badge(*badge));
}

void BadgeUI::initGui()
{
    QFile file(":/gui/badge.ui");
    file.open(QFile::ReadOnly);
    QWidget *formWidget = QUiLoader().load(&file, this);
    file.close();
    ui_name = formWidget->findChild<KLineEdit*>("varName");
    ui_obey= formWidget->findChild<KIntNumInput*>("varObey");
    ui_face = formWidget->findChild<KComboBox*>("varFace");
    ui_badge = formWidget->findChild<KComboBox*>("varBadge");
    ui_stat = formWidget->findChild<QTableWidget*>("varStat");
    ui_statMultiplier = formWidget->findChild<FractionWidget*>("varStatMultiplier");
    connect(ui_name, SIGNAL(textChanged(QString)), this, SLOT(nameChanged(QString)));
    connect(ui_obey, SIGNAL(valueChanged(int)), this, SLOT(obeyChanged(int)));
    connect(ui_face, SIGNAL(currentIndexChanged(int)), this, SLOT(faceChanged(int)));
    connect(ui_badge, SIGNAL(currentIndexChanged(int)), this, SLOT(badgeChanged(int)));
    connect(ui_stat, SIGNAL(currentCellChanged(int,int,int,int)), this, SLOT(statChanged(int)));
    connect(ui_statMultiplier, SIGNAL(valueChanged(Sigcore::Fraction)), this, SLOT(statMultiplierChanged(Sigcore::Fraction)));
    QVBoxLayout* layout = new QVBoxLayout;
    layout->addWidget(formWidget);
    setLayout(layout);
    ui_stat->horizontalHeader()->setResizeMode(QHeaderView::Stretch);
}

void BadgeUI::refreshGui()
{
    int maxHeight = 0;
    int maxWidth = 0;
    ui_obey->setMaximum(game()->rules()->maxLevel());
    const bool blockedFace = ui_face->blockSignals(true);
    const bool blockedBadge = ui_badge->blockSignals(true);
    ui_face->clear();
    ui_badge->clear();
    for (int i = 0; i < game()->spriteCount(); ++i)
    {
        const Sprite* sprite = game()->sprite(i);
        QPixmap icon;
        icon.loadFromData(sprite->sprite());
        maxHeight = qMax(maxHeight, icon.height());
        maxWidth = qMax(maxWidth, icon.width());
        ui_face->addItem(icon, sprite->name());
        ui_badge->addItem(icon, sprite->name());
    }
    ui_face->blockSignals(blockedFace);
    ui_badge->blockSignals(blockedBadge);
    const QSize maxSize(maxWidth, maxHeight);
    ui_face->setIconSize(maxSize);
    ui_badge->setIconSize(maxSize);
    const bool isSplit = game()->rules()->specialSplit();
    ui_stat->clear();
    ui_stat->setRowCount((isSplit ? ST_SpecialDefense : ST_Special) - ST_Attack + 1);
    ui_stat->setVerticalHeaderLabels((isSplit ? StatGSCStr : StatRBYStr).mid(ST_Attack, (isSplit ? ST_SpecialDefense : ST_Special) - ST_Attack + 1));
    ui_stat->setColumnCount(1);
    ui_stat->setHorizontalHeaderLabels(QStringList() << "Multiplier");
    ui_stat->verticalHeaderItem(ST_Attack - ST_Attack)->setData(Qt::UserRole, QVariant::fromValue(ST_Attack));
    ui_stat->verticalHeaderItem(ST_Defense - ST_Attack)->setData(Qt::UserRole, QVariant::fromValue(ST_Defense));
    ui_stat->verticalHeaderItem(ST_Speed - ST_Attack)->setData(Qt::UserRole, QVariant::fromValue(ST_Speed));
    if (isSplit)
    {
        ui_stat->verticalHeaderItem(ST_SpecialAttack - ST_Attack)->setData(Qt::UserRole, QVariant::fromValue(ST_SpecialAttack));
        ui_stat->verticalHeaderItem(ST_SpecialDefense - ST_Attack)->setData(Qt::UserRole, QVariant::fromValue(ST_SpecialDefense));
    }
    else
        ui_stat->verticalHeaderItem(ST_Special - ST_Attack)->setData(Qt::UserRole, QVariant::fromValue(ST_Special));
    QTableWidgetItem* item = new QTableWidgetItem;
    item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
    ui_stat->setItem(ST_Attack - ST_Attack, 0, item);
    ui_stat->setCurrentItem(item);
    item = new QTableWidgetItem;
    item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
    ui_stat->setItem(ST_Defense- ST_Attack, 0, item);
    item = new QTableWidgetItem;
    item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
    ui_stat->setItem(ST_Speed - ST_Attack, 0, item);
    if (isSplit)
    {
        item = new QTableWidgetItem;
        item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
        ui_stat->setItem(ST_SpecialAttack - ST_Attack, 0, item);
        item = new QTableWidgetItem;
        item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
        ui_stat->setItem(ST_SpecialDefense - ST_Attack, 0, item);
    }
    else
    {
        item = new QTableWidgetItem;
        item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
        ui_stat->setItem(ST_Special - ST_Attack, 0, item);
    }
}

void BadgeUI::setGui()
{
    ui_name->setText(qobject_cast<Badge*>(modified())->name());
    ui_obey->setValue(qobject_cast<Badge*>(modified())->obey());
    ui_face->setCurrentIndex(game()->spriteIndex(qobject_cast<Badge*>(modified())->face()));
    ui_badge->setCurrentIndex(game()->spriteIndex(qobject_cast<Badge*>(modified())->badge()));
    ui_stat->item(ST_Attack - ST_Attack, 0)->setData(Qt::DisplayRole, QString::number(qobject_cast<Badge*>(modified())->stat(ST_Attack), 'g', 7));
    ui_stat->item(ST_Defense - ST_Attack, 0)->setData(Qt::DisplayRole, QString::number(qobject_cast<Badge*>(modified())->stat(ST_Defense), 'g', 7));
    ui_stat->item(ST_Speed - ST_Attack, 0)->setData(Qt::DisplayRole, QString::number(qobject_cast<Badge*>(modified())->stat(ST_Speed), 'g', 7));
    if (game()->rules()->specialSplit())
    {
        ui_stat->item(ST_SpecialAttack - ST_Attack, 0)->setData(Qt::DisplayRole, QString::number(qobject_cast<Badge*>(modified())->stat(ST_SpecialAttack), 'g', 7));
        ui_stat->item(ST_SpecialDefense - ST_Attack, 0)->setData(Qt::DisplayRole, QString::number(qobject_cast<Badge*>(modified())->stat(ST_SpecialDefense), 'g', 7));
    }
    else
        ui_stat->item(ST_Special - ST_Attack, 0)->setData(Qt::DisplayRole, QString::number(qobject_cast<Badge*>(modified())->stat(ST_Special), 'g', 7));
    ui_statMultiplier->setValue(qobject_cast<Badge*>(modified())->stat(ui_stat->verticalHeaderItem(ui_stat->currentRow())->data(Qt::UserRole).value<Stat>()));
}

void BadgeUI::apply()
{
    *qobject_cast<Badge*>(original()) = *qobject_cast<Badge*>(modified());
    emit(changed(false));
}

void BadgeUI::discard()
{
    *qobject_cast<Badge*>(modified()) = *qobject_cast<Badge*>(original());
    setGui();
    emit(changed(false));
}

void BadgeUI::nameChanged(const QString& name)
{
    const int cursor = ui_name->cursorPosition();
    qobject_cast<Badge*>(modified())->setName(name);
    ui_name->setCursorPosition(cursor);
}

void BadgeUI::obeyChanged(const int obey)
{
    qobject_cast<Badge*>(modified())->setObey(obey);
}

void BadgeUI::faceChanged(const int face)
{
    if (0 <= face)
        qobject_cast<Badge*>(modified())->setFace(game()->sprite(face)->id());
}

void BadgeUI::badgeChanged(const int badge)
{
    if (0 <= badge)
        qobject_cast<Badge*>(modified())->setBadge(game()->sprite(badge)->id());
}

void BadgeUI::statChanged(const int row)
{
    ui_statMultiplier->setValue(qobject_cast<Badge*>(modified())->stat(ui_stat->verticalHeaderItem(row)->data(Qt::UserRole).value<Stat>()));
}

void BadgeUI::statMultiplierChanged(const Fraction& multiplier)
{
    qobject_cast<Badge*>(modified())->setStat(ui_stat->verticalHeaderItem(ui_stat->currentRow())->data(Qt::UserRole).value<Stat>(), multiplier);
}