npad.cpp 57 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343
  1. // SPDX-FileCopyrightText: Copyright 2018 yuzu Emulator Project
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  3. #include <algorithm>
  4. #include <array>
  5. #include <chrono>
  6. #include <cstring>
  7. #include "common/assert.h"
  8. #include "common/bit_field.h"
  9. #include "common/common_types.h"
  10. #include "common/logging/log.h"
  11. #include "common/settings.h"
  12. #include "core/core_timing.h"
  13. #include "core/hle/kernel/k_event.h"
  14. #include "core/hle/kernel/k_readable_event.h"
  15. #include "core/hle/service/kernel_helpers.h"
  16. #include "hid_core/frontend/emulated_controller.h"
  17. #include "hid_core/hid_core.h"
  18. #include "hid_core/hid_result.h"
  19. #include "hid_core/hid_util.h"
  20. #include "hid_core/resources/applet_resource.h"
  21. #include "hid_core/resources/npad/npad.h"
  22. #include "hid_core/resources/npad/npad_vibration.h"
  23. #include "hid_core/resources/shared_memory_format.h"
  24. namespace Service::HID {
  25. NPad::NPad(Core::HID::HIDCore& hid_core_, KernelHelpers::ServiceContext& service_context_)
  26. : hid_core{hid_core_}, service_context{service_context_}, npad_resource{service_context} {
  27. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; ++aruid_index) {
  28. for (std::size_t i = 0; i < controller_data[aruid_index].size(); ++i) {
  29. auto& controller = controller_data[aruid_index][i];
  30. controller.device = hid_core.GetEmulatedControllerByIndex(i);
  31. Core::HID::ControllerUpdateCallback engine_callback{
  32. .on_change =
  33. [this, i](Core::HID::ControllerTriggerType type) { ControllerUpdate(type, i); },
  34. .is_npad_service = true,
  35. };
  36. controller.callback_key = controller.device->SetCallback(engine_callback);
  37. }
  38. }
  39. for (std::size_t i = 0; i < abstracted_pads.size(); ++i) {
  40. abstracted_pads[i] = AbstractPad{};
  41. abstracted_pads[i].SetNpadId(IndexToNpadIdType(i));
  42. }
  43. }
  44. NPad::~NPad() {
  45. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; ++aruid_index) {
  46. for (std::size_t i = 0; i < controller_data[aruid_index].size(); ++i) {
  47. auto& controller = controller_data[aruid_index][i];
  48. controller.device->DeleteCallback(controller.callback_key);
  49. }
  50. }
  51. }
  52. Result NPad::Activate() {
  53. if (ref_counter == std::numeric_limits<s32>::max() - 1) {
  54. return ResultNpadResourceOverflow;
  55. }
  56. if (ref_counter == 0) {
  57. std::scoped_lock lock{mutex};
  58. // TODO: Activate handlers and AbstractedPad
  59. }
  60. ref_counter++;
  61. return ResultSuccess;
  62. }
  63. Result NPad::Activate(u64 aruid) {
  64. std::scoped_lock lock{mutex};
  65. std::scoped_lock shared_lock{*applet_resource_holder.shared_mutex};
  66. auto* data = applet_resource_holder.applet_resource->GetAruidData(aruid);
  67. const auto aruid_index = applet_resource_holder.applet_resource->GetIndexFromAruid(aruid);
  68. if (data == nullptr || !data->flag.is_assigned) {
  69. return ResultSuccess;
  70. }
  71. for (std::size_t i = 0; i < controller_data[aruid_index].size(); ++i) {
  72. auto& controller = controller_data[aruid_index][i];
  73. controller.shared_memory = &data->shared_memory_format->npad.npad_entry[i].internal_state;
  74. }
  75. // Prefill controller buffers
  76. for (auto& controller : controller_data[aruid_index]) {
  77. auto* npad = controller.shared_memory;
  78. npad->fullkey_color = {
  79. .attribute = ColorAttribute::NoController,
  80. .fullkey = {},
  81. };
  82. npad->joycon_color = {
  83. .attribute = ColorAttribute::NoController,
  84. .left = {},
  85. .right = {},
  86. };
  87. // HW seems to initialize the first 19 entries
  88. for (std::size_t i = 0; i < 19; ++i) {
  89. WriteEmptyEntry(npad);
  90. }
  91. controller.is_active = true;
  92. }
  93. return ResultSuccess;
  94. }
  95. Result NPad::ActivateNpadResource() {
  96. return npad_resource.Activate();
  97. }
  98. Result NPad::ActivateNpadResource(u64 aruid) {
  99. return npad_resource.Activate(aruid);
  100. }
  101. void NPad::ControllerUpdate(Core::HID::ControllerTriggerType type, std::size_t controller_idx) {
  102. if (type == Core::HID::ControllerTriggerType::All) {
  103. ControllerUpdate(Core::HID::ControllerTriggerType::Connected, controller_idx);
  104. ControllerUpdate(Core::HID::ControllerTriggerType::Battery, controller_idx);
  105. return;
  106. }
  107. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; aruid_index++) {
  108. if (controller_idx >= controller_data[aruid_index].size()) {
  109. return;
  110. }
  111. auto* data = applet_resource_holder.applet_resource->GetAruidDataByIndex(aruid_index);
  112. if (!data->flag.is_assigned) {
  113. continue;
  114. }
  115. auto& controller = controller_data[aruid_index][controller_idx];
  116. const auto is_connected = controller.device->IsConnected();
  117. const auto npad_type = controller.device->GetNpadStyleIndex();
  118. const auto npad_id = controller.device->GetNpadIdType();
  119. switch (type) {
  120. case Core::HID::ControllerTriggerType::Connected:
  121. case Core::HID::ControllerTriggerType::Disconnected:
  122. if (is_connected == controller.is_connected) {
  123. return;
  124. }
  125. UpdateControllerAt(data->aruid, npad_type, npad_id, is_connected);
  126. break;
  127. case Core::HID::ControllerTriggerType::Battery: {
  128. if (!controller.device->IsConnected()) {
  129. return;
  130. }
  131. auto* shared_memory = controller.shared_memory;
  132. const auto& battery_level = controller.device->GetBattery();
  133. shared_memory->battery_level_dual = battery_level.dual.battery_level;
  134. shared_memory->battery_level_left = battery_level.left.battery_level;
  135. shared_memory->battery_level_right = battery_level.right.battery_level;
  136. break;
  137. }
  138. default:
  139. break;
  140. }
  141. }
  142. }
  143. void NPad::InitNewlyAddedController(u64 aruid, Core::HID::NpadIdType npad_id) {
  144. auto& controller = GetControllerFromNpadIdType(aruid, npad_id);
  145. if (!npad_resource.IsControllerSupported(aruid, controller.device->GetNpadStyleIndex())) {
  146. return;
  147. }
  148. LOG_DEBUG(Service_HID, "Npad connected {}", npad_id);
  149. const auto controller_type = controller.device->GetNpadStyleIndex();
  150. const auto& body_colors = controller.device->GetColors();
  151. const auto& battery_level = controller.device->GetBattery();
  152. auto* shared_memory = controller.shared_memory;
  153. if (controller_type == Core::HID::NpadStyleIndex::None) {
  154. npad_resource.SignalStyleSetUpdateEvent(aruid, npad_id);
  155. return;
  156. }
  157. // Reset memory values
  158. shared_memory->style_tag.raw = Core::HID::NpadStyleSet::None;
  159. shared_memory->device_type.raw = 0;
  160. shared_memory->system_properties.raw = 0;
  161. shared_memory->joycon_color.attribute = ColorAttribute::NoController;
  162. shared_memory->joycon_color.attribute = ColorAttribute::NoController;
  163. shared_memory->fullkey_color = {};
  164. shared_memory->joycon_color.left = {};
  165. shared_memory->joycon_color.right = {};
  166. shared_memory->battery_level_dual = {};
  167. shared_memory->battery_level_left = {};
  168. shared_memory->battery_level_right = {};
  169. switch (controller_type) {
  170. case Core::HID::NpadStyleIndex::None:
  171. ASSERT(false);
  172. break;
  173. case Core::HID::NpadStyleIndex::Fullkey:
  174. shared_memory->fullkey_color.attribute = ColorAttribute::Ok;
  175. shared_memory->fullkey_color.fullkey = body_colors.fullkey;
  176. shared_memory->battery_level_dual = battery_level.dual.battery_level;
  177. shared_memory->style_tag.fullkey.Assign(1);
  178. shared_memory->device_type.fullkey.Assign(1);
  179. shared_memory->system_properties.is_vertical.Assign(1);
  180. shared_memory->system_properties.use_plus.Assign(1);
  181. shared_memory->system_properties.use_minus.Assign(1);
  182. shared_memory->system_properties.is_charging_joy_dual.Assign(
  183. battery_level.dual.is_charging);
  184. shared_memory->applet_footer_type = AppletFooterUiType::SwitchProController;
  185. shared_memory->sixaxis_fullkey_properties.is_newly_assigned.Assign(1);
  186. break;
  187. case Core::HID::NpadStyleIndex::Handheld:
  188. shared_memory->fullkey_color.attribute = ColorAttribute::Ok;
  189. shared_memory->joycon_color.attribute = ColorAttribute::Ok;
  190. shared_memory->fullkey_color.fullkey = body_colors.fullkey;
  191. shared_memory->joycon_color.left = body_colors.left;
  192. shared_memory->joycon_color.right = body_colors.right;
  193. shared_memory->style_tag.handheld.Assign(1);
  194. shared_memory->device_type.handheld_left.Assign(1);
  195. shared_memory->device_type.handheld_right.Assign(1);
  196. shared_memory->system_properties.is_vertical.Assign(1);
  197. shared_memory->system_properties.use_plus.Assign(1);
  198. shared_memory->system_properties.use_minus.Assign(1);
  199. shared_memory->system_properties.use_directional_buttons.Assign(1);
  200. shared_memory->system_properties.is_charging_joy_dual.Assign(
  201. battery_level.left.is_charging);
  202. shared_memory->system_properties.is_charging_joy_left.Assign(
  203. battery_level.left.is_charging);
  204. shared_memory->system_properties.is_charging_joy_right.Assign(
  205. battery_level.right.is_charging);
  206. shared_memory->assignment_mode = NpadJoyAssignmentMode::Dual;
  207. shared_memory->applet_footer_type = AppletFooterUiType::HandheldJoyConLeftJoyConRight;
  208. shared_memory->sixaxis_handheld_properties.is_newly_assigned.Assign(1);
  209. break;
  210. case Core::HID::NpadStyleIndex::JoyconDual:
  211. shared_memory->fullkey_color.attribute = ColorAttribute::Ok;
  212. shared_memory->joycon_color.attribute = ColorAttribute::Ok;
  213. shared_memory->style_tag.joycon_dual.Assign(1);
  214. if (controller.is_dual_left_connected) {
  215. shared_memory->joycon_color.left = body_colors.left;
  216. shared_memory->battery_level_left = battery_level.left.battery_level;
  217. shared_memory->device_type.joycon_left.Assign(1);
  218. shared_memory->system_properties.use_minus.Assign(1);
  219. shared_memory->system_properties.is_charging_joy_left.Assign(
  220. battery_level.left.is_charging);
  221. shared_memory->sixaxis_dual_left_properties.is_newly_assigned.Assign(1);
  222. }
  223. if (controller.is_dual_right_connected) {
  224. shared_memory->joycon_color.right = body_colors.right;
  225. shared_memory->battery_level_right = battery_level.right.battery_level;
  226. shared_memory->device_type.joycon_right.Assign(1);
  227. shared_memory->system_properties.use_plus.Assign(1);
  228. shared_memory->system_properties.is_charging_joy_right.Assign(
  229. battery_level.right.is_charging);
  230. shared_memory->sixaxis_dual_right_properties.is_newly_assigned.Assign(1);
  231. }
  232. shared_memory->system_properties.use_directional_buttons.Assign(1);
  233. shared_memory->system_properties.is_vertical.Assign(1);
  234. shared_memory->assignment_mode = NpadJoyAssignmentMode::Dual;
  235. if (controller.is_dual_left_connected && controller.is_dual_right_connected) {
  236. shared_memory->applet_footer_type = AppletFooterUiType::JoyDual;
  237. shared_memory->fullkey_color.fullkey = body_colors.left;
  238. shared_memory->battery_level_dual = battery_level.left.battery_level;
  239. shared_memory->system_properties.is_charging_joy_dual.Assign(
  240. battery_level.left.is_charging);
  241. } else if (controller.is_dual_left_connected) {
  242. shared_memory->applet_footer_type = AppletFooterUiType::JoyDualLeftOnly;
  243. shared_memory->fullkey_color.fullkey = body_colors.left;
  244. shared_memory->battery_level_dual = battery_level.left.battery_level;
  245. shared_memory->system_properties.is_charging_joy_dual.Assign(
  246. battery_level.left.is_charging);
  247. } else {
  248. shared_memory->applet_footer_type = AppletFooterUiType::JoyDualRightOnly;
  249. shared_memory->fullkey_color.fullkey = body_colors.right;
  250. shared_memory->battery_level_dual = battery_level.right.battery_level;
  251. shared_memory->system_properties.is_charging_joy_dual.Assign(
  252. battery_level.right.is_charging);
  253. }
  254. break;
  255. case Core::HID::NpadStyleIndex::JoyconLeft:
  256. shared_memory->fullkey_color.attribute = ColorAttribute::Ok;
  257. shared_memory->fullkey_color.fullkey = body_colors.left;
  258. shared_memory->joycon_color.attribute = ColorAttribute::Ok;
  259. shared_memory->joycon_color.left = body_colors.left;
  260. shared_memory->battery_level_dual = battery_level.left.battery_level;
  261. shared_memory->style_tag.joycon_left.Assign(1);
  262. shared_memory->device_type.joycon_left.Assign(1);
  263. shared_memory->system_properties.is_horizontal.Assign(1);
  264. shared_memory->system_properties.use_minus.Assign(1);
  265. shared_memory->system_properties.is_charging_joy_left.Assign(
  266. battery_level.left.is_charging);
  267. shared_memory->applet_footer_type = AppletFooterUiType::JoyLeftHorizontal;
  268. shared_memory->sixaxis_left_properties.is_newly_assigned.Assign(1);
  269. break;
  270. case Core::HID::NpadStyleIndex::JoyconRight:
  271. shared_memory->fullkey_color.attribute = ColorAttribute::Ok;
  272. shared_memory->fullkey_color.fullkey = body_colors.right;
  273. shared_memory->joycon_color.attribute = ColorAttribute::Ok;
  274. shared_memory->joycon_color.right = body_colors.right;
  275. shared_memory->battery_level_right = battery_level.right.battery_level;
  276. shared_memory->style_tag.joycon_right.Assign(1);
  277. shared_memory->device_type.joycon_right.Assign(1);
  278. shared_memory->system_properties.is_horizontal.Assign(1);
  279. shared_memory->system_properties.use_plus.Assign(1);
  280. shared_memory->system_properties.is_charging_joy_right.Assign(
  281. battery_level.right.is_charging);
  282. shared_memory->applet_footer_type = AppletFooterUiType::JoyRightHorizontal;
  283. shared_memory->sixaxis_right_properties.is_newly_assigned.Assign(1);
  284. break;
  285. case Core::HID::NpadStyleIndex::GameCube:
  286. shared_memory->style_tag.gamecube.Assign(1);
  287. shared_memory->device_type.fullkey.Assign(1);
  288. shared_memory->system_properties.is_vertical.Assign(1);
  289. shared_memory->system_properties.use_plus.Assign(1);
  290. break;
  291. case Core::HID::NpadStyleIndex::Pokeball:
  292. shared_memory->style_tag.palma.Assign(1);
  293. shared_memory->device_type.palma.Assign(1);
  294. shared_memory->sixaxis_fullkey_properties.is_newly_assigned.Assign(1);
  295. break;
  296. case Core::HID::NpadStyleIndex::NES:
  297. shared_memory->style_tag.lark.Assign(1);
  298. shared_memory->device_type.fullkey.Assign(1);
  299. break;
  300. case Core::HID::NpadStyleIndex::SNES:
  301. shared_memory->style_tag.lucia.Assign(1);
  302. shared_memory->device_type.fullkey.Assign(1);
  303. shared_memory->applet_footer_type = AppletFooterUiType::Lucia;
  304. break;
  305. case Core::HID::NpadStyleIndex::N64:
  306. shared_memory->style_tag.lagoon.Assign(1);
  307. shared_memory->device_type.fullkey.Assign(1);
  308. shared_memory->applet_footer_type = AppletFooterUiType::Lagon;
  309. break;
  310. case Core::HID::NpadStyleIndex::SegaGenesis:
  311. shared_memory->style_tag.lager.Assign(1);
  312. shared_memory->device_type.fullkey.Assign(1);
  313. break;
  314. default:
  315. break;
  316. }
  317. controller.is_connected = true;
  318. controller.device->Connect();
  319. controller.device->SetLedPattern();
  320. if (controller_type == Core::HID::NpadStyleIndex::JoyconDual) {
  321. if (controller.is_dual_left_connected) {
  322. controller.device->SetPollingMode(Core::HID::EmulatedDeviceIndex::LeftIndex,
  323. Common::Input::PollingMode::Active);
  324. }
  325. if (controller.is_dual_right_connected) {
  326. controller.device->SetPollingMode(Core::HID::EmulatedDeviceIndex::RightIndex,
  327. Common::Input::PollingMode::Active);
  328. }
  329. } else {
  330. controller.device->SetPollingMode(Core::HID::EmulatedDeviceIndex::AllDevices,
  331. Common::Input::PollingMode::Active);
  332. }
  333. npad_resource.SignalStyleSetUpdateEvent(aruid, npad_id);
  334. WriteEmptyEntry(controller.shared_memory);
  335. hid_core.SetLastActiveController(npad_id);
  336. abstracted_pads[NpadIdTypeToIndex(npad_id)].Update();
  337. }
  338. void NPad::WriteEmptyEntry(NpadInternalState* npad) {
  339. NPadGenericState dummy_pad_state{};
  340. NpadGcTriggerState dummy_gc_state{};
  341. dummy_pad_state.sampling_number = npad->fullkey_lifo.ReadCurrentEntry().sampling_number + 1;
  342. npad->fullkey_lifo.WriteNextEntry(dummy_pad_state);
  343. dummy_pad_state.sampling_number = npad->handheld_lifo.ReadCurrentEntry().sampling_number + 1;
  344. npad->handheld_lifo.WriteNextEntry(dummy_pad_state);
  345. dummy_pad_state.sampling_number = npad->joy_dual_lifo.ReadCurrentEntry().sampling_number + 1;
  346. npad->joy_dual_lifo.WriteNextEntry(dummy_pad_state);
  347. dummy_pad_state.sampling_number = npad->joy_left_lifo.ReadCurrentEntry().sampling_number + 1;
  348. npad->joy_left_lifo.WriteNextEntry(dummy_pad_state);
  349. dummy_pad_state.sampling_number = npad->joy_right_lifo.ReadCurrentEntry().sampling_number + 1;
  350. npad->joy_right_lifo.WriteNextEntry(dummy_pad_state);
  351. dummy_pad_state.sampling_number = npad->palma_lifo.ReadCurrentEntry().sampling_number + 1;
  352. npad->palma_lifo.WriteNextEntry(dummy_pad_state);
  353. dummy_pad_state.sampling_number = npad->system_ext_lifo.ReadCurrentEntry().sampling_number + 1;
  354. npad->system_ext_lifo.WriteNextEntry(dummy_pad_state);
  355. dummy_gc_state.sampling_number = npad->gc_trigger_lifo.ReadCurrentEntry().sampling_number + 1;
  356. npad->gc_trigger_lifo.WriteNextEntry(dummy_gc_state);
  357. }
  358. void NPad::RequestPadStateUpdate(u64 aruid, Core::HID::NpadIdType npad_id) {
  359. std::scoped_lock lock{*applet_resource_holder.shared_mutex};
  360. auto& controller = GetControllerFromNpadIdType(aruid, npad_id);
  361. const auto controller_type = controller.device->GetNpadStyleIndex();
  362. if (!controller.device->IsConnected() && controller.is_connected) {
  363. DisconnectNpad(aruid, npad_id);
  364. return;
  365. }
  366. if (!controller.device->IsConnected()) {
  367. return;
  368. }
  369. if (controller.device->IsConnected() && !controller.is_connected) {
  370. InitNewlyAddedController(aruid, npad_id);
  371. }
  372. // This function is unique to yuzu for the turbo buttons and motion to work properly
  373. controller.device->StatusUpdate();
  374. auto& pad_entry = controller.npad_pad_state;
  375. auto& trigger_entry = controller.npad_trigger_state;
  376. const auto button_state = controller.device->GetNpadButtons();
  377. const auto stick_state = controller.device->GetSticks();
  378. using btn = Core::HID::NpadButton;
  379. pad_entry.npad_buttons.raw = btn::None;
  380. if (controller_type != Core::HID::NpadStyleIndex::JoyconLeft) {
  381. constexpr btn right_button_mask = btn::A | btn::B | btn::X | btn::Y | btn::StickR | btn::R |
  382. btn::ZR | btn::Plus | btn::StickRLeft | btn::StickRUp |
  383. btn::StickRRight | btn::StickRDown;
  384. pad_entry.npad_buttons.raw = button_state.raw & right_button_mask;
  385. pad_entry.r_stick = stick_state.right;
  386. }
  387. if (controller_type != Core::HID::NpadStyleIndex::JoyconRight) {
  388. constexpr btn left_button_mask =
  389. btn::Left | btn::Up | btn::Right | btn::Down | btn::StickL | btn::L | btn::ZL |
  390. btn::Minus | btn::StickLLeft | btn::StickLUp | btn::StickLRight | btn::StickLDown;
  391. pad_entry.npad_buttons.raw |= button_state.raw & left_button_mask;
  392. pad_entry.l_stick = stick_state.left;
  393. }
  394. if (controller_type == Core::HID::NpadStyleIndex::JoyconLeft ||
  395. controller_type == Core::HID::NpadStyleIndex::JoyconDual) {
  396. pad_entry.npad_buttons.left_sl.Assign(button_state.left_sl);
  397. pad_entry.npad_buttons.left_sr.Assign(button_state.left_sr);
  398. }
  399. if (controller_type == Core::HID::NpadStyleIndex::JoyconRight ||
  400. controller_type == Core::HID::NpadStyleIndex::JoyconDual) {
  401. pad_entry.npad_buttons.right_sl.Assign(button_state.right_sl);
  402. pad_entry.npad_buttons.right_sr.Assign(button_state.right_sr);
  403. }
  404. if (controller_type == Core::HID::NpadStyleIndex::GameCube) {
  405. const auto& trigger_state = controller.device->GetTriggers();
  406. trigger_entry.l_analog = trigger_state.left;
  407. trigger_entry.r_analog = trigger_state.right;
  408. pad_entry.npad_buttons.zl.Assign(false);
  409. pad_entry.npad_buttons.zr.Assign(button_state.r);
  410. pad_entry.npad_buttons.l.Assign(button_state.zl);
  411. pad_entry.npad_buttons.r.Assign(button_state.zr);
  412. }
  413. if (pad_entry.npad_buttons.raw != Core::HID::NpadButton::None) {
  414. hid_core.SetLastActiveController(npad_id);
  415. }
  416. }
  417. void NPad::OnUpdate(const Core::Timing::CoreTiming& core_timing) {
  418. if (ref_counter == 0) {
  419. return;
  420. }
  421. std::scoped_lock lock{*applet_resource_holder.shared_mutex};
  422. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; ++aruid_index) {
  423. const auto* data = applet_resource_holder.applet_resource->GetAruidDataByIndex(aruid_index);
  424. const auto aruid = data->aruid;
  425. if (!data->flag.is_assigned) {
  426. continue;
  427. }
  428. bool is_set{};
  429. npad_resource.IsSupportedNpadStyleSet(is_set, aruid);
  430. // Wait until style is defined
  431. if (!is_set) {
  432. continue;
  433. }
  434. for (std::size_t i = 0; i < controller_data[aruid_index].size(); ++i) {
  435. auto& controller = controller_data[aruid_index][i];
  436. controller.shared_memory =
  437. &data->shared_memory_format->npad.npad_entry[i].internal_state;
  438. auto* npad = controller.shared_memory;
  439. const auto& controller_type = controller.device->GetNpadStyleIndex();
  440. if (controller_type == Core::HID::NpadStyleIndex::None ||
  441. !controller.device->IsConnected()) {
  442. continue;
  443. }
  444. if (!data->flag.enable_pad_input) {
  445. continue;
  446. }
  447. if (!controller.is_active) {
  448. continue;
  449. }
  450. RequestPadStateUpdate(aruid, controller.device->GetNpadIdType());
  451. auto& pad_state = controller.npad_pad_state;
  452. auto& libnx_state = controller.npad_libnx_state;
  453. auto& trigger_state = controller.npad_trigger_state;
  454. // LibNX exclusively uses this section, so we always update it since LibNX doesn't
  455. // activate any controllers.
  456. libnx_state.connection_status.raw = 0;
  457. libnx_state.connection_status.is_connected.Assign(1);
  458. switch (controller_type) {
  459. case Core::HID::NpadStyleIndex::None:
  460. ASSERT(false);
  461. break;
  462. case Core::HID::NpadStyleIndex::Fullkey:
  463. case Core::HID::NpadStyleIndex::NES:
  464. case Core::HID::NpadStyleIndex::SNES:
  465. case Core::HID::NpadStyleIndex::N64:
  466. case Core::HID::NpadStyleIndex::SegaGenesis:
  467. pad_state.connection_status.raw = 0;
  468. pad_state.connection_status.is_connected.Assign(1);
  469. pad_state.connection_status.is_wired.Assign(1);
  470. libnx_state.connection_status.is_wired.Assign(1);
  471. pad_state.sampling_number =
  472. npad->fullkey_lifo.ReadCurrentEntry().state.sampling_number + 1;
  473. npad->fullkey_lifo.WriteNextEntry(pad_state);
  474. break;
  475. case Core::HID::NpadStyleIndex::Handheld:
  476. pad_state.connection_status.raw = 0;
  477. pad_state.connection_status.is_connected.Assign(1);
  478. pad_state.connection_status.is_wired.Assign(1);
  479. pad_state.connection_status.is_left_connected.Assign(1);
  480. pad_state.connection_status.is_right_connected.Assign(1);
  481. pad_state.connection_status.is_left_wired.Assign(1);
  482. pad_state.connection_status.is_right_wired.Assign(1);
  483. libnx_state.connection_status.is_wired.Assign(1);
  484. libnx_state.connection_status.is_left_connected.Assign(1);
  485. libnx_state.connection_status.is_right_connected.Assign(1);
  486. libnx_state.connection_status.is_left_wired.Assign(1);
  487. libnx_state.connection_status.is_right_wired.Assign(1);
  488. pad_state.sampling_number =
  489. npad->handheld_lifo.ReadCurrentEntry().state.sampling_number + 1;
  490. npad->handheld_lifo.WriteNextEntry(pad_state);
  491. break;
  492. case Core::HID::NpadStyleIndex::JoyconDual:
  493. pad_state.connection_status.raw = 0;
  494. pad_state.connection_status.is_connected.Assign(1);
  495. if (controller.is_dual_left_connected) {
  496. pad_state.connection_status.is_left_connected.Assign(1);
  497. libnx_state.connection_status.is_left_connected.Assign(1);
  498. }
  499. if (controller.is_dual_right_connected) {
  500. pad_state.connection_status.is_right_connected.Assign(1);
  501. libnx_state.connection_status.is_right_connected.Assign(1);
  502. }
  503. pad_state.sampling_number =
  504. npad->joy_dual_lifo.ReadCurrentEntry().state.sampling_number + 1;
  505. npad->joy_dual_lifo.WriteNextEntry(pad_state);
  506. break;
  507. case Core::HID::NpadStyleIndex::JoyconLeft:
  508. pad_state.connection_status.raw = 0;
  509. pad_state.connection_status.is_connected.Assign(1);
  510. pad_state.connection_status.is_left_connected.Assign(1);
  511. libnx_state.connection_status.is_left_connected.Assign(1);
  512. pad_state.sampling_number =
  513. npad->joy_left_lifo.ReadCurrentEntry().state.sampling_number + 1;
  514. npad->joy_left_lifo.WriteNextEntry(pad_state);
  515. break;
  516. case Core::HID::NpadStyleIndex::JoyconRight:
  517. pad_state.connection_status.raw = 0;
  518. pad_state.connection_status.is_connected.Assign(1);
  519. pad_state.connection_status.is_right_connected.Assign(1);
  520. libnx_state.connection_status.is_right_connected.Assign(1);
  521. pad_state.sampling_number =
  522. npad->joy_right_lifo.ReadCurrentEntry().state.sampling_number + 1;
  523. npad->joy_right_lifo.WriteNextEntry(pad_state);
  524. break;
  525. case Core::HID::NpadStyleIndex::GameCube:
  526. pad_state.connection_status.raw = 0;
  527. pad_state.connection_status.is_connected.Assign(1);
  528. pad_state.connection_status.is_wired.Assign(1);
  529. libnx_state.connection_status.is_wired.Assign(1);
  530. pad_state.sampling_number =
  531. npad->fullkey_lifo.ReadCurrentEntry().state.sampling_number + 1;
  532. trigger_state.sampling_number =
  533. npad->gc_trigger_lifo.ReadCurrentEntry().state.sampling_number + 1;
  534. npad->fullkey_lifo.WriteNextEntry(pad_state);
  535. npad->gc_trigger_lifo.WriteNextEntry(trigger_state);
  536. break;
  537. case Core::HID::NpadStyleIndex::Pokeball:
  538. pad_state.connection_status.raw = 0;
  539. pad_state.connection_status.is_connected.Assign(1);
  540. pad_state.sampling_number =
  541. npad->palma_lifo.ReadCurrentEntry().state.sampling_number + 1;
  542. npad->palma_lifo.WriteNextEntry(pad_state);
  543. break;
  544. default:
  545. break;
  546. }
  547. libnx_state.npad_buttons.raw = pad_state.npad_buttons.raw;
  548. libnx_state.l_stick = pad_state.l_stick;
  549. libnx_state.r_stick = pad_state.r_stick;
  550. libnx_state.sampling_number =
  551. npad->system_ext_lifo.ReadCurrentEntry().state.sampling_number + 1;
  552. npad->system_ext_lifo.WriteNextEntry(libnx_state);
  553. press_state |= static_cast<u64>(pad_state.npad_buttons.raw);
  554. }
  555. }
  556. }
  557. Result NPad::SetSupportedNpadStyleSet(u64 aruid, Core::HID::NpadStyleSet supported_style_set) {
  558. std::scoped_lock lock{mutex};
  559. hid_core.SetSupportedStyleTag({supported_style_set});
  560. const Result result = npad_resource.SetSupportedNpadStyleSet(aruid, supported_style_set);
  561. if (result.IsSuccess()) {
  562. OnUpdate({});
  563. }
  564. return result;
  565. }
  566. Result NPad::GetSupportedNpadStyleSet(u64 aruid,
  567. Core::HID::NpadStyleSet& out_supported_style_set) const {
  568. std::scoped_lock lock{mutex};
  569. const Result result = npad_resource.GetSupportedNpadStyleSet(out_supported_style_set, aruid);
  570. if (result == ResultUndefinedStyleset) {
  571. out_supported_style_set = Core::HID::NpadStyleSet::None;
  572. return ResultSuccess;
  573. }
  574. return result;
  575. }
  576. Result NPad::GetMaskedSupportedNpadStyleSet(
  577. u64 aruid, Core::HID::NpadStyleSet& out_supported_style_set) const {
  578. std::scoped_lock lock{mutex};
  579. const Result result =
  580. npad_resource.GetMaskedSupportedNpadStyleSet(out_supported_style_set, aruid);
  581. if (result == ResultUndefinedStyleset) {
  582. out_supported_style_set = Core::HID::NpadStyleSet::None;
  583. return ResultSuccess;
  584. }
  585. return result;
  586. }
  587. Result NPad::SetSupportedNpadIdType(u64 aruid,
  588. std::span<const Core::HID::NpadIdType> supported_npad_list) {
  589. std::scoped_lock lock{mutex};
  590. if (supported_npad_list.size() > MaxSupportedNpadIdTypes) {
  591. return ResultInvalidArraySize;
  592. }
  593. Result result = npad_resource.SetSupportedNpadIdType(aruid, supported_npad_list);
  594. if (result.IsSuccess()) {
  595. OnUpdate({});
  596. }
  597. return result;
  598. }
  599. Result NPad::SetNpadJoyHoldType(u64 aruid, NpadJoyHoldType hold_type) {
  600. std::scoped_lock lock{mutex};
  601. return npad_resource.SetNpadJoyHoldType(aruid, hold_type);
  602. }
  603. Result NPad::GetNpadJoyHoldType(u64 aruid, NpadJoyHoldType& out_hold_type) const {
  604. std::scoped_lock lock{mutex};
  605. return npad_resource.GetNpadJoyHoldType(out_hold_type, aruid);
  606. }
  607. Result NPad::SetNpadHandheldActivationMode(u64 aruid, NpadHandheldActivationMode mode) {
  608. std::scoped_lock lock{mutex};
  609. Result result = npad_resource.SetNpadHandheldActivationMode(aruid, mode);
  610. if (result.IsSuccess()) {
  611. OnUpdate({});
  612. }
  613. return result;
  614. }
  615. Result NPad::GetNpadHandheldActivationMode(u64 aruid, NpadHandheldActivationMode& out_mode) const {
  616. std::scoped_lock lock{mutex};
  617. return npad_resource.GetNpadHandheldActivationMode(out_mode, aruid);
  618. }
  619. bool NPad::SetNpadMode(u64 aruid, Core::HID::NpadIdType& new_npad_id, Core::HID::NpadIdType npad_id,
  620. NpadJoyDeviceType npad_device_type, NpadJoyAssignmentMode assignment_mode) {
  621. if (!IsNpadIdValid(npad_id)) {
  622. LOG_ERROR(Service_HID, "Invalid NpadIdType npad_id:{}", npad_id);
  623. return false;
  624. }
  625. auto& controller = GetControllerFromNpadIdType(aruid, npad_id);
  626. if (controller.shared_memory->assignment_mode != assignment_mode) {
  627. controller.shared_memory->assignment_mode = assignment_mode;
  628. }
  629. if (!controller.device->IsConnected()) {
  630. return false;
  631. }
  632. if (assignment_mode == NpadJoyAssignmentMode::Dual) {
  633. if (controller.device->GetNpadStyleIndex() == Core::HID::NpadStyleIndex::JoyconLeft) {
  634. DisconnectNpad(aruid, npad_id);
  635. controller.is_dual_left_connected = true;
  636. controller.is_dual_right_connected = false;
  637. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconDual, npad_id, true);
  638. return false;
  639. }
  640. if (controller.device->GetNpadStyleIndex() == Core::HID::NpadStyleIndex::JoyconRight) {
  641. DisconnectNpad(aruid, npad_id);
  642. controller.is_dual_left_connected = false;
  643. controller.is_dual_right_connected = true;
  644. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconDual, npad_id, true);
  645. return false;
  646. }
  647. return false;
  648. }
  649. // This is for NpadJoyAssignmentMode::Single
  650. // Only JoyconDual get affected by this function
  651. if (controller.device->GetNpadStyleIndex() != Core::HID::NpadStyleIndex::JoyconDual) {
  652. return false;
  653. }
  654. if (controller.is_dual_left_connected && !controller.is_dual_right_connected) {
  655. DisconnectNpad(aruid, npad_id);
  656. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconLeft, npad_id, true);
  657. return false;
  658. }
  659. if (!controller.is_dual_left_connected && controller.is_dual_right_connected) {
  660. DisconnectNpad(aruid, npad_id);
  661. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconRight, npad_id, true);
  662. return false;
  663. }
  664. // We have two controllers connected to the same npad_id we need to split them
  665. new_npad_id = hid_core.GetFirstDisconnectedNpadId();
  666. auto& controller_2 = GetControllerFromNpadIdType(aruid, new_npad_id);
  667. DisconnectNpad(aruid, npad_id);
  668. if (npad_device_type == NpadJoyDeviceType::Left) {
  669. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconLeft, npad_id, true);
  670. controller_2.is_dual_left_connected = false;
  671. controller_2.is_dual_right_connected = true;
  672. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconDual, new_npad_id, true);
  673. } else {
  674. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconRight, npad_id, true);
  675. controller_2.is_dual_left_connected = true;
  676. controller_2.is_dual_right_connected = false;
  677. UpdateControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconDual, new_npad_id, true);
  678. }
  679. return true;
  680. }
  681. Result NPad::AcquireNpadStyleSetUpdateEventHandle(u64 aruid, Kernel::KReadableEvent** out_event,
  682. Core::HID::NpadIdType npad_id) {
  683. std::scoped_lock lock{mutex};
  684. return npad_resource.AcquireNpadStyleSetUpdateEventHandle(aruid, out_event, npad_id);
  685. }
  686. void NPad::AddNewControllerAt(u64 aruid, Core::HID::NpadStyleIndex controller,
  687. Core::HID::NpadIdType npad_id) {
  688. UpdateControllerAt(aruid, controller, npad_id, true);
  689. }
  690. void NPad::UpdateControllerAt(u64 aruid, Core::HID::NpadStyleIndex type,
  691. Core::HID::NpadIdType npad_id, bool connected) {
  692. auto& controller = GetControllerFromNpadIdType(aruid, npad_id);
  693. if (!connected) {
  694. DisconnectNpad(aruid, npad_id);
  695. return;
  696. }
  697. controller.device->SetNpadStyleIndex(type);
  698. InitNewlyAddedController(aruid, npad_id);
  699. }
  700. Result NPad::DisconnectNpad(u64 aruid, Core::HID::NpadIdType npad_id) {
  701. if (!IsNpadIdValid(npad_id)) {
  702. LOG_ERROR(Service_HID, "Invalid NpadIdType npad_id:{}", npad_id);
  703. return ResultInvalidNpadId;
  704. }
  705. LOG_DEBUG(Service_HID, "Npad disconnected {}", npad_id);
  706. auto& controller = GetControllerFromNpadIdType(aruid, npad_id);
  707. auto* shared_memory = controller.shared_memory;
  708. // Don't reset shared_memory->assignment_mode this value is persistent
  709. shared_memory->style_tag.raw = Core::HID::NpadStyleSet::None; // Zero out
  710. shared_memory->device_type.raw = 0;
  711. shared_memory->system_properties.raw = 0;
  712. shared_memory->button_properties.raw = 0;
  713. shared_memory->sixaxis_fullkey_properties.raw = 0;
  714. shared_memory->sixaxis_handheld_properties.raw = 0;
  715. shared_memory->sixaxis_dual_left_properties.raw = 0;
  716. shared_memory->sixaxis_dual_right_properties.raw = 0;
  717. shared_memory->sixaxis_left_properties.raw = 0;
  718. shared_memory->sixaxis_right_properties.raw = 0;
  719. shared_memory->battery_level_dual = Core::HID::NpadBatteryLevel::Empty;
  720. shared_memory->battery_level_left = Core::HID::NpadBatteryLevel::Empty;
  721. shared_memory->battery_level_right = Core::HID::NpadBatteryLevel::Empty;
  722. shared_memory->fullkey_color = {
  723. .attribute = ColorAttribute::NoController,
  724. .fullkey = {},
  725. };
  726. shared_memory->joycon_color = {
  727. .attribute = ColorAttribute::NoController,
  728. .left = {},
  729. .right = {},
  730. };
  731. shared_memory->applet_footer_type = AppletFooterUiType::None;
  732. controller.is_dual_left_connected = true;
  733. controller.is_dual_right_connected = true;
  734. controller.is_connected = false;
  735. controller.device->Disconnect();
  736. npad_resource.SignalStyleSetUpdateEvent(aruid, npad_id);
  737. WriteEmptyEntry(shared_memory);
  738. return ResultSuccess;
  739. }
  740. Result NPad::IsFirmwareUpdateAvailableForSixAxisSensor(
  741. u64 aruid, const Core::HID::SixAxisSensorHandle& sixaxis_handle,
  742. bool& is_firmware_available) const {
  743. const auto is_valid = IsSixaxisHandleValid(sixaxis_handle);
  744. if (is_valid.IsError()) {
  745. LOG_ERROR(Service_HID, "Invalid handle, error_code={}", is_valid.raw);
  746. return is_valid;
  747. }
  748. const auto& sixaxis_properties = GetSixaxisProperties(aruid, sixaxis_handle);
  749. is_firmware_available = sixaxis_properties.is_firmware_update_available != 0;
  750. return ResultSuccess;
  751. }
  752. Result NPad::ResetIsSixAxisSensorDeviceNewlyAssigned(
  753. u64 aruid, const Core::HID::SixAxisSensorHandle& sixaxis_handle) {
  754. const auto is_valid = IsSixaxisHandleValid(sixaxis_handle);
  755. if (is_valid.IsError()) {
  756. LOG_ERROR(Service_HID, "Invalid handle, error_code={}", is_valid.raw);
  757. return is_valid;
  758. }
  759. auto& sixaxis_properties = GetSixaxisProperties(aruid, sixaxis_handle);
  760. sixaxis_properties.is_newly_assigned.Assign(0);
  761. return ResultSuccess;
  762. }
  763. Result NPad::MergeSingleJoyAsDualJoy(u64 aruid, Core::HID::NpadIdType npad_id_1,
  764. Core::HID::NpadIdType npad_id_2) {
  765. if (!IsNpadIdValid(npad_id_1) || !IsNpadIdValid(npad_id_2)) {
  766. LOG_ERROR(Service_HID, "Invalid NpadIdType npad_id_1:{}, npad_id_2:{}", npad_id_1,
  767. npad_id_2);
  768. return ResultInvalidNpadId;
  769. }
  770. auto& controller_1 = GetControllerFromNpadIdType(aruid, npad_id_1);
  771. auto& controller_2 = GetControllerFromNpadIdType(aruid, npad_id_2);
  772. auto controller_style_1 = controller_1.device->GetNpadStyleIndex();
  773. auto controller_style_2 = controller_2.device->GetNpadStyleIndex();
  774. // Simplify this code by converting dualjoycon with only a side connected to single joycons
  775. if (controller_style_1 == Core::HID::NpadStyleIndex::JoyconDual) {
  776. if (controller_1.is_dual_left_connected && !controller_1.is_dual_right_connected) {
  777. controller_style_1 = Core::HID::NpadStyleIndex::JoyconLeft;
  778. }
  779. if (!controller_1.is_dual_left_connected && controller_1.is_dual_right_connected) {
  780. controller_style_1 = Core::HID::NpadStyleIndex::JoyconRight;
  781. }
  782. }
  783. if (controller_style_2 == Core::HID::NpadStyleIndex::JoyconDual) {
  784. if (controller_2.is_dual_left_connected && !controller_2.is_dual_right_connected) {
  785. controller_style_2 = Core::HID::NpadStyleIndex::JoyconLeft;
  786. }
  787. if (!controller_2.is_dual_left_connected && controller_2.is_dual_right_connected) {
  788. controller_style_2 = Core::HID::NpadStyleIndex::JoyconRight;
  789. }
  790. }
  791. // Invalid merge errors
  792. if (controller_style_1 == Core::HID::NpadStyleIndex::JoyconDual ||
  793. controller_style_2 == Core::HID::NpadStyleIndex::JoyconDual) {
  794. return NpadIsDualJoycon;
  795. }
  796. if (controller_style_1 == Core::HID::NpadStyleIndex::JoyconLeft &&
  797. controller_style_2 == Core::HID::NpadStyleIndex::JoyconLeft) {
  798. return NpadIsSameType;
  799. }
  800. if (controller_style_1 == Core::HID::NpadStyleIndex::JoyconRight &&
  801. controller_style_2 == Core::HID::NpadStyleIndex::JoyconRight) {
  802. return NpadIsSameType;
  803. }
  804. // These exceptions are handled as if they where dual joycon
  805. if (controller_style_1 != Core::HID::NpadStyleIndex::JoyconLeft &&
  806. controller_style_1 != Core::HID::NpadStyleIndex::JoyconRight) {
  807. return NpadIsDualJoycon;
  808. }
  809. if (controller_style_2 != Core::HID::NpadStyleIndex::JoyconLeft &&
  810. controller_style_2 != Core::HID::NpadStyleIndex::JoyconRight) {
  811. return NpadIsDualJoycon;
  812. }
  813. // Disconnect the joycons and connect them as dual joycon at the first index.
  814. DisconnectNpad(aruid, npad_id_1);
  815. DisconnectNpad(aruid, npad_id_2);
  816. controller_1.is_dual_left_connected = true;
  817. controller_1.is_dual_right_connected = true;
  818. AddNewControllerAt(aruid, Core::HID::NpadStyleIndex::JoyconDual, npad_id_1);
  819. return ResultSuccess;
  820. }
  821. Result NPad::StartLrAssignmentMode(u64 aruid) {
  822. std::scoped_lock lock{mutex};
  823. bool is_enabled{};
  824. Result result = npad_resource.GetLrAssignmentMode(is_enabled, aruid);
  825. if (result.IsSuccess() && is_enabled == false) {
  826. result = npad_resource.SetLrAssignmentMode(aruid, true);
  827. }
  828. return result;
  829. }
  830. Result NPad::StopLrAssignmentMode(u64 aruid) {
  831. std::scoped_lock lock{mutex};
  832. bool is_enabled{};
  833. Result result = npad_resource.GetLrAssignmentMode(is_enabled, aruid);
  834. if (result.IsSuccess() && is_enabled == true) {
  835. result = npad_resource.SetLrAssignmentMode(aruid, false);
  836. }
  837. return result;
  838. }
  839. Result NPad::SwapNpadAssignment(u64 aruid, Core::HID::NpadIdType npad_id_1,
  840. Core::HID::NpadIdType npad_id_2) {
  841. if (!IsNpadIdValid(npad_id_1) || !IsNpadIdValid(npad_id_2)) {
  842. LOG_ERROR(Service_HID, "Invalid NpadIdType npad_id_1:{}, npad_id_2:{}", npad_id_1,
  843. npad_id_2);
  844. return ResultInvalidNpadId;
  845. }
  846. if (npad_id_1 == Core::HID::NpadIdType::Handheld ||
  847. npad_id_2 == Core::HID::NpadIdType::Handheld || npad_id_1 == Core::HID::NpadIdType::Other ||
  848. npad_id_2 == Core::HID::NpadIdType::Other) {
  849. return ResultSuccess;
  850. }
  851. const auto& controller_1 = GetControllerFromNpadIdType(aruid, npad_id_1).device;
  852. const auto& controller_2 = GetControllerFromNpadIdType(aruid, npad_id_2).device;
  853. const auto type_index_1 = controller_1->GetNpadStyleIndex();
  854. const auto type_index_2 = controller_2->GetNpadStyleIndex();
  855. const auto is_connected_1 = controller_1->IsConnected();
  856. const auto is_connected_2 = controller_2->IsConnected();
  857. if (!npad_resource.IsControllerSupported(aruid, type_index_1) && is_connected_1) {
  858. return ResultNpadNotConnected;
  859. }
  860. if (!npad_resource.IsControllerSupported(aruid, type_index_2) && is_connected_2) {
  861. return ResultNpadNotConnected;
  862. }
  863. UpdateControllerAt(aruid, type_index_2, npad_id_1, is_connected_2);
  864. UpdateControllerAt(aruid, type_index_1, npad_id_2, is_connected_1);
  865. return ResultSuccess;
  866. }
  867. Result NPad::IsUnintendedHomeButtonInputProtectionEnabled(bool& out_is_enabled, u64 aruid,
  868. Core::HID::NpadIdType npad_id) const {
  869. std::scoped_lock lock{mutex};
  870. return npad_resource.GetHomeProtectionEnabled(out_is_enabled, aruid, npad_id);
  871. }
  872. Result NPad::EnableUnintendedHomeButtonInputProtection(u64 aruid, Core::HID::NpadIdType npad_id,
  873. bool is_enabled) {
  874. std::scoped_lock lock{mutex};
  875. return npad_resource.SetHomeProtectionEnabled(aruid, npad_id, is_enabled);
  876. }
  877. void NPad::SetNpadAnalogStickUseCenterClamp(u64 aruid, bool is_enabled) {
  878. std::scoped_lock lock{mutex};
  879. npad_resource.SetNpadAnalogStickUseCenterClamp(aruid, is_enabled);
  880. }
  881. void NPad::ClearAllConnectedControllers() {
  882. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; aruid_index++) {
  883. for (auto& controller : controller_data[aruid_index]) {
  884. if (controller.device->IsConnected() &&
  885. controller.device->GetNpadStyleIndex() != Core::HID::NpadStyleIndex::None) {
  886. controller.device->Disconnect();
  887. controller.device->SetNpadStyleIndex(Core::HID::NpadStyleIndex::None);
  888. }
  889. }
  890. }
  891. }
  892. void NPad::DisconnectAllConnectedControllers() {
  893. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; aruid_index++) {
  894. for (auto& controller : controller_data[aruid_index]) {
  895. controller.device->Disconnect();
  896. }
  897. }
  898. }
  899. void NPad::ConnectAllDisconnectedControllers() {
  900. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; aruid_index++) {
  901. for (auto& controller : controller_data[aruid_index]) {
  902. if (controller.device->GetNpadStyleIndex() != Core::HID::NpadStyleIndex::None &&
  903. !controller.device->IsConnected()) {
  904. controller.device->Connect();
  905. }
  906. }
  907. }
  908. }
  909. void NPad::ClearAllControllers() {
  910. for (std::size_t aruid_index = 0; aruid_index < AruidIndexMax; aruid_index++) {
  911. for (auto& controller : controller_data[aruid_index]) {
  912. controller.device->Disconnect();
  913. controller.device->SetNpadStyleIndex(Core::HID::NpadStyleIndex::None);
  914. }
  915. }
  916. }
  917. Core::HID::NpadButton NPad::GetAndResetPressState() {
  918. return static_cast<Core::HID::NpadButton>(press_state.exchange(0));
  919. }
  920. Result NPad::ApplyNpadSystemCommonPolicy(u64 aruid) {
  921. std::scoped_lock lock{mutex};
  922. const Result result = npad_resource.ApplyNpadSystemCommonPolicy(aruid, false);
  923. if (result.IsSuccess()) {
  924. OnUpdate({});
  925. }
  926. return result;
  927. }
  928. Result NPad::ApplyNpadSystemCommonPolicyFull(u64 aruid) {
  929. std::scoped_lock lock{mutex};
  930. const Result result = npad_resource.ApplyNpadSystemCommonPolicy(aruid, true);
  931. if (result.IsSuccess()) {
  932. OnUpdate({});
  933. }
  934. return result;
  935. }
  936. Result NPad::ClearNpadSystemCommonPolicy(u64 aruid) {
  937. std::scoped_lock lock{mutex};
  938. const Result result = npad_resource.ClearNpadSystemCommonPolicy(aruid);
  939. if (result.IsSuccess()) {
  940. OnUpdate({});
  941. }
  942. return result;
  943. }
  944. void NPad::SetRevision(u64 aruid, NpadRevision revision) {
  945. npad_resource.SetNpadRevision(aruid, revision);
  946. }
  947. NpadRevision NPad::GetRevision(u64 aruid) {
  948. return npad_resource.GetNpadRevision(aruid);
  949. }
  950. Result NPad::RegisterAppletResourceUserId(u64 aruid) {
  951. return npad_resource.RegisterAppletResourceUserId(aruid);
  952. }
  953. void NPad::UnregisterAppletResourceUserId(u64 aruid) {
  954. // TODO: Remove this once abstract pad is emulated properly
  955. const auto aruid_index = npad_resource.GetIndexFromAruid(aruid);
  956. for (auto& controller : controller_data[aruid_index]) {
  957. controller.is_active = false;
  958. controller.is_connected = false;
  959. controller.shared_memory = nullptr;
  960. }
  961. npad_resource.UnregisterAppletResourceUserId(aruid);
  962. }
  963. void NPad::SetNpadExternals(std::shared_ptr<AppletResource> resource,
  964. std::recursive_mutex* shared_mutex,
  965. std::shared_ptr<HandheldConfig> handheld_config,
  966. Kernel::KEvent* input_event, std::mutex* input_mutex,
  967. std::shared_ptr<Service::Set::ISystemSettingsServer> settings) {
  968. applet_resource_holder.applet_resource = resource;
  969. applet_resource_holder.shared_mutex = shared_mutex;
  970. applet_resource_holder.shared_npad_resource = &npad_resource;
  971. applet_resource_holder.handheld_config = handheld_config;
  972. applet_resource_holder.input_event = input_event;
  973. applet_resource_holder.input_mutex = input_mutex;
  974. vibration_handler.SetSettingsService(settings);
  975. for (auto& abstract_pad : abstracted_pads) {
  976. abstract_pad.SetExternals(&applet_resource_holder, nullptr, nullptr, nullptr, nullptr,
  977. &vibration_handler, &hid_core);
  978. }
  979. }
  980. NPad::NpadControllerData& NPad::GetControllerFromHandle(
  981. u64 aruid, const Core::HID::SixAxisSensorHandle& device_handle) {
  982. const auto npad_id = static_cast<Core::HID::NpadIdType>(device_handle.npad_id);
  983. return GetControllerFromNpadIdType(aruid, npad_id);
  984. }
  985. const NPad::NpadControllerData& NPad::GetControllerFromHandle(
  986. u64 aruid, const Core::HID::SixAxisSensorHandle& device_handle) const {
  987. const auto npad_id = static_cast<Core::HID::NpadIdType>(device_handle.npad_id);
  988. return GetControllerFromNpadIdType(aruid, npad_id);
  989. }
  990. NPad::NpadControllerData& NPad::GetControllerFromNpadIdType(u64 aruid,
  991. Core::HID::NpadIdType npad_id) {
  992. if (!IsNpadIdValid(npad_id)) {
  993. LOG_ERROR(Service_HID, "Invalid NpadIdType npad_id:{}", npad_id);
  994. npad_id = Core::HID::NpadIdType::Player1;
  995. }
  996. const auto npad_index = NpadIdTypeToIndex(npad_id);
  997. const auto aruid_index = applet_resource_holder.applet_resource->GetIndexFromAruid(aruid);
  998. return controller_data[aruid_index][npad_index];
  999. }
  1000. const NPad::NpadControllerData& NPad::GetControllerFromNpadIdType(
  1001. u64 aruid, Core::HID::NpadIdType npad_id) const {
  1002. if (!IsNpadIdValid(npad_id)) {
  1003. LOG_ERROR(Service_HID, "Invalid NpadIdType npad_id:{}", npad_id);
  1004. npad_id = Core::HID::NpadIdType::Player1;
  1005. }
  1006. const auto npad_index = NpadIdTypeToIndex(npad_id);
  1007. const auto aruid_index = applet_resource_holder.applet_resource->GetIndexFromAruid(aruid);
  1008. return controller_data[aruid_index][npad_index];
  1009. }
  1010. Core::HID::SixAxisSensorProperties& NPad::GetSixaxisProperties(
  1011. u64 aruid, const Core::HID::SixAxisSensorHandle& sixaxis_handle) {
  1012. auto& controller = GetControllerFromHandle(aruid, sixaxis_handle);
  1013. switch (sixaxis_handle.npad_type) {
  1014. case Core::HID::NpadStyleIndex::Fullkey:
  1015. case Core::HID::NpadStyleIndex::Pokeball:
  1016. return controller.shared_memory->sixaxis_fullkey_properties;
  1017. case Core::HID::NpadStyleIndex::Handheld:
  1018. return controller.shared_memory->sixaxis_handheld_properties;
  1019. case Core::HID::NpadStyleIndex::JoyconDual:
  1020. if (sixaxis_handle.device_index == Core::HID::DeviceIndex::Left) {
  1021. return controller.shared_memory->sixaxis_dual_left_properties;
  1022. }
  1023. return controller.shared_memory->sixaxis_dual_right_properties;
  1024. case Core::HID::NpadStyleIndex::JoyconLeft:
  1025. return controller.shared_memory->sixaxis_left_properties;
  1026. case Core::HID::NpadStyleIndex::JoyconRight:
  1027. return controller.shared_memory->sixaxis_right_properties;
  1028. default:
  1029. return controller.shared_memory->sixaxis_fullkey_properties;
  1030. }
  1031. }
  1032. const Core::HID::SixAxisSensorProperties& NPad::GetSixaxisProperties(
  1033. u64 aruid, const Core::HID::SixAxisSensorHandle& sixaxis_handle) const {
  1034. const auto& controller = GetControllerFromHandle(aruid, sixaxis_handle);
  1035. switch (sixaxis_handle.npad_type) {
  1036. case Core::HID::NpadStyleIndex::Fullkey:
  1037. case Core::HID::NpadStyleIndex::Pokeball:
  1038. return controller.shared_memory->sixaxis_fullkey_properties;
  1039. case Core::HID::NpadStyleIndex::Handheld:
  1040. return controller.shared_memory->sixaxis_handheld_properties;
  1041. case Core::HID::NpadStyleIndex::JoyconDual:
  1042. if (sixaxis_handle.device_index == Core::HID::DeviceIndex::Left) {
  1043. return controller.shared_memory->sixaxis_dual_left_properties;
  1044. }
  1045. return controller.shared_memory->sixaxis_dual_right_properties;
  1046. case Core::HID::NpadStyleIndex::JoyconLeft:
  1047. return controller.shared_memory->sixaxis_left_properties;
  1048. case Core::HID::NpadStyleIndex::JoyconRight:
  1049. return controller.shared_memory->sixaxis_right_properties;
  1050. default:
  1051. return controller.shared_memory->sixaxis_fullkey_properties;
  1052. }
  1053. }
  1054. AppletDetailedUiType NPad::GetAppletDetailedUiType(Core::HID::NpadIdType npad_id) {
  1055. const auto aruid = applet_resource_holder.applet_resource->GetActiveAruid();
  1056. const auto& shared_memory = GetControllerFromNpadIdType(aruid, npad_id).shared_memory;
  1057. return {
  1058. .ui_variant = 0,
  1059. .footer = shared_memory->applet_footer_type,
  1060. };
  1061. }
  1062. Result NPad::SetNpadCaptureButtonAssignment(u64 aruid, Core::HID::NpadStyleSet npad_style_set,
  1063. Core::HID::NpadButton button_assignment) {
  1064. std::scoped_lock lock{mutex};
  1065. return npad_resource.SetNpadCaptureButtonAssignment(aruid, npad_style_set, button_assignment);
  1066. }
  1067. Result NPad::ClearNpadCaptureButtonAssignment(u64 aruid) {
  1068. std::scoped_lock lock{mutex};
  1069. return npad_resource.ClearNpadCaptureButtonAssignment(aruid);
  1070. }
  1071. std::size_t NPad::GetNpadCaptureButtonAssignment(std::span<Core::HID::NpadButton> out_list,
  1072. u64 aruid) const {
  1073. std::scoped_lock lock{mutex};
  1074. return npad_resource.GetNpadCaptureButtonAssignment(out_list, aruid);
  1075. }
  1076. Result NPad::SetNpadSystemExtStateEnabled(u64 aruid, bool is_enabled) {
  1077. std::scoped_lock lock{mutex};
  1078. const auto result = npad_resource.SetNpadSystemExtStateEnabled(aruid, is_enabled);
  1079. if (result.IsSuccess()) {
  1080. std::scoped_lock shared_lock{*applet_resource_holder.shared_mutex};
  1081. // TODO: abstracted_pad->EnableAppletToGetInput(aruid);
  1082. }
  1083. return result;
  1084. }
  1085. Result NPad::AssigningSingleOnSlSrPress(u64 aruid, bool is_enabled) {
  1086. std::scoped_lock lock{mutex};
  1087. bool is_currently_enabled{};
  1088. Result result = npad_resource.IsAssigningSingleOnSlSrPressEnabled(is_currently_enabled, aruid);
  1089. if (result.IsSuccess() && is_enabled != is_currently_enabled) {
  1090. result = npad_resource.SetAssigningSingleOnSlSrPress(aruid, is_enabled);
  1091. }
  1092. return result;
  1093. }
  1094. Result NPad::GetLastActiveNpad(Core::HID::NpadIdType& out_npad_id) const {
  1095. std::scoped_lock lock{mutex};
  1096. out_npad_id = hid_core.GetLastActiveController();
  1097. return ResultSuccess;
  1098. }
  1099. NpadVibration* NPad::GetVibrationHandler() {
  1100. return &vibration_handler;
  1101. }
  1102. std::vector<NpadVibrationBase*> NPad::GetAllVibrationDevices() {
  1103. std::vector<NpadVibrationBase*> vibration_devices;
  1104. for (auto& abstract_pad : abstracted_pads) {
  1105. auto* left_device = abstract_pad.GetVibrationDevice(Core::HID::DeviceIndex::Left);
  1106. auto* right_device = abstract_pad.GetVibrationDevice(Core::HID::DeviceIndex::Right);
  1107. auto* n64_device = abstract_pad.GetGCVibrationDevice();
  1108. auto* gc_device = abstract_pad.GetGCVibrationDevice();
  1109. if (left_device != nullptr) {
  1110. vibration_devices.emplace_back(left_device);
  1111. }
  1112. if (right_device != nullptr) {
  1113. vibration_devices.emplace_back(right_device);
  1114. }
  1115. if (n64_device != nullptr) {
  1116. vibration_devices.emplace_back(n64_device);
  1117. }
  1118. if (gc_device != nullptr) {
  1119. vibration_devices.emplace_back(gc_device);
  1120. }
  1121. }
  1122. return vibration_devices;
  1123. }
  1124. NpadVibrationBase* NPad::GetVibrationDevice(const Core::HID::VibrationDeviceHandle& handle) {
  1125. if (IsVibrationHandleValid(handle).IsError()) {
  1126. return nullptr;
  1127. }
  1128. const auto npad_index = NpadIdTypeToIndex(static_cast<Core::HID::NpadIdType>(handle.npad_id));
  1129. const auto style_inde = static_cast<Core::HID::NpadStyleIndex>(handle.npad_type);
  1130. if (style_inde == Core::HID::NpadStyleIndex::GameCube) {
  1131. return abstracted_pads[npad_index].GetGCVibrationDevice();
  1132. }
  1133. if (style_inde == Core::HID::NpadStyleIndex::N64) {
  1134. return abstracted_pads[npad_index].GetN64VibrationDevice();
  1135. }
  1136. return abstracted_pads[npad_index].GetVibrationDevice(handle.device_index);
  1137. }
  1138. NpadN64VibrationDevice* NPad::GetN64VibrationDevice(
  1139. const Core::HID::VibrationDeviceHandle& handle) {
  1140. if (IsVibrationHandleValid(handle).IsError()) {
  1141. return nullptr;
  1142. }
  1143. const auto npad_index = NpadIdTypeToIndex(static_cast<Core::HID::NpadIdType>(handle.npad_id));
  1144. const auto style_inde = static_cast<Core::HID::NpadStyleIndex>(handle.npad_type);
  1145. if (style_inde != Core::HID::NpadStyleIndex::N64) {
  1146. return nullptr;
  1147. }
  1148. return abstracted_pads[npad_index].GetN64VibrationDevice();
  1149. }
  1150. NpadVibrationDevice* NPad::GetNSVibrationDevice(const Core::HID::VibrationDeviceHandle& handle) {
  1151. if (IsVibrationHandleValid(handle).IsError()) {
  1152. return nullptr;
  1153. }
  1154. const auto npad_index = NpadIdTypeToIndex(static_cast<Core::HID::NpadIdType>(handle.npad_id));
  1155. const auto style_inde = static_cast<Core::HID::NpadStyleIndex>(handle.npad_type);
  1156. if (style_inde == Core::HID::NpadStyleIndex::GameCube ||
  1157. style_inde == Core::HID::NpadStyleIndex::N64) {
  1158. return nullptr;
  1159. }
  1160. return abstracted_pads[npad_index].GetVibrationDevice(handle.device_index);
  1161. }
  1162. NpadGcVibrationDevice* NPad::GetGcVibrationDevice(const Core::HID::VibrationDeviceHandle& handle) {
  1163. if (IsVibrationHandleValid(handle).IsError()) {
  1164. return nullptr;
  1165. }
  1166. const auto npad_index = NpadIdTypeToIndex(static_cast<Core::HID::NpadIdType>(handle.npad_id));
  1167. const auto style_inde = static_cast<Core::HID::NpadStyleIndex>(handle.npad_type);
  1168. if (style_inde != Core::HID::NpadStyleIndex::GameCube) {
  1169. return nullptr;
  1170. }
  1171. return abstracted_pads[npad_index].GetGCVibrationDevice();
  1172. }
  1173. void NPad::UpdateHandheldAbstractState() {
  1174. std::scoped_lock lock{mutex};
  1175. abstracted_pads[NpadIdTypeToIndex(Core::HID::NpadIdType::Handheld)].Update();
  1176. }
  1177. void NPad::EnableAppletToGetInput(u64 aruid) {
  1178. std::scoped_lock lock{mutex};
  1179. std::scoped_lock shared_lock{*applet_resource_holder.shared_mutex};
  1180. for (auto& abstract_pad : abstracted_pads) {
  1181. abstract_pad.EnableAppletToGetInput(aruid);
  1182. }
  1183. }
  1184. } // namespace Service::HID