Skip to content

Add auto-detection for color LED deck variants in examples #578

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gemenerik merged 1 commit into from
Nov 26, 2025
Merged

Add auto-detection for color LED deck variants in examples #578

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 30 additions & 7 deletions examples/color_led_deck/color_led_cycle.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -94,17 +94,40 @@ def pack_wrgb(input: wrgb) -> int:
with SyncCrazyflie(URI, cf=Crazyflie(rw_cache='./cache')) as scf:
cf = scf.cf

# Detect which color LED deck is present (bottom or top)
# We check for whichever deck is attached and use the first one found
# Check for bottom-facing deck first
if str(cf.param.get_value('deck.bcColorLedBot')) == '1':
deck_params = {
'color': 'colorLedBot.wrgb8888',
'brightness': 'colorLedBot.brightCorr',
'throttle': 'colorLedBot.throttlePct',
'temp': 'colorLedBot.deckTemp'
}
print('Detected bottom-facing color LED deck')
# Check for top-facing deck
elif str(cf.param.get_value('deck.bcColorLedTop')) == '1':
deck_params = {
'color': 'colorLedTop.wrgb8888',
'brightness': 'colorLedTop.brightCorr',
'throttle': 'colorLedTop.throttlePct',
'temp': 'colorLedTop.deckTemp'
}
print('Detected top-facing color LED deck')
else:
raise RuntimeError('No color LED deck detected!')

# Thermal status callback
def thermal_status_callback(timestamp, data, logconf):
throttle_pct = data['colorled.throttlePct']
throttle_pct = data[deck_params['throttle']]
if throttle_pct > 0:
temp = data['colorled.deckTemp']
temp = data[deck_params['temp']]
print(f'WARNING: Thermal throttling active! Temp: {temp}°C, Throttle: {throttle_pct}%')

# Setup log configuration for thermal monitoring
log_conf = LogConfig(name='ThermalStatus', period_in_ms=100)
log_conf.add_variable('colorled.deckTemp', 'uint8_t')
log_conf.add_variable('colorled.throttlePct', 'uint8_t')
log_conf.add_variable(deck_params['temp'], 'uint8_t')
log_conf.add_variable(deck_params['throttle'], 'uint8_t')

cf.log.add_config(log_conf)
log_conf.data_received_cb.add_callback(thermal_status_callback)
Expand All @@ -113,7 +136,7 @@ def thermal_status_callback(timestamp, data, logconf):
# Brightness correction: balances luminance across W/R/G/B channels
# Set to 1 (enabled, default) for perceptually uniform colors
# Set to 0 (disabled) for maximum brightness per channel
cf.param.set_value('colorled.brightnessCorr', 1)
cf.param.set_value(deck_params['brightness'], 1)
time.sleep(0.1)

try:
Expand All @@ -123,9 +146,9 @@ def thermal_status_callback(timestamp, data, logconf):
color: wrgb = cycle_colors(i)
# print(color.r, color.g, color.b)
color_uint32 = pack_wrgb(color)
cf.param.set_value('colorled.wrgb8888', str(color_uint32))
cf.param.set_value(deck_params['color'], str(color_uint32))
time.sleep(0.01)
except KeyboardInterrupt:
print('\nStopping and turning off LED...')
cf.param.set_value('colorled.wrgb8888', '0')
cf.param.set_value(deck_params['color'], '0')
time.sleep(0.1)
37 changes: 30 additions & 7 deletions examples/color_led_deck/color_led_set.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,17 +59,40 @@ def pack_wrgb(input: wrgb | rgb) -> int:
with SyncCrazyflie(URI, cf=Crazyflie(rw_cache='./cache')) as scf:
cf = scf.cf

# Detect which color LED deck is present (bottom or top)
# We check for whichever deck is attached and use the first one found
# Check for bottom-facing deck first
if str(cf.param.get_value('deck.bcColorLedBot')) == '1':
deck_params = {
'color': 'colorLedBot.wrgb8888',
'brightness': 'colorLedBot.brightCorr',
'throttle': 'colorLedBot.throttlePct',
'temp': 'colorLedBot.deckTemp'
}
print('Detected bottom-facing color LED deck')
# Check for top-facing deck
elif str(cf.param.get_value('deck.bcColorLedTop')) == '1':
deck_params = {
'color': 'colorLedTop.wrgb8888',
'brightness': 'colorLedTop.brightCorr',
'throttle': 'colorLedTop.throttlePct',
'temp': 'colorLedTop.deckTemp'
}
print('Detected top-facing color LED deck')
else:
raise RuntimeError('No color LED deck detected!')

# Thermal status callback
def thermal_status_callback(timestamp, data, logconf):
throttle_pct = data['colorled.throttlePct']
throttle_pct = data[deck_params['throttle']]
if throttle_pct > 0:
temp = data['colorled.deckTemp']
temp = data[deck_params['temp']]
print(f'WARNING: Thermal throttling active! Temp: {temp}°C, Throttle: {throttle_pct}%')

# Setup log configuration for thermal monitoring
log_conf = LogConfig(name='ThermalStatus', period_in_ms=100)
log_conf.add_variable('colorled.deckTemp', 'uint8_t')
log_conf.add_variable('colorled.throttlePct', 'uint8_t')
log_conf.add_variable(deck_params['temp'], 'uint8_t')
log_conf.add_variable(deck_params['throttle'], 'uint8_t')

cf.log.add_config(log_conf)
log_conf.data_received_cb.add_callback(thermal_status_callback)
Expand All @@ -78,7 +101,7 @@ def thermal_status_callback(timestamp, data, logconf):
# Brightness correction: balances luminance across R/G/B/W channels
# Set to 1 (enabled, default) for perceptually uniform colors
# Set to 0 (disabled) for maximum brightness per channel
cf.param.set_value('colorled.brightnessCorr', 1)
cf.param.set_value(deck_params['brightness'], 1)
time.sleep(0.1)

try:
Expand All @@ -97,13 +120,13 @@ def thermal_status_callback(timestamp, data, logconf):
# ========================================

color_uint32 = pack_wrgb(color)
cf.param.set_value('colorled.wrgb8888', color_uint32)
cf.param.set_value(deck_params['color'], color_uint32)
time.sleep(0.01)
print(f'Setting LED to R={color.r}, G={color.g}, B={color.b}, W={color.w}')
print('Press Ctrl-C to turn off LED and exit.')
while True:
time.sleep(0.1)
except KeyboardInterrupt:
print('\nStopping and turning off LED...')
cf.param.set_value('colorled.wrgb8888', 0)
cf.param.set_value(deck_params['color'], 0)
time.sleep(0.1)