tmc: SET_TMC_FIELD VELOCITY

Ability to specify `VELOCITY` as a parameter for SET_TMC_FIELD. Useful for configuring at runtime the TSTEP based fields of the driver.

Signed-off-by: Alex Voinea <voinea.dragos.alexandru@gmail.com>

1 files changed, 4 insertions, 1 deletions

diff --git a/klippy/extras/tmc_uart.py b/klippy/extras/tmc_uart.py
index fe51303e..4d5ec1d5 100644
--- a/klippy/extras/tmc_uart.py
+++ b/klippy/extras/tmc_uart.py
@@ -210,7 +210,7 @@ def lookup_tmc_uart_bitbang(config, max_addr):
 
 # Helper code for communicating via TMC uart
 class MCU_TMC_uart:
-    def __init__(self, config, name_to_reg, fields, max_addr=0):
+    def __init__(self, config, name_to_reg, fields, max_addr, tmc_frequency):
         self.printer = config.get_printer()
         self.name = config.get_name().split()[-1]
         self.name_to_reg = name_to_reg
@@ -219,6 +219,7 @@ class MCU_TMC_uart:
         self.instance_id, self.addr, self.mcu_uart = lookup_tmc_uart_bitbang(
             config, max_addr)
         self.mutex = self.mcu_uart.mutex
+        self.tmc_frequency = tmc_frequency
     def get_fields(self):
         return self.fields
     def _do_get_register(self, reg_name):
@@ -250,3 +251,5 @@ class MCU_TMC_uart:
                     return
         raise self.printer.command_error(
             "Unable to write tmc uart '%s' register %s" % (self.name, reg_name))
+    def get_tmc_frequency(self):
+        return self.tmc_frequency