udp_client.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650
  1. // SPDX-FileCopyrightText: 2018 Citra Emulator Project
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  3. #include <random>
  4. #include <boost/asio.hpp>
  5. #include <fmt/format.h>
  6. #include "common/logging/log.h"
  7. #include "common/param_package.h"
  8. #include "common/settings.h"
  9. #include "input_common/drivers/udp_client.h"
  10. #include "input_common/helpers/udp_protocol.h"
  11. using boost::asio::ip::udp;
  12. namespace InputCommon::CemuhookUDP {
  13. struct SocketCallback {
  14. std::function<void(Response::Version)> version;
  15. std::function<void(Response::PortInfo)> port_info;
  16. std::function<void(Response::PadData)> pad_data;
  17. };
  18. class Socket {
  19. public:
  20. using clock = std::chrono::system_clock;
  21. explicit Socket(const std::string& host, u16 port, SocketCallback callback_)
  22. : callback(std::move(callback_)), timer(io_service),
  23. socket(io_service, udp::endpoint(udp::v4(), 0)), client_id(GenerateRandomClientId()) {
  24. boost::system::error_code ec{};
  25. auto ipv4 = boost::asio::ip::make_address_v4(host, ec);
  26. if (ec.value() != boost::system::errc::success) {
  27. LOG_ERROR(Input, "Invalid IPv4 address \"{}\" provided to socket", host);
  28. ipv4 = boost::asio::ip::address_v4{};
  29. }
  30. send_endpoint = {udp::endpoint(ipv4, port)};
  31. }
  32. void Stop() {
  33. io_service.stop();
  34. }
  35. void Loop() {
  36. io_service.run();
  37. }
  38. void StartSend(const clock::time_point& from) {
  39. timer.expires_at(from + std::chrono::seconds(3));
  40. timer.async_wait([this](const boost::system::error_code& error) { HandleSend(error); });
  41. }
  42. void StartReceive() {
  43. socket.async_receive_from(
  44. boost::asio::buffer(receive_buffer), receive_endpoint,
  45. [this](const boost::system::error_code& error, std::size_t bytes_transferred) {
  46. HandleReceive(error, bytes_transferred);
  47. });
  48. }
  49. private:
  50. u32 GenerateRandomClientId() const {
  51. std::random_device device;
  52. return device();
  53. }
  54. void HandleReceive(const boost::system::error_code&, std::size_t bytes_transferred) {
  55. if (auto type = Response::Validate(receive_buffer.data(), bytes_transferred)) {
  56. switch (*type) {
  57. case Type::Version: {
  58. Response::Version version;
  59. std::memcpy(&version, &receive_buffer[sizeof(Header)], sizeof(Response::Version));
  60. callback.version(std::move(version));
  61. break;
  62. }
  63. case Type::PortInfo: {
  64. Response::PortInfo port_info;
  65. std::memcpy(&port_info, &receive_buffer[sizeof(Header)],
  66. sizeof(Response::PortInfo));
  67. callback.port_info(std::move(port_info));
  68. break;
  69. }
  70. case Type::PadData: {
  71. Response::PadData pad_data;
  72. std::memcpy(&pad_data, &receive_buffer[sizeof(Header)], sizeof(Response::PadData));
  73. callback.pad_data(std::move(pad_data));
  74. break;
  75. }
  76. }
  77. }
  78. StartReceive();
  79. }
  80. void HandleSend(const boost::system::error_code&) {
  81. boost::system::error_code _ignored{};
  82. // Send a request for getting port info for the pad
  83. const Request::PortInfo port_info{4, {0, 1, 2, 3}};
  84. const auto port_message = Request::Create(port_info, client_id);
  85. std::memcpy(&send_buffer1, &port_message, PORT_INFO_SIZE);
  86. socket.send_to(boost::asio::buffer(send_buffer1), send_endpoint, {}, _ignored);
  87. // Send a request for getting pad data for the pad
  88. const Request::PadData pad_data{
  89. Request::RegisterFlags::AllPads,
  90. 0,
  91. EMPTY_MAC_ADDRESS,
  92. };
  93. const auto pad_message = Request::Create(pad_data, client_id);
  94. std::memcpy(send_buffer2.data(), &pad_message, PAD_DATA_SIZE);
  95. socket.send_to(boost::asio::buffer(send_buffer2), send_endpoint, {}, _ignored);
  96. StartSend(timer.expiry());
  97. }
  98. SocketCallback callback;
  99. boost::asio::io_service io_service;
  100. boost::asio::basic_waitable_timer<clock> timer;
  101. udp::socket socket;
  102. const u32 client_id;
  103. static constexpr std::size_t PORT_INFO_SIZE = sizeof(Message<Request::PortInfo>);
  104. static constexpr std::size_t PAD_DATA_SIZE = sizeof(Message<Request::PadData>);
  105. std::array<u8, PORT_INFO_SIZE> send_buffer1;
  106. std::array<u8, PAD_DATA_SIZE> send_buffer2;
  107. udp::endpoint send_endpoint;
  108. std::array<u8, MAX_PACKET_SIZE> receive_buffer;
  109. udp::endpoint receive_endpoint;
  110. };
  111. static void SocketLoop(Socket* socket) {
  112. socket->StartReceive();
  113. socket->StartSend(Socket::clock::now());
  114. socket->Loop();
  115. }
  116. UDPClient::UDPClient(std::string input_engine_) : InputEngine(std::move(input_engine_)) {
  117. LOG_INFO(Input, "Udp Initialization started");
  118. ReloadSockets();
  119. }
  120. UDPClient::~UDPClient() {
  121. Reset();
  122. }
  123. UDPClient::ClientConnection::ClientConnection() = default;
  124. UDPClient::ClientConnection::~ClientConnection() = default;
  125. void UDPClient::ReloadSockets() {
  126. Reset();
  127. std::stringstream servers_ss(Settings::values.udp_input_servers.GetValue());
  128. std::string server_token;
  129. std::size_t client = 0;
  130. while (std::getline(servers_ss, server_token, ',')) {
  131. if (client == MAX_UDP_CLIENTS) {
  132. break;
  133. }
  134. std::stringstream server_ss(server_token);
  135. std::string token;
  136. std::getline(server_ss, token, ':');
  137. std::string udp_input_address = token;
  138. std::getline(server_ss, token, ':');
  139. char* temp;
  140. const u16 udp_input_port = static_cast<u16>(std::strtol(token.c_str(), &temp, 0));
  141. if (*temp != '\0') {
  142. LOG_ERROR(Input, "Port number is not valid {}", token);
  143. continue;
  144. }
  145. const std::size_t client_number = GetClientNumber(udp_input_address, udp_input_port);
  146. if (client_number != MAX_UDP_CLIENTS) {
  147. LOG_ERROR(Input, "Duplicated UDP servers found");
  148. continue;
  149. }
  150. StartCommunication(client++, udp_input_address, udp_input_port);
  151. }
  152. }
  153. std::size_t UDPClient::GetClientNumber(std::string_view host, u16 port) const {
  154. for (std::size_t client = 0; client < clients.size(); client++) {
  155. if (clients[client].active == -1) {
  156. continue;
  157. }
  158. if (clients[client].host == host && clients[client].port == port) {
  159. return client;
  160. }
  161. }
  162. return MAX_UDP_CLIENTS;
  163. }
  164. Common::Input::BatteryLevel UDPClient::GetBatteryLevel(Response::Battery battery) const {
  165. switch (battery) {
  166. case Response::Battery::Dying:
  167. return Common::Input::BatteryLevel::Empty;
  168. case Response::Battery::Low:
  169. return Common::Input::BatteryLevel::Critical;
  170. case Response::Battery::Medium:
  171. return Common::Input::BatteryLevel::Low;
  172. case Response::Battery::High:
  173. return Common::Input::BatteryLevel::Medium;
  174. case Response::Battery::Full:
  175. case Response::Battery::Charged:
  176. return Common::Input::BatteryLevel::Full;
  177. case Response::Battery::Charging:
  178. default:
  179. return Common::Input::BatteryLevel::Charging;
  180. }
  181. }
  182. void UDPClient::OnVersion([[maybe_unused]] Response::Version data) {
  183. LOG_TRACE(Input, "Version packet received: {}", data.version);
  184. }
  185. void UDPClient::OnPortInfo([[maybe_unused]] Response::PortInfo data) {
  186. LOG_TRACE(Input, "PortInfo packet received: {}", data.model);
  187. }
  188. void UDPClient::OnPadData(Response::PadData data, std::size_t client) {
  189. const std::size_t pad_index = (client * PADS_PER_CLIENT) + data.info.id;
  190. if (pad_index >= pads.size()) {
  191. LOG_ERROR(Input, "Invalid pad id {}", data.info.id);
  192. return;
  193. }
  194. LOG_TRACE(Input, "PadData packet received");
  195. if (data.packet_counter == pads[pad_index].packet_sequence) {
  196. LOG_WARNING(
  197. Input,
  198. "PadData packet dropped because its stale info. Current count: {} Packet count: {}",
  199. pads[pad_index].packet_sequence, data.packet_counter);
  200. pads[pad_index].connected = false;
  201. return;
  202. }
  203. clients[client].active = 1;
  204. pads[pad_index].connected = true;
  205. pads[pad_index].packet_sequence = data.packet_counter;
  206. const auto now = std::chrono::steady_clock::now();
  207. const auto time_difference = static_cast<u64>(
  208. std::chrono::duration_cast<std::chrono::microseconds>(now - pads[pad_index].last_update)
  209. .count());
  210. pads[pad_index].last_update = now;
  211. // Gyroscope values are not it the correct scale from better joy.
  212. // Dividing by 312 allows us to make one full turn = 1 turn
  213. // This must be a configurable valued called sensitivity
  214. const float gyro_scale = 1.0f / 312.0f;
  215. const BasicMotion motion{
  216. .gyro_x = data.gyro.pitch * gyro_scale,
  217. .gyro_y = data.gyro.roll * gyro_scale,
  218. .gyro_z = -data.gyro.yaw * gyro_scale,
  219. .accel_x = data.accel.x,
  220. .accel_y = -data.accel.z,
  221. .accel_z = data.accel.y,
  222. .delta_timestamp = time_difference,
  223. };
  224. const PadIdentifier identifier = GetPadIdentifier(pad_index);
  225. SetMotion(identifier, 0, motion);
  226. for (std::size_t id = 0; id < data.touch.size(); ++id) {
  227. const auto touch_pad = data.touch[id];
  228. const auto touch_axis_x_id =
  229. static_cast<int>(id == 0 ? PadAxes::Touch1X : PadAxes::Touch2X);
  230. const auto touch_axis_y_id =
  231. static_cast<int>(id == 0 ? PadAxes::Touch1Y : PadAxes::Touch2Y);
  232. const auto touch_button_id =
  233. static_cast<int>(id == 0 ? PadButton::Touch1 : PadButton::Touch2);
  234. // TODO: Use custom calibration per device
  235. const Common::ParamPackage touch_param(Settings::values.touch_device.GetValue());
  236. const u16 min_x = static_cast<u16>(touch_param.Get("min_x", 100));
  237. const u16 min_y = static_cast<u16>(touch_param.Get("min_y", 50));
  238. const u16 max_x = static_cast<u16>(touch_param.Get("max_x", 1800));
  239. const u16 max_y = static_cast<u16>(touch_param.Get("max_y", 850));
  240. const f32 x =
  241. static_cast<f32>(std::clamp(static_cast<u16>(touch_pad.x), min_x, max_x) - min_x) /
  242. static_cast<f32>(max_x - min_x);
  243. const f32 y =
  244. static_cast<f32>(std::clamp(static_cast<u16>(touch_pad.y), min_y, max_y) - min_y) /
  245. static_cast<f32>(max_y - min_y);
  246. if (touch_pad.is_active) {
  247. SetAxis(identifier, touch_axis_x_id, x);
  248. SetAxis(identifier, touch_axis_y_id, y);
  249. SetButton(identifier, touch_button_id, true);
  250. continue;
  251. }
  252. SetAxis(identifier, touch_axis_x_id, 0);
  253. SetAxis(identifier, touch_axis_y_id, 0);
  254. SetButton(identifier, touch_button_id, false);
  255. }
  256. SetAxis(identifier, static_cast<int>(PadAxes::LeftStickX),
  257. (data.left_stick_x - 127.0f) / 127.0f);
  258. SetAxis(identifier, static_cast<int>(PadAxes::LeftStickY),
  259. (data.left_stick_y - 127.0f) / 127.0f);
  260. SetAxis(identifier, static_cast<int>(PadAxes::RightStickX),
  261. (data.right_stick_x - 127.0f) / 127.0f);
  262. SetAxis(identifier, static_cast<int>(PadAxes::RightStickY),
  263. (data.right_stick_y - 127.0f) / 127.0f);
  264. static constexpr std::array<PadButton, 16> buttons{
  265. PadButton::Share, PadButton::L3, PadButton::R3, PadButton::Options,
  266. PadButton::Up, PadButton::Right, PadButton::Down, PadButton::Left,
  267. PadButton::L2, PadButton::R2, PadButton::L1, PadButton::R1,
  268. PadButton::Triangle, PadButton::Circle, PadButton::Cross, PadButton::Square};
  269. for (std::size_t i = 0; i < buttons.size(); ++i) {
  270. const bool button_status = (data.digital_button & (1U << i)) != 0;
  271. const int button = static_cast<int>(buttons[i]);
  272. SetButton(identifier, button, button_status);
  273. }
  274. SetButton(identifier, static_cast<int>(PadButton::Home), data.home != 0);
  275. SetButton(identifier, static_cast<int>(PadButton::TouchHardPress), data.touch_hard_press != 0);
  276. SetBattery(identifier, GetBatteryLevel(data.info.battery));
  277. }
  278. void UDPClient::StartCommunication(std::size_t client, const std::string& host, u16 port) {
  279. SocketCallback callback{[this](Response::Version version) { OnVersion(version); },
  280. [this](Response::PortInfo info) { OnPortInfo(info); },
  281. [this, client](Response::PadData data) { OnPadData(data, client); }};
  282. LOG_INFO(Input, "Starting communication with UDP input server on {}:{}", host, port);
  283. clients[client].uuid = GetHostUUID(host);
  284. clients[client].host = host;
  285. clients[client].port = port;
  286. clients[client].active = 0;
  287. clients[client].socket = std::make_unique<Socket>(host, port, callback);
  288. clients[client].thread = std::thread{SocketLoop, clients[client].socket.get()};
  289. for (std::size_t index = 0; index < PADS_PER_CLIENT; ++index) {
  290. const PadIdentifier identifier = GetPadIdentifier(client * PADS_PER_CLIENT + index);
  291. PreSetController(identifier);
  292. }
  293. }
  294. PadIdentifier UDPClient::GetPadIdentifier(std::size_t pad_index) const {
  295. const std::size_t client = pad_index / PADS_PER_CLIENT;
  296. return {
  297. .guid = clients[client].uuid,
  298. .port = static_cast<std::size_t>(clients[client].port),
  299. .pad = pad_index,
  300. };
  301. }
  302. Common::UUID UDPClient::GetHostUUID(const std::string& host) const {
  303. const auto ip = boost::asio::ip::make_address_v4(host);
  304. const auto hex_host = fmt::format("00000000-0000-0000-0000-0000{:06x}", ip.to_uint());
  305. return Common::UUID{hex_host};
  306. }
  307. void UDPClient::Reset() {
  308. for (auto& client : clients) {
  309. if (client.thread.joinable()) {
  310. client.active = -1;
  311. client.socket->Stop();
  312. client.thread.join();
  313. }
  314. }
  315. }
  316. std::vector<Common::ParamPackage> UDPClient::GetInputDevices() const {
  317. std::vector<Common::ParamPackage> devices;
  318. if (!Settings::values.enable_udp_controller) {
  319. return devices;
  320. }
  321. for (std::size_t client = 0; client < clients.size(); client++) {
  322. if (clients[client].active != 1) {
  323. continue;
  324. }
  325. for (std::size_t index = 0; index < PADS_PER_CLIENT; ++index) {
  326. const std::size_t pad_index = client * PADS_PER_CLIENT + index;
  327. if (!pads[pad_index].connected) {
  328. continue;
  329. }
  330. const auto pad_identifier = GetPadIdentifier(pad_index);
  331. Common::ParamPackage identifier{};
  332. identifier.Set("engine", GetEngineName());
  333. identifier.Set("display", fmt::format("UDP Controller {}", pad_identifier.pad));
  334. identifier.Set("guid", pad_identifier.guid.RawString());
  335. identifier.Set("port", static_cast<int>(pad_identifier.port));
  336. identifier.Set("pad", static_cast<int>(pad_identifier.pad));
  337. devices.emplace_back(identifier);
  338. }
  339. }
  340. return devices;
  341. }
  342. ButtonMapping UDPClient::GetButtonMappingForDevice(const Common::ParamPackage& params) {
  343. // This list excludes any button that can't be really mapped
  344. static constexpr std::array<std::pair<Settings::NativeButton::Values, PadButton>, 20>
  345. switch_to_dsu_button = {
  346. std::pair{Settings::NativeButton::A, PadButton::Circle},
  347. {Settings::NativeButton::B, PadButton::Cross},
  348. {Settings::NativeButton::X, PadButton::Triangle},
  349. {Settings::NativeButton::Y, PadButton::Square},
  350. {Settings::NativeButton::Plus, PadButton::Options},
  351. {Settings::NativeButton::Minus, PadButton::Share},
  352. {Settings::NativeButton::DLeft, PadButton::Left},
  353. {Settings::NativeButton::DUp, PadButton::Up},
  354. {Settings::NativeButton::DRight, PadButton::Right},
  355. {Settings::NativeButton::DDown, PadButton::Down},
  356. {Settings::NativeButton::L, PadButton::L1},
  357. {Settings::NativeButton::R, PadButton::R1},
  358. {Settings::NativeButton::ZL, PadButton::L2},
  359. {Settings::NativeButton::ZR, PadButton::R2},
  360. {Settings::NativeButton::SL, PadButton::L2},
  361. {Settings::NativeButton::SR, PadButton::R2},
  362. {Settings::NativeButton::LStick, PadButton::L3},
  363. {Settings::NativeButton::RStick, PadButton::R3},
  364. {Settings::NativeButton::Home, PadButton::Home},
  365. {Settings::NativeButton::Screenshot, PadButton::TouchHardPress},
  366. };
  367. if (!params.Has("guid") || !params.Has("port") || !params.Has("pad")) {
  368. return {};
  369. }
  370. ButtonMapping mapping{};
  371. for (const auto& [switch_button, dsu_button] : switch_to_dsu_button) {
  372. Common::ParamPackage button_params{};
  373. button_params.Set("engine", GetEngineName());
  374. button_params.Set("guid", params.Get("guid", ""));
  375. button_params.Set("port", params.Get("port", 0));
  376. button_params.Set("pad", params.Get("pad", 0));
  377. button_params.Set("button", static_cast<int>(dsu_button));
  378. mapping.insert_or_assign(switch_button, std::move(button_params));
  379. }
  380. return mapping;
  381. }
  382. AnalogMapping UDPClient::GetAnalogMappingForDevice(const Common::ParamPackage& params) {
  383. if (!params.Has("guid") || !params.Has("port") || !params.Has("pad")) {
  384. return {};
  385. }
  386. AnalogMapping mapping = {};
  387. Common::ParamPackage left_analog_params;
  388. left_analog_params.Set("engine", GetEngineName());
  389. left_analog_params.Set("guid", params.Get("guid", ""));
  390. left_analog_params.Set("port", params.Get("port", 0));
  391. left_analog_params.Set("pad", params.Get("pad", 0));
  392. left_analog_params.Set("axis_x", static_cast<int>(PadAxes::LeftStickX));
  393. left_analog_params.Set("axis_y", static_cast<int>(PadAxes::LeftStickY));
  394. mapping.insert_or_assign(Settings::NativeAnalog::LStick, std::move(left_analog_params));
  395. Common::ParamPackage right_analog_params;
  396. right_analog_params.Set("engine", GetEngineName());
  397. right_analog_params.Set("guid", params.Get("guid", ""));
  398. right_analog_params.Set("port", params.Get("port", 0));
  399. right_analog_params.Set("pad", params.Get("pad", 0));
  400. right_analog_params.Set("axis_x", static_cast<int>(PadAxes::RightStickX));
  401. right_analog_params.Set("axis_y", static_cast<int>(PadAxes::RightStickY));
  402. mapping.insert_or_assign(Settings::NativeAnalog::RStick, std::move(right_analog_params));
  403. return mapping;
  404. }
  405. MotionMapping UDPClient::GetMotionMappingForDevice(const Common::ParamPackage& params) {
  406. if (!params.Has("guid") || !params.Has("port") || !params.Has("pad")) {
  407. return {};
  408. }
  409. MotionMapping mapping = {};
  410. Common::ParamPackage left_motion_params;
  411. left_motion_params.Set("engine", GetEngineName());
  412. left_motion_params.Set("guid", params.Get("guid", ""));
  413. left_motion_params.Set("port", params.Get("port", 0));
  414. left_motion_params.Set("pad", params.Get("pad", 0));
  415. left_motion_params.Set("motion", 0);
  416. Common::ParamPackage right_motion_params;
  417. right_motion_params.Set("engine", GetEngineName());
  418. right_motion_params.Set("guid", params.Get("guid", ""));
  419. right_motion_params.Set("port", params.Get("port", 0));
  420. right_motion_params.Set("pad", params.Get("pad", 0));
  421. right_motion_params.Set("motion", 0);
  422. mapping.insert_or_assign(Settings::NativeMotion::MotionLeft, std::move(left_motion_params));
  423. mapping.insert_or_assign(Settings::NativeMotion::MotionRight, std::move(right_motion_params));
  424. return mapping;
  425. }
  426. Common::Input::ButtonNames UDPClient::GetUIButtonName(const Common::ParamPackage& params) const {
  427. PadButton button = static_cast<PadButton>(params.Get("button", 0));
  428. switch (button) {
  429. case PadButton::Left:
  430. return Common::Input::ButtonNames::ButtonLeft;
  431. case PadButton::Right:
  432. return Common::Input::ButtonNames::ButtonRight;
  433. case PadButton::Down:
  434. return Common::Input::ButtonNames::ButtonDown;
  435. case PadButton::Up:
  436. return Common::Input::ButtonNames::ButtonUp;
  437. case PadButton::L1:
  438. return Common::Input::ButtonNames::L1;
  439. case PadButton::L2:
  440. return Common::Input::ButtonNames::L2;
  441. case PadButton::L3:
  442. return Common::Input::ButtonNames::L3;
  443. case PadButton::R1:
  444. return Common::Input::ButtonNames::R1;
  445. case PadButton::R2:
  446. return Common::Input::ButtonNames::R2;
  447. case PadButton::R3:
  448. return Common::Input::ButtonNames::R3;
  449. case PadButton::Circle:
  450. return Common::Input::ButtonNames::Circle;
  451. case PadButton::Cross:
  452. return Common::Input::ButtonNames::Cross;
  453. case PadButton::Square:
  454. return Common::Input::ButtonNames::Square;
  455. case PadButton::Triangle:
  456. return Common::Input::ButtonNames::Triangle;
  457. case PadButton::Share:
  458. return Common::Input::ButtonNames::Share;
  459. case PadButton::Options:
  460. return Common::Input::ButtonNames::Options;
  461. case PadButton::Home:
  462. return Common::Input::ButtonNames::Home;
  463. case PadButton::Touch1:
  464. case PadButton::Touch2:
  465. case PadButton::TouchHardPress:
  466. return Common::Input::ButtonNames::Touch;
  467. default:
  468. return Common::Input::ButtonNames::Undefined;
  469. }
  470. }
  471. Common::Input::ButtonNames UDPClient::GetUIName(const Common::ParamPackage& params) const {
  472. if (params.Has("button")) {
  473. return GetUIButtonName(params);
  474. }
  475. if (params.Has("axis")) {
  476. return Common::Input::ButtonNames::Value;
  477. }
  478. if (params.Has("motion")) {
  479. return Common::Input::ButtonNames::Engine;
  480. }
  481. return Common::Input::ButtonNames::Invalid;
  482. }
  483. bool UDPClient::IsStickInverted(const Common::ParamPackage& params) {
  484. if (!params.Has("guid") || !params.Has("port") || !params.Has("pad")) {
  485. return false;
  486. }
  487. const auto x_axis = static_cast<PadAxes>(params.Get("axis_x", 0));
  488. const auto y_axis = static_cast<PadAxes>(params.Get("axis_y", 0));
  489. if (x_axis != PadAxes::LeftStickY && x_axis != PadAxes::RightStickY) {
  490. return false;
  491. }
  492. if (y_axis != PadAxes::LeftStickX && y_axis != PadAxes::RightStickX) {
  493. return false;
  494. }
  495. return true;
  496. }
  497. void TestCommunication(const std::string& host, u16 port,
  498. const std::function<void()>& success_callback,
  499. const std::function<void()>& failure_callback) {
  500. std::thread([=] {
  501. Common::Event success_event;
  502. SocketCallback callback{
  503. .version = [](Response::Version) {},
  504. .port_info = [](Response::PortInfo) {},
  505. .pad_data = [&](Response::PadData) { success_event.Set(); },
  506. };
  507. Socket socket{host, port, std::move(callback)};
  508. std::thread worker_thread{SocketLoop, &socket};
  509. const bool result =
  510. success_event.WaitUntil(std::chrono::steady_clock::now() + std::chrono::seconds(10));
  511. socket.Stop();
  512. worker_thread.join();
  513. if (result) {
  514. success_callback();
  515. } else {
  516. failure_callback();
  517. }
  518. }).detach();
  519. }
  520. CalibrationConfigurationJob::CalibrationConfigurationJob(
  521. const std::string& host, u16 port, std::function<void(Status)> status_callback,
  522. std::function<void(u16, u16, u16, u16)> data_callback) {
  523. std::thread([=, this] {
  524. u16 min_x{UINT16_MAX};
  525. u16 min_y{UINT16_MAX};
  526. u16 max_x{};
  527. u16 max_y{};
  528. Status current_status{Status::Initialized};
  529. SocketCallback callback{[](Response::Version) {}, [](Response::PortInfo) {},
  530. [&](Response::PadData data) {
  531. constexpr u16 CALIBRATION_THRESHOLD = 100;
  532. if (current_status == Status::Initialized) {
  533. // Receiving data means the communication is ready now
  534. current_status = Status::Ready;
  535. status_callback(current_status);
  536. }
  537. if (data.touch[0].is_active == 0) {
  538. return;
  539. }
  540. LOG_DEBUG(Input, "Current touch: {} {}", data.touch[0].x,
  541. data.touch[0].y);
  542. min_x = std::min(min_x, static_cast<u16>(data.touch[0].x));
  543. min_y = std::min(min_y, static_cast<u16>(data.touch[0].y));
  544. if (current_status == Status::Ready) {
  545. // First touch - min data (min_x/min_y)
  546. current_status = Status::Stage1Completed;
  547. status_callback(current_status);
  548. }
  549. if (data.touch[0].x - min_x > CALIBRATION_THRESHOLD &&
  550. data.touch[0].y - min_y > CALIBRATION_THRESHOLD) {
  551. // Set the current position as max value and finishes
  552. // configuration
  553. max_x = data.touch[0].x;
  554. max_y = data.touch[0].y;
  555. current_status = Status::Completed;
  556. data_callback(min_x, min_y, max_x, max_y);
  557. status_callback(current_status);
  558. complete_event.Set();
  559. }
  560. }};
  561. Socket socket{host, port, std::move(callback)};
  562. std::thread worker_thread{SocketLoop, &socket};
  563. complete_event.Wait();
  564. socket.Stop();
  565. worker_thread.join();
  566. }).detach();
  567. }
  568. CalibrationConfigurationJob::~CalibrationConfigurationJob() {
  569. Stop();
  570. }
  571. void CalibrationConfigurationJob::Stop() {
  572. complete_event.Set();
  573. }
  574. } // namespace InputCommon::CemuhookUDP