wiiuse.h

Go to the documentation of this file.

#ifndef __WIIUSE_H__
#define __WIIUSE_H__

#if defined(_WIN32)
 /* windows */
 #include <windows.h>
#elif defined(GEKKO)
 /* wii */
 #include <bte/bte.h>
#elif defined(__linux__)
 /* nix */
 #include <bluetooth/bluetooth.h>
#endif

#ifdef WIIUSE_INTERNAL_H_INCLUDED
 #define WCONST
#else
 #define WCONST const
#endif

/* led bit masks */
#define WIIMOTE_LED_NONE 0x00
#define WIIMOTE_LED_1 0x10
#define WIIMOTE_LED_2 0x20
#define WIIMOTE_LED_3 0x40
#define WIIMOTE_LED_4 0x80

/* button codes */
#define WIIMOTE_BUTTON_TWO 0x0001
#define WIIMOTE_BUTTON_ONE 0x0002
#define WIIMOTE_BUTTON_B 0x0004
#define WIIMOTE_BUTTON_A 0x0008
#define WIIMOTE_BUTTON_MINUS 0x0010
#define WIIMOTE_BUTTON_ZACCEL_BIT6 0x0020
#define WIIMOTE_BUTTON_ZACCEL_BIT7 0x0040
#define WIIMOTE_BUTTON_HOME 0x0080
#define WIIMOTE_BUTTON_LEFT 0x0100
#define WIIMOTE_BUTTON_RIGHT 0x0200
#define WIIMOTE_BUTTON_DOWN 0x0400
#define WIIMOTE_BUTTON_UP 0x0800
#define WIIMOTE_BUTTON_PLUS 0x1000
#define WIIMOTE_BUTTON_ZACCEL_BIT4 0x2000
#define WIIMOTE_BUTTON_ZACCEL_BIT5 0x4000
#define WIIMOTE_BUTTON_UNKNOWN 0x8000
#define WIIMOTE_BUTTON_ALL 0x1F9F

/* nunchul button codes */
#define NUNCHUK_BUTTON_Z 0x01
#define NUNCHUK_BUTTON_C 0x02
#define NUNCHUK_BUTTON_ALL 0x03

/* classic controller button codes */
#define CLASSIC_CTRL_BUTTON_UP 0x0001
#define CLASSIC_CTRL_BUTTON_LEFT 0x0002
#define CLASSIC_CTRL_BUTTON_ZR 0x0004
#define CLASSIC_CTRL_BUTTON_X 0x0008
#define CLASSIC_CTRL_BUTTON_A 0x0010
#define CLASSIC_CTRL_BUTTON_Y 0x0020
#define CLASSIC_CTRL_BUTTON_B 0x0040
#define CLASSIC_CTRL_BUTTON_ZL 0x0080
#define CLASSIC_CTRL_BUTTON_FULL_R 0x0200
#define CLASSIC_CTRL_BUTTON_PLUS 0x0400
#define CLASSIC_CTRL_BUTTON_HOME 0x0800
#define CLASSIC_CTRL_BUTTON_MINUS 0x1000
#define CLASSIC_CTRL_BUTTON_FULL_L 0x2000
#define CLASSIC_CTRL_BUTTON_DOWN 0x4000
#define CLASSIC_CTRL_BUTTON_RIGHT 0x8000
#define CLASSIC_CTRL_BUTTON_ALL 0xFEFF

/* guitar hero 3 button codes */
#define GUITAR_HERO_3_BUTTON_STRUM_UP 0x0001
#define GUITAR_HERO_3_BUTTON_YELLOW 0x0008
#define GUITAR_HERO_3_BUTTON_GREEN 0x0010
#define GUITAR_HERO_3_BUTTON_BLUE 0x0020
#define GUITAR_HERO_3_BUTTON_RED 0x0040
#define GUITAR_HERO_3_BUTTON_ORANGE 0x0080
#define GUITAR_HERO_3_BUTTON_PLUS 0x0400
#define GUITAR_HERO_3_BUTTON_MINUS 0x1000
#define GUITAR_HERO_3_BUTTON_STRUM_DOWN 0x4000
#define GUITAR_HERO_3_BUTTON_ALL 0xFEFF

/* guitar hero world tour touch bar codes */
#define GUITAR_HERO_3_TOUCH_AVAILABLE 0x1000
#define GUITAR_HERO_3_TOUCH_GREEN 0x1001
#define GUITAR_HERO_3_TOUCH_RED 0x1002
#define GUITAR_HERO_3_TOUCH_YELLOW 0x1004
#define GUITAR_HERO_3_TOUCH_BLUE 0x1008
#define GUITAR_HERO_3_TOUCH_ORANGE 0x1010

/* wiimote option flags */
#define WIIUSE_SMOOTHING 0x01
#define WIIUSE_CONTINUOUS 0x02
#define WIIUSE_ACCEL_THRESH 0x04
#define WIIUSE_IR_THRESH 0x08
#define WIIUSE_JS_THRESH 0x10
#define WIIUSE_INIT_FLAGS WIIUSE_SMOOTHING

#define WIIUSE_ORIENT_PRECISION 100.0f

/* expansion codes */
#define EXP_NONE 0
#define EXP_NUNCHUK 1
#define EXP_CLASSIC 2
#define EXP_GUITAR_HERO_3 3
#define EXP_WII_BOARD 4
#define EXP_MOTION_PLUS 5

/* IR correction types */
typedef enum ir_position_t {
 WIIUSE_IR_ABOVE,
 WIIUSE_IR_BELOW
} ir_position_t;

#define IS_PRESSED(dev, button) ((dev->btns & button) == button)

#define IS_HELD(dev, button) ((dev->btns_held & button) == button)

#define IS_RELEASED(dev, button) ((dev->btns_released & button) == button)

#define IS_JUST_PRESSED(dev, button) (IS_PRESSED(dev, button) && !IS_HELD(dev, button))

#define WIIUSE_GET_IR_SENSITIVITY(dev, lvl) \
 do { \
 if ((wm->state & 0x01000) == 0x01000) *lvl = 1; \
 else if ((wm->state & 0x02000) == 0x02000) *lvl = 2; \
 else if ((wm->state & 0x04000) == 0x04000) *lvl = 3; \
 else if ((wm->state & 0x08000) == 0x08000) *lvl = 4; \
 else if ((wm->state & 0x10000) == 0x10000) *lvl = 5; \
 else *lvl = 0; \
 } while (0)

#define WIIUSE_USING_ACC(wm) ((wm->state & 0x00100) == 0x00100)
#define WIIUSE_USING_EXP(wm) ((wm->state & 0x00200) == 0x00200)
#define WIIUSE_USING_IR(wm) ((wm->state & 0x00400) == 0x00400)
#define WIIUSE_USING_SPEAKER(wm) ((wm->state & 0x00800) == 0x00800)

#define WIIUSE_IS_LED_SET(wm, num) ((wm->leds & WIIMOTE_LED_##num) == WIIMOTE_LED_##num)

/*
 * Largest known payload is 21 bytes.
 * Add 2 for the prefix and round up to a power of 2.
 */
#define MAX_PAYLOAD 32

/*
 * This is left over from an old hack, but it may actually
 * be a useful feature to keep so it wasn't removed.
 */
#ifdef WIN32
 #define WIIMOTE_DEFAULT_TIMEOUT 100
 #define WIIMOTE_EXP_TIMEOUT 100
#endif

typedef unsigned char ubyte;
typedef char sbyte;
typedef unsigned short uword;
typedef short sword;
typedef unsigned int uint;
typedef char sint;


struct wiimote_t;
struct vec3b_t;
struct orient_t;
struct gforce_t;

#ifdef GEKKO
 typedef void (*wii_event_cb)(struct wiimote_t*, s32 event);
#endif

typedef void (*wiiuse_data_cb)(struct wiimote_t* wm, ubyte* data, unsigned short len);

typedef enum data_req_s
{
 REQ_READY = 0,
 REQ_SENT,
 REQ_DONE
} data_req_s;

struct data_req_t {
 lwp_node node;
 ubyte data[48]; 
 unsigned int len;
 data_req_s state; 
 wiiuse_data_cb cb; 
 struct data_req_t *next;
};

typedef void (*cmd_blk_cb)(struct wiimote_t *wm,ubyte *data,uword len);

typedef enum cmd_blk_s
{
 CMD_READY = 0,
 CMD_SENT,
 CMD_DONE
} cmd_blk_s;

struct cmd_blk_t
{
 lwp_node node;

 ubyte data[48];
 uint len;

 cmd_blk_s state;
 cmd_blk_cb cb;

 struct cmd_blk_t *next;
};


typedef struct vec2b_t {
 ubyte x, y;
} vec2b_t;


typedef struct vec3b_t {
 ubyte x, y, z;
} vec3b_t;

typedef struct vec3w_t {
 uword x, y, z;
} vec3w_t;


typedef struct vec3f_t {
 float x, y, z;
} vec3f_t;


typedef struct orient_t {
 float roll; 
 float pitch; 
 float yaw;

 float a_roll; 
 float a_pitch; 
} orient_t;


typedef struct gforce_t {
 float x, y, z;
} gforce_t;


typedef struct accel_t {
 struct vec3w_t cal_zero; 
 struct vec3w_t cal_g; 
 float st_roll; 
 float st_pitch; 
 float st_alpha; 
} accel_t;


typedef struct ir_dot_t {
 ubyte visible; 
 short rx; 
 short ry; 
 ubyte size; 
} ir_dot_t;


typedef struct fdot_t {
 float x,y;
} fdot_t;

typedef struct sb_t {
 fdot_t dots[2];
 fdot_t acc_dots[2];
 fdot_t rot_dots[2];
 float angle;
 float off_angle;
 float score;
} sb_t;

typedef enum aspect_t {
 WIIUSE_ASPECT_4_3,
 WIIUSE_ASPECT_16_9
} aspect_t;


typedef struct ir_t {
 struct ir_dot_t dot[4]; 
 ubyte num_dots; 
 int state; 
 int raw_valid; 
 sb_t sensorbar; 
 float ax; 
 float ay; 
 float distance; 
 float z; 
 float angle; 
 int smooth_valid; 
 float sx; 
 float sy; 
 float error_cnt; 
 float glitch_cnt; 
 int valid; 
 float x; 
 float y; 
 enum aspect_t aspect; 
 enum ir_position_t pos; 
 unsigned int vres[2]; 
 int offset[2]; 
} ir_t;


typedef struct joystick_t {
 struct vec2b_t max; 
 struct vec2b_t min; 
 struct vec2b_t center; 
 struct vec2b_t pos; 
 float ang; 
 float mag; 
} joystick_t;


typedef struct nunchuk_t {
 struct accel_t accel_calib; 
 struct joystick_t js; 
 int* flags; 
 ubyte btns; 
 ubyte btns_last; 
 ubyte btns_held; 
 ubyte btns_released; 
 struct vec3w_t accel; 
 struct orient_t orient; 
 struct gforce_t gforce; 
} nunchuk_t;


typedef struct classic_ctrl_t {
 short btns; 
 short btns_last; 
 short btns_held; 
 short btns_released; 
 ubyte rs_raw;
 ubyte ls_raw;

 float r_shoulder; 
 float l_shoulder; 
 struct joystick_t ljs; 
 struct joystick_t rjs; 
 ubyte type; 
} classic_ctrl_t;


typedef struct guitar_hero_3_t {
 short btns; 
 short btns_last; 
 short btns_held; 
 short btns_released; 
 ubyte wb_raw;
 float whammy_bar; 
 ubyte tb_raw;
 int touch_bar; 
 struct joystick_t js; 
} guitar_hero_3_t;

typedef struct wii_board_t {
 float tl; /* Interpolated */
 float tr;
 float bl;
 float br; /* End interp */
 short rtl; /* RAW */
 short rtr;
 short rbl;
 short rbr; /* /RAW */
 short ctl[3]; /* Calibration */
 short ctr[3];
 short cbl[3];
 short cbr[3]; /* /Calibration */
 float x;
 float y;
} wii_board_t;

typedef struct motion_plus_t
{
 short rx, ry, rz;
 ubyte status;
 ubyte ext;
} motion_plus_t;

typedef struct expansion_t {
 int type; 
 union {
 struct nunchuk_t nunchuk;
 struct classic_ctrl_t classic;
 struct guitar_hero_3_t gh3;
 struct wii_board_t wb;
 struct motion_plus_t mp;
 };
} expansion_t;


typedef enum win_bt_stack_t {
 WIIUSE_STACK_UNKNOWN,
 WIIUSE_STACK_MS,
 WIIUSE_STACK_BLUESOLEIL
} win_bt_stack_t;


typedef struct wiimote_state_t {
 unsigned short btns;

 struct ir_t ir;
 struct vec3w_t accel;
 struct expansion_t exp;
} wiimote_state_t;


typedef enum WIIUSE_EVENT_TYPE {
 WIIUSE_NONE = 0,
 WIIUSE_EVENT,
 WIIUSE_STATUS,
 WIIUSE_CONNECT,
 WIIUSE_DISCONNECT,
 WIIUSE_UNEXPECTED_DISCONNECT,
 WIIUSE_READ_DATA,
 WIIUSE_ACK,
 WIIUSE_NUNCHUK_INSERTED,
 WIIUSE_NUNCHUK_REMOVED,
 WIIUSE_CLASSIC_CTRL_INSERTED,
 WIIUSE_CLASSIC_CTRL_REMOVED,
 WIIUSE_GUITAR_HERO_3_CTRL_INSERTED,
 WIIUSE_GUITAR_HERO_3_CTRL_REMOVED,
 WIIUSE_WII_BOARD_INSERTED,
 WIIUSE_WII_BOARD_REMOVED,
 WIIUSE_MOTION_PLUS_ACTIVATED,
 WIIUSE_MOTION_PLUS_REMOVED
} WIIUSE_EVENT_TYPE;

typedef struct wiimote_t {
 WCONST int unid; 
 #if defined(_WIN32)
 WCONST HANDLE dev_handle; 
 WCONST OVERLAPPED hid_overlap; 
 WCONST enum win_bt_stack_t stack; 
 WCONST int timeout; 
 WCONST ubyte normal_timeout; 
 WCONST ubyte exp_timeout; 
 #elif defined(GEKKO)
 WCONST lwp_queue cmdq;
 WCONST struct bd_addr bdaddr; 
 WCONST char bdaddr_str[18]; 
 WCONST struct bte_pcb *sock; 
 WCONST wii_event_cb event_cb; 
 #elif defined(unix)
 WCONST bdaddr_t bdaddr; 
 WCONST char bdaddr_str[18]; 
 WCONST int out_sock; 
 WCONST int in_sock; 
 #endif

 WCONST int state; 
 WCONST ubyte leds; 
#ifdef GEKKO
 WCONST ubyte battery_level; 
#else
 WCONST float battery_level; 
#endif

 WCONST int flags; 
 WCONST ubyte handshake_state; 
 WCONST ubyte expansion_state; 
 WCONST struct data_req_t* data_req; 
 WCONST struct cmd_blk_t *cmd_head;
 WCONST struct cmd_blk_t *cmd_tail;

 WCONST struct accel_t accel_calib; 
 WCONST struct expansion_t exp; 
 WCONST struct vec3w_t accel; 
 WCONST struct orient_t orient; 
 WCONST struct gforce_t gforce; 
 WCONST struct ir_t ir; 
 WCONST unsigned short btns; 
 WCONST unsigned short btns_last; 
 WCONST unsigned short btns_held; 
 WCONST unsigned short btns_released; 
 WCONST struct wiimote_state_t lstate; 
 WCONST WIIUSE_EVENT_TYPE event; 
 WCONST ubyte event_buf[MAX_PAYLOAD]; 
 WCONST ubyte motion_plus_id[6];
} wiimote;

#if defined(GEKKO)

typedef struct wiimote_listen_t {
 WCONST struct bd_addr bdaddr;
 WCONST struct bte_pcb *sock;
 WCONST struct wiimote_t *(*assign_cb)(struct bd_addr *bdaddr);
 WCONST struct wiimote_t *wm;
} wiimote_listen;
#endif

/*****************************************
 *
 * Include API specific stuff
 *
 *****************************************/

#ifdef _WIN32
 #define WIIUSE_EXPORT_DECL __declspec(dllexport)
 #define WIIUSE_IMPORT_DECL __declspec(dllimport)
#else
 #define WIIUSE_EXPORT_DECL
 #define WIIUSE_IMPORT_DECL
#endif

#ifdef WIIUSE_COMPILE_LIB
 #define WIIUSE_EXPORT WIIUSE_EXPORT_DECL
#else
 #define WIIUSE_EXPORT WIIUSE_IMPORT_DECL
#endif

#ifdef __cplusplus
extern "C" {
#endif

/* wiiuse.c */
WIIUSE_EXPORT extern const char* wiiuse_version();

#ifndef GEKKO
WIIUSE_EXPORT extern struct wiimote_t** wiiuse_init(int wiimotes);
#else
WIIUSE_EXPORT extern int wiiuse_register(struct wiimote_listen_t *wml, struct bd_addr *bdaddr, struct wiimote_t *(*assign_cb)(struct bd_addr *bdaddr));
WIIUSE_EXPORT extern struct wiimote_t** wiiuse_init(int wiimotes, wii_event_cb event_cb);
WIIUSE_EXPORT extern void wiiuse_sensorbar_enable(int enable);
#endif

WIIUSE_EXPORT extern void wiiuse_disconnected(struct wiimote_t* wm);
WIIUSE_EXPORT extern void wiiuse_cleanup(struct wiimote_t** wm, int wiimotes);
WIIUSE_EXPORT extern void wiiuse_rumble(struct wiimote_t* wm, int status);
WIIUSE_EXPORT extern void wiiuse_toggle_rumble(struct wiimote_t* wm);
WIIUSE_EXPORT extern void wiiuse_set_leds(struct wiimote_t* wm, int leds,cmd_blk_cb cb);
WIIUSE_EXPORT extern void wiiuse_motion_sensing(struct wiimote_t* wm, int status);
WIIUSE_EXPORT extern int wiiuse_read_data(struct wiimote_t* wm, ubyte* buffer, unsigned int offset, unsigned short len, cmd_blk_cb cb);
WIIUSE_EXPORT extern int wiiuse_write_data(struct wiimote_t *wm,unsigned int addr,ubyte *data,ubyte len,cmd_blk_cb cb);
WIIUSE_EXPORT extern void wiiuse_status(struct wiimote_t *wm,cmd_blk_cb cb);
WIIUSE_EXPORT extern struct wiimote_t* wiiuse_get_by_id(struct wiimote_t** wm, int wiimotes, int unid);
WIIUSE_EXPORT extern int wiiuse_set_flags(struct wiimote_t* wm, int enable, int disable);
WIIUSE_EXPORT extern float wiiuse_set_smooth_alpha(struct wiimote_t* wm, float alpha);
WIIUSE_EXPORT extern void wiiuse_set_bluetooth_stack(struct wiimote_t** wm, int wiimotes, enum win_bt_stack_t type);
WIIUSE_EXPORT extern void wiiuse_resync(struct wiimote_t* wm);
WIIUSE_EXPORT extern void wiiuse_set_timeout(struct wiimote_t** wm, int wiimotes, ubyte normal_timeout, ubyte exp_timeout);
WIIUSE_EXPORT extern int wiiuse_write_streamdata(struct wiimote_t *wm,ubyte *data,ubyte len,cmd_blk_cb cb);

/* connect.c */
WIIUSE_EXPORT extern int wiiuse_find(struct wiimote_t** wm, int max_wiimotes, int timeout);
WIIUSE_EXPORT extern int wiiuse_connect(struct wiimote_t** wm, int wiimotes);
WIIUSE_EXPORT extern void wiiuse_disconnect(struct wiimote_t* wm);

/* events.c */
WIIUSE_EXPORT extern int wiiuse_poll(struct wiimote_t** wm, int wiimotes);

/* ir.c */
WIIUSE_EXPORT extern void wiiuse_set_ir_mode(struct wiimote_t *wm);
WIIUSE_EXPORT extern void wiiuse_set_ir(struct wiimote_t* wm, int status);
WIIUSE_EXPORT extern void wiiuse_set_ir_vres(struct wiimote_t* wm, unsigned int x, unsigned int y);
WIIUSE_EXPORT extern void wiiuse_set_ir_position(struct wiimote_t* wm, enum ir_position_t pos);
WIIUSE_EXPORT extern void wiiuse_set_aspect_ratio(struct wiimote_t* wm, enum aspect_t aspect);
WIIUSE_EXPORT extern void wiiuse_set_ir_sensitivity(struct wiimote_t* wm, int level);

/* motion_plus.c */
WIIUSE_EXPORT extern void wiiuse_set_motion_plus(struct wiimote_t *wm, int status);

/* speaker.c */
WIIUSE_EXPORT extern void wiiuse_set_speaker(struct wiimote_t *wm, int status);

#ifdef __cplusplus
}
#endif

#endif