MissionControl/mc_mitm/source/controllers/wii_controller.hpp

535 lines
17 KiB
C++

/*
* 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 WiiControllerLEDs {
WiiControllerLEDs_P1 = 0x10,
WiiControllerLEDs_P2 = 0x20,
WiiControllerLEDs_P3 = 0x40,
WiiControllerLEDs_P4 = 0x80,
};
enum WiiControllerOrientation {
WiiControllerOrientation_Horizontal,
WiiControllerOrientation_Vertical
};
enum WiiExtensionController {
WiiExtensionController_None,
WiiExtensionController_Nunchuck,
WiiExtensionController_ClassicPro,
WiiExtensionController_WiiUPro,
WiiExtensionController_MotionPlus,
WiiExtensionController_MotionPlusNunchuckPassthrough,
WiiExtensionController_MotionPlusClassicControllerPassthrough,
WiiExtensionController_TaTaCon,
WiiExtensionController_Unrecognised,
};
enum MotionPlusStatus {
MotionPlusStatus_None,
MotionPlusStatus_Uninitialised,
MotionPlusStatus_Inactive,
MotionPlusStatus_Active,
};
struct WiiButtonData {
union {
uint8_t raw[2];
struct {
uint8_t dpad_left : 1;
uint8_t dpad_right : 1;
uint8_t dpad_down : 1;
uint8_t dpad_up : 1;
uint8_t plus : 1;
uint8_t : 0;
uint8_t two : 1;
uint8_t one : 1;
uint8_t B : 1;
uint8_t A : 1;
uint8_t minus : 1;
uint8_t : 2;
uint8_t home : 1;
};
};
} __attribute__ ((__packed__));
struct WiiAccelerometerData {
uint8_t x;
uint8_t y;
uint8_t z;
} __attribute__ ((__packed__));
struct WiiAccelerometerCalibrationData {
uint16_t acc_x_0g;
uint16_t acc_y_0g;
uint16_t acc_z_0g;
uint16_t acc_x_1g;
uint16_t acc_y_1g;
uint16_t acc_z_1g;
} __attribute__ ((__packed__));
struct WiiClassicControllerExtensionData {
struct {
uint8_t left_stick_x : 6;
uint8_t right_stick_x_43 : 2;
uint8_t left_stick_y : 6;
uint8_t right_stick_x_21 : 2;
uint8_t right_stick_y : 5;
uint8_t left_trigger_43 : 2;
uint8_t right_stick_x_0 : 1;
uint8_t right_trigger : 5;
uint8_t left_trigger_20 : 3;
};
struct {
uint8_t : 1;
uint8_t R : 1;
uint8_t plus : 1;
uint8_t home : 1;
uint8_t minus : 1;
uint8_t L : 1;
uint8_t dpad_down : 1;
uint8_t dpad_right : 1;
uint8_t dpad_up : 1;
uint8_t dpad_left : 1;
uint8_t ZR : 1;
uint8_t X : 1;
uint8_t A : 1;
uint8_t Y : 1;
uint8_t B : 1;
uint8_t ZL : 1;
} buttons;
} __attribute__ ((__packed__));
struct WiiClassicControllerPassthroughExtensionData {
union {
struct {
uint8_t : 1;
uint8_t left_stick_x_51 : 5;
uint8_t right_stick_x_43 : 2;
uint8_t : 1;
uint8_t left_stick_y_51 : 5;
uint8_t right_stick_x_21 : 2;
uint8_t right_stick_y : 5;
uint8_t left_trigger_43 : 2;
uint8_t right_stick_x_0 : 1;
uint8_t right_trigger : 5;
uint8_t left_trigger_20 : 3;
uint8_t extension_connected : 1;
uint8_t : 0;
uint8_t : 1;
uint8_t motionplus_report : 1;
uint8_t : 0;
};
struct {
uint8_t dpad_up : 1;
uint8_t : 0;
uint8_t dpad_left : 1;
uint8_t : 0;
uint8_t pad[2];
uint8_t : 1;
uint8_t R : 1;
uint8_t plus : 1;
uint8_t home : 1;
uint8_t minus : 1;
uint8_t L : 1;
uint8_t dpad_down : 1;
uint8_t dpad_right : 1;
uint8_t : 2;
uint8_t ZR : 1;
uint8_t X : 1;
uint8_t A : 1;
uint8_t Y : 1;
uint8_t B : 1;
uint8_t ZL : 1;
} buttons;
};
} __attribute__ ((__packed__));
struct WiiNunchuckExtensionData {
uint8_t stick_x;
uint8_t stick_y;
uint8_t accel_x_92;
uint8_t accel_y_92;
uint8_t accel_z_92;
uint8_t Z : 1;
uint8_t C : 1;
uint8_t accel_x_10 : 2;
uint8_t accel_y_10 : 2;
uint8_t accel_z_10 : 2;
} __attribute__ ((__packed__));
struct WiiNunchuckPassthroughExtensionData {
uint8_t stick_x;
uint8_t stick_y;
uint8_t accel_x_92;
uint8_t accel_y_92;
uint8_t extension_connected : 1;
uint8_t accel_z_93 : 7;
uint8_t : 1;
uint8_t motionplus_report : 1;
uint8_t Z : 1;
uint8_t C : 1;
uint8_t accel_x_1 : 1;
uint8_t accel_y_1 : 1;
uint8_t accel_z_21 : 2;
} __attribute__ ((__packed__));
struct WiiUProButtonData {
uint8_t : 1;
uint8_t R : 1;
uint8_t plus : 1;
uint8_t home : 1;
uint8_t minus : 1;
uint8_t L : 1;
uint8_t dpad_down : 1;
uint8_t dpad_right : 1;
uint8_t dpad_up : 1;
uint8_t dpad_left : 1;
uint8_t ZR : 1;
uint8_t X : 1;
uint8_t A : 1;
uint8_t Y : 1;
uint8_t B : 1;
uint8_t ZL : 1;
uint8_t rstick_press : 1;
uint8_t lstick_press : 1;
uint8_t charging : 1;
uint8_t usb_connected : 1;
uint8_t battery : 3;
uint8_t : 1;
} __attribute__ ((__packed__));
struct WiiUProExtensionData {
uint16_t left_stick_x;
uint16_t right_stick_x;
uint16_t left_stick_y;
uint16_t right_stick_y;
WiiUProButtonData buttons;
} __attribute__ ((__packed__));
struct MotionPlusExtensionData {
uint8_t yaw_speed_lo;
uint8_t roll_speed_lo;
uint8_t pitch_speed_lo;
uint8_t pitch_slow_mode : 1;
uint8_t yaw_slow_mode : 1;
uint8_t yaw_speed_hi : 6;
uint8_t extension_connected : 1;
uint8_t roll_slow_mode : 1;
uint8_t roll_speed_hi : 6;
uint8_t : 1;
uint8_t motionplus_report : 1;
uint8_t pitch_speed_hi : 6;
} __attribute__ ((__packed__));
struct MotionPlusCalibration {
uint16_t yaw_zero;
uint16_t roll_zero;
uint16_t pitch_zero;
uint16_t yaw_scale;
uint16_t roll_scale;
uint16_t pitch_scale;
uint8_t degrees_div_6;
} __attribute__ ((__packed__));
struct MotionPlusCalibrationData {
MotionPlusCalibration fast;
MotionPlusCalibration slow;
} __attribute__ ((__packed__));
struct TaTaConExtensionData {
uint8_t _unk0[5];
uint8_t : 3;
uint8_t R_rim : 1;
uint8_t R_center : 1;
uint8_t L_rim : 1;
uint8_t L_center : 1;
uint8_t : 0;
} __attribute__ ((__packed__));
struct WiiOutputReport0x10 {
uint8_t rumble : 1;
uint8_t : 0;
} __attribute__ ((__packed__));
struct WiiOutputReport0x11 {
uint8_t rumble : 1;
uint8_t : 3;
uint8_t leds : 4;
} __attribute__ ((__packed__));
struct WiiOutputReport0x12 {
uint8_t rumble : 1;
uint8_t : 0;
uint8_t report_mode;
} __attribute__ ((__packed__));
struct WiiOutputReport0x14 {
uint8_t : 5;
uint8_t speaker_enable : 1;
uint8_t : 0;
} __attribute__ ((__packed__));
struct WiiOutputReport0x15 {
uint8_t rumble : 1;
uint8_t : 0;
} __attribute__ ((__packed__));
struct WiiOutputReport0x16 {
uint32_t address;
uint8_t size;
uint8_t data[16];
} __attribute__ ((__packed__));
struct WiiOutputReport0x17 {
uint32_t address;
uint16_t size;
} __attribute__ ((__packed__));
struct WiiOutputReport0x18 {
uint8_t size;
uint8_t speaker_data[20];
} __attribute__ ((__packed__));
struct WiiOutputReport0x19 {
uint8_t : 5;
uint8_t speaker_mute : 1;
uint8_t : 0;
} __attribute__ ((__packed__));
struct WiiInputReport0x20 {
WiiButtonData buttons;
uint8_t battery_critical : 1;
uint8_t extension_connected : 1;
uint8_t speaker_enabled : 1;
uint8_t ir_enabled : 1;
uint8_t led_state : 4;
uint8_t _pad[2];
uint8_t battery;
} __attribute__ ((__packed__));
struct WiiInputReport0x21 {
WiiButtonData buttons;
uint8_t error : 4;
uint8_t size : 4;
uint16_t address;
uint8_t data[16];
} __attribute__ ((__packed__));
struct WiiInputReport0x22 {
WiiButtonData buttons;
uint8_t report_id;
uint8_t error;
} __attribute__ ((__packed__));
struct WiiInputReport0x30 {
WiiButtonData buttons;
} __attribute__ ((__packed__));
struct WiiInputReport0x31 {
WiiButtonData buttons;
WiiAccelerometerData accel;
} __attribute__ ((__packed__));
struct WiiInputReport0x32 {
WiiButtonData buttons;
uint8_t extension[8];
} __attribute__ ((__packed__));
struct WiiInputReport0x33 {
WiiButtonData buttons;
WiiAccelerometerData accel;
uint8_t ir[12];
} __attribute__ ((__packed__));
struct WiiInputReport0x34 {
WiiButtonData buttons;
uint8_t extension[19];
} __attribute__ ((__packed__));
struct WiiInputReport0x35 {
WiiButtonData buttons;
WiiAccelerometerData accel;
uint8_t extension[16];
} __attribute__ ((__packed__));
struct WiiInputReport0x36 {
WiiButtonData buttons;
uint8_t ir[10];
uint8_t extension[9];
} __attribute__ ((__packed__));
struct WiiInputReport0x37 {
WiiButtonData buttons;
WiiAccelerometerData accel;
uint8_t ir[10];
uint8_t extension[6];
} __attribute__ ((__packed__));
struct WiiInputReport0x3d {
uint8_t extension[21];
} __attribute__ ((__packed__));
struct WiiInputReport0x3e {
WiiButtonData buttons;
uint8_t accel;
uint8_t ir[18];
} __attribute__ ((__packed__));
struct WiiInputReport0x3f {
WiiButtonData buttons;
uint8_t accel;
uint8_t ir[18];
} __attribute__ ((__packed__));
struct WiiReportData {
uint8_t id;
union {
WiiOutputReport0x10 output0x10;
WiiOutputReport0x11 output0x11;
WiiOutputReport0x12 output0x12;
WiiOutputReport0x14 output0x14;
WiiOutputReport0x15 output0x15;
WiiOutputReport0x16 output0x16;
WiiOutputReport0x17 output0x17;
WiiOutputReport0x18 output0x18;
WiiOutputReport0x19 output0x19;
WiiInputReport0x20 input0x20;
WiiInputReport0x21 input0x21;
WiiInputReport0x22 input0x22;
WiiInputReport0x30 input0x30;
WiiInputReport0x31 input0x31;
WiiInputReport0x32 input0x32;
WiiInputReport0x33 input0x33;
WiiInputReport0x34 input0x34;
WiiInputReport0x35 input0x35;
WiiInputReport0x36 input0x36;
WiiInputReport0x37 input0x37;
WiiInputReport0x3d input0x3d;
WiiInputReport0x3e input0x3e;
WiiInputReport0x3f input0x3f;
};
} __attribute__ ((__packed__));
class WiiController : public EmulatedSwitchController {
public:
static constexpr const HardwareID hardware_ids[] = {
{0x057e, 0x0306}, // Official Wiimote
{0x057e, 0x0330}, // Official Wii U Pro Controller
};
WiiController(const bluetooth::Address *address, HardwareID id)
: EmulatedSwitchController(address, id)
, m_orientation(WiiControllerOrientation_Horizontal)
, m_extension(WiiExtensionController_None)
, m_rumble_state(0)
, m_mp_extension_flag(false)
, m_debounce_counter(0) { }
Result Initialize(void);
Result SetVibration(const SwitchRumbleData *rumble_data);
Result CancelVibration(void);
Result SetPlayerLed(uint8_t led_mask);
void ProcessInputData(const bluetooth::HidReport *report) override;
protected:
void MapInputReport0x20(const WiiReportData *src);
void MapInputReport0x21(const WiiReportData *src);
void MapInputReport0x22(const WiiReportData *src);
void MapInputReport0x30(const WiiReportData *src);
void MapInputReport0x31(const WiiReportData *src);
void MapInputReport0x32(const WiiReportData *src);
void MapInputReport0x34(const WiiReportData *src);
void MapInputReport0x35(const WiiReportData *src);
void MapInputReport0x3d(const WiiReportData *src);
void MapCoreButtons(const WiiButtonData *buttons);
void MapAccelerometerData(const WiiAccelerometerData *accel, const WiiButtonData *buttons);
void MapExtensionBytes(const uint8_t ext[]);
void MapNunchuckExtension(const uint8_t ext[]);
void MapClassicControllerExtension(const uint8_t ext[]);
void MapWiiUProControllerExtension(const uint8_t ext[]);
void MapTaTaConExtension(const uint8_t ext[]);
void MapMotionPlusExtension(const uint8_t ext[]);
void MapNunchuckExtensionPassthroughMode(const uint8_t ext[]);
void MapClassicControllerExtensionPassthroughMode(const uint8_t ext[]);
void HandleStatusReport(const WiiReportData *wii_report);
WiiExtensionController GetExtensionControllerType();
MotionPlusStatus GetMotionPlusStatus();
Result GetAccelerometerCalibration(WiiAccelerometerCalibrationData *calibration);
Result GetMotionPlusCalibration(MotionPlusCalibrationData *calibration);
Result SetReportMode(uint8_t mode);
Result QueryStatus();
Result InitializeStandardExtension();
Result InitializeMotionPlus();
Result ActivateMotionPlus();
Result ActivateMotionPlusNunchuckPassthrough();
Result ActivateMotionPlusClassicPassthrough();
Result DeactivateMotionPlus();
Result UpdateMotionPlusExtensionStatus(bool extension_connected);
Result WriteMemory(uint32_t write_addr, const void *data, uint8_t size);
Result ReadMemory(uint32_t read_addr, uint16_t size, void *out_data);
WiiControllerOrientation m_orientation;
WiiExtensionController m_extension;
bool m_rumble_state;
bool m_mp_extension_flag;
u64 m_debounce_counter;
WiiAccelerometerCalibrationData m_accel_calibration;
MotionPlusCalibrationData m_gyro_calibration;
};
}