/*

    SDL - Simple DirectMedia Layer

    Copyright (C) 1997-2010 Sam Lantinga

    This library is free software; you can redistribute it and/or

    modify it under the terms of the GNU Lesser General Public

    License as published by the Free Software Foundation; either

    version 2.1 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public

    License along with this library; if not, write to the Free Software

    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

    Sam Lantinga

    slouken@libsdl.org

*/

#include "SDL_config.h"

#ifdef SDL_JOYSTICK_USBHID

/*

 * Joystick driver for the uhid(4) interface found in OpenBSD,

 * NetBSD and FreeBSD.

 *

 * Maintainer: <vedge at csoft.org>

 */

#include <sys/param.h>

#include <unistd.h>

#include <fcntl.h>

#include <errno.h>

#ifndef __FreeBSD_kernel_version

#define __FreeBSD_kernel_version __FreeBSD_version

#endif

#if defined(HAVE_USB_H)

#include <usb.h>

#endif

#ifdef __DragonFly__

#include <bus/usb/usb.h>

#include <bus/usb/usbhid.h>

#else

#include <dev/usb/usb.h>

#include <dev/usb/usbhid.h>

#endif

#if defined(HAVE_USBHID_H)

#include <usbhid.h>

#elif defined(HAVE_LIBUSB_H)

#include <libusb.h>

#elif defined(HAVE_LIBUSBHID_H)

#include <libusbhid.h>

#endif

#ifdef __FREEBSD__

#ifndef __DragonFly__

#include <osreldate.h>

#endif

#include <sys/joystick.h>

#endif

#if SDL_JOYSTICK_USBHID_MACHINE_JOYSTICK_H

#include <machine/joystick.h>

#endif

#include "SDL_joystick.h"

#include "../SDL_sysjoystick.h"

#include "../SDL_joystick_c.h"

#define MAX_UHID_JOYS	4

#define MAX_JOY_JOYS	2

#define MAX_JOYS	(MAX_UHID_JOYS + MAX_JOY_JOYS)

#if defined(__FREEBSD__) && (__FreeBSD_kernel_version > 800063)

struct usb_ctl_report {

    int     ucr_report;

    u_char  ucr_data[1024]; /* filled data size will vary */

};

#endif

struct report

{

    struct usb_ctl_report *buf; /* Buffer */

    size_t size;                /* Buffer size */

    int rid;                    /* Report ID */

    enum

    {

        SREPORT_UNINIT,

        SREPORT_CLEAN,

        SREPORT_DIRTY

    } status;

};

static struct

{

    int uhid_report;

    hid_kind_t kind;

    const char *name;

} const repinfo[] = {

    {UHID_INPUT_REPORT, hid_input, "input"},

    {UHID_OUTPUT_REPORT, hid_output, "output"},

    {UHID_FEATURE_REPORT, hid_feature, "feature"}

};

enum

{

    REPORT_INPUT = 0,

    REPORT_OUTPUT = 1,

    REPORT_FEATURE = 2

};

enum

{

    JOYAXE_X,

    JOYAXE_Y,

    JOYAXE_Z,

    JOYAXE_SLIDER,

    JOYAXE_WHEEL,

    JOYAXE_RX,

    JOYAXE_RY,

    JOYAXE_RZ,

    JOYAXE_count

};

struct joystick_hwdata

{

    int fd;

    char *path;

    enum

    {

        BSDJOY_UHID,            /* uhid(4) */

        BSDJOY_JOY              /* joy(4) */

    } type;

    struct report_desc *repdesc;

    struct report inreport;

    int axis_map[JOYAXE_count]; /* map present JOYAXE_* to 0,1,.. */

};

static char *joynames[MAX_JOYS];

static char *joydevnames[MAX_JOYS];

static int report_alloc(struct report *, struct report_desc *, int);

static void report_free(struct report *);

#if defined(USBHID_UCR_DATA) || (defined(__FREEBSD__) && (__FreeBSD_kernel_version > 800063))

#define REP_BUF_DATA(rep) ((rep)->buf->ucr_data)

#else

#define REP_BUF_DATA(rep) ((rep)->buf->data)

#endif

int

SDL_SYS_JoystickInit(void)

{

    char s[16];

    int i, fd;

    SDL_numjoysticks = 0;

    SDL_memset(joynames, 0, sizeof(joynames));

    SDL_memset(joydevnames, 0, sizeof(joydevnames));

    for (i = 0; i < MAX_UHID_JOYS; i++) {

        SDL_Joystick nj;

        SDL_snprintf(s, SDL_arraysize(s), "/dev/uhid%d", i);

        nj.index = SDL_numjoysticks;

        joynames[nj.index] = strdup(s);

        if (SDL_SYS_JoystickOpen(&nj) == 0) {

            SDL_SYS_JoystickClose(&nj);

            SDL_numjoysticks++;

        } else {

            SDL_free(joynames[nj.index]);

            joynames[nj.index] = NULL;

        }

    }

    for (i = 0; i < MAX_JOY_JOYS; i++) {

        SDL_snprintf(s, SDL_arraysize(s), "/dev/joy%d", i);

        fd = open(s, O_RDONLY);

        if (fd != -1) {

            joynames[SDL_numjoysticks++] = strdup(s);

            close(fd);

        }

    }

    /* Read the default USB HID usage table. */

    hid_init(NULL);

    return (SDL_numjoysticks);

}

const char *

SDL_SYS_JoystickName(int index)

{

    if (joydevnames[index] != NULL) {

        return (joydevnames[index]);

    }

    return (joynames[index]);

}

static int

usage_to_joyaxe(unsigned usage)

{

    int joyaxe;

    switch (usage) {

    case HUG_X:

        joyaxe = JOYAXE_X;

        break;

    case HUG_Y:

        joyaxe = JOYAXE_Y;

        break;

    case HUG_Z:

        joyaxe = JOYAXE_Z;

        break;

    case HUG_SLIDER:

        joyaxe = JOYAXE_SLIDER;

        break;

    case HUG_WHEEL:

        joyaxe = JOYAXE_WHEEL;

        break;

    case HUG_RX:

        joyaxe = JOYAXE_RX;

        break;

    case HUG_RY:

        joyaxe = JOYAXE_RY;

        break;

    case HUG_RZ:

        joyaxe = JOYAXE_RZ;

        break;

    default:

        joyaxe = -1;

    }

    return joyaxe;

}

static unsigned

hatval_to_sdl(Sint32 hatval)

{

    static const unsigned hat_dir_map[8] = {

        SDL_HAT_UP, SDL_HAT_RIGHTUP, SDL_HAT_RIGHT, SDL_HAT_RIGHTDOWN,

        SDL_HAT_DOWN, SDL_HAT_LEFTDOWN, SDL_HAT_LEFT, SDL_HAT_LEFTUP

    };

    unsigned result;

    if ((hatval & 7) == hatval)

        result = hat_dir_map[hatval];

    else

        result = SDL_HAT_CENTERED;

    return result;

}

int

SDL_SYS_JoystickOpen(SDL_Joystick * joy)

{

    char *path = joynames[joy->index];

    struct joystick_hwdata *hw;

    struct hid_item hitem;

    struct hid_data *hdata;

    struct report *rep;

    int fd;

    int i;

    fd = open(path, O_RDONLY);

    if (fd == -1) {

        SDL_SetError("%s: %s", path, strerror(errno));

        return (-1);

    }

    hw = (struct joystick_hwdata *)

        SDL_malloc(sizeof(struct joystick_hwdata));

    if (hw == NULL) {

        SDL_OutOfMemory();

        close(fd);

        return (-1);

    }

    joy->hwdata = hw;

    hw->fd = fd;

    hw->path = strdup(path);

    if (!SDL_strncmp(path, "/dev/joy", 8)) {

        hw->type = BSDJOY_JOY;

        joy->naxes = 2;

        joy->nbuttons = 2;

        joy->nhats = 0;

        joy->nballs = 0;

        joydevnames[joy->index] = strdup("Gameport joystick");

        goto usbend;

    } else {

        hw->type = BSDJOY_UHID;

    }

    {

        int ax;

        for (ax = 0; ax < JOYAXE_count; ax++)

            hw->axis_map[ax] = -1;

    }

    hw->repdesc = hid_get_report_desc(fd);

    if (hw->repdesc == NULL) {

        SDL_SetError("%s: USB_GET_REPORT_DESC: %s", hw->path,

                     strerror(errno));

        goto usberr;

    }

#if defined(__FREEBSD__) && (__FreeBSD_kernel_version > 800063)

    rep->rid = hid_get_report_id(fd);

    if (rep->rid < 0) {

#else

    if (ioctl(fd, USB_GET_REPORT_ID, &rep->rid) < 0) {

#endif

        rep->rid = -1;          /* XXX */

    }

    if (report_alloc(rep, hw->repdesc, REPORT_INPUT) < 0) {

        goto usberr;

    }

    if (rep->size <= 0) {

        SDL_SetError("%s: Input report descriptor has invalid length",

                     hw->path);

        goto usberr;

    }

#if defined(USBHID_NEW) || (defined(__FREEBSD__) && __FreeBSD_kernel_version >= 500111)

    hdata = hid_start_parse(hw->repdesc, 1 << hid_input, rep->rid);

#else

    hdata = hid_start_parse(hw->repdesc, 1 << hid_input);

#endif

    if (hdata == NULL) {

        SDL_SetError("%s: Cannot start HID parser", hw->path);

        goto usberr;

    }

    joy->naxes = 0;

    joy->nbuttons = 0;

    joy->nhats = 0;

    joy->nballs = 0;

    for (i = 0; i < JOYAXE_count; i++)

        hw->axis_map[i] = -1;

    while (hid_get_item(hdata, &hitem) > 0) {

        char *sp;

        const char *s;

        switch (hitem.kind) {

        case hid_collection:

            switch (HID_PAGE(hitem.usage)) {

            case HUP_GENERIC_DESKTOP:

                switch (HID_USAGE(hitem.usage)) {

                case HUG_JOYSTICK:

                case HUG_GAME_PAD:

                    s = hid_usage_in_page(hitem.usage);

                    sp = SDL_malloc(SDL_strlen(s) + 5);

                    SDL_snprintf(sp, SDL_strlen(s) + 5, "%s (%d)",

                                 s, joy->index);

                    joydevnames[joy->index] = sp;

                }

            }

            break;

        case hid_input:

            switch (HID_PAGE(hitem.usage)) {

            case HUP_GENERIC_DESKTOP:

                {

                    unsigned usage = HID_USAGE(hitem.usage);

                    int joyaxe = usage_to_joyaxe(usage);

                    if (joyaxe >= 0) {

                        hw->axis_map[joyaxe] = 1;

                    } else if (usage == HUG_HAT_SWITCH) {

                        joy->nhats++;

                    }

                    break;

                }

            case HUP_BUTTON:

                joy->nbuttons++;

                break;

            default:

                break;

            }

            break;

        default:

            break;

        }

    }

    hid_end_parse(hdata);

    for (i = 0; i < JOYAXE_count; i++)

        if (hw->axis_map[i] > 0)

            hw->axis_map[i] = joy->naxes++;

  usbend:

    /* The poll blocks the event thread. */

    fcntl(fd, F_SETFL, O_NONBLOCK);

    return (0);

  usberr:

    close(hw->fd);

    SDL_free(hw->path);

    SDL_free(hw);

    return (-1);

}

void

SDL_SYS_JoystickUpdate(SDL_Joystick * joy)

{

    struct hid_item hitem;

    struct hid_data *hdata;

    struct report *rep;

    int nbutton, naxe = -1;

    Sint32 v;

#if defined(__FREEBSD__) || SDL_JOYSTICK_USBHID_MACHINE_JOYSTICK_H

    struct joystick gameport;

    static int x, y, xmin = 0xffff, ymin = 0xffff, xmax = 0, ymax = 0;

    if (joy->hwdata->type == BSDJOY_JOY) {

        if (read(joy->hwdata->fd, &gameport, sizeof gameport) !=

            sizeof gameport)

            return;

        if (abs(x - gameport.x) > 8) {

            x = gameport.x;

            if (x < xmin) {

                xmin = x;

            }

            if (x > xmax) {

                xmax = x;

            }

            if (xmin == xmax) {

                xmin--;

                xmax++;

            }

            v = (Sint32) x;

            v -= (xmax + xmin + 1) / 2;

            v *= 32768 / ((xmax - xmin + 1) / 2);

            SDL_PrivateJoystickAxis(joy, 0, v);

        }

        if (abs(y - gameport.y) > 8) {

            y = gameport.y;

            if (y < ymin) {

                ymin = y;

            }

            if (y > ymax) {

                ymax = y;

            }

            if (ymin == ymax) {

                ymin--;

                ymax++;

            }

            v = (Sint32) y;

            v -= (ymax + ymin + 1) / 2;

            v *= 32768 / ((ymax - ymin + 1) / 2);

            SDL_PrivateJoystickAxis(joy, 1, v);

        }

        if (gameport.b1 != joy->buttons[0]) {

            SDL_PrivateJoystickButton(joy, 0, gameport.b1);

        }

        if (gameport.b2 != joy->buttons[1]) {

            SDL_PrivateJoystickButton(joy, 1, gameport.b2);

        }

        return;

    }

#endif /* defined(__FREEBSD__) || SDL_JOYSTICK_USBHID_MACHINE_JOYSTICK_H */

    rep = &joy->hwdata->inreport;

    if (read(joy->hwdata->fd, REP_BUF_DATA(rep), rep->size) != rep->size) {

        return;

    }

#if defined(USBHID_NEW) || (defined(__FREEBSD__) && __FreeBSD_kernel_version >= 500111)

    hdata = hid_start_parse(joy->hwdata->repdesc, 1 << hid_input, rep->rid);

#else

    hdata = hid_start_parse(joy->hwdata->repdesc, 1 << hid_input);

#endif

    if (hdata == NULL) {

        fprintf(stderr, "%s: Cannot start HID parser\n", joy->hwdata->path);

        return;

    }

    for (nbutton = 0; hid_get_item(hdata, &hitem) > 0;) {

        switch (hitem.kind) {

        case hid_input:

            switch (HID_PAGE(hitem.usage)) {

            case HUP_GENERIC_DESKTOP:

                {

                    unsigned usage = HID_USAGE(hitem.usage);

                    int joyaxe = usage_to_joyaxe(usage);

                    if (joyaxe >= 0) {

                        naxe = joy->hwdata->axis_map[joyaxe];

                        /* scaleaxe */

                        v = (Sint32) hid_get_data(REP_BUF_DATA(rep), &hitem);