/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include <QLabel>

#include <QGridLayout>

#include <QHBoxLayout>

#include <QGraphicsScene>

#include <QGroupBox>

#include <QSizePolicy>

#include <QPainterPath>

#include "pagegamestats.h"

#include "team.h"

FitGraphicsView::FitGraphicsView(QWidget* parent) : QGraphicsView(parent)

{

}

void FitGraphicsView::resizeEvent(QResizeEvent * event)

{

    Q_UNUSED(event);

    fitInView(sceneRect());

}

QLayout * PageGameStats::bodyLayoutDefinition()

{

    kindOfPoints = QString("");

    defaultGraphTitle = true;

    pageLayout = new QGridLayout();

    pageLayout->setRowStretch(0, 1);

    pageLayout->setRowStretch(1, 20);

    pageLayout->setVerticalSpacing(20);

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

    gbDetails = new QGroupBox(this);

    gbDetails->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Preferred);

    QVBoxLayout * gbl = new QVBoxLayout;

    // details

    labelGameStats = new QLabel(this);

    labelDetails = new QLabel(this);

    labelDetails->setTextFormat(Qt::RichText);

    labelDetails->setText("<h1><img src=\":/res/StatsD.png\"> " + PageGameStats::tr("Details").toHtmlEscaped() + "</h1>");

    labelDetails->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

    labelGameStats->setTextFormat(Qt::RichText);

    labelGameStats->setAlignment(Qt::AlignTop);

    labelGameStats->setWordWrap(true);

    gbl->addWidget(labelDetails);

    gbl->addWidget(labelGameStats);

    gbDetails->setLayout(gbl);

    pageLayout->addWidget(gbDetails, 1, 1);

    // graph

    graphic = new FitGraphicsView(gbDetails);

    graphic->setObjectName("gameStatsView");

    labelGraphTitle = new QLabel(this);

    labelGraphTitle->setTextFormat(Qt::RichText);

    labelGraphTitle->setText("<h1><img src=\":/res/StatsH.png\"> " + PageGameStats::tr("Health graph").toHtmlEscaped() + "</h1>");

    labelGraphTitle->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

    gbl->addWidget(labelGraphTitle);

    gbl->addWidget(graphic);

    graphic->scale(1.0, -1.0);

    graphic->setBackgroundBrush(QBrush(Qt::black));

    graphic->setRenderHint(QPainter::Antialiasing, true);

    labelGameWin = new QLabel(this);

    labelGameWin->setTextFormat(Qt::RichText);

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

    // ranking box

    gbRanks = new QGroupBox(this);

    gbRanks->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Preferred);

    gbl = new QVBoxLayout;

    labelGameRank = new QLabel(gbRanks);

    QLabel* l = new QLabel(this);

    l->setTextFormat(Qt::RichText);

    l->setText("<h1><img src=\":/res/StatsR.png\"> " + PageGameStats::tr("Ranking").toHtmlEscaped() + "</h1>");

    l->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

    gbl->addWidget(l);

    gbl->addWidget(labelGameRank);

    gbRanks->setLayout(gbl);

    labelGameRank->setTextFormat(Qt::RichText);

    labelGameRank->setAlignment(Qt::AlignTop);

    pageLayout->addWidget(gbRanks, 1, 0);

    return pageLayout;

}

//TODO button placement, image etc

QLayout * PageGameStats::footerLayoutDefinition()

{

    QHBoxLayout * bottomLayout = new QHBoxLayout();

    mainNote = new QLabel(this);

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

    mainNote->setWordWrap(true);

    bottomLayout->addWidget(mainNote, 0);

    bottomLayout->setStretch(0,1);

    btnRestart = addButton(":/res/Start.png", bottomLayout, 1, true);

    btnRestart->setWhatsThis(tr("Play again"));

    btnRestart->setFixedWidth(58);

    btnRestart->setFixedHeight(81);

    btnRestart->setStyleSheet("QPushButton{margin-top:24px}");

    btnSave = addButton(":/res/Save.png", bottomLayout, 2, true);

    saveDemoBtnEnabled(true);

    btnSave->setStyleSheet("QPushButton{margin: 24px 0 0 0;}");

    return bottomLayout;

}

void PageGameStats::connectSignals()

{

    connect(this, SIGNAL(pageEnter()), this, SLOT(renderStats()));

    connect(btnSave, SIGNAL(clicked()), this, SIGNAL(saveDemoRequested()));

    connect(btnRestart, SIGNAL(clicked()), this, SIGNAL(restartGameRequested()));

}

PageGameStats::PageGameStats(QWidget* parent) : AbstractPage(parent)

{

    initPage();

}

void PageGameStats::AddStatText(const QString & msg)

{

    labelGameStats->setText(labelGameStats->text() + msg);

    labelDetails->show();

    labelGameStats->show();

    gbDetails->show();

}

void PageGameStats::clear()

{

    labelGameStats->setText("");

    healthPoints.clear();

    labelGameRank->setText("");

    labelGameWin->setText("");

    playerPosition = 0;

    scriptPlayerPosition = 0;

    lastColor = 0;

    graphic->hide();

    labelDetails->hide();

    labelGameStats->hide();

    gbDetails->hide();

    gbRanks->hide();

    pageLayout->setColumnStretch(0, 0);

    pageLayout->setColumnStretch(1, 0);

    pageLayout->setHorizontalSpacing(0);

}

void PageGameStats::restartBtnVisible(bool visible)

{

    btnRestart->setVisible(visible);

}

void PageGameStats::saveDemoBtnEnabled(bool enabled)

{

    btnSave->setEnabled(enabled);

    if (enabled)

        btnSave->setWhatsThis(tr("Save demo"));

    else

        btnSave->setWhatsThis(tr("Save demo (unavailable because the /lua command was used)"));

}

void PageGameStats::renderStats()

{

    if(defaultGraphTitle) {

        labelGraphTitle->setText("<h1><img src=\":/res/StatsH.png\"> " + PageGameStats::tr("Health graph").toHtmlEscaped() + "</h1>");

    } else {

        defaultGraphTitle = true;

    }

    // if not health data sent

    if(healthPoints.size() == 0) {

        labelGraphTitle->hide();

        graphic->hide();

    } else {

        graphic->setScene(Q_NULLPTR);

        gbDetails->show();

        m_scene.reset(new QGraphicsScene(this));

        // min and max value across the entire chart

        qint32 minValue = 0;

        qint32 maxValue = 0;

        bool minMaxValuesInitialized = false;

        // max data points per clan

        int maxDataPoints = 0;

        for(QMap<qint32, QVector<qint32> >::const_iterator i = healthPoints.constBegin(); i != healthPoints.constEnd(); ++i)

        {

            maxDataPoints = qMax(maxDataPoints, i.value().size());

        }

        /* There must be at least 2 data points for any clan,

           otherwise there's not much to look at. ;-) */

        if(maxDataPoints < 2) {

            labelGraphTitle->hide();

            graphic->hide();

            applySpacing();

            return;

        }

        QMap<qint32, QVector<qint32> >::const_iterator i = healthPoints.constBegin();

        while (i != healthPoints.constEnd())

        {

            qint32 c = i.key();

            const QVector<qint32>& hps = i.value();

            QPainterPath path;

            if (hps.size()) {

                path.moveTo(0, hps[0]);

                if(minMaxValuesInitialized) {

                    minValue = qMin(minValue, hps[0]);

                    maxValue = qMax(maxValue, hps[0]);

                } else {

                    minValue = hps[0];

                    maxValue = hps[0];

                    minMaxValuesInitialized = true;

                }

            }

            for(int t = 0; t < hps.size(); ++t) {

                path.lineTo(t, hps[t]);

                maxValue = qMax(maxValue, hps[t]);

                minValue = qMin(minValue, hps[t]);

            }

            // Draw clan health/score graph lines

            QColor col = QColor(c);

            // Special pen for very dark clan colors

            if (!(col.red() >= cInvertTextColorAt || col.green() >= cInvertTextColorAt || col.blue() >= cInvertTextColorAt))

            {

                QPen pen_marker(QColor(255, 255, 255));

                pen_marker.setWidth(3);

                pen_marker.setStyle(Qt::DotLine);

                pen_marker.setCosmetic(true);

                m_scene->addPath(path, pen_marker);

            }

            // Regular pen

            QPen pen(col);

            pen.setWidth(2);

            pen.setCosmetic(true);

            m_scene->addPath(path, pen);

            ++i;

        }

        graphic->setScene(m_scene.data());

        // Calculate the bounding box of the final chart

        qint32 sceneMinY = minValue;

        qint32 sceneMaxY = maxValue;

        // If all values are 0 or greater, make sure to include 0 at the bottom.

        if(sceneMinY >= 0 && sceneMaxY >= 0)

            sceneMinY = 0;

        // If all values are equal, we must increase sceneMaxY, otherwise the scene rect

        // would have a height of 0 and will screw up

        if(sceneMinY == sceneMaxY)

            sceneMaxY++;

        graphic->setSceneRect(0, sceneMinY, maxDataPoints-1, sceneMaxY - sceneMinY);

        graphic->fitInView(graphic->sceneRect());

        graphic->show();

        labelGraphTitle->show();

        gbDetails->show();

    }

    applySpacing();

}

void PageGameStats::applySpacing()

{

    if (!labelGameStats->isHidden())

    {

        labelGraphTitle->setText("<br>" + labelGraphTitle->text());

    }

    if ((!gbDetails->isHidden()) && (!gbRanks->isHidden()))

    {

        pageLayout->setColumnStretch(0, 1);

        pageLayout->setColumnStretch(1, 1);

        pageLayout->setHorizontalSpacing(20);

    }

}

void PageGameStats::GameStats(char type, const QString & info)

{

    switch(type)

    {

        case 'r' :

        {

            labelGameWin->setText(QString("<h1 align=\"center\">%1</h1>").arg(info.toHtmlEscaped()));

            break;

        }

        case 'D' :

        {

            int i = info.indexOf(' ');

            int num = info.left(i).toInt();

            QString message = "<p><img src=\":/res/StatsBestShot.png\"> " + PageGameStats::tr("The best shot award was won by <b>%1</b> with <b>%2</b> pts.", "", num).arg(info.mid(i + 1).toHtmlEscaped(), info.left(i)) + "</p>";

            AddStatText(message);

            break;

        }

        case 'k' :

        {

            int i = info.indexOf(' ');

            int num = info.left(i).toInt();

            QString message = "<p><img src=\":/res/StatsBestKiller.png\"> " + PageGameStats::tr("The best killer is <b>%1</b> with <b>%2</b> kills in a turn.", "", num).arg(info.mid(i + 1).toHtmlEscaped(), info.left(i)) + "</p>";

            AddStatText(message);

            break;

        }

        case 'K' :

        {

            int num = info.toInt();

            QString message = "<p><img src=\":/res/StatsHedgehogsKilled.png\"> " +  PageGameStats::tr("A total of <b>%1</b> hedgehog(s) were killed during this round.", "", num).arg(num) + "</p>";

            AddStatText(message);

            break;

        }

        case 'H' :

        {

            int i = info.indexOf(' ');

            quint32 clan = info.left(i).toInt();

            qint32 hp = info.mid(i + 1).toInt();

            healthPoints[clan].append(hp);

            break;

        }

        case 'g' :

        {

            // TODO: change default picture or add change pic capability

            defaultGraphTitle = false;

            labelGraphTitle->setText("<h1><img src=\":/res/StatsR.png\"> " + info.toHtmlEscaped() + "</h1>");

            break;

        }

        case 'T':   // local team stats

        {

            // unused

            break;

        }

        case 'p' :

        {

            kindOfPoints = info;

            break;

        }

        case 'P' :

        {

            int i = info.indexOf(' ');

            playerPosition++;

            QString color = info.left(i);

            quint32 c = color.toInt();

            QColor clanColor = QColor(qRgb((c >> 16) & 255, (c >> 8) & 255, c & 255));

            QString playerinfo = info.mid(i + 1);

            i = playerinfo.indexOf(' ');

            QString killsString = playerinfo.left(i);

            int kills = killsString.toInt();

            QString playername = playerinfo.mid(i + 1);

            QString image;

            if (lastColor == c) playerPosition--;

            lastColor = c;

            unsigned int realPlayerPosition;

            if(scriptPlayerPosition == 0)

                realPlayerPosition = playerPosition;

            else

                realPlayerPosition = scriptPlayerPosition;

            switch (realPlayerPosition)

            {

                case 1:

                    image = "<img src=\":/res/StatsMedal1.png\">";

                    break;

                case 2:

                    image = "<img src=\":/res/StatsMedal2.png\">";

                    break;

                case 3:

                    image = "<img src=\":/res/StatsMedal3.png\">";

                    break;

                default:

                    image = "<img src=\":/res/StatsMedal4.png\">";

                    break;

            }

            QString message;

            QString killstring;

            if(kindOfPoints.isEmpty()) {

                //: Number of kills in stats screen, written after the team name

                killstring = PageGameStats::tr("(%1 kill)", "", kills).arg(kills);

            } else if (kindOfPoints == "!POINTS") {

                //: Number of points in stats screen, written after the team name

                killstring = PageGameStats::tr("(%1 point(s))", "", kills).arg(kills);

            } else if (kindOfPoints == "!TIME") {

                //: Time in seconds

                killstring = PageGameStats::tr("(%L1 second(s))", "", kills).arg((double) kills/1000, 0, 'f', 3);

            } else if (kindOfPoints.startsWith("!TIME") && kindOfPoints.length() == 6) {

                int len = kindOfPoints.at(6).digitValue();

                if(len != -1)

                    killstring = PageGameStats::tr("(%L1 second(s))", "", kills).arg((double) kills/1000, 0, 'f', len);

                else

                    qWarning("SendStat: siPointType received with !TIME and invalid number length!");

            } else if (kindOfPoints == "!CRATES") {

                killstring = PageGameStats::tr("(%1 crate(s))", "", kills).arg(kills);

            } else if (kindOfPoints == "!EMPTY") {

                killstring = QString("");

            } else {

                //: For custom number of points in the stats screen, written after the team name. %1 is the number, %2 is the word. Example: “4 points”

                killstring = PageGameStats::tr("(%1 %2)", "", kills).arg(kills).arg(kindOfPoints);

            }

            kindOfPoints = QString("");

            message = QString("<p><h2>%1 %2. <font color=\"%4\">%3</font> ").arg(image, QString::number(realPlayerPosition), playername.toHtmlEscaped(), clanColor.name().toHtmlEscaped()) + killstring.toHtmlEscaped() + "</h2></p>";

            labelGameRank->setText(labelGameRank->text() + message);

            scriptPlayerPosition = 0;

            gbRanks->show();

            break;

        }

        case 's' :

        {

            int i = info.indexOf(' ');

            int num = info.left(i).toInt();

            QString message = "<p><img src=\":/res/StatsMostSelfDamage.png\"> " + PageGameStats::tr("<b>%1</b> thought it's good to shoot their own hedgehogs for <b>%2</b> pts.", "", num).arg(info.mid(i + 1).toHtmlEscaped()).arg(num) + "</p>";

            AddStatText(message);

            break;

        }

        case 'S' :

        {

            int i = info.indexOf(' ');

            int num = info.left(i).toInt();

            QString message = "<p><img src=\":/res/StatsSelfKilled.png\"> " + PageGameStats::tr("<b>%1</b> killed <b>%2</b> of their own hedgehogs.", "", num).arg(info.mid(i + 1).toHtmlEscaped()).arg(num) + "</p>";

            AddStatText(message);

            break;

        }

        case 'B' :

        {

            int i = info.indexOf(' ');

            int num = info.left(i).toInt();

            QString message = "<p><img src=\":/res/StatsSkipped.png\"> " + PageGameStats::tr("<b>%1</b> was scared and skipped turn <b>%2</b> times.", "", num).arg(info.mid(i + 1).toHtmlEscaped()).arg(num) + "</p>";

            AddStatText(message);

            break;

        }

        case 'c' :

        {

            QString message = "<p><img src=\":/res/StatsCustomAchievement.png\"> "+info.toHtmlEscaped()+" </p>";

            AddStatText(message);

            break;

        }

        case 'R' :

        {

            scriptPlayerPosition = info.toInt();

            break;

        }

        case 'h' :

        {

            QString message = "<p><img src=\":/res/StatsEverAfter.png\"> " + PageGameStats::tr("With everyone having the same clan color, there was no reason to fight. And so the hedgehogs happily lived in peace ever after.").toHtmlEscaped() + "</p>";

            AddStatText(message);

            break;

        }

    }

}