Added basic controller LED (e.g. Dualshock 4) support

mmatyas · Dec 28, 2023 · 6cf4ec3 · 6cf4ec3
1 parent 9860315
commit 6cf4ec3
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/src/common/gfx.cpp b/src/common/gfx.cpp
@@ -637,3 +637,10 @@ bool gfx_loadimage(gfxSprite * gSprite, const std::string& f, Uint8 r, Uint8 g,
 
     return fRet;
 }
+
+void gfx_setjoystickteamcolor(SDL_Joystick * joystick, unsigned short team, Uint8 brightness)
+{
+    uint8_t r = 0, g = 0, b = 0;
+    gfx.getPalette().copyColorSchemeTo(team, 0, 5, r, g, b);
+    SDL_JoystickSetLED(joystick, ((short)brightness + 1) * r >> 8, ((short)brightness + 1) * g >> 8, ((short)brightness + 1) * b >> 8);
+}
diff --git a/src/common/gfx.h b/src/common/gfx.h
@@ -60,4 +60,6 @@ bool gfx_loadimage(gfxSprite * gSprite, const std::string& f, bool fWrap = true,
 bool gfx_loadimage(gfxSprite * gSprite, const std::string& f, Uint8 alpha, bool fWrap = true, bool fUseAccel = true);
 bool gfx_loadimage(gfxSprite * gSprite, const std::string& f, Uint8 r, Uint8 g, Uint8 b, bool fWrap = true, bool fUseAccel = true);
 
+void gfx_setjoystickteamcolor(SDL_Joystick * joystick, unsigned short team, Uint8 brightness);
+
 #endif // GFX_H
diff --git a/src/smw/ui/MI_TeamSelect.cpp b/src/smw/ui/MI_TeamSelect.cpp
@@ -1,5 +1,6 @@
 #include "MI_TeamSelect.h"
 
+#include "gfx.h"
 #include "Game.h"
 #include "GameValues.h"
 #include "ResourceManager.h"
@@ -223,6 +224,10 @@ MenuCodeEnum MI_TeamSelect::SendInput(CPlayerInput * playerInput)
                 }
             }
         }
+        if (DEVICE_KEYBOARD != playerInput->inputControls[iPlayer]->iDevice) {
+            short team = GetTeam(iPlayer);
+            gfx_setjoystickteamcolor(SDL_JoystickFromPlayerIndex(playerInput->inputControls[iPlayer]->iDevice), team, fReady[iPlayer] ? 200 : 100);
+        }
     }
 
     return MENU_CODE_NONE;