diff options
| -rw-r--r-- | klippy/extras/bltouch.py | 5 | ||||
| -rw-r--r-- | klippy/extras/probe.py | 5 |
2 files changed, 6 insertions, 4 deletions
diff --git a/klippy/extras/bltouch.py b/klippy/extras/bltouch.py index 6415ce1a..3dc46044 100644 --- a/klippy/extras/bltouch.py +++ b/klippy/extras/bltouch.py @@ -28,6 +28,8 @@ class BLTouchEndstopWrapper: self.printer = config.get_printer() self.printer.register_event_handler("klippy:connect", self.handle_connect) + self.printer.register_event_handler('klippy:mcu_identify', + self.handle_mcu_identify) self.position_endstop = config.getfloat('z_offset') self.stow_on_each_sample = config.getboolean('stow_on_each_sample', True) @@ -45,7 +47,6 @@ class BLTouchEndstopWrapper: pin = config.get('sensor_pin') pin_params = ppins.lookup_pin(pin, can_invert=True, can_pullup=True) mcu = pin_params['chip'] - mcu.register_config_callback(self._build_config) self.mcu_endstop = mcu.setup_pin('endstop', pin_params) # output mode omodes = {'5V': '5V', 'OD': 'OD', None: None} @@ -72,7 +73,7 @@ class BLTouchEndstopWrapper: desc=self.cmd_BLTOUCH_STORE_help) # multi probes state self.multi = 'OFF' - def _build_config(self): + def handle_mcu_identify(self): kin = self.printer.lookup_object('toolhead').get_kinematics() for stepper in kin.get_steppers(): if stepper.is_active_axis('z'): diff --git a/klippy/extras/probe.py b/klippy/extras/probe.py index a8fc36a6..700a8955 100644 --- a/klippy/extras/probe.py +++ b/klippy/extras/probe.py @@ -280,8 +280,9 @@ class ProbeEndstopWrapper: pin = config.get('pin') pin_params = ppins.lookup_pin(pin, can_invert=True, can_pullup=True) mcu = pin_params['chip'] - mcu.register_config_callback(self._build_config) self.mcu_endstop = mcu.setup_pin('endstop', pin_params) + self.printer.register_event_handler('klippy:mcu_identify', + self._handle_mcu_identify) # Wrappers self.get_mcu = self.mcu_endstop.get_mcu self.add_stepper = self.mcu_endstop.add_stepper @@ -291,7 +292,7 @@ class ProbeEndstopWrapper: self.query_endstop = self.mcu_endstop.query_endstop # multi probes state self.multi = 'OFF' - def _build_config(self): + def _handle_mcu_identify(self): kin = self.printer.lookup_object('toolhead').get_kinematics() for stepper in kin.get_steppers(): if stepper.is_active_axis('z'): |