stepper_enable: refactor to mux commands

Signed-off-by: Timofey Titovets <nefelim4ag@gmail.com>
This commit is contained in:
Timofey Titovets 2025-12-02 22:00:36 +01:00
parent 906900e9cd
commit 92a4263e4f

View file

@ -81,11 +81,12 @@ class PrinterStepperEnable:
gcode = self.printer.lookup_object('gcode')
gcode.register_command("M18", self.cmd_M18)
gcode.register_command("M84", self.cmd_M18)
gcode.register_command("SET_STEPPER_ENABLE",
self.cmd_SET_STEPPER_ENABLE,
desc=self.cmd_SET_STEPPER_ENABLE_help)
def register_stepper(self, config, mcu_stepper):
name = mcu_stepper.get_name()
gcode = self.printer.lookup_object('gcode')
gcode.register_mux_command('SET_STEPPER_ENABLE', "STEPPER", name,
self.cmd_SET_STEPPER_ENABLE,
desc=self.cmd_SET_STEPPER_ENABLE_help)
enable = setup_enable_pin(self.printer, config.get('enable_pin', None))
self.enable_lines[name] = EnableTracking(mcu_stepper, enable)
def set_motors_enable(self, stepper_names, enable):
@ -128,11 +129,7 @@ class PrinterStepperEnable:
self.motor_off()
cmd_SET_STEPPER_ENABLE_help = "Enable/disable individual stepper by name"
def cmd_SET_STEPPER_ENABLE(self, gcmd):
stepper_name = gcmd.get('STEPPER', None)
if stepper_name not in self.enable_lines:
gcmd.respond_info('SET_STEPPER_ENABLE: Invalid stepper "%s"'
% (stepper_name,))
return
stepper_name = gcmd.get('STEPPER')
stepper_enable = gcmd.get_int('ENABLE', 1)
self.set_motors_enable([stepper_name], stepper_enable)
if stepper_enable: