#include "hardware.h"

#include <SPI.h>

#if defined(SD_CS)
#include <SD.h>
#endif

#if defined(USE_SPIFFS)
#include <SPIFFS.h>
#endif

#include <UTFT.h>

#if defined(SPIRAM_CS)
#include <SpiRAM.h>
#include "spiram.h"
spiram sram(SPIRAM_SIZE);
#endif

#include "ps2drv.h"
#include "memory.h"
#include "CPU.h"

Memory memory;
PS2Driver ps2;

UTFT utft(TFT_MODEL, TFT_RS, TFT_WR, TFT_CS, TFT_RST);
static CPU *_cpu;

bool hardware_reset() {
	bool success = true;

#if defined(SPIRAM_CS)
	extern SPIClass SPIRAM_DEV;
	SPIRAM_DEV.begin();
	SPIRAM_DEV.setModule(SPIRAM_SPI);
	SPIRAM_DEV.setClockDivider(1);
	SPIRAM_DEV.setDataMode(SPI_MODE0);
#endif

#if defined(SD_CS)
	success = SD.begin(SD_CS, 2, SD_SPI);
	pinMode(SPI_CS, OUTPUT);	// without this, the SPI-RAM isn't seen
#endif

#if defined(USE_SPIFFS)
	success = SPIFFS.begin();
#endif

#if defined(TFT_BACKLIGHT)
	digitalWrite(TFT_BACKLIGHT, HIGH);
#endif

	_cpu->reset();
	return success;
}

void hardware_init(CPU &cpu) {
	_cpu = &cpu;
	memory.begin();
	ps2.begin(KBD_DATA, KBD_IRQ);

#if defined(TFT_BACKLIGHT)
	pinMode(TFT_BACKLIGHT, OUTPUT);
#endif

#if defined(SD_CS)
	pinMode(SD_CS, OUTPUT);
	digitalWrite(SD_CS, HIGH);
#endif

#if defined(SPIRAM_CS)
	pinMode(SPIRAM_CS, OUTPUT);
	digitalWrite(SPIRAM_CS, HIGH);
#endif
}

void hardware_checkpoint(Stream &s) {
	unsigned ds = 0;
	for (unsigned i = 0; i < 0x10000; i += ds) {
		Memory::Device *dev = memory.get(i);
		dev->checkpoint(s);
		ds = dev->pages() * Memory::page_size;
	}
	_cpu->checkpoint(s);
}

void hardware_restore(Stream &s) {
	unsigned ds = 0;
	for (unsigned i = 0; i < 0x10000; i += ds) {
		Memory::Device *dev = memory.get(i);
		dev->restore(s);
		ds = dev->pages() * Memory::page_size;
	}
	_cpu->restore(s);
}