mirror of
				https://github.com/Klipper3d/klipper.git
				synced 2025-11-03 20:05:49 +01:00 
			
		
		
		
	Add ability to clear the MCU shutdown flag from the console. Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
		
			
				
	
	
		
			390 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			390 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# Parse gcode commands
 | 
						|
#
 | 
						|
# Copyright (C) 2016  Kevin O'Connor <kevin@koconnor.net>
 | 
						|
#
 | 
						|
# This file may be distributed under the terms of the GNU GPLv3 license.
 | 
						|
import os, re, logging, collections
 | 
						|
import homing
 | 
						|
 | 
						|
# Parse out incoming GCode and find and translate head movements
 | 
						|
class GCodeParser:
 | 
						|
    RETRY_TIME = 0.100
 | 
						|
    def __init__(self, printer, fd, is_fileinput=False):
 | 
						|
        self.printer = printer
 | 
						|
        self.fd = fd
 | 
						|
        self.is_fileinput = is_fileinput
 | 
						|
        # Input handling
 | 
						|
        self.reactor = printer.reactor
 | 
						|
        self.fd_handle = None
 | 
						|
        if not is_fileinput:
 | 
						|
            self.fd_handle = self.reactor.register_fd(self.fd, self.process_data)
 | 
						|
        self.input_commands = [""]
 | 
						|
        self.bytes_read = 0
 | 
						|
        self.input_log = collections.deque([], 50)
 | 
						|
        # Busy handling
 | 
						|
        self.busy_timer = self.reactor.register_timer(self.busy_handler)
 | 
						|
        self.busy_state = None
 | 
						|
        # Command handling
 | 
						|
        self.gcode_handlers = {}
 | 
						|
        self.is_printer_ready = False
 | 
						|
        self.need_ack = False
 | 
						|
        self.toolhead = self.heater_nozzle = self.heater_bed = self.fan = None
 | 
						|
        self.speed = 1.0
 | 
						|
        self.absolutecoord = self.absoluteextrude = True
 | 
						|
        self.base_position = [0.0, 0.0, 0.0, 0.0]
 | 
						|
        self.last_position = [0.0, 0.0, 0.0, 0.0]
 | 
						|
        self.homing_add = [0.0, 0.0, 0.0, 0.0]
 | 
						|
        self.axis2pos = {'X': 0, 'Y': 1, 'Z': 2, 'E': 3}
 | 
						|
        self.build_handlers()
 | 
						|
    def build_config(self):
 | 
						|
        self.toolhead = self.printer.objects['toolhead']
 | 
						|
        self.heater_nozzle = None
 | 
						|
        extruder = self.printer.objects.get('extruder')
 | 
						|
        if extruder:
 | 
						|
            self.heater_nozzle = extruder.heater
 | 
						|
        self.heater_bed = self.printer.objects.get('heater_bed')
 | 
						|
        self.fan = self.printer.objects.get('fan')
 | 
						|
    def build_handlers(self):
 | 
						|
        handlers = ['G1', 'G4', 'G20', 'G21', 'G28', 'G90', 'G91', 'G92',
 | 
						|
                    'M18', 'M82', 'M83', 'M105', 'M110', 'M114', 'M206',
 | 
						|
                    'HELP', 'QUERY_ENDSTOPS', 'RESTART', 'CLEAR_SHUTDOWN']
 | 
						|
        if self.heater_nozzle is not None:
 | 
						|
            handlers.extend(['M104', 'M109', 'PID_TUNE'])
 | 
						|
        if self.heater_bed is not None:
 | 
						|
            handlers.extend(['M140', 'M190'])
 | 
						|
        if self.fan is not None:
 | 
						|
            handlers.extend(['M106', 'M107'])
 | 
						|
        if not self.is_printer_ready:
 | 
						|
            handlers = [h for h in handlers
 | 
						|
                        if getattr(self, 'cmd_'+h+'_when_not_ready', False)]
 | 
						|
        self.gcode_handlers = dict((h, getattr(self, 'cmd_'+h))
 | 
						|
                                   for h in handlers)
 | 
						|
        for h, f in self.gcode_handlers.items():
 | 
						|
            aliases = getattr(self, 'cmd_'+h+'_aliases', [])
 | 
						|
            self.gcode_handlers.update(dict([(a, f) for a in aliases]))
 | 
						|
    def stats(self, eventtime):
 | 
						|
        return "gcodein=%d" % (self.bytes_read,)
 | 
						|
    def set_printer_ready(self, is_ready):
 | 
						|
        if self.is_printer_ready == is_ready:
 | 
						|
            return
 | 
						|
        self.is_printer_ready = is_ready
 | 
						|
        self.build_handlers()
 | 
						|
        if is_ready and self.is_fileinput and self.fd_handle is None:
 | 
						|
            self.fd_handle = self.reactor.register_fd(self.fd, self.process_data)
 | 
						|
    def motor_heater_off(self):
 | 
						|
        if self.toolhead is not None:
 | 
						|
            self.toolhead.motor_off()
 | 
						|
        if self.heater_nozzle is not None:
 | 
						|
            self.heater_nozzle.set_temp(0., 0.)
 | 
						|
        if self.heater_bed is not None:
 | 
						|
            self.heater_bed.set_temp(0., 0.)
 | 
						|
    def dump_debug(self):
 | 
						|
        logging.info("Dumping gcode input %d blocks" % (
 | 
						|
            len(self.input_log),))
 | 
						|
        for eventtime, data in self.input_log:
 | 
						|
            logging.info("Read %f: %s" % (eventtime, repr(data)))
 | 
						|
    # Parse input into commands
 | 
						|
    args_r = re.compile('([a-zA-Z_]+|[a-zA-Z*])')
 | 
						|
    def process_commands(self, eventtime):
 | 
						|
        i = -1
 | 
						|
        for i in range(len(self.input_commands)-1):
 | 
						|
            line = self.input_commands[i]
 | 
						|
            # Ignore comments and leading/trailing spaces
 | 
						|
            line = origline = line.strip()
 | 
						|
            cpos = line.find(';')
 | 
						|
            if cpos >= 0:
 | 
						|
                line = line[:cpos]
 | 
						|
            # Break command into parts
 | 
						|
            parts = self.args_r.split(line)[1:]
 | 
						|
            params = dict((parts[i].upper(), parts[i+1].strip())
 | 
						|
                          for i in range(0, len(parts), 2))
 | 
						|
            params['#original'] = origline
 | 
						|
            if parts and parts[0].upper() == 'N':
 | 
						|
                # Skip line number at start of command
 | 
						|
                del parts[:2]
 | 
						|
            if not parts:
 | 
						|
                self.cmd_default(params)
 | 
						|
                continue
 | 
						|
            params['#command'] = cmd = parts[0].upper() + parts[1].strip()
 | 
						|
            # Invoke handler for command
 | 
						|
            self.need_ack = True
 | 
						|
            handler = self.gcode_handlers.get(cmd, self.cmd_default)
 | 
						|
            try:
 | 
						|
                handler(params)
 | 
						|
            except:
 | 
						|
                logging.exception("Exception in command handler")
 | 
						|
                self.toolhead.force_shutdown()
 | 
						|
                self.respond_error('Internal error on command:"%s"' % (cmd,))
 | 
						|
            # Check if machine can process next command or must stall input
 | 
						|
            if self.busy_state is not None:
 | 
						|
                break
 | 
						|
            if self.is_printer_ready and self.toolhead.check_busy(eventtime):
 | 
						|
                self.set_busy(self.toolhead)
 | 
						|
                break
 | 
						|
            self.ack()
 | 
						|
        del self.input_commands[:i+1]
 | 
						|
    def process_data(self, eventtime):
 | 
						|
        if self.busy_state is not None:
 | 
						|
            self.reactor.unregister_fd(self.fd_handle)
 | 
						|
            self.fd_handle = None
 | 
						|
            return
 | 
						|
        data = os.read(self.fd, 4096)
 | 
						|
        self.input_log.append((eventtime, data))
 | 
						|
        self.bytes_read += len(data)
 | 
						|
        lines = data.split('\n')
 | 
						|
        lines[0] = self.input_commands[0] + lines[0]
 | 
						|
        self.input_commands = lines
 | 
						|
        self.process_commands(eventtime)
 | 
						|
        if not data and self.is_fileinput:
 | 
						|
            self.motor_heater_off()
 | 
						|
            self.printer.request_exit_eof()
 | 
						|
    # Response handling
 | 
						|
    def ack(self, msg=None):
 | 
						|
        if not self.need_ack or self.is_fileinput:
 | 
						|
            return
 | 
						|
        if msg:
 | 
						|
            os.write(self.fd, "ok %s\n" % (msg,))
 | 
						|
        else:
 | 
						|
            os.write(self.fd, "ok\n")
 | 
						|
        self.need_ack = False
 | 
						|
    def respond(self, msg):
 | 
						|
        logging.debug(msg)
 | 
						|
        if self.is_fileinput:
 | 
						|
            return
 | 
						|
        os.write(self.fd, msg+"\n")
 | 
						|
    def respond_error(self, msg):
 | 
						|
        lines = msg.strip().split('\n')
 | 
						|
        for line in lines[:-1]:
 | 
						|
            self.respond('// %s' % (line.strip(),))
 | 
						|
        self.respond('!! %s' % (lines[-1].strip(),))
 | 
						|
    # Busy handling
 | 
						|
    def set_busy(self, busy_handler):
 | 
						|
        self.busy_state = busy_handler
 | 
						|
        self.reactor.update_timer(self.busy_timer, self.reactor.NOW)
 | 
						|
    def busy_handler(self, eventtime):
 | 
						|
        try:
 | 
						|
            busy = self.busy_state.check_busy(eventtime)
 | 
						|
        except homing.EndstopError, e:
 | 
						|
            self.respond_error(str(e))
 | 
						|
            busy = False
 | 
						|
        except:
 | 
						|
            logging.exception("Exception in busy handler")
 | 
						|
            self.toolhead.force_shutdown()
 | 
						|
            self.respond_error('Internal error in busy handler')
 | 
						|
            busy = False
 | 
						|
        if busy:
 | 
						|
            self.toolhead.reset_motor_off_time(eventtime)
 | 
						|
            return eventtime + self.RETRY_TIME
 | 
						|
        self.busy_state = None
 | 
						|
        self.ack()
 | 
						|
        self.process_commands(eventtime)
 | 
						|
        if self.busy_state is not None:
 | 
						|
            return self.reactor.NOW
 | 
						|
        if self.fd_handle is None:
 | 
						|
            self.fd_handle = self.reactor.register_fd(self.fd, self.process_data)
 | 
						|
        return self.reactor.NEVER
 | 
						|
    # Temperature wrappers
 | 
						|
    def get_temp(self):
 | 
						|
        if not self.is_printer_ready:
 | 
						|
            return "T:0"
 | 
						|
        # T:XXX /YYY B:XXX /YYY
 | 
						|
        out = []
 | 
						|
        if self.heater_nozzle:
 | 
						|
            cur, target = self.heater_nozzle.get_temp()
 | 
						|
            out.append("T:%.1f /%.1f" % (cur, target))
 | 
						|
        if self.heater_bed:
 | 
						|
            cur, target = self.heater_bed.get_temp()
 | 
						|
            out.append("B:%.1f /%.1f" % (cur, target))
 | 
						|
        return " ".join(out)
 | 
						|
    def bg_temp(self, heater):
 | 
						|
        # Wrapper class for check_busy() that periodically prints current temp
 | 
						|
        class temp_busy_handler_wrapper:
 | 
						|
            gcode = self
 | 
						|
            last_temp_time = 0.
 | 
						|
            cur_heater = heater
 | 
						|
            def check_busy(self, eventtime):
 | 
						|
                if eventtime > self.last_temp_time + 1.0:
 | 
						|
                    self.gcode.respond(self.gcode.get_temp())
 | 
						|
                    self.last_temp_time = eventtime
 | 
						|
                return self.cur_heater.check_busy(eventtime)
 | 
						|
        if self.is_fileinput:
 | 
						|
            return
 | 
						|
        self.set_busy(temp_busy_handler_wrapper())
 | 
						|
    def set_temp(self, heater, params, wait=False):
 | 
						|
        print_time = self.toolhead.get_last_move_time()
 | 
						|
        temp = float(params.get('S', '0'))
 | 
						|
        heater.set_temp(print_time, temp)
 | 
						|
        if wait:
 | 
						|
            self.bg_temp(heater)
 | 
						|
    # Individual command handlers
 | 
						|
    def cmd_default(self, params):
 | 
						|
        if not self.is_printer_ready:
 | 
						|
            self.respond_error(self.printer.get_state_message())
 | 
						|
            return
 | 
						|
        cmd = params.get('#command')
 | 
						|
        if not cmd:
 | 
						|
            logging.debug(params['#original'])
 | 
						|
            return
 | 
						|
        self.respond('echo:Unknown command:"%s"' % (cmd,))
 | 
						|
    cmd_G1_aliases = ['G0']
 | 
						|
    def cmd_G1(self, params):
 | 
						|
        # Move
 | 
						|
        for a, p in self.axis2pos.items():
 | 
						|
            if a in params:
 | 
						|
                v = float(params[a])
 | 
						|
                if not self.absolutecoord or (p>2 and not self.absoluteextrude):
 | 
						|
                    # value relative to position of last move
 | 
						|
                    self.last_position[p] += v
 | 
						|
                else:
 | 
						|
                    # value relative to base coordinate position
 | 
						|
                    self.last_position[p] = v + self.base_position[p]
 | 
						|
        if 'F' in params:
 | 
						|
            self.speed = float(params['F']) / 60.
 | 
						|
        try:
 | 
						|
            self.toolhead.move(self.last_position, self.speed)
 | 
						|
        except homing.EndstopError, e:
 | 
						|
            self.respond_error(str(e))
 | 
						|
            self.last_position = self.toolhead.get_position()
 | 
						|
    def cmd_G4(self, params):
 | 
						|
        # Dwell
 | 
						|
        if 'S' in params:
 | 
						|
            delay = float(params['S'])
 | 
						|
        else:
 | 
						|
            delay = float(params.get('P', '0')) / 1000.
 | 
						|
        self.toolhead.dwell(delay)
 | 
						|
    def cmd_G20(self, params):
 | 
						|
        # Set units to inches
 | 
						|
        self.respond_error('Machine does not support G20 (inches) command')
 | 
						|
    def cmd_G21(self, params):
 | 
						|
        # Set units to millimeters
 | 
						|
        pass
 | 
						|
    def cmd_G28(self, params):
 | 
						|
        # Move to origin
 | 
						|
        axes = []
 | 
						|
        for axis in 'XYZ':
 | 
						|
            if axis in params:
 | 
						|
                axes.append(self.axis2pos[axis])
 | 
						|
        if not axes:
 | 
						|
            axes = [0, 1, 2]
 | 
						|
        homing_state = homing.Homing(self.toolhead, axes)
 | 
						|
        if self.is_fileinput:
 | 
						|
            homing_state.set_no_verify_retract()
 | 
						|
        self.toolhead.home(homing_state)
 | 
						|
        def axes_update(homing_state):
 | 
						|
            newpos = self.toolhead.get_position()
 | 
						|
            for axis in homing_state.get_axes():
 | 
						|
                self.last_position[axis] = newpos[axis]
 | 
						|
                self.base_position[axis] = -self.homing_add[axis]
 | 
						|
        homing_state.plan_axes_update(axes_update)
 | 
						|
        self.set_busy(homing_state)
 | 
						|
    def cmd_G90(self, params):
 | 
						|
        # Use absolute coordinates
 | 
						|
        self.absolutecoord = True
 | 
						|
    def cmd_G91(self, params):
 | 
						|
        # Use relative coordinates
 | 
						|
        self.absolutecoord = False
 | 
						|
    def cmd_G92(self, params):
 | 
						|
        # Set position
 | 
						|
        mcount = 0
 | 
						|
        for a, p in self.axis2pos.items():
 | 
						|
            if a in params:
 | 
						|
                self.base_position[p] = self.last_position[p] - float(params[a])
 | 
						|
                mcount += 1
 | 
						|
        if not mcount:
 | 
						|
            self.base_position = list(self.last_position)
 | 
						|
    def cmd_M82(self, params):
 | 
						|
        # Use absolute distances for extrusion
 | 
						|
        self.absoluteextrude = True
 | 
						|
    def cmd_M83(self, params):
 | 
						|
        # Use relative distances for extrusion
 | 
						|
        self.absoluteextrude = False
 | 
						|
    cmd_M18_aliases = ["M84"]
 | 
						|
    def cmd_M18(self, params):
 | 
						|
        # Turn off motors
 | 
						|
        self.toolhead.motor_off()
 | 
						|
    cmd_M105_when_not_ready = True
 | 
						|
    def cmd_M105(self, params):
 | 
						|
        # Get Extruder Temperature
 | 
						|
        self.ack(self.get_temp())
 | 
						|
    def cmd_M104(self, params):
 | 
						|
        # Set Extruder Temperature
 | 
						|
        self.set_temp(self.heater_nozzle, params)
 | 
						|
    def cmd_M109(self, params):
 | 
						|
        # Set Extruder Temperature and Wait
 | 
						|
        self.set_temp(self.heater_nozzle, params, wait=True)
 | 
						|
    cmd_M110_when_not_ready = True
 | 
						|
    def cmd_M110(self, params):
 | 
						|
        # Set Current Line Number
 | 
						|
        pass
 | 
						|
    cmd_M114_when_not_ready = True
 | 
						|
    def cmd_M114(self, params):
 | 
						|
        # Get Current Position
 | 
						|
        if self.toolhead is None:
 | 
						|
            self.cmd_default(params)
 | 
						|
            return
 | 
						|
        kinpos = self.toolhead.get_position()
 | 
						|
        self.respond("X:%.3f Y:%.3f Z:%.3f E:%.3f Count X:%.3f Y:%.3f Z:%.3f" % (
 | 
						|
            self.last_position[0], self.last_position[1],
 | 
						|
            self.last_position[2], self.last_position[3],
 | 
						|
            kinpos[0], kinpos[1], kinpos[2]))
 | 
						|
    def cmd_M140(self, params):
 | 
						|
        # Set Bed Temperature
 | 
						|
        self.set_temp(self.heater_bed, params)
 | 
						|
    def cmd_M190(self, params):
 | 
						|
        # Set Bed Temperature and Wait
 | 
						|
        self.set_temp(self.heater_bed, params, wait=True)
 | 
						|
    def cmd_M106(self, params):
 | 
						|
        # Set fan speed
 | 
						|
        print_time = self.toolhead.get_last_move_time()
 | 
						|
        self.fan.set_speed(print_time, float(params.get('S', '255')) / 255.)
 | 
						|
    def cmd_M107(self, params):
 | 
						|
        # Turn fan off
 | 
						|
        print_time = self.toolhead.get_last_move_time()
 | 
						|
        self.fan.set_speed(print_time, 0)
 | 
						|
    def cmd_M206(self, params):
 | 
						|
        # Set home offset
 | 
						|
        for a, p in self.axis2pos.items():
 | 
						|
            if a in params:
 | 
						|
                v = float(params[a])
 | 
						|
                self.base_position[p] += self.homing_add[p] - v
 | 
						|
                self.homing_add[p] = v
 | 
						|
    cmd_QUERY_ENDSTOPS_help = "Report on the status of each endstop"
 | 
						|
    cmd_QUERY_ENDSTOPS_aliases = ["M119"]
 | 
						|
    def cmd_QUERY_ENDSTOPS(self, params):
 | 
						|
        # Get Endstop Status
 | 
						|
        if self.is_fileinput:
 | 
						|
            return
 | 
						|
        print_time = self.toolhead.get_last_move_time()
 | 
						|
        query_state = homing.QueryEndstops(print_time, self.respond)
 | 
						|
        self.toolhead.query_endstops(query_state)
 | 
						|
        self.set_busy(query_state)
 | 
						|
    cmd_PID_TUNE_help = "Run PID Tuning"
 | 
						|
    cmd_PID_TUNE_aliases = ["M303"]
 | 
						|
    def cmd_PID_TUNE(self, params):
 | 
						|
        # Run PID tuning
 | 
						|
        heater = int(params.get('E', '0'))
 | 
						|
        heater = {0: self.heater_nozzle, -1: self.heater_bed}[heater]
 | 
						|
        temp = float(params.get('S', '60'))
 | 
						|
        heater.start_auto_tune(temp)
 | 
						|
        self.bg_temp(heater)
 | 
						|
    cmd_CLEAR_SHUTDOWN_when_not_ready = True
 | 
						|
    cmd_CLEAR_SHUTDOWN_help = "Clear firmware shutdown and restart"
 | 
						|
    def cmd_CLEAR_SHUTDOWN(self, params):
 | 
						|
        if self.toolhead is None:
 | 
						|
            self.cmd_default(params)
 | 
						|
            return
 | 
						|
        self.printer.mcu.clear_shutdown()
 | 
						|
        self.printer.request_restart()
 | 
						|
    cmd_RESTART_when_not_ready = True
 | 
						|
    cmd_RESTART_help = "Reload config file and restart host software"
 | 
						|
    def cmd_RESTART(self, params):
 | 
						|
        self.printer.request_restart()
 | 
						|
    cmd_HELP_when_not_ready = True
 | 
						|
    def cmd_HELP(self, params):
 | 
						|
        cmdhelp = ["// Available extended commands:"]
 | 
						|
        for cmd in self.gcode_handlers:
 | 
						|
            desc = getattr(self, 'cmd_'+cmd+'_help', None)
 | 
						|
            if desc is not None:
 | 
						|
                cmdhelp.append("%-10s: %s" % (cmd, desc))
 | 
						|
        self.respond("\n// ".join(cmdhelp))
 |