tmc2208: Use field definitions during driver init

Reduce the amount of bit manipulations by using the FieldHelper class.

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 7 insertions, 2 deletions

diff --git a/klippy/extras/tmc2130.py b/klippy/extras/tmc2130.py
index a49d55df..061e9271 100644
--- a/klippy/extras/tmc2130.py
+++ b/klippy/extras/tmc2130.py
@@ -45,13 +45,18 @@ class FieldHelper:
     def __init__(self, all_fields, field_formatters={}):
         self.all_fields = all_fields
         self.field_formatters = field_formatters
+        self.field_to_register = { f: r for r, fields in self.all_fields.items()
+                                   for f in fields }
+    def lookup_register(self, field_name):
+        # Return the name of the register containing the given field
+        return self.field_to_register[field_name]
     def get_field(self, reg_name, field_name, reg_value):
         # Returns value of the register field
-        mask = self.all_fields.get(reg_name, {})[field_name]
+        mask = self.all_fields[reg_name][field_name]
         return (reg_value & mask) >> ffs(mask)
     def set_field(self, reg_name, field_name, reg_value, field_value):
         # Returns register value with field bits filled with supplied value
-        mask = self.all_fields.get(reg_name, {})[field_name]
+        mask = self.all_fields[reg_name][field_name]
         return (reg_value & ~mask) | ((field_value << ffs(mask)) & mask)
     def pretty_format(self, reg_name, value):
         # Provide a string description of a register