/*

 * 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

 */

#include <QPushButton>

#include <QBuffer>

#include <QUuid>

#include <QBitmap>

#include <QPainter>

#include <QLinearGradient>

#include <QColor>

#include <QTextStream>

#include <QLabel>

#include <QListView>

#include <QVBoxLayout>

#include <QIcon>

#include <QLineEdit>

#include <QStringListModel>

#include <QListWidget>

#include <QListWidgetItem>

#include <QDebug>

#include <QFile>

#include <QFileDialog>

#include <QInputDialog>

#include <QMessageBox>

#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)

{

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

    hhLimit = 18;

    templateFilter = 0;

    m_master = true;

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

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

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

    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 */

    QHBoxLayout * typeLayout = new QHBoxLayout();

    QHBoxLayout * seedLayout = new QHBoxLayout();

    QHBoxLayout * twoColumnLayout = new QHBoxLayout();

    QVBoxLayout * leftLayout = new QVBoxLayout();

    QVBoxLayout * rightLayout = new QVBoxLayout();

    twoColumnLayout->addLayout(leftLayout, 0);

    twoColumnLayout->addStretch(1);

    twoColumnLayout->addLayout(rightLayout, 0);

    QVBoxLayout * drawnControls = new QVBoxLayout();

    leftLayout->addLayout(typeLayout, 0);

    rightLayout->addLayout(seedLayout, 0);

    /* Map type combobox */

    typeLayout->setSpacing(10);

    typeLayout->addWidget(new QLabel(tr("Map type:")), 0);

    cType = new QComboBox(this);

    typeLayout->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);

    connect(cType, SIGNAL(currentIndexChanged(int)), this, SLOT(mapTypeChanged(int)));

    m_childWidgets << cType;

    /* Randomize button */

    seedLayout->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());

    seedLayout->addWidget(btnRandomize, 1);

    /* Seed button */

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

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

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

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

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

    seedLayout->addWidget(btnSeed, 0);

    /* Map preview label */

    QLabel * lblMapPreviewText = new QLabel(this);

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

    leftLayout->addWidget(lblMapPreviewText, 0);

    leftLayout->addSpacing(2);

    /* Map Preview */

    mapPreview = new QLabel(this);

    mapPreview->setObjectName("mapPreview");

    mapPreview->setFixedSize(256, 128);

    leftLayout->addWidget(mapPreview, 0);

    /* Bottom-Left layout */

    QVBoxLayout * bottomLeftLayout = new QVBoxLayout();

    leftLayout->addLayout(bottomLeftLayout, 1);

    /* Map list label */

    lblMapList = new QLabel();

    rightLayout->addWidget(lblMapList, 0);

    /* Static maps list */

    staticMapList = new QListView;

    staticMapList->setModel(m_staticMapModel);

    rightLayout->addWidget(staticMapList, 1);

    staticMapList->setEditTriggers(QAbstractItemView::NoEditTriggers);

    m_childWidgets << staticMapList;

    QItemSelectionModel * staticSelectionModel = staticMapList->selectionModel();

    connect(staticSelectionModel,

            SIGNAL(currentRowChanged(const QModelIndex &, const QModelIndex &)),

            this,

            SLOT(staticMapChanged(const QModelIndex &, const QModelIndex &)));

    /* Mission maps list */

    missionMapList = new QListView;

    missionMapList->setModel(m_missionMapModel);

    missionMapList->setEditTriggers(QAbstractItemView::NoEditTriggers);

    rightLayout->addWidget(missionMapList, 1);

    m_childWidgets << missionMapList;

    QItemSelectionModel * missionSelectionModel = missionMapList->selectionModel();

    connect(missionSelectionModel,

            SIGNAL(currentRowChanged(const QModelIndex &, const QModelIndex &)),

            this,

            SLOT(missionMapChanged(const QModelIndex &, const QModelIndex &)));

    /* Map load and edit buttons */

    drawnControls->addStretch(1);

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

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

    drawnControls->addWidget(btnLoadMap, 0);

    m_childWidgets << btnLoadMap;

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

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

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

    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();

    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();

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

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

    new QListWidgetItem(tr("Largetunnels"), 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);

    /* Mission description */

    lblDesc = new QLabel();

    lblDesc->setWordWrap(true);

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

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

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

    bottomLeftLayout->addWidget(lblDesc, 1);

    /* Add stretch above theme button */

    bottomLeftLayout->addStretch(1);

    /* Theme chooser */

    btnTheme = new QPushButton();

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

    m_childWidgets << btnTheme;

    bottomLeftLayout->addWidget(btnTheme, 0);

    /* Add everything to main layout */

    mainLayout.addLayout(twoColumnLayout, 0);

    /* Set defaults */

    setRandomTheme();

    setRandomSeed();

    setMazeSize(0);

    setTemplateFilter(0);

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

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

    updateTheme(m_themeModel->index(0, 0));

    changeMapType(MapModel::GeneratedMap);

}

void HWMapContainer::setImage(const QImage newImage)

{

    QPixmap px(m_previewSize);

    QPixmap pxres(px.size());

    QPainter p(&pxres);

    px.fill(Qt::yellow);

    QBitmap bm = QBitmap::fromImage(newImage);

    px.setMask(bm);

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

    p.drawPixmap(QPoint(0, 0), px);

    addInfoToPreview(pxres);

    pMap = 0;

    cType->setEnabled(isMaster());

}

void HWMapContainer::setHHLimit(int newHHLimit)

{

    hhLimit = newHHLimit;

}

// Should this add text to identify map size?

void HWMapContainer::addInfoToPreview(QPixmap image)

{

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

    finalImage.fill(QColor(0, 0, 0, 0));

    QPainter p(&finalImage);

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

    //p.setPen(QColor(0xf4,0x9e,0xe9));

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

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

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

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

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

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

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

    mapPreview->setPixmap(finalImage);

}

void HWMapContainer::askForGeneratedPreview()

{

    pMap = new HWMap(this);

    connect(pMap, SIGNAL(ImageReceived(const QImage)), this, SLOT(setImage(const QImage)));

    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()

                  );

    setHHLimit(0);

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

    QPixmap waitImage(m_previewSize);

    QPainter p(&waitImage);

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

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

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

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

    addInfoToPreview(waitImage);

    cType->setEnabled(false);

}

QString HWMapContainer::getCurrentSeed() const

{

    return m_seed;

}

QString HWMapContainer::getCurrentMap() const

{

    switch (m_mapInfo.type)

    {

        case MapModel::StaticMap:

        case MapModel::MissionMap:

            return m_curMap;

        default:

            return QString();

    }

}

QString HWMapContainer::getCurrentTheme() const

{

    return(m_theme);

}

bool HWMapContainer::getCurrentIsMission() const

{

    return(m_mapInfo.type == MapModel::MissionMap);

}

int HWMapContainer::getCurrentHHLimit() const

{

    return hhLimit;

}

QString HWMapContainer::getCurrentScheme() const

{

    return(m_mapInfo.scheme);

}

QString HWMapContainer::getCurrentWeapons() const

{

    return(m_mapInfo.weapons);

}

quint32 HWMapContainer::getTemplateFilter() const

{

    return generationStyles->currentRow();

}

void HWMapContainer::resizeEvent ( QResizeEvent * event )

{

    Q_UNUSED(event);

}

void HWMapContainer::intSetSeed(const QString & seed)

{

    m_seed = seed;

}

void HWMapContainer::setSeed(const QString & seed)

{

    intSetSeed(seed);

    if ((m_mapInfo.type == MapModel::GeneratedMap) || (m_mapInfo.type == MapModel::GeneratedMaze))

        updatePreview();

}

void HWMapContainer::intSetMap(const QString & map)

{

    if (map == "+rnd+")

    {

        changeMapType(MapModel::GeneratedMap);

    }

    else if (map == "+maze+")

    {

        changeMapType(MapModel::GeneratedMaze);

    }

    else if (map == "+drawn+")

    {

        changeMapType(MapModel::HandDrawnMap);

    }

    else if (m_staticMapModel->mapExists(map))

    {

        changeMapType(MapModel::StaticMap, m_staticMapModel->index(m_staticMapModel->findMap(map), 0));

    }

    else if (m_missionMapModel->mapExists(map))

    {

        changeMapType(MapModel::MissionMap, m_missionMapModel->index(m_missionMapModel->findMap(map), 0));

    } else

    {

        qDebug() << "HWMapContainer::intSetMap: Map doesn't exist: " << map;

    }

}

void HWMapContainer::setMap(const QString & map)

{

    if ((m_mapInfo.type == MapModel::Invalid) || (map != m_mapInfo.name))

        intSetMap(map);

}

void HWMapContainer::setTheme(const QString & theme)

{

    QModelIndexList mdl = m_themeModel->match(m_themeModel->index(0), Qt::DisplayRole, theme);

    if(mdl.size())

        updateTheme(mdl.at(0));

    else

        intSetIconlessTheme(theme);

}

void HWMapContainer::setRandomMap()

{

    setRandomSeed();

    switch(m_mapInfo.type)

    {

        case MapModel::GeneratedMap:

        case MapModel::GeneratedMaze:

            setRandomTheme();

            break;

        case MapModel::MissionMap:

            missionMapChanged(m_missionMapModel->index(rand() % m_missionMapModel->rowCount(), 0));

            break;

        case MapModel::StaticMap:

            staticMapChanged(m_staticMapModel->index(rand() % m_staticMapModel->rowCount(), 0));

            break;

        default:

            break;

    }

}

void HWMapContainer::setRandomSeed()

{

    setSeed(QUuid::createUuid().toString());

    emit seedChanged(m_seed);

}

void HWMapContainer::setRandomTheme()

{

    if(!m_themeModel->rowCount()) return;

    quint32 themeNum = rand() % m_themeModel->rowCount();

    updateTheme(m_themeModel->index(themeNum));

}

void HWMapContainer::intSetTemplateFilter(int filter)

{

    generationStyles->setCurrentRow(filter);

    emit newTemplateFilter(filter);

}

void HWMapContainer::setTemplateFilter(int filter)

{