Skip to content

Commit

Permalink
Split drivers into seperate files
Browse files Browse the repository at this point in the history
  • Loading branch information
@sanjay900
sanjay900 committed Oct 17, 2023
1 parent e308bd0 commit 3980502
Show file tree
Hide file tree
Showing 6 changed files with 802 additions and 540 deletions.
3 changes: 3 additions & 0 deletions include/usb_device_drivers.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,5 +31,8 @@ static inline bool usb_driver_is_comaptible(u16 vid, u16 pid, const struct devic

extern const usb_device_driver_t ds3_usb_device_driver;
extern const usb_device_driver_t ds4_usb_device_driver;
extern const usb_device_driver_t gh_guitar_usb_device_driver;
extern const usb_device_driver_t gh_drum_usb_device_driver;
extern const usb_device_driver_t turntable_usb_device_driver;

#endif
262 changes: 262 additions & 0 deletions source/usb_drivers/dj_hero_turntable.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,262 @@
#include "button_map.h"
#include "usb_device_drivers.h"
#include "usb.h"
#include "utils.h"
#include "wiimote.h"

struct turntable_input_report {
u8 : 4;
u8 triangle_euphoria : 1;
u8 circle : 1;
u8 cross : 1;
u8 square : 1;

u8 : 3;
u8 ps : 1;
u8 : 2;
u8 start : 1;
u8 select : 1;

u8 hat;
u8 unused;
u8 unused2;
u8 left_turn_table_velocity;
u8 right_turn_table_velocity;
u8 pressure_yellow;
u8 pressure_red;
u8 pressure_green;
u8 pressure_blue;
u8 pressure_kick;
u8 pressure_orange;
u8 unused3[2];
u16 effects_knob;
u16 cross_fader;
u16 right_green : 1;
u16 right_red : 1;
u16 right_blue : 1;
u16 left_green : 1;
u16 left_red : 1;
u16 left_blue : 1;
u16 : 3;
u16 table_neutral : 1;
u16 : 6;
u16 : 16;

} ATTRIBUTE_PACKED;

enum turntable_buttons_e {
TURNTABLE_BUTTON_SQUARE,
TURNTABLE_BUTTON_CROSS,
TURNTABLE_BUTTON_CIRCLE,
TURNTABLE_BUTTON_TRIANGLE_EUPHORIA,
TURNTABLE_BUTTON_SELECT,
TURNTABLE_BUTTON_START,
TURNTABLE_BUTTON_LEFT_GREEN,
TURNTABLE_BUTTON_LEFT_RED,
TURNTABLE_BUTTON_LEFT_BLUE,
TURNTABLE_BUTTON_RIGHT_GREEN,
TURNTABLE_BUTTON_RIGHT_RED,
TURNTABLE_BUTTON_RIGHT_BLUE,
TURNTABLE_BUTTON_UP,
TURNTABLE_BUTTON_DOWN,
TURNTABLE_BUTTON_LEFT,
TURNTABLE_BUTTON_RIGHT,
TURNTABLE_BUTTON_PS,
TURNTABLE_BUTTON__NUM
};

enum turntable_analog_axis_e {
TURNTABLE_ANALOG_AXIS_LEFT_VELOCITY,
TURNTABLE_ANALOG_AXIS_RIGHT_VELOCITY,
TURNTABLE_ANALOG_AXIS_CROSS_FADER,
TURNTABLE_ANALOG_AXIS_EFFECTS_KNOB,
TURNTABLE_ANALOG_AXIS__NUM
};

#define MAX_ANALOG_AXIS TURNTABLE_ANALOG_AXIS__NUM

struct turntable_private_data_t {
struct {
u32 buttons;
u8 analog_axis[MAX_ANALOG_AXIS];
} input;
u8 leds;
};
static_assert(sizeof(struct turntable_private_data_t) <= USB_INPUT_DEVICE_PRIVATE_DATA_SIZE);

static const struct {
u16 wiimote_button_map[TURNTABLE_BUTTON__NUM];
u16 turntable_button_map[TURNTABLE_BUTTON__NUM];
u8 turntable_analog_axis_map[TURNTABLE_ANALOG_AXIS__NUM];
} turntable_mapping =
{
.wiimote_button_map = {
[TURNTABLE_BUTTON_PS] = WIIMOTE_BUTTON_HOME,
},
.turntable_analog_axis_map = {
[TURNTABLE_ANALOG_AXIS_LEFT_VELOCITY] = BM_TURNTABLE_ANALOG_LEFT_TURNTABLE_VELOCITY,
[TURNTABLE_ANALOG_AXIS_RIGHT_VELOCITY] = BM_TURNTABLE_ANALOG_RIGHT_TURNTABLE_VELOCITY,
[TURNTABLE_ANALOG_AXIS_CROSS_FADER] = BM_TURNTABLE_ANALOG_CROSS_FADER,
[TURNTABLE_ANALOG_AXIS_EFFECTS_KNOB] = BM_TURNTABLE_ANALOG_EFFECTS_DIAL,
},
.turntable_button_map = {
[TURNTABLE_BUTTON_LEFT_GREEN] = TURNTABLE_CTRL_BUTTON_LEFT_GREEN,
[TURNTABLE_BUTTON_LEFT_RED] = TURNTABLE_CTRL_BUTTON_LEFT_RED,
[TURNTABLE_BUTTON_LEFT_BLUE] = TURNTABLE_CTRL_BUTTON_LEFT_BLUE,
[TURNTABLE_BUTTON_RIGHT_GREEN] = TURNTABLE_CTRL_BUTTON_RIGHT_GREEN,
[TURNTABLE_BUTTON_RIGHT_RED] = TURNTABLE_CTRL_BUTTON_RIGHT_RED,
[TURNTABLE_BUTTON_RIGHT_BLUE] = TURNTABLE_CTRL_BUTTON_RIGHT_BLUE,
[TURNTABLE_BUTTON_CROSS] = TURNTABLE_CTRL_BUTTON_LEFT_GREEN | TURNTABLE_CTRL_BUTTON_RIGHT_GREEN,
[TURNTABLE_BUTTON_CIRCLE] = TURNTABLE_CTRL_BUTTON_LEFT_RED | TURNTABLE_CTRL_BUTTON_RIGHT_RED,
[TURNTABLE_BUTTON_SQUARE] = TURNTABLE_CTRL_BUTTON_LEFT_BLUE | TURNTABLE_CTRL_BUTTON_RIGHT_BLUE,
[TURNTABLE_BUTTON_START] = TURNTABLE_CTRL_BUTTON_PLUS,
[TURNTABLE_BUTTON_SELECT] = TURNTABLE_CTRL_BUTTON_MINUS,
}};

static inline void turntable_get_buttons(const struct turntable_input_report *report, u32 *buttons) {
u32 mask = 0;

#define MAP(field, button) \
if (report->field) \
mask |= BIT(button);

MAP(left_green, TURNTABLE_BUTTON_LEFT_GREEN)
MAP(left_red, TURNTABLE_BUTTON_LEFT_RED)
MAP(left_blue, TURNTABLE_BUTTON_LEFT_BLUE)
MAP(right_green, TURNTABLE_BUTTON_RIGHT_GREEN)
MAP(right_red, TURNTABLE_BUTTON_RIGHT_RED)
MAP(right_blue, TURNTABLE_BUTTON_RIGHT_BLUE)
MAP(cross, TURNTABLE_BUTTON_CROSS)
MAP(circle, TURNTABLE_BUTTON_CIRCLE)
MAP(square, TURNTABLE_BUTTON_SQUARE)
MAP(triangle_euphoria, TURNTABLE_BUTTON_TRIANGLE_EUPHORIA)
MAP(select, TURNTABLE_BUTTON_SELECT)
MAP(ps, TURNTABLE_BUTTON_PS)
#undef MAP
if (report->hat == 0 || report->hat == 1 || report->hat == 7) {
mask |= BIT(TURNTABLE_BUTTON_UP);
}
if (report->hat == 1 || report->hat == 2 || report->hat == 3) {
mask |= BIT(TURNTABLE_BUTTON_LEFT);
}
if (report->hat == 3 || report->hat == 4 || report->hat == 5) {
mask |= BIT(TURNTABLE_BUTTON_DOWN);
}
if (report->hat == 5 || report->hat == 6 || report->hat == 7) {
mask |= BIT(TURNTABLE_BUTTON_RIGHT);
}

*buttons = mask;
}


static inline void turntable_get_analog_axis(const struct turntable_input_report *report,
u8 analog_axis[static MAX_ANALOG_AXIS]) {
analog_axis[TURNTABLE_ANALOG_AXIS_CROSS_FADER] = report->cross_fader;
analog_axis[TURNTABLE_ANALOG_AXIS_EFFECTS_KNOB] = report->effects_knob;
analog_axis[TURNTABLE_ANALOG_AXIS_LEFT_VELOCITY] = report->left_turn_table_velocity;
analog_axis[TURNTABLE_ANALOG_AXIS_RIGHT_VELOCITY] = report->right_turn_table_velocity;
}


static inline int turntable_request_data(usb_input_device_t *device)
{
return usb_device_driver_issue_intr_transfer_async(device, device->endpoint_address, device->usb_async_resp,
device->max_packet_len);
}

static int turntable_driver_update_leds(usb_input_device_t *device)
{
// TODO: this
return 0;
}


bool turntable_driver_ops_probe(u16 vid, u16 pid)
{
static const struct device_id_t compatible[] = {
{SONY_INST_VID, DJ_TURNTABLE_PID}
};

return usb_driver_is_comaptible(vid, pid, compatible, ARRAY_SIZE(compatible));
}

int turntable_driver_ops_init(usb_input_device_t *device, u16 vid, u16 pid)
{
int ret;
struct turntable_private_data_t *priv = (void *)device->private_data;

priv->leds = 0;

fake_wiimote_set_extension(device->wiimote, WIIMOTE_EXT_TURNTABLE);

ret = turntable_request_data(device);
if (ret < 0)
return ret;

return 0;
}

int turntable_driver_ops_disconnect(usb_input_device_t *device)
{
struct turntable_private_data_t *priv = (void *)device->private_data;

priv->leds = 0;

return turntable_driver_update_leds(device);
}

int turntable_driver_ops_slot_changed(usb_input_device_t *device, u8 slot)
{
struct turntable_private_data_t *priv = (void *)device->private_data;

priv->leds = slot;

return turntable_driver_update_leds(device);
}

bool turntable_report_input(usb_input_device_t *device)
{
struct turntable_private_data_t *priv = (void *)device->private_data;
u16 wiimote_buttons = 0;
u16 acc_x, acc_y, acc_z;
union wiimote_extension_data_t extension_data;

bm_map_wiimote(TURNTABLE_BUTTON__NUM, priv->input.buttons,
turntable_mapping.wiimote_button_map,
&wiimote_buttons);
acc_x = ACCEL_ZERO_G;
acc_y = ACCEL_ZERO_G;
acc_z = ACCEL_ZERO_G;

fake_wiimote_report_accelerometer(device->wiimote, acc_x, acc_y, acc_z);

bm_map_turntable(TURNTABLE_BUTTON__NUM, priv->input.buttons,
TURNTABLE_ANALOG_AXIS__NUM, priv->input.analog_axis,
turntable_mapping.turntable_button_map,
turntable_mapping.turntable_analog_axis_map,
&extension_data.turntable);
fake_wiimote_report_input_ext(device->wiimote, wiimote_buttons,
&extension_data, sizeof(extension_data.turntable));

return true;
}

int turntable_driver_ops_usb_async_resp(usb_input_device_t *device)
{
struct turntable_private_data_t *priv = (void *)device->private_data;
struct turntable_input_report *report = (void *)device->usb_async_resp;
turntable_get_buttons(report, &priv->input.buttons);
turntable_get_analog_axis(report, priv->input.analog_axis);

return turntable_request_data(device);
}

const usb_device_driver_t turntable_usb_device_driver = {
.probe = turntable_driver_ops_probe,
.init = turntable_driver_ops_init,
.disconnect = turntable_driver_ops_disconnect,
.slot_changed = turntable_driver_ops_slot_changed,
.report_input = turntable_report_input,
.usb_async_resp = turntable_driver_ops_usb_async_resp,
};
Loading

0 comments on commit 3980502

Please sign in to comment.