diff options
Diffstat (limited to 'klippy/extras')
| -rw-r--r-- | klippy/extras/force_move.py | 5 | ||||
| -rw-r--r-- | klippy/extras/stepper_enable.py | 8 |
2 files changed, 6 insertions, 7 deletions
diff --git a/klippy/extras/force_move.py b/klippy/extras/force_move.py index ec13f6ed..d57e60b1 100644 --- a/klippy/extras/force_move.py +++ b/klippy/extras/force_move.py @@ -49,9 +49,8 @@ class ForceMove: gcode.register_command('SET_KINEMATIC_POSITION', self.cmd_SET_KINEMATIC_POSITION, desc=self.cmd_SET_KINEMATIC_POSITION_help) - def register_stepper(self, stepper): - name = stepper.get_name() - self.steppers[name] = stepper + def register_stepper(self, config, mcu_stepper): + self.steppers[mcu_stepper.get_name()] = mcu_stepper def lookup_stepper(self, name): if name not in self.steppers: raise self.printer.config_error("Unknown stepper %s" % (name,)) diff --git a/klippy/extras/stepper_enable.py b/klippy/extras/stepper_enable.py index cd1d34c6..e8b41c0a 100644 --- a/klippy/extras/stepper_enable.py +++ b/klippy/extras/stepper_enable.py @@ -84,10 +84,10 @@ class PrinterStepperEnable: gcode.register_command("SET_STEPPER_ENABLE", self.cmd_SET_STEPPER_ENABLE, desc=self.cmd_SET_STEPPER_ENABLE_help) - def register_stepper(self, stepper, pin): - name = stepper.get_name() - enable = setup_enable_pin(self.printer, pin) - self.enable_lines[name] = EnableTracking(stepper, enable) + def register_stepper(self, config, mcu_stepper): + name = mcu_stepper.get_name() + enable = setup_enable_pin(self.printer, config.get('enable_pin', None)) + self.enable_lines[name] = EnableTracking(mcu_stepper, enable) def motor_off(self): toolhead = self.printer.lookup_object('toolhead') toolhead.dwell(DISABLE_STALL_TIME) |