MissionControl/mc_mitm/source/controllers/dualsense_controller.hpp

/*
 * Copyright (c) 2020-2021 ndeadly
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once
#include "emulated_switch_controller.hpp"

namespace ams::controller {

    enum DualsenseDPadDirection {
        DualsenseDPad_N,
        DualsenseDPad_NE,
        DualsenseDPad_E,
        DualsenseDPad_SE,
        DualsenseDPad_S,
        DualsenseDPad_SW,
        DualsenseDPad_W,
        DualsenseDPad_NW,
        DualsenseDPad_Released
    };

    struct DualsenseStickData {
        uint8_t x;
        uint8_t y;
    } __attribute__((packed));

    struct DualsenseButtonData {
        uint8_t dpad       : 4;
        uint8_t square     : 1;
        uint8_t cross      : 1;
        uint8_t circle     : 1;
        uint8_t triangle   : 1;

        uint8_t L1         : 1;
        uint8_t R1         : 1;
        uint8_t L2         : 1;
        uint8_t R2         : 1;
        uint8_t share      : 1;
        uint8_t options    : 1;
        uint8_t L3         : 1;
        uint8_t R3         : 1;
        
        uint8_t ps         : 1;
        uint8_t tpad       : 1;
        uint8_t counter    : 6;
    } __attribute__((packed));

    struct DualsenseRumbleData {
        uint8_t amp_motor_left;
        uint8_t amp_motor_right;
    } __attribute__((packed));

    struct DualsenseImuCalibrationData {
        struct {
            int16_t pitch_bias;
            int16_t yaw_bias;
            int16_t roll_bias;
            int16_t pitch_max;
            int16_t pitch_min;
            int16_t yaw_max;
            int16_t yaw_min;
            int16_t roll_max;
            int16_t roll_min;
            int16_t speed_max;
            int16_t speed_min;
        } gyro;
        
        struct {
            int16_t x_max;
            int16_t x_min;
            int16_t y_max;
            int16_t y_min;
            int16_t z_max;
            int16_t z_min;
        } acc;
    } __attribute__((packed));

    struct DualsenseFeatureReport0x05 {
        DualsenseImuCalibrationData calibration;
        uint32_t crc;
    } __attribute__((packed));

    struct DualsenseOutputReport0x31 {
        struct {
            uint8_t data[75];
        };
        uint32_t crc;
    } __attribute__((packed));

    struct DualsenseInputReport0x01 {
        DualsenseStickData      left_stick;
        DualsenseStickData      right_stick;
        DualsenseButtonData     buttons;
        uint8_t                 left_trigger;
        uint8_t                 right_trigger;
    } __attribute__((packed));

    struct DualsenseInputReport0x31 {
        uint8_t                 _unk0;
        DualsenseStickData      left_stick;
        DualsenseStickData      right_stick;
        uint8_t                 left_trigger;
        uint8_t                 right_trigger;
        uint8_t                 counter;
        DualsenseButtonData     buttons;
        uint8_t                 _unk1[5];
        int16_t                 vel_x;
        int16_t                 vel_y;
        int16_t                 vel_z;
        int16_t                 acc_x;
        int16_t                 acc_y;
        int16_t                 acc_z;
        uint8_t                 _unk2[25];

        uint8_t battery_level    : 4;
        uint8_t usb              : 1;
        uint8_t full             : 1;
        uint8_t                  : 0;
    } __attribute__((packed));

    struct DualsenseReportData {
        uint8_t id;
        union {
            DualsenseFeatureReport0x05 feature0x05;
            DualsenseInputReport0x01 input0x01;
            DualsenseInputReport0x31 input0x31;
        };
    } __attribute__((packed));

    class DualsenseController : public EmulatedSwitchController {

        public:
            static constexpr const HardwareID hardware_ids[] = {
                {0x054c, 0x0ce6}    // Sony Dualsense Controller
            };

            DualsenseController(const bluetooth::Address *address, HardwareID id)
            : EmulatedSwitchController(address, id)
            , m_led_flags(0)
            , m_led_colour({0, 0, 0})
            , m_rumble_state({0, 0}) { }

            Result Initialize(void);
            Result SetVibration(const SwitchRumbleData *rumble_data);
            Result CancelVibration(void);
            Result SetPlayerLed(uint8_t led_mask);
            Result SetLightbarColour(RGBColour colour);

            void ProcessInputData(const bluetooth::HidReport *report) override;

        private:
            void MapInputReport0x01(const DualsenseReportData *src);
            void MapInputReport0x31(const DualsenseReportData *src);

            void MapButtons(const DualsenseButtonData *buttons);

            Result GetCalibrationData(DualsenseImuCalibrationData *calibration);
            Result PushRumbleLedState(void);

            uint8_t m_led_flags;
            RGBColour m_led_colour;
            DualsenseRumbleData m_rumble_state;

            DualsenseImuCalibrationData m_motion_calibration;
    };

}