~sithlord48/hyne/master

/****************************************************************************
 ** Hyne Final Fantasy VIII Save Editor
 ** Copyright (C) 2009-2013 Arzel Jérôme <myst6re@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, either version 3 of the License, or
 ** (at your option) any later version.
 **
 ** 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, see <http://www.gnu.org/licenses/>.
 ****************************************************************************/

#include "HeaderDialog.h"
#include "Config.h"

HeaderDialog::HeaderDialog(SaveData *saveData, QWidget *parent, ViewType viewType)
	: QDialog(parent, Qt::Dialog | Qt::WindowCloseButtonHint), saveData(saveData), viewType(viewType)
{
	setWindowTitle(tr("Propriétés"));

	group1 = new QGroupBox(tr("Memory card header"), this);

	exists_lbl = new QLabel(tr("Existe :"));
	exists_lbl->setTextFormat(Qt::PlainText);
	exists = new QCheckBox;

	QLabel *country_lbl = new QLabel(tr("Région :"));
	country_lbl->setTextFormat(Qt::PlainText);
	country = new QComboBox;
	country->addItem(QIcon(":/images/jp.png"), tr("Japon"), COUNTRY_JP);
	country->addItem(QIcon(":/images/us.png"), tr("Amérique"), COUNTRY_US);
	country->addItem(QIcon(":/images/eu.png"), tr("Europe"), COUNTRY_EU);
	country->addItem(tr("Invalide"), '\x00');

	country->setCurrentIndex(Config::valueVar(Config::LastCountry).toInt());

	QLabel *code_lbl = new QLabel(tr("Code :"));
	code_lbl->setTextFormat(Qt::PlainText);
	code = new QComboBox;
	code->setEditable(true);
	code->lineEdit()->setMaxLength(10);
	fillCode(code);

	QString lastGameCode = Config::value(Config::LastGameCode);
	if(!lastGameCode.isEmpty()) {
		setCode(lastGameCode);
	}

	id_lbl = new QLabel(tr("Identifiant :"));
	id_lbl->setTextFormat(Qt::PlainText);
	id = new QComboBox;
	id->setEditable(true);
	id->lineEdit()->setMaxLength(8);
	fillId(id, saveData->id());

	QGridLayout *layout1 = new QGridLayout(group1);
	layout1->addWidget(exists_lbl, 0, 0);
	layout1->addWidget(exists, 0, 1);
	layout1->addWidget(country_lbl, 1, 0);
	layout1->addWidget(country, 1, 1);
	layout1->addWidget(code_lbl, 2, 0);
	layout1->addWidget(code, 2, 1);
	layout1->addWidget(id_lbl, 3, 0);
	layout1->addWidget(id, 3, 1);

	group2 = new QGroupBox(tr("Save header"), this);

	desc = new QLineEdit();
	desc->setMaxLength(64);

	desc_auto = new QCheckBox(tr("Auto."));

	QLabel *bloc_lbl = new QLabel(tr("Blocs utilisés :"));
	bloc = new QLabel();
	bloc->setTextFormat(Qt::PlainText);

	QLabel *desc_lbl = new QLabel(tr("Description :"));
	desc_lbl->setTextFormat(Qt::PlainText);

	QList<SaveIconData> saveIcons;
	for(int i=0 ; i<16 ; ++i) {
		QFile iconFile(QString(":/data/icon%1.psico").arg(i));
		if(iconFile.open(QIODevice::ReadOnly)) {
			saveIcons.append(SaveIconData(iconFile.readAll()));
			iconFile.close();
		}
	}

	QLabel *icon1_lbl = new QLabel(tr("Icône :"));
	icon1_lbl->setTextFormat(Qt::PlainText);
	icon1 = new QComboBox();
	icon1->setIconSize(QSize(16, 16));
	foreach(const SaveIconData &saveIcon, saveIcons) {
		icon1->addItem(QIcon(saveIcon.icon()), QString(), saveIcon.data().left(160));
	}
	QPushButton *icon1_saveButton = new QPushButton();
	QIcon saveIcon(QApplication::style()->standardIcon(QStyle::SP_DialogSaveButton));
	icon1_saveButton->setIcon(saveIcon);
	icon1_saveButton->setFlat(true);
	icon2_lbl = new QLabel(tr("Icône additionnel :"));
	icon2_lbl->setTextFormat(Qt::PlainText);
	icon2 = new QComboBox();
	icon2->setIconSize(QSize(32, 32));
	foreach(const SaveIconData &saveIcon, saveIcons) {
		icon2->addItem(QIcon(saveIcon.icon(0, true)), QString(), saveIcon.data().mid(160));
	}
	icon2_saveButton = new QPushButton();
	icon2_saveButton->setIcon(saveIcon);
	icon2_saveButton->setFlat(true);

	QGridLayout *layout2 = new QGridLayout(group2);
	layout2->addWidget(bloc_lbl, 0, 0);
	layout2->addWidget(bloc, 0, 1, 1, 2);
	layout2->addWidget(desc_lbl, 1, 0);
	layout2->addWidget(desc, 1, 1, 1, 2);
	layout2->addWidget(desc_auto, 1, 3);
	layout2->addWidget(icon1_lbl, 2, 0);
	layout2->addWidget(icon1, 2, 1);
	layout2->addWidget(icon1_saveButton, 2, 2, Qt::AlignRight);
	layout2->addWidget(icon2_lbl, 3, 0);
	layout2->addWidget(icon2, 3, 1);
	layout2->addWidget(icon2_saveButton, 3, 2, Qt::AlignRight);

	QDialogButtonBox *buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Close);

	QGridLayout *layout = new QGridLayout(this);
	layout->addWidget(group1, 0, 0);
	layout->addWidget(group2, 1, 0);
	layout->setRowStretch(2, 1);
	layout->addWidget(buttonBox, 3, 0);

	connect(buttonBox, SIGNAL(accepted()), SLOT(accept()));
	connect(buttonBox, SIGNAL(rejected()), SLOT(reject()));
	connect(icon1_saveButton, SIGNAL(released()), SLOT(saveIcon1()));
	connect(icon2_saveButton, SIGNAL(released()), SLOT(saveIcon2()));
	connect(desc_auto, SIGNAL(toggled(bool)), desc, SLOT(setDisabled(bool)));

	fill();
}

void HeaderDialog::fillCode(QComboBox *code)
{
	QIcon jp(":/images/jp.png");
	code->addItem(jp, "SLPSP01880");//japanese
	code->addItem(jp, "SCPSP45375");//japanese (???)
	code->addItem(jp, "SLPMP87384");//japanese (Ultimate Hits)
	code->addItem(QIcon(":/images/us.png"), "SLUSP00892");//american
	code->addItem(QIcon(":/images/uk.png"), "SLESP02080");//english
	code->addItem(QIcon(":/images/fr.png"), "SLESP02081");//french
	code->addItem(QIcon(":/images/de.png"), "SLESP02082");//german
	code->addItem(QIcon(":/images/it.png"), "SLESP02083");//italian
	code->addItem(QIcon(":/images/sp.png"), "SLESP02084");//spanish
}

void HeaderDialog::fillId(QComboBox *id, int saveId)
{
	id->addItem(QIcon(":/images/jp.png"), QString("FF08%1").arg(saveId, 2, 10, QChar('0')));
	id->addItem(QIcon(":/images/us.png"), QString("0426%1").arg(saveId, 2, 10, QChar('0')));
	id->addItem(QIcon(":/images/eu.png"), QString("0520%1").arg(saveId, 2, 10, QChar('0')));
}

void HeaderDialog::fill()
{
	setWindowIcon(QIcon(saveData->saveIcon().icon()));

	if(viewType == NormalView && !saveData->hasMCHeader()) {
		group1->hide();
	}
	else {
		// Fill group1 (MCHeader)

		if(viewType != NormalView) {
			exists->setChecked(true);
		}
		else {
			exists->setChecked(!saveData->isDelete());
			setId(saveData->MCHeaderId());
		}

		if(viewType != CreateView || saveData->hasMCHeader()) {
			setCountry(saveData->MCHeaderCountry());
			setCode(saveData->MCHeaderCode());
		}
	}

	if(!saveData->hasExistsInfos()) {
		exists->hide();
		exists_lbl->hide();
	}

	if(viewType == RestoreView || viewType == CreateView) {
		group2->hide();

		exists->hide();
		exists_lbl->hide();
		id_lbl->hide();
		id->hide();
		country->setMaxCount(3);
	}
	else {
		// Fill group2 (SC_header)

		desc->setText(saveData->shortDescription());
		desc_auto->setChecked(saveData->isDescriptionAuto());
		desc_auto->setVisible(saveData->isFF8());
		bloc->setText(QString::number(saveData->blockCount()));
		QImage pix = saveData->saveIcon().icon().toImage();
		int currentIndex = -1;
		for(int i=0 ; i<icon1->count() ; ++i) {
			if(icon1->itemIcon(i).pixmap(16).toImage() == pix) {
				currentIndex = i;
				break;
			}
		}
		if(currentIndex == -1) {
			icon1->addItem(QIcon(QPixmap::fromImage(pix)), QString(), QByteArray());
			currentIndex = icon1->count()-1;
		}
		icon1->setCurrentIndex(currentIndex);
		if(saveData->isFF8()) {
			pix = saveData->saveIcon().icon(0, true).toImage();
			currentIndex = -1;
			for(int i=0 ; i<icon2->count() ; ++i) {
				if(icon2->itemIcon(i).pixmap(32).toImage() == pix) {
					currentIndex = i;
					break;
				}
			}
			if(currentIndex == -1) {
				icon2->addItem(QIcon(QPixmap::fromImage(pix)), QString(), QByteArray());
				currentIndex = icon1->count()-1;
			}
			icon2->setCurrentIndex(currentIndex);
		}
		else {
			icon2_lbl->hide();
			icon2->hide();
			icon2_saveButton->hide();
		}
	}
}

void HeaderDialog::setCountry(char c)
{
	int index;
	if((index=country->findData(c)) != -1)
	{
		country->setCurrentIndex(index);
		if(index != 3)		country->setMaxCount(3);
	}
	else
	{
		if(viewType == RestoreView) {
			country->setCurrentIndex(0);
		}
		else {
			country->setItemData(3, c);
			country->setCurrentIndex(3);//Invalid
		}
	}
}

void HeaderDialog::setCode(const QString &codestr)
{
	int index;
	if(codestr.size()==10 && (index=code->findText(codestr)) != -1)
	{
		code->setCurrentIndex(index);
	}
	else if(viewType == NormalView) {
		code->insertItem(0, codestr);
		code->setCurrentIndex(0);
	}
}

void HeaderDialog::setId(const QString &idStr)
{
	if(idStr.startsWith("FF08")) {
		id->setCurrentIndex(0);
	} else if(idStr.startsWith("0426")) {
		id->setCurrentIndex(1);
	} else if(idStr.startsWith("0520")) {
		id->setCurrentIndex(2);
	} else {
		id->addItem(idStr);
		id->setCurrentIndex(id->count()-1);
	}
	id->lineEdit()->setText(idStr);
}

void HeaderDialog::accept()
{
	if(viewType != NormalView || saveData->hasMCHeader()) {
		Config::setValue(Config::LastCountry, country->currentIndex());
		Config::setValue(Config::LastGameCode, code->currentText());

		// Edit MC Header
		saveData->setMCHeader(exists->isChecked(),
							  country->itemData(country->currentIndex()).toInt(),
							  code->currentText(),
							  id->isVisible() ? id->currentText() : QString());
	}

	// Edit short description
	saveData->setShortDescription(desc->text());
	saveData->setDescriptionAuto(desc_auto->isChecked());

	// Edit icon 1
	SaveIconData saveIcon = saveData->saveIcon();
	QByteArray itemData = icon1->itemData(icon1->currentIndex()).toByteArray();
	if(!itemData.isEmpty()) {
		saveIcon.setData(itemData + saveIcon.data().mid(itemData.size()));
		saveData->setSaveIcon(saveIcon);
	}

	if(saveData->isFF8()) {
		// Edit icon 2
		itemData = icon2->itemData(icon2->currentIndex()).toByteArray();
		if(!itemData.isEmpty()) {
			saveIcon.setData(saveIcon.data().left(160) + itemData);
			saveData->setSaveIcon(saveIcon);
		}
	}

	QDialog::accept();
}

void HeaderDialog::saveIcon(bool chocobo_world_icon)
{
	QString path, savePathIcon = Config::value(Config::SavePathIcon);

	path = savePathIcon.isEmpty() ? QString() : savePathIcon % "/";
	path = QFileDialog::getSaveFileName(this, tr("Enregistrer sous"), path % QString("icon%1%2.png").arg(saveData->id()+1).arg(chocobo_world_icon ? "b" : ""), tr("Image PNG (*.png);;Image JPG (*.jpg *.jpeg);;Image BMP (*.bmp)"));
	if(path.isEmpty())		return;

	int index = path.lastIndexOf('/');
	Config::setValue(Config::SavePathIcon, index == -1 ? path : path.left(index));

	if(!saveData->saveIcon().icon(0, chocobo_world_icon).save(path)) {
		QMessageBox::warning(this, tr("Erreur"), tr("Format incorrect."));
	}
}