2019-09-06 15:09:30 +00:00
|
|
|
// Copyright 2019 Dolphin Emulator Project
|
2021-07-05 01:22:19 +00:00
|
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2019-10-26 16:05:16 +00:00
|
|
|
#include "InputCommon/ControllerInterface/DualShockUDPClient/DualShockUDPClient.h"
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2019-11-22 21:07:12 +00:00
|
|
|
#include <algorithm>
|
|
|
|
#include <array>
|
2019-09-06 15:09:30 +00:00
|
|
|
#include <chrono>
|
2019-11-22 21:07:12 +00:00
|
|
|
#include <tuple>
|
2019-09-06 15:09:30 +00:00
|
|
|
|
|
|
|
#include <SFML/Network/SocketSelector.hpp>
|
|
|
|
#include <SFML/Network/UdpSocket.hpp>
|
2020-05-09 20:44:45 +00:00
|
|
|
#include <fmt/format.h>
|
2019-09-06 15:09:30 +00:00
|
|
|
|
|
|
|
#include "Common/Config/Config.h"
|
|
|
|
#include "Common/Flag.h"
|
|
|
|
#include "Common/Logging/Log.h"
|
|
|
|
#include "Common/MathUtil.h"
|
|
|
|
#include "Common/Random.h"
|
2021-05-15 09:28:30 +00:00
|
|
|
#include "Common/ScopeGuard.h"
|
2020-05-09 20:44:45 +00:00
|
|
|
#include "Common/StringUtil.h"
|
2019-09-06 15:09:30 +00:00
|
|
|
#include "Common/Thread.h"
|
|
|
|
#include "Core/CoreTiming.h"
|
|
|
|
#include "InputCommon/ControllerInterface/ControllerInterface.h"
|
2019-10-26 16:05:16 +00:00
|
|
|
#include "InputCommon/ControllerInterface/DualShockUDPClient/DualShockUDPProto.h"
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2019-10-26 16:05:16 +00:00
|
|
|
namespace ciface::DualShockUDPClient
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-01-02 17:32:53 +00:00
|
|
|
constexpr std::string_view DUALSHOCKUDP_SOURCE_NAME = "DSUClient";
|
|
|
|
|
2019-11-22 20:55:11 +00:00
|
|
|
namespace Settings
|
|
|
|
{
|
2020-07-13 05:26:43 +00:00
|
|
|
const Config::Info<std::string> SERVER_ADDRESS{
|
|
|
|
{Config::System::DualShockUDPClient, "Server", "IPAddress"}, ""};
|
|
|
|
const Config::Info<int> SERVER_PORT{{Config::System::DualShockUDPClient, "Server", "Port"}, 0};
|
2020-05-09 20:44:45 +00:00
|
|
|
const Config::Info<std::string> SERVERS{{Config::System::DualShockUDPClient, "Server", "Entries"},
|
|
|
|
""};
|
|
|
|
const Config::Info<bool> SERVERS_ENABLED{{Config::System::DualShockUDPClient, "Server", "Enabled"},
|
|
|
|
false};
|
2019-11-22 20:55:11 +00:00
|
|
|
} // namespace Settings
|
|
|
|
|
2019-11-22 21:49:43 +00:00
|
|
|
// Clock type used for querying timeframes
|
|
|
|
using SteadyClock = std::chrono::steady_clock;
|
|
|
|
|
2019-11-17 16:49:54 +00:00
|
|
|
class Device final : public Core::Device
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
private:
|
|
|
|
template <class T>
|
2019-11-17 16:49:54 +00:00
|
|
|
class Button final : public Input
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-11-17 16:49:54 +00:00
|
|
|
Button(const char* name, const T& buttons, T mask)
|
|
|
|
: m_name(name), m_buttons(buttons), m_mask(mask)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
std::string GetName() const override { return m_name; }
|
|
|
|
ControlState GetState() const override { return (m_buttons & m_mask) != 0; }
|
|
|
|
|
|
|
|
private:
|
2019-11-17 16:49:54 +00:00
|
|
|
const char* const m_name;
|
2019-09-06 15:09:30 +00:00
|
|
|
const T& m_buttons;
|
2019-11-17 16:49:54 +00:00
|
|
|
const T m_mask;
|
2019-09-06 15:09:30 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
template <class T>
|
2019-11-17 16:49:54 +00:00
|
|
|
class AnalogInput : public Input
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-11-17 16:49:54 +00:00
|
|
|
AnalogInput(const char* name, const T& input, ControlState range, ControlState offset = 0)
|
|
|
|
: m_name(name), m_input(input), m_range(range), m_offset(offset)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
}
|
2019-11-17 16:49:54 +00:00
|
|
|
std::string GetName() const final override { return m_name; }
|
|
|
|
ControlState GetState() const final override
|
|
|
|
{
|
|
|
|
return (ControlState(m_input) + m_offset) / m_range;
|
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
|
|
|
|
private:
|
2019-11-17 16:49:54 +00:00
|
|
|
const char* m_name;
|
2019-09-06 15:09:30 +00:00
|
|
|
const T& m_input;
|
|
|
|
const ControlState m_range;
|
|
|
|
const ControlState m_offset;
|
|
|
|
};
|
|
|
|
|
2019-11-17 16:49:54 +00:00
|
|
|
class TouchInput final : public AnalogInput<int>
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-11-17 16:49:54 +00:00
|
|
|
using AnalogInput::AnalogInput;
|
2020-02-22 16:27:43 +00:00
|
|
|
bool IsDetectable() const override { return false; }
|
2019-09-06 15:09:30 +00:00
|
|
|
};
|
|
|
|
|
2019-11-17 23:10:35 +00:00
|
|
|
class MotionInput final : public AnalogInput<float>
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-11-17 16:49:54 +00:00
|
|
|
using AnalogInput::AnalogInput;
|
2020-02-22 16:27:43 +00:00
|
|
|
bool IsDetectable() const override { return false; }
|
2019-09-06 15:09:30 +00:00
|
|
|
};
|
|
|
|
|
2019-11-17 16:49:54 +00:00
|
|
|
using AccelerometerInput = MotionInput;
|
|
|
|
using GyroInput = MotionInput;
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2019-11-17 16:42:49 +00:00
|
|
|
class BatteryInput final : public Input
|
|
|
|
{
|
2019-11-17 16:49:54 +00:00
|
|
|
public:
|
2019-11-17 16:42:49 +00:00
|
|
|
using BatteryState = Proto::DsBattery;
|
|
|
|
|
|
|
|
BatteryInput(const BatteryState& battery) : m_battery(battery) {}
|
|
|
|
|
|
|
|
std::string GetName() const override { return "Battery"; }
|
|
|
|
|
|
|
|
ControlState GetState() const override
|
|
|
|
{
|
|
|
|
switch (m_battery)
|
|
|
|
{
|
2021-05-10 19:43:41 +00:00
|
|
|
case BatteryState::Charging: // We don't actually know the battery level in this case
|
2019-11-17 16:42:49 +00:00
|
|
|
case BatteryState::Charged:
|
|
|
|
return BATTERY_INPUT_MAX_VALUE;
|
|
|
|
default:
|
|
|
|
return ControlState(m_battery) / ControlState(BatteryState::Full) * BATTERY_INPUT_MAX_VALUE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-22 16:27:43 +00:00
|
|
|
bool IsDetectable() const override { return false; }
|
2019-11-17 16:42:49 +00:00
|
|
|
|
|
|
|
private:
|
|
|
|
const BatteryState& m_battery;
|
|
|
|
};
|
|
|
|
|
2019-09-06 15:09:30 +00:00
|
|
|
public:
|
|
|
|
void UpdateInput() override;
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
Device(std::string name, int index, std::string server_address, u16 server_port, u32 client_uid);
|
2019-09-06 15:09:30 +00:00
|
|
|
|
|
|
|
std::string GetName() const final override;
|
|
|
|
std::string GetSource() const final override;
|
|
|
|
std::optional<int> GetPreferredId() const final override;
|
2021-05-15 09:14:11 +00:00
|
|
|
// Always add these at the end, given their hotplug nature
|
2022-03-02 21:40:20 +00:00
|
|
|
int GetSortPriority() const override { return -5; }
|
2019-09-06 15:09:30 +00:00
|
|
|
|
|
|
|
private:
|
2021-05-10 19:43:41 +00:00
|
|
|
void ResetPadData();
|
|
|
|
|
2020-05-09 20:44:45 +00:00
|
|
|
const std::string m_name;
|
2019-09-06 15:09:30 +00:00
|
|
|
const int m_index;
|
|
|
|
sf::UdpSocket m_socket;
|
2019-11-22 21:49:43 +00:00
|
|
|
SteadyClock::time_point m_next_reregister = SteadyClock::time_point::min();
|
2019-11-22 20:51:25 +00:00
|
|
|
Proto::MessageType::PadDataResponse m_pad_data{};
|
|
|
|
Proto::Touch m_prev_touch{};
|
|
|
|
bool m_prev_touch_valid = false;
|
|
|
|
int m_touch_x = 0;
|
|
|
|
int m_touch_y = 0;
|
2022-10-22 22:33:21 +00:00
|
|
|
const std::string m_server_address;
|
|
|
|
const u16 m_server_port;
|
2021-05-10 19:43:41 +00:00
|
|
|
|
2023-02-16 03:23:47 +00:00
|
|
|
s16 m_touch_x_min = 0;
|
|
|
|
s16 m_touch_y_min = 0;
|
|
|
|
s16 m_touch_x_max = 0;
|
|
|
|
s16 m_touch_y_max = 0;
|
2022-10-22 22:33:21 +00:00
|
|
|
|
|
|
|
const u32 m_client_uid;
|
2019-09-06 15:09:30 +00:00
|
|
|
};
|
|
|
|
|
2019-11-09 00:23:22 +00:00
|
|
|
using MathUtil::GRAVITY_ACCELERATION;
|
2019-11-22 20:55:11 +00:00
|
|
|
constexpr auto SERVER_REREGISTER_INTERVAL = std::chrono::seconds{1};
|
|
|
|
constexpr auto SERVER_LISTPORTS_INTERVAL = std::chrono::seconds{1};
|
|
|
|
constexpr int TOUCH_X_AXIS_MAX = 1000;
|
|
|
|
constexpr int TOUCH_Y_AXIS_MAX = 500;
|
2021-05-15 09:28:52 +00:00
|
|
|
constexpr auto THREAD_MAX_WAIT_INTERVAL = std::chrono::milliseconds{250};
|
|
|
|
constexpr auto SERVER_UNRESPONSIVE_INTERVAL = std::chrono::seconds{1}; // Can be 0
|
|
|
|
constexpr u32 SERVER_ASKED_PADS = 4;
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2020-05-09 20:44:45 +00:00
|
|
|
struct Server
|
|
|
|
{
|
|
|
|
Server(std::string description, std::string address, u16 port)
|
|
|
|
: m_description{std::move(description)}, m_address{std::move(address)}, m_port{port}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
Server(const Server&) = delete;
|
|
|
|
Server(Server&& other) noexcept
|
|
|
|
{
|
|
|
|
m_description = std::move(other.m_description);
|
|
|
|
m_address = std::move(other.m_address);
|
|
|
|
m_port = other.m_port;
|
|
|
|
m_port_info = std::move(other.m_port_info);
|
|
|
|
}
|
|
|
|
|
|
|
|
Server& operator=(const Server&) = delete;
|
|
|
|
Server& operator=(Server&&) = delete;
|
|
|
|
|
|
|
|
~Server() = default;
|
|
|
|
|
|
|
|
std::string m_description;
|
|
|
|
std::string m_address;
|
2023-02-16 03:23:47 +00:00
|
|
|
u16 m_port = 0;
|
2021-09-04 04:43:19 +00:00
|
|
|
std::array<Proto::MessageType::PortInfo, Proto::PORT_COUNT> m_port_info{};
|
2020-05-09 20:44:45 +00:00
|
|
|
sf::UdpSocket m_socket;
|
2021-05-15 09:28:52 +00:00
|
|
|
SteadyClock::time_point m_disconnect_time = SteadyClock::now();
|
2020-05-09 20:44:45 +00:00
|
|
|
};
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
class InputBackend final : public ciface::InputBackend
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
InputBackend(ControllerInterface* controller_interface);
|
|
|
|
~InputBackend();
|
|
|
|
void PopulateDevices() override;
|
|
|
|
|
|
|
|
private:
|
|
|
|
void ConfigChanged();
|
|
|
|
void Restart();
|
|
|
|
|
|
|
|
void HotplugThreadFunc();
|
|
|
|
void StartHotplugThread();
|
|
|
|
void StopHotplugThread();
|
|
|
|
|
2023-02-16 03:23:47 +00:00
|
|
|
bool m_servers_enabled = false;
|
2022-10-22 22:33:21 +00:00
|
|
|
std::vector<Server> m_servers;
|
2023-02-16 03:23:47 +00:00
|
|
|
u32 m_client_uid = 0;
|
2022-10-22 22:33:21 +00:00
|
|
|
SteadyClock::time_point m_next_listports_time;
|
|
|
|
std::thread m_hotplug_thread;
|
|
|
|
Common::Flag m_hotplug_thread_running;
|
2023-08-16 19:37:12 +00:00
|
|
|
Config::ConfigChangedCallbackID m_config_change_callback_id;
|
2022-10-22 22:33:21 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
std::unique_ptr<ciface::InputBackend> CreateInputBackend(ControllerInterface* controller_interface)
|
|
|
|
{
|
|
|
|
return std::make_unique<InputBackend>(controller_interface);
|
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
|
|
|
|
static bool IsSameController(const Proto::MessageType::PortInfo& a,
|
|
|
|
const Proto::MessageType::PortInfo& b)
|
|
|
|
{
|
|
|
|
// compare everything but battery_status
|
2019-11-22 21:07:12 +00:00
|
|
|
return std::tie(a.pad_id, a.pad_state, a.model, a.connection_type, a.pad_mac_address) ==
|
|
|
|
std::tie(b.pad_id, b.pad_state, b.model, b.connection_type, b.pad_mac_address);
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
void InputBackend::HotplugThreadFunc()
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2019-10-26 16:05:16 +00:00
|
|
|
Common::SetCurrentThreadName("DualShockUDPClient Hotplug Thread");
|
2021-05-04 20:37:06 +00:00
|
|
|
INFO_LOG_FMT(CONTROLLERINTERFACE, "DualShockUDPClient hotplug thread started");
|
2021-05-15 09:28:30 +00:00
|
|
|
Common::ScopeGuard thread_stop_guard{
|
|
|
|
[] { INFO_LOG_FMT(CONTROLLERINTERFACE, "DualShockUDPClient hotplug thread stopped"); }};
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
std::vector<bool> timed_out_servers(m_servers.size(), false);
|
2021-05-15 09:28:52 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
while (m_hotplug_thread_running.IsSet())
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-05-15 09:28:52 +00:00
|
|
|
using namespace std::chrono;
|
|
|
|
using namespace std::chrono_literals;
|
|
|
|
|
2019-11-22 21:49:43 +00:00
|
|
|
const auto now = SteadyClock::now();
|
2022-10-22 22:33:21 +00:00
|
|
|
if (now >= m_next_listports_time)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2022-10-22 22:33:21 +00:00
|
|
|
m_next_listports_time = now + SERVER_LISTPORTS_INTERVAL;
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
for (size_t i = 0; i < m_servers.size(); ++i)
|
2020-05-09 20:44:45 +00:00
|
|
|
{
|
2022-10-22 22:33:21 +00:00
|
|
|
auto& server = m_servers[i];
|
|
|
|
Proto::Message<Proto::MessageType::ListPorts> msg(m_client_uid);
|
2020-05-09 20:44:45 +00:00
|
|
|
auto& list_ports = msg.m_message;
|
2021-05-15 09:28:52 +00:00
|
|
|
// We ask for x possible devices. We will receive a message for every connected device.
|
|
|
|
list_ports.pad_request_count = SERVER_ASKED_PADS;
|
|
|
|
list_ports.pad_ids = {0, 1, 2, 3};
|
2020-05-09 20:44:45 +00:00
|
|
|
msg.Finish();
|
|
|
|
if (server.m_socket.send(&list_ports, sizeof list_ports, server.m_address, server.m_port) !=
|
|
|
|
sf::Socket::Status::Done)
|
|
|
|
{
|
2021-05-04 20:37:06 +00:00
|
|
|
ERROR_LOG_FMT(CONTROLLERINTERFACE, "DualShockUDPClient HotplugThreadFunc send failed");
|
2020-05-09 20:44:45 +00:00
|
|
|
}
|
2021-05-15 09:28:52 +00:00
|
|
|
timed_out_servers[i] = true;
|
2020-05-09 20:44:45 +00:00
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
2021-03-03 23:57:18 +00:00
|
|
|
sf::SocketSelector selector;
|
2022-10-22 22:33:21 +00:00
|
|
|
for (auto& server : m_servers)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-03-03 23:57:18 +00:00
|
|
|
selector.add(server.m_socket);
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
auto timeout = duration_cast<milliseconds>(m_next_listports_time - SteadyClock::now());
|
2021-03-03 23:57:18 +00:00
|
|
|
|
2021-05-15 09:28:52 +00:00
|
|
|
// Receive controller port info within a time from our request.
|
|
|
|
// Run this even if we sent no new requests, to disconnect devices,
|
|
|
|
// sleep (wait) the thread and catch old responses.
|
|
|
|
do
|
2021-03-03 23:57:18 +00:00
|
|
|
{
|
2021-05-15 09:28:52 +00:00
|
|
|
// Selector's wait treats a timeout of zero as infinite timeout, which we don't want,
|
|
|
|
// but we also don't want risk waiting for the whole SERVER_LISTPORTS_INTERVAL and hang
|
|
|
|
// the thead trying to close this one in case we received no answers.
|
|
|
|
const auto current_timeout = std::max(std::min(timeout, THREAD_MAX_WAIT_INTERVAL), 1ms);
|
|
|
|
timeout -= current_timeout;
|
|
|
|
// This will return at the first answer
|
|
|
|
if (selector.wait(sf::milliseconds(current_timeout.count())))
|
2021-03-03 23:57:18 +00:00
|
|
|
{
|
2021-05-15 09:28:52 +00:00
|
|
|
// Now check all the servers because we don't know which one(s) sent a reply
|
2022-10-22 22:33:21 +00:00
|
|
|
for (size_t i = 0; i < m_servers.size(); ++i)
|
2021-05-15 09:28:52 +00:00
|
|
|
{
|
2022-10-22 22:33:21 +00:00
|
|
|
auto& server = m_servers[i];
|
2021-05-15 09:28:52 +00:00
|
|
|
if (!selector.isReady(server.m_socket))
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
Proto::Message<Proto::MessageType::FromServer> msg;
|
|
|
|
std::size_t received_bytes;
|
|
|
|
sf::IpAddress sender;
|
|
|
|
u16 port;
|
|
|
|
if (server.m_socket.receive(&msg, sizeof(msg), received_bytes, sender, port) !=
|
|
|
|
sf::Socket::Status::Done)
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (auto port_info = msg.CheckAndCastTo<Proto::MessageType::PortInfo>())
|
|
|
|
{
|
|
|
|
server.m_disconnect_time = SteadyClock::now() + SERVER_UNRESPONSIVE_INTERVAL;
|
|
|
|
// We have receive at least one valid update, that's enough. This is needed to avoid
|
|
|
|
// false positive when checking for disconnection in case our thread waited too long
|
|
|
|
timed_out_servers[i] = false;
|
|
|
|
|
|
|
|
const bool port_changed =
|
|
|
|
!IsSameController(*port_info, server.m_port_info[port_info->pad_id]);
|
|
|
|
if (port_changed)
|
|
|
|
{
|
|
|
|
server.m_port_info[port_info->pad_id] = *port_info;
|
|
|
|
// Just remove and re-add all the devices for simplicity
|
2022-10-22 22:33:21 +00:00
|
|
|
GetControllerInterface().PlatformPopulateDevices([this] { PopulateDevices(); });
|
2021-05-15 09:28:52 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-03-03 23:57:18 +00:00
|
|
|
}
|
2022-10-22 22:33:21 +00:00
|
|
|
if (!m_hotplug_thread_running.IsSet()) // Avoid hanging the thread for too long
|
2021-05-15 09:28:52 +00:00
|
|
|
return;
|
|
|
|
} while (timeout > 0ms);
|
2021-03-03 23:57:18 +00:00
|
|
|
|
2021-05-15 09:28:52 +00:00
|
|
|
// If we have failed to receive any information from the server (or even send it),
|
|
|
|
// disconnect all devices from it (after enough time has elapsed, to avoid false positives).
|
2022-10-22 22:33:21 +00:00
|
|
|
for (size_t i = 0; i < m_servers.size(); ++i)
|
2021-05-15 09:28:52 +00:00
|
|
|
{
|
2022-10-22 22:33:21 +00:00
|
|
|
auto& server = m_servers[i];
|
2021-05-15 09:28:52 +00:00
|
|
|
if (timed_out_servers[i] && SteadyClock::now() >= server.m_disconnect_time)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-05-15 09:28:52 +00:00
|
|
|
bool any_connected = false;
|
|
|
|
for (size_t port_index = 0; port_index < server.m_port_info.size(); port_index++)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-05-15 09:28:52 +00:00
|
|
|
any_connected = any_connected ||
|
|
|
|
server.m_port_info[port_index].pad_state == Proto::DsState::Connected;
|
|
|
|
server.m_port_info[port_index] = {};
|
|
|
|
server.m_port_info[port_index].pad_id = static_cast<u8>(port_index);
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
2021-05-15 09:28:52 +00:00
|
|
|
// We can't only remove devices added by this server as we wouldn't know which they are
|
|
|
|
if (any_connected)
|
2022-10-22 22:33:21 +00:00
|
|
|
GetControllerInterface().PlatformPopulateDevices([this] { PopulateDevices(); });
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
void InputBackend::StartHotplugThread()
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
// Mark the thread as running.
|
2022-10-22 22:33:21 +00:00
|
|
|
if (!m_hotplug_thread_running.TestAndSet())
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
// It was already running.
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
m_hotplug_thread = std::thread(&InputBackend::HotplugThreadFunc, this);
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
void InputBackend::StopHotplugThread()
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
// Tell the hotplug thread to stop.
|
2022-10-22 22:33:21 +00:00
|
|
|
if (!m_hotplug_thread_running.TestAndClear())
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
// It wasn't running, we're done.
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
m_hotplug_thread.join();
|
2021-05-15 09:28:30 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
for (auto& server : m_servers)
|
2020-05-09 20:44:45 +00:00
|
|
|
{
|
|
|
|
server.m_socket.unbind(); // interrupt blocking socket
|
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
2021-05-15 09:28:30 +00:00
|
|
|
// Also just start
|
2022-10-22 22:33:21 +00:00
|
|
|
void InputBackend::Restart()
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-05-04 20:37:06 +00:00
|
|
|
INFO_LOG_FMT(CONTROLLERINTERFACE, "DualShockUDPClient Restart");
|
2019-10-22 23:49:48 +00:00
|
|
|
|
|
|
|
StopHotplugThread();
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
for (auto& server : m_servers)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2020-05-09 20:44:45 +00:00
|
|
|
for (size_t port_index = 0; port_index < server.m_port_info.size(); port_index++)
|
|
|
|
{
|
|
|
|
server.m_port_info[port_index] = {};
|
|
|
|
server.m_port_info[port_index].pad_id = static_cast<u8>(port_index);
|
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
2021-05-15 09:28:30 +00:00
|
|
|
// Only removes devices as servers have been cleaned
|
2022-10-22 22:33:21 +00:00
|
|
|
GetControllerInterface().PlatformPopulateDevices([this] { PopulateDevices(); });
|
2019-10-22 23:49:48 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
m_client_uid = Common::Random::GenerateValue<u32>();
|
|
|
|
m_next_listports_time = SteadyClock::now();
|
2021-05-15 09:25:31 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
if (m_servers_enabled && !m_servers.empty())
|
2019-09-06 15:09:30 +00:00
|
|
|
StartHotplugThread();
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
void InputBackend::ConfigChanged()
|
2019-10-22 23:49:48 +00:00
|
|
|
{
|
2020-05-09 20:44:45 +00:00
|
|
|
const bool servers_enabled = Config::Get(Settings::SERVERS_ENABLED);
|
|
|
|
const std::string servers_setting = Config::Get(Settings::SERVERS);
|
|
|
|
|
|
|
|
std::string new_servers_setting;
|
2022-10-22 22:33:21 +00:00
|
|
|
for (const auto& server : m_servers)
|
2020-05-09 20:44:45 +00:00
|
|
|
{
|
|
|
|
new_servers_setting +=
|
|
|
|
fmt::format("{}:{}:{};", server.m_description, server.m_address, server.m_port);
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
if (servers_enabled != m_servers_enabled || servers_setting != new_servers_setting)
|
2019-10-22 23:49:48 +00:00
|
|
|
{
|
2022-10-22 22:33:21 +00:00
|
|
|
// Stop the thread before writing to m_servers
|
2021-05-15 09:28:30 +00:00
|
|
|
StopHotplugThread();
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
m_servers_enabled = servers_enabled;
|
|
|
|
m_servers.clear();
|
2020-05-09 20:44:45 +00:00
|
|
|
|
|
|
|
const auto server_details = SplitString(servers_setting, ';');
|
|
|
|
for (const auto& server_detail : server_details)
|
|
|
|
{
|
|
|
|
const auto server_info = SplitString(server_detail, ':');
|
|
|
|
if (server_info.size() < 3)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
const std::string description = server_info[0];
|
|
|
|
const std::string server_address = server_info[1];
|
|
|
|
const auto port = std::stoi(server_info[2]);
|
|
|
|
if (port >= std::numeric_limits<u16>::max())
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
u16 server_port = static_cast<u16>(port);
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
m_servers.emplace_back(description, server_address, server_port);
|
2020-05-09 20:44:45 +00:00
|
|
|
}
|
2019-10-22 23:49:48 +00:00
|
|
|
Restart();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
InputBackend::InputBackend(ControllerInterface* controller_interface)
|
|
|
|
: ciface::InputBackend(controller_interface)
|
2019-10-22 23:49:48 +00:00
|
|
|
{
|
2020-07-13 05:26:43 +00:00
|
|
|
// The following is added for backwards compatibility
|
|
|
|
const auto server_address_setting = Config::Get(Settings::SERVER_ADDRESS);
|
|
|
|
const auto server_port_setting = Config::Get(Settings::SERVER_PORT);
|
|
|
|
|
|
|
|
if (!server_address_setting.empty() && server_port_setting != 0)
|
|
|
|
{
|
|
|
|
const auto& servers_setting = Config::Get(ciface::DualShockUDPClient::Settings::SERVERS);
|
|
|
|
Config::SetBaseOrCurrent(ciface::DualShockUDPClient::Settings::SERVERS,
|
|
|
|
servers_setting + fmt::format("{}:{}:{};", "DS4",
|
|
|
|
server_address_setting,
|
|
|
|
server_port_setting));
|
|
|
|
Config::SetBase(Settings::SERVER_ADDRESS, "");
|
|
|
|
Config::SetBase(Settings::SERVER_PORT, 0);
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
m_config_change_callback_id =
|
|
|
|
Config::AddConfigChangedCallback(std::bind(&InputBackend::ConfigChanged, this));
|
|
|
|
// Call it immediately to load settings
|
|
|
|
ConfigChanged();
|
2019-10-22 23:49:48 +00:00
|
|
|
}
|
|
|
|
|
2021-05-15 09:28:30 +00:00
|
|
|
// This can be called by the host thread as well as the hotplug thread, concurrently.
|
|
|
|
// So use PlatformPopulateDevices().
|
2022-10-22 22:33:21 +00:00
|
|
|
// m_servers is already safe because it can only be modified when the DSU thread is not running,
|
2021-05-15 09:28:30 +00:00
|
|
|
// from the main thread
|
2022-10-22 22:33:21 +00:00
|
|
|
void InputBackend::PopulateDevices()
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2021-05-04 20:37:06 +00:00
|
|
|
INFO_LOG_FMT(CONTROLLERINTERFACE, "DualShockUDPClient PopulateDevices");
|
2019-09-06 15:09:30 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
// m_servers has already been updated so we can't use it to know which devices we removed,
|
2021-01-02 17:32:53 +00:00
|
|
|
// also it's good to remove all of them before adding new ones so that their id will be set
|
|
|
|
// correctly if they have the same name
|
2022-10-22 22:33:21 +00:00
|
|
|
GetControllerInterface().RemoveDevice(
|
2021-01-02 17:32:53 +00:00
|
|
|
[](const auto* dev) { return dev->GetSource() == DUALSHOCKUDP_SOURCE_NAME; });
|
|
|
|
|
2021-05-15 09:28:30 +00:00
|
|
|
// Users might have created more than one server on the same IP/Port.
|
|
|
|
// Devices might end up being duplicated (if the server responds two all requests)
|
|
|
|
// but they won't conflict.
|
2022-10-22 22:33:21 +00:00
|
|
|
for (const auto& server : m_servers)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2020-05-09 20:44:45 +00:00
|
|
|
for (size_t port_index = 0; port_index < server.m_port_info.size(); port_index++)
|
|
|
|
{
|
|
|
|
const Proto::MessageType::PortInfo& port_info = server.m_port_info[port_index];
|
|
|
|
if (port_info.pad_state != Proto::DsState::Connected)
|
|
|
|
continue;
|
2019-11-22 21:07:12 +00:00
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
GetControllerInterface().AddDevice(
|
|
|
|
std::make_shared<Device>(server.m_description, static_cast<int>(port_index),
|
|
|
|
server.m_address, server.m_port, m_client_uid));
|
2020-05-09 20:44:45 +00:00
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
InputBackend::~InputBackend()
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
2022-10-22 22:33:21 +00:00
|
|
|
Config::RemoveConfigChangedCallback(m_config_change_callback_id);
|
2019-09-06 15:09:30 +00:00
|
|
|
StopHotplugThread();
|
|
|
|
}
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
Device::Device(std::string name, int index, std::string server_address, u16 server_port,
|
|
|
|
u32 client_uid)
|
2020-05-09 20:44:45 +00:00
|
|
|
: m_name{std::move(name)}, m_index{index}, m_server_address{std::move(server_address)},
|
2022-10-22 22:33:21 +00:00
|
|
|
m_server_port{server_port}, m_client_uid(client_uid)
|
2019-09-06 15:09:30 +00:00
|
|
|
{
|
|
|
|
m_socket.setBlocking(false);
|
|
|
|
|
|
|
|
AddInput(new AnalogInput<u8>("Pad W", m_pad_data.button_dpad_left_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Pad S", m_pad_data.button_dpad_down_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Pad E", m_pad_data.button_dpad_right_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Pad N", m_pad_data.button_dpad_up_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Square", m_pad_data.button_square_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Cross", m_pad_data.button_cross_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Circle", m_pad_data.button_circle_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("Triangle", m_pad_data.button_triangle_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("L1", m_pad_data.button_l1_analog, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("R1", m_pad_data.button_r1_analog, 255));
|
|
|
|
|
|
|
|
AddInput(new AnalogInput<u8>("L2", m_pad_data.trigger_l2, 255));
|
|
|
|
AddInput(new AnalogInput<u8>("R2", m_pad_data.trigger_r2, 255));
|
|
|
|
|
|
|
|
AddInput(new Button<u8>("L3", m_pad_data.button_states1, 0x2));
|
|
|
|
AddInput(new Button<u8>("R3", m_pad_data.button_states1, 0x4));
|
|
|
|
AddInput(new Button<u8>("Share", m_pad_data.button_states1, 0x1));
|
|
|
|
AddInput(new Button<u8>("Options", m_pad_data.button_states1, 0x8));
|
|
|
|
AddInput(new Button<u8>("PS", m_pad_data.button_ps, 0x1));
|
|
|
|
AddInput(new Button<u8>("Touch Button", m_pad_data.button_touch, 0x1));
|
|
|
|
|
|
|
|
AddInput(new AnalogInput<u8>("Left X-", m_pad_data.left_stick_x, -128, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Left X+", m_pad_data.left_stick_x, 127, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Left Y-", m_pad_data.left_stick_y_inverted, -128, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Left Y+", m_pad_data.left_stick_y_inverted, 127, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Right X-", m_pad_data.right_stick_x, -128, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Right X+", m_pad_data.right_stick_x, 127, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Right Y-", m_pad_data.right_stick_y_inverted, -128, -128));
|
|
|
|
AddInput(new AnalogInput<u8>("Right Y+", m_pad_data.right_stick_y_inverted, 127, -128));
|
|
|
|
|
|
|
|
AddInput(new TouchInput("Touch X-", m_touch_x, -TOUCH_X_AXIS_MAX));
|
|
|
|
AddInput(new TouchInput("Touch X+", m_touch_x, TOUCH_X_AXIS_MAX));
|
|
|
|
AddInput(new TouchInput("Touch Y-", m_touch_y, -TOUCH_Y_AXIS_MAX));
|
|
|
|
AddInput(new TouchInput("Touch Y+", m_touch_y, TOUCH_Y_AXIS_MAX));
|
|
|
|
|
2019-11-17 23:10:35 +00:00
|
|
|
// Convert Gs to meters per second squared
|
|
|
|
constexpr auto accel_scale = 1.0 / GRAVITY_ACCELERATION;
|
|
|
|
|
|
|
|
AddInput(new AccelerometerInput("Accel Up", m_pad_data.accelerometer_y_g, -accel_scale));
|
|
|
|
AddInput(new AccelerometerInput("Accel Down", m_pad_data.accelerometer_y_g, accel_scale));
|
|
|
|
AddInput(new AccelerometerInput("Accel Left", m_pad_data.accelerometer_x_g, accel_scale));
|
|
|
|
AddInput(new AccelerometerInput("Accel Right", m_pad_data.accelerometer_x_g, -accel_scale));
|
|
|
|
AddInput(new AccelerometerInput("Accel Forward", m_pad_data.accelerometer_z_g, accel_scale));
|
|
|
|
AddInput(new AccelerometerInput("Accel Backward", m_pad_data.accelerometer_z_g, -accel_scale));
|
|
|
|
|
|
|
|
// Convert degrees per second to radians per second
|
|
|
|
constexpr auto gyro_scale = 360.0 / MathUtil::TAU;
|
|
|
|
|
|
|
|
AddInput(new GyroInput("Gyro Pitch Up", m_pad_data.gyro_pitch_deg_s, gyro_scale));
|
|
|
|
AddInput(new GyroInput("Gyro Pitch Down", m_pad_data.gyro_pitch_deg_s, -gyro_scale));
|
|
|
|
AddInput(new GyroInput("Gyro Roll Left", m_pad_data.gyro_roll_deg_s, -gyro_scale));
|
|
|
|
AddInput(new GyroInput("Gyro Roll Right", m_pad_data.gyro_roll_deg_s, gyro_scale));
|
|
|
|
AddInput(new GyroInput("Gyro Yaw Left", m_pad_data.gyro_yaw_deg_s, -gyro_scale));
|
|
|
|
AddInput(new GyroInput("Gyro Yaw Right", m_pad_data.gyro_yaw_deg_s, gyro_scale));
|
2021-05-10 19:43:41 +00:00
|
|
|
|
|
|
|
AddInput(new BatteryInput(m_pad_data.battery_status));
|
|
|
|
|
|
|
|
m_touch_x_min = 0;
|
|
|
|
m_touch_y_min = 0;
|
|
|
|
// DS4 touchpad max values
|
|
|
|
m_touch_x_max = 1919;
|
|
|
|
m_touch_y_max = 941;
|
|
|
|
|
|
|
|
ResetPadData();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Device::ResetPadData()
|
|
|
|
{
|
|
|
|
m_pad_data = Proto::MessageType::PadDataResponse{};
|
|
|
|
|
|
|
|
// Make sure they start from resting values, not from 0
|
|
|
|
m_touch_x = m_touch_x_min + ((m_touch_x_max - m_touch_x_min) / 2.0);
|
|
|
|
m_touch_y = m_touch_y_min + ((m_touch_y_max - m_touch_y_min) / 2.0);
|
|
|
|
m_pad_data.left_stick_x = 128;
|
|
|
|
m_pad_data.left_stick_y_inverted = 128;
|
|
|
|
m_pad_data.right_stick_x = 128;
|
|
|
|
m_pad_data.right_stick_y_inverted = 128;
|
|
|
|
m_pad_data.touch1.x = m_touch_x;
|
|
|
|
m_pad_data.touch1.y = m_touch_y;
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
std::string Device::GetName() const
|
|
|
|
{
|
2020-05-09 20:44:45 +00:00
|
|
|
return m_name;
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
std::string Device::GetSource() const
|
|
|
|
{
|
2021-01-02 17:32:53 +00:00
|
|
|
return std::string(DUALSHOCKUDP_SOURCE_NAME);
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void Device::UpdateInput()
|
|
|
|
{
|
|
|
|
// Regularly tell the UDP server to feed us controller data
|
2019-11-22 21:49:43 +00:00
|
|
|
const auto now = SteadyClock::now();
|
2019-09-06 15:09:30 +00:00
|
|
|
if (now >= m_next_reregister)
|
|
|
|
{
|
|
|
|
m_next_reregister = now + SERVER_REREGISTER_INTERVAL;
|
|
|
|
|
2022-10-22 22:33:21 +00:00
|
|
|
Proto::Message<Proto::MessageType::PadDataRequest> msg(m_client_uid);
|
2019-09-06 15:09:30 +00:00
|
|
|
auto& data_req = msg.m_message;
|
|
|
|
data_req.register_flags = Proto::RegisterFlags::PadID;
|
|
|
|
data_req.pad_id_to_register = m_index;
|
|
|
|
msg.Finish();
|
2020-05-09 20:44:45 +00:00
|
|
|
if (m_socket.send(&data_req, sizeof(data_req), m_server_address, m_server_port) !=
|
2019-09-06 15:09:30 +00:00
|
|
|
sf::Socket::Status::Done)
|
2020-10-22 21:49:29 +00:00
|
|
|
{
|
2021-05-04 20:37:06 +00:00
|
|
|
ERROR_LOG_FMT(CONTROLLERINTERFACE, "DualShockUDPClient UpdateInput send failed");
|
2020-10-22 21:49:29 +00:00
|
|
|
}
|
2019-09-06 15:09:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Receive and handle controller data
|
|
|
|
Proto::Message<Proto::MessageType::FromServer> msg;
|
|
|
|
std::size_t received_bytes;
|
|
|
|
sf::IpAddress sender;
|
|
|
|
u16 port;
|
|
|
|
while (m_socket.receive(&msg, sizeof msg, received_bytes, sender, port) ==
|
|
|
|
sf::Socket::Status::Done)
|
|
|
|
{
|
|
|
|
if (auto pad_data = msg.CheckAndCastTo<Proto::MessageType::PadDataResponse>())
|
|
|
|
{
|
|
|
|
m_pad_data = *pad_data;
|
|
|
|
|
|
|
|
// Update touch pad relative coordinates
|
|
|
|
if (m_pad_data.touch1.id != m_prev_touch.id)
|
|
|
|
m_prev_touch_valid = false;
|
|
|
|
if (m_prev_touch_valid)
|
|
|
|
{
|
|
|
|
m_touch_x += m_pad_data.touch1.x - m_prev_touch.x;
|
|
|
|
m_touch_y += m_pad_data.touch1.y - m_prev_touch.y;
|
|
|
|
m_touch_x = std::clamp(m_touch_x, -TOUCH_X_AXIS_MAX, TOUCH_X_AXIS_MAX);
|
|
|
|
m_touch_y = std::clamp(m_touch_y, -TOUCH_Y_AXIS_MAX, TOUCH_Y_AXIS_MAX);
|
|
|
|
}
|
|
|
|
m_prev_touch = m_pad_data.touch1;
|
|
|
|
m_prev_touch_valid = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::optional<int> Device::GetPreferredId() const
|
|
|
|
{
|
|
|
|
return m_index;
|
|
|
|
}
|
|
|
|
|
2019-10-26 16:05:16 +00:00
|
|
|
} // namespace ciface::DualShockUDPClient
|