238 lines
11 KiB
Python

import re
import logging
import gi
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
class Panel(ScreenPanel):
distances = ['.1', '.5', '1', '5', '10', '25', '50']
distance = distances[-2]
def __init__(self, screen, title):
super().__init__(screen, title)
if self.ks_printer_cfg is not None:
dis = self.ks_printer_cfg.get("move_distances", '')
if re.match(r'^[0-9,\.\s]+$', dis):
dis = [str(i.strip()) for i in dis.split(',')]
if 1 < len(dis) <= 7:
self.distances = dis
self.distance = self.distances[-2]
self.settings = {}
self.menu = ['move_menu']
self.buttons = {
'x+': self._gtk.Button("arrow-right", "X+", "color1"),
'x-': self._gtk.Button("arrow-left", "X-", "color1"),
'y+': self._gtk.Button("arrow-up", "Y+", "color2"),
'y-': self._gtk.Button("arrow-down", "Y-", "color2"),
'z+': self._gtk.Button("z-farther", "Z+", "color3"),
'z-': self._gtk.Button("z-closer", "Z-", "color3"),
'home': self._gtk.Button("home", _("Home"), "color4"),
'motors_off': self._gtk.Button("motor-off", _("Disable Motors"), "color4"),
}
self.buttons['x+'].connect("clicked", self.move, "X", "+")
self.buttons['x-'].connect("clicked", self.move, "X", "-")
self.buttons['y+'].connect("clicked", self.move, "Y", "+")
self.buttons['y-'].connect("clicked", self.move, "Y", "-")
self.buttons['z+'].connect("clicked", self.move, "Z", "+")
self.buttons['z-'].connect("clicked", self.move, "Z", "-")
self.buttons['home'].connect("clicked", self.home)
script = {"script": "M18"}
self.buttons['motors_off'].connect("clicked", self._screen._confirm_send_action,
_("Are you sure you wish to disable motors?"),
"printer.gcode.script", script)
adjust = self._gtk.Button("settings", None, "color2", 1, Gtk.PositionType.LEFT, 1)
adjust.connect("clicked", self.load_menu, 'options', _('Settings'))
adjust.set_hexpand(False)
grid = Gtk.Grid(row_homogeneous=True, column_homogeneous=True)
if self._screen.vertical_mode:
if self._screen.lang_ltr:
grid.attach(self.buttons['x+'], 2, 1, 1, 1)
grid.attach(self.buttons['x-'], 0, 1, 1, 1)
grid.attach(self.buttons['z+'], 2, 2, 1, 1)
grid.attach(self.buttons['z-'], 0, 2, 1, 1)
else:
grid.attach(self.buttons['x+'], 0, 1, 1, 1)
grid.attach(self.buttons['x-'], 2, 1, 1, 1)
grid.attach(self.buttons['z+'], 0, 2, 1, 1)
grid.attach(self.buttons['z-'], 2, 2, 1, 1)
grid.attach(adjust, 1, 2, 1, 1)
grid.attach(self.buttons['y+'], 1, 0, 1, 1)
grid.attach(self.buttons['y-'], 1, 1, 1, 1)
else:
if self._screen.lang_ltr:
grid.attach(self.buttons['x+'], 2, 1, 1, 1)
grid.attach(self.buttons['x-'], 0, 1, 1, 1)
else:
grid.attach(self.buttons['x+'], 0, 1, 1, 1)
grid.attach(self.buttons['x-'], 2, 1, 1, 1)
grid.attach(self.buttons['y+'], 1, 0, 1, 1)
grid.attach(self.buttons['y-'], 1, 1, 1, 1)
grid.attach(self.buttons['z+'], 3, 0, 1, 1)
grid.attach(self.buttons['z-'], 3, 1, 1, 1)
grid.attach(self.buttons['home'], 0, 0, 1, 1)
grid.attach(self.buttons['motors_off'], 2, 0, 1, 1)
distgrid = Gtk.Grid()
for j, i in enumerate(self.distances):
self.labels[i] = self._gtk.Button(label=i)
self.labels[i].set_direction(Gtk.TextDirection.LTR)
self.labels[i].connect("clicked", self.change_distance, i)
ctx = self.labels[i].get_style_context()
ctx.add_class("horizontal_togglebuttons")
if i == self.distance:
ctx.add_class("horizontal_togglebuttons_active")
distgrid.attach(self.labels[i], j, 0, 1, 1)
for p in ('pos_x', 'pos_y', 'pos_z'):
self.labels[p] = Gtk.Label()
self.labels['move_dist'] = Gtk.Label(label=_("Move Distance (mm)"))
bottomgrid = Gtk.Grid(row_homogeneous=True, column_homogeneous=True)
bottomgrid.set_direction(Gtk.TextDirection.LTR)
bottomgrid.attach(self.labels['pos_x'], 0, 0, 1, 1)
bottomgrid.attach(self.labels['pos_y'], 1, 0, 1, 1)
bottomgrid.attach(self.labels['pos_z'], 2, 0, 1, 1)
bottomgrid.attach(self.labels['move_dist'], 0, 1, 3, 1)
if not self._screen.vertical_mode:
bottomgrid.attach(adjust, 3, 0, 1, 2)
self.labels['move_menu'] = Gtk.Grid(row_homogeneous=True, column_homogeneous=True)
self.labels['move_menu'].attach(grid, 0, 0, 1, 3)
self.labels['move_menu'].attach(bottomgrid, 0, 3, 1, 1)
self.labels['move_menu'].attach(distgrid, 0, 4, 1, 1)
self.content.add(self.labels['move_menu'])
printer_cfg = self._printer.get_config_section("printer")
# The max_velocity parameter is not optional in klipper config.
# The minimum is 1, but least 2 values are needed to create a scale
max_velocity = max(int(float(printer_cfg["max_velocity"])), 2)
if "max_z_velocity" in printer_cfg:
max_z_velocity = max(int(float(printer_cfg["max_z_velocity"])), 2)
else:
max_z_velocity = max_velocity
configurable_options = [
{"invert_x": {"section": "main", "name": _("Invert X"), "type": "binary", "value": "False"}},
{"invert_y": {"section": "main", "name": _("Invert Y"), "type": "binary", "value": "False"}},
{"invert_z": {"section": "main", "name": _("Invert Z"), "type": "binary", "value": "False"}},
{"move_speed_xy": {
"section": "main", "name": _("XY Speed (mm/s)"), "type": "scale", "value": "50",
"range": [1, max_velocity], "step": 1}},
{"move_speed_z": {
"section": "main", "name": _("Z Speed (mm/s)"), "type": "scale", "value": "10",
"range": [1, max_z_velocity], "step": 1}}
]
self.labels['options_menu'] = self._gtk.ScrolledWindow()
self.labels['options'] = Gtk.Grid()
self.labels['options_menu'].add(self.labels['options'])
self.options = {}
for option in configurable_options:
name = list(option)[0]
self.options.update(self.add_option('options', self.settings, name, option[name]))
def process_update(self, action, data):
if action != "notify_status_update":
return
if "toolhead" in data and "max_velocity" in data["toolhead"]:
max_vel = max(int(float(data["toolhead"]["max_velocity"])), 2)
adj = self.options['move_speed_xy'].get_adjustment()
adj.set_upper(max_vel)
if "gcode_move" in data or "toolhead" in data and "homed_axes" in data["toolhead"]:
homed_axes = self._printer.get_stat("toolhead", "homed_axes")
for i, axis in enumerate(('x', 'y', 'z')):
if axis not in homed_axes:
self.labels[f"pos_{axis}"].set_text(f"{axis.upper()}: ?")
elif "gcode_move" in data and "gcode_position" in data["gcode_move"]:
self.labels[f"pos_{axis}"].set_text(
f"{axis.upper()}: {data['gcode_move']['gcode_position'][i]:.2f}")
def change_distance(self, widget, distance):
logging.info(f"### Distance {distance}")
self.labels[f"{self.distance}"].get_style_context().remove_class("horizontal_togglebuttons_active")
self.labels[f"{distance}"].get_style_context().add_class("horizontal_togglebuttons_active")
self.distance = distance
def move(self, widget, axis, direction):
if self._config.get_config()['main'].getboolean(f"invert_{axis.lower()}", False):
direction = "-" if direction == "+" else "+"
dist = f"{direction}{self.distance}"
config_key = "move_speed_z" if axis == "Z" else "move_speed_xy"
speed = None if self.ks_printer_cfg is None else self.ks_printer_cfg.getint(config_key, None)
if speed is None:
speed = self._config.get_config()['main'].getint(config_key, 20)
speed = 60 * max(1, speed)
script = f"{KlippyGcodes.MOVE_RELATIVE}\nG0 {axis}{dist} F{speed}"
self._screen._send_action(widget, "printer.gcode.script", {"script": script})
if self._printer.get_stat("gcode_move", "absolute_coordinates"):
self._screen._ws.klippy.gcode_script("G90")
def add_option(self, boxname, opt_array, opt_name, option):
name = Gtk.Label(hexpand=True, vexpand=True, halign=Gtk.Align.START, valign=Gtk.Align.CENTER, wrap=True)
name.set_markup(f"<big><b>{option['name']}</b></big>")
name.set_line_wrap_mode(Pango.WrapMode.WORD_CHAR)
dev = Gtk.Box(spacing=5,
hexpand=True, vexpand=False, valign=Gtk.Align.CENTER)
dev.get_style_context().add_class("frame-item")
dev.add(name)
setting = {}
if option['type'] == "binary":
box = Gtk.Box(hexpand=False)
switch = Gtk.Switch(hexpand=False, vexpand=False,
width_request=round(self._gtk.font_size * 7),
height_request=round(self._gtk.font_size * 3.5),
active=self._config.get_config().getboolean(option['section'], opt_name))
switch.connect("notify::active", self.switch_config_option, option['section'], opt_name)
setting = {opt_name: switch}
box.add(switch)
dev.add(box)
elif option['type'] == "scale":
dev.set_orientation(Gtk.Orientation.VERTICAL)
scale = Gtk.Scale.new_with_range(Gtk.Orientation.HORIZONTAL,
min=option['range'][0], max=option['range'][1], step=option['step'])
scale.set_hexpand(True)
scale.set_value(int(self._config.get_config().get(option['section'], opt_name, fallback=option['value'])))
scale.set_digits(0)
scale.connect("button-release-event", self.scale_moved, option['section'], opt_name)
setting = {opt_name: scale}
dev.add(scale)
opt_array[opt_name] = {
"name": option['name'],
"row": dev
}
opts = sorted(list(opt_array), key=lambda x: opt_array[x]['name'])
pos = opts.index(opt_name)
self.labels[boxname].insert_row(pos)
self.labels[boxname].attach(opt_array[opt_name]['row'], 0, pos, 1, 1)
self.labels[boxname].show_all()
return setting
def back(self):
if len(self.menu) > 1:
self.unload_menu()
return True
return False
def home(self, widget):
if "delta" in self._printer.get_config_section("printer")['kinematics']:
self._screen._send_action(widget, "printer.gcode.script", {"script": 'G28'})
return
name = "homing"
disname = self._screen._config.get_menu_name("move", name)
menuitems = self._screen._config.get_menu_items("move", name)
self._screen.show_panel("menu", disname, items=menuitems)