diff --git a/klippy/extras/tmc_uart.py b/klippy/extras/tmc_uart.py index b87dfca9..32e74b24 100644 --- a/klippy/extras/tmc_uart.py +++ b/klippy/extras/tmc_uart.py @@ -53,13 +53,30 @@ class MCU_analog_mux: # TMC uart communication ###################################################################### +# Share mutexes so only one active tmc_uart command on a single mcu at +# a time. This helps limit cpu usage on slower micro-controllers. +class PrinterTMCUartMutexes: + def __init__(self): + self.mcu_to_mutex = {} +def lookup_tmc_uart_mutex(mcu): + printer = mcu.get_printer() + pmutexes = printer.lookup_object('tmc_uart', None) + if pmutexes is None: + pmutexes = PrinterTMCUartMutexes() + printer.add_object('tmc_uart', pmutexes) + mutex = pmutexes.mcu_to_mutex.get(mcu) + if mutex is None: + mutex = printer.get_reactor().mutex() + pmutexes.mcu_to_mutex[mcu] = mutex + return mutex + TMC_BAUD_RATE = 9000 # Code for sending messages on a TMC uart class MCU_TMC_uart_bitbang: def __init__(self, rx_pin_params, tx_pin_params, select_pins_desc): self.mcu = rx_pin_params['chip'] - self.mutex = self.mcu.get_printer().get_reactor().mutex() + self.mutex = lookup_tmc_uart_mutex(self.mcu) self.pullup = rx_pin_params['pullup'] self.rx_pin = rx_pin_params['pin'] self.tx_pin = tx_pin_params['pin']