/*
 * Hedgewars, a free turn based strategy game
 * Copyright (c) 2008 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 <QDir>
#include <QPixmap>
#include <QPainter>
#include "hwconsts.h"
#include "hats.h"

HatsModel::HatsModel(QObject* parent) :
  QAbstractListModel(parent)
{
	QPixmap hhpix = QPixmap(datadir->absolutePath() + "/Graphics/Hedgehog/Idle.png").copy(0, 0, 32, 32);

	QDir tmpdir;
	tmpdir.cd(datadir->absolutePath());
	tmpdir.cd("Graphics");
	tmpdir.cd("Hats");

	tmpdir.setFilter(QDir::Files);


	QStringList hatsList = tmpdir.entryList(QStringList("*.png"));
	for (QStringList::Iterator it = hatsList.begin(); it != hatsList.end(); ++it )
	{
		QString str = (*it).replace(QRegExp("^(.*)\\.png"), "\\1");
		QPixmap pix(datadir->absolutePath() + "/Graphics/Hats/" + str + ".png");

		QPixmap tmppix(32, 37);
		tmppix.fill(QColor(Qt::transparent));
		
		QPainter painter(&tmppix);
		painter.drawPixmap(QPoint(0, 5), hhpix);
		painter.drawPixmap(QPoint(0, 0), pix.copy(0, 0, 32, 32));
		painter.end();

		hats.append(qMakePair(str, QIcon(tmppix)));
	}
}

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

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

/*int HatsModel::columnCount(const QModelIndex & parent) const
{
	if (parent.isValid())
		return 0;
	else
		return 2;
}
*/
QVariant HatsModel::data(const QModelIndex &index,
                         int role) const
{
	if (!index.isValid() || index.row() < 0
		|| index.row() >= hats.size()
		|| (role != Qt::DisplayRole && role != Qt::DecorationRole))
		return QVariant();

	if (role == Qt::DisplayRole)
		return hats.at(index.row()).first;
	else // role == Qt::DecorationRole
		return hats.at(index.row()).second;
}