OpenPuck/OpenPuck/mode_ps5.cpp

#include "mode_ps5.h"
#include "triton.h"
#include "gamepad_util.h"
#include "config.h"
#include <Adafruit_TinyUSB.h>
#include <Arduino.h>
#include <string.h>

Ps5Controller g_ps5Ctl;

static const uint8_t PS5_HID_DESC[]={
  0x05,0x01,0x09,0x05,0xA1,0x01,0x85,0x01,0x09,0x30,0x09,0x31,0x09,0x32,0x09,0x35,
  0x09,0x33,0x09,0x34,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x06,0x81,0x02,0x06,
  0x00,0xFF,0x09,0x20,0x95,0x01,0x81,0x02,0x05,0x01,0x09,0x39,0x15,0x00,0x25,0x07,
  0x35,0x00,0x46,0x3B,0x01,0x65,0x14,0x75,0x04,0x95,0x01,0x81,0x42,0x65,0x00,0x05,
  0x09,0x19,0x01,0x29,0x0F,0x15,0x00,0x25,0x01,0x75,0x01,0x95,0x0F,0x81,0x02,0x06,
  0x00,0xFF,0x09,0x21,0x95,0x0D,0x81,0x02,0x06,0x00,0xFF,0x09,0x22,0x15,0x00,0x26,
  0xFF,0x00,0x75,0x08,0x95,0x34,0x81,0x02,0x85,0x02,0x09,0x23,0x95,0x2F,0x91,0x02,
  0x85,0x05,0x09,0x33,0x95,0x28,0xB1,0x02,0x85,0x08,0x09,0x34,0x95,0x2F,0xB1,0x02,
  0x85,0x09,0x09,0x24,0x95,0x13,0xB1,0x02,0x85,0x0A,0x09,0x25,0x95,0x1A,0xB1,0x02,
  0x85,0x20,0x09,0x26,0x95,0x3F,0xB1,0x02,0x85,0x21,0x09,0x27,0x95,0x04,0xB1,0x02,
  0x85,0x22,0x09,0x40,0x95,0x3F,0xB1,0x02,0x85,0x80,0x09,0x28,0x95,0x3F,0xB1,0x02,
  0x85,0x81,0x09,0x29,0x95,0x3F,0xB1,0x02,0x85,0x82,0x09,0x2A,0x95,0x09,0xB1,0x02,
  0x85,0x83,0x09,0x2B,0x95,0x3F,0xB1,0x02,0x85,0x84,0x09,0x2C,0x95,0x3F,0xB1,0x02,
  0x85,0x85,0x09,0x2D,0x95,0x02,0xB1,0x02,0x85,0xA0,0x09,0x2E,0x95,0x01,0xB1,0x02,
  0x85,0xE0,0x09,0x2F,0x95,0x3F,0xB1,0x02,0x85,0xF0,0x09,0x30,0x95,0x3F,0xB1,0x02,
  0x85,0xF1,0x09,0x31,0x95,0x3F,0xB1,0x02,0x85,0xF2,0x09,0x32,0x95,0x0F,0xB1,0x02,
  0x85,0xF4,0x09,0x35,0x95,0x3F,0xB1,0x02,0x85,0xF5,0x09,0x36,0x95,0x03,0xB1,0x02,0xC0
};
#define PS5_TOUCH_H 1080
#define PS5_STATUS_USB 0x1A   // charging + level 10 (~100%)
static unsigned long g_ps5LastMs=0;
static Adafruit_USBD_HID g_ps5;

static void ps5Build(uint8_t out[63]){
  uint32_t b=psButtonsFromSteam(g_in.buttons);
  bool lTouch=(b&TB_LPADT)||(b&TB_LPADC), rTouch=(b&TB_RPADT)||(b&TB_RPADC);
  memset(out,0,63);
  out[0]=swStick(g_in.lx,false); out[1]=swStick(g_in.ly,true);
  out[2]=swStick(g_in.rx,false); out[3]=swStick(g_in.ry,true);
  out[4]=g_in.lt; out[5]=g_in.rt;
  static uint8_t seq=0; out[6]=seq++;
  out[7]=psHatNibble(b)|psFaceNibble(b);
  out[8]=psShouldersByte(b);
  out[9]=((b&TB_STEAM)?0x01:0)|((b&TB_LPADC||b&TB_RPADC)?0x02:0);
  out[15]=g_in.gx&0xFF; out[16]=g_in.gx>>8;
  out[17]=g_in.gz&0xFF; out[18]=g_in.gz>>8;
  out[19]=g_in.gy&0xFF; out[20]=g_in.gy>>8;
  out[21]=g_in.ax&0xFF; out[22]=g_in.ax>>8;
  out[23]=g_in.ay&0xFF; out[24]=g_in.ay>>8;
  out[25]=g_in.az&0xFF; out[26]=g_in.az>>8;
  uint16_t lx, ly, rx, ry;
  steamPadsToTouch(b, PS5_TOUCH_H, g_in.lpx, g_in.lpy, g_in.rpx, g_in.rpy, &lx, &ly, &rx, &ry);
  touchPackPads(out+32, lTouch, rTouch, lx, ly, rx, ry);
  out[52]=PS5_STATUS_USB;
}

void Ps5Controller::begin(){
  USBDevice.setID(0x054C, 0x0CE6);
  USBDevice.setDeviceVersion(0x0101);   // bumped from 0x0100 for the added wake-mouse interface (Windows caches config by VID:PID:bcdDevice)
  USBDevice.setManufacturerDescriptor("Sony Interactive Entertainment");
  USBDevice.setProductDescriptor("DualSense Wireless Controller");
  g_ps5.enableOutEndpoint(true);
  g_ps5.setReportDescriptor(PS5_HID_DESC, sizeof PS5_HID_DESC);
  g_ps5.setPollInterval(4);
  g_ps5.begin();
}
void Ps5Controller::task(){
  if(!g_ps5.ready()) return;
  if(millis()-g_ps5LastMs<USB_STREAM_MS) return; g_ps5LastMs=millis();
  uint8_t p[63]; ps5Build(p);
  g_ps5.sendReport(0x01, p, sizeof p);
}