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