Search.cpp

/****************************************************************************
 ** Deling Final Fantasy VIII Field Editor
 ** Copyright (C) 2009-2012 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 "Search.h"

Search::Search(QTreeWidget *fieldList, QWidget *parent)
	: QDialog(parent, Qt::Tool), fieldArchive(NULL), fieldList(fieldList)
{
	fieldID = -1;
	textID = from = groupID = methodID = opcodeID = 0;
	setWindowTitle(tr("Rechercher"));
	setWindowModality(Qt::NonModal);
	setSizeGripEnabled(true);

	tabWidget = new QTabWidget(this);
	tabWidget->addTab(textPageWidget(), tr("Textes"));
	tabWidget->addTab(scriptPageWidget(), tr("Scripts"));

	buttonNext = new QPushButton(tr("Chercher le suivant"), this);
	buttonPrev = new QPushButton(tr("Chercher le précédent"), this);
	buttonPrev->setAutoDefault(false);
	buttonNext->setAutoDefault(false);
	buttonNext->setEnabled(false);
	buttonPrev->setEnabled(false);
	buttonNext->setDefault(true);

	new QShortcut(QKeySequence::FindNext, this, SLOT(findNext()), 0, Qt::ApplicationShortcut);
	new QShortcut(QKeySequence::FindPrevious, this, SLOT(findPrev()), 0, Qt::ApplicationShortcut);

	// buttonNext.width == buttonPrev.width
	if(buttonPrev->sizeHint().width() > buttonNext->sizeHint().width())
		buttonNext->setFixedSize(buttonPrev->sizeHint());
	else
		buttonPrev->setFixedSize(buttonNext->sizeHint());
	
	QGridLayout *layout = new QGridLayout(this);
	layout->addWidget(tabWidget, 0, 0, 1, 2);
	layout->addWidget(buttonPrev, 1, 0, Qt::AlignRight);
	layout->addWidget(buttonNext, 1, 1, Qt::AlignLeft);
	QMargins margins = layout->contentsMargins();
	margins.setTop(0);
	margins.setLeft(0);
	margins.setRight(0);
	layout->setContentsMargins(margins);

	connect(buttonNext, SIGNAL(released()), SLOT(findNext()));
	connect(buttonPrev, SIGNAL(released()), SLOT(findPrev()));
	connect(tabWidget, SIGNAL(currentChanged(int)), SLOT(setFocus()));
	connect(searchTextField, SIGNAL(textEdited(QString)), searchScriptTextField, SLOT(setText(QString)));
	connect(searchScriptTextField, SIGNAL(textEdited(QString)), searchTextField, SLOT(setText(QString)));
}

Qt::CaseSensitivity Search::sensitivity() const
{
	QCheckBox *checkBox = currentIndex() == Text
						  ? this->checkBox
						  : this->scriptCheckBox;
	if(checkBox->isChecked())
		return Qt::CaseSensitive;
	else
		return Qt::CaseInsensitive;
}

QString Search::text() const
{
	QLineEdit *lineEdit = currentIndex() == Text
						  ? this->searchTextField
						  : this->searchScriptTextField;
	return lineEdit->text();
}

FieldArchive::Sorting Search::sorting() const
{
	switch(fieldList->sortColumn()) {
	case 1:
		return FieldArchive::SortByDesc;
	case 2:
		return FieldArchive::SortByMapId;
	default:
		return FieldArchive::SortByName;
	}
}

QRegExp Search::regexp() const
{
	QCheckBox *checkBox = currentIndex() == Text
						  ? this->checkBox2
						  : this->scriptCheckBox2;

	if(checkBox->isChecked()) {
		return QRegExp(text(), sensitivity());
	}
	return QRegExp(QRegExp::escape(text()), sensitivity());
}

QWidget *Search::textPageWidget()
{
	QWidget *ret = new QWidget(this);

	searchTextField = new QLineEdit(ret);
	searchTextField->setFocus();
	checkBox = new QCheckBox(tr("Sensible à la casse"), ret);
	checkBox2 = new QCheckBox(tr("Expression régulière"), ret);

	QVBoxLayout *layout = new QVBoxLayout(ret);
	layout->addWidget(searchTextField);
	layout->addWidget(checkBox);
	layout->addWidget(checkBox2);
	layout->addStretch(1);

	return ret;
}

QWidget *Search::scriptPageWidget()
{
	QWidget *ret = new QWidget(this);

	typeScriptChoice = new QComboBox(ret);
	typeScriptChoice->addItem(tr("Texte"));
	typeScriptChoice->addItem(tr("Opcode"));
	typeScriptChoice->addItem(tr("Variable"));
	typeScriptChoice->addItem(tr("Exécution"));

	scriptStacked = new QStackedWidget(ret);
	scriptStacked->addWidget(scriptPageWidget1());
	scriptStacked->addWidget(scriptPageWidget2());
	scriptStacked->addWidget(scriptPageWidget3());
	scriptStacked->addWidget(scriptPageWidget4());

	QVBoxLayout *layout = new QVBoxLayout(ret);
	layout->addWidget(typeScriptChoice);
	layout->addWidget(scriptStacked);
	layout->addStretch(1);
	layout->setContentsMargins(QMargins());

	connect(typeScriptChoice, SIGNAL(currentIndexChanged(int)), scriptStacked, SLOT(setCurrentIndex(int)));
	connect(typeScriptChoice, SIGNAL(currentIndexChanged(int)), SLOT(setFocus()));

	return ret;
}

QWidget *Search::scriptPageWidget1()
{
	QWidget *ret = new QWidget(this);

	searchScriptTextField = new QLineEdit(ret);
	searchScriptTextField->setFocus();
	scriptCheckBox = new QCheckBox(tr("Sensible à la casse"), ret);
	scriptCheckBox2 = new QCheckBox(tr("Expression régulière"), ret);

	QVBoxLayout *layout = new QVBoxLayout(ret);
	layout->addWidget(searchScriptTextField);
	layout->addWidget(scriptCheckBox);
	layout->addWidget(scriptCheckBox2);
	layout->addStretch(1);

	return ret;
}

QWidget *Search::scriptPageWidget2()
{
	QWidget *ret = new QWidget(this);

	searchOpcode = new QComboBox(ret);
	for(int i=0 ; i<376 ; ++i) {
		searchOpcode->addItem(QString("%1 - %2").arg(i, 3, 16, QChar('0')).arg(JsmOpcode::opcodes[i]));
	}

	searchOpcodeValue = new QSpinBox(ret);
	searchOpcodeValue->setRange(-1, 162777216);// 0 -> 2^24
	searchOpcodeValue->setValue(-1);

	QGridLayout *layout = new QGridLayout(ret);
	layout->addWidget(searchOpcode, 0, 0, 1, 2);
	layout->addWidget(new QLabel(tr("Paramètre :"), ret), 1, 0);
	layout->addWidget(searchOpcodeValue, 1, 1);
	layout->setRowStretch(2, 1);
	layout->setColumnStretch(1, 1);

	return ret;
}

QWidget *Search::scriptPageWidget3()
{
	QWidget *ret = new QWidget(this);

	selectScriptVar = new QComboBox(ret);
	for(int i=0 ; i<1536 ; ++i)
		selectScriptVar->addItem(QString("%1 - %2").arg(i, 4, 10, QChar('0')).arg(Config::value(QString("var%1").arg(i)).toString()));
	popScriptVar = new QCheckBox(tr("Pop uniquement"), ret);

	QVBoxLayout *layout = new QVBoxLayout(ret);
	layout->addWidget(selectScriptVar, 0, Qt::AlignTop);
	layout->addWidget(popScriptVar, 0, Qt::AlignTop);

	return ret;
}

QWidget *Search::scriptPageWidget4()
{
	QWidget *ret = new QWidget(this);

	selectScriptGroup = new QSpinBox(ret);
	selectScriptGroup->setRange(0, 256);
	selectScriptLabel = new QSpinBox(ret);
	selectScriptLabel->setRange(0, 65535);

	QGridLayout *layout = new QGridLayout(ret);
	layout->addWidget(new QLabel(tr("Groupe id"),ret), 0, 0);
	layout->addWidget(new QLabel(tr("Label"),ret), 0, 1);
	layout->addWidget(selectScriptGroup, 1, 0);
	layout->addWidget(selectScriptLabel, 1, 1);

	return ret;
}

Search::Tabs Search::currentIndex() const
{
	return Tabs(tabWidget->currentIndex());
}

void Search::setCurrentIndex(int index)
{
	if(index < 2) {
		tabWidget->setCurrentIndex(index);
	}
}

void Search::focusInEvent(QFocusEvent *)
{
	if(currentIndex() == Text) {
		searchTextField->setFocus();
		searchTextField->selectAll();
	} else if (currentIndex() == Script) {
		searchScriptTextField->setFocus();
		searchScriptTextField->selectAll();
	}
}

void Search::setFieldArchive(FieldArchive *fieldArchive)
{
	this->fieldArchive = fieldArchive;
	buttonNext->setEnabled(fieldArchive!=NULL);
	buttonPrev->setEnabled(fieldArchive!=NULL);
}

void Search::setFieldId(int fieldID)
{
//	if(fieldID != this->fieldID)
//	{
		this->fieldID = fieldID;
//		qDebug() << "FieldID=" << fieldID << "(Search::setFieldId)";
		this->textID = -1;
//		qDebug() << "textID=" << textID << "(Search::setFieldId::textID=0)";
		this->from = -1;
//	}
}

void Search::setTextId(int textID)
{
//	if(textID != this->textID)
//	{
		this->textID = textID;
//		qDebug() << "textID=" << textID << "(Search::setTextId)";
		this->from = -1;
//	}
}

void Search::setFrom(int from)
{
	this->from = from;
}

void Search::setSearchText(const QString &text)
{
	if(text.isEmpty())	return;
	searchTextField->setText(text);
	searchTextField->selectAll();
	searchScriptTextField->setText(text);
	searchScriptTextField->selectAll();
}

void Search::setSearchOpcode(int opcode)
{
	searchOpcode->setCurrentIndex(opcode);
}

void Search::findNext()
{
	bool found = false;
	if(fieldArchive==NULL)	return;

	buttonNext->setEnabled(false);
	buttonNext->setDefault(true);

	if(currentIndex() == Text)
	{
		found = findNextText();
	}
	else if(currentIndex() == Script)
	{
		found = findNextScript();
	}

	if(!found)
	{
		QMessageBox::information(this, windowTitle(), tr("Dernier fichier,\npoursuite de la recherche dans le premier fichier."));
	}

	buttonNext->setEnabled(true);
}

void Search::findPrev()
{
	bool found = false;
	if(fieldArchive==NULL)	return;

	buttonPrev->setEnabled(false);
	buttonPrev->setDefault(true);

	if(currentIndex() == Text)
	{
		found = findPrevText();
	}
	else if(currentIndex() == Script)
	{
		found = findPrevScript();
	}

	if(!found)
	{
		QMessageBox::information(this, windowTitle(), tr("Premier fichier,\npoursuite de la recherche dans le dernier fichier."));
	}

	buttonPrev->setEnabled(true);
}

bool Search::findNextText()
{
	int size;
	FieldArchive::Sorting sort = sorting();

	++from;

	if(fieldArchive->searchText(regexp(), fieldID, textID, from, size, sort))
	{
//		qDebug() << "from=" << from << "(MsdFile::findNextText::from=index)";
		emit foundText(fieldID, textID, from, size);
		// "found" signal may change 'from' value
//		qDebug() << "from=" << from << "(MsdFile::findNextText::from++)";
		return true;
	}

	textID = from = fieldID = -1;
//	qDebug() << "FieldID=" << fieldID << "(Search::findNextText::0)";
//	qDebug() << "textID=" << textID << "(Search::findNextText::0)";
//	qDebug() << "from=" << from << "(MsdFile::findNextText::0)";

	return false;
}

bool Search::findPrevText()
{
	int index, size;
	FieldArchive::Sorting sort = sorting();

	--from;

	if(fieldArchive->searchTextReverse(regexp(), fieldID, textID, from, index, size, sort))
	{
//		qDebug() << "from=" << from << "(Search::findPrevText::from=index)";
		emit foundText(fieldID, textID, index, size);
		return true;
	}

	fieldID = textID = 2147483647;
	from = 0;
//	qDebug() << "FieldID=" << fieldID << "(Search::findPrevText::0)";
//	qDebug() << "textID=" << textID << "(Search::findPrevText::0)";
//	qDebug() << "from=" << from << "(Search::findPrevText::0)";

	return false;
}

bool Search::findNextScript()
{
	int sav;
	bool found = false;
	FieldArchive::Sorting sort = sorting();

	if(typeScriptChoice->currentIndex() == 0) {
		found = fieldArchive->searchScriptText(regexp(), fieldID, groupID, methodID, opcodeID, sort);
	}
	else if(typeScriptChoice->currentIndex() == 1) {
		found = fieldArchive->searchScript(1, searchOpcode->currentIndex() | (searchOpcodeValue->value() << 16), fieldID, groupID, methodID, opcodeID, sort);
	}
	else if(typeScriptChoice->currentIndex() == 2) {
		found = fieldArchive->searchScript(2, (popScriptVar->isChecked() << 31) | (selectScriptVar->currentIndex() & 0x7FFFFFFF), fieldID, groupID, methodID, opcodeID, sort);
	}
	else if(typeScriptChoice->currentIndex() == 3) {
		Field *field = fieldArchive->getField(fieldID);
		if(field != NULL && field->hasJsmFile()) {
			found = field->getJsmFile()->search(3, (selectScriptGroup->value() & 0xFFFF) | ((selectScriptLabel->value() & 0xFFFF) << 16), groupID, methodID, opcodeID);
		}
		if(!found) {
			groupID = methodID = opcodeID = 0;
			return false;
		}
	}

	if(found)
	{
		sav = opcodeID;
		emit foundOpcode(fieldID, groupID, methodID, opcodeID);
		// "found" signal may change 'opcodeID' value
		opcodeID = sav + 1;
		return true;
	}

	groupID = methodID = opcodeID = 0;
	fieldID = -1;

	return false;
}

bool Search::findPrevScript()
{
	int sav;
	bool found = false;
	FieldArchive::Sorting sort = sorting();

	if(typeScriptChoice->currentIndex() == 0) {
		found = fieldArchive->searchScriptTextReverse(regexp(), fieldID, groupID, methodID, opcodeID, sort);
	}
	else if(typeScriptChoice->currentIndex() == 1) {
		found = fieldArchive->searchScriptReverse(1, searchOpcode->currentIndex(), fieldID, groupID, methodID, opcodeID, sort);
	}
	else if(typeScriptChoice->currentIndex() == 2) {
		found = fieldArchive->searchScriptReverse(2, (popScriptVar->isChecked() << 31) | (selectScriptVar->currentIndex() & 0x7FFFFFFF), fieldID, groupID, methodID, opcodeID, sort);
	}
	else if(typeScriptChoice->currentIndex() == 3) {
		found = fieldArchive->searchScriptReverse(3, selectScriptGroup->value() | (selectScriptLabel->value() << 16), fieldID, groupID, methodID, opcodeID, sort);
	}

	if(found)
	{
		sav = opcodeID;
		emit foundOpcode(fieldID, groupID, methodID, opcodeID);
		// "found" signal may change 'opcodeID' value
		opcodeID = sav - 1;
		return true;
	}

	fieldID = groupID = methodID = opcodeID = -1;

	return false;
}