#pragma once

#include <cstdint>
#include <vector>
#include <string>

#include <Ram.h>
#include <Bus.h>
#include <Register.h>
#include <UnusedMemory.h>

#include "LR35902.h"
#include "IoRegisters.h"

namespace EightBit {
	namespace GameBoy {
		class Bus : public EightBit::Bus {
		public:

			enum CartridgeType {
				ROM = 0,
				ROM_MBC1 = 1,
				ROM_MBC1_RAM = 2,
				ROM_MBC1_RAM_BATTERY = 3,
			};

			enum {
				CyclesPerSecond = 4 * 1024 * 1024,
				FramesPerSecond = 60,
				CyclesPerFrame = CyclesPerSecond / FramesPerSecond,
				TotalLineCount = 154,
				CyclesPerLine = CyclesPerFrame / TotalLineCount,
				RomPageSize = 0x4000
			};

			Bus() noexcept;

			auto& CPU() { return m_cpu; }
			auto& VRAM() { return m_videoRam; }
			auto& OAMRAM() { return m_oamRam; }
			auto& IO() { return m_ioPorts; }

			void reset();

			void disableGameRom() { m_disableGameRom = true; }
			void enableGameRom() { m_disableGameRom = false; }

			bool gameRomDisabled() const { return m_disableGameRom; }
			bool gameRomEnabled() const { return !gameRomDisabled(); }

			void loadBootRom(const std::string& path);
			void loadGameRom(const std::string& path);

			int runRasterLines();
			int runVerticalBlankLines();

		protected:
			virtual MemoryMapping mapping(uint16_t address) override;

		private:
			LR35902 m_cpu;

			Rom m_bootRom = 0x100;							// 0x0000 - 0x00ff
			std::vector<Rom> m_gameRomBanks;				// 0x0000 - 0x3fff, 0x4000 - 0x7fff (switchable)
			Ram m_videoRam = 0x2000;						// 0x8000 - 0x9fff
			std::vector<Ram> m_ramBanks;					// 0xa000 - 0xbfff (switchable)
			UnusedMemory m_unmapped2000 = { 0x2000, 0xff };	// 0xa000 - 0xbfff
			Ram m_lowInternalRam = 0x2000;					// 0xc000 - 0xdfff (mirrored at 0xe000)
			Ram m_oamRam = 0xa0;							// 0xfe00 - 0xfe9f
			UnusedMemory m_unmapped60 = { 0x60, 0xff };		// 0xfea0 - 0xfeff
			IoRegisters m_ioPorts;							// 0xff00 - 0xff7f
			Ram m_highInternalRam = 0x80;					// 0xff80 - 0xffff

			bool m_enabledLCD = false;

			bool m_disableGameRom = false;

			bool m_rom = false;
			bool m_banked = false;
			bool m_ram = false;
			bool m_battery = false;

			bool m_higherRomBank = true;
			bool m_ramBankSwitching = false;

			int m_romBank = 1;
			int m_ramBank = 0;

			void validateCartridgeType();

			void Bus_WrittenByte(EightBit::EventArgs);

			int runRasterLines(int lines);
			int runVerticalBlankLines(int lines);
			int runRasterLine(int limit);
		};
	}
}