fix(bw): unshifted trims on XLite radios should always adjust elevato…

…r and aileron (#5580)
EdgeTX · Oct 8, 2024 · fcc75bd · fcc75bd
1 parent 4c82ead
commit fcc75bd
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 19 deletions.
diff --git a/radio/src/boards/generic_stm32/inputs.cpp b/radio/src/boards/generic_stm32/inputs.cpp
@@ -41,12 +41,5 @@ __weak uint32_t readKeys()
 
 __weak uint32_t readTrims()
 {
-  uint32_t trims = _read_trims();
-
-#if defined(PCBXLITE)
-  if (_read_keys() & (1 << KEY_SHIFT))
-    return ((trims & 0x03) << 6) | ((trims & 0x0c) << 2);
-#endif
-
-  return trims;
+  return _read_trims();
 }
diff --git a/radio/src/keys.cpp b/radio/src/keys.cpp
@@ -28,7 +28,7 @@
 #include "hal/rotary_encoder.h"
 #include "dataconstants.h"
 
-#if !defined(BOOT) && defined(USE_HATS_AS_KEYS)
+#if !defined(BOOT) && (defined(USE_HATS_AS_KEYS) || defined(PCBXLITE))
 #include "edgetx.h"
 #endif
 
@@ -277,14 +277,38 @@ bool waitKeysReleased()
   return true;
 }
 
+#if defined(PCBXLITE) && !defined(BOOT)
+uint32_t _readTrims()
+{
+  uint32_t trims = readTrims();
+
+  uint8_t lr = trims & 0x3;
+  uint8_t ud = trims & 0xc;
+  bool shift = readKeys() & (1 << KEY_SHIFT);
+  // Mode 1 or 2 - AIL on right stick
+  bool ailRight = g_eeGeneral.stickMode < 2;
+  // Mode 2 or 4 - ELE on right stick
+  bool eleRight = (g_eeGeneral.stickMode & 1) == 1;
+  // Ensure non-shifted trims are AIL and ELE
+  if (ailRight == !shift) lr <<= 6;
+  if (eleRight == !shift) ud <<= 2;
+
+  return lr | ud;
+}
+
+#define READ_TRIMS() _readTrims()
+#else
+#define READ_TRIMS() readTrims()
+#endif
+
 bool keyDown()
 {
   return readKeys() || readTrims();
 }
 
 bool trimDown(uint8_t idx)
 {
-  return readTrims() & (1 << idx);
+  return READ_TRIMS() & (1 << idx);
 }
 
 uint8_t keysGetState(uint8_t key)
@@ -431,7 +455,7 @@ bool keysPollingCycle()
     trims_input = readTrims();
   }
 #else
-  trims_input = readTrims();
+  trims_input = READ_TRIMS();
 #endif
 
   for (int i = 0; i < MAX_KEYS; i++) {

diff --git a/radio/src/targets/simu/simpgmspace.cpp b/radio/src/targets/simu/simpgmspace.cpp
@@ -484,21 +484,16 @@ uint32_t readKeys()
 
 uint32_t readTrims()
 {
-  uint32_t result = 0;
+  uint32_t trims = 0;
 
   for (int i = 0; i < keysGetMaxTrims() * 2; i++) {
     if (trimsStates[i]) {
       // TRACE("trim pressed %d", i);
-      result |= 1 << i;
+      trims |= 1 << i;
     }
   }
 
-#if defined(PCBXLITE)
-  if (keysStates[KEY_SHIFT])
-    result = ((result & 0x03) << 6) | ((result & 0x0c) << 2);
-#endif
-
-  return result;
+  return trims;
 }
 
 int usbPlugged() { return false; }