/*
 *  C64_x.i - Put the pieces together, X specific stuff
 *
 *  Frodo (C) 1994-1997,2002 Christian Bauer
 *  Unix stuff by Bernd Schmidt/Lutz Vieweg
 */

#include "main.h"

#include <sys/types.h>
#include <dirent.h>

#if defined(GEKKO)
#include <wiiuse/wpad.h>
#include <ogc/lwp_watchdog.h>
#define SAVES_PATH "/apps/frodo/saves"
#define IMAGE_PATH "/apps/frodo/images"
#define TMP_PATH "/apps/frodo/tmp"
#else
#define SAVES_PATH "saves"
#define IMAGE_PATH "images"
#define TMP_PATH "tmp"
#endif

#include "menutexts.h"

/* TODO: */
extern char *fixme_tmp_network_client;
extern char *fixme_tmp_network_server;


static struct timeval tv_start;

/*
 *  Constructor, system-dependent things
 */
void C64::c64_ctor1(void)
{
	// Initialize joystick variables
#ifdef HAVE_LINUX_JOYSTICK_H
	joyfd[0] = joyfd[1] = -1;
	joy_minx = joy_miny = 32767;
	joy_maxx = joy_maxy = -32768;
#endif

	this->fake_key_sequence = false;
	this->fake_key_index = 0;
	this->fake_key_keytime = 4;
        this->fake_key_str = "\nLOAD \"*\",8,1\nRUN\n";

	this->prefs_changed = false;
	memset(this->save_game_name, 0, sizeof(this->save_game_name));
	strcpy(this->save_game_name, "unknown");

	this->virtual_keyboard = new VirtualKeyboard(real_screen, this->menu_font);
	virtual_keyboard = this->virtual_keyboard;

	strncpy(this->server_hostname, "c64-network.game-host.org",
			sizeof(this->server_hostname));
	this->server_port = 46214;
	this->network_connection_type = NONE;
	this->peer = NULL;

	if (fixme_tmp_network_server) {
		strcpy(this->server_hostname, fixme_tmp_network_server);
		this->peer = new Network(this->server_hostname, this->server_port, true);
		this->network_connection_type = MASTER;
		printf("Waiting for connection\n");
		if (this->peer->Connect() == false)
		{
			printf("No client connected. Bye\n");
			delete this->peer;
			this->peer = NULL;
		}
	}
	if (fixme_tmp_network_client)
	{
		strcpy(this->server_hostname, fixme_tmp_network_client);
		this->peer = new Network(this->server_hostname, this->server_port, false);
		this->network_connection_type = CLIENT;
		if (this->peer->Connect() == false)
		{
			printf("Could not connect to server. Bye\n");
			delete this->peer;
			this->peer = NULL;
		}
	}
}

void C64::c64_ctor2(void)
{
	gettimeofday(&tv_start, NULL);
}


/*
 *  Destructor, system-dependent things
 */

void C64::c64_dtor(void)
{
}

void C64::select_disc(Prefs *np, bool start)
{
	const char *name;
	const char *d64_name = NULL;

	if (!start)
		name = menu_select_file(IMAGE_PATH);
	else
		name = menu_select_file_start(IMAGE_PATH, &d64_name);

	if (name== NULL)
		return;

	if (strcmp(name, "None") == 0)
	{
		strcpy(np->DrivePath[0], "\0");
		strcpy(this->save_game_name, "unknown");
	}
	else
	{
		strncpy(np->DrivePath[0], name, 255);
		strncpy(this->save_game_name, name, 255);
		if (strstr(name, ".prg") || strstr(name, ".PRG") ||
				strstr(name, ".p00") || strstr(name, ".P00") ||
				strstr(name, ".s00") || strstr(name, ".S00")) {
			FILE *src, *dst;

			/* Clean temp dir first (we only want one file) */
			unlink(TMP_PATH"/a");

			src = fopen(np->DrivePath[0], "r");
			if (src != NULL)
			{
				snprintf(np->DrivePath[0], 255, "%s", TMP_PATH);

				/* Special handling of .prg: Copy to TMP_PATH and
				 * load that as a dir */
				dst = fopen(TMP_PATH"/a", "w");
				if (dst)
				{
					Uint8 buf[1024];
					size_t v;

					do {
						v = fread(buf, 1, 1024, src);
						fwrite(buf, 1, v, dst);
					} while (v > 0);
					fclose(dst);
				}
				fclose(src);
			}
		}

		NewPrefs(np);
		ThePrefs = *np;
	}
	this->prefs_changed = true;

	/* Cleanup*/
	free((void*)name);

	/* And maybe start the game */
	if (d64_name)
	{
		static char buf[255];

		snprintf(buf, 255, "\nLOAD \"%s\",8,1\nRUN\n", d64_name);
		this->start_fake_key_sequence((const char*)buf);
		free((void*)d64_name);
	}
	else if (start)
		this->start_fake_key_sequence("\nLOAD \"*\",8,1\nRUN\n");
}


char *C64::bind_one_key(Prefs *np, int which)
{
	static const char *which_to_button_name[N_WIIMOTE_BINDINGS] = {
			"up", "down", "left", "right",
			"2", "1", "A", "B", "+", "-",
			"classic up", "classic down", "classic left", "classic right", "classic a",
			"classic b", "classic X", "classic Y", "classic L",
			"classic R", "classic ZR", "classic ZL" };
	static char strs[N_WIIMOTE_BINDINGS][255];
	char *out = strs[which];
	int cur = np->JoystickKeyBinding[which];

	snprintf(out, 255, "Bind to %s (now %s)", which_to_button_name[which],
			this->virtual_keyboard->keycode_to_string(cur));

	return out;
}

void C64::bind_keys(Prefs *np)
{
	const char *bind_key_messages[N_WIIMOTE_BINDINGS + 1];
	bool has_classic_controller = false;

#if defined(GEKKO)
        WPADData *wpad, *wpad_other;

        wpad = WPAD_Data(0);
        wpad_other = WPAD_Data(1);

        if (wpad->exp.type == WPAD_EXP_CLASSIC ||
        		wpad_other->exp.type == WPAD_EXP_CLASSIC)
        	has_classic_controller = true;
#endif

        memset(bind_key_messages, 0, sizeof(const char*) * (N_WIIMOTE_BINDINGS + 1));

        for (int i = 0; i < (has_classic_controller ? N_WIIMOTE_BINDINGS : CLASSIC_UP); i++)
        	bind_key_messages[i] = this->bind_one_key(np, i);

	int opt = menu_select(bind_key_messages, NULL);
	if (opt >= 0)
	{
	        int key;
	        bool shifted;

		key = this->virtual_keyboard->get_key();
		/* -2 means abort */
		if (key != np->JoystickKeyBinding[opt] && key != -2)
		{
			this->prefs_changed = true;
			np->JoystickKeyBinding[opt] = key;
		}
	}
}

void C64::networking_menu(Prefs *np)
{
	int opt;

	do
	{
		char buf[3][255];
		const char *network_client_messages[] = {
				buf[0],                  /* 0 */
				buf[1],                  /* 1 */
				buf[2],                  /* 2 */
				"Host a game",           /* 3 */
				"Connect as client",     /* 4 */
				NULL,
		};

		snprintf(buf[0], 255, "Set username (%s)",
				np->NetworkName);
		snprintf(buf[1], 255, "Server hostname (%s)",
				this->server_hostname);
		snprintf(buf[2], 255, "Port (%d)",
				this->server_port);
		opt = menu_select(network_client_messages, NULL);

		if (opt >= 0 && opt <= 2)
		{
			const char *m = this->virtual_keyboard->get_string();

			if (m)
			{
				if (opt == 0)
				{
					memset(np->NetworkName, 0,
							sizeof(np->NetworkName));
					strncpy(np->NetworkName, m,
							sizeof(np->NetworkName));
				} else if (opt == 1)
					strncpy(this->server_hostname, m,
							sizeof(this->server_hostname));
				if (opt == 2)
					this->server_port = atoi(m);
			}
		}
		else if (opt == 3 || opt == 4) {
			bool master = (opt == 3);

			this->peer = new Network(this->server_hostname,
					this->server_port, master);
			this->network_connection_type = master ? MASTER_CONNECT : CLIENT;
			if (this->network_connection_type == CLIENT &&
					this->peer->Connect() == false)
			{
				delete this->peer;
				this->peer = NULL;
				this->network_connection_type = NONE;
			}
		}
	} while (opt == 1 || opt == 2);

	this->prefs_changed = true;
}

void C64::advanced_options(Prefs *np)
{
	int submenus[4] = { np->DisplayOption, 0,
			np->SpriteCollisions, 0 };

#define SPEED_95 30
#define SPEED_100 20
#define SPEED_110 18

	switch (np->MsPerFrame)
	{
	case SPEED_95:
		submenus[1] = 0; break;
	case SPEED_110:
		submenus[1] = 2; break;
	default:
		/* If it has some other value... */
		submenus[1] = 1; break;
	}

	int opt = menu_select(new_advanced_options_menu_messages,
			submenus);
	if (opt >= 0)
	{
		np->DisplayOption = submenus[0];
		np->SpriteCollisions = submenus[2];

		switch(submenus[1])
		{
		case 0:
			np->MsPerFrame = SPEED_95; break;
		case 1:
			np->MsPerFrame = SPEED_100; break;
		case 2:
		default:
			np->MsPerFrame = SPEED_110; break;
		}

		this->prefs_changed = true;
	}
}

void C64::other_options(Prefs *np)
{
	int old_swap = ThePrefs.JoystickSwap == true ? 1 : 0; 
	int submenus[3] = { old_swap, !np->Emul1541Proc, 0 };

	int opt = menu_select(new_options_menu_messages,
			submenus);
	if (opt >= 0)
	{
		np->Emul1541Proc = submenus[1] == 0 ? true : false;

		this->prefs_changed = true;
	}
}

/* From dreamcast port but heavily modified */
void C64::run_fake_key_sequence()
{
	int kc = this->virtual_keyboard->char_to_keycode(this->fake_key_str[this->fake_key_index]);

	TheDisplay->FakeKeyPress(kc, TheCIA1->KeyMatrix, TheCIA1->RevMatrix);

	this->fake_key_keytime --;
        if (this->fake_key_keytime == 0)
        {
                this->fake_key_keytime = 4;
                this->fake_key_index ++;
        	TheDisplay->FakeKeyPress(-1, TheCIA1->KeyMatrix, TheCIA1->RevMatrix);

		if (this->fake_key_str[this->fake_key_index] == '\0')
                {
                        this->fake_key_sequence = false;
                        this->fake_key_index = 0;
                        this->fake_key_keytime = 5;
                }
        }
}

void C64::start_fake_key_sequence(const char *str)
{
	this->fake_key_str = str;
	this->fake_key_sequence = true;
}

void C64::select_fake_key_sequence(Prefs *np)
{
	static const char *fake_key_sequences[] = {
			"\nLOAD \"*\",8,1\nRUN\n",
			"\nLOAD \"$\",8\n",
			"\nLIST\n",
			NULL};
	const char *fake_key_messages[] = {
			"LOAD \"*\",8,1  and  RUN",
			"LOAD \"$\",8",
			"LIST",
			NULL};
	int opt;

	opt = menu_select(fake_key_messages, NULL);
	if (opt < 0)
		return;

	this->start_fake_key_sequence(fake_key_sequences[opt]);
}

void C64::save_load_state(Prefs *np, int opt)
{
	switch(opt)
	{
	case 1: /* save */
	{
		char save_buf[255];
		char prefs_buf[255];

		snprintf(save_buf, 255, "%s.sav",
				this->save_game_name);
		snprintf(prefs_buf, 255, "%s.prefs", save_buf);

		this->SaveSnapshot(save_buf);
		np->Save(prefs_buf);
	} break;
	case 0: /* load/delete */
	case 2:
	{
		const char *name = menu_select_file(SAVES_PATH);
		char save_buf[255];
		char prefs_buf[255];

		if (name == NULL)
			break;

		snprintf(save_buf, 255, "%s", name);
		snprintf(prefs_buf, 255, "%s.prefs", save_buf);
		if (opt == 2)
		{
			unlink(save_buf);
			unlink(prefs_buf);
		}
		else /* Load the snapshot */
		{
			this->LoadSnapshot(save_buf);
			np->Load(prefs_buf);
			this->prefs_changed = true;
		}

		free((void*)name);
	} break;
	default:
		break;
	}
}

/*
 *  Start main emulation thread
 */

void C64::Run(void)
{
	// Reset chips
	TheCPU->Reset();
	TheSID->Reset();
	TheCIA1->Reset();
	TheCIA2->Reset();
	TheCPU1541->Reset();

	// Patch kernal IEC routines
	orig_kernal_1d84 = Kernal[0x1d84];
	orig_kernal_1d85 = Kernal[0x1d85];
	PatchKernal(ThePrefs.FastReset, ThePrefs.Emul1541Proc);

	quit_thyself = false;
	thread_func();
}

void C64::network_vblank()
{
        static uint32_t last_time_update;
#if defined(GEKKO)
        Uint32 now = ticks_to_millisecs(gettime());
#else
        Uint32 now = SDL_GetTicks();
#endif

        if (this->peer) {
        	Uint8 *master = this->TheDisplay->BitmapBase();
        	Network *remote = this->peer;
		uint8 *js;
        	static bool has_throttled;

        	if (this->quit_thyself)
		{
        		if (this->network_connection_type != MASTER_CONNECT)
        			remote->Disconnect();
			delete remote;
			this->peer = NULL;
			return;
		}

        	remote->Tick( now - last_time_update );
        	if (this->network_connection_type == MASTER) {
        		if (ThePrefs.JoystickSwap)
        			js = &TheCIA1->Joystick1;
        		else
        			js = &TheCIA1->Joystick2;
        	} else if (this->network_connection_type == MASTER_CONNECT) {
        		network_connection_error_t err = this->peer->ConnectFSM();

        		TheDisplay->display_status_string("WAITING FOR CONNECTION...", 1);

        		if (err == OK) {
        			this->network_connection_type = MASTER;
                		TheDisplay->display_status_string("CLIENT CONNECTED!", 1);
        		}
        		else if (err != AGAIN_ERROR)
        		{
        			delete remote;
        			this->peer = NULL;
        		}
        		return;
        	} else {
        		if (ThePrefs.JoystickSwap)
        			js = &TheCIA1->Joystick2;
        		else
        			js = &TheCIA1->Joystick1;
        	}

        	/* Has the peer sent any data? */
        	if (remote->ReceiveUpdate() == true)
        	{
        		if (remote->DecodeUpdate(this->TheDisplay,
        				js, this->TheSID) == false)
        		{
        			/* Disconnect or sending crap, remove this guy! */
        			delete remote;
        			this->peer = NULL;
        			return;
        		}
                	if (this->network_connection_type == CLIENT)
                		this->TheDisplay->Update(remote->GetScreen());
        	}
		remote->ResetNetworkUpdate();

        	/* Encode and send updates to the other side (what is determined by 
        	 * if this is the master or not) */
		if (this->network_connection_type == MASTER)
		{
			/* Skip this frame if the data rate is too high */
			if (remote->ThrottleTraffic())
				has_throttled = true;
			else
				remote->EncodeDisplay(master, remote->GetScreen());
        	}

		char *msg = TheDisplay->GetTextMessage();
		if (msg)
			remote->EncodeTextMessage(msg);

		remote->EncodeJoystickUpdate(*js);
		remote->EncodeSound();

		if (remote->SendUpdate() == false)
        	{
        		/* Disconnect or broken data */
        		printf("Could not send update\n");
        	}
		else
			remote->ResetNetworkUpdate();

        	if (1)
        	{
        		static uint32_t last_traffic_update;

        		if (last_time_update - last_traffic_update > 300)
        		{
        			TheDisplay->NetworkTrafficMeter(remote->GetKbps() / (8 * 1024.0),
        					has_throttled);
        			last_traffic_update = now;
        			has_throttled = false;
        		}
        	}
        }

        last_time_update = now;
}

/*
 *  Vertical blank: Poll keyboard and joysticks, update window
 */

void C64::VBlank(bool draw_frame)
{
	/* From Acorn port */
	static uint32_t lastFrame;
        uint32_t now;
        uint8 j1, j2;

#if defined(GEKKO)
	WPAD_ScanPads();
#endif

	// Poll joysticks
	j1 = poll_joystick(0);
	j2 = poll_joystick(1);

	// Poll keyboard
	TheDisplay->PollKeyboard(TheCIA1->KeyMatrix, TheCIA1->RevMatrix, &joykey);
	if (TheDisplay->quit_requested)
		quit_thyself = true;

	if (this->fake_key_sequence)
		this->run_fake_key_sequence();
#ifndef GEKKO
	// Joystick keyboard emulation
	if (TheDisplay->NumLock())
		j1 &= joykey;
	else
		j2 &= joykey;
#endif

	if (this->network_connection_type == MASTER)
	{
		/* Only poll one joystick for network servers */
		if (ThePrefs.JoystickSwap)
			TheCIA1->Joystick2 = j2;
		else
			TheCIA1->Joystick1 = j2;
	}
	else if (this->network_connection_type == CLIENT)
	{
		Uint8 which = j2;

		/* Set both joysticks to the updated value */
		if (j2 != 0xff)
			which = j2;
		TheCIA1->Joystick1 = which;
		TheCIA1->Joystick2 = which;
	}
	else
	{
		TheCIA1->Joystick1 = j1;
		TheCIA1->Joystick2 = j2;
	}

	// Count TOD clocks
	TheCIA1->CountTOD();
	TheCIA2->CountTOD();

	// Update window if needed
	if (draw_frame && this->network_connection_type != CLIENT) {
		TheDisplay->Update();
	}

	if (this->have_a_break) {
		int submenus[3] = {1, 0, 0};
		int opt;

		Prefs np = ThePrefs;
		this->prefs_changed = false;

		TheSID->PauseSound();
		opt = menu_select(new_main_menu_messages, submenus);

		switch(opt)
		{
		case 2: /* Insert disc/tape */
			this->select_disc(&np, submenus[0] == 1);
			break;
		case 4: /* Save / load game */
			this->save_load_state(&np, submenus[1]);
			break;
		case 6: /* Bind keys to joystick */
			switch (submenus[2])
			{
			case 0: /* type */
			{
				const char *seq = this->virtual_keyboard->get_string();

				if (seq != NULL)
					this->start_fake_key_sequence(seq);
			} break;
			case 1: /* Macro */
				this->select_fake_key_sequence(&np); break;
			default:
			case 2: /* Bind to controller */
				this->bind_keys(&np); break;
			}
			break;
		case 9: /* Reset the C64 */
			Reset();
			break;
		case 10: /* Networking */
			this->networking_menu(&np);
			break;
		case 11: /* Other options */
			this->other_options(&np);
			break;
		case 12: /* Advanced options */
			this->advanced_options(&np);
			break;
		case 13:
		{
			menu_select(welcome, NULL);
		} break;
		case 15: /* Quit */
			quit_thyself = true;				
			break;
		case -1:
		default:
			break;
		}
		if (submenus[0] == 0)
			np.JoystickSwap = false;
		else
			np.JoystickSwap = true;

		if (this->prefs_changed)
		{
			this->NewPrefs(&np);
			ThePrefs = np;
		}
		TheDisplay->FakeKeyPress(-1, TheCIA1->KeyMatrix,
				TheCIA1->RevMatrix);

		this->have_a_break = false;
		if (this->quit_thyself)
			ThePrefs.Save((const char*)PREFS_PATH);
	}
	this->network_vblank();

#if defined(GEKKO)
	if (this->quit_thyself)
		SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);
        now = ticks_to_millisecs(gettime());
#else
        now = SDL_GetTicks();
#endif

        if ( (now - lastFrame) < ThePrefs.MsPerFrame) {
        	usleep( (ThePrefs.MsPerFrame - (now - lastFrame)) * 1000);
        }
        lastFrame = now;
}

/*
 * The emulation's main loop
 */

void C64::thread_func(void)
{
	int linecnt = 0;

#ifdef FRODO_SC
	while (!quit_thyself) {

		// The order of calls is important here
		if (TheVIC->EmulateCycle())
			TheSID->EmulateLine();
		TheCIA1->CheckIRQs();
		TheCIA2->CheckIRQs();
		TheCIA1->EmulateCycle();
		TheCIA2->EmulateCycle();
		TheCPU->EmulateCycle();

		if (ThePrefs.Emul1541Proc) {
			TheCPU1541->CountVIATimers(1);
			if (!TheCPU1541->Idle)
				TheCPU1541->EmulateCycle();
		}
		CycleCounter++;
#else
	while (!quit_thyself) {

		// The order of calls is important here
		int cycles = TheVIC->EmulateLine();
		TheSID->EmulateLine();
#if !PRECISE_CIA_CYCLES
		TheCIA1->EmulateLine(ThePrefs.CIACycles);
		TheCIA2->EmulateLine(ThePrefs.CIACycles);
#endif

		if (ThePrefs.Emul1541Proc) {
			int cycles_1541 = ThePrefs.FloppyCycles;
			TheCPU1541->CountVIATimers(cycles_1541);

			if (!TheCPU1541->Idle) {
				// 1541 processor active, alternately execute
				//  6502 and 6510 instructions until both have
				//  used up their cycles
				while (cycles >= 0 || cycles_1541 >= 0)
					if (cycles > cycles_1541)
						cycles -= TheCPU->EmulateLine(1);
					else
						cycles_1541 -= TheCPU1541->EmulateLine(1);
			} else
				TheCPU->EmulateLine(cycles);
		} else
			// 1541 processor disabled, only emulate 6510
			TheCPU->EmulateLine(cycles);
#endif
		linecnt++;
	}
}