/*
 * Hedgewars, a free turn based strategy game
 * Copyright (c) 2009 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 <QDebug>
#include <QModelIndex>

#include "ammoSchemeModel.h"
#include "hwconsts.h"

QList<QVariant> defaultScheme = QList<QVariant>()
		<< QVariant("Default")     // name           0
		<< QVariant(false)         // fortsmode      1
		<< QVariant(false)         // team divide    2
		<< QVariant(false)         // solid land     3
		<< QVariant(false)         // border         4
		<< QVariant(false)         // low gravity    5
		<< QVariant(false)         // laser sight    6
		<< QVariant(false)         // invulnerable   7
		<< QVariant(true)          // add mines      8
		<< QVariant(false)         // vampiric       9
		<< QVariant(false)         // karma          10
		<< QVariant(false)         // artillery      11
		<< QVariant(100)           // damage modfier 12
		<< QVariant(45)            // turn time      13
		<< QVariant(100)           // init health    14
		<< QVariant(15)            // sudden death   15
		<< QVariant(5)             // case prob      16
		<< QVariant(3)             //  mines time    17
		<< QVariant(4)             //  landadds      18
		;

AmmoSchemeModel::AmmoSchemeModel(QObject* parent, const QString & fileName) :
	QAbstractTableModel(parent),
	numberOfDefaultSchemes(4),
	fileConfig(fileName, QSettings::IniFormat)
{
	QStringList predefSchemesNames;
	predefSchemesNames
		<< "Default"
		<< "Pro mode"
		<< "Shoppa"
		<< "Basketball"
		<< "Minefield";

	spNames = QStringList()
		<< "name"             //  0
		<< "fortsmode"        //  1
		<< "divteams"         //  2
		<< "solidland"        //  3
		<< "border"           //  4
		<< "lowgrav"          //  5
		<< "laser"            //  6
		<< "invulnerability"  //  7
		<< "mines"            //  8
		<< "vampiric"         //  9
		<< "karma"            // 10
		<< "artillery"        // 11
		<< "damagefactor"     // 12
		<< "turntime"         // 13
		<< "health"           // 14
		<< "suddendeath"      // 15
		<< "caseprobability"  // 16
		<< "minestime"        // 17
		<< "landadds"         // 18
		;

	QList<QVariant> proMode;
	proMode
		<< predefSchemesNames[1]   // name           0
		<< QVariant(false)         // fortsmode      1
		<< QVariant(false)         // team divide    2
		<< QVariant(false)         // solid land     3
		<< QVariant(false)         // border         4
		<< QVariant(false)         // low gravity    5
		<< QVariant(false)         // laser sight    6
		<< QVariant(false)         // invulnerable   7
		<< QVariant(false)         // add mines      8
		<< QVariant(false)         // vampiric       9
		<< QVariant(false)         // karma          10
		<< QVariant(false)         // artillery      11
		<< QVariant(100)           // damage modfier 12
		<< QVariant(15)            // turn time      13
		<< QVariant(100)           // init health    14
		<< QVariant(15)            // sudden death   15
		<< QVariant(0)             // case prob      16
		<< QVariant(3)             //  mines time    17
		<< QVariant(4)             //  landadds      18
		;

	QList<QVariant> shoppa;
	shoppa
		<< predefSchemesNames[2]   // name           0
		<< QVariant(false)         // fortsmode      1
		<< QVariant(false)         // team divide    2
		<< QVariant(true)          // solid land     3
		<< QVariant(true)          // border         4
		<< QVariant(false)         // low gravity    5
		<< QVariant(false)         // laser sight    6
		<< QVariant(false)         // invulnerable   7
		<< QVariant(false)         // add mines      8
		<< QVariant(false)         // vampiric       9
		<< QVariant(false)         // karma          10
		<< QVariant(false)         // artillery      11
		<< QVariant(100)           // damage modfier 12
		<< QVariant(30)            // turn time      13
		<< QVariant(100)           // init health    14
		<< QVariant(50)            // sudden death   15
		<< QVariant(1)             // case prob      16
		<< QVariant(3)             //  mines time    17
		<< QVariant(4)             //  landadds      18
		;

	QList<QVariant> basketball;
	basketball
		<< predefSchemesNames[3]   // name           0
		<< QVariant(false)         // fortsmode      1
		<< QVariant(false)         // team divide    2
		<< QVariant(true)          // solid land     3
		<< QVariant(true)          // border         4
		<< QVariant(true)          // low gravity    5
		<< QVariant(false)         // laser sight    6
		<< QVariant(true)          // invulnerable   7
		<< QVariant(false)         // add mines      8
		<< QVariant(false)         // vampiric       9
		<< QVariant(false)         // karma          10
		<< QVariant(false)         // artillery      11
		<< QVariant(100)           // damage modfier 12
		<< QVariant(30)            // turn time      13
		<< QVariant(100)           // init health    14
		<< QVariant(15)            // sudden death   15
		<< QVariant(0)             // case prob      16
		<< QVariant(3)             //  mines time    17
		<< QVariant(4)             //  landadds      18
		;

	QList<QVariant> minefield;
	minefield
		<< predefSchemesNames[4]   // name           0
		<< QVariant(false)         // fortsmode      1
		<< QVariant(false)         // team divide    2
		<< QVariant(false)         // solid land     3
		<< QVariant(false)         // border         4
		<< QVariant(false)         // low gravity    5
		<< QVariant(false)         // laser sight    6
		<< QVariant(false)         // invulnerable   7
		<< QVariant(true)          // add mines      8
		<< QVariant(false)         // vampiric       9
		<< QVariant(false)         // karma          10
		<< QVariant(false)         // artillery      11
		<< QVariant(150)           // damage modfier 12
		<< QVariant(30)            // turn time      13
		<< QVariant(50)            // init health    14
		<< QVariant(15)            // sudden death   15
		<< QVariant(0)             // case prob      16
		<< QVariant(0)             //  mines time    17
		<< QVariant(50)            //  landadds      18
		;

	schemes.append(defaultScheme);
	schemes.append(proMode);
	schemes.append(shoppa);
	schemes.append(basketball);
	schemes.append(minefield);


	int size = fileConfig.beginReadArray("schemes");
	for (int i = 0; i < size; ++i) {
		fileConfig.setArrayIndex(i);

		if (!predefSchemesNames.contains(fileConfig.value(spNames[0]).toString()))
		{
			QList<QVariant> scheme;

			for (int k = 0; k < spNames.size(); ++k)
				scheme << fileConfig.value(spNames[k], defaultScheme[k]);

			schemes.append(scheme);
		}
	}
	fileConfig.endArray();
}

QVariant AmmoSchemeModel::headerData(int section, Qt::Orientation orientation, int role) const
{
	return QVariant();
}

int AmmoSchemeModel::rowCount(const QModelIndex &parent) const
{
	if (parent.isValid())
		return 0;
	else
		return schemes.size();
}

int AmmoSchemeModel::columnCount(const QModelIndex & parent) const
{
	if (parent.isValid())
		return 0;
	else
		return defaultScheme.size();
}

Qt::ItemFlags AmmoSchemeModel::flags(const QModelIndex & index) const
{
	return
		Qt::ItemIsEnabled
		| Qt::ItemIsSelectable
		| Qt::ItemIsEditable;
}

bool AmmoSchemeModel::setData(const QModelIndex & index, const QVariant & value, int role)
{
	if (!index.isValid() || index.row() < numberOfDefaultSchemes
		|| index.row() >= schemes.size()
		|| index.column() >= defaultScheme.size()
		|| role != Qt::EditRole)
		return false;

	schemes[index.row()][index.column()] = value;

	emit dataChanged(index, index);
	return true;
}

bool AmmoSchemeModel::insertRows(int row, int count, const QModelIndex & parent)
{
	beginInsertRows(parent, row, row);

	QList<QVariant> newScheme = defaultScheme;
	newScheme[0] = QVariant(tr("new"));

	schemes.insert(row, newScheme);

	endInsertRows();

	return true;
}

bool AmmoSchemeModel::removeRows(int row, int count, const QModelIndex & parent)
{
	if(count != 1
		|| row < numberOfDefaultSchemes
		|| row >= schemes.size())
		return false;

	beginRemoveRows(parent, row, row);

	schemes.removeAt(row);

	endRemoveRows();

	return true;
}

QVariant AmmoSchemeModel::data(const QModelIndex &index, int role) const
{
	if (!index.isValid() || index.row() < 0
		|| index.row() >= schemes.size()
		|| index.column() >= defaultScheme.size()
		|| (role != Qt::EditRole && role != Qt::DisplayRole)
		)
		return QVariant();

	return schemes[index.row()][index.column()];
}

void AmmoSchemeModel::Save()
{
	fileConfig.beginWriteArray("schemes", schemes.size());

	for (int i = 0; i < schemes.size(); ++i) {
		fileConfig.setArrayIndex(i);

		QList<QVariant> scheme = schemes[i];

		for (int k = 0; k < scheme.size(); ++k)
			fileConfig.setValue(spNames[k], scheme[k]);
	}
	fileConfig.endArray();
}


NetAmmoSchemeModel::NetAmmoSchemeModel(QObject * parent) :
	QAbstractTableModel(parent)
{
	netScheme = defaultScheme;
}

QVariant NetAmmoSchemeModel::headerData(int section, Qt::Orientation orientation, int role) const
{
	return QVariant();
}

int NetAmmoSchemeModel::rowCount(const QModelIndex & parent) const
{
	if (parent.isValid())
		return 0;
	else
		return 1;
}

int NetAmmoSchemeModel::columnCount(const QModelIndex & parent) const
{
	if (parent.isValid())
		return 0;
	else
		return defaultScheme.size();
}

QVariant NetAmmoSchemeModel::data(const QModelIndex &index, int role) const
{
	if (!index.isValid() || index.row() < 0
		|| index.row() > 1
		|| index.column() >= defaultScheme.size()
		|| (role != Qt::EditRole && role != Qt::DisplayRole)
		)
		return QVariant();

	return netScheme[index.column()];
}

void NetAmmoSchemeModel::setNetSchemeConfig(QStringList & cfg)
{
	if(cfg.size() != netScheme.size())
	{
		qWarning("Incorrect scheme cfg size");
		return;
	}

	for(int i = 0; i < cfg.size(); ++i)
		netScheme[i] = QVariant(cfg[i]);

	reset();
}