1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
# Support for "dotstar" leds
#
# Copyright (C) 2019-2024 Kevin O'Connor <kevin@koconnor.net>
#
# This file may be distributed under the terms of the GNU GPLv3 license.
from . import bus, led
BACKGROUND_PRIORITY_CLOCK = 0x7FFFFFFF00000000
class PrinterDotstar:
def __init__(self, config):
self.printer = printer = config.get_printer()
# Configure a software spi bus
ppins = printer.lookup_object("pins")
data_pin_params = ppins.lookup_pin(config.get("data_pin"))
clock_pin_params = ppins.lookup_pin(config.get("clock_pin"))
mcu = data_pin_params["chip"]
if mcu is not clock_pin_params["chip"]:
raise config.error("Dotstar pins must be on same mcu")
sw_spi_pins = (
data_pin_params["pin"],
data_pin_params["pin"],
clock_pin_params["pin"],
)
self.spi = bus.MCU_SPI(mcu, None, None, 0, 500000, sw_spi_pins)
# Initialize color data
self.chain_count = config.getint("chain_count", 1, minval=1)
self.led_helper = led.LEDHelper(config, self.update_leds, self.chain_count)
self.prev_data = None
# Register commands
printer.register_event_handler("klippy:connect", self.handle_connect)
def handle_connect(self):
self.update_leds(self.led_helper.get_status()["color_data"], None)
def update_leds(self, led_state, print_time):
if led_state == self.prev_data:
return
self.prev_data = led_state
# Build data to send
data = [0] * ((len(led_state) + 2) * 4)
for i, (red, green, blue, white) in enumerate(led_state):
idx = (i + 1) * 4
data[idx] = 0xFF
data[idx + 1] = int(blue * 255.0 + 0.5)
data[idx + 2] = int(green * 255.0 + 0.5)
data[idx + 3] = int(red * 255.0 + 0.5)
data[-4] = data[-3] = data[-2] = data[-1] = 0xFF
# Transmit update
minclock = 0
if print_time is not None:
minclock = self.spi.get_mcu().print_time_to_clock(print_time)
for d in [data[i : i + 20] for i in range(0, len(data), 20)]:
self.spi.spi_send(d, minclock=minclock, reqclock=BACKGROUND_PRIORITY_CLOCK)
def get_status(self, eventtime):
return self.led_helper.get_status(eventtime)
def load_config_prefix(config):
return PrinterDotstar(config)
|
