diff options
| author | Kevin O'Connor <kevin@koconnor.net> | 2018-07-13 17:14:33 -0400 |
|---|---|---|
| committer | Kevin O'Connor <kevin@koconnor.net> | 2018-07-13 17:14:33 -0400 |
| commit | 2af73d1006befdb9ffbdb43aae3a32b356caea72 (patch) | |
| tree | 41b40004e0ea2c1bf49666a904f9f4f5958a6b04 | |
| parent | a31c31aed43a6852880ec3e47f0621c811b0e2d0 (diff) | |
| download | kutter-2af73d1006befdb9ffbdb43aae3a32b356caea72.tar.gz kutter-2af73d1006befdb9ffbdb43aae3a32b356caea72.tar.xz kutter-2af73d1006befdb9ffbdb43aae3a32b356caea72.zip | |
probe: Simplify mcu_probe stepper registration
Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
| -rw-r--r-- | klippy/extras/probe.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/klippy/extras/probe.py b/klippy/extras/probe.py index 26eaa932..64bddedc 100644 --- a/klippy/extras/probe.py +++ b/klippy/extras/probe.py @@ -43,12 +43,10 @@ class PrinterProbe: self.gcode.register_command( 'QUERY_PROBE', self.cmd_QUERY_PROBE, desc=self.cmd_QUERY_PROBE_help) def build_config(self): - toolhead = self.printer.lookup_object('toolhead') - z_rails = toolhead.get_kinematics().get_rails("Z") - for rail in z_rails: - for mcu_endstop, name in rail.get_endstops(): - for mcu_stepper in mcu_endstop.get_steppers(): - self.mcu_probe.add_stepper(mcu_stepper) + kin = self.printer.lookup_object('toolhead').get_kinematics() + for rail in kin.get_rails("Z"): + for stepper in rail.get_steppers(): + stepper.add_to_endstop(self.mcu_probe) def setup_pin(self, pin_params): if (pin_params['pin'] != 'z_virtual_endstop' or pin_params['type'] != 'endstop'): |