diff options
Diffstat (limited to 'klippy/extras/tmc2208.py')
| -rw-r--r-- | klippy/extras/tmc2208.py | 18 |
1 files changed, 6 insertions, 12 deletions
diff --git a/klippy/extras/tmc2208.py b/klippy/extras/tmc2208.py index 6ea0c056..4bcdabd8 100644 --- a/klippy/extras/tmc2208.py +++ b/klippy/extras/tmc2208.py @@ -189,7 +189,7 @@ class TMC2208: self.mcu_tmc = tmc_uart.MCU_TMC_uart(config, Registers, self.fields) # Register commands cmdhelper = tmc.TMCCommandHelper(config, self.mcu_tmc) - cmdhelper.setup_register_dump(self.query_registers) + cmdhelper.setup_register_dump(ReadRegisters, self.read_translate) # Setup basic register values self.fields.set_field("pdn_disable", True) self.fields.set_field("mstep_reg_select", True) @@ -215,17 +215,11 @@ class TMC2208: set_config_field(config, "pwm_autograd", True) set_config_field(config, "PWM_REG", 8) set_config_field(config, "PWM_LIM", 12) - def query_registers(self, print_time=0.): - out = [] - for reg_name in ReadRegisters: - val = self.mcu_tmc.get_register(reg_name) - # IOIN has different mappings depending on the driver type - # (SEL_A field of IOIN reg) - if reg_name == "IOIN": - drv_type = self.fields.get_field("SEL_A", val) - reg_name = "IOIN@TMC220x" if drv_type else "IOIN@TMC222x" - out.append((reg_name, val)) - return out + def read_translate(self, reg_name, val): + if reg_name == "IOIN": + drv_type = self.fields.get_field("SEL_A", val) + reg_name = "IOIN@TMC220x" if drv_type else "IOIN@TMC222x" + return reg_name, val def load_config_prefix(config): return TMC2208(config) |