input_common: Implement custom joycon driver

2021-12-01 14:46:25 -06:00
parent 7f77aafe41
commit d27a4b3e07
8 changed files with 1514 additions and 9 deletions
@@ -0,0 +1,126 @@
+// Copyright 2021 yuzu Emulator Project
+// Licensed under GPLv2 or any later version
+// Refer to the license.txt file included
+
+#pragma once
+
+#include <span>
+#include <thread>
+
+#include "input_common/helpers/joycon_protocol.h"
+#include "input_common/input_engine.h"
+
+namespace InputCommon {
+
+class Joycons final : public InputCommon::InputEngine {
+public:
+    explicit Joycons(const std::string& input_engine_);
+
+    ~Joycons();
+
+    /// Used for automapping features
+    std::vector<Common::ParamPackage> GetInputDevices() const override;
+    ButtonMapping GetButtonMappingForDevice(const Common::ParamPackage& params) override;
+    AnalogMapping GetAnalogMappingForDevice(const Common::ParamPackage& params) override;
+    MotionMapping GetMotionMappingForDevice(const Common::ParamPackage& params) override;
+    Common::Input::ButtonNames GetUIName(const Common::ParamPackage& params) const override;
+
+    Common::Input::VibrationError SetRumble(
+        const PadIdentifier& identifier, const Common::Input::VibrationStatus vibration) override;
+
+private:
+    enum class PadAxes {
+        LeftStickX,
+        LeftStickY,
+        RightStickX,
+        RightStickY,
+        Undefined,
+    };
+
+    enum class PadMotion {
+        LeftMotion,
+        RightMotion,
+        ProMotion,
+        Undefined,
+    };
+
+    struct JoyconData {
+        Joycon::JoyconHandle joycon_handle{nullptr, 0};
+
+        // Harware config
+        Joycon::GyroSensitivity gyro_sensitivity;
+        Joycon::GyroPerformance gyro_performance;
+        Joycon::AccelerometerSensitivity accelerometer_sensitivity;
+        Joycon::AccelerometerPerformance accelerometer_performance;
+        Joycon::ReportMode mode;
+        bool imu_enabled;
+        bool rumble_enabled;
+        u8 leds;
+
+        // Fixed value info
+        f32 version;
+        std::array<u8, 6> mac;
+        Joycon::CalibrationData calibration;
+        Joycon::ControllerType type;
+        std::array<u8, 15> serial_number;
+        Joycon::Color color;
+        std::size_t port;
+
+        // Polled data
+        u32 raw_temperature;
+        u8 raw_battery;
+        Common::Input::VibrationStatus hd_rumble;
+    };
+    void InputThread(std::stop_token stop_token);
+    void ScanThread(std::stop_token stop_token);
+
+    /// Resets status of device connected to port
+    void ResetDeviceType(std::size_t port);
+
+    /// Captures JC Adapter endpoint address,
+    void GetJCEndpoint();
+
+    /// For shutting down, clear all data, join all threads, release usb
+    void Reset();
+
+    bool IsPayloadCorrect(int status, std::span<const u8> buffer);
+
+    /// For use in initialization, querying devices to find the adapter
+    void Setup();
+
+    bool DeviceConnected(std::size_t port) const;
+
+    void UpdateJoyconData(JoyconData& jc, std::span<const u8> buffer);
+    void UpdateLeftPadInput(JoyconData& jc, std::span<const u8> buffer);
+    void UpdateRightPadInput(JoyconData& jc, std::span<const u8> buffer);
+    void UpdateProPadInput(JoyconData& jc, std::span<const u8> buffer);
+
+    // Reads gyro and accelerometer values
+    f32 GetAxisValue(u16 raw_value, Joycon::JoyStickAxisCalibration calibration) const;
+    f32 GetAccelerometerValue(s16 raw_value, Joycon::ImuSensorCalibration cal,
+                              Joycon::AccelerometerSensitivity sen) const;
+    f32 GetGyroValue(s16 raw_value, Joycon::ImuSensorCalibration cal,
+                     Joycon::GyroSensitivity sen) const;
+
+    s16 GetRawIMUValues(size_t sensor, size_t axis, std::span<const u8> buffer) const;
+    BasicMotion GetMotionInput(std::span<const u8> buffer, Joycon::ImuCalibration calibration,
+                               Joycon::AccelerometerSensitivity accel_sensitivity,
+                               Joycon::GyroSensitivity gyro_sensitivity) const;
+
+    Joycon::JoyconHandle& GetHandle(PadIdentifier identifier);
+    PadIdentifier GetIdentifier(std::size_t port) const;
+
+    std::string JoyconName(std::size_t port) const;
+
+    Common::Input::ButtonNames GetUIButtonName(const Common::ParamPackage& params) const;
+
+    std::jthread input_thread;
+    std::jthread scan_thread;
+    bool input_thread_running{};
+    bool scan_thread_running{};
+    u8 input_error_counter{};
+
+    std::array<JoyconData, 8> joycon_data{};
+};
+
+} // namespace InputCommon