/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2006-2010 Andrey Korotaev <unC0Rr@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; version 2 of the License

 *

 * 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, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

 */

#include <QGridLayout>

#include <QPushButton>

#include <QGroupBox>

#include <QComboBox>

#include <QCheckBox>

#include <QLabel>

#include <QToolBox>

#include <QLineEdit>

#include <QListWidget>

#include <QApplication>

#include <QSpinBox>

#include <QTextEdit>

#include <QRadioButton>

#include <QTableView>

#include <QMessageBox>

#include <QHeaderView>

#include <QTabWidget>

#include <QTextBrowser>

#include <QTableWidget>

#include <QAction>

#include <QMenu>

#include <QDataWidgetMapper>

#include "ammoSchemeModel.h"

#include "pages.h"

#include "sdlkeys.h"

#include "hwconsts.h"

#include "gamecfgwidget.h"

#include "teamselect.h"

#include "gamecfgwidget.h"

#include "SquareLabel.h"

#include "mapContainer.h"

#include "about.h"

#include "fpsedit.h"

#include "netserverslist.h"

#include "netudpwidget.h"

#include "chatwidget.h"

#include "playrecordpage.h"

#include "selectWeapon.h"

#include "igbox.h"

#include "hats.h"

#include "misc.h"

#include "togglebutton.h"

#include "hwform.h"

#include "SDLs.h"

PageMain::PageMain(QWidget* parent) :

  AbstractPage(parent)

{

    if(frontendEffects) setAttribute(Qt::WA_NoSystemBackground, true);

    QGridLayout * pageLayout = new QGridLayout(this);

    //pageLayout->setColumnStretch(0, 1);

    //pageLayout->setColumnStretch(1, 2);

    //pageLayout->setColumnStretch(2, 1);

    //QPushButton* btnLogo = addButton(":/res/HedgewarsTitle.png", pageLayout, 0, 0, 1, 4, true);

    //pageLayout->setAlignment(btnLogo, Qt::AlignHCenter);

    pageLayout->setRowStretch(0, 1);

    pageLayout->setRowStretch(1, 1);

    pageLayout->setRowStretch(2, 0);

    pageLayout->setRowStretch(3, 1);

    pageLayout->setRowStretch(4, 1);

    BtnSinglePlayer = addButton(":/res/LocalPlay.png", pageLayout, 2, 0, 1, 2, true);

    BtnSinglePlayer->setToolTip(tr("Local Game (Play a game on a single computer)"));

    pageLayout->setAlignment(BtnSinglePlayer, Qt::AlignHCenter);

    BtnNet = addButton(":/res/NetworkPlay.png", pageLayout, 2, 2, 1, 2, true);

    BtnNet->setToolTip(tr("Network Game (Play a game across a network)"));

    pageLayout->setAlignment(BtnNet, Qt::AlignHCenter);

    if(isDevBuild)

    {

        QLabel* devNote = new QLabel(this);

        devNote->setText(QLabel::tr("This SVN build is 'work in progress' and may not be compatible with other versions of the game.\nSome features might be broken or incomplete. Use at your own risk!"));

        devNote->setAlignment(Qt::AlignHCenter | Qt::AlignVCenter);

        pageLayout->addWidget(devNote, 4, 1, 1, 2);

    }

    BtnSetup = addButton(":/res/Settings.png", pageLayout, 4, 3, true);

    //BtnInfo = addButton(":/res/About.png", pageLayout, 3, 1, 1, 2, true);

    BtnInfo = addButton(":/res/HedgewarsTitle.png", pageLayout, 0, 0, 1, 4, true);

    BtnInfo->setStyleSheet("border: transparent;background: transparent;");

    pageLayout->setAlignment(BtnInfo, Qt::AlignHCenter);

    //pageLayout->setAlignment(BtnInfo, Qt::AlignHCenter);

    BtnExit = addButton(":/res/Exit.png", pageLayout, 4, 0, 1, 1, true);

}

PageEditTeam::PageEditTeam(QWidget* parent, SDLInteraction * sdli) :

  AbstractPage(parent)

{

    mySdli = sdli;

    QGridLayout * pageLayout = new QGridLayout(this);

    QTabWidget * tbw = new QTabWidget(this);

    QWidget * page1 = new QWidget(this);

    QWidget * page2 = new QWidget(this);

    tbw->addTab(page1, tr("General"));

    tbw->addTab(page2, tr("Advanced"));

    pageLayout->addWidget(tbw, 0, 0, 1, 3);

    BtnTeamDiscard = addButton(":/res/Exit.png", pageLayout, 1, 0, true);

    BtnTeamSave = addButton(":/res/Save.png", pageLayout, 1, 2, true);;

    BtnTeamSave->setStyleSheet("QPushButton{margin: 12px 0px 12px 0px;}");

    QHBoxLayout * page1Layout = new QHBoxLayout(page1);

    page1Layout->setAlignment(Qt::AlignTop);

    QGridLayout * page2Layout = new QGridLayout(page2);

// ====== Page 1 ======

    QVBoxLayout * vbox1 = new QVBoxLayout();

    QVBoxLayout * vbox2 = new QVBoxLayout();

    page1Layout->addLayout(vbox1);

    page1Layout->addLayout(vbox2);

    GBoxHedgehogs = new QGroupBox(this);

    GBoxHedgehogs->setTitle(QGroupBox::tr("Team Members"));

    GBoxHedgehogs->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

    QGridLayout * GBHLayout = new QGridLayout(GBoxHedgehogs);

    signalMapper = new QSignalMapper(this);

    HatsModel * hatsModel = new HatsModel(GBoxHedgehogs);

    for(int i = 0; i < 8; i++)

    {

        HHHats[i] = new QComboBox(GBoxHedgehogs);

        HHHats[i]->setModel(hatsModel);

        HHHats[i]->setIconSize(QSize(32, 37));

        //HHHats[i]->setSizeAdjustPolicy(QComboBox::AdjustToContents);

        //HHHats[i]->setModelColumn(1);

        //HHHats[i]->setMinimumWidth(132);

        GBHLayout->addWidget(HHHats[i], i, 0);

        HHNameEdit[i] = new QLineEdit(GBoxHedgehogs);

        HHNameEdit[i]->setMaxLength(64);

        HHNameEdit[i]->setMinimumWidth(120);

        GBHLayout->addWidget(HHNameEdit[i], i, 1);

        randButton[i] = addButton(":/res/dice.png", GBHLayout, i, 3, true);

        connect(randButton[i], SIGNAL(clicked()), signalMapper, SLOT(map()));

            signalMapper->setMapping(randButton[i], i);

    }

    randTeamButton = addButton(QPushButton::tr("Random Team"), GBHLayout, 9, false);

    vbox1->addWidget(GBoxHedgehogs);

    GBoxTeam = new QGroupBox(this);

    GBoxTeam->setTitle(QGroupBox::tr("Team Settings"));

    GBoxTeam->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

    QGridLayout * GBTLayout = new QGridLayout(GBoxTeam);

    QLabel * tmpLabel = new QLabel(GBoxTeam);

    tmpLabel->setText(QLabel::tr("Name"));

    GBTLayout->addWidget(tmpLabel, 0, 0);

    tmpLabel = new QLabel(GBoxTeam);

    tmpLabel->setText(QLabel::tr("Type"));

    GBTLayout->addWidget(tmpLabel, 1, 0);

    tmpLabel = new QLabel(GBoxTeam);

    tmpLabel->setText(QLabel::tr("Grave"));

    GBTLayout->addWidget(tmpLabel, 2, 0);

    tmpLabel = new QLabel(GBoxTeam);

    tmpLabel->setText(QLabel::tr("Flag"));

    GBTLayout->addWidget(tmpLabel, 3, 0);

    tmpLabel = new QLabel(GBoxTeam);

    tmpLabel->setText(QLabel::tr("Voice"));

    GBTLayout->addWidget(tmpLabel, 4, 0);

    TeamNameEdit = new QLineEdit(GBoxTeam);

    TeamNameEdit->setMaxLength(64);

    GBTLayout->addWidget(TeamNameEdit, 0, 1);

    vbox2->addWidget(GBoxTeam);

    CBTeamLvl = new QComboBox(GBoxTeam);

    CBTeamLvl->setIconSize(QSize(48, 48));

    CBTeamLvl->addItem(QIcon(":/res/botlevels/0.png"), QComboBox::tr("Human"));

    for(int i = 5; i > 0; i--)

        CBTeamLvl->addItem(

                QIcon(QString(":/res/botlevels/%1.png").arg(6 - i)),

                QString("%1 %2").arg(QComboBox::tr("Level")).arg(i)

                );

    GBTLayout->addWidget(CBTeamLvl, 1, 1);

    CBGrave = new QComboBox(GBoxTeam);

    CBGrave->setMaxCount(65535);

    CBGrave->setIconSize(QSize(32, 32));

    GBTLayout->addWidget(CBGrave, 2, 1);

    CBFlag = new QComboBox(GBoxTeam);

    CBFlag->setMaxCount(65535);

    CBFlag->setIconSize(QSize(22, 15));

    GBTLayout->addWidget(CBFlag, 3, 1);

    {

        QHBoxLayout * hbox = new QHBoxLayout();

        CBVoicepack = new QComboBox(GBoxTeam);

        {

            QDir tmpdir;

            tmpdir.cd(datadir->absolutePath());

            tmpdir.cd("Sounds/voices");

            QStringList list = tmpdir.entryList(QDir::AllDirs | QDir::NoDotAndDotDot, QDir::Name);

            CBVoicepack->addItems(list);

        }

        hbox->addWidget(CBVoicepack, 100);

        BtnTestSound = addButton(":/res/PlaySound.png", hbox, 1, true);

        hbox->setStretchFactor(BtnTestSound, 1);

        connect(BtnTestSound, SIGNAL(clicked()), this, SLOT(testSound()));

        GBTLayout->addLayout(hbox, 4, 1);

    }

    GBoxFort = new QGroupBox(this);

    GBoxFort->setTitle(QGroupBox::tr("Fort"));

    QGridLayout * GBFLayout = new QGridLayout(GBoxFort);

    CBFort = new QComboBox(GBoxFort);

    CBFort->setMaxCount(65535);

    GBFLayout->addWidget(CBFort, 0, 0);

    FortPreview = new SquareLabel(GBoxFort);

    FortPreview->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

    FortPreview->setMinimumSize(128, 128);

    FortPreview->setPixmap(QPixmap());

    // perhaps due to handling its own paintevents, SquareLabel doesn't play nice with the stars

    //FortPreview->setAttribute(Qt::WA_PaintOnScreen, true);

    GBFLayout->addWidget(FortPreview, 1, 0);

    vbox2->addWidget(GBoxFort);

    QDir tmpdir;

    tmpdir.cd(datadir->absolutePath());

    tmpdir.cd("Forts");

    tmpdir.setFilter(QDir::Files);

    connect(CBFort, SIGNAL(currentIndexChanged(const QString &)), this, SLOT(CBFort_activated(const QString &)));

    CBFort->addItems(tmpdir.entryList(QStringList("*L.png")).replaceInStrings(QRegExp("^(.*)L\\.png"), "\\1"));

    tmpdir.cd("../Graphics/Graves");

    QStringList list = tmpdir.entryList(QStringList("*.png"));

    for (QStringList::Iterator it = list.begin(); it != list.end(); ++it )

    {

        QPixmap pix(datadir->absolutePath() + "/Graphics/Graves/" + *it);

        QIcon icon(pix.copy(0, 0, 32, 32));

        CBGrave->addItem(icon, (*it).replace(QRegExp("^(.*)\\.png"), "\\1"));

    }

    tmpdir.cd(datadir->absolutePath());

    tmpdir.cd("Graphics/Flags");

    list = tmpdir.entryList(QStringList("*.png"));

    for (QStringList::Iterator it = list.begin(); it != list.end(); ++it )

    {

        QPixmap pix(datadir->absolutePath() + "/Graphics/Flags/" + *it);

        QIcon icon(pix.copy(0, 0, 22, 15));

        if(it->compare("cpu.png")) // skip cpu flag

            CBFlag->addItem(icon, (*it).replace(QRegExp("^(.*)\\.png"), "\\1"));

    }

    vbox1->addStretch();

    vbox2->addStretch();

// ====== Page 2 ======

    GBoxBinds = new QGroupBox(this);

    GBoxBinds->setTitle(QGroupBox::tr("Key binds"));

    QGridLayout * GBBLayout = new QGridLayout(GBoxBinds);

    BindsBox = new QToolBox(GBoxBinds);

    BindsBox->setLineWidth(0);

    GBBLayout->addWidget(BindsBox);

    page2Layout->addWidget(GBoxBinds, 0, 0);

    quint16 i = 0;

    quint16 num = 0;

    QWidget * curW = NULL;

    QGridLayout * pagelayout = NULL;

    QLabel* l = NULL;

    while (i < BINDS_NUMBER) {

        if(cbinds[i].category != NULL)

        {

            if(curW != NULL)

            {

                l = new QLabel(curW);

                l->setText("");

                pagelayout->addWidget(l, num++, 0, 1, 2);

            }

            curW = new QWidget(this);

            BindsBox->addItem(curW, QApplication::translate("binds (categories)", cbinds[i].category));

            pagelayout = new QGridLayout(curW);

            num = 0;

        }

        if(cbinds[i].description != NULL)

        {

            l = new QLabel(curW);

            l->setText((num > 0 ? QString("\n") : QString("")) + QApplication::translate("binds (descriptions)", cbinds[i].description));

            pagelayout->addWidget(l, num++, 0, 1, 2);

        }

        l = new QLabel(curW);

        l->setText(QApplication::translate("binds", cbinds[i].name));

        l->setAlignment(Qt::AlignRight);

        pagelayout->addWidget(l, num, 0);

        CBBind[i] = new QComboBox(curW);

        for(int j = 0; sdlkeys[j][1][0] != '\0'; j++)

            CBBind[i]->addItem(QApplication::translate("binds (keys)", sdlkeys[j][1]).contains(": ") ? QApplication::translate("binds (keys)", sdlkeys[j][1]) : QApplication::translate("binds (keys)", "Keyboard") + QString(": ") + QApplication::translate("binds (keys)", sdlkeys[j][1]), sdlkeys[j][0]);

        pagelayout->addWidget(CBBind[i++], num++, 1);

    }

}

void PageEditTeam::CBFort_activated(const QString & fortname)

{

    QPixmap pix(datadir->absolutePath() + "/Forts/" + fortname + "L.png");

    FortPreview->setPixmap(pix);

}

void PageEditTeam::testSound()

{

    Mix_Chunk *sound;

    QDir tmpdir;

    mySdli->SDLMusicInit();

    tmpdir.cd(datadir->absolutePath());

    tmpdir.cd("Sounds/voices");

    tmpdir.cd(CBVoicepack->currentText());

    QStringList list = tmpdir.entryList(QStringList() << "Illgetyou.ogg" << "Incoming.ogg" << "Stupid.ogg" << "Coward.ogg" << "Firstblood.ogg", QDir::Files);

    if (list.size()) {

        sound = Mix_LoadWAV(QString(tmpdir.absolutePath() + "/" + list[rand() % list.size()]).toLocal8Bit().constData());

        Mix_PlayChannel(-1, sound, 0);

    }

}

PageMultiplayer::PageMultiplayer(QWidget* parent) :

  AbstractPage(parent)

{

    QGridLayout * pageLayout = new QGridLayout(this);

    BtnBack = addButton(":/res/Exit.png", pageLayout, 2, 0, true);

    gameCFG = new GameCFGWidget(this);

    pageLayout->addWidget(gameCFG, 0, 0, 1, 2);

    pageLayout->setRowStretch(1, 1);

    teamsSelect = new TeamSelWidget(this);

    pageLayout->addWidget(teamsSelect, 0, 2, 2, 2);

    BtnStartMPGame = addButton(tr("Start"), pageLayout, 2, 3);

}

PageOptions::PageOptions(QWidget* parent) :

  AbstractPage(parent)

{

    QGridLayout * pageLayout = new QGridLayout(this);

    pageLayout->setColumnStretch(0, 100);

    pageLayout->setColumnStretch(1, 100);

    pageLayout->setColumnStretch(2, 100);

    pageLayout->setRowStretch(0, 0);

    pageLayout->setRowStretch(1, 100);

    pageLayout->setRowStretch(2, 0);

    pageLayout->setContentsMargins(7, 7, 7, 0);

    pageLayout->setSpacing(0);

    QGroupBox * gbTwoBoxes = new QGroupBox(this);

    pageLayout->addWidget(gbTwoBoxes, 0, 0, 1, 3);

    QGridLayout * gbTBLayout = new QGridLayout(gbTwoBoxes);

    gbTBLayout->setMargin(0);

    gbTBLayout->setSpacing(0);

    gbTBLayout->setAlignment(Qt::AlignTop | Qt::AlignLeft);

    QPixmap pmNew(":/res/new.png");

    QPixmap pmEdit(":/res/edit.png");

    QPixmap pmDelete(":/res/delete.png");

        {

            teamsBox = new IconedGroupBox(this);

            //teamsBox->setContentTopPadding(0);

            //teamsBox->setAttribute(Qt::WA_PaintOnScreen, true);

            teamsBox->setIcon(QIcon(":/res/teamicon.png"));

            teamsBox->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

            teamsBox->setTitle(QGroupBox::tr("Teams"));

            QGridLayout * GBTlayout = new QGridLayout(teamsBox);

            CBTeamName = new QComboBox(teamsBox);

            GBTlayout->addWidget(CBTeamName, 0, 0);

            BtnNewTeam = new QPushButton(teamsBox);

            BtnNewTeam->setToolTip(tr("New team"));

            BtnNewTeam->setIconSize(pmNew.size());

            BtnNewTeam->setIcon(pmNew);

            BtnNewTeam->setMaximumWidth(pmNew.width() + 6);

            GBTlayout->addWidget(BtnNewTeam, 0, 1);

            BtnEditTeam = new QPushButton(teamsBox);

            BtnEditTeam->setToolTip(tr("Edit team"));

            BtnEditTeam->setIconSize(pmEdit.size());

            BtnEditTeam->setIcon(pmEdit);

            BtnEditTeam->setMaximumWidth(pmEdit.width() + 6);

            GBTlayout->addWidget(BtnEditTeam, 0, 2);

            BtnDeleteTeam = new QPushButton(teamsBox);

            BtnDeleteTeam->setToolTip(tr("Delete team"));

            BtnDeleteTeam->setIconSize(pmDelete.size());

            BtnDeleteTeam->setIcon(pmDelete);

            BtnDeleteTeam->setMaximumWidth(pmDelete.width() + 6);

            GBTlayout->addWidget(BtnDeleteTeam, 0, 3);

            gbTBLayout->addWidget(teamsBox, 0, 0);

        }

        {

            // TODO: This box should contain controls for all schemes: game modes and weapons

            IconedGroupBox* groupWeapons = new IconedGroupBox(this);

            //groupWeapons->setContentTopPadding(0);

            groupWeapons->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

            groupWeapons->setIcon(QIcon(":/res/weaponsicon.png"));

            //groupWeapons->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

            groupWeapons->setTitle(QGroupBox::tr("Weapons"));

            QGridLayout * WeaponsLayout = new QGridLayout(groupWeapons);

            WeaponsName = new QComboBox(groupWeapons);

            WeaponsLayout->addWidget(WeaponsName, 0, 0);

            WeaponsButt = new QPushButton(groupWeapons);

            WeaponsButt->setToolTip(tr("New weapon scheme"));

            WeaponsButt->setIconSize(pmNew.size());

            WeaponsButt->setIcon(pmNew);

            WeaponsButt->setMaximumWidth(pmNew.width() + 6);

            WeaponsLayout->addWidget(WeaponsButt, 0, 1);

            WeaponEdit = new QPushButton(groupWeapons);

            WeaponEdit->setToolTip(tr("Edit weapon scheme"));

            WeaponEdit->setIconSize(pmEdit.size());

            WeaponEdit->setIcon(pmEdit);

            WeaponEdit->setMaximumWidth(pmEdit.width() + 6);

            WeaponsLayout->addWidget(WeaponEdit, 0, 2);

            WeaponDelete = new QPushButton(groupWeapons);

            WeaponDelete->setToolTip(tr("Delete weapon scheme"));

            WeaponDelete->setIconSize(pmDelete.size());

            WeaponDelete->setIcon(pmDelete);

            WeaponDelete->setMaximumWidth(pmDelete.width() + 6);

            WeaponDelete->setEnabled(false);

            WeaponDelete->setVisible(false); // hide for now

            WeaponsLayout->addWidget(WeaponDelete, 0, 3);

            WeaponTooltip = new QCheckBox(this);

            WeaponTooltip->setText(QCheckBox::tr("Show ammo menu tooltips"));

            WeaponsLayout->addWidget(WeaponTooltip, 1, 0, 1, 3);

            gbTBLayout->addWidget(groupWeapons, 1, 0);

        }

        {

            IconedGroupBox* groupMisc = new IconedGroupBox(this);

            //groupMisc->setContentTopPadding(0);

            groupMisc->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

            groupMisc->setIcon(QIcon(":/res/miscicon.png"));

            //groupMisc->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

            groupMisc->setTitle(QGroupBox::tr("Misc"));

            QGridLayout * MiscLayout = new QGridLayout(groupMisc);

            labelNN = new QLabel(groupMisc);

            labelNN->setText(QLabel::tr("Net nick"));

            MiscLayout->addWidget(labelNN, 0, 0);

            editNetNick = new QLineEdit(groupMisc);

            editNetNick->setMaxLength(20);