// (C) 2018 by Folkert van Heusden
// Released under Apache License v2.0
#include <errno.h>
#include <string.h>

#include "bus.h"
#include "cpu.h"
#include "error.h"
#if defined(ESP32)
#include "esp32.h"
#endif
#include "gen.h"
#include "rk05.h"
#include "utils.h"


const char * const regnames[] = { 
	"RK05_DS drivestatus",
	"RK05_ERROR        ",
	"RK05_CS ctrlstatus",
	"RK05_WC word count",
	"RK05_BA busaddress",
	"RK05_DA disk addrs",
	"RK05_DATABUF      "
	};

rk05::rk05(const std::string & file, bus *const b) : b(b)
{
	memset(registers, 0x00, sizeof registers);
	memset(xfer_buffer, 0x00, sizeof xfer_buffer);

#if defined(ESP32)
	Serial.print(F("MISO: "));
	Serial.println(int(MISO));
	Serial.print(F("MOSI: "));
	Serial.println(int(MOSI));
	Serial.print(F("SCK : "));
	Serial.println(int(SCK));
	Serial.print(F("SS  : "));
	Serial.println(int(SS));

	Serial.println(F("Files on SD-card:"));

	if (!sd.begin(SS, SD_SCK_MHZ(15)))
		sd.initErrorHalt();

	sd.ls("/", LS_DATE | LS_SIZE | LS_R);

	while(Serial.available())
		Serial.read();

	std::string selected_file = read_terminal_line("Enter filename: ");

	Serial.print(F("Opening file: "));
	Serial.println(selected_file.c_str());

	if (!fh.open(selected_file.c_str(), O_RDWR))
		sd.errorHalt(F("rk05: open failed"));
#else
	fh = fopen(file.c_str(), "rb");
	if (!fh)
		error_exit(true, "rk05: cannot open \"%s\"", file.c_str());
#endif
}

rk05::~rk05()
{
#if defined(ESP32)
	fh.close();
#else
	fclose(fh);
#endif
}

uint8_t rk05::readByte(const uint16_t addr)
{
	uint16_t v = readWord(addr & ~1);

	if (addr & 1)
		return v >> 8;

	return v;
}

uint16_t rk05::readWord(const uint16_t addr)
{
	const int reg = (addr - RK05_BASE) / 2;

	D(fprintf(stderr, "RK05 read %s/%o: ", reg[regnames], addr);)

	if (addr == RK05_DS) {		// 0177400
		setBit(registers[reg], 11, true); // disk on-line
		setBit(registers[reg], 8, true); // sector ok
		setBit(registers[reg], 7, true); // drive ready
		setBit(registers[reg], 6, true); // seek ready
		setBit(registers[reg], 4, true); // heads in position
	}
	else if (addr == RK05_ERROR)	// 0177402
		registers[reg] = 0;
	else if (addr == RK05_CS) {	// 0177404
		setBit(registers[reg], 15, false); // clear error
		setBit(registers[reg], 14, false); // clear hard error
		setBit(registers[reg], 7, true); // controller ready
	}

	uint16_t vtemp = registers[reg];

	if (addr == RK05_CS)
		setBit(registers[reg], 0, false); // clear go

	return vtemp;
}

void rk05::writeByte(const uint16_t addr, const uint8_t v)
{
	uint16_t vtemp = registers[(addr - RK05_BASE) / 2];

	if (addr & 1) {
		vtemp &= ~0xff00;
		vtemp |= v << 8;
	}
	else {
		vtemp &= ~0x00ff;
		vtemp |= v;
	}

	writeWord(addr, vtemp);
}

void rk05::writeWord(const uint16_t addr, uint16_t v)
{
#if defined(ESP32)
	digitalWrite(LED_BUILTIN, LOW);
#endif

	const int reg = (addr - RK05_BASE) / 2;
	D(fprintf(stderr, "RK05 write %s/%o: %o\n", regnames[reg], addr, v);)

	D(fprintf(stderr, "set register %o to %o\n", addr, v);)
	registers[reg] = v;

	if (addr == RK05_CS) {
		if (v & 1) { // GO
			const int func = (v >> 1) & 7; // FUNCTION
			int16_t wc = registers[(RK05_WC - RK05_BASE) / 2];
			const size_t reclen = wc < 0 ? (-wc * 2) : wc * 2;
			D(fprintf(stderr, "RK05 rec len %zd\n", reclen);)

			uint16_t dummy = registers[(RK05_DA - RK05_BASE) / 2];
			uint8_t sector = dummy & 0b1111;
			uint8_t surface = (dummy >> 4) & 1;
			int track = (dummy >> 4) & 511;
			uint16_t cylinder = (dummy >> 5) & 255;
			D(fprintf(stderr, "RK05 position sec %d surf %d cyl %d\n", sector, surface, cylinder);)

			const int diskoff = track * 12 + sector;

			const int diskoffb = diskoff * 512; // RK05 is high density
			const uint16_t memoff = registers[(RK05_BA - RK05_BASE) / 2];

			D(fprintf(stderr, "invoke %d\n", func);)

			if (func == 0) { // controller reset
			}
			else if (func == 1) { // write
				D(fprintf(stderr, "RK05 writing %zo bytes to offset %o (%d dec)\n", reclen, diskoffb, diskoffb);)

				uint32_t p = reclen; // FIXME
				for(size_t i=0; i<reclen; i++)
					xfer_buffer[i] = b -> readByte(memoff + i);

#if defined(ESP32)
				if (!fh.seek(diskoffb))
					fprintf(stderr, "RK05 seek error %s\n", strerror(errno));
				if (fh.write(xfer_buffer, reclen) != reclen)
                                        fprintf(stderr, "RK05 fwrite error %s\n", strerror(errno));
#else
				if (fseek(fh, diskoffb, SEEK_SET) == -1)
					fprintf(stderr, "RK05 seek error %s\n", strerror(errno));
				if (fwrite(xfer_buffer, 1, reclen, fh) != reclen)
					fprintf(stderr, "RK05 fwrite error %s\n", strerror(errno));
#endif

				if (v & 2048)
					D(fprintf(stderr, "RK05 inhibit BA increase\n");)
				else
					registers[(RK05_BA - RK05_BASE) / 2] += p;

				if (++sector >= 12) {
					sector = 0;
					if (++surface >= 2) {
						surface = 0;
						cylinder++;
					}
				}

				registers[(RK05_DA - RK05_BASE) / 2] = sector | (surface << 4) | (cylinder << 5);
			}
			else if (func == 2) { // read
				D(fprintf(stderr, "RK05 reading %zo bytes from offset %o (%d dec) to %o\n", reclen, diskoffb, diskoffb, memoff);)

#if defined(ESP32)
				if (!fh.seek(diskoffb))
					fprintf(stderr, "RK05 seek error %s\n", strerror(errno));
#else
				if (fseek(fh, diskoffb, SEEK_SET) == -1)
					fprintf(stderr, "RK05 seek error %s\n", strerror(errno));
#endif
				
				uint32_t temp = reclen;
				uint32_t p = memoff;
				while(temp > 0) {
					uint32_t cur = std::min(uint32_t(sizeof xfer_buffer), temp);

#if defined(ESP32)
					yield();

					if (fh.read(xfer_buffer, cur) != size_t(cur))
						D(fprintf(stderr, "RK05 fread error: %s\n", strerror(errno));)
#else
					if (fread(xfer_buffer, 1, cur, fh) != size_t(cur))
						D(fprintf(stderr, "RK05 fread error: %s\n", strerror(errno));)
#endif

					for(uint32_t i=0; i<cur; i++) {
						if (p < 0160000)
							b -> writeByte(p, xfer_buffer[i]);
						p++;
					}

					temp -= cur;
				}

				if (v & 2048)
					D(fprintf(stderr, "RK05 inhibit BA increase\n");)
				else
					registers[(RK05_BA - RK05_BASE) / 2] += p;

				if (++sector >= 12) {
					sector = 0;
					if (++surface >= 2) {
						surface = 0;
						cylinder++;
					}
				}

				registers[(RK05_DA - RK05_BASE) / 2] = sector | (surface << 4) | (cylinder << 5);
			}
			else if (func == 4) {
				D(fprintf(stderr, "RK05 seek to offset %o\n", diskoffb);)
			}
			else if (func == 7)
				D(fprintf(stderr, "RK05 write lock\n");)
			else {
				D(fprintf(stderr, "RK05 command %d UNHANDLED\n", func);)
			}

			registers[(RK05_WC - RK05_BASE) / 2] = 0;

			registers[(RK05_DS - RK05_BASE) / 2] |= 64;  // drive ready
			registers[(RK05_CS - RK05_BASE) / 2] |= 128;  // control ready

			if (v & 64) { // bit 6, invoke interrupt when done vector address 220, see http://www.pdp-11.nl/peripherals/disk/rk05-info.html
				b->getCpu()->trap(0220);
			}
		}
	}

#if defined(ESP32)
	digitalWrite(LED_BUILTIN, HIGH);
#endif
}