/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2006-2012 Igor Ulyanov <iulyanov@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 <QBitmap>

#include <QBuffer>

#include <QColor>

#include <QDebug>

#include <QFile>

#include <QFileDialog>

#include <QIcon>

#include <QInputDialog>

#include <QLabel>

#include <QLinearGradient>

#include <QLineEdit>

#include <QListView>

#include <QListWidget>

#include <QListWidgetItem>

#include <QMessageBox>

#include <QPainter>

#include <QPushButton>

#include <QSlider>

#include <QStringListModel>

#include <QTextStream>

#include <QUuid>

#include <QVBoxLayout>

#include "hwconsts.h"

#include "mapContainer.h"

#include "themeprompt.h"

#include "seedprompt.h"

#include "igbox.h"

#include "HWApplication.h"

#include "ThemeModel.h"

HWMapContainer::HWMapContainer(QWidget * parent) :

    QWidget(parent),

    mainLayout(this),

    pMap(0),

    mapgen(MAPGEN_REGULAR),

    m_previewSize(256, 128)

{

    // don't show preview anything until first show event

    m_previewEnabled = false;

    m_missionsViewSetup = false;

    m_staticViewSetup = false;

    m_script = QString();

    m_prevMapFeatureSize = 12;

    m_mapFeatureSize = 12;

    m_withoutDLC = false;

    m_missingMap = false;

    hhSmall.load(":/res/hh_small.png");

    hhLimit = 18;

    templateFilter = 0;

    m_master = true;

    linearGradNormal = QLinearGradient(QPoint(128, 0), QPoint(128, 128));

    linearGradNormal.setColorAt(1, QColor(0, 0, 192));

    linearGradNormal.setColorAt(0, QColor(66, 115, 225));

    linearGradLoading = QLinearGradient(QPoint(128, 0), QPoint(128, 128));

    linearGradLoading.setColorAt(1, QColor(58, 58, 137));

    linearGradLoading.setColorAt(0, QColor(90, 109, 153));

    linearGradMapError = QLinearGradient(QPoint(128, 0), QPoint(128, 128));

    linearGradMapError.setColorAt(1, QColor(255, 1, 0));

    linearGradMapError.setColorAt(0, QColor(255, 119, 0));

    linearGradNoPreview = QLinearGradient(QPoint(128, 0), QPoint(128, 128));

    linearGradNoPreview.setColorAt(1, QColor(15, 9, 72));

    linearGradNoPreview.setColorAt(0, QColor(15, 9, 72));

    mainLayout.setContentsMargins(HWApplication::style()->pixelMetric(QStyle::PM_LayoutLeftMargin),

                                  10,

                                  HWApplication::style()->pixelMetric(QStyle::PM_LayoutRightMargin),

                                  HWApplication::style()->pixelMetric(QStyle::PM_LayoutBottomMargin));

    m_staticMapModel = DataManager::instance().staticMapModel();

    m_missionMapModel = DataManager::instance().missionMapModel();

    m_themeModel = DataManager::instance().themeModel();

    /* Layouts */

    QWidget * topWidget = new QWidget();

    QHBoxLayout * topLayout = new QHBoxLayout(topWidget);

    topWidget->setContentsMargins(0, 0, 0, 0);

    topLayout->setContentsMargins(0, 0, 0, 0);

    twoColumnLayout = new QHBoxLayout();

    QVBoxLayout * leftLayout = new QVBoxLayout();

    leftLayout->setAlignment(Qt::AlignLeft);

    QVBoxLayout * rightLayout = new QVBoxLayout();

    twoColumnLayout->addLayout(leftLayout, 0);

    twoColumnLayout->addLayout(rightLayout, 0);

    QVBoxLayout * drawnControls = new QVBoxLayout();

    /* Map type label */

    QLabel* lblMapType = new QLabel(tr("Map type:"));

    topLayout->setSpacing(10);

    topLayout->addWidget(lblMapType, 0);

    m_childWidgets << lblMapType;

    /* Map type combobox */

    cType = new QComboBox(this);

    topLayout->addWidget(cType, 1);

    cType->insertItem(0, tr("Image map"), MapModel::StaticMap);

    cType->insertItem(1, tr("Mission map"), MapModel::MissionMap);

    cType->insertItem(2, tr("Hand-drawn"), MapModel::HandDrawnMap);

    cType->insertItem(3, tr("Randomly generated"), MapModel::GeneratedMap);

    cType->insertItem(4, tr("Random maze"), MapModel::GeneratedMaze);

    cType->insertItem(5, tr("Random perlin"), MapModel::GeneratedPerlin);

    cType->insertItem(6, tr("Forts"), MapModel::FortsMap);

    m_childWidgets << cType;

    /* Randomize button */

    topLayout->addStretch(1);

    const QIcon& lp = QIcon(":/res/dice.png");

    QSize sz = lp.actualSize(QSize(65535, 65535));

    btnRandomize = new QPushButton();

    btnRandomize->setText(tr("Random"));

    btnRandomize->setIcon(lp);

    btnRandomize->setFixedHeight(30);

    btnRandomize->setIconSize(sz);

    btnRandomize->setFlat(true);

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

    connect(btnRandomize, SIGNAL(clicked()), this, SLOT(setRandomMap()));

    m_childWidgets << btnRandomize;

    btnRandomize->setStyleSheet("padding: 5px;");

    btnRandomize->setFixedHeight(cType->height());

    topLayout->addWidget(btnRandomize, 1);

    /* Seed button */

    btnSeed = new QPushButton(parentWidget()->parentWidget());

    //: Refers to the "random seed"; the source of randomness in the game

    btnSeed->setText(tr("Seed"));

    btnSeed->setWhatsThis(tr("View and edit the seed, the source of randomness in the game"));

    btnSeed->setStyleSheet("padding: 5px;");

    btnSeed->setFixedHeight(cType->height());

    connect(btnSeed, SIGNAL(clicked()), this, SLOT(showSeedPrompt()));

    topLayout->addWidget(btnSeed, 0);

    /* Map preview label */

    QLabel * lblMapPreviewText = new QLabel(this);

    lblMapPreviewText->setText(tr("Map preview:"));

    leftLayout->addWidget(lblMapPreviewText, 0);

    m_childWidgets << lblMapPreviewText;

    /* Map Preview */

    mapPreview = new QPushButton(this);

    mapPreview->setObjectName("mapPreview");

    mapPreview->setFlat(true);

    mapPreview->setFixedSize(256 + 6, 128 + 6);

    mapPreview->setContentsMargins(0, 0, 0, 0);

    leftLayout->addWidget(mapPreview, 0);

    connect(mapPreview, SIGNAL(clicked()), this, SLOT(previewClicked()));

    m_childWidgets << mapPreview;

    /* Bottom-Left layout */

    QVBoxLayout * bottomLeftLayout = new QVBoxLayout();

    leftLayout->addLayout(bottomLeftLayout, 1);

    /* Map list label */

    lblMapList = new QLabel(this);

    lblMapList->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Minimum);

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

    rightLayout->addWidget(lblMapList, 0);

    m_childWidgets << lblMapList;

    /* Static maps list */

    staticMapList = new QListView(this);

    rightLayout->addWidget(staticMapList, 1);

    m_childWidgets << staticMapList;

    /* Mission maps list */

    missionMapList = new QListView(this);

    rightLayout->addWidget(missionMapList, 1);

    m_childWidgets << missionMapList;

    /* Map name (when not room master) */

    /* We use a QTextEdit instead of QLabel because it is able

       to wrap at any character. */

    teMapName = new QTextEdit(this);

    teMapName->setObjectName("mapName");

    teMapName->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Minimum);

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

    /* Boilerplate to emulate a QLabel */

    teMapName->setReadOnly(true);

    teMapName->setAcceptRichText(false);

    teMapName->setFrameStyle(QFrame::NoFrame);

    teMapName->setStyleSheet("background-color: transparent");

    teMapName->setLineWrapMode(QTextEdit::WidgetWidth);

    teMapName->setWordWrapMode(QTextOption::WrapAtWordBoundaryOrAnywhere);

    rightLayout->addWidget(teMapName, 1);

    m_childWidgets << teMapName;

    /* Map load and edit buttons */

    drawnControls->addStretch(1);

    QPixmap pmLoad(":/res/Load.png");

    QIcon iconLoad = QIcon(pmLoad);

    sz = iconLoad.actualSize(QSize(48, 48));

    btnLoadMap = new QPushButton(tr("Load"));

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

    btnLoadMap->setWhatsThis(tr("Load map drawing"));

    btnLoadMap->setStyleSheet("padding: 5px;");

    btnLoadMap->setFixedHeight(50);

    btnLoadMap->setIcon(iconLoad);

    btnLoadMap->setIconSize(sz);

    btnLoadMap->setFlat(true);

    drawnControls->addWidget(btnLoadMap, 0);

    m_childWidgets << btnLoadMap;

    connect(btnLoadMap, SIGNAL(clicked()), this, SLOT(loadDrawing()));

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

    QIcon iconEdit = QIcon(pmEdit);

    sz = iconEdit.actualSize(QSize(48, 48));

    btnEditMap = new QPushButton(tr("Edit"));

    btnEditMap->setWhatsThis(tr("Edit map drawing"));

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

    btnEditMap->setStyleSheet("padding: 5px;");

    btnEditMap->setFixedHeight(50);

    btnEditMap->setIcon(iconEdit);

    btnEditMap->setIconSize(sz);

    btnEditMap->setFlat(true);

    drawnControls->addWidget(btnEditMap, 0);

    m_childWidgets << btnEditMap;

    connect(btnEditMap, SIGNAL(clicked()), this, SIGNAL(drawMapRequested()));

    drawnControls->addStretch(1);

    rightLayout->addLayout(drawnControls);

    /* Generator style list */

    generationStyles = new QListWidget(this);

    new QListWidgetItem(tr("All"), generationStyles);

    new QListWidgetItem(tr("Small"), generationStyles);

    new QListWidgetItem(tr("Medium"), generationStyles);

    new QListWidgetItem(tr("Large"), generationStyles);

    new QListWidgetItem(tr("Cavern"), generationStyles);

    new QListWidgetItem(tr("Wacky"), generationStyles);

    connect(generationStyles, SIGNAL(currentRowChanged(int)), this, SLOT(setTemplateFilter(int)));

    m_childWidgets << generationStyles;

    rightLayout->addWidget(generationStyles, 1);

    /* Maze style list */

    mazeStyles = new QListWidget(this);

    new QListWidgetItem(tr("Small tunnels"), mazeStyles);

    new QListWidgetItem(tr("Medium tunnels"), mazeStyles);

    new QListWidgetItem(tr("Large tunnels"), mazeStyles);

    new QListWidgetItem(tr("Small islands"), mazeStyles);

    new QListWidgetItem(tr("Medium islands"), mazeStyles);

    new QListWidgetItem(tr("Large islands"), mazeStyles);

    connect(mazeStyles, SIGNAL(currentRowChanged(int)), this, SLOT(setMazeSize(int)));

    m_childWidgets << mazeStyles;

    rightLayout->addWidget(mazeStyles, 1);

    mapFeatureSize = new QSlider(Qt::Horizontal, this);

    mapFeatureSize->setObjectName("mapFeatureSize");

    //mapFeatureSize->setTickPosition(QSlider::TicksBelow);

    mapFeatureSize->setMaximum(25);

    mapFeatureSize->setMinimum(1);

    //mapFeatureSize->setFixedWidth(259);

    mapFeatureSize->setPageStep(5);

    mapFeatureSize->setValue(m_mapFeatureSize);

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

    bottomLeftLayout->addWidget(mapFeatureSize, 0);

    connect(mapFeatureSize, SIGNAL(valueChanged(int)), this, SLOT(setFeatureSize(int)));

    m_childWidgets << mapFeatureSize;

    /* Mission description */

    lblDesc = new QLabel();

    lblDesc->setWordWrap(true);

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

    lblDesc->setAlignment(Qt::AlignBottom | Qt::AlignLeft);

    lblDesc->setStyleSheet("font: 10px;");

    bottomLeftLayout->addWidget(lblDesc, 100);

    /* Add stretch above theme button */

    bottomLeftLayout->addStretch(1);

    /* Theme chooser */

    QHBoxLayout * themeHBox = new QHBoxLayout();

    btnRandTheme = new QPushButton();

    btnRandTheme->setWhatsThis(tr("Randomize the theme"));

    btnRandTheme->setIcon(lp);

    btnRandTheme->setIconSize(QSize(24, 24));

    btnRandTheme->setFixedHeight(30);

    btnRandTheme->setFixedWidth(30);

    connect(btnRandTheme, SIGNAL(clicked()), this, SLOT(setRandomTheme()));

    m_childWidgets << btnRandTheme;

    themeHBox->addWidget(btnRandTheme, 0);

    btnTheme = new QPushButton(this);

    btnTheme->setWhatsThis(tr("Choose a theme"));

    btnTheme->setFlat(true);

    btnTheme->setIconSize(QSize(30, 30));

    btnTheme->setFixedHeight(30);

    btnTheme->setMaximumWidth(222);

    connect(btnTheme, SIGNAL(clicked()), this, SLOT(showThemePrompt()));

    m_childWidgets << btnTheme;

    themeHBox->addWidget(btnTheme, 1);

    bottomLeftLayout->addLayout(themeHBox);

    /* Add everything to main layout */

    mainLayout.addWidget(topWidget, 0);

    mainLayout.addLayout(twoColumnLayout, 1);

    /* Set defaults */

    setRandomSeed();

    setMazeSize(0);

    setTemplateFilter(0);

    staticMapChanged(m_staticMapModel->index(0, 0));

    missionMapChanged(m_missionMapModel->index(0, 0));

    changeMapType(MapModel::GeneratedMap);

}

void HWMapContainer::onImageReceived(const QPixmap &newImage)

{

    // When image received from the engine.

    switch (m_mapInfo.type)

    {

        case MapModel::GeneratedMap:

        case MapModel::GeneratedMaze:

        case MapModel::GeneratedPerlin:

        case MapModel::HandDrawnMap:

        case MapModel::FortsMap:

            setImage(newImage);

            break;

        // Throw away image if we have switched the map mode in the meantime

        default:

            return;

    }

}

void HWMapContainer::setImage(const QPixmap &newImage)

{

    addInfoToPreview(newImage);

    pMap = 0;

    cType->setEnabled(isMaster());

}

void HWMapContainer::setImage(const QPixmap &newImage, const QLinearGradient &linearGrad, bool showHHLimit)

{

    addInfoToPreview(newImage, linearGrad, showHHLimit);

    pMap = 0;

    cType->setEnabled(isMaster());

}

void HWMapContainer::setHHLimit(int newHHLimit)

{

    hhLimit = newHHLimit;

}

void HWMapContainer::addInfoToPreview(const QPixmap &image)

{

    addInfoToPreview(image, linearGradNormal, true);

}

// Should this add text to identify map size?

void HWMapContainer::addInfoToPreview(const QPixmap &image, const QLinearGradient &linearGrad, bool drawHHLimit)

{

    QPixmap finalImage = QPixmap(image.size());

    QPainter p(&finalImage);

    p.fillRect(finalImage.rect(), linearGrad);

    p.drawPixmap(finalImage.rect(), image);

    if (drawHHLimit)

    {

        p.setPen(QColor(0xff,0xcc,0x00));

        p.setBrush(QColor(0, 0, 0));

        p.setFont(QFont("MS Shell Dlg", 10));

        p.drawRect(finalImage.rect().width() - hhSmall.rect().width() - 28, 3, 40, 20);

        QString text = (hhLimit > 0) ? QString::number(hhLimit) : "?";

        p.drawText(finalImage.rect().width() - hhSmall.rect().width() - 14 - (hhLimit > 9 ? 10 : 0), 18, text);

        p.drawPixmap(finalImage.rect().width() - hhSmall.rect().width() - 5, 5, hhSmall.rect().width(), hhSmall.rect().height(), hhSmall);

    }

    // Set the map preview image. Make sure it is always colored the same,

    // no matter if disabled or not.

    QIcon mapPreviewIcon = QIcon();

    mapPreviewIcon.addPixmap(finalImage, QIcon::Normal);

    mapPreviewIcon.addPixmap(finalImage, QIcon::Disabled);

    mapPreview->setIcon(mapPreviewIcon);

    mapPreview->setIconSize(finalImage.size());

}

void HWMapContainer::askForGeneratedPreview()

{

    pMap = new HWMap(this);

    connect(pMap, SIGNAL(ImageReceived(QPixmap)), this, SLOT(onImageReceived(const QPixmap)));

    connect(pMap, SIGNAL(HHLimitReceived(int)), this, SLOT(setHHLimit(int)));

    connect(pMap, SIGNAL(destroyed(QObject *)), this, SLOT(onPreviewMapDestroyed(QObject *)));

    pMap->getImage(m_seed,

                   getTemplateFilter(),

                   get_mapgen(),

                   getMazeSize(),

                   getDrawnMapData(),

                   m_script,

                   m_scriptparam,

                   m_mapFeatureSize

                  );

    setHHLimit(0);

    QPixmap waitImage(m_previewSize);

    waitImage.fill(Qt::transparent);

    QPainter p(&waitImage);

    const QPixmap waitIcon(":/res/iconTime.png");

    int x = (waitImage.width() - waitIcon.width()) / 2;

    int y = (waitImage.height() - waitIcon.height()) / 2;

    p.drawPixmap(QPoint(x, y), waitIcon);

    setImage(waitImage, linearGradLoading, false);

    cType->setEnabled(false);

}

void HWMapContainer::previewClicked()

{

    if (isMaster()) // should only perform these if master, but disabling the button when not, causes an unattractive preview.

        switch (m_mapInfo.type)

        {

            case MapModel::HandDrawnMap:

                emit drawMapRequested();

                break;

            default:

                setRandomMap();

                break;

        }