AP_SerialLED: support set_num_neopixel_rgb()

andyp1per · Sep 29, 2023 · b8ec902 · b8ec902
1 parent a6e0429
commit b8ec902
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/libraries/AP_SerialLED/AP_SerialLED.cpp b/libraries/AP_SerialLED/AP_SerialLED.cpp
@@ -28,11 +28,19 @@ extern const AP_HAL::HAL& hal;
 AP_SerialLED AP_SerialLED::singleton;
 
 // set number of NeoPixels per pin
-bool AP_SerialLED::set_num_neopixel(uint8_t chan, uint8_t num_leds, bool rgb)
+bool AP_SerialLED::set_num_neopixel(uint8_t chan, uint8_t num_leds)
 {
     if (chan >= 1 && chan <= 16 && num_leds <= AP_SERIALLED_MAX_LEDS) {
-        return hal.rcout->set_serial_led_num_LEDs(chan-1, num_leds,
-            rgb ? AP_HAL::RCOutput::MODE_NEOPIXELRGB : AP_HAL::RCOutput::MODE_NEOPIXEL);
+        return hal.rcout->set_serial_led_num_LEDs(chan-1, num_leds, AP_HAL::RCOutput::MODE_NEOPIXEL);
+    }
+    return false;
+}
+
+// set number of NeoPixels per pin in RGB mode
+bool AP_SerialLED::set_num_neopixel_rgb(uint8_t chan, uint8_t num_leds)
+{
+    if (chan >= 1 && chan <= 16 && num_leds <= AP_SERIALLED_MAX_LEDS) {
+        return hal.rcout->set_serial_led_num_LEDs(chan-1, num_leds, AP_HAL::RCOutput::MODE_NEOPIXELRGB);
     }
     return false;
 }

diff --git a/libraries/AP_SerialLED/AP_SerialLED.h b/libraries/AP_SerialLED/AP_SerialLED.h
@@ -33,8 +33,10 @@
 class AP_SerialLED {
 public:
     // set number of LEDs per pin
-    bool set_num_neopixel(uint8_t chan, uint8_t num_leds, bool rgb = false);
+    bool set_num_neopixel(uint8_t chan, uint8_t num_leds);
     bool set_num_profiled(uint8_t chan, uint8_t num_leds);
+    // set number of LEDs per pin in RGB mode
+    bool set_num_neopixel_rgb(uint8_t chan, uint8_t num_leds);
 
     // set RGB value on mask of LEDs. chan is PWM output, 1..16
     void set_RGB_mask(uint8_t chan, uint32_t ledmask, uint8_t red, uint8_t green, uint8_t blue);