inputmodule-control: Add command for debug mode

Signed-off-by: Daniel Schaefer <[email protected]>

Author: JohnAZoidberg

fl16-inputmodules/src/control.rs

@@ -69,6 +69,7 @@ pub enum CommandVals {
     SetPowerMode = 0x1B,
     AnimationPeriod = 0x1C,
     PwmFreq = 0x1E,
+    DebugMode = 0x1F,
     Version = 0x20,
 }
 
@@ -205,6 +206,8 @@ pub enum Command {
     #[cfg(feature = "ledmatrix")]
     SetPwmFreq(PwmFreqArg),
     GetPwmFreq,
+    SetDebugMode(bool),
+    GetDebugMode,
     _Unknown,
 }
 
@@ -385,6 +388,11 @@ pub fn parse_module_command(count: usize, buf: &[u8]) -> Option<Command> {
                     Some(Command::GetPwmFreq)
                 }
             }
+            Some(CommandVals::DebugMode) => Some(if let Some(debug_mode) = arg {
+                Command::SetDebugMode(debug_mode == 1)
+            } else {
+                Command::GetDebugMode
+            }),
             _ => None,
         }
     } else {
@@ -614,6 +622,15 @@ pub fn handle_command(
             response[0] = state.pwm_freq as u8;
             Some(response)
         }
+        Command::SetDebugMode(arg) => {
+            state.debug_mode = *arg;
+            None
+        }
+        Command::GetDebugMode => {
+            let mut response: [u8; 32] = [0; 32];
+            response[0] = state.debug_mode as u8;
+            Some(response)
+        }
         _ => handle_generic_command(command),
     }
 }

inputmodule-control/src/inputmodule.rs

@@ -48,6 +48,7 @@ enum Command {
     PowerMode = 0x1B,
     AnimationPeriod = 0x1C,
     PwmFreq = 0x1E,
+    DebugMode = 0x1F,
     Version = 0x20,
 }
 
@@ -225,6 +226,9 @@ pub fn serial_commands(args: &crate::ClapCli) {
                 if let Some(freq) = ledmatrix_args.pwm_freq {
                     pwm_freq_cmd(serialdev, freq);
                 }
+                if let Some(debug_mode) = ledmatrix_args.debug_mode {
+                    debug_mode_cmd(serialdev, debug_mode);
+                }
 
                 if ledmatrix_args.stop_game {
                     simple_cmd(
@@ -446,6 +450,26 @@ fn sleeping_cmd(serialdev: &str, arg: Option<bool>) {
     }
 }
 
+fn debug_mode_cmd(serialdev: &str, arg: Option<bool>) {
+    let mut port = serialport::new(serialdev, 115_200)
+        .timeout(SERIAL_TIMEOUT)
+        .open()
+        .expect("Failed to open port");
+
+    if let Some(enable_debug) = arg {
+        simple_cmd_port(&mut port, Command::DebugMode, &[u8::from(enable_debug)]);
+    } else {
+        simple_cmd_port(&mut port, Command::DebugMode, &[]);
+
+        let mut response: Vec<u8> = vec![0; 32];
+        port.read_exact(response.as_mut_slice())
+            .expect("Found no data!");
+
+        let debug_mode: bool = response[0] == 1;
+        println!("Debug Mode enabled: {debug_mode}");
+    }
+}
+
 fn brightness_cmd(serialdev: &str, arg: Option<u8>) {
     let mut port = serialport::new(serialdev, 115_200)
         .timeout(SERIAL_TIMEOUT)

inputmodule-control/src/ledmatrix.rs

@@ -131,6 +131,10 @@ pub struct LedMatrixSubcommand {
     #[clap(value_enum)]
     pub pwm_freq: Option<Option<u16>>,
 
+    /// Set debug mode or get current mode, if no value provided
+    #[arg(long)]
+    pub debug_mode: Option<Option<bool>>,
+
     /// Crash the firmware (TESTING ONLY!)
     #[arg(long)]
     pub panic: bool,

ledmatrix_control.py

@@ -47,6 +47,7 @@ class CommandVals(IntEnum):
     SetFps = 0x1A
     SetPowerMode = 0x1B
     PwmFreq = 0x1E
+    DebugMode = 0x1F
     Version = 0x20