diff --git a/libraries/AP_Notify/AP_Notify.cpp b/libraries/AP_Notify/AP_Notify.cpp
index 21aaa62361..fd1fa8f983 100644
--- a/libraries/AP_Notify/AP_Notify.cpp
+++ b/libraries/AP_Notify/AP_Notify.cpp
@@ -38,6 +38,7 @@
#include "AP_BoardLED2.h"
#include "ProfiLED.h"
#include "ScriptingLED.h"
+#include "DShotLED.h"
extern const AP_HAL::HAL& hal;
@@ -96,6 +97,14 @@ AP_Notify *AP_Notify::_singleton;
#define BUZZER_ENABLE_DEFAULT 1
#endif
+#ifndef BUILD_DEFAULT_BUZZER_TYPE
+#if CONFIG_HAL_BOARD == HAL_BOARD_CHIBIOS && HAL_DSHOT_ALARM
+ #define BUILD_DEFAULT_BUZZER_TYPE (BUZZER_ENABLE_DEFAULT | (HAL_DSHOT_ALARM << 1))
+#else
+ #define BUILD_DEFAULT_BUZZER_TYPE BUZZER_ENABLE_DEFAULT
+#endif
+#endif
+
#ifndef NOTIFY_LED_BRIGHT_DEFAULT
#define NOTIFY_LED_BRIGHT_DEFAULT RGB_LED_HIGH
#endif
@@ -122,12 +131,12 @@ const AP_Param::GroupInfo AP_Notify::var_info[] = {
// @User: Advanced
AP_GROUPINFO("LED_BRIGHT", 0, AP_Notify, _rgb_led_brightness, NOTIFY_LED_BRIGHT_DEFAULT),
- // @Param: BUZZ_ENABLE
- // @DisplayName: Buzzer enable
- // @Description: Enable or disable the buzzer.
- // @Values: 0:Disable,1:Enable
+ // @Param: BUZZ_TYPES
+ // @DisplayName: Buzzer Driver Types
+ // @Description: Controls what types of Buzzer will be enabled
+ // @Bitmask: 0:Built-in buzzer, 1:DShot
// @User: Advanced
- AP_GROUPINFO("BUZZ_ENABLE", 1, AP_Notify, _buzzer_enable, BUZZER_ENABLE_DEFAULT),
+ AP_GROUPINFO("BUZZ_TYPES", 1, AP_Notify, _buzzer_type, BUILD_DEFAULT_BUZZER_TYPE),
// @Param: LED_OVERRIDE
// @DisplayName: Specifies colour source for the RGBLed
@@ -164,7 +173,7 @@ const AP_Param::GroupInfo AP_Notify::var_info[] = {
// @Param: LED_TYPES
// @DisplayName: LED Driver Types
// @Description: Controls what types of LEDs will be enabled
- // @Bitmask: 0:Built-in LED, 1:Internal ToshibaLED, 2:External ToshibaLED, 3:External PCA9685, 4:Oreo LED, 5:UAVCAN, 6:NCP5623 External, 7:NCP5623 Internal, 8:NeoPixel, 9:ProfiLED, 10:Scripting
+ // @Bitmask: 0:Built-in LED, 1:Internal ToshibaLED, 2:External ToshibaLED, 3:External PCA9685, 4:Oreo LED, 5:UAVCAN, 6:NCP5623 External, 7:NCP5623 Internal, 8:NeoPixel, 9:ProfiLED, 10:Scripting, 11:DShot
// @User: Advanced
AP_GROUPINFO("LED_TYPES", 6, AP_Notify, _led_type, BUILD_DEFAULT_LED_TYPE),
@@ -313,6 +322,11 @@ void AP_Notify::add_backends(void)
#endif
break;
+ case Notify_LED_DShot:
+#if HAL_SUPPORT_RCOUT_SERIAL
+ ADD_BACKEND(new DShotLED());
+#endif
+ break;
}
}
@@ -323,7 +337,7 @@ void AP_Notify::add_backends(void)
// ChibiOS noise makers
#if CONFIG_HAL_BOARD == HAL_BOARD_CHIBIOS
ADD_BACKEND(new Buzzer());
-#ifdef HAL_PWM_ALARM
+#if defined(HAL_PWM_ALARM) || HAL_DSHOT_ALARM
ADD_BACKEND(new AP_ToneAlarm());
#endif
diff --git a/libraries/AP_Notify/AP_Notify.h b/libraries/AP_Notify/AP_Notify.h
index b992a3413e..335614f4a7 100644
--- a/libraries/AP_Notify/AP_Notify.h
+++ b/libraries/AP_Notify/AP_Notify.h
@@ -72,6 +72,7 @@ public:
Notify_LED_NeoPixel = (1 << 8), // NeoPixel 5050 AdaFruit 1655 SK6812 Worldsemi WS2812B
Notify_LED_ProfiLED = (1 << 9), // ProfiLED
Notify_LED_Scripting = (1 << 10),// Colour accessor for scripting
+ Notify_LED_DShot = (1 << 11),// Use dshot commands to set ESC LEDs
Notify_LED_MAX
};
@@ -155,7 +156,9 @@ public:
// play a tune string
static void play_tune(const char *tune);
- bool buzzer_enabled() const { return _buzzer_enable; }
+ bool buzzer_enabled() const { return _buzzer_type != 0; }
+
+ uint8_t get_buzzer_types() const { return _buzzer_type; }
// set flight mode string
void set_flight_mode_str(const char *str);
@@ -189,7 +192,7 @@ private:
// parameters
AP_Int8 _rgb_led_brightness;
AP_Int8 _rgb_led_override;
- AP_Int8 _buzzer_enable;
+ AP_Int8 _buzzer_type;
AP_Int8 _display_type;
AP_Int8 _oreo_theme;
AP_Int8 _buzzer_pin;
diff --git a/libraries/AP_Notify/DShotLED.cpp b/libraries/AP_Notify/DShotLED.cpp
new file mode 100644
index 0000000000..798905a990
--- /dev/null
+++ b/libraries/AP_Notify/DShotLED.cpp
@@ -0,0 +1,35 @@
+/*
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program. If not, see .
+ */
+
+#include "AP_Notify/AP_Notify.h"
+#include
+#include "DShotLED.h"
+#include "SRV_Channel/SRV_Channel.h"
+
+extern const AP_HAL::HAL& hal;
+
+bool DShotLED::hw_set_rgb(uint8_t red, uint8_t green, uint8_t blue)
+{
+ // don't play the motor LEDs while flying
+ if (hal.util->get_soft_armed() || hal.rcout->get_dshot_esc_type() != AP_HAL::RCOutput::DSHOT_ESC_BLHELI) {
+ return false;
+ }
+
+ hal.rcout->send_dshot_command(red ? AP_HAL::RCOutput::DSHOT_LED1_ON : AP_HAL::RCOutput::DSHOT_LED1_OFF);
+ hal.rcout->send_dshot_command(green ? AP_HAL::RCOutput::DSHOT_LED2_ON : AP_HAL::RCOutput::DSHOT_LED2_OFF);
+ hal.rcout->send_dshot_command(blue ? AP_HAL::RCOutput::DSHOT_LED0_ON : AP_HAL::RCOutput::DSHOT_LED0_OFF);
+
+ return true;
+}
\ No newline at end of file
diff --git a/libraries/AP_Notify/DShotLED.h b/libraries/AP_Notify/DShotLED.h
new file mode 100644
index 0000000000..1caf109d32
--- /dev/null
+++ b/libraries/AP_Notify/DShotLED.h
@@ -0,0 +1,29 @@
+/*
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program. If not, see .
+ */
+#pragma once
+
+#include "RGBLed.h"
+#include "DShotLED.h"
+#include
+
+class DShotLED : public RGBLed {
+public:
+ DShotLED() : RGBLed(0, 1, 1, 1) {}
+
+ bool init(void) override { return true; }
+
+protected:
+ bool hw_set_rgb(uint8_t r, uint8_t g, uint8_t b) override;
+};
diff --git a/libraries/AP_Notify/ToneAlarm.cpp b/libraries/AP_Notify/ToneAlarm.cpp
index 408e8ce1f2..00c0d547fd 100644
--- a/libraries/AP_Notify/ToneAlarm.cpp
+++ b/libraries/AP_Notify/ToneAlarm.cpp
@@ -105,7 +105,7 @@ bool AP_ToneAlarm::init()
if (pNotify->buzzer_enabled() == false) {
return false;
}
- if (!hal.util->toneAlarm_init()) {
+ if (!hal.util->toneAlarm_init(pNotify->get_buzzer_types())) {
return false;
}