#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "../WindCore/wind.h"
#include <QTimer>
#include <QKeyEvent>
#include "../WindCore/decoder.h"

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    emu = new Emu;
	emu->loadROM("/Users/ash/src/psion/Sys$rom.bin");

    timer = new QTimer(this);
    timer->setInterval(1000/64);
    connect(timer, SIGNAL(timeout()), SLOT(execTimer()));

	updateScreen();
}

MainWindow::~MainWindow()
{
    delete ui;
}

void MainWindow::updateScreen()
{
    ui->cycleCounter->setText(QString("Cycles: %1").arg(emu->currentCycles()));

	updateMemory();

    ui->regsLabel->setText(
				QString("R0: %1 / R1: %2 / R2: %3 / R3: %4 / R4: %5 / R5: %6 / R6: %7 / R7: %8\nR8: %9 / R9: %10 / R10:%11 / R11:%12 / R12:%13 / SP: %14 / LR: %15 / PC: %16")
                .arg(emu->getGPR(0), 8, 16)
                .arg(emu->getGPR(1), 8, 16)
                .arg(emu->getGPR(2), 8, 16)
                .arg(emu->getGPR(3), 8, 16)
                .arg(emu->getGPR(4), 8, 16)
                .arg(emu->getGPR(5), 8, 16)
                .arg(emu->getGPR(6), 8, 16)
                .arg(emu->getGPR(7), 8, 16)
                .arg(emu->getGPR(8), 8, 16)
                .arg(emu->getGPR(9), 8, 16)
                .arg(emu->getGPR(10), 8, 16)
                .arg(emu->getGPR(11), 8, 16)
                .arg(emu->getGPR(12), 8, 16)
                .arg(emu->getGPR(13), 8, 16)
                .arg(emu->getGPR(14), 8, 16)
                .arg(emu->getGPR(15), 8, 16)
                );

    // show a crude disassembly
    const int context = 8 * 4;
    uint32_t pc = emu->getGPR(15) - 4;
    uint32_t minCode = pc - context;
	if (minCode >= (UINT32_MAX - context))
        minCode = 0;
    uint32_t maxCode = pc + context;
	if (maxCode < context)
        maxCode = UINT32_MAX;

	QStringList codeLines;
	for (uint32_t addr = minCode; addr >= minCode && addr <= maxCode; addr += 4) {
		const char *prefix = (addr == pc) ? "==>" : "   ";
        struct ARMInstructionInfo info;
        char buffer[512];

		auto result = emu->readVirtual(addr, ARM710a::V32);
		if (result.first.has_value()) {
			uint32_t opcode = result.first.value();
			ARMDecodeARM(opcode, &info);
			ARMDisassemble(&info, addr, buffer, sizeof(buffer));
			codeLines.append(QString("%1 %2 | %3 | %4").arg(prefix).arg(addr, 8, 16).arg(opcode, 8, 16).arg(buffer));
		}
    }
	ui->codeLabel->setText(codeLines.join('\n'));

    // now, the actual screen
    const uint8_t *lcdBuf = emu->getLCDBuffer();
    if (lcdBuf) {
        QImage img(640, 240, QImage::Format_Grayscale8);

        // fetch palette
        int bpp = 1 << (lcdBuf[1] >> 4);
        int ppb = 8 / bpp;
        uint16_t palette[16];
        for (int i = 0; i < 16; i++)
            palette[i] = lcdBuf[i*2] | ((lcdBuf[i*2+1] << 8) & 0xF00);

        // build our image out
        int lineWidth = (img.width() * bpp) / 8;
        for (int y = 0; y < img.height(); y++) {
            uint8_t *scanline = img.scanLine(y);
            int lineOffs = 0x20 + (lineWidth * y);
            for (int x = 0; x < img.width(); x++) {
                uint8_t byte = lcdBuf[lineOffs + (x / ppb)];
                int shift = (x & (ppb - 1)) * bpp;
				int mask = (1 << bpp) - 1;
                int palIdx = (byte >> shift) & mask;
                int palValue = palette[palIdx];

				palValue |= (palValue << 4);
				scanline[x] = palValue ^ 0xFF;
            }
        }

        ui->screen->setPixmap(QPixmap::fromImage(std::move(img)));
    }
}


static int resolveKey(int key) {
    switch (key) {
    case Qt::Key_6: return 0;
    case Qt::Key_5: return 1;
    case Qt::Key_4: return 2;
    case Qt::Key_3: return 3;
    case Qt::Key_2: return 4;
    case Qt::Key_1: return 5;
    // missing 6: F13/rec

    case Qt::Key_Apostrophe: return 7;
    case Qt::Key_Backspace: return 8;
    case Qt::Key_0: return 9;
    case Qt::Key_9: return 10;
    case Qt::Key_8: return 11;
    case Qt::Key_7: return 12;
    // missing 13: F15/play

    case Qt::Key_Y: return 14;
    case Qt::Key_T: return 15;
    case Qt::Key_R: return 16;
    case Qt::Key_E: return 17;
    case Qt::Key_W: return 18;
    case Qt::Key_Q: return 19;
    case Qt::Key_Escape: return 20;

    case Qt::Key_Enter: return 21;
    case Qt::Key_Return: return 21;
    case Qt::Key_L: return 22;
    case Qt::Key_P: return 23;
    case Qt::Key_O: return 24;
    case Qt::Key_I: return 25;
    case Qt::Key_U: return 26;
    case Qt::Key_Alt: return 27; // actually Menu

    case Qt::Key_G: return 28;
    case Qt::Key_F: return 29;
    case Qt::Key_D: return 30;
    case Qt::Key_S: return 31;
    case Qt::Key_A: return 32;
    case Qt::Key_Tab: return 33;
#ifdef Q_OS_MAC
    case Qt::Key_Meta: return 34; // Control -> Control
#else
    case Qt::Key_Control: return 34; // Control -> Control
#endif

    case Qt::Key_Down: return 35;
    case Qt::Key_Period: return 36;
    case Qt::Key_M: return 37;
    case Qt::Key_K: return 38;
    case Qt::Key_J: return 39;
    case Qt::Key_H: return 40;
#ifdef Q_OS_MAC
    case Qt::Key_Control: return 41; // Command -> Fn
#else
    case Qt::Key_Meta: return 41; // Super -> Fn
#endif

    case Qt::Key_N: return 42;
    case Qt::Key_B: return 43;
    case Qt::Key_V: return 44;
    case Qt::Key_C: return 45;
    case Qt::Key_X: return 46;
    case Qt::Key_Z: return 47;
    case Qt::Key_Shift: return 48;

    case Qt::Key_Right: return 49;
    case Qt::Key_Left: return 50;
    case Qt::Key_Comma: return 51;
    case Qt::Key_Up: return 52;
    case Qt::Key_Space: return 53;
    // missing 54: F14/stop
    // missing 55: another Shift
    }
    return -1;
}


void MainWindow::keyPressEvent(QKeyEvent *event)
{
    int k = resolveKey(event->key());
    if (k >= 0)
        emu->keyboardKeys[k] = true;
}

void MainWindow::keyReleaseEvent(QKeyEvent *event)
{
    int k = resolveKey(event->key());
	if (k >= 0)
        emu->keyboardKeys[k] = false;
}




void MainWindow::on_startButton_clicked()
{
    timer->start();
    ui->startButton->setEnabled(false);
    ui->stopButton->setEnabled(true);
    ui->stepInsnButton->setEnabled(false);
    ui->stepTickButton->setEnabled(false);
}

void MainWindow::on_stopButton_clicked()
{
    timer->stop();
    ui->startButton->setEnabled(true);
    ui->stopButton->setEnabled(false);
    ui->stepInsnButton->setEnabled(true);
    ui->stepTickButton->setEnabled(true);
}

void MainWindow::on_stepTickButton_clicked()
{
    emu->executeUntil(emu->currentCycles() + (CLOCK_SPEED * 2));
    updateScreen();
}

void MainWindow::on_stepInsnButton_clicked()
{
    emu->executeUntil(emu->currentCycles() + 1);
    updateScreen();
}

void MainWindow::execTimer()
{
    emu->executeUntil(emu->currentCycles() + (CLOCK_SPEED / 64));
    updateScreen();
}

void MainWindow::on_addBreakButton_clicked()
{
	uint32_t addr = ui->breakpointAddress->text().toUInt(nullptr, 16);
	emu->breakpoints().insert(addr);
	updateBreakpointsList();
}

void MainWindow::on_removeBreakButton_clicked()
{
	uint32_t addr = ui->breakpointAddress->text().toUInt(nullptr, 16);
	emu->breakpoints().erase(addr);
	updateBreakpointsList();
}

void MainWindow::updateBreakpointsList()
{
	ui->breakpointsList->clear();
	for (uint32_t addr : emu->breakpoints()) {
		ui->breakpointsList->addItem(QString::number(addr, 16));
	}
}

void MainWindow::on_memoryViewAddress_textEdited(const QString &)
{
	updateMemory();
}

void MainWindow::updateMemory()
{
	uint32_t virtBase = ui->memoryViewAddress->text().toUInt(nullptr, 16) & ~0xFF;
	auto physBaseOpt = emu->virtToPhys(virtBase);
	auto physBase = physBaseOpt.value_or(0xFFFFFFFF);
	bool ok = physBaseOpt.has_value() && emu->isPhysAddressValid(physBase);
	if (ok && (virtBase != physBase))
		ui->physicalAddressLabel->setText(QStringLiteral("Physical: %1").arg(physBase, 8, 16, QLatin1Char('0')));

	uint8_t block[0x100];
	if (ok) {
		for (int i = 0; i < 0x100; i++) {
			block[i] = emu->readPhysical(physBase + i, ARM710a::V8).value();
		}
	}

	QStringList output;
	for (int row = 0; row < 16; row++) {
		QString outLine;
		outLine.reserve(8 + 2 + (2 * 16) + 3 + 16);
		outLine.append(QStringLiteral("%1 |").arg(virtBase + (row * 16), 8, 16));
		for (int col = 0; col < 16; col++) {
			if (ok)
				outLine.append(QStringLiteral(" %1").arg(block[row*16+col], 2, 16, QLatin1Char('0')));
			else
				outLine.append(QStringLiteral(" ??"));
		}
		outLine.append(QStringLiteral(" | "));
		for (int col = 0; col < 16; col++) {
			uint8_t byte = block[row*16+col];
			if (!ok)
				outLine.append('?');
			else if (byte >= 0x20 && byte <= 0x7E)
				outLine.append(byte);
			else
				outLine.append('.');
		}
		output.append(outLine);
	}

	ui->memoryViewLabel->setText(output.join('\n'));
}

void MainWindow::on_memoryAdd1_clicked() { adjustMemoryAddress(1); }
void MainWindow::on_memoryAdd4_clicked() { adjustMemoryAddress(4); }
void MainWindow::on_memoryAdd10_clicked() { adjustMemoryAddress(0x10); }
void MainWindow::on_memoryAdd100_clicked() { adjustMemoryAddress(0x100); }
void MainWindow::on_memorySub1_clicked() { adjustMemoryAddress(-1); }
void MainWindow::on_memorySub4_clicked() { adjustMemoryAddress(-4); }
void MainWindow::on_memorySub10_clicked() { adjustMemoryAddress(-0x10); }
void MainWindow::on_memorySub100_clicked() { adjustMemoryAddress(-0x100); }

void MainWindow::adjustMemoryAddress(int offset) {
	uint32_t address = ui->memoryViewAddress->text().toUInt(nullptr, 16);
	address += offset;
	ui->memoryViewAddress->setText(QString("%1").arg(address, 8, 16, QLatin1Char('0')));
	updateMemory();
}

void MainWindow::on_writeByteButton_clicked()
{
	uint32_t address = ui->memoryViewAddress->text().toUInt(nullptr, 16);
	uint8_t value = (uint8_t)ui->memoryWriteValue->text().toUInt(nullptr, 16);
	emu->writeVirtual(value, address, ARM710a::V8);
	updateMemory();
}

void MainWindow::on_writeDwordButton_clicked()
{
	uint32_t address = ui->memoryViewAddress->text().toUInt(nullptr, 16);
	uint32_t value = ui->memoryWriteValue->text().toUInt(nullptr, 16);
	emu->writeVirtual(value, address, ARM710a::V32);
	updateMemory();
}