#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <termios.h>
#include <fcntl.h>
#include <string.h>
#include <sys/errno.h>
#include <sys/ioctl.h>
#include <sys/types.h>

#define __u32 unsigned int
#define __u8 unsigned char
#include <linux/joystick.h>

#include "../../shared/debug.h"
#include "../../shared/str.h"

#define NAME_LENGTH 128

#include "lcd.h"
#include "joy.h"

//////////////////////////////////////////////////////////////////////////
////////////////////// Base "class" to derive from ///////////////////////
//////////////////////////////////////////////////////////////////////////

lcd_logical_driver *joy;

int fd;

struct js_event js;

char axes = 2;
char buttons = 2;
int jsversion = 0x000800;
char jsname[NAME_LENGTH] = "Unknown";

int *axis;
int *button;

// Configured for a Gravis Gamepad  (2 axis, 4 button)
char *axismap = "EFGHIJKLMNOPQRST";
char *buttonmap = "BDACEFGHIJKLMNOP";

////////////////////////////////////////////////////////////
// init() should set up any device-specific stuff, and
// point all the function pointers.
int
joy_init (struct lcd_logical_driver *driver, char *args)
{
	char device[256];
	char *argv[64];
	int argc, i, j;

	joy = driver;

	strcpy (device, "/dev/js0");

	argc = get_args (argv, args, 64);

	for (i = 0; i < argc; i++) {
		//printf("Arg(%i): %s\n", i, argv[i]);
		if (0 == strcmp (argv[i], "-d") || 0 == strcmp (argv[i], "--device")) {
			if (i + 1 > argc) {
				fprintf (stderr, "joy_init: %s requires an argument\n", argv[i]);
				return -1;
			}
			strcpy (device, argv[++i]);
		} else if (0 == strcmp (argv[i], "-a") || 0 == strcmp (argv[i], "--axes")) {
			if (i + 1 > argc) {
				fprintf (stderr, "joy_init: %s requires an argument\n", argv[i]);
				return -1;
			}
			strncpy (axismap, argv[++i], 16);
		} else if (0 == strcmp (argv[i], "-b") || 0 == strcmp (argv[i], "--buttons")) {
			if (i + 1 > argc) {
				fprintf (stderr, "joy_init: %s requires an argument\n", argv[i]);
				return -1;
			}
			strncpy (buttonmap, argv[++i], 16);
		} else if (0 == strcmp (argv[i], "-h") || 0 == strcmp (argv[i], "--help")) {
			printf ("LCDproc Joystick input driver\n" "\t-d\t--device\tSelect the input device to use [/dev/js0]\n" "\t-a\t--axes\t\tModify the axis map [%s]\n" "\t-b\t--buttons\tModify the button map [%s]\n" "\t-h\t--help\t\tShow this help information\n", axismap, buttonmap);
			return -1;
		} else {
			printf ("Invalid parameter: %s\n", argv[i]);
		}

	}

	driver->getkey = joy_getkey;
	driver->close = joy_close;

	/*  FIXME:  This crashes!
	   if(args)
	   {
	   if(strlen(args) > 0)
	   strcpy(device, args);
	   }
	 */

	fd = open (device, O_RDONLY);
	fcntl (fd, F_SETFL, O_NONBLOCK);

	if (fd < 0)
		return -1;

	ioctl (fd, JSIOCGVERSION, &jsversion);
	ioctl (fd, JSIOCGAXES, &axes);
	ioctl (fd, JSIOCGBUTTONS, &buttons);
	ioctl (fd, JSIOCGNAME (NAME_LENGTH), jsname);

	debug ("Joystick (%s) has %d axes and %d buttons. Driver version is %d.%d.%d.\n", jsname, axes, buttons, jsversion >> 16, (jsversion >> 8) & 0xff, jsversion & 0xff);

	axis = calloc (axes, sizeof (int));
	button = calloc (buttons, sizeof (char));

	return fd;						  // 200 is arbitrary.  (must be 1 or more)
}

void
joy_close ()
{
	if (joy->framebuf != NULL)
		free (joy->framebuf);
	close (fd);

	joy->framebuf = NULL;

	// Why do I have so much trouble getting memory freed without segfaults??
	//if(axis) free(axis);
	//if(button) free(button);

}

//////////////////////////////////////////////////////////////////////
// Tries to read a character from an input device...
//
// Return 0 for "nothing available".
//
char
joy_getkey ()
{
	int i;
	int err;

	err = read (fd, &js, sizeof (struct js_event));
	if (err <= 0)
		return 0;
	if (err != sizeof (struct js_event)) {
		fprintf (stderr, "\nerror reading joystick\n");
		return 0;
	}
//   if(js.type & JS_EVENT_INIT) return 0;

	switch (js.type & ~JS_EVENT_INIT) {
	case JS_EVENT_BUTTON:
		button[js.number] = js.value;
		break;
	case JS_EVENT_AXIS:
		axis[js.number] = js.value;
		break;
	}

	if (buttons) {
		//printf("Buttons: ");
		for (i = 0; i < buttons; i++)
			//printf("%2d:%s ", i, button[i] ? "on " : "off");
			if (button[i])
				return buttonmap[i];
	}

	if (axes) {
		//printf("Axes: ");
		for (i = 0; i < axes; i++) {
			//printf("%2d:%6d ", i, axis[i]);
			// Eliminate noise...
			if (axis[i] > 20000)
				return axismap[(2 * i) + 1];
			if (axis[i] < -20000)
				return axismap[(2 * i)];
		}
	}

	return 0;
}