import gi import logging import re gi.require_version("Gtk", "3.0") from gi.repository import Gtk, Pango from ks_includes.KlippyGcodes import KlippyGcodes from ks_includes.screen_panel import ScreenPanel def create_panel(*args): return BedLevelPanel(*args) class BedLevelPanel(ScreenPanel): def __init__(self, screen, title, back=True): super().__init__(screen, title, back) self.response_count = 0 self.panel_name = "" self.screw_dict = {} self.screws = [] self.y_cnt = 0 self.x_cnt = 0 self.x_offset = 0 self.y_offset = 0 def initialize(self, panel_name): self.panel_name = panel_name self.labels['dm'] = self._gtk.ButtonImage("motor-off", _("Disable XY"), "color3") self.labels['dm'].connect("clicked", self.disable_motors) screw_positions = [] rotation = None grid = self._gtk.HomogeneousGrid() grid.attach(self.labels['dm'], 0, 0, 1, 1) if "screws_tilt_adjust" in self._screen.printer.get_config_section_list(): self.labels['screws'] = self._gtk.ButtonImage("refresh", _("Screws Adjust"), "color4") self.labels['screws'].connect("clicked", self.screws_tilt_calculate) grid.attach(self.labels['screws'], 0, 1, 1, 1) self.screws = self._get_screws("screws_tilt_adjust") logging.info(f"screws_tilt_adjust: {self.screws}") if "bltouch" in self._screen.printer.get_config_section_list(): self._get_offsets("bltouch") elif "probe" in self._screen.printer.get_config_section_list(): self._get_offsets("probe") new_screws = [] # bed_screws uses NOZZLE positions # screws_tilt_adjust uses PROBE positions and to be offseted for the buttons to work equal to bed_screws for screw in self.screws: new_screws.append([round(screw[0] + self.x_offset, 1), round(screw[1] + self.y_offset, 1)]) self.screws = new_screws logging.info(f"screws with offset: {self.screws}") elif "bed_screws" in self._screen.printer.get_config_section_list(): self.screws = self._get_screws("bed_screws") logging.info(f"bed_screws: {self.screws}") # get dimensions x_positions = {x[0] for x in self.screws} y_positions = {y[1] for y in self.screws} logging.info(f"X: {x_positions}\nY: {y_positions}") self.x_cnt = len(x_positions) self.y_cnt = len(y_positions) min_x = min(x_positions) max_x = max(x_positions) min_y = min(y_positions) max_y = max(y_positions) fl = [min_x, min_y] bl = [min_x, max_y] br = [max_x, max_y] fr = [max_x, min_y] if self.x_cnt == 3: mid_x = [x for x in list(zip(*self.screws))[0] if x not in (min_x, max_x)][0] fm = [mid_x, min_y] bm = [mid_x, max_y] else: fm = bm = None if self.y_cnt == 3: mid_y = [y for y in list(zip(*self.screws))[1] if y not in (min_y, max_y)][0] lm = [min_x, mid_y] rm = [max_x, mid_y] else: lm = rm = None logging.debug(f"Using {len(self.screws)}-screw locations [x,y] [{self.x_cnt}x{self.y_cnt}]") self.labels['bl'] = self._gtk.ButtonImage("bed-level-t-l", scale=2.5) self.labels['br'] = self._gtk.ButtonImage("bed-level-t-r", scale=2.5) self.labels['fl'] = self._gtk.ButtonImage("bed-level-b-l", scale=2.5) self.labels['fr'] = self._gtk.ButtonImage("bed-level-b-r", scale=2.5) self.labels['lm'] = self._gtk.ButtonImage("bed-level-l-m", scale=2.5) self.labels['rm'] = self._gtk.ButtonImage("bed-level-r-m", scale=2.5) self.labels['fm'] = self._gtk.ButtonImage("bed-level-b-m", scale=2.5) self.labels['bm'] = self._gtk.ButtonImage("bed-level-t-m", scale=2.5) valid_positions = True printer_cfg = self._config.get_printer_config(self._screen.connected_printer) if printer_cfg is not None: logging.info(f"printer {printer_cfg}") screw_positions = printer_cfg.get("screw_positions", "") screw_positions = [str(i.strip()) for i in screw_positions.split(',')] logging.info(f"Positions: {screw_positions}") for screw in screw_positions: if screw not in ("bl", "fl", "fr", "br", "bm", "fm", "lm", "rm", ""): logging.error(f"Unknown screw: {screw}") self._screen.show_popup_message(_("Unknown screw position") + ": %s" % screw) valid_positions = False if not(3 <= len(screw_positions) <= 8): valid_positions = False rotation = printer_cfg.getint("screw_rotation", 0) logging.info(f"Rotation: {rotation}") else: valid_positions = False if 'bed_screws' in self._config.get_config(): rotation = self._config.get_config()['bed_screws'].getint("rotation", 0) logging.debug(f"Rotation: {rotation}") bedgrid = Gtk.Grid() nscrews = len(self.screws) if valid_positions: if "bl" in screw_positions: bedgrid.attach(self.labels['bl'], 1, 0, 1, 1) if "fl" in screw_positions: bedgrid.attach(self.labels['fl'], 1, 2, 1, 1) if "fr" in screw_positions: bedgrid.attach(self.labels['fr'], 3, 2, 1, 1) if "br" in screw_positions: bedgrid.attach(self.labels['br'], 3, 0, 1, 1) if "bm" in screw_positions: bedgrid.attach(self.labels['bm'], 2, 0, 1, 1) if "fm" in screw_positions: bedgrid.attach(self.labels['fm'], 2, 2, 1, 1) if "lm" in screw_positions: bedgrid.attach(self.labels['lm'], 1, 1, 1, 1) if "rm" in screw_positions: bedgrid.attach(self.labels['rm'], 3, 1, 1, 1) elif nscrews in (4, 6, 8): bedgrid.attach(self.labels['bl'], 1, 0, 1, 1) bedgrid.attach(self.labels['fl'], 1, 2, 1, 1) bedgrid.attach(self.labels['fr'], 3, 2, 1, 1) bedgrid.attach(self.labels['br'], 3, 0, 1, 1) if self.x_cnt == 3: bedgrid.attach(self.labels['bm'], 2, 0, 1, 1) bedgrid.attach(self.labels['fm'], 2, 2, 1, 1) if self.y_cnt == 3: bedgrid.attach(self.labels['lm'], 1, 1, 1, 1) bedgrid.attach(self.labels['rm'], 3, 1, 1, 1) else: label = Gtk.Label( _("Bed screw configuration:") + f" {nscrews}\n\n" + _("Not supported for auto-detection, it needs to be configured in klipperscreen.conf") ) label.set_line_wrap_mode(Pango.WrapMode.WORD_CHAR) label.set_line_wrap(True) grid.attach(label, 1, 0, 3, 2) self.content.add(grid) return if rotation == 90: # fl lm bl # fm bm # fr rm br self.labels['bl'].connect("clicked", self.go_to_position, fl) self.labels['bm'].connect("clicked", self.go_to_position, lm) self.labels['br'].connect("clicked", self.go_to_position, bl) self.labels['rm'].connect("clicked", self.go_to_position, bm) self.labels['fr'].connect("clicked", self.go_to_position, br) self.labels['fm'].connect("clicked", self.go_to_position, rm) self.labels['fl'].connect("clicked", self.go_to_position, fr) self.labels['lm'].connect("clicked", self.go_to_position, fm) self.screw_dict = { 'bl': fl, 'bm': lm, 'br': bl, 'rm': bm, 'fr': br, 'fm': rm, 'fl': fr, 'lm': fm } elif rotation == 180: # fr fm fl # rm lm # br bm bl self.labels['bl'].connect("clicked", self.go_to_position, fr) self.labels['bm'].connect("clicked", self.go_to_position, fm) self.labels['br'].connect("clicked", self.go_to_position, fl) self.labels['rm'].connect("clicked", self.go_to_position, lm) self.labels['fr'].connect("clicked", self.go_to_position, bl) self.labels['fm'].connect("clicked", self.go_to_position, bm) self.labels['fl'].connect("clicked", self.go_to_position, br) self.labels['lm'].connect("clicked", self.go_to_position, rm) self.screw_dict = { 'bl': fr, 'bm': fm, 'br': fl, 'rm': lm, 'fr': bl, 'fm': bm, 'fl': br, 'lm': rm } elif rotation == 270: # br rm fr # bm fm # bl lm fl self.labels['bl'].connect("clicked", self.go_to_position, br) self.labels['bm'].connect("clicked", self.go_to_position, rm) self.labels['br'].connect("clicked", self.go_to_position, fr) self.labels['rm'].connect("clicked", self.go_to_position, fm) self.labels['fr'].connect("clicked", self.go_to_position, fl) self.labels['fm'].connect("clicked", self.go_to_position, lm) self.labels['fl'].connect("clicked", self.go_to_position, bl) self.labels['lm'].connect("clicked", self.go_to_position, bm) self.screw_dict = { 'bl': br, 'bm': rm, 'br': fr, 'rm': fm, 'fr': fl, 'fm': lm, 'fl': bl, 'lm': bm } else: # bl bm br # lm rm # fl fm fr self.labels['bl'].connect("clicked", self.go_to_position, bl) self.labels['bm'].connect("clicked", self.go_to_position, bm) self.labels['br'].connect("clicked", self.go_to_position, br) self.labels['rm'].connect("clicked", self.go_to_position, rm) self.labels['fr'].connect("clicked", self.go_to_position, fr) self.labels['fm'].connect("clicked", self.go_to_position, fm) self.labels['fl'].connect("clicked", self.go_to_position, fl) self.labels['lm'].connect("clicked", self.go_to_position, lm) self.screw_dict = { 'bl': bl, 'bm': bm, 'br': br, 'rm': rm, 'fr': fr, 'fm': fm, 'fl': fl, 'lm': lm } grid.attach(bedgrid, 1, 0, 3, 2) self.content.add(grid) def activate(self): for key, value in self.screw_dict.items(): self.labels[key].set_label(f"{value}") if self._printer.config_section_exists("screws_tilt_adjust"): self.labels['screws'].set_sensitive(True) def go_to_position(self, widget, position): if self._screen.printer.get_stat("toolhead", "homed_axes") != "xyz": self._screen._ws.klippy.gcode_script(KlippyGcodes.HOME) logging.debug(f"Going to position: {position}") script = [ f"{KlippyGcodes.MOVE_ABSOLUTE}", "G1 Z7 F800\n", f"G1 X{position[0]} Y{position[1]} F3600\n", "G1 Z.1 F300\n" ] self._screen._ws.klippy.gcode_script( "\n".join(script) ) def disable_motors(self, widget): self._screen._ws.klippy.gcode_script( "M18" # Disable motors ) def process_update(self, action, data): if action != "notify_gcode_response": return if data.startswith('!!'): self.response_count = 0 self.labels['screws'].set_sensitive(True) return result = re.match( "^// (.*) : [xX= ]+([\\-0-9\\.]+), [yY= ]+([\\-0-9\\.]+), [zZ= ]+[\\-0-9\\.]+ :" + " (Adjust ->|adjust) ([CW]+ [0-9:]+)", data ) # screws_tilt_adjust uses PROBE positions and was offseted for the buttons to work equal to bed_screws # for the result we need to undo the offset if result: x = round(float(result[2]) + self.x_offset, 1) y = round(float(result[3]) + self.y_offset, 1) for key, value in self.screw_dict.items(): if value and x == value[0] and y == value[1]: logging.debug(f"X: {x} Y: {y} Adjust: {result[5]} Pos: {key}") self.labels[key].set_label(result[5]) break self.response_count += 1 if self.response_count >= len(self.screws) - 1: self.labels['screws'].set_sensitive(True) else: result = re.match( "^// (.*) : [xX= ]+([\\-0-9\\.]+), [yY= ]+([\\-0-9\\.]+), [zZ= ]+[\\-0-9\\.]", data ) # screws_tilt_adjust uses PROBE positions and was offseted for the buttons to work equal to bed_screws # for the result we need to undo the offset if result and re.search('base', result[1]): x = round(float(result[2]) + self.x_offset, 1) y = round(float(result[3]) + self.y_offset, 1) logging.debug(f"X: {x} Y: {y} is the reference") for key, value in self.screw_dict.items(): if value and x == value[0] and y == value[1]: logging.debug(f"X: {x} Y: {y} Pos: {key}") self.labels[key].set_label(_("Reference")) def _get_screws(self, config_section_name): screws = [] config_section = self._screen.printer.get_config_section(config_section_name) for item in config_section: logging.debug(f"{config_section_name}: {config_section[item]}") result = re.match(r"([\-0-9\.]+)\s*,\s*([\-0-9\.]+)", config_section[item]) if result: screws.append([ round(float(result[1]), 1), round(float(result[2]), 1) ]) return sorted(screws, key=lambda s: (float(s[1]), float(s[0]))) def _get_offsets(self, section): probe = self._screen.printer.get_config_section(section) if "x_offset" in probe: self.x_offset = round(float(probe['x_offset']), 1) if "y_offset" in probe: self.y_offset = round(float(probe['y_offset']), 1) logging.debug(f"{section} offset X: {self.x_offset} Y: {self.y_offset}") def screws_tilt_calculate(self, widget): if self._screen.printer.get_stat("toolhead", "homed_axes") != "xyz": self._screen._ws.klippy.gcode_script(KlippyGcodes.HOME) self.response_count = 0 self.labels['screws'].set_sensitive(False) self._screen._ws.klippy.gcode_script("SCREWS_TILT_CALCULATE")