tmc: Support reporting tmc2240 temperature in driver get_status()

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
This commit is contained in:
Kevin O'Connor 2023-03-28 16:34:58 -04:00
parent 22ee889f41
commit 83308a1051
2 changed files with 22 additions and 5 deletions

View File

@ -458,6 +458,9 @@ objects (eg, `[tmc2208 stepper_x]`):
- `drv_status`: The results of the last driver status query. (Only - `drv_status`: The results of the last driver status query. (Only
non-zero fields are reported.) This field will be null if the driver non-zero fields are reported.) This field will be null if the driver
is not enabled (and thus is not periodically queried). is not enabled (and thus is not periodically queried).
- `temperature`: The internal temperature reported by the driver. This
field will be null if the driver is not enabled or if the driver
does not support temperature reporting.
- `run_current`: The currently set run current. - `run_current`: The currently set run current.
- `hold_current`: The currently set hold current. - `hold_current`: The currently set hold current.

View File

@ -93,7 +93,7 @@ class TMCErrorCheck:
self.mcu_tmc = mcu_tmc self.mcu_tmc = mcu_tmc
self.fields = mcu_tmc.get_fields() self.fields = mcu_tmc.get_fields()
self.check_timer = None self.check_timer = None
self.last_drv_status = self.last_status = None self.last_drv_status = self.last_drv_fields = None
# Setup for GSTAT query # Setup for GSTAT query
reg_name = self.fields.lookup_register("drv_err") reg_name = self.fields.lookup_register("drv_err")
if reg_name is not None: if reg_name is not None:
@ -122,6 +122,9 @@ class TMCErrorCheck:
if f in err_fields: if f in err_fields:
err_mask |= self.fields.all_fields[reg_name][f] err_mask |= self.fields.all_fields[reg_name][f]
self.drv_status_reg_info = [0, reg_name, mask, err_mask, cs_actual_mask] self.drv_status_reg_info = [0, reg_name, mask, err_mask, cs_actual_mask]
# Setup for temperature query
self.adc_temp = None
self.adc_temp_reg = self.fields.lookup_register("adc_temp")
def _query_register(self, reg_info, try_clear=False): def _query_register(self, reg_info, try_clear=False):
last_value, reg_name, mask, err_mask, cs_actual_mask = reg_info last_value, reg_name, mask, err_mask, cs_actual_mask = reg_info
cleared_flags = 0 cleared_flags = 0
@ -161,11 +164,20 @@ class TMCErrorCheck:
cleared_flags |= val & err_mask cleared_flags |= val & err_mask
self.mcu_tmc.set_register(reg_name, val & err_mask) self.mcu_tmc.set_register(reg_name, val & err_mask)
return cleared_flags return cleared_flags
def _query_temperature(self):
try:
self.adc_temp = self.mcu_tmc.get_register(self.adc_temp_reg)
except self.printer.command_error as e:
# Ignore comms error for temperature
self.adc_temp = None
return
def _do_periodic_check(self, eventtime): def _do_periodic_check(self, eventtime):
try: try:
self._query_register(self.drv_status_reg_info) self._query_register(self.drv_status_reg_info)
if self.gstat_reg_info is not None: if self.gstat_reg_info is not None:
self._query_register(self.gstat_reg_info) self._query_register(self.gstat_reg_info)
if self.adc_temp_reg is not None:
self._query_temperature()
except self.printer.command_error as e: except self.printer.command_error as e:
self.printer.invoke_shutdown(str(e)) self.printer.invoke_shutdown(str(e))
return self.printer.get_reactor().NEVER return self.printer.get_reactor().NEVER
@ -194,14 +206,16 @@ class TMCErrorCheck:
return False return False
def get_status(self, eventtime=None): def get_status(self, eventtime=None):
if self.check_timer is None: if self.check_timer is None:
return {'drv_status': None} return {'drv_status': None, 'temperature': None}
temp = None
if self.adc_temp is not None:
temp = round((self.adc_temp - 2038) / 7.7, 2)
last_value, reg_name = self.drv_status_reg_info[:2] last_value, reg_name = self.drv_status_reg_info[:2]
if last_value != self.last_drv_status: if last_value != self.last_drv_status:
self.last_drv_status = last_value self.last_drv_status = last_value
fields = self.fields.get_reg_fields(reg_name, last_value) fields = self.fields.get_reg_fields(reg_name, last_value)
fields = {n: v for n, v in fields.items() if v} self.last_drv_fields = {n: v for n, v in fields.items() if v}
self.last_status = {'drv_status': fields} return {'drv_status': self.last_drv_fields, 'temperature': temp}
return self.last_status
###################################################################### ######################################################################