/* Copyright (c) 2013-2017 Jeffrey Pfau
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "MemorySearch.h"

#include <mgba/core/core.h>

#include "CoreController.h"
#include "MemoryView.h"

using namespace QGBA;

MemorySearch::MemorySearch(std::shared_ptr<CoreController> controller, QWidget* parent)
	: QWidget(parent)
	, m_controller(controller)
{
	m_ui.setupUi(this);

	mCoreMemorySearchResultsInit(&m_results, 0);
	connect(m_ui.search, &QPushButton::clicked, this, &MemorySearch::search);
	connect(m_ui.searchWithin, &QPushButton::clicked, this, &MemorySearch::searchWithin);
	connect(m_ui.refresh, &QPushButton::clicked, this, &MemorySearch::refresh);
	connect(m_ui.numHex, &QPushButton::clicked, this, &MemorySearch::refresh);
	connect(m_ui.numDec, &QPushButton::clicked, this, &MemorySearch::refresh);
	connect(m_ui.viewMem, &QPushButton::clicked, this, &MemorySearch::openMemory);

	connect(controller.get(), &CoreController::stopping, this, &QWidget::close);
}

MemorySearch::~MemorySearch() {
	mCoreMemorySearchResultsDeinit(&m_results);
}

bool MemorySearch::createParams(mCoreMemorySearchParams* params) {
	params->memoryFlags = mCORE_MEMORY_RW;
	mCore* core = m_controller->thread()->core;

	QByteArray string;
	bool ok = false;
	if (m_ui.typeNum->isChecked()) {
		if (m_ui.bits8->isChecked()) {
			params->type = mCORE_MEMORY_SEARCH_8;
		}
		if (m_ui.bits16->isChecked()) {
			params->type = mCORE_MEMORY_SEARCH_16;
		}
		if (m_ui.bits32->isChecked()) {
			params->type = mCORE_MEMORY_SEARCH_32;
		}
		if (m_ui.numHex->isChecked()) {
			uint32_t v = m_ui.value->text().toUInt(&ok, 16);
			if (ok) {
				switch (params->type) {
				case mCORE_MEMORY_SEARCH_8:
					ok = v < 0x100;
					params->value8 = v;
					break;
				case mCORE_MEMORY_SEARCH_16:
					ok = v < 0x10000;
					params->value16 = v;
					break;
				case mCORE_MEMORY_SEARCH_32:
					params->value32 = v;
					break;
				default:
					ok = false;
				}
			}
		}
		if (m_ui.numDec->isChecked()) {
			uint32_t v = m_ui.value->text().toUInt(&ok, 10);
			if (ok) {
				switch (params->type) {
				case mCORE_MEMORY_SEARCH_8:
					ok = v < 0x100;
					params->value8 = v;
					break;
				case mCORE_MEMORY_SEARCH_16:
					ok = v < 0x10000;
					params->value16 = v;
					break;
				case mCORE_MEMORY_SEARCH_32:
					params->value32 = v;
					break;
				default:
					ok = false;
				}
			}
		}
		if (m_ui.numGuess->isChecked()) {
			params->type = mCORE_MEMORY_SEARCH_GUESS;
			m_string = m_ui.value->text().toLocal8Bit();
			params->valueStr = m_string.constData();
			ok = true;
		}
	}
	if (m_ui.typeStr->isChecked()) {
		params->type = mCORE_MEMORY_SEARCH_STRING;
		m_string = m_ui.value->text().toLocal8Bit();
		params->valueStr = m_string.constData();
		ok = true;
	}
	return ok;
}

void MemorySearch::search() {
	mCoreMemorySearchResultsClear(&m_results);

	mCoreMemorySearchParams params;

	CoreController::Interrupter interrupter(m_controller);
	mCore* core = m_controller->thread()->core;

	if (createParams(&params)) {
		mCoreMemorySearch(core, &params, &m_results, LIMIT);
	}

	refresh();
}

void MemorySearch::searchWithin() {
	mCoreMemorySearchParams params;

	CoreController::Interrupter interrupter(m_controller);
	mCore* core = m_controller->thread()->core;

	if (createParams(&params)) {
		mCoreMemorySearchRepeat(core, &params, &m_results);
	}

	refresh();
}

void MemorySearch::refresh() {
	CoreController::Interrupter interrupter(m_controller);
	mCore* core = m_controller->thread()->core;

	m_ui.results->clearContents();
	m_ui.results->setRowCount(mCoreMemorySearchResultsSize(&m_results));
	for (size_t i = 0; i < mCoreMemorySearchResultsSize(&m_results); ++i) {
		mCoreMemorySearchResult* result = mCoreMemorySearchResultsGetPointer(&m_results, i);
		QTableWidgetItem* item = new QTableWidgetItem(QString("%1").arg(result->address, 8, 16, QChar('0')));
		m_ui.results->setItem(i, 0, item);
		QTableWidgetItem* type;
		if (m_ui.numHex->isChecked()) {
			switch (result->type) {
			case mCORE_MEMORY_SEARCH_8:
				item = new QTableWidgetItem(QString("%1").arg(core->rawRead8(core, result->address, result->segment), 2, 16, QChar('0')));
				break;
			case mCORE_MEMORY_SEARCH_16:
				item = new QTableWidgetItem(QString("%1").arg(core->rawRead16(core, result->address, result->segment), 4, 16, QChar('0')));
				break;
			case mCORE_MEMORY_SEARCH_GUESS:
			case mCORE_MEMORY_SEARCH_32:
				item = new QTableWidgetItem(QString("%1").arg(core->rawRead32(core, result->address, result->segment), 8, 16, QChar('0')));
				break;
			case mCORE_MEMORY_SEARCH_STRING:
				item = new QTableWidgetItem("?"); // TODO
			}
		} else {
			switch (result->type) {
			case mCORE_MEMORY_SEARCH_8:
				item = new QTableWidgetItem(QString::number(core->rawRead8(core, result->address, result->segment)));
				break;
			case mCORE_MEMORY_SEARCH_16:
				item = new QTableWidgetItem(QString::number(core->rawRead16(core, result->address, result->segment)));
				break;
			case mCORE_MEMORY_SEARCH_GUESS:
			case mCORE_MEMORY_SEARCH_32:
				item = new QTableWidgetItem(QString::number(core->rawRead32(core, result->address, result->segment)));
				break;
			case mCORE_MEMORY_SEARCH_STRING:
				item = new QTableWidgetItem("?"); // TODO
			}
		}
		QString divisor;
		if (result->guessDivisor > 1) {
			divisor = tr(" (⅟%0×)").arg(result->guessDivisor);
		}
		switch (result->type) {
		case mCORE_MEMORY_SEARCH_8:
			type = new QTableWidgetItem(tr("1 byte%0").arg(divisor));
			break;
		case mCORE_MEMORY_SEARCH_16:
			type = new QTableWidgetItem(tr("2 bytes%0").arg(divisor));
			break;
		case mCORE_MEMORY_SEARCH_GUESS:
		case mCORE_MEMORY_SEARCH_32:
			type = new QTableWidgetItem(tr("4 bytes%0").arg(divisor));
			break;
		case mCORE_MEMORY_SEARCH_STRING:
			item = new QTableWidgetItem("?"); // TODO
		}
		m_ui.results->setItem(i, 1, item);
		m_ui.results->setItem(i, 2, type);
	}
	m_ui.results->sortItems(0);
	m_ui.results->resizeColumnsToContents();
	m_ui.results->resizeRowsToContents();
}

void MemorySearch::openMemory() {
	auto items = m_ui.results->selectedItems();
	if (items.empty()) {
		return;
	}
	QTableWidgetItem* item = items[0];
	uint32_t address = item->text().toUInt(nullptr, 16);

	MemoryView* memView = new MemoryView(m_controller);
	memView->jumpToAddress(address);

	memView->setAttribute(Qt::WA_DeleteOnClose);
	memView->show();
}