95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
"""
|
||
protocol.py - Command builder for common cheap BLE LED controller protocols.
|
||
|
||
These controllers all speak one of a handful of similar protocols.
|
||
We'll confirm which format the KAIYU uses after sniffing/probing.
|
||
|
||
Supported protocol families:
|
||
- 0x7E family (most common, used by ELK-BLEDOM, LEDBLE, etc.)
|
||
- 0x56 family (older MagicLight / JACKYLED style)
|
||
"""
|
||
|
||
|
||
class Protocol7E:
|
||
"""
|
||
The 0x7E protocol family.
|
||
|
||
Packet format: 7e 00 <CMD> <P1> <P2> <P3> <P4> <LED> ef
|
||
CMD 0x04 = power on/off
|
||
CMD 0x05 = set colour (P1=mode, P2=R, P3=G, P4=B)
|
||
CMD 0x01 = set brightness (P1=0x00–0x64)
|
||
CMD 0x03 = set effect/speed
|
||
"""
|
||
|
||
# Colour mode byte for CMD 0x05
|
||
MODE_STATIC = 0x03
|
||
MODE_EFFECT = 0x02
|
||
|
||
# LED channel — 0xFF = all channels
|
||
LED_ALL = 0xFF
|
||
|
||
@staticmethod
|
||
def set_color(r: int, g: int, b: int) -> bytes:
|
||
r, g, b = _clamp(r), _clamp(g), _clamp(b)
|
||
return bytes([0x7e, 0x00, 0x05, Protocol7E.MODE_STATIC, r, g, b, 0x00, 0xef])
|
||
|
||
@staticmethod
|
||
def set_brightness(level: int) -> bytes:
|
||
"""level: 0–100"""
|
||
level = max(0, min(100, level))
|
||
return bytes([0x7e, 0x00, 0x01, level, 0x00, 0x00, 0x00, Protocol7E.LED_ALL, 0xef])
|
||
|
||
@staticmethod
|
||
def turn_on() -> bytes:
|
||
return bytes([0x7e, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00, Protocol7E.LED_ALL, 0xef])
|
||
|
||
@staticmethod
|
||
def turn_off() -> bytes:
|
||
return bytes([0x7e, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0xef])
|
||
|
||
@staticmethod
|
||
def set_effect(effect_id: int, speed: int = 0x03) -> bytes:
|
||
"""
|
||
effect_id: 0x80–0x8b (varies by controller)
|
||
speed: 0x01 (fast) – 0x0a (slow)
|
||
"""
|
||
return bytes([0x7e, 0x00, 0x03, effect_id, speed, 0x00, 0x00, Protocol7E.LED_ALL, 0xef])
|
||
|
||
|
||
class Protocol56:
|
||
"""
|
||
The 0x56 protocol family (older MagicLight style).
|
||
|
||
RGB command : 56 RR GG BB 00 f0 aa
|
||
White cmd : 56 00 00 00 WW 0f aa
|
||
Power on : cc 23 33
|
||
Power off : cc 24 33
|
||
"""
|
||
|
||
@staticmethod
|
||
def set_color(r: int, g: int, b: int) -> bytes:
|
||
r, g, b = _clamp(r), _clamp(g), _clamp(b)
|
||
return bytes([0x56, r, g, b, 0x00, 0xf0, 0xaa])
|
||
|
||
@staticmethod
|
||
def set_white(level: int) -> bytes:
|
||
level = _clamp(level)
|
||
return bytes([0x56, 0x00, 0x00, 0x00, level, 0x0f, 0xaa])
|
||
|
||
@staticmethod
|
||
def turn_on() -> bytes:
|
||
return bytes([0xcc, 0x23, 0x33])
|
||
|
||
@staticmethod
|
||
def turn_off() -> bytes:
|
||
return bytes([0xcc, 0x24, 0x33])
|
||
|
||
|
||
def _clamp(v: int) -> int:
|
||
return max(0, min(255, v))
|
||
|
||
|
||
# ── Convenience ───────────────────────────────────────────────────────────────
|
||
# Default to 7E family — update after confirming with scanner
|
||
ActiveProtocol = Protocol7E
|