picodrive/platform/common/input.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "input.h"
#include "../linux/in_evdev.h"

typedef struct
{
	int drv_id;
	void *drv_data;
	int *binds;
	char *name;
	int probed:1;
} in_dev_t;

static in_drv_t in_drivers[IN_DRVID_COUNT];
static in_dev_t in_devices[IN_MAX_DEVS];
static int in_dev_count = 0;

#define DRV(id) in_drivers[(unsigned)(id) < IN_DRVID_COUNT ? (id) : 0]

static int in_bind_count(int drv_id)
{
	int count = DRV(drv_id).get_bind_count();
	if (count <= 0)
		printf("input: failed to get bind count for drv %d\n", drv_id);

	return count;
}

static int *in_alloc_binds(int drv_id)
{
	int count, *binds;

	count = in_bind_count(drv_id);
	if (count <= 0)
		return NULL;

	binds = calloc(count * 2, sizeof(binds[0]));
	if (binds == NULL)
		return NULL;

	DRV(drv_id).get_def_binds(binds + count);
	memcpy(binds, binds + count, count * sizeof(binds[0]));

	return binds;
}

static void in_free(in_dev_t *dev)
{
	if (dev->probed)
		DRV(dev->drv_id).free(dev->drv_data);
	dev->probed = 0;
	dev->drv_data = NULL;
	free(dev->name);
	dev->name = NULL;
	free(dev->binds);
	dev->binds = NULL;
}

/* to be called by drivers */
void in_register(const char *nname, int drv_id, void *drv_data)
{
	int i, ret, dupe_count = 0, *binds;
	char name[256], *name_end, *tmp;

	strncpy(name, nname, sizeof(name));
	name[sizeof(name)-12] = 0;
	name_end = name + strlen(name);

	for (i = 0; i < in_dev_count; i++)
	{
		if (in_devices[i].name == NULL)
			continue;
		if (strcmp(in_devices[i].name, name) == 0)
		{
			if (in_devices[i].probed) {
				dupe_count++;
				sprintf(name_end, " [%d]", dupe_count);
				continue;
			}
			goto update;
		}
	}

	if (i >= IN_MAX_DEVS)
	{
		/* try to find unused device */
		for (i = 0; i < IN_MAX_DEVS; i++)
			if (!in_devices[i].probed) break;
		if (i >= IN_MAX_DEVS) {
			printf("input: too many devices, can't add %s\n", name);
			return;
		}
		in_free(&in_devices[i]);
	}

	tmp = strdup(name);
	if (tmp == NULL)
		return;

	binds = in_alloc_binds(drv_id);
	if (binds == NULL) {
		free(tmp);
		return;
	}

	in_devices[i].name = tmp;
	in_devices[i].binds = binds;
	if (i + 1 > in_dev_count)
		in_dev_count = i + 1;

	printf("input: new device #%d \"%s\"\n", i, name);
update:
	in_devices[i].probed = 1;
	in_devices[i].drv_id = drv_id;
	in_devices[i].drv_data = drv_data;

	if (in_devices[i].binds != NULL) {
		ret = DRV(drv_id).clean_binds(drv_data, in_devices[i].binds);
		if (ret == 0) {
			/* no useable binds */
			free(in_devices[i].binds);
			in_devices[i].binds = NULL;
		}
	}
}

void in_probe(void)
{
	int i;
	for (i = 0; i < in_dev_count; i++)
		in_devices[i].probed = 0;

	for (i = 1; i < IN_DRVID_COUNT; i++)
		in_drivers[i].probe();

	/* get rid of devs without binds and probes */
	for (i = 0; i < in_dev_count; i++) {
		if (!in_devices[i].probed && in_devices[i].binds == NULL) {
			in_dev_count--;
			if (i < in_dev_count) {
				free(in_devices[i].name);
				memmove(&in_devices[i], &in_devices[i+1],
					(in_dev_count - i) * sizeof(in_devices[0]));
			}
		}
	}
}

int in_update(void)
{
	int i, result = 0;

	for (i = 0; i < in_dev_count; i++) {
		if (in_devices[i].probed && in_devices[i].binds != NULL) {
			switch (in_devices[i].drv_id) {
#ifdef IN_EVDEV
			case IN_DRVID_EVDEV:
				result |= in_evdev_update(in_devices[i].drv_data, in_devices[i].binds);
				break;
#endif
			}
		}
	}

	return result;
}

static void **in_collect_drvdata(int drv_id, int *count)
{
	static void *data[IN_MAX_DEVS];
	int i;

	for (*count = i = 0; i < in_dev_count; i++) {
		if (in_devices[i].drv_id == drv_id && in_devices[i].probed)
			data[(*count)++] = in_devices[i].drv_data;
	}

	return data;
}

void in_set_blocking(int is_blocking)
{
	int i;

	for (i = 0; i < in_dev_count; i++) {
		if (in_devices[i].probed)
			DRV(in_devices[i].drv_id).set_blocking(in_devices[i].drv_data, is_blocking);
	}
}

/* 
 * update with wait for a press, return keycode
 * only can use 1 drv here..
 */
int in_update_keycode(int *dev_id, int *is_down)
{
	int result = 0;
#ifdef IN_EVDEV
	void **data;
	int i, id = 0, count = 0;

	data = in_collect_drvdata(IN_DRVID_EVDEV, &count);
	if (count == 0) {
		/* don't deadlock, fail */
		printf("input: failed to find devices to read\n");
		exit(1);
	}

	result = in_evdev_update_keycode(data, count, &id, is_down);

	if (dev_id != NULL) {
		for (i = id; i < in_dev_count; i++) {
			if (in_devices[i].drv_data == data[id]) {
				*dev_id = i;
				break;
			}
		}
	}
#else
#error no menu read handlers
#endif

	return result;
}

/* 
 * same as above, only return bitfield of BTN_*
 */
int in_update_menu(void)
{
	static int keys_active = 0;
	int keys_old = keys_active;

	while (1)
	{
		int code, is_down = 0, dev_id = 0;
		code = in_update_keycode(&dev_id, &is_down);
		code = DRV(in_devices[dev_id].drv_id).menu_translate(code);
		if (code == 0) continue;

		if (is_down)
			keys_active |=  code;
		else
			keys_active &= ~code;

		if (keys_old != keys_active)
			break;
	}

	return keys_active;
}

const int *in_get_dev_binds(int dev_id)
{
	if (dev_id < 0 || dev_id >= IN_MAX_DEVS)
		return NULL;

	return in_devices[dev_id].binds;
}

const int *in_get_dev_def_binds(int dev_id)
{
	int count;

	if (dev_id < 0 || dev_id >= IN_MAX_DEVS)
		return NULL;

	count = in_bind_count(in_devices[dev_id].drv_id);
	return in_devices[dev_id].binds + count;
}

int in_get_dev_bind_count(int dev_id)
{
	if (dev_id < 0 || dev_id >= IN_MAX_DEVS)
		return 0;

	return in_bind_count(in_devices[dev_id].drv_id);
}

const char *in_get_dev_name(int dev_id)
{
	if (dev_id < 0 || dev_id >= IN_MAX_DEVS)
		return NULL;

	return in_devices[dev_id].name;
}

const char *in_get_key_name(int dev_id, int keycode)
{
	if (dev_id < 0 || dev_id >= IN_MAX_DEVS)
		return "Unkn0";

	return DRV(in_devices[dev_id].drv_id).get_key_name(keycode);
}

/* returns device id, or -1 on error */
int in_config_parse_dev(const char *name)
{
	int drv_id = -1, i;

	for (i = 0; i < IN_DRVID_COUNT; i++) {
		int len = strlen(in_drivers[i].prefix);
		if (strncmp(name, in_drivers[i].prefix, len) == 0) {
			drv_id = i;
			break;
		}
	}

	if (drv_id < 0) {
		printf("input: missing driver for %s\n", name);
		return -1;
	}

	for (i = 0; i < in_dev_count; i++)
	{
		if (in_devices[i].name == NULL)
			continue;
		if (strcmp(in_devices[i].name, name) == 0)
			return i;
	}

	if (i >= IN_MAX_DEVS)
	{
		/* try to find unused device */
		for (i = 0; i < IN_MAX_DEVS; i++)
			if (in_devices[i].name == NULL) break;
		if (i >= IN_MAX_DEVS) {
			printf("input: too many devices, can't add %s\n", name);
			return -1;
		}
	}

	memset(&in_devices[i], 0, sizeof(in_devices[i]));

	in_devices[i].name = strdup(name);
	if (in_devices[i].name == NULL)
		return -1;

	if (i + 1 > in_dev_count)
		in_dev_count = i + 1;
	in_devices[i].drv_id = drv_id;

	return i;
}

void in_config_start(void)
{
	int i;

	/* mark all default binds, so they get overwritten by func below */
	for (i = 0; i < IN_MAX_DEVS; i++) {
		int n, count, *binds, *def_binds;

		if (in_devices[i].binds == NULL)
			continue;

		count = in_bind_count(in_devices[i].drv_id);
		binds = in_devices[i].binds;
		def_binds = binds + count;

		for (n = 0; n < count; n++)
			if (binds[n] == def_binds[n])
				binds[n] = -1;
	}
}

int in_config_bind_key(int dev_id, const char *key, int binds)
{
	int kc;

	if (dev_id < 0 || dev_id >= IN_MAX_DEVS)
		return -1;

	if (in_devices[dev_id].binds == NULL) {
		in_devices[dev_id].binds = in_alloc_binds(in_devices[dev_id].drv_id);
		if (in_devices[dev_id].binds == NULL)
			return -1;
		in_config_start();
	}

	kc = DRV(in_devices[dev_id].drv_id).get_key_code(key);
	if (kc < 0) {
		printf("input: bad key: %s\n", key);
		return -1;
	}

	if (in_devices[dev_id].binds[kc] == -1)
		in_devices[dev_id].binds[kc] = 0;
	in_devices[dev_id].binds[kc] |= binds;

	return 0;
}

void in_config_end(void)
{
	int i;

	for (i = 0; i < IN_MAX_DEVS; i++) {
		int n, ret, count, *binds, *def_binds;

		if (in_devices[i].binds == NULL)
			continue;

		count = in_bind_count(in_devices[i].drv_id);
		binds = in_devices[i].binds;
		def_binds = binds + count;

		for (n = 0; n < count; n++)
			if (binds[n] == -1)
				binds[n] = def_binds[n];

		if (in_devices[i].drv_data == NULL)
			continue;

		ret = DRV(in_devices[i].drv_id).clean_binds(in_devices[i].drv_data, binds);
		if (ret == 0) {
			/* no useable binds */
			free(binds);
			in_devices[i].binds = NULL;
		}
	}
}

void in_debug_dump(void)
{
	int i;

	printf("# drv probed binds name\n");
	for (i = 0; i < IN_MAX_DEVS; i++) {
		in_dev_t *d = &in_devices[i];
		if (!d->probed && d->name == NULL && d->binds == NULL)
			continue;
		printf("%d %3d %6c %5c %s\n", i, d->drv_id, d->probed ? 'y' : 'n',
			d->binds ? 'y' : 'n', d->name);
	}
}

/* handlers for unknown/not_preset drivers */

static void in_def_probe(void) {}
static void in_def_free(void *drv_data) {}
static int  in_def_get_bind_count(void) { return 0; }
static void in_def_get_def_binds(int *binds) {}
static int  in_def_clean_binds(void *drv_data, int *binds) { return 0; }
static void in_def_set_blocking(void *data, int y) {}
static int  in_def_menu_translate(int keycode) { return keycode; }
static int  in_def_get_key_code(const char *key_name) { return 0; }
static const char *in_def_get_key_name(int keycode) { return NULL; }

void in_init(void)
{
	int i;

	memset(in_drivers, 0, sizeof(in_drivers));
	memset(in_devices, 0, sizeof(in_devices));
	in_dev_count = 0;

	for (i = 0; i < IN_DRVID_COUNT; i++) {
		in_drivers[i].prefix = "none:";
		in_drivers[i].probe = in_def_probe;
		in_drivers[i].free = in_def_free;
		in_drivers[i].get_bind_count = in_def_get_bind_count;
		in_drivers[i].get_def_binds = in_def_get_def_binds;
		in_drivers[i].clean_binds = in_def_clean_binds;
		in_drivers[i].set_blocking = in_def_set_blocking;
		in_drivers[i].menu_translate = in_def_menu_translate;
		in_drivers[i].get_key_code = in_def_get_key_code;
		in_drivers[i].get_key_name = in_def_get_key_name;
	}

#ifdef IN_EVDEV
	in_evdev_init(&in_drivers[IN_DRVID_EVDEV]);
#endif
}

#if 0
int main(void)
{
	int ret;

	in_init();
	in_probe();

	in_set_blocking(1);

#if 1
	while (1) {
		int dev = 0, down;
		ret = in_update_keycode(&dev, &down);
		printf("#%i: %i %i (%s)\n", dev, down, ret, in_get_key_name(dev, ret));
	}
#else
	while (1) {
		ret = in_update_menu();
		printf("%08x\n", ret);
	}
#endif

	return 0;
}
#endif