201 lines
6.8 KiB
Python
201 lines
6.8 KiB
Python
import logging
|
|
|
|
import gi
|
|
|
|
gi.require_version("Gtk", "3.0")
|
|
from gi.repository import Gtk, GLib, Pango
|
|
|
|
from ks_includes.screen_panel import ScreenPanel
|
|
|
|
|
|
def create_panel(*args):
|
|
return MacroPanel(*args)
|
|
|
|
|
|
class MacroPanel(ScreenPanel):
|
|
def __init__(self, screen, title, back=True):
|
|
super().__init__(screen, title, back)
|
|
self.sort_reverse = False
|
|
self.sort_lbl = _("Name")
|
|
self.sort_btn = self._gtk.ButtonImage("arrow-up", self.sort_lbl, "color1", .5, Gtk.PositionType.RIGHT, 1)
|
|
self.sort_btn.connect("clicked", self.change_sort)
|
|
self.sort_btn.set_hexpand(True)
|
|
self.allmacros = {}
|
|
self.loaded_macros = []
|
|
self.macros = {}
|
|
self.menu = ['macros_menu']
|
|
sort = Gtk.Label(_("Sort:"))
|
|
sort.set_hexpand(False)
|
|
|
|
adjust = self._gtk.ButtonImage("settings", None, "color2", 1, Gtk.PositionType.LEFT, 1)
|
|
adjust.connect("clicked", self.load_menu, 'options')
|
|
adjust.set_hexpand(False)
|
|
|
|
sbox = Gtk.Box()
|
|
sbox.set_vexpand(False)
|
|
sbox.pack_start(sort, False, False, 5)
|
|
sbox.pack_start(self.sort_btn, True, True, 5)
|
|
sbox.pack_start(adjust, True, True, 5)
|
|
|
|
self.labels['macros_list'] = self._gtk.ScrolledWindow()
|
|
self.labels['macros'] = Gtk.Grid()
|
|
self.labels['macros_list'].add(self.labels['macros'])
|
|
|
|
self.labels['macros_menu'] = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
|
|
self.labels['macros_menu'].set_vexpand(True)
|
|
self.labels['macros_menu'].pack_start(sbox, False, False, 0)
|
|
self.labels['macros_menu'].pack_start(self.labels['macros_list'], True, True, 0)
|
|
|
|
self.content.add(self.labels['macros_menu'])
|
|
self.labels['options_menu'] = self._gtk.ScrolledWindow()
|
|
self.labels['options'] = Gtk.Grid()
|
|
self.labels['options_menu'].add(self.labels['options'])
|
|
|
|
def activate(self):
|
|
while len(self.menu) > 1:
|
|
self.unload_menu()
|
|
self.reload_macros()
|
|
|
|
def add_gcode_macro(self, macro):
|
|
# Support for hiding macros by name
|
|
if macro.startswith("_"):
|
|
return
|
|
|
|
name = Gtk.Label()
|
|
name.set_markup(f"<big><b>{macro}</b></big>")
|
|
name.set_hexpand(True)
|
|
name.set_vexpand(True)
|
|
name.set_halign(Gtk.Align.START)
|
|
name.set_valign(Gtk.Align.CENTER)
|
|
name.set_line_wrap(True)
|
|
name.set_line_wrap_mode(Pango.WrapMode.WORD_CHAR)
|
|
|
|
btn = self._gtk.ButtonImage("resume", style="color3")
|
|
btn.connect("clicked", self.run_gcode_macro, macro)
|
|
btn.set_hexpand(False)
|
|
btn.set_halign(Gtk.Align.END)
|
|
|
|
labels = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
|
|
labels.add(name)
|
|
|
|
dev = Gtk.Box(spacing=5)
|
|
dev.add(labels)
|
|
dev.add(btn)
|
|
|
|
frame = Gtk.Frame()
|
|
frame.get_style_context().add_class("frame-item")
|
|
frame.add(dev)
|
|
|
|
self.macros[macro] = {
|
|
"row": frame
|
|
}
|
|
|
|
macros = sorted(self.macros, reverse=self.sort_reverse, key=str.casefold)
|
|
pos = macros.index(macro)
|
|
|
|
self.loaded_macros.append(macro)
|
|
self.labels['macros'].insert_row(pos)
|
|
self.labels['macros'].attach(self.macros[macro]['row'], 0, pos, 1, 1)
|
|
|
|
def run_gcode_macro(self, widget, macro):
|
|
self._screen._ws.klippy.gcode_script(macro)
|
|
|
|
def change_sort(self, widget):
|
|
self.sort_reverse ^= True
|
|
if self.sort_reverse:
|
|
self.sort_btn.set_image(self._gtk.Image("arrow-down", self._gtk.img_scale * .5))
|
|
else:
|
|
self.sort_btn.set_image(self._gtk.Image("arrow-up", self._gtk.img_scale * .5))
|
|
self.sort_btn.show()
|
|
|
|
GLib.idle_add(self.reload_macros)
|
|
|
|
def reload_macros(self):
|
|
self.labels['macros'].remove_column(0)
|
|
self.macros = {}
|
|
self.loaded_macros = []
|
|
self.allmacros = {}
|
|
self.labels['options'].remove_column(0)
|
|
self.load_gcode_macros()
|
|
|
|
def load_gcode_macros(self):
|
|
macros = self._screen.printer.get_gcode_macros()
|
|
section_name = f"displayed_macros {self._screen.connected_printer}"
|
|
logging.info(f"Macro section name [{section_name}]")
|
|
|
|
for x in macros:
|
|
macro = x[12:].strip()
|
|
|
|
if macro in self.loaded_macros:
|
|
continue
|
|
|
|
if (section_name not in self._config.get_config().sections() or
|
|
self._config.get_config().getboolean(section_name, macro.lower(), fallback=True)):
|
|
self.add_gcode_macro(macro)
|
|
|
|
for macro in self._printer.get_config_section_list("gcode_macro "):
|
|
macro = macro[12:]
|
|
# Support for hiding macros by name
|
|
if macro.startswith("_"):
|
|
continue
|
|
|
|
self.allmacros[macro] = {
|
|
"name": macro,
|
|
"section": f"displayed_macros {self._screen.connected_printer}",
|
|
}
|
|
for macro in list(self.allmacros):
|
|
self.add_option('options', self.allmacros, macro, self.allmacros[macro])
|
|
|
|
self.labels['macros'].show_all()
|
|
|
|
def add_option(self, boxname, opt_array, opt_name, option):
|
|
name = Gtk.Label()
|
|
name.set_markup(f"<big><b>{option['name']}</b></big>")
|
|
name.set_hexpand(True)
|
|
name.set_vexpand(True)
|
|
name.set_halign(Gtk.Align.START)
|
|
name.set_valign(Gtk.Align.CENTER)
|
|
name.set_line_wrap(True)
|
|
name.set_line_wrap_mode(Pango.WrapMode.WORD_CHAR)
|
|
|
|
box = Gtk.Box()
|
|
box.set_vexpand(False)
|
|
switch = Gtk.Switch()
|
|
switch.set_hexpand(False)
|
|
switch.set_vexpand(False)
|
|
switch.set_active(self._config.get_config().getboolean(option['section'], opt_name, fallback=True))
|
|
switch.connect("notify::active", self.switch_config_option, option['section'], opt_name)
|
|
switch.set_property("width-request", round(self._gtk.get_font_size() * 7))
|
|
switch.set_property("height-request", round(self._gtk.get_font_size() * 3.5))
|
|
box.add(switch)
|
|
|
|
dev = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=5)
|
|
dev.set_hexpand(True)
|
|
dev.set_vexpand(False)
|
|
dev.set_valign(Gtk.Align.CENTER)
|
|
dev.add(name)
|
|
dev.add(box)
|
|
|
|
frame = Gtk.Frame()
|
|
frame.get_style_context().add_class("frame-item")
|
|
frame.add(dev)
|
|
frame.show_all()
|
|
opt_array[opt_name] = {
|
|
"name": option['name'],
|
|
"row": frame
|
|
}
|
|
|
|
opts = sorted(self.allmacros, key=str.casefold)
|
|
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()
|
|
|
|
def back(self):
|
|
if len(self.menu) > 1:
|
|
self.unload_menu()
|
|
self.reload_macros()
|
|
return True
|
|
return False
|