| /* |
| Simple DirectMedia Layer |
| Copyright (C) 1997-2020 Sam Lantinga <slouken@libsdl.org> |
| |
| This software is provided 'as-is', without any express or implied |
| warranty. In no event will the authors be held liable for any damages |
| arising from the use of this software. |
| |
| Permission is granted to anyone to use this software for any purpose, |
| including commercial applications, and to alter it and redistribute it |
| freely, subject to the following restrictions: |
| |
| 1. The origin of this software must not be misrepresented; you must not |
| claim that you wrote the original software. If you use this software |
| in a product, an acknowledgment in the product documentation would be |
| appreciated but is not required. |
| 2. Altered source versions must be plainly marked as such, and must not be |
| misrepresented as being the original software. |
| 3. This notice may not be removed or altered from any source distribution. |
| */ |
| #include "../../SDL_internal.h" |
| |
| #ifdef SDL_JOYSTICK_HIDAPI |
| |
| #include "SDL_hints.h" |
| #include "SDL_events.h" |
| #include "SDL_joystick.h" |
| #include "SDL_gamecontroller.h" |
| #include "../SDL_sysjoystick.h" |
| #include "SDL_hidapijoystick_c.h" |
| #include "SDL_hidapi_rumble.h" |
| |
| |
| #ifdef SDL_JOYSTICK_HIDAPI_PS5 |
| |
| /* Define this if you want to log all packets from the controller */ |
| /*#define DEBUG_PS5_PROTOCOL*/ |
| |
| /* Define this if you want to log calibration data */ |
| /*#define DEBUG_PS5_CALIBRATION*/ |
| |
| #define GYRO_RES_PER_DEGREE 1024.0f |
| #define ACCEL_RES_PER_G 8192.0f |
| |
| #define LOAD16(A, B) (Sint16)((Uint16)(A) | (((Uint16)(B)) << 8)) |
| |
| typedef enum |
| { |
| k_EPS5ReportIdState = 0x01, |
| k_EPS5ReportIdUsbEffects = 0x02, |
| k_EPS5ReportIdBluetoothEffects = 0x31, |
| k_EPS5ReportIdBluetoothState = 0x31, |
| } EPS5ReportId; |
| |
| typedef enum |
| { |
| k_EPS5FeatureReportIdCalibration = 0x05, |
| k_EPS5FeatureReportIdSerialNumber = 0x09, |
| } EPS5FeatureReportId; |
| |
| typedef struct |
| { |
| Uint8 ucLeftJoystickX; |
| Uint8 ucLeftJoystickY; |
| Uint8 ucRightJoystickX; |
| Uint8 ucRightJoystickY; |
| Uint8 rgucButtonsHatAndCounter[3]; |
| Uint8 ucTriggerLeft; |
| Uint8 ucTriggerRight; |
| } PS5SimpleStatePacket_t; |
| |
| typedef struct |
| { |
| Uint8 ucLeftJoystickX; /* 0 */ |
| Uint8 ucLeftJoystickY; /* 1 */ |
| Uint8 ucRightJoystickX; /* 2 */ |
| Uint8 ucRightJoystickY; /* 3 */ |
| Uint8 ucTriggerLeft; /* 4 */ |
| Uint8 ucTriggerRight; /* 5 */ |
| Uint8 ucCounter; /* 6 */ |
| Uint8 rgucButtonsAndHat[3]; /* 7 */ |
| Uint8 ucZero; /* 10 */ |
| Uint8 rgucPacketSequence[4]; /* 11 - 32 bit little endian */ |
| Uint8 rgucGyroX[2]; /* 15 */ |
| Uint8 rgucGyroY[2]; /* 17 */ |
| Uint8 rgucGyroZ[2]; /* 19 */ |
| Uint8 rgucAccelX[2]; /* 21 */ |
| Uint8 rgucAccelY[2]; /* 23 */ |
| Uint8 rgucAccelZ[2]; /* 25 */ |
| Uint8 rgucTimer1[4]; /* 27 - 32 bit little endian */ |
| Uint8 ucBatteryTemp; /* 31 */ |
| Uint8 ucTouchpadCounter1; /* 32 - high bit clear + counter */ |
| Uint8 rgucTouchpadData1[3]; /* 33 - X/Y, 12 bits per axis */ |
| Uint8 ucTouchpadCounter2; /* 36 - high bit clear + counter */ |
| Uint8 rgucTouchpadData2[3]; /* 37 - X/Y, 12 bits per axis */ |
| Uint8 rgucUnknown1[8]; /* 40 */ |
| Uint8 rgucTimer2[4]; /* 48 - 32 bit little endian */ |
| Uint8 ucBatteryLevel; /* 52 */ |
| Uint8 ucConnectState; /* 53 - 0x08 = USB, 0x01 = headphone */ |
| |
| /* There's more unknown data at the end, and a 32-bit CRC on Bluetooth */ |
| } PS5StatePacket_t; |
| |
| typedef struct |
| { |
| Uint8 ucEnableBits1; /* 0 */ |
| Uint8 ucEnableBits2; /* 1 */ |
| Uint8 ucRumbleRight; /* 2 */ |
| Uint8 ucRumbleLeft; /* 3 */ |
| Uint8 ucHeadphoneVolume; /* 4 */ |
| Uint8 ucSpeakerVolume; /* 5 */ |
| Uint8 ucMicrophoneVolume; /* 6 */ |
| Uint8 ucAudioEnableBits; /* 7 */ |
| Uint8 ucMicLightMode; /* 8 */ |
| Uint8 ucAudioMuteBits; /* 9 */ |
| Uint8 rgucRightTriggerEffect[11]; /* 10 */ |
| Uint8 rgucLeftTriggerEffect[11]; /* 21 */ |
| Uint8 rgucUnknown1[6]; /* 32 */ |
| Uint8 ucLedFlags; /* 38 */ |
| Uint8 rgucUnknown2[2]; /* 39 */ |
| Uint8 ucLedAnim; /* 41 */ |
| Uint8 ucLedBrightness; /* 42 */ |
| Uint8 ucPadLights; /* 43 */ |
| Uint8 ucLedRed; /* 44 */ |
| Uint8 ucLedGreen; /* 45 */ |
| Uint8 ucLedBlue; /* 46 */ |
| } DS5EffectsState_t; |
| |
| typedef enum { |
| k_EDS5EffectNone, |
| k_EDS5EffectRumbleStart, |
| k_EDS5EffectRumble, |
| k_EDS5EffectLED, |
| k_EDS5EffectPadLights, |
| k_EDS5EffectMicLight, |
| } EDS5Effect; |
| |
| typedef struct { |
| Sint16 bias; |
| float sensitivity; |
| } IMUCalibrationData; |
| |
| typedef struct { |
| SDL_bool is_bluetooth; |
| SDL_bool report_sensors; |
| SDL_bool hardware_calibration; |
| IMUCalibrationData calibration[6]; |
| int player_index; |
| Uint8 rumble_left; |
| Uint8 rumble_right; |
| SDL_bool color_set; |
| Uint8 led_red; |
| Uint8 led_green; |
| Uint8 led_blue; |
| union |
| { |
| PS5SimpleStatePacket_t simple; |
| PS5StatePacket_t state; |
| } last_state; |
| } SDL_DriverPS5_Context; |
| |
| |
| static SDL_bool |
| HIDAPI_DriverPS5_IsSupportedDevice(const char *name, SDL_GameControllerType type, Uint16 vendor_id, Uint16 product_id, Uint16 version, int interface_number, int interface_class, int interface_subclass, int interface_protocol) |
| { |
| return (type == SDL_CONTROLLER_TYPE_PS5); |
| } |
| |
| static const char * |
| HIDAPI_DriverPS5_GetDeviceName(Uint16 vendor_id, Uint16 product_id) |
| { |
| if (vendor_id == USB_VENDOR_SONY) { |
| return "PS5 Controller"; |
| } |
| return NULL; |
| } |
| |
| static int ReadFeatureReport(hid_device *dev, Uint8 report_id, Uint8 *report, size_t length) |
| { |
| SDL_memset(report, 0, length); |
| report[0] = report_id; |
| return hid_get_feature_report(dev, report, length); |
| } |
| |
| static void |
| SetLedsForPlayerIndex(DS5EffectsState_t *effects, int player_index) |
| { |
| /* This list is the same as what hid-sony.c uses in the Linux kernel. |
| The first 4 values correspond to what the PS4 assigns. |
| */ |
| static const Uint8 colors[7][3] = { |
| { 0x00, 0x00, 0x40 }, /* Blue */ |
| { 0x40, 0x00, 0x00 }, /* Red */ |
| { 0x00, 0x40, 0x00 }, /* Green */ |
| { 0x20, 0x00, 0x20 }, /* Pink */ |
| { 0x02, 0x01, 0x00 }, /* Orange */ |
| { 0x00, 0x01, 0x01 }, /* Teal */ |
| { 0x01, 0x01, 0x01 } /* White */ |
| }; |
| |
| if (player_index >= 0) { |
| player_index %= SDL_arraysize(colors); |
| } else { |
| player_index = 0; |
| } |
| |
| effects->ucLedRed = colors[player_index][0]; |
| effects->ucLedGreen = colors[player_index][1]; |
| effects->ucLedBlue = colors[player_index][2]; |
| } |
| |
| static SDL_bool |
| HIDAPI_DriverPS5_InitDevice(SDL_HIDAPI_Device *device) |
| { |
| return HIDAPI_JoystickConnected(device, NULL, SDL_FALSE); |
| } |
| |
| static int |
| HIDAPI_DriverPS5_GetDevicePlayerIndex(SDL_HIDAPI_Device *device, SDL_JoystickID instance_id) |
| { |
| return -1; |
| } |
| |
| static void |
| HIDAPI_DriverPS5_LoadCalibrationData(SDL_HIDAPI_Device *device) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| int i, size; |
| Uint8 data[USB_PACKET_LENGTH]; |
| |
| size = ReadFeatureReport(device->dev, k_EPS5FeatureReportIdCalibration, data, sizeof(data)); |
| if (size < 35) { |
| #ifdef DEBUG_PS5_CALIBRATION |
| SDL_Log("Short read of calibration data: %d, ignoring calibration\n", size); |
| #endif |
| return; |
| } |
| |
| { |
| Sint16 sGyroPitchBias, sGyroYawBias, sGyroRollBias; |
| Sint16 sGyroPitchPlus, sGyroPitchMinus; |
| Sint16 sGyroYawPlus, sGyroYawMinus; |
| Sint16 sGyroRollPlus, sGyroRollMinus; |
| Sint16 sGyroSpeedPlus, sGyroSpeedMinus; |
| |
| Sint16 sAccXPlus, sAccXMinus; |
| Sint16 sAccYPlus, sAccYMinus; |
| Sint16 sAccZPlus, sAccZMinus; |
| |
| float flNumerator; |
| Sint16 sRange2g; |
| |
| #ifdef DEBUG_PS5_CALIBRATION |
| HIDAPI_DumpPacket("PS5 calibration packet: size = %d", data, size); |
| #endif |
| |
| sGyroPitchBias = LOAD16(data[1], data[2]); |
| sGyroYawBias = LOAD16(data[3], data[4]); |
| sGyroRollBias = LOAD16(data[5], data[6]); |
| |
| sGyroPitchPlus = LOAD16(data[7], data[8]); |
| sGyroPitchMinus = LOAD16(data[9], data[10]); |
| sGyroYawPlus = LOAD16(data[11], data[12]); |
| sGyroYawMinus = LOAD16(data[13], data[14]); |
| sGyroRollPlus = LOAD16(data[15], data[16]); |
| sGyroRollMinus = LOAD16(data[17], data[18]); |
| |
| sGyroSpeedPlus = LOAD16(data[19], data[20]); |
| sGyroSpeedMinus = LOAD16(data[21], data[22]); |
| |
| sAccXPlus = LOAD16(data[23], data[24]); |
| sAccXMinus = LOAD16(data[25], data[26]); |
| sAccYPlus = LOAD16(data[27], data[28]); |
| sAccYMinus = LOAD16(data[29], data[30]); |
| sAccZPlus = LOAD16(data[31], data[32]); |
| sAccZMinus = LOAD16(data[33], data[34]); |
| |
| flNumerator = (sGyroSpeedPlus + sGyroSpeedMinus) * GYRO_RES_PER_DEGREE; |
| ctx->calibration[0].bias = sGyroPitchBias; |
| ctx->calibration[0].sensitivity = flNumerator / (sGyroPitchPlus - sGyroPitchMinus); |
| |
| ctx->calibration[1].bias = sGyroYawBias; |
| ctx->calibration[1].sensitivity = flNumerator / (sGyroYawPlus - sGyroYawMinus); |
| |
| ctx->calibration[2].bias = sGyroRollBias; |
| ctx->calibration[2].sensitivity = flNumerator / (sGyroRollPlus - sGyroRollMinus); |
| |
| sRange2g = sAccXPlus - sAccXMinus; |
| ctx->calibration[3].bias = sAccXPlus - sRange2g / 2; |
| ctx->calibration[3].sensitivity = 2.0f * ACCEL_RES_PER_G / (float)sRange2g; |
| |
| sRange2g = sAccYPlus - sAccYMinus; |
| ctx->calibration[4].bias = sAccYPlus - sRange2g / 2; |
| ctx->calibration[4].sensitivity = 2.0f * ACCEL_RES_PER_G / (float)sRange2g; |
| |
| sRange2g = sAccZPlus - sAccZMinus; |
| ctx->calibration[5].bias = sAccZPlus - sRange2g / 2; |
| ctx->calibration[5].sensitivity = 2.0f * ACCEL_RES_PER_G / (float)sRange2g; |
| |
| ctx->hardware_calibration = SDL_TRUE; |
| for (i = 0; i < 6; ++i) { |
| float divisor = (i < 3 ? 64.0f : 1.0f); |
| #ifdef DEBUG_PS5_CALIBRATION |
| SDL_Log("calibration[%d] bias = %d, sensitivity = %f\n", i, ctx->calibration[i].bias, ctx->calibration[i].sensitivity); |
| #endif |
| /* Some controllers have a bad calibration */ |
| if ((SDL_abs(ctx->calibration[i].bias) > 1024) || (SDL_fabs(1.0f - ctx->calibration[i].sensitivity / divisor) > 0.5f)) { |
| #ifdef DEBUG_PS5_CALIBRATION |
| SDL_Log("invalid calibration, ignoring\n"); |
| #endif |
| ctx->hardware_calibration = SDL_FALSE; |
| } |
| } |
| } |
| } |
| |
| static float |
| HIDAPI_DriverPS5_ApplyCalibrationData(SDL_DriverPS5_Context *ctx, int index, Sint16 value) |
| { |
| float result; |
| |
| if (ctx->hardware_calibration) { |
| IMUCalibrationData *calibration = &ctx->calibration[index]; |
| |
| result = (value - calibration->bias) * calibration->sensitivity; |
| } else { |
| result = value; |
| } |
| |
| /* Convert the raw data to the units expected by SDL */ |
| if (index < 3) { |
| result = (result / GYRO_RES_PER_DEGREE) * (float)M_PI / 180.0f; |
| } else { |
| result = (result / ACCEL_RES_PER_G) * SDL_STANDARD_GRAVITY; |
| } |
| return result; |
| } |
| |
| static int |
| HIDAPI_DriverPS5_UpdateEffects(SDL_HIDAPI_Device *device, EDS5Effect effect) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| DS5EffectsState_t *effects; |
| Uint8 data[78]; |
| int report_size, offset; |
| Uint8 *pending_data; |
| int *pending_size; |
| int maximum_size; |
| |
| |
| SDL_zero(data); |
| |
| if (ctx->is_bluetooth) { |
| data[0] = k_EPS5ReportIdBluetoothEffects; |
| data[1] = 0x02; /* Magic value */ |
| |
| report_size = 78; |
| offset = 2; |
| } else { |
| data[0] = k_EPS5ReportIdUsbEffects; |
| |
| report_size = 48; |
| offset = 1; |
| } |
| effects = (DS5EffectsState_t *)&data[offset]; |
| |
| if (ctx->rumble_left || ctx->rumble_right) { |
| effects->ucEnableBits1 |= 0x01; /* Enable rumble emulation */ |
| effects->ucEnableBits1 |= 0x02; /* Disable audio haptics */ |
| |
| /* Shift to reduce effective rumble strength to match Xbox controllers */ |
| effects->ucRumbleLeft = ctx->rumble_left >> 1; |
| effects->ucRumbleRight = ctx->rumble_right >> 1; |
| } else { |
| /* Leaving emulated rumble bits off will restore audio haptics */ |
| } |
| |
| switch (effect) { |
| case k_EDS5EffectRumbleStart: |
| effects->ucEnableBits1 |= 0x02; /* Disable audio haptics */ |
| break; |
| case k_EDS5EffectRumble: |
| /* Already handled above */ |
| break; |
| case k_EDS5EffectLED: |
| effects->ucEnableBits2 |= 0x04; /* Enable LED color */ |
| |
| /* Populate the LED state with the appropriate color from our lookup table */ |
| if (ctx->color_set) { |
| effects->ucLedRed = ctx->led_red; |
| effects->ucLedGreen = ctx->led_green; |
| effects->ucLedBlue = ctx->led_blue; |
| } else { |
| SetLedsForPlayerIndex(effects, ctx->player_index); |
| } |
| break; |
| case k_EDS5EffectPadLights: |
| effects->ucEnableBits2 |= 0x10; /* Enable touchpad lights */ |
| |
| effects->ucPadLights = 0x00; /* Bitmask, 0x1F enables all lights, 0x20 changes instantly instead of fade */ |
| break; |
| case k_EDS5EffectMicLight: |
| effects->ucEnableBits2 |= 0x01; /* Enable microphone light */ |
| |
| effects->ucMicLightMode = 0; /* Bitmask, 0x00 = off, 0x01 = solid, 0x02 = pulse */ |
| break; |
| default: |
| break; |
| } |
| |
| if (ctx->is_bluetooth) { |
| /* Bluetooth reports need a CRC at the end of the packet (at least on Linux) */ |
| Uint8 ubHdr = 0xA2; /* hidp header is part of the CRC calculation */ |
| Uint32 unCRC; |
| unCRC = SDL_crc32(0, &ubHdr, 1); |
| unCRC = SDL_crc32(unCRC, data, (size_t)(report_size - sizeof(unCRC))); |
| SDL_memcpy(&data[report_size - sizeof(unCRC)], &unCRC, sizeof(unCRC)); |
| } |
| |
| if (SDL_HIDAPI_LockRumble() < 0) { |
| return -1; |
| } |
| |
| /* See if we can update an existing pending request */ |
| if (SDL_HIDAPI_GetPendingRumbleLocked(device, &pending_data, &pending_size, &maximum_size)) { |
| DS5EffectsState_t *pending_effects = (DS5EffectsState_t *)&pending_data[offset]; |
| if (report_size == *pending_size && |
| effects->ucEnableBits1 == pending_effects->ucEnableBits1 && |
| effects->ucEnableBits2 == pending_effects->ucEnableBits2) { |
| /* We're simply updating the data for this request */ |
| SDL_memcpy(pending_data, data, report_size); |
| SDL_HIDAPI_UnlockRumble(); |
| return 0; |
| } |
| } |
| |
| return SDL_HIDAPI_SendRumbleAndUnlock(device, data, report_size); |
| } |
| |
| static void |
| HIDAPI_DriverPS5_SetBluetooth(SDL_HIDAPI_Device *device, SDL_bool is_bluetooth) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| |
| if (ctx->is_bluetooth != is_bluetooth) { |
| ctx->is_bluetooth = is_bluetooth; |
| HIDAPI_DriverPS5_UpdateEffects(device, k_EDS5EffectLED); |
| } |
| } |
| |
| static void |
| HIDAPI_DriverPS5_SetDevicePlayerIndex(SDL_HIDAPI_Device *device, SDL_JoystickID instance_id, int player_index) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| |
| if (!ctx) { |
| return; |
| } |
| |
| ctx->player_index = player_index; |
| |
| /* This will set the new LED state based on the new player index */ |
| HIDAPI_DriverPS5_UpdateEffects(device, k_EDS5EffectLED); |
| } |
| |
| static SDL_bool |
| HIDAPI_DriverPS5_OpenJoystick(SDL_HIDAPI_Device *device, SDL_Joystick *joystick) |
| { |
| SDL_DriverPS5_Context *ctx; |
| Uint8 data[USB_PACKET_LENGTH]; |
| |
| ctx = (SDL_DriverPS5_Context *)SDL_calloc(1, sizeof(*ctx)); |
| if (!ctx) { |
| SDL_OutOfMemory(); |
| return SDL_FALSE; |
| } |
| |
| device->dev = hid_open_path(device->path, 0); |
| if (!device->dev) { |
| SDL_free(ctx); |
| SDL_SetError("Couldn't open %s", device->path); |
| return SDL_FALSE; |
| } |
| device->context = ctx; |
| |
| /* Read the serial number (Bluetooth address in reverse byte order) |
| This will also enable enhanced reports over Bluetooth |
| */ |
| if (ReadFeatureReport(device->dev, k_EPS5FeatureReportIdSerialNumber, data, sizeof(data)) >= 7) { |
| char serial[18]; |
| |
| SDL_snprintf(serial, sizeof(serial), "%.2x-%.2x-%.2x-%.2x-%.2x-%.2x", |
| data[6], data[5], data[4], data[3], data[2], data[1]); |
| joystick->serial = SDL_strdup(serial); |
| } |
| |
| /* Initialize player index (needed for setting LEDs) */ |
| ctx->player_index = SDL_JoystickGetPlayerIndex(joystick); |
| |
| /* Initialize LED and effect state */ |
| HIDAPI_DriverPS5_UpdateEffects(device, k_EDS5EffectLED); |
| |
| /* Initialize the joystick capabilities */ |
| joystick->nbuttons = 17; |
| joystick->naxes = SDL_CONTROLLER_AXIS_MAX; |
| joystick->epowerlevel = SDL_JOYSTICK_POWER_WIRED; |
| |
| SDL_PrivateJoystickAddTouchpad(joystick, 2); |
| SDL_PrivateJoystickAddSensor(joystick, SDL_SENSOR_GYRO); |
| SDL_PrivateJoystickAddSensor(joystick, SDL_SENSOR_ACCEL); |
| |
| return SDL_TRUE; |
| } |
| |
| static int |
| HIDAPI_DriverPS5_RumbleJoystick(SDL_HIDAPI_Device *device, SDL_Joystick *joystick, Uint16 low_frequency_rumble, Uint16 high_frequency_rumble) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| |
| if (!ctx->rumble_left && !ctx->rumble_right) { |
| HIDAPI_DriverPS5_UpdateEffects(device, k_EDS5EffectRumbleStart); |
| } |
| |
| ctx->rumble_left = (low_frequency_rumble >> 8); |
| ctx->rumble_right = (high_frequency_rumble >> 8); |
| |
| return HIDAPI_DriverPS5_UpdateEffects(device, k_EDS5EffectRumble); |
| } |
| |
| static int |
| HIDAPI_DriverPS5_RumbleJoystickTriggers(SDL_HIDAPI_Device *device, SDL_Joystick *joystick, Uint16 left_rumble, Uint16 right_rumble) |
| { |
| return SDL_Unsupported(); |
| } |
| |
| static SDL_bool |
| HIDAPI_DriverPS5_HasJoystickLED(SDL_HIDAPI_Device *device, SDL_Joystick *joystick) |
| { |
| return SDL_FALSE; |
| } |
| |
| static int |
| HIDAPI_DriverPS5_SetJoystickLED(SDL_HIDAPI_Device *device, SDL_Joystick *joystick, Uint8 red, Uint8 green, Uint8 blue) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| |
| ctx->color_set = SDL_TRUE; |
| ctx->led_red = red; |
| ctx->led_green = green; |
| ctx->led_blue = blue; |
| |
| return HIDAPI_DriverPS5_UpdateEffects(device, k_EDS5EffectLED); |
| } |
| |
| static int |
| HIDAPI_DriverPS5_SetJoystickSensorsEnabled(SDL_HIDAPI_Device *device, SDL_Joystick *joystick, SDL_bool enabled) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| |
| if (enabled) { |
| HIDAPI_DriverPS5_LoadCalibrationData(device); |
| } |
| ctx->report_sensors = enabled; |
| |
| return 0; |
| } |
| |
| static void |
| HIDAPI_DriverPS5_HandleSimpleStatePacket(SDL_Joystick *joystick, hid_device *dev, SDL_DriverPS5_Context *ctx, PS5SimpleStatePacket_t *packet) |
| { |
| Sint16 axis; |
| |
| if (ctx->last_state.simple.rgucButtonsHatAndCounter[0] != packet->rgucButtonsHatAndCounter[0]) { |
| { |
| Uint8 data = (packet->rgucButtonsHatAndCounter[0] >> 4); |
| |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_X, (data & 0x01) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_A, (data & 0x02) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_B, (data & 0x04) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_Y, (data & 0x08) ? SDL_PRESSED : SDL_RELEASED); |
| } |
| { |
| Uint8 data = (packet->rgucButtonsHatAndCounter[0] & 0x0F); |
| SDL_bool dpad_up = SDL_FALSE; |
| SDL_bool dpad_down = SDL_FALSE; |
| SDL_bool dpad_left = SDL_FALSE; |
| SDL_bool dpad_right = SDL_FALSE; |
| |
| switch (data) { |
| case 0: |
| dpad_up = SDL_TRUE; |
| break; |
| case 1: |
| dpad_up = SDL_TRUE; |
| dpad_right = SDL_TRUE; |
| break; |
| case 2: |
| dpad_right = SDL_TRUE; |
| break; |
| case 3: |
| dpad_right = SDL_TRUE; |
| dpad_down = SDL_TRUE; |
| break; |
| case 4: |
| dpad_down = SDL_TRUE; |
| break; |
| case 5: |
| dpad_left = SDL_TRUE; |
| dpad_down = SDL_TRUE; |
| break; |
| case 6: |
| dpad_left = SDL_TRUE; |
| break; |
| case 7: |
| dpad_up = SDL_TRUE; |
| dpad_left = SDL_TRUE; |
| break; |
| default: |
| break; |
| } |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_DOWN, dpad_down); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_UP, dpad_up); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_RIGHT, dpad_right); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_LEFT, dpad_left); |
| } |
| } |
| |
| if (ctx->last_state.simple.rgucButtonsHatAndCounter[1] != packet->rgucButtonsHatAndCounter[1]) { |
| Uint8 data = packet->rgucButtonsHatAndCounter[1]; |
| |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_LEFTSHOULDER, (data & 0x01) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_RIGHTSHOULDER, (data & 0x02) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_BACK, (data & 0x10) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_START, (data & 0x20) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_LEFTSTICK, (data & 0x40) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_RIGHTSTICK, (data & 0x80) ? SDL_PRESSED : SDL_RELEASED); |
| } |
| |
| if (ctx->last_state.simple.rgucButtonsHatAndCounter[2] != packet->rgucButtonsHatAndCounter[2]) { |
| Uint8 data = (packet->rgucButtonsHatAndCounter[2] & 0x03); |
| |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_GUIDE, (data & 0x01) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, 15, (data & 0x02) ? SDL_PRESSED : SDL_RELEASED); |
| } |
| |
| axis = ((int)packet->ucTriggerLeft * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_TRIGGERLEFT, axis); |
| axis = ((int)packet->ucTriggerRight * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_TRIGGERRIGHT, axis); |
| axis = ((int)packet->ucLeftJoystickX * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_LEFTX, axis); |
| axis = ((int)packet->ucLeftJoystickY * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_LEFTY, axis); |
| axis = ((int)packet->ucRightJoystickX * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_RIGHTX, axis); |
| axis = ((int)packet->ucRightJoystickY * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_RIGHTY, axis); |
| |
| SDL_memcpy(&ctx->last_state.simple, packet, sizeof(ctx->last_state.simple)); |
| } |
| |
| static void |
| HIDAPI_DriverPS5_HandleStatePacket(SDL_Joystick *joystick, hid_device *dev, SDL_DriverPS5_Context *ctx, PS5StatePacket_t *packet) |
| { |
| static const float TOUCHPAD_SCALEX = 1.0f / 1920; |
| static const float TOUCHPAD_SCALEY = 1.0f / 1070; |
| Sint16 axis; |
| Uint8 touchpad_state; |
| int touchpad_x, touchpad_y; |
| |
| if (ctx->last_state.state.rgucButtonsAndHat[0] != packet->rgucButtonsAndHat[0]) { |
| { |
| Uint8 data = (packet->rgucButtonsAndHat[0] >> 4); |
| |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_X, (data & 0x01) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_A, (data & 0x02) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_B, (data & 0x04) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_Y, (data & 0x08) ? SDL_PRESSED : SDL_RELEASED); |
| } |
| { |
| Uint8 data = (packet->rgucButtonsAndHat[0] & 0x0F); |
| SDL_bool dpad_up = SDL_FALSE; |
| SDL_bool dpad_down = SDL_FALSE; |
| SDL_bool dpad_left = SDL_FALSE; |
| SDL_bool dpad_right = SDL_FALSE; |
| |
| switch (data) { |
| case 0: |
| dpad_up = SDL_TRUE; |
| break; |
| case 1: |
| dpad_up = SDL_TRUE; |
| dpad_right = SDL_TRUE; |
| break; |
| case 2: |
| dpad_right = SDL_TRUE; |
| break; |
| case 3: |
| dpad_right = SDL_TRUE; |
| dpad_down = SDL_TRUE; |
| break; |
| case 4: |
| dpad_down = SDL_TRUE; |
| break; |
| case 5: |
| dpad_left = SDL_TRUE; |
| dpad_down = SDL_TRUE; |
| break; |
| case 6: |
| dpad_left = SDL_TRUE; |
| break; |
| case 7: |
| dpad_up = SDL_TRUE; |
| dpad_left = SDL_TRUE; |
| break; |
| default: |
| break; |
| } |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_DOWN, dpad_down); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_UP, dpad_up); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_RIGHT, dpad_right); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_DPAD_LEFT, dpad_left); |
| } |
| } |
| |
| if (ctx->last_state.state.rgucButtonsAndHat[1] != packet->rgucButtonsAndHat[1]) { |
| Uint8 data = packet->rgucButtonsAndHat[1]; |
| |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_LEFTSHOULDER, (data & 0x01) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_RIGHTSHOULDER, (data & 0x02) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_BACK, (data & 0x10) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_START, (data & 0x20) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_LEFTSTICK, (data & 0x40) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_RIGHTSTICK, (data & 0x80) ? SDL_PRESSED : SDL_RELEASED); |
| } |
| |
| if (ctx->last_state.state.rgucButtonsAndHat[2] != packet->rgucButtonsAndHat[2]) { |
| Uint8 data = packet->rgucButtonsAndHat[2]; |
| |
| SDL_PrivateJoystickButton(joystick, SDL_CONTROLLER_BUTTON_GUIDE, (data & 0x01) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, 15, (data & 0x04) ? SDL_PRESSED : SDL_RELEASED); |
| SDL_PrivateJoystickButton(joystick, 16, (data & 0x02) ? SDL_PRESSED : SDL_RELEASED); |
| } |
| |
| axis = ((int)packet->ucTriggerLeft * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_TRIGGERLEFT, axis); |
| axis = ((int)packet->ucTriggerRight * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_TRIGGERRIGHT, axis); |
| axis = ((int)packet->ucLeftJoystickX * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_LEFTX, axis); |
| axis = ((int)packet->ucLeftJoystickY * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_LEFTY, axis); |
| axis = ((int)packet->ucRightJoystickX * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_RIGHTX, axis); |
| axis = ((int)packet->ucRightJoystickY * 257) - 32768; |
| SDL_PrivateJoystickAxis(joystick, SDL_CONTROLLER_AXIS_RIGHTY, axis); |
| |
| if (packet->ucBatteryLevel & 0x10) { |
| /* 0x20 set means fully charged */ |
| joystick->epowerlevel = SDL_JOYSTICK_POWER_WIRED; |
| } else { |
| /* Battery level ranges from 0 to 10 */ |
| int level = (packet->ucBatteryLevel & 0xF); |
| if (level == 0) { |
| joystick->epowerlevel = SDL_JOYSTICK_POWER_EMPTY; |
| } else if (level <= 2) { |
| joystick->epowerlevel = SDL_JOYSTICK_POWER_LOW; |
| } else if (level <= 7) { |
| joystick->epowerlevel = SDL_JOYSTICK_POWER_MEDIUM; |
| } else { |
| joystick->epowerlevel = SDL_JOYSTICK_POWER_FULL; |
| } |
| } |
| |
| touchpad_state = ((packet->ucTouchpadCounter1 & 0x80) == 0) ? SDL_PRESSED : SDL_RELEASED; |
| touchpad_x = packet->rgucTouchpadData1[0] | (((int)packet->rgucTouchpadData1[1] & 0x0F) << 8); |
| touchpad_y = (packet->rgucTouchpadData1[1] >> 4) | ((int)packet->rgucTouchpadData1[2] << 4); |
| SDL_PrivateJoystickTouchpad(joystick, 0, 0, touchpad_state, touchpad_x * TOUCHPAD_SCALEX, touchpad_y * TOUCHPAD_SCALEY, touchpad_state ? 1.0f : 0.0f); |
| |
| touchpad_state = ((packet->ucTouchpadCounter2 & 0x80) == 0) ? SDL_PRESSED : SDL_RELEASED; |
| touchpad_x = packet->rgucTouchpadData2[0] | (((int)packet->rgucTouchpadData2[1] & 0x0F) << 8); |
| touchpad_y = (packet->rgucTouchpadData2[1] >> 4) | ((int)packet->rgucTouchpadData2[2] << 4); |
| SDL_PrivateJoystickTouchpad(joystick, 0, 1, touchpad_state, touchpad_x * TOUCHPAD_SCALEX, touchpad_y * TOUCHPAD_SCALEY, touchpad_state ? 1.0f : 0.0f); |
| |
| if (ctx->report_sensors) { |
| float data[3]; |
| |
| data[0] = HIDAPI_DriverPS5_ApplyCalibrationData(ctx, 0, LOAD16(packet->rgucGyroX[0], packet->rgucGyroX[1])); |
| data[1] = HIDAPI_DriverPS5_ApplyCalibrationData(ctx, 1, LOAD16(packet->rgucGyroY[0], packet->rgucGyroY[1])); |
| data[2] = HIDAPI_DriverPS5_ApplyCalibrationData(ctx, 2, LOAD16(packet->rgucGyroZ[0], packet->rgucGyroZ[1])); |
| SDL_PrivateJoystickSensor(joystick, SDL_SENSOR_GYRO, data, 3); |
| |
| data[0] = HIDAPI_DriverPS5_ApplyCalibrationData(ctx, 3, LOAD16(packet->rgucAccelX[0], packet->rgucAccelX[1])); |
| data[1] = HIDAPI_DriverPS5_ApplyCalibrationData(ctx, 4, LOAD16(packet->rgucAccelY[0], packet->rgucAccelY[1])); |
| data[2] = HIDAPI_DriverPS5_ApplyCalibrationData(ctx, 5, LOAD16(packet->rgucAccelZ[0], packet->rgucAccelZ[1])); |
| SDL_PrivateJoystickSensor(joystick, SDL_SENSOR_ACCEL, data, 3); |
| } |
| |
| SDL_memcpy(&ctx->last_state.state, packet, sizeof(ctx->last_state.state)); |
| } |
| |
| static SDL_bool |
| HIDAPI_DriverPS5_UpdateDevice(SDL_HIDAPI_Device *device) |
| { |
| SDL_DriverPS5_Context *ctx = (SDL_DriverPS5_Context *)device->context; |
| SDL_Joystick *joystick = NULL; |
| Uint8 data[USB_PACKET_LENGTH*2]; |
| int size; |
| |
| if (device->num_joysticks > 0) { |
| joystick = SDL_JoystickFromInstanceID(device->joysticks[0]); |
| } |
| if (!joystick) { |
| return SDL_FALSE; |
| } |
| |
| while ((size = hid_read_timeout(device->dev, data, sizeof(data), 0)) > 0) { |
| #ifdef DEBUG_PS5_PROTOCOL |
| HIDAPI_DumpPacket("PS5 packet: size = %d", data, size); |
| #endif |
| switch (data[0]) { |
| case k_EPS5ReportIdState: |
| if (size == 10) { |
| HIDAPI_DriverPS5_SetBluetooth(device, SDL_TRUE); /* Simple state packet over Bluetooth */ |
| HIDAPI_DriverPS5_HandleSimpleStatePacket(joystick, device->dev, ctx, (PS5SimpleStatePacket_t *)&data[1]); |
| } else { |
| HIDAPI_DriverPS5_SetBluetooth(device, SDL_FALSE); |
| HIDAPI_DriverPS5_HandleStatePacket(joystick, device->dev, ctx, (PS5StatePacket_t *)&data[1]); |
| } |
| break; |
| case k_EPS5ReportIdBluetoothState: |
| HIDAPI_DriverPS5_SetBluetooth(device, SDL_TRUE); |
| HIDAPI_DriverPS5_HandleStatePacket(joystick, device->dev, ctx, (PS5StatePacket_t *)&data[2]); |
| break; |
| default: |
| #ifdef DEBUG_JOYSTICK |
| SDL_Log("Unknown PS5 packet: 0x%.2x\n", data[0]); |
| #endif |
| break; |
| } |
| } |
| |
| if (size < 0) { |
| /* Read error, device is disconnected */ |
| HIDAPI_JoystickDisconnected(device, joystick->instance_id, SDL_FALSE); |
| } |
| return (size >= 0); |
| } |
| |
| static void |
| HIDAPI_DriverPS5_CloseJoystick(SDL_HIDAPI_Device *device, SDL_Joystick *joystick) |
| { |
| hid_close(device->dev); |
| device->dev = NULL; |
| |
| SDL_free(device->context); |
| device->context = NULL; |
| } |
| |
| static void |
| HIDAPI_DriverPS5_FreeDevice(SDL_HIDAPI_Device *device) |
| { |
| } |
| |
| SDL_HIDAPI_DeviceDriver SDL_HIDAPI_DriverPS5 = |
| { |
| SDL_HINT_JOYSTICK_HIDAPI_PS5, |
| SDL_TRUE, |
| HIDAPI_DriverPS5_IsSupportedDevice, |
| HIDAPI_DriverPS5_GetDeviceName, |
| HIDAPI_DriverPS5_InitDevice, |
| HIDAPI_DriverPS5_GetDevicePlayerIndex, |
| HIDAPI_DriverPS5_SetDevicePlayerIndex, |
| HIDAPI_DriverPS5_UpdateDevice, |
| HIDAPI_DriverPS5_OpenJoystick, |
| HIDAPI_DriverPS5_RumbleJoystick, |
| HIDAPI_DriverPS5_RumbleJoystickTriggers, |
| HIDAPI_DriverPS5_HasJoystickLED, |
| HIDAPI_DriverPS5_SetJoystickLED, |
| HIDAPI_DriverPS5_SetJoystickSensorsEnabled, |
| HIDAPI_DriverPS5_CloseJoystick, |
| HIDAPI_DriverPS5_FreeDevice, |
| NULL |
| }; |
| |
| #endif /* SDL_JOYSTICK_HIDAPI_PS5 */ |
| |
| #endif /* SDL_JOYSTICK_HIDAPI */ |
| |
| /* vi: set ts=4 sw=4 expandtab: */ |
